HitInZ0TanLambdaBoxUsingZ.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun, Oliver Frost                                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/hough/boxes/Z0TanLambdaBox.h>
 
#include <tracking/trackFindingCDC/eventdata/hits/CDCRecoHit3D.h>
 
#include <tracking/trackFindingCDC/topology/CDCWire.h>
#include <tracking/trackFindingCDC/topology/WireLine.h>
 
#include <tracking/trackFindingCDC/numerics/Weight.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class HitInZ0TanLambdaBoxUsingZ {
    public:
 
      using HoughBox = Z0TanLambdaBox;
 
      Weight operator()(const CDCRecoHit3D& recoHit,
                        const HoughBox* z0TanLambdaBox)
      {
        const CDCWire& wire = recoHit.getWire();
        const WireLine& wireLine = wire.getWireLine();
 
        float lowerZ0 = z0TanLambdaBox->getLowerZ0();
        float upperZ0 = z0TanLambdaBox->getUpperZ0();
        const float centerZ0 = 0.5 * (lowerZ0 + upperZ0);
 
        float lowerTanLambda = z0TanLambdaBox->getLowerTanLambda();
        float upperTanLambda = z0TanLambdaBox->getUpperTanLambda();
        const float centerTanLambda = 0.5 * (lowerTanLambda + upperTanLambda);
 
        float perpS = recoHit.getArcLength2D();
        const Vector2D& recoPosition = recoHit.getRecoPos2D();
 
        float hitZ = centerTanLambda * perpS + centerZ0;
 
        Vector2D pos2D = wireLine.nominalPos2DAtZ(hitZ);
 
        float distanceToRecoPosition = (pos2D - recoPosition).norm();
 
        return exp(-distanceToRecoPosition);
      }
 
      static bool compareDistances(const HoughBox& z0TanLambdaBox, const CDCRecoHit3D& lhsRecoHit, const CDCRecoHit3D& rhsRecoHit)
      {
        const double z0Mean = (z0TanLambdaBox.getLowerZ0() + z0TanLambdaBox.getUpperZ0()) / 2.0;
        const double tanLambdaMean = (z0TanLambdaBox.getLowerTanLambda() + z0TanLambdaBox.getUpperTanLambda()) / 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 = lhsS * tanLambdaMean + z0Mean - lhsZ;
        const double rhsZDistance = rhsS * tanLambdaMean + z0Mean - rhsZ;
 
        return lhsZDistance < rhsZDistance;
      }
 
      static const char* debugLine() { return "[0] + [1] * x";}
    };
  }
}