TruthSegmentMaker.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRUTHSEGMENTMAKER_TRUTHSEGMENTMAKER_H
#define MUONTRUTHSEGMENTMAKER_TRUTHSEGMENTMAKER_H
 
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
#include "xAODMuonSimHit/MuonSimHitContainer.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
#include "xAODMuon/MuonSegmentContainer.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
namespace MuonR4{
  /*** @brief The TruthSegmentMaker collects hits inside the chamber stemming from the same HepMC::GenParticle. 
   *          The particle state at the first sim hit is propagated via a simple straight line onto the plane 
   *          z=0 in the sector frame. The parameters are then translated into the global frame and a xAOD::MuonSegment
   *          is created. The segment hit summary is written based on the sim hit type. If a gasGap has eta & phi measurements,
   *          the counters in the respective categories are each increased for the given sim hit. Finally, a vector of ElementLinks
   *          pointing to the sim hits is decorated to the segment. */
  class TruthSegmentMaker : public AthReentrantAlgorithm {
      public:
          using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
          ~TruthSegmentMaker() = default;
 
          StatusCode initialize() override final;
          StatusCode execute(const EventContext& ctx) const override;
      
      private:
          template <class ContainerType> StatusCode retrieveContainer(const EventContext& ctx,
                                                                      const SG::ReadHandleKey<ContainerType>& key,
                                                                      const ContainerType* & contToPush) const;
          Amg::Transform3D toChamber(const ActsGeometryContext& gctx,
                                     const Identifier& chanId) const;
          ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "IdHelperSvc",  "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
          SG::ReadHandleKeyArray<xAOD::MuonSimHitContainer> m_readKeys{this, "SimHitKeys", {}};
          SG::ReadHandleKey<ActsGeometryContext> m_geoCtxKey{this, "AlignmentKey", "ActsAlignment", "cond handle key"};
          SG::WriteHandleKey<xAOD::MuonSegmentContainer> m_segmentKey{this, "WriteKey", "TruthSegmentsR4"};
          SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_eleLinkKey{this, "SimHitLink", m_segmentKey, "simHitLinks"};
          SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_ptKey{this, "PtKey", m_segmentKey, "pt"};
          SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_locParKey{this, "LocParKey", m_segmentKey,"localSegPars"};
          SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_qKey{this, "qKey", m_segmentKey, "charge"};
          Gaudi::Property<bool> m_useOnlyMuonHits{this, "useOnlyMuonHits", true};
          const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};
  };
}
#endif