TrackFinderMCTruthRecoTracksModule.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 <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/MCParticle.h>
#include <cdc/dataobjects/CDCHit.h>
#include <cdc/dataobjects/CDCSimHit.h>
#include <pxd/dataobjects/PXDTrueHit.h>
#include <pxd/dataobjects/PXDCluster.h>
#include <svd/dataobjects/SVDTrueHit.h>
#include <svd/dataobjects/SVDCluster.h>
#include <tracking/dataobjects/RecoTrack.h>
 
#include <TMatrixDSym.h>
#include <string>
 
namespace Belle2 {
  class TrackFinderMCTruthRecoTracksModule : public Module {
 
  public:
    TrackFinderMCTruthRecoTracksModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
    void endRun() override;
 
  private:
 
    template<class THit, class TSimHit>
    bool isWithinNLoops(double Bz, const THit* aHit, double nLoops);
 
    template<class THit, class TSimHit>
    bool didParticleExitCDC(const THit* aHit);
 
    bool m_usePXDHits;                                          
    bool m_useSVDHits;                                          
    bool m_useCDCHits;                                          
    bool m_useOnlyAxialCDCHits;                                 
    bool m_useOnlyBeforeTOP;                                    
    float m_useNLoops;                                          
    bool m_useReassignedHits;                                   
    bool m_useSecondCDCHits;                  
    bool m_enforceTrueHit;                                      
    std::vector<std::string>
    m_whichParticles;                  
    int m_particleProperties =
      0;                               
    double m_energyCut;                                         
    bool m_neutrals;                                            
    bool m_mergeDecayInFlight;                                  
    bool m_setTimeSeed;                                         
    double m_smearing;                                          
    std::vector<double>
    m_smearingCov;                          
    TMatrixDSym
    m_initialCov;                                   
    int m_notEnoughtHitsCounter =
      0;                                
    int m_noTrueHitCounter =
      0;                                     
    int m_nRecoTracks = 0;                                          
    std::string m_recoTracksStoreArrayName;                          
    int m_minPXDHits;                                           
    int m_minSVDHits;                                           
    int m_minCDCAxialHits;                                      
    int m_minCDCStereoHits;                                     
    bool m_allowFirstCDCSuperLayerOnly;                         
    int m_minimalNdf;                                           
    std::vector<int>
    m_fromPdgCodes;                            
    std::vector<int>
    m_particlePdgCodes;                        
    bool m_onlyCheckDirectParentPdgCode = false;
 
    bool m_mcParticlesPresent =
      false; 
    double m_splitAfterDeltaT; 
    bool m_discardAuxiliaryHits = false; 
    std::vector<uint> m_param_useCDCSuperLayers;  
    std::vector<uint> m_param_useCDCLayers;       
    std::vector<uint> m_param_ignoreCDCLayers;    
    std::array<bool, 9> m_useCDCSuperLayers{};   
    std::array<bool, 56> m_useCDCLayers{};       
    StoreArray<MCParticle> m_MCParticles;   
    StoreArray<RecoTrack> m_RecoTracks;     
    StoreArray<PXDTrueHit> m_PXDTrueHits;   
    StoreArray<PXDCluster> m_PXDClusters;   
    StoreArray<SVDTrueHit> m_SVDTrueHits;   
    StoreArray<SVDCluster> m_SVDClusters;   
    StoreArray<CDCHit> m_CDCHits;           
    StoreArray<CDCSimHit> m_CDCSimHits;     
  };
}