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>
 
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::vector<GTouchable> touchableVector;
 
    bool isThisANewTouchable(const std::shared_ptr<GTouchable>& thisTouchable);
 
    GHitsCollection* gHitsCollection;
 
    GHit* getHitInHitCollectionUsingTouchable(const std::shared_ptr<GTouchable>& gtouchable);
 
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;
    }
};