MdcROOTGeo.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 MdcROOTGeo
 *                 inherit from class SubDetectorROOTGeo
 */

#ifndef Mdc_ROOT_GEO_H
#define Mdc_ROOT_GEO_H

#include <map>

#include <TGeoVolume.h>
#include <TGeoNode.h>
#include <TGeoPhysicalNode.h>

#include "SubDetectorROOTGeo.h"
#include "Mdc2DWire.h"

class MdcROOTGeo : public SubDetectorROOTGeo
{
 public:
  
  MdcROOTGeo();

  ~MdcROOTGeo();

  void InitFromGDML( const char *gdmlFile, const char *setupName );

  void InitFromROOT( TGeoVolume *vol );

  void Init2DGeometry();

  void SetNode();
  
  int CorrectReplica(int layer, int replica);

  void SetVolumeDefaultVis();

  void SetAllVisible();

  void SetQuarterVisible();

  void SetHalfVisible();

  void SetPhysicalNode();
  
  //void SetPhysicalDefaultVis();

  void SetDetector();

  void SetHits();
  void SetTMatch(Bool_t input);
  void SetQMatch(Bool_t input);
  
  void SetVisMdcDetector();

  void SetVisMdcHits();

  int GetSegmentNb() { return m_kSegment; }

  int GetLayerNb() { return m_kLayer; }

  int GetReplicaNb(int layer) { return m_kReplica[layer]; }

  void SetVolumeMdc(TGeoVolume *vol) { m_Mdc = vol; }
  
  TGeoVolume *GetVolumeMdc() { return m_Mdc; }
  
  TGeoVolume *GetVolumeSegment( int segment );

  TGeoVolume *GetVolumeLayer( int layer );
  
  TGeoVolume *GetVolumeReplica( int layer );

  TGeoNode *GetSegment( int segment, int no );
  
  TGeoNode *GetLayer( int layer );
  
  TGeoNode *GetReplica( int layer, int replica );

  TGeoPhysicalNode *GetPhysicalSegment( int segment );

  TGeoPhysicalNode *GetPhysicalReplica( int layer, int replica );

  Mdc2DWire *Get2DWire( Int_t layer, Int_t replica ); 

  Int_t GetSimuLayer(Int_t trueLayer);
  
  Bool_t IsHit(Int_t layer, Int_t wire);

  void Draw(Option_t *option);

  void DrawHits(Option_t *option);

 private:

  static const int m_kSegment        = 66;
  static const int m_kStereoLayerIn  = 8;
  static const int m_kAxialLayerIn   = 12;
  static const int m_kStereoLayerOut = 16;
  static const int m_kAxialLayerOut  = 7;
  static const int m_kLayer          = 50;
  static const int m_kTrueLayer      = 43; // m_kLayer-m_kAxialLayerOut
  static const int m_kReplicaMax     = 288;
  static const int m_kReplica[m_kLayer];
  static const int m_kStereoDir[m_kTrueLayer];
  static const int m_kCorrectLayer   = 10;
  static const int m_kiCorrectLayer[m_kCorrectLayer];
  static const int m_kiCorrectReplica[m_kCorrectLayer];
  Bool_t k_TMatch;
  Bool_t k_QMatch;

  static const int MDC_TIME_FACTOR   = 100000;
  static const int MDC_CHARGE_FACTOR = 1000000;

  int m_MdcColor;
  int m_segmentColor;
  int m_hypeColor;
  int m_tubeColor;
  int m_twistedTubsColor;
  int m_replicaColor;

  std::map<int ,int> m_CorrectMap;
  typedef std::map<int, int> intMap;

  TGeoVolume       *m_Mdc;
  TGeoNode         *m_NodeLayer[m_kLayer];
  TGeoNode         *m_NodeReplica[m_kLayer][m_kReplicaMax];
  TGeoPhysicalNode *m_PhysicalSegment[m_kSegment*2];
  TGeoPhysicalNode *m_PhysicalReplica[m_kLayer][m_kReplicaMax];

  const TObjArray  *m_MdcDigiCol;

  // 2D
  BesCircle2D        *m_MdcXY;
  BesCircle2D        *m_MdcXYSuper[4];
  Mdc2DWire          *m_Mdc2DWire[m_kTrueLayer][m_kReplicaMax];
    
  BesPolygon2D       *m_MdcZR[2];
};

#endif   /* Mdc_ROOT_GEO_H */

---