Cgem2DStrip.h

Go to the documentation of this file.

#ifndef CGEM_2D_STRIP_H
#define CGEM_2D_STRIP_H
//Author: Long Peixun
 
#include <TNamed.h>
#include <TAttLine.h>
#include <TAttFill.h>
#include <TPaveLabel.h>
#include <TPaveText.h>
#include <TVector3.h>
 
#include "BesCircle2D.h"
#include "BesPolygon2D.h"
#include "BesMarker2D.h"
#include "BesPaveText.h"
 
//Enum of CGEM strip type
enum ECgemStripType
{
    CGEM_X_STRIP = 0,
    CGEM_V_STRIP = 1
};
 
//Cgem 2D Strip base class
class Cgem2DStripBase : public TNamed, public TAttLine, public TAttFill
{
public:
    Cgem2DStripBase(const char* name, const char* title, Int_t layer, Int_t sheet, Int_t strip);
    virtual ~Cgem2DStripBase();
 
    virtual void  SetStyle() {}
    virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) = 0;       //pure virtual
    virtual void  ExecuteEvent(Int_t event, Int_t px, Int_t py) = 0; //pure virtual
 
    virtual void  SetHighlighted(bool status = true) { fHighlighted = status; }
    virtual void  SetFired(bool status = true) { fFired = status; }
    virtual bool  IsHighlighted() const { return fHighlighted; }
    virtual bool  IsFired() const { return fFired; }
 
    virtual void  AddInfo(TString info) { fInfoCon.push_back(info); }
    virtual void  CloseInfo() = 0;                  //pure virtual
    virtual void  ClearInfo();
 
    virtual void Draw(Option_t *option = "") = 0;   //pure virtual
    virtual void Paint(Option_t *option = "") = 0;  //pure virtual
    virtual ECgemStripType GetType() const = 0;      //pure virtual
    virtual char* GetObjectInfo(Int_t px, Int_t py) const;
 
    virtual Int_t GetLayer() const { return fLayer; }
    virtual Int_t GetSheet() const { return fSheet; }
    virtual Int_t GetStrip() const { return fStrip; }
 
protected:
    std::vector<TString> fInfoCon;
    BesPaveText *fTip;
 
    Int_t fLayer;
    Int_t fSheet;
    Int_t fStrip;
    bool fHighlighted;
    bool fFired;
 
    //style
    Int_t lcStrip;  //line color
    Int_t lwStrip;  //line width
    Int_t fcStrip;  //fill color
    Int_t fsStrip;  //fill style
 
    Int_t lcStripFired;
    Int_t lwStripFired;
    Int_t fcStripFired;
    Int_t fsStripFired;
 
    Int_t lcStripHL;
    Int_t lwStripHL;
    Int_t fcStripHL;
    Int_t fsStripHL;
 
    ClassDef(Cgem2DStripBase, 1)
};
 
 
//----------------------------------------------------------------
//Cgem X-strip class
class Cgem2DXStrip : public Cgem2DStripBase
{
public:
    Cgem2DXStrip(const char* name, const char* title,
        Double_t len, Double_t wid, Double_t phi, Double_t zmin, Double_t r,
        Double_t size, Double_t thick, Int_t layer, Int_t sheet, Int_t strip,
        Double_t rUF);  //rUF: radius in unfolded view
    virtual ~Cgem2DXStrip();
 
    virtual void  CloseInfo();
    virtual void  SetStyle();
    virtual Int_t DistancetoPrimitive(Int_t px, Int_t py);
    virtual void  ExecuteEvent(Int_t event, Int_t px, Int_t py);
 
    virtual void  Draw(Option_t *option = "");
    virtual void  Paint(Option_t *option = "");
    virtual ECgemStripType GetType() const { return CGEM_X_STRIP; }
 
protected:
    BesPolygon2D* fXStripXY;
    BesPolygon2D* fXStripZR;
    BesPolygon2D* fXStripUF;
 
    BesMarker2D* fXYHighlight;
    Int_t cHLMarker;
    //fired X-strip will be drawn larger in XY view
    Double_t fFiredSizeFactor;   //fired X-strip size factor
 
    ClassDef(Cgem2DXStrip, 1)
};
 
 
//-----------------------------------------------------------------
//This class is used to calculate points of V-strip
class CgemVStripCalculator
{
private:
    double m_RV;        //Radius of V-strip
    double m_Length;    //Length of sheet
    double m_Width;     //Width of sheet
    double m_Angle;     //Stereo angle of V-strip
    double m_Xmin;      //Start X of sheet
    double m_Zmin;      //Start Z of sheet
    double m_Pitch;     //Pitch of V-strip
 
    double delta;       //Distance of neighbor points in drawing
public:
    //Return status code
    static const int NORMAL = 0;
    static const int INVALID_VID = -1;
    static const int ADJUST_TO_END = 1;
    static const int ADJUST_TO_START = 2;
 
    CgemVStripCalculator(double r, double len, double wid, double ang, double xmin, double zmin, double pit);
    void setDelta(double sd) { delta = sd; }
    double getPointPhi(double* pos) const;
    bool isPointInSheet(double phi, double z) const;
    double getZFromVPhi(double v, double phi) const;
 
    //Calculating by VID
    // int getStartVID() const;
    // int getStartPointFromVID(int VID, double& phi, double& z) const;
    // int getEndPointFromVID(int VID, double& phi, double& z) const;
    // double getStripLengthFromVID(int VID) const;
    // int getPointNumberFromVID(int VID) const;
    // int getNthPointFromVID(int VID, int i, double& phi, double& z) const;
 
    //Calculating by V
    int getStartPointFromV(double v, double& phi, double& z) const;
    int getEndPointFromV(double v, double& phi, double& z) const;
    double getStripLengthFromV(double v) const;
    int getPointNumberFromV(double v) const;
    int getNthPointFromV(double v, int i, double& phi, double& z) const;
};
 
 
//-----------------------------------------------------------------
//Cgem V-strip class
class Cgem2DVStrip : public Cgem2DStripBase
{
public:
    Cgem2DVStrip(const char* name, const char* title, double v,
        double r, double len, double wid, double ang, double xmin, double zmin, double pit,
        Double_t size, Double_t thick, Int_t layer, Int_t sheet, Int_t strip,
        Double_t rUF);  //rUF: radius in unfolded view
    virtual ~Cgem2DVStrip();
 
    virtual void  CloseInfo();
    virtual void  SetStyle();
    virtual Int_t DistancetoPrimitive(Int_t px, Int_t py);
    virtual void  ExecuteEvent(Int_t event, Int_t px, Int_t py);
 
    virtual void  Draw(Option_t *option = "");
    virtual void  Paint(Option_t *option = "");
    virtual ECgemStripType GetType() const { return CGEM_V_STRIP; }
 
protected:
    BesPolygon2D* fVStripXY;
    BesPolygon2D* fVStripZR;
    BesPolygon2D* fVStripUF;
 
    ClassDef(Cgem2DVStrip, 1)
};
 
#endif