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 <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_param_calculateROI = false;
 
      // StoreArrays
      std::string m_param_storePXDInterceptsName = "DATCONPXDIntercepts";
      std::string m_param_storeROIsName = "DATCONROIs";
      StoreArray<PXDIntercept> m_storePXDIntercepts;
      StoreArray<ROIid> m_storeROIs;
 
 
      //  Extrapolation parameters
      double m_param_tolerancePhi = 0.15;
      double m_param_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_param_radiusCorrectionFactor = 4.0;
      double m_param_sinPhiCorrectionFactor = 0.0;
      double m_param_cosPhiCorrectionFactor = 0.0;
 
 
      //  ROI calculation parameters
      double m_param_minimumROISizeUL1 = 40;
      double m_param_minimumROISizeVL1 = 40;
      double m_param_minimumROISizeUL2 = 40;
      double m_param_minimumROISizeVL2 = 40;
 
      double m_param_multiplierUL1 = 600;
      double m_param_multiplierUL2 = 800;
      double m_param_multiplierVL1 = 0.8;
      double m_param_multiplierVL2 = 1.2;
 
      unsigned short m_param_maximumROISizeU = 120;
      unsigned short m_param_maximumROISizeV = 192;
 
      //  Constants used during extrapolation to PXD and ROI calculation
      // ATTENTION: hard coded values taken and derived from pxd/data/PXD-Components.xml
      const double m_const_centerZShiftLayer1[2] = {3.68255, -0.88255};
      const double m_const_centerZShiftLayer2[2] = {5.01455, -1.21455};
      const double m_const_sensorMinY = -0.36;
      const double m_const_sensorMaxY =  0.89;
      const double m_const_activeSensorWidth = (m_const_sensorMaxY - m_const_sensorMinY);
      const double m_const_shiftY = (m_const_sensorMaxY + m_const_sensorMinY) / 2.0;
      const double m_const_layerRadius[2] = {1.42854, 2.21218};
      const double m_const_activeSensorLength[2] = {4.48, 6.144};
      const double m_const_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 m_const_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;
 
    };
 
  }
}