INDRAGeometryBuilder.h

 
#ifndef __INDRAGEOMETRYBUILDER_H
#define __INDRAGEOMETRYBUILDER_H
 
#include "KVBase.h"
#include "KVDetector.h"
#include "TVector3.h"
#include "TGeoVolume.h"
#include "TGeoMatrix.h"
#include "KVNumberList.h"
#include "KVNameValueList.h"
#include <unordered_map>
 
class TGeoManager;
class INDRAGeometryBuilder : public KVBase {
public:
   using theta_phi = std::pair<double, double>;
private:
    TString fDetName;
    TVector3 fInnerFront[4];
    TVector3 fOuterFront[4];
    TVector3 fFrameFront[4];
    KVDetector fCurrentDetector;
    TString fCurrentDetectorType;
    Int_t fActiveLayer;
    KVMaterial fFrameMat;
    TVector3 fInnerCentre;
    TVector3 fOuterCentre;
    TVector3 fFrameCentre;
    Int_t Ndets;
    Double_t thetaMin, thetaMax, deltaPhi, phi0;
    TGeoVolume* fFrameVolume;
    TGeoVolume* fDetVolume;
   Int_t fInnerPads, fOuterPads;
   Int_t fInnerRing, fOuterRing;
   Int_t fInnerDmod, fOuterDmod;
   Int_t fInnerModmin, fOuterModmin;
   Int_t fModmax;
   Double_t fRingCentreDistance;
   TGeoTranslation* fDetectorPosition;
   TGeoVolumeAssembly* fEtalonVol;
   Double_t fEtalonTheta[10];
   Double_t fEtalonPhi[10];
   void CorrectCoordinates(Double_t*, Double_t&, Double_t&);
   Bool_t CheckDetectorPresent(TString detname);
#ifndef __CINT__
   struct detector_properties {
      std::string name;
      long double thick = 0.0;
      int ring = 0;
      int module = 0;
      theta_phi thetaminmax;
      theta_phi phiminmax;
      bool present = false;
   };
    std::unordered_map<std::string, detector_properties> detector_map;
   std::vector<detector_properties> read_telescope_infos(
           const Char_t* prefix, Int_t ring, Int_t mod, Int_t ntel,
           double thmin, double thmax, double phi);
#endif
   TString fDataSet;
   Int_t fCurrentRun;
    TEnv fStrucInfos;
 
   void read_indra_struct_file();
   void read_layer_infos(const Char_t* name);
   void read_ring_infos(Int_t number, const Char_t* prefix);
 
   Bool_t fWithPhoswich = kFALSE;
public:
 
   INDRAGeometryBuilder(const TString& _data_set, Int_t current_run = -1);
 
   void ReadDetCAO(const Char_t* detname, Int_t ring);
   void CalculateBackPlaneCoordinates(TVector3* frontcoords, TVector3 centre, Double_t depth, TVector3* backcoords);
   void CalculateCentre(TVector3* corners, TVector3& centre);
   void CalculateCornersInPlane(TVector3* plane, Double_t thetamin,
                                Double_t thetamax, Double_t phimin, Double_t phimax, TVector3* corners);
 
   std::vector<theta_phi> GetDirectionsToTestForImport() const;
   Bool_t CheckImportResults(const KVSeqCollection*);
 
   void Print(Option_t* = "") const override;
 
   void MakeFrame(TString det_type, Int_t ring_num);
   void MakeDetector(const Char_t* det, TVector3* som, TVector3 cen);
   void PlaceFrame(Double_t phi, Int_t copy_no);
   void PlaceDetector();
   void MakeRing(const Char_t* det, int ring);
   void MakeEtalon(int RING);
   void ReflectPad(TVector3* orig, Double_t phicentre, TVector3* newpad);
   void CloseAndDraw();
 
   void Build(Bool_t withTarget = kTRUE, Bool_t closeGeometry = kTRUE);
   void BuildEtalonVolumes();
   void BuildTarget();
   void Build(const KVNumberList& rings, const KVNameValueList& detectors);
   void TransformToOwnFrame(TVector3* orig, TVector3& centre, TVector3* ownframe);
 
   Bool_t SetDetectorThickness(const TString& detname, Double_t thickness);
 
   ClassDefOverride(INDRAGeometryBuilder, 0) //Build INDRA geometry from Huguet CAO infos
};
 
#endif