TRestDetectorGeometry.h

 
#ifndef RestCore_TRestDetectorGeometry
#define RestCore_TRestDetectorGeometry
 
#include <TGeoManager.h>
#include <TGeoMaterial.h>
#include <TGeoVolume.h>
#include <TObject.h>
 
#include <iostream>
#include <vector>
 
#include "TRestDetectorGas.h"
 
#if defined USE_Garfield
 
#if defined USE_Garfield_OLD
#include "ComponentBase.hh"
#else
#include "Component.hh"
#endif
 
#include "GeometryRoot.hh"
#include "Sensor.hh"
 
#if defined USE_Garfield_OLD
typedef Garfield::ComponentBase Component;
#else
typedef Garfield::Component Component;
#endif
 
#endif
 
class TRestDetectorGeometry : public TGeoManager {
   protected:
#if defined USE_Garfield
    Garfield::GeometryRoot* fGfGeometry;       
    std::vector<Component*> vGfComponent;      
    std::vector<Garfield::Sensor*> vGfSensor;  
    TGeoNode* fDriftElec;                      
    std::vector<TGeoNode*> vReadoutElec;       
 
#endif
 
   public:
    // Constructor
    TRestDetectorGeometry();
    // Destructor
    virtual ~TRestDetectorGeometry();
 
    void InitGfGeometry();
 
#if defined USE_Garfield
 
    Garfield::GeometryRoot* GetGfGeometry() { return fGfGeometry; }
 
    void SetGfGeoMedium(const char* geoMatName, TRestDetectorGas* gas) {
        fGfGeometry->SetMedium(geoMatName, gas->GetGasMedium());
    }
 
    void AddGfComponent(Component* c) {
        c->SetGeometry(fGfGeometry);
        vGfComponent.push_back(c);
    }
 
    void AddGfSensor(Garfield::Sensor* s) { vGfSensor.push_back(s); }
 
    void SetDriftElecNode(TGeoNode* n) { fDriftElec = n; }
 
    void AddReadoutElecNode(TGeoNode* n) { vReadoutElec.push_back(n); }
 
    TGeoNode* GetDriftElecNode() { return fDriftElec; }
 
    int GetNReadoutElecNodes() { return vReadoutElec.size(); }
 
    TGeoNode* GetReadoutElecNode(unsigned int i) {
        if (i < vReadoutElec.size())
            return vReadoutElec[i];
        else
            return nullptr;
    }
 
    Component* GetGfComponent(unsigned int i) {
        if (i < vGfComponent.size())
            return vGfComponent[i];
        else
            return nullptr;
    }
 
    int GetNbOfGfComponent() { return vGfComponent.size(); }
 
    Garfield::Sensor* GetGfSensor(unsigned int i) {
        if (i < vGfSensor.size())
            return vGfSensor[i];
        else
            return nullptr;
    }
 
    int GetNbOfGfSensor() { return vGfSensor.size(); }
 
    Garfield::Medium* GetGfMedium(double x, double y, double z) {
        FindNode(x / 10., y / 10., z / 10.);  // needed due to a bug in TGeoNavigator
        return fGfGeometry->GetMedium(x / 10., y / 10., z / 10.);
    }  // mm to cm
#endif
 
    void PrintGeometry();
 
    ClassDef(TRestDetectorGeometry, 1);  // REST event superclass
};
 
#endif