MuonSensitiveDetector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONSENSITIVEDETECTORSR4_MUONSENSITIVEDETECTOR_H
#define MUONSENSITIVEDETECTORSR4_MUONSENSITIVEDETECTOR_H
 
#include <GeoPrimitives/GeoPrimitives.h>
 
#include <StoreGate/WriteHandle.h>
#include <MuonReadoutGeometryR4/MuonDetectorManager.h>
#include <xAODMuonSimHit/MuonSimHitContainer.h>
#include <xAODMuonSimHit/MuonSimHitAuxContainer.h>
#include <HitManagement/AthenaHitsVector.h>
#include <HitManagement/HitCollectionMap.h>
#include <AthenaBaseComps/AthMessaging.h>
 
#include <G4VSensitiveDetector.hh>

namespace MuonG4R4 {
    struct MuonSimHitsVec : public  HitsVectorBase {
        MuonSimHitsVec() {
            container->setStore(auxContainer.get());
        }

        std::unique_ptr<xAOD::MuonSimHitContainer> container{std::make_unique<xAOD::MuonSimHitContainer>()};
        std::unique_ptr<xAOD::MuonSimHitAuxContainer> auxContainer{std::make_unique<xAOD::MuonSimHitAuxContainer>()};
    };
 
    class MuonSensitiveDetector : public G4VSensitiveDetector, public AthMessaging {
        public:
            MuonSensitiveDetector(const std::string& name, 
                                  const std::string& output_key,
                                  const std::string& trf_storeKey,
                                  const MuonGMR4::MuonDetectorManager* detMgr);
            ~MuonSensitiveDetector() = default;

            virtual void Initialize(G4HCofThisEvent* HCE) override final;
 
        private:
            std::string m_writeKey{};
            xAOD::MuonSimHitContainer* m_outContainer{};
            SG::ReadHandleKey<ActsTrk::DetectorAlignStore> m_trfCacheKey;
        protected:
            bool processStep(const G4Step* step) const;

            ActsTrk::GeometryContext getGeoContext(const EventContext& ctx) const;

            xAOD::MuonSimHit* lastSnapShot(const Identifier& gasGapId,
                                           const G4Step* hitStep);

            xAOD::MuonSimHit* propagateAndSaveStrip(const Identifier& hitId,
                                                    const Amg::Transform3D& toGasGap,
                                                    const G4Step* hitStep);

            xAOD::MuonSimHit* saveHit(const Identifier& hitId,
                                      const Amg::Vector3D& hitPos,
                                      const Amg::Vector3D& hitDir,
                                      const double globTime,
                                      const G4Step* hitStep);

            const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};
 
            
    };
}
template<> inline StatusCode HitCollectionMap::Record<MuonG4R4::MuonSimHitsVec>(std::string const& sgKey, std::string const& hitCollectionName, EventContext const& ctx) {
    SG::WriteHandle<xAOD::MuonSimHitContainer> writeHandle{sgKey, ctx};
    auto simHitVec = Extract<MuonG4R4::MuonSimHitsVec>(hitCollectionName);
    if (!simHitVec) {
        THROW_EXCEPTION("The Muon sim hit collection "<<hitCollectionName<<" does not exist");
    }
    return writeHandle.record(std::move(simHitVec->container), 
                       std::move(simHitVec->auxContainer));
 
}
 
#endif