STEER/TFndGeometry.h

// @(#)fROOT/STEER:$Name:  $:$Id: TFndGeometry.h,v 1.19 2007/09/05 11:09:42 Diego_Faso Exp $
// Author: Diego Faso <mailto:faso@to.infn.it>, 2006/07/05

#ifndef FROOT_TFndGeometry
#define FROOT_TFndGeometry

//                                                                         //
//                             TFndGeometry                                //
//                                                                         //
//                                                                         //

#include "TApplication.h"
#include "TGeoManager.h"

#include "TObject.h"
#include "TCanvas.h"

#include "FROOT.h"
using namespace FROOT;

class TFndGeometry: public TObject{

 public:
  
  // if you change the following "enum" check also TFndEvd::fDetectorVisibililty
  enum EFndGeoLayers { FGEO_BEAM_PIPE = 0,                 // Beam-pipe beryllium part
                       FGEO_TOFINO,                        // TOFINO active part (scintillators)
                       FGEO_ISIM, FGEO_TARGETS, FGEO_OSIM, // Vertex region
                       FGEO_INNER_LMD, FGEO_OUTER_LMD,     // inner and outer layers of LMD
                       FGEO_STRAW_TUBES,                   // View region occupied by straw tubes (all layers)
                       FGEO_STRAW_TUBES_1,FGEO_STRAW_TUBES_2,FGEO_STRAW_TUBES_3, // every straw-tubes layer
                       FGEO_STRAW_TUBES_4,FGEO_STRAW_TUBES_5,FGEO_STRAW_TUBES_6, // (memory consuming!!!)
                       FGEO_TOFONE };                         // TOFONE active part (scintillators)
  
  enum EFndGeoSetup { FGEO_2003 = 1, FGEO_2006 };
  
  TString GetLayerName(EFndGeoLayers lay)
    {
      switch(lay){
      case FGEO_BEAM_PIPE: return "BEAM-PIPE";
      case FGEO_TOFINO: return "TOFINO";
      case FGEO_ISIM: return "ISIM";
      case FGEO_TARGETS: return "TARGETS";
      case FGEO_OSIM: return "OSIM";
      case FGEO_INNER_LMD: return "INNER-LMD";
      case FGEO_OUTER_LMD: return "OUTER-LMD";
      case FGEO_STRAW_TUBES: return "STB";
      case FGEO_STRAW_TUBES_1: return "STB Layer 1";
      case FGEO_STRAW_TUBES_2: return "STB Layer 2";
      case FGEO_STRAW_TUBES_3: return "STB Layer 3";
      case FGEO_STRAW_TUBES_4: return "STB Layer 4";
      case FGEO_STRAW_TUBES_5: return "STB Layer 5";
      case FGEO_STRAW_TUBES_6: return "STB Layer 6";
      case FGEO_TOFONE: return "TOFONE";
      default: return 0;
      } 
    }

  // the following definitions are needed to drive
  // the event-display by using gFndGeom
  typedef struct{
    Int_t det; 
    Int_t ch;
  } FinEvdHit_t;
  enum EFndGeoHitMode { FGEO_HIT_DETECTED = 0, FGEO_HIT_RECONSTRUCTED };


 protected:
  Int_t fVerboseLevel;
  Int_t fSetup; // data-taking setup (see TFndGeometry::EFndGeoSetup)
  TGeoNode   *fPromptNode; // The prompt-node ("FIDA_1") is used a lot of times by several methods (do not delete it)
  TGeoVolume *fPromptVolume; // The prompt-volume ("FIDA") is used a lot of times by several methods (do not delete it)
  TGeoNode   *fCurDetNode; // Current detector node (set by the "SelectDetectorNode" method)

  TGeoNode *fTgtNodes[8]; // node array for targets (they change depending on the data-taking [2003/2006])

  // --- roto-traslation matrices (loaded one time only from gGeoManager)
  // TOF
  TGeoHMatrix *fGeoMatr_Top_Tof[2][K_N_TOFONE_SLABS]; // MRS->Tof.RF (0:TOFINO; 1:TOFONE)
  // SIL
  TGeoMatrix *fGeoMatr_Top_Sil[2][K_N_Sil_Modules]; // indices: [layer:isim/osim][module]
  // LMD
  TGeoMatrix *fGeoMatr_Top_Lmd[K_N_LMD_LAYERS][K_N_LMD_CHAMS_PER_LAYER]; // MRS->Lmd.RF
  TGeoMatrix *fGeoMatr_InnerLmd_Wire[K_N_LMD_CHAMS_PER_LAYER][K_N_LMD_INNER_WIRES]; // InnerLmd.RF->Wire.RF
  TGeoMatrix *fGeoMatr_OuterLmd_Wire[K_N_LMD_CHAMS_PER_LAYER][K_N_LMD_OUTER_WIRES]; // OuterLmd.RF->Wire.RF
  TGeoHMatrix *fGeoMatr_Top_InnerLmdWire[K_N_LMD_CHAMS_PER_LAYER][K_N_LMD_INNER_WIRES]; // MRS->InnerWire.RF
  TGeoHMatrix *fGeoMatr_Top_OuterLmdWire[K_N_LMD_CHAMS_PER_LAYER][K_N_LMD_OUTER_WIRES]; // MRS->OuterWire.RF
  // STB
  TGeoMatrix *fGeoMatr_Top_StbWire[K_N_STB_CROWNS][K_N_STB_TUBES]; // MRS->StbWire.RF
  

  TCanvas *fMainCanvas;  // canvas for drawing the geometry

  
 private:
  void Init();
  TString GetSetupName(Int_t set){
    switch(set){
    case (Int_t)FGEO_2003: return "2003 (first data-taking)";
    case (Int_t)FGEO_2006: return "2006 (second data-taking)";
    default:
      Error("GetSetupName","Maybe you have just found a BUG!");
      gApplication->Terminate();
    }
  }

  void LoadGeoMatrices();

  Int_t PrintNode(TGeoNode *node,Int_t id_subnum,const Int_t level);

  void SetStbLayVis(Int_t lay,Bool_t visible,Bool_t detailed);
  void SetLmdLayVis(Int_t lay,Bool_t visible,Bool_t detailed);
    
 public:
  
  TFndGeometry();
  ~TFndGeometry();
  void SetVerboseLevel(Int_t lev) { fVerboseLevel = lev; }
  TString GetCurrentSetup() { return GetSetupName(fSetup); }
  TCanvas *GetMainCanvas() { return fMainCanvas; }

  TGeoNode   *GetPromptNode() { return fPromptNode; }
  TGeoVolume *GetPromptVolume() { return fPromptVolume; }
  TGeoNode *GetCurrentNode() { return fCurDetNode; } // previousely set by SelectDetectorNode
  TGeoNode *SelectDetectorNode(const Int_t &det, const Int_t &ch);

  static Double_t GetTofinoSlabAngle() { return (Double_t) ( 360./(Double_t)(K_N_TOFINO_SLABS) ); } // angle of every slab (deg)
  static Double_t GetTofoneSlabAngle() { return (Double_t) ( 360./(Double_t)(K_N_TOFONE_SLABS) ); } // angle of every slab (deg)
  static Double_t GetTofinoRotAngles(Int_t slab);
  static Double_t GetTofoneRotAngles(Int_t slab);

  // --- Reference System changes
  static Double_t SilPositionStrip(const Int_t &side, const Double_t &barcha);

  void RefSys_TofSlabToMRS(const Int_t &channel,const Double_t* local, Double_t* master);
  void RefSys_SilModToMRS(const Int_t &layer,const Int_t &module,const Double_t* local, Double_t* master);
  void RefSys_LmdWireToMRS(const Int_t &channel,const Double_t* local, Double_t* master);
  void RefSys_StbWireToMRS(const Int_t &channel,const Double_t* local, Double_t* master);

  static Int_t ConvertFromGeant3(TString root_geom_filename="fin_spec_geom.root", TString geant3_geom_path="fin_spec.geom",Bool_t CleanTmp=kTRUE);
  void PrintStructure(const Int_t levels=7);
  void PrintGeoMatrices(Int_t det,Int_t verbose_lev); // local verbose level is used

  Int_t LoadGeometry(TString filename="$FROOTSYS/geom/fin_spec_geom.root");
  Int_t ShowGeometry(Option_t *option="ogl",Int_t vis_level=7);

  void SetLayerVisibility(Int_t layer,Bool_t visible,Bool_t detailed=kFALSE);
  void SetStructureVisibility(Bool_t visible,Int_t detail_lev=0);

  Int_t SetDefaultColors();
  Int_t SetDefaultVisibility();


  // The following pure virtual methods were used in order to 
  // drive the event display by using the global pointer gFndGeom
  //   (see TFndEvd for the relative documentation)
  // they are now commented in order to provide the possibility
  // of loading geometry without creating an object of TFndEvd (March 13, 2007 - D. Faso)
  /*   virtual void HitDetector(Int_t det, Int_t ch, Int_t mode=FGEO_HIT_DETECTED,Bool_t refresh_display=kTRUE) = 0; */
  /*   virtual void RemoveHit(Int_t det, Int_t ch,Bool_t refresh_display=kTRUE) = 0; */
  /*   virtual void RemoveAllHits(Bool_t reset_default_view=kFALSE,Bool_t refresh_display=kTRUE) = 0;   */
  /*   virtual void PrintHits(Bool_t Selected) = 0; */
  /*   virtual void DisplayRandomEvent(Bool_t keep_current_hits=kFALSE) = 0; */
  
  ClassDef(TFndGeometry,0)
};
R__EXTERN  TFndGeometry *gFndGeom;
#endif // FROOT_TFndGeometry

---