gsd.h

Go to the documentation of this file.

#pragma once
 
// geant4
#include "G4VSensitiveDetector.hh"
 
// gemc
#include <gemc/goptions/goptions.h>
#include <gemc/gdynamicDigitization/gdynamicdigitization.h>
#include <gemc/gbase/gbase.h>
 
// c++
#include <unordered_map>



constexpr int ERR_DYNAMICPLUGINNOTFOUND = 2001;

constexpr int ERR_HITNOTFOUNDINCOLLECTION = 2002;

constexpr int ERR_NOCOLLECTION = 2003;

constexpr const char* GSENSITIVE_LOGGER = "gsd";

 
namespace gsensitivedetector {

inline GOptions defineOptions() { return GOptions(GSENSITIVE_LOGGER); }
 
} // namespace gsensitivedetector

using GHitsCollection = G4THitsCollection<GHit>;

class GSensitiveDetector : public GBase<GSensitiveDetector>, public G4VSensitiveDetector {
 
public:
    GSensitiveDetector(const std::string&               sdName,
                       const std::shared_ptr<GOptions>& goptions);

    void Initialize(G4HCofThisEvent* g4hc) override;

    G4bool ProcessHits(G4Step* thisStep, G4TouchableHistory* g4th) override;

    void EndOfEvent(G4HCofThisEvent* g4HitCollection) override;

    void assign_digi_routine(std::shared_ptr<GDynamicDigitization> digi_routine) { digitization_routine = digi_routine; }

    void resetTouchableMap() { gTouchableMap.clear(); }
 
private:
    std::shared_ptr<GDynamicDigitization> digitization_routine;

    std::map<std::string, std::shared_ptr<GTouchable>> gTouchableMap;

    inline std::shared_ptr<GTouchable> getGTouchable(const G4Step* thisStep) {
        std::string vname = thisStep->GetPreStepPoint()->GetTouchable()->GetVolume()->GetName();
 
        auto it = gTouchableMap.find(vname);
        // Return a per-step copy: ProcessHits and processTouchable mutate trackId, pid and the
        // time-cell index, so the registry entry must stay pristine across steps and events.
        if (it != gTouchableMap.end()) { return std::make_shared<GTouchable>(*it->second); }
        // If not found, log an error. The calling code assumes a valid pointer.
        log->error(ERR_DYNAMICPLUGINNOTFOUND, "GTouchable for volume " + vname + " not found in gTouchableMap");
    }

    std::unordered_map<std::string, GHit*> hitsByCellKey;

    GHitsCollection* gHitsCollection;
 
public:
    inline void registerGVolumeTouchable(const std::string& name, std::shared_ptr<GTouchable> gt) {
        log->info(2, "Registering touchable gvolume <" + name + "> with value: " + gt->getIdentityString());
 
        // Store the GTouchable in the map; this module retains a shared ownership reference.
        gTouchableMap[name] = gt;
    }
};