PRD_TruthTrajectoryBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/

// PRD_TruthTrajectoryBuilder.h, (c) ATLAS Detector software
 
#ifndef TRK_TRUTHTRACKTOOLS_PRD_TRUTHTRAJECTORYBUILDER_H
#define TRK_TRUTHTRACKTOOLS_PRD_TRUTHTRAJECTORYBUILDER_H 1
 
// Gaudi
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
// Trk includes
#include "TrkTruthTrackInterfaces/IPRD_TruthTrajectoryBuilder.h"
#include "TrkTruthTrackInterfaces/PRD_TruthTrajectory.h"
#include "TrkTruthData/PRD_MultiTruthCollection.h"
#include "TrkTruthTrackInterfaces/IPRD_TruthTrajectoryManipulator.h"
#include "TrkToolInterfaces/IPRD_Provider.h"
 
class AtlasDetectorID;
 
#include "AtlasHepMC/GenParticle_fwd.h"
  
namespace Trk {
    
  class PrepRawData;
    
     
  class PRD_TruthTrajectoryBuilder : public AthAlgTool, virtual public IPRD_TruthTrajectoryBuilder {
 
     public:     
        //** Constructor with parameters */
       PRD_TruthTrajectoryBuilder( const std::string& t, const std::string& n, const IInterface* p );
 
       // Athena algtool's Hooks
       virtual StatusCode  initialize() override;
       virtual StatusCode  finalize() override;

       virtual std::map< HepMC::ConstGenParticlePtr, PRD_TruthTrajectory > truthTrajectories() const override;

       virtual StatusCode refreshEvent() override;
 
     private:
       const AtlasDetectorID*                               m_idHelper;                         
                                                                                                
        ToolHandle<IPRD_Provider>                           m_idPrdProvider{this, "InDetPRD_Provider", ""};                    
        ToolHandle<IPRD_Provider>                           m_msPrdProvider{this, "MuonPRD_Provider", ""};                    
        
        ToolHandleArray<IPRD_TruthTrajectoryManipulator>    m_prdTruthTrajectoryManipulators{this, "PRD_TruthTrajectoryManipulators", {}};   
        
        SG::ReadHandleKeyArray<PRD_MultiTruthCollection>    m_prdMultiTruthCollectionNames{this,"PRD_MultiTruthCollections",{"PixelClusterTruth","SCT_ClusterTruth","TRT_DriftCircleTruth"}, "PRD multi truth collection names this builder is working on"};
        std::vector<const PRD_MultiTruthCollection*>        m_prdMultiTruthCollections;         
        
        Gaudi::Property<double>                             m_minPt{this,"MinimumPt",400.,"minimum pT to be even considered"};
        Gaudi::Property<bool>                               m_geantinos{this,"Geantinos",false,"Track geantinos or not"};
        
  };
 
} // end of namespace
 
#endif // TRK_TRUTHTRACKTOOLS_PRD_TRUTHTRAJECTORYBUILDER_H