TRestGeant4GeometryInfo.h

 
#ifndef REST_TRESTGEANT4GEOMETRYINFO_H
#define REST_TRESTGEANT4GEOMETRYINFO_H
 
#include <TRestStringOutput.h>
#include <TRotation.h>
#include <TString.h>
#include <TVector3.h>
 
#include <map>
#include <set>
#include <vector>
 
class G4VPhysicalVolume;
 
class TRestGeant4GeometryInfo {
    ClassDef(TRestGeant4GeometryInfo, 5);
 
   private:
    bool fIsAssembly = false;
    TString fPathSeparator = "_";
 
    std::map<Int_t, TString> fVolumeNameMap = {};
    std::map<TString, Int_t> fVolumeNameReverseMap = {};
 
    void PopulateFromGeant4World(const G4VPhysicalVolume*);
 
    inline void InitializeOnDetectorConstruction(const TString& gdmlFilename,
                                                 const G4VPhysicalVolume* world) {
        PopulateFromGdml(gdmlFilename);
        PopulateFromGeant4World(world);
    }
 
   public:
    std::vector<TString> fGdmlNewPhysicalNames;
 
    std::vector<TString> fGdmlLogicalNames;
 
    std::map<TString, TString>
        fNewPhysicalToGeant4PhysicalNameMap;  // reverse map of old fGeant4PhysicalNameToNewPhysicalNameMap
 
    std::map<TString, TString> fGeant4AssemblyImprintToGdmlNameMap;
 
    std::map<TString, std::map<TString, TString>> fGdmlAssemblyToChildrenGeant4ToGdmlPhysicalNameMap;
 
    std::map<TString, TString> fGeant4AssemblyImprintToAssemblyLogicalNameMap;
 
    std::map<TString, TString> fPhysicalToLogicalVolumeMap;
 
    std::map<TString, std::vector<TString>> fLogicalToPhysicalMap;
 
    std::map<TString, TString> fLogicalToMaterialMap;
 
    std::map<TString, TVector3> fPhysicalToPositionInWorldMap;
 
    std::map<TString, TRotation> fPhysicalToRotationInWorldMap;
 
   public:
    inline TRestGeant4GeometryInfo() = default;
 
    void PopulateFromGdml(const TString&);
 
    TString GetAlternativePathFromGeant4Path(const TString&) const;
    TString GetAlternativeNameFromGeant4PhysicalName(const TString&) const;
    std::set<TString> GetAlternativeNamesFromGeant4PhysicalName(const TString&) const;
    TString GetGeant4PhysicalNameFromAlternativeName(const TString&) const;
 
    std::vector<TString> GetAllLogicalVolumes() const;
    std::vector<TString> GetAllPhysicalVolumes() const;
    std::vector<TString> GetAllAlternativePhysicalVolumes() const;
 
    std::vector<TString> GetAllLogicalVolumesMatchingExpression(const TString&) const;
    std::vector<TString> GetAllPhysicalVolumesMatchingExpression(const TString&) const;
 
    inline bool IsValidGdmlName(const TString& volume) const {
        for (const auto& name : fGdmlNewPhysicalNames) {
            if (name == volume) {
                return true;
            }
        }
        return false;
    }
 
    inline bool IsValidGeant4PhysicalVolume(const TString& volume) const {
        return fPhysicalToLogicalVolumeMap.count(volume) > 0;
    }
 
    inline bool IsValidPhysicalVolume(const TString& volume) const {
        return fNewPhysicalToGeant4PhysicalNameMap.count(volume) > 0;
    }
 
    inline bool IsValidLogicalVolume(const TString& volume) const {
        return fLogicalToPhysicalMap.count(volume) > 0;
    }
 
    inline std::vector<TString> GetAllPhysicalVolumesFromLogical(const TString& logicalVolume) const {
        if (IsValidLogicalVolume(logicalVolume)) {
            return fLogicalToPhysicalMap.at(logicalVolume);
        }
        return {};
    }
 
    inline TVector3 GetPosition(const TString& volume) const {
        return fPhysicalToPositionInWorldMap.at(volume);
    }
 
    inline TRotation GetRotation(const TString& volume) const {
        return fPhysicalToRotationInWorldMap.at(volume);
    }
 
    inline bool IsAssembly() const { return fIsAssembly; }
    inline TString GetPathSeparator() const { return fPathSeparator; }
    void SetPathSeparator(const TString& separator) { fPathSeparator = separator; }
    void InsertVolumeName(Int_t id, const TString& volumeName);
 
    TString GetVolumeFromID(Int_t id) const;
    Int_t GetIDFromVolume(const TString& volumeName) const;
 
    void Print(bool multiLine = false) const;
 
    friend class DetectorConstruction;
};
 
#endif  // REST_TRESTGEANT4GEOMETRYINFO_H