SegmentInZ0TanLambdaBox.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/eventdata/segments/CDCSegment3D.h>
#include <tracking/trackFindingCDC/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;
 
      Weight operator()(const std::pair<CDCSegment3D, 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;
      }
    };
  }
}