SegmentInZ0TanLambdaBox.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
#include <tracking/trackingUtilities/eventdata/segments/CDCSegment3D.h>
#include <tracking/trackingUtilities/eventdata/trajectories/CDCTrajectorySZ.h>
#include <tracking/trackFindingCDC/hough/boxes/Z0TanLambdaBox.h>
#include <tracking/trackFindingCDC/hough/baseelements/SameSignChecker.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {

    class SegmentInZ0TanLambdaBox {
    public:

      using HoughBox = Z0TanLambdaBox;

      TrackingUtilities::Weight operator()(const std::pair<TrackingUtilities::CDCSegment3D, TrackingUtilities::CDCTrajectorySZ>&
                                           segmentWithTrajectorySZ,
                                           const HoughBox* z0TanLambdaBox)
      {
        float lowerZ0 = z0TanLambdaBox->getLowerZ0();
        float upperZ0 = z0TanLambdaBox->getUpperZ0();
 
        float lowerTanLambda = z0TanLambdaBox->getLowerTanLambda();
        float upperTanLambda = z0TanLambdaBox->getUpperTanLambda();
 
        const CDCTrajectorySZ& szTrajectory = segmentWithTrajectorySZ.second;
        const CDCSegment3D& segment = segmentWithTrajectorySZ.first;
 
        float trajectoryZ0 = szTrajectory.getZ0();
        float trajectoryTanLambda = szTrajectory.getTanLambda();
 
        if (std::isnan(trajectoryZ0) or std::isnan(trajectoryTanLambda)) {
          return NAN;
        }
 
        if (SameSignChecker::isIn(trajectoryZ0, trajectoryTanLambda, lowerZ0, upperZ0, lowerTanLambda, upperTanLambda)) {
          return 0.6 * segment.size();
        }
 
        return NAN;
      }
    };
  }
}