ROIFinder.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/trackFindingCDC/findlets/base/Findlet.h>
#include <tracking/dataobjects/ROIid.h>
#include <tracking/dataobjects/PXDIntercept.h>
#include <tracking/trackFindingVXD/trackQualityEstimators/QualityEstimatorBase.h>
#include <framework/datastore/StoreArray.h>
#include <framework/geometry/B2Vector3.h>
 
#include <string>
#include <vector>
 
namespace Belle2 {
  class ModuleParamList;
  class SpacePointTrackCand;
 
  namespace vxdHoughTracking {
 
    class ROIFinder : public TrackFindingCDC::Findlet<const SpacePointTrackCand> {
      using Super = TrackFindingCDC::Findlet<const SpacePointTrackCand>;
 
    public:
      ROIFinder();
 
      ~ROIFinder();
 
      void exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix) override;
 
      void apply(const std::vector<SpacePointTrackCand>& finalTracks) override;
 
      void initialize() override;
 
      void beginRun() override;
 
      void beginEvent() override;
 
    private:
      bool m_calculateROI = false;
 
      // StoreArrays
      std::string m_storePXDInterceptsName = "SVDHoughPXDIntercepts";
      std::string m_storeROIsName = "SVDHoughROIs";
      StoreArray<PXDIntercept> m_storePXDIntercepts;
      StoreArray<ROIid> m_storeROIs;
 
 
      bool m_refit = true;
      bool m_addVirtualIP = true;
      std::string m_ROIFitMethod = "helixFit";
 
      std::unique_ptr<QualityEstimatorBase> m_estimator;
 
 
      //  Extrapolation parameters
      double m_tolerancePhi = 0.15;
      double m_toleranceZ = 0.5;
 
      //  The extrapolation has two charge dependent biases:
      //  a) the residuals show a ~1/R bias, that is larger for low pT = small track radii
      //  b) the residuals show a larger sin(phi) and a smaller cos(phi modulation)
      //  Both of these can be corrected for
      double m_radiusCorrectionFactorL1 = -2.0;
      double m_radiusCorrectionFactorL2 = -5.0;
      double m_sinPhiCorrectionFactor = 0.0;
      double m_cosPhiCorrectionFactor = 0.0;
 
      double m_zPositionCorrectionFactor = 1.0;
 
 
      //  ROI calculation parameters
      double m_minimumROISizeUL1 = 40;
      double m_minimumROISizeVL1 = 40;
      double m_minimumROISizeUL2 = 35;
      double m_minimumROISizeVL2 = 30;
 
      double m_multiplierUL1 = 500;
      double m_multiplierUL2 = 600;
      double m_multiplierVL1 = 0.8;
      double m_multiplierVL2 = 0.8;
 
      unsigned short m_maximumROISizeU = 100;
      unsigned short m_maximumROISizeV = 100;
 
      //  Constants used during extrapolation to PXD and ROI calculation
      // ATTENTION: hard coded values taken and derived from pxd/data/PXD-Components.xml
      const double c_centerZShiftLayer1[2] = {3.68255, -0.88255};
      const double c_centerZShiftLayer2[2] = {5.01455, -1.21455};
      const double c_sensorMinY = -0.36;
      const double c_sensorMaxY =  0.89;
      const double c_activeSensorWidth = (c_sensorMaxY - c_sensorMinY);
      const double c_shiftY = (c_sensorMaxY + c_sensorMinY) / 2.0;
      const double c_layerRadius[2] = {1.42854, 2.21218};
      const double c_activeSensorLength[2] = {4.48, 6.144};
      const double c_ladderPhiL1[8] = {0., 0.25 * M_PI, M_PI_2, 0.75 * M_PI, M_PI, -0.75 * M_PI, -M_PI_2, -0.25 * M_PI};
      const double c_ladderPhiL2[12] = {0., 1. / 6. * M_PI, 1. / 3. * M_PI,  M_PI_2, 2. / 3. * M_PI, 5. / 6. * M_PI,  M_PI, -5. / 6. * M_PI, -2. / 3. * M_PI, -M_PI_2, -1. / 3. * M_PI, -1. / 6. * M_PI};
 
      double m_bFieldZ = 1.5;
 
 
      B2Vector3D m_BeamSpotPosition;
      B2Vector3D m_BeamSpotPositionError;
 
    };
 
  }
}