MuonCombinedInDetExtensionAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUONCOMBINEDALGS_MUONCOMBINEDINDETEXTENSIONALG_H
#define MUONCOMBINEDALGS_MUONCOMBINEDINDETEXTENSIONALG_H
 
#include <string>
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "MuonCombinedEvent/InDetCandidateCollection.h"
#include "MuonCombinedEvent/InDetCandidateToTagMap.h"
#include "MuonCombinedToolInterfaces/IMuonCombinedInDetExtensionTool.h"
 
#include "MuonPrepRawData/MMPrepDataContainer.h"
#include "MuonPrepRawData/TgcPrepDataContainer.h"
#include "MuonPrepRawData/MdtPrepDataContainer.h"
#include "MuonPrepRawData/CscPrepDataContainer.h"
#include "MuonPrepRawData/TgcPrepDataContainer.h"
#include "MuonPrepRawData/RpcPrepDataContainer.h"
#include "MuonPrepRawData/sTgcPrepDataContainer.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkTrack/TrackCollection.h"
 
// uses (further down the call chain) the MuPatHitTool that has a mutable cache of pointers to-be-deleted at the end of the event
// thus, currently, the MuonCombinedInDetExtensionAlg cannot become an AthReentrantAlgorithm
class MuonCombinedInDetExtensionAlg : public AthReentrantAlgorithm {
public:
    MuonCombinedInDetExtensionAlg(const std::string& name, ISvcLocator* pSvcLocator);
    ~MuonCombinedInDetExtensionAlg() = default;
 
    StatusCode initialize() override;
    StatusCode execute(const EventContext& ctx) const override;
 
private:
    template <class ContType> StatusCode loadPrdContainer(const EventContext& ctx , const SG::ReadHandleKey<ContType>& key, const ContType* & target_ptr) const;
    template <class ContType> StatusCode record(const EventContext& ctx, const SG::WriteHandleKey<ContType>& key, ContType* & target_ptr) const;
    ToolHandle<MuonCombined::IMuonCombinedInDetExtensionTool> m_muonCombinedInDetExtensionTool{
        this, "MuonCombinedInDetExtensionTool", ""};
    SG::ReadHandleKey<InDetCandidateCollection> m_indetCandidateCollectionName{
        this,
        "InDetCandidateLocation",
        "InDetCandidates",
        "ID candidates",
    };
    SG::ReadHandleKey<Muon::MdtPrepDataContainer> m_MDT_ContainerName{
        this,
        "MDTPrepDataLocation",
        "MDT_DriftCircles",
        "MDT prep data",
    };
    SG::ReadHandleKey<Muon::CscPrepDataContainer> m_CSC_ContainerName{
        this,
        "CSCPrepDataLocation",
        "CSC_Clusters",
        "CSC prep data",
    };
    SG::ReadHandleKey<Muon::RpcPrepDataContainer> m_RPC_ContainerName{
        this,
        "RPCPrepDataLocation",
        "RPC_Measurements",
        "RPC prep data",
    };
    SG::ReadHandleKey<Muon::TgcPrepDataContainer> m_TGC_ContainerName{
        this,
        "TGCPrepDataLocation",
        "TGC_Measurements",
        "TGC prep data",
    };
    SG::ReadHandleKey<Muon::sTgcPrepDataContainer> m_sTGC_ContainerName{
        this,
        "sTGCPrepDataLocation",
        "STGC_Measurements",
        "sTGC prep data",
    };
    SG::ReadHandleKey<Muon::MMPrepDataContainer> m_MM_ContainerName{
        this,
        "MMPrepDataLocation",
        "MM_Measurements",
        "MM prep data",
    };
    SG::WriteHandleKey<MuonCombined::InDetCandidateToTagMap> m_tagMap{
        this,
        "TagMap",
        "muGirlTagMap",
        "tag map",
    };
    SG::WriteHandleKey<TrackCollection> m_combTracks{
        this,
        "CombinedTrackCollection",
        "MuGirlCombinedTracks",
        "combined track collection",
    };
    SG::WriteHandleKey<TrackCollection> m_METracks{
        this,
        "METrackCollection",
        "MuGirlMETracks",
        "ME track collection",
    };
    SG::WriteHandleKey<Trk::SegmentCollection> m_segments{
        this,
        "SegmentCollection",
        "",
        "specify segment collection",
    };
 
    Gaudi::Property<bool> m_usePRDs{this, "usePRDs", false};
    Gaudi::Property<bool> m_hasCSC{this, "HasCSC", true};
    Gaudi::Property<bool> m_hasSTGC{this, "HasSTgc", true};
    Gaudi::Property<bool> m_hasMM{this, "HasMM", true};
};
 
#endif