ECLTrackClusterMatchingModule.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 <TF1.h>
 
#include <ecl/dataobjects/ECLCalDigit.h>
#include <ecl/dataobjects/ECLShower.h>
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <tracking/dataobjects/ExtHit.h>
 
namespace Belle2 {
  class ECLTrackClusterMatchingParameterizations;
  class ECLTrackClusterMatchingThresholds;
 
  class ECLTrackClusterMatchingModule : public Module {
  public:
    ECLTrackClusterMatchingModule();
 
    virtual ~ECLTrackClusterMatchingModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
  private:
    bool isECLEnterHit(const ExtHit& extHit) const;
 
    bool isECLHit(const ExtHit& extHit) const;
 
    double showerQuality(double deltaPhi, double deltaTheta, double pt, int eclDetectorRegion, int hitStatus) const;
 
    double phiConsistency(double deltaPhi, double pt, int eclDetectorRegion, int hitStatus) const;
 
    double thetaConsistency(double deltaTheta, double pt, int eclDetectorRegion, int hitStatus) const;
 
    bool trackTowardsGap(double theta) const; 
    void optimizedPTMatchingConsistency(double theta, double pt); 
    StoreArray<ExtHit> m_extHits; 
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 
    StoreArray<ECLCluster> m_eclClusters; 
    StoreArray<ECLShower> m_eclShowers; 
    StoreArray<ECLCalDigit> m_eclCalDigits; 
    DBObjPtr<ECLTrackClusterMatchingParameterizations> m_matchingParameterizations; 
    TF1 f_phiRMSFWDCROSS; 
    TF1 f_phiRMSFWDDL; 
    TF1 f_phiRMSFWDNEAR; 
    TF1 f_phiRMSBRLCROSS; 
    TF1 f_phiRMSBRLDL; 
    TF1 f_phiRMSBRLNEAR; 
    TF1 f_phiRMSBWDCROSS; 
    TF1 f_phiRMSBWDDL; 
    TF1 f_phiRMSBWDNEAR; 
    TF1 f_thetaRMSFWDCROSS; 
    TF1 f_thetaRMSFWDDL; 
    TF1 f_thetaRMSFWDNEAR; 
    TF1 f_thetaRMSBRLCROSS; 
    TF1 f_thetaRMSBRLDL; 
    TF1 f_thetaRMSBRLNEAR; 
    TF1 f_thetaRMSBWDCROSS; 
    TF1 f_thetaRMSBWDDL; 
    TF1 f_thetaRMSBWDNEAR; 
    DBObjPtr<ECLTrackClusterMatchingThresholds> m_matchingThresholds; 
    std::vector<std::pair<double, double>> m_matchingThresholdValuesFWD; 
    std::vector<std::pair<double, std::pair<double, double>>> m_matchingThresholdValuesBRL; 
    std::vector<std::pair<double, double>> m_matchingThresholdValuesBWD; 
    bool m_angularDistanceMatching;
    bool m_useOptimizedMatchingConsistency; 
    bool m_skipZeroChargeTracks; 
    double m_matchingConsistency; 
    double m_matchingPTThreshold; 
    double m_brlEdgeTheta; 
    int m_minimalCDCHits; 
  };
} //Belle2