MuonTruthDecorationAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRUTHPARTICLEALGS_MUONTRUTHDECORATIONALG_H
#define TRUTHPARTICLEALGS_MUONTRUTHDECORATIONALG_H
 
#include <map>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "GeneratorObjects/xAODTruthParticleLink.h"
#include "MCTruthClassifier/IMCTruthClassifier.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonRecHelperTools/MuonEDMPrinterTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/WriteHandleKeyArray.h"
#include "TrackRecord/TrackRecordCollection.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkTruthData/PRD_MultiTruthCollection.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthVertex.h"
 
class MuonSimDataCollection;
class CscSimDataCollection;
 
namespace MuonGM {
    class MuonDetectorManager;
}
 
namespace Muon {
 
    class MuonTruthDecorationAlg : public AthReentrantAlgorithm {
    public:
        typedef std::map<Muon::MuonStationIndex::ChIndex, std::vector<Identifier> > ChamberIdMap;
 
        // Constructor with parameters:
        MuonTruthDecorationAlg(const std::string& name, ISvcLocator* pSvcLocator);
 
        // Basic algorithm methods:
        virtual StatusCode initialize() override;
        virtual StatusCode execute(const EventContext& ctx) const override;
 
    private:
        StatusCode addTrackRecords(const EventContext& ctx, 
                                   xAOD::TruthParticle& truthParticle) const;
        
 
        StatusCode addHitCounts(const EventContext& ctx, 
                                xAOD::TruthParticle& truthParticle, 
                                ChamberIdMap& ids) const;
 
        void addHitIDVectors(xAOD::TruthParticle& truthParticle, const MuonTruthDecorationAlg::ChamberIdMap& ids) const;
        
        StatusCode createSegments(const EventContext& ctx, 
                                  const ElementLink<xAOD::TruthParticleContainer>& truthLink,
                                  SG::WriteHandle<xAOD::MuonSegmentContainer>& segmentContainer,
                                  const ChamberIdMap& ids) const;
 
        SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthParticleContainerName{this, "TruthParticleContainerName", "TruthParticles"};
        SG::WriteHandleKey<xAOD::TruthParticleContainer> m_muonTruthParticleContainerName{this, "MuonTruthParticleContainerName",
                                                                                          "MuonTruthParticles"};
        SG::WriteHandleKey<xAOD::MuonSegmentContainer> m_muonTruthSegmentContainerName{this, "MuonTruthSegmentName", "MuonTruthSegments"};
 
        SG::ReadHandleKeyArray<TrackRecordCollection> m_trackRecordCollectionNames{
            this, "TrackRecordCollectionNames", {"CaloEntryLayer", "MuonEntryLayer", "MuonExitLayer"}};
        SG::ReadHandleKeyArray<PRD_MultiTruthCollection> m_PRD_TruthNames{
            this,
            "PRD_TruthMaps",
            {"CSC_TruthMap", "RPC_TruthMap", "TGC_TruthMap", "MDT_TruthMap"},
            "remove NSW by default for now, can always be changed in the configuration"};
        SG::ReadHandleKeyArray<MuonSimDataCollection> m_SDO_TruthNames{
            this, "SDOs", {"RPC_SDO", "TGC_SDO", "MDT_SDO"}, "remove NSW by default for now, can always be changed in the configuration"};
        SG::ReadHandleKey<CscSimDataCollection> m_CSC_SDO_TruthNames{this, "CSCSDOs", "CSC_SDO"};
 
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
        ToolHandle<IMCTruthClassifier> m_truthClassifier{this, "MCTruthClassifier", "MCTruthClassifier/MCTruthClassifier"};
        ToolHandle<Trk::IExtrapolator> m_extrapolator{this, "Extrapolator", "Trk::Extrapolator/AtlasExtrapolator"};
 
        Gaudi::Property<bool> m_createTruthSegment{this, "CreateTruthSegments", true};
 
            // MuonDetectorManager from the conditions store
        SG::ReadCondHandleKey<MuonGM::MuonDetectorManager> m_detMgrKey{this, "DetectorManagerKey", "MuonDetectorManager",
                                                                      "Key of input MuonDetectorManager condition data"};
 
    };
 
}  // namespace Muon
 
#endif  // TRUTHPARTICLEALGS_MUONTRUTHDECORATIONALG_H