OMSimHitManager.hh

Go to the documentation of this file.

 
#pragma once
 
#include "OMSimPMTResponse.hh"
 
#include <G4ThreeVector.hh>
#include <fstream>
#include <G4AutoLock.hh>
#include <G4Threading.hh>
 
struct HitStats
{
    std::vector<G4long> eventId;                         
    std::vector<G4double> hitTime;                       
    std::vector<G4double> flightTime;                    
    std::vector<G4double> pathLenght;                    
    std::vector<G4double> energy;                        
    std::vector<G4int> PMTnr;                            
    std::vector<G4ThreeVector> direction;                
    std::vector<G4ThreeVector> localPosition;            
    std::vector<G4ThreeVector> globalPosition;           
    std::vector<G4double> generationDetectionDistance;   
    std::vector<OMSimPMTResponse::PMTPulse> PMTresponse; 
};
 
class OMSimHitManager
{
    OMSimHitManager();
    ~OMSimHitManager() = default;
    OMSimHitManager(const OMSimHitManager &) = delete;
    OMSimHitManager &operator=(const OMSimHitManager &) = delete;
 
public:
    static void init();
    static void shutdown();
    static OMSimHitManager &getInstance();
 
    void appendHitInfo(
        G4int p_eventid,
        G4double pGlobalTime,
        G4double pLocalTime,
        G4double pTrackLength,
        G4double pEnergy,
        G4int pPMTHitNumber,
        G4ThreeVector pMomentumDirection,
        G4ThreeVector pGlobalPos,
        G4ThreeVector pLocalPos,
        G4double pDistance,
        OMSimPMTResponse::PMTPulse pResponse,
        G4int pModuleIndex = 0);
 
    void reset();
    std::vector<double> countMergedHits(int pModuleIndex = 0, bool pDEweight = false);
    void setNumberOfPMTs(int pNumberOfPMTs, int pModuleIndex = 0);
    HitStats getMergedHitsOfModule(int pModuleIndex = 0);
    HitStats getSingleThreadHitsOfModule(int pModuleIndex = 0);
    bool areThereHitsInModuleSingleThread(int pModuleIndex = 0);
    void sortHitStatsByTime(HitStats &pHits);
    std::vector<int> calculateMultiplicity(const G4double pTimeWindow, int pModuleNumber = 0);
    G4int getNextDetectorIndex() { return ++m_currentIndex; }
    G4int getNumberOfModules() { return m_currentIndex + 1; }
 
    void mergeThreadData();
 
    std::map<G4int, HitStats> m_moduleHits; 
 
private:
    std::map<G4int, G4int> m_numberOfPMTs; 
    G4int m_currentIndex;
    static G4Mutex m_mutex;
    static OMSimHitManager *m_instance;
    struct ThreadLocalData
    {
        std::map<G4int, HitStats> moduleHits;
    };
    G4ThreadLocal static ThreadLocalData *m_threadData;
};
 
inline OMSimHitManager *g_hitManager = nullptr;