MuonTruthSegmentCreationAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
  ** @brief This alg receives the container of truth muons, selects good muons according to 
  *         their origin, creates segments, and store them into a new container. Segments are 
  *         decorated with number and type of hits available, link to the 
  *         corresponding truth muon, position and direction of the truth segment, and
  *         sector, chamber layer, station eta, and technology of the chamber where the 
  *         segment has been recorded.
*/
 
#pragma once
 
#include <map>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "Identifier/Identifier.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "StoreGate/ReadDecorHandleKeyArray.h"
 
class MuonSimDataCollection;
class CscSimDataCollection;
 
namespace MuonGM {
    class MuonDetectorManager;
}
 
namespace Muon {
 
    class MuonTruthSegmentCreationAlg : public AthReentrantAlgorithm {
    public:
 
        // Constructor with parameters:
        using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
        // Basic algorithm methods:
        virtual StatusCode initialize() override;
        virtual StatusCode execute(const EventContext& ctx) const override;

        using ChamberIdMap = std::map<Muon::MuonStationIndex::ChIndex, std::vector<Identifier>>;
 
    private:

        StatusCode fillChamberIdMap(const EventContext& ctx, const xAOD::TruthParticle& truthParticle, ChamberIdMap& ids) const;

        StatusCode createSegments(const EventContext& ctx, 
                                  const ElementLink<xAOD::TruthParticleContainer>& truthLink,
                                  const ChamberIdMap& ids,
                                  xAOD::MuonSegmentContainer& segmentContainer) const;
                          
        SG::ReadHandleKey<xAOD::TruthParticleContainer> m_muonTruth{this, "muonTruth", "MuonTruthParticles"};
        SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> m_truthOriginKey{this, "truthOriginKey", m_muonTruth, "truthOrigin"};

        SG::ReadDecorHandleKeyArray<xAOD::TruthParticleContainer, std::vector<unsigned long long>> m_truthHitsKeyArray{this, "truthHitsKeyArray", { }};
   
        SG::WriteHandleKey<xAOD::MuonSegmentContainer> m_muonTruthSegmentContainerName{this, "MuonTruthSegmentName", "MuonTruthSegments"};
   
        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"};
 
        SG::ReadCondHandleKey<MuonGM::MuonDetectorManager> m_detMgrKey{this, "DetectorManagerKey", "MuonDetectorManager",
                                                                      "Key of input MuonDetectorManager condition data"};
 
    };
 
}  // namespace Muon