CgemROOTGeo.h

Go to the documentation of this file.

/*
 *      2018/1/26   Long Peixun     IHEP
 *                  Cgem Geometry General for EventDisplay
 *                  Inherit from class SubDetectorROOTGeo
*/
 
#ifndef CGEM_ROOT_GEO_H
#define CGEM_ROOT_GEO_H
 
#include <TGeoVolume.h>
//#include <TGeoNode.h>
//#include <TGeoPhysicalNode.h>
#include <TPolyLine3D.h>
 
#include "SubDetectorROOTGeo.h"
#include "Cgem2DStrip.h"
 
/*
 * Class CgemROOTGeo contains all of the objects necessary to describe the CGEM geometry.
 * Author: Long Peixun
*/
 
class CgemROOTGeo : public SubDetectorROOTGeo
{
public:
    //Constructor
    CgemROOTGeo();
    //Destructor
    virtual ~CgemROOTGeo();
 
    //Initialize ROOTGeo from GDML
    void InitFromGDML(const char *gdmlFile, const char *setupName);
    //Initialize ROOTGeo from TGeoVolume CGEM
    void InitFromROOT(TGeoVolume *vol);
 
    //Initialize Geometry
    void InitGeometry();
 
    //Set the pointers to theirs nodes
    void SetNode();
    //Set default visual attributes
    void SetVolumeDefaultVis();
    //Set default 3D color
    void SetPhysicalDefaultVis();
 
    //Set all visible;
    void SetAllVisible();
    //Set quater visible
    void SetQuarterVisible();
    //Set half visible;
    void SetHalfVisible();
 
    //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();
    //Long Peixun's update: clear all hits
    void ClearHits();
 
    //Set CGEM default detector visibility
    void SetVisCgemDetector();
    //Set CGEM hits visibility
    void SetVisCgemHits();
 
    //Set CGEM volume, while initializing from ROOT
    void SetVolumeCgem(TGeoVolume *vol) { m_CGEM = vol; }
    //Get CGEM volume
    TGeoVolume *GetVolumeCgem() const { return m_CGEM; }
    //Get 2D strip
    Cgem2DStripBase* Get2DStrip(bool is_x, int layer, int sheet, int strip) const;
 
    //Get CGEM's anode, cathode and foil
    TGeoNode* GetCgemAnode(int layer);
    TGeoNode* GetCgemCathode(int layer);
    TGeoNode* GetCgemFoil(int layer, int foil);
 
    //Draw function
    void Draw(Option_t *option);
    //Draw unfolded function
    void DrawUF(int layer, Option_t *option = "");
    //Draw 2D hits
    void DrawHits(Option_t *option);
    //Draw 3D hits
    void Draw3DHits(Option_t *option);
 
    //These 4 functions are used for CGEM clusters calculation in BesEvent
    double calcZFromLayerSheetVIDPhi(int layer, int sheet, double v, double phi) const;
    double getRFromLayerSheet(int layer, int sheet) const;
    double getSheetWidth(int layer, int sheet) const;
    double isPointInSheet(int layer, int sheet, double phi, double z) const;
private:
    //Constants of CGEM
    static const int m_kCgemLayer = 3;
    static const int m_kCgemFoil = 3;
 
    //CGEM nodes
    TGeoVolume   *m_CGEM;
    TGeoNode     *m_NodeCgemAnode[m_kCgemLayer];
    TGeoNode     *m_NodeCgemCathode[m_kCgemLayer];
    TGeoNode     *m_NodeCgemFoil[m_kCgemLayer][m_kCgemFoil];
 
    //CGEM strip colors
    int m_CgemColor;
    int m_XstripColor;      //3D X-strip color
    int m_VstripColor;      //3D V-strip color
 
    //CGEM geometry information
    Double_t     RX[m_kCgemLayer][2];
    Double_t     RV[m_kCgemLayer][2];
    Double_t     StereoAngle[m_kCgemLayer][2];
    Double_t     SheetLength[m_kCgemLayer][2];
    Double_t     SheetWidth[m_kCgemLayer][2];
    Double_t     XMin[m_kCgemLayer][2];
    Double_t     ZMin[m_kCgemLayer][2];
    Double_t     XPitch[m_kCgemLayer][2];
    Double_t     VPitch[m_kCgemLayer][2];
    Int_t        NXStrip[m_kCgemLayer][2];      //X-strips number of each layer and sheet
    Int_t        NVStrip[m_kCgemLayer][2];      //V-strips number of each layer and sheet
    Double_t     *XStripPhi[m_kCgemLayer][2];   //phi of each X-strip
    Double_t     *VStripV[m_kCgemLayer][2];     //New Geom: V of each V-strip
 
    //CGEM 2D geometry
    //XY:
    BesCircle2D  *m_CgemXY;
    BesCircle2D  *m_CgemCathodeXY[m_kCgemLayer];
    BesCircle2D  *m_CgemAnodeXY[m_kCgemLayer];
    BesCircle2D  *m_CgemFoilXY[m_kCgemLayer][m_kCgemFoil];
    Cgem2DXStrip **m_CgemXStrip[m_kCgemLayer][2];
 
    //ZR:
    BesPolygon2D *m_CgemZR[2];
    BesPolygon2D *m_CgemCathodeZR[m_kCgemLayer][2];
    BesPolygon2D *m_CgemAnodeZR[m_kCgemLayer][2];
    BesPolygon2D *m_CgemFoilZR[m_kCgemLayer][m_kCgemFoil][2];
    Cgem2DVStrip **m_CgemVStrip[m_kCgemLayer][2];
 
    //CGEM unfolded geometry
    BesPolygon2D *m_CgemSheet[m_kCgemLayer][2];
 
    //Hits array
    const TObjArray* m_CgemDigiCol;
};
 
#endif  //CGEM_ROOT_GEO_H