InstallArea/include/BesVisLib/BesVisLib/TofROOTGeo.h

Go to the documentation of this file.

//$id$
/*
 *    2004/11/29   Zhengyun You      Peking University
 *                 Tof Geometry General for EventDisplay
 *
 *    2004/12/9    Zhengyun You      Peking University
 *                 named from TofGeo to TofROOTGeo
 *                 inherit from class SubDetectorROOTGeo
 */
 
#ifndef Tof_ROOT_GEO_H
#define Tof_ROOT_GEO_H
 
#include <TGeoVolume.h>
#include <TGeoNode.h>
#include <TGeoPhysicalNode.h>
 
#include "SubDetectorROOTGeo.h"
#include "Tof2DScin.h"
 
/**
 * Class TofGeo contains all of the objects necessary to describe the
 * tof geometry.
 *
 * @author Zhengyun You \URL{mailto:[email protected]}
 *
 */
 
class TofROOTGeo : public SubDetectorROOTGeo
{
 public:
  
  /// Constructor.
  TofROOTGeo();
 
  /// Destructor.
  ~TofROOTGeo();
 
  /// Initialize ROOTGeo from GDML.
  void InitFromGDML( const char *gdmlFile, const char *setupName );
 
  /// Initialize ROOTGeo from TGeoVolume logicalTof.
  void InitFromROOT( TGeoVolume *vol );
 
  /// Initialize 2D Geometry
  void Init2DGeometry();
 
  /// Set the pointers to theirs nodes; 
  void SetNode();
  
  /// Set default visual attributes;
  void SetVolumeDefaultVis();
 
  /// Set all visible;
  void SetAllVisible();
  void SetTMatch(Bool_t input);
  void SetQMatch(Bool_t input);
 
  /// Set quater visible;
  void SetQuarterVisible();
 
  /// Set half visible;
  void SetHalfVisible();
 
  /// Set noend visible;
  void SetNoEndVisible();
 
  /// Set the pointers to the physical nodes; 
  void SetPhysicalNode();
  
  /// Draw Detecor (what is detector depends on you)
  void SetDetector();
 
  /// Set all physicalNodes corresponding to digiCol;
  void SetHits();
  
  /// Set Tof detector visibility;
  void SetVisTofDetector();
 
  /// Set Tof hits visibility;
  void SetVisTofHits();
 
  /// Get number of part;
  int GetPartNb();
 
  /// Get number of scintillators on each part;
  int GetScinNb(int part);
 
  // Get number of layer on part;
  int GetLayerNb(int part);
 
  /// Set Tof volume, while initializing from ROOT;
  void SetVolumeTof(TGeoVolume *vol) { m_Tof = vol; }
  
  /// Get Tof volume;
  TGeoVolume *GetVolumeTof() { return m_Tof; }
  
  /// Get assembly volume; part=0,2 for AssemblyEc, 1 for AssemblyBr;  
  //TGeoVolumeAssembly *GetVolumeAssembly( int part );
  
  /// Get part volume;
  TGeoVolume *GetVolumePart( int part);
 
  /// Get PVF volume;
  TGeoVolume *GetVolumePVF( int part, int layer );
 
  /// Get Al volume;
  TGeoVolume *GetVolumeAl( int part, int layer );
 
  /// Get scintillator volume;
  TGeoVolume *GetVolumeScin( int part, int layer );
 
  /// Get bucket volume;
  TGeoVolume *GetVolumeBucket( int part, int layer );
 
  /// Get PVF node;
  TGeoNode *GetPVF( int part, int layer, int scin );
 
  /// Get Al node;
  TGeoNode *GetAl( int part, int layer );
 
  /// Get scintillator node;
  TGeoNode *GetScin( int part, int layer );
 
  /// Get bucket node, 0 for west bucket, 1 for the east one; 
  //TGeoNode *GetBucket( int part, int layer, int scin, int bucket );
 
  /// Get scintillator physical node; 
  TGeoPhysicalNode *GetPhysicalScin( int part, int layer, int scin );
 
  /// Get Tof2DScin;
  Tof2DScin *Get2DScin( Int_t part, Int_t layer, Int_t scin ); 
 
  /// Get part no of a scintillator physcial node
  Int_t GetPart(TGeoPhysicalNode* phyNode);
 
  /// Draw function
  void Draw(Option_t *option);
 
  /// Draw 2D hits
  void DrawHits(Option_t *option);
 
 private:
  static const int m_kPart     = 3;
  static const int m_kLayerBr  = 2;
  static const int m_kLayerEc  = 1;
  static const int m_kScinBr   = 88;
  static const int m_kScinEc   = 48;
  static const int m_kBucketEc = 1;
  static const int m_kBucketBr = 2;
 
  static const int TOF_TIME_FACTOR   = 1000000;
  static const int TOF_CHARGE_FACTOR = 1000000; 
   
  Bool_t k_TMatch;
  Bool_t k_QMatch;
 
  int m_TofColor;
  int m_BucketColor;
  int m_ScinColor;
 
  TGeoVolume       *m_Tof;
  TGeoNode         *m_NodePVF[m_kPart][m_kLayerBr][m_kScinBr];
  TGeoNode         *m_NodeAl[m_kPart][m_kLayerBr];
  TGeoNode         *m_NodeScin[m_kPart][m_kLayerBr];
  TGeoPhysicalNode *m_PhysicalScin[m_kPart][m_kLayerBr][m_kScinBr];
 
  const TObjArray  *m_TofDigiCol;
 
  // 2D
  Tof2DScin        *m_Tof2DScin[m_kPart][m_kLayerBr][m_kScinBr];
 
};
 
#endif   /* Tof_ROOT_GEO_H */