HitInQuadraticBox.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun, Oliver Frost, Dmitrii Neverov                *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
#include <tracking/trackFindingCDC/eventdata/hits/CDCRecoHit3D.h>
#include <tracking/trackFindingCDC/hough/boxes/PQBox.h>
#include <tracking/trackFindingCDC/hough/baseelements/SameSignChecker.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class HitInQuadraticBox {
    public:
 
      using HoughBox = PQBox;
 
      Weight operator()(const CDCRecoHit3D& recoHit,
                        const HoughBox* pqBox)
      {
        float lowerP = pqBox->getLowerP();
        float upperP = pqBox->getUpperP();
 
        float lowerQ = pqBox->getLowerQ();
        float upperQ = pqBox->getUpperQ();
 
        float perpS = recoHit.getArcLength2D();
        float reconstructedZ = recoHit.getRecoZ();
 
        float distLowerPLowerQ = (lowerP + 4 * lowerQ) * perpS - lowerQ / 25 * perpS * perpS - reconstructedZ;
        float distUpperPLowerQ = (upperP + 4 * lowerQ) * perpS - lowerQ / 25 * perpS * perpS - reconstructedZ;
        float distLowerPUpperQ = (lowerP + 4 * upperQ) * perpS - upperQ / 25 * perpS * perpS - reconstructedZ;
        float distUpperPUpperQ = (upperP + 4 * upperQ) * perpS - upperQ / 25 * perpS * perpS - reconstructedZ;
 
        const bool sameSign = SameSignChecker::sameSign(distLowerPLowerQ, distUpperPLowerQ,
                                                        distLowerPUpperQ, distUpperPUpperQ);
        if (not sameSign) {
          return 1.0;
        } else {
          return NAN;
        }
      }
 
      static bool compareDistances(const HoughBox& pqBox, const CDCRecoHit3D& lhsRecoHit, const CDCRecoHit3D& rhsRecoHit)
      {
        const double pMean = (pqBox.getLowerP() + pqBox.getUpperP()) / 2.0;
        const double qMean = (pqBox.getLowerQ() + pqBox.getUpperQ()) / 2.0;
 
        const double lhsZ = lhsRecoHit.getRecoZ();
        const double rhsZ = rhsRecoHit.getRecoZ();
 
        const double lhsS = lhsRecoHit.getArcLength2D();
        const double rhsS = rhsRecoHit.getArcLength2D();
 
        const double lhsZDistance = (pMean + 4 * qMean) * lhsS - qMean / 25 * lhsS * lhsS - lhsZ;
        const double rhsZDistance = (pMean + 4 * qMean) * rhsS - qMean / 25 * rhsS * rhsS - rhsZ;
 
        return lhsZDistance < rhsZDistance;
      }
 
      static const char* debugLine() { return "([0] + 4*[1])*x - [1] / 25 * x * x";}
    };
  }
}