CDCGeometry.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <vector>
#include <string>
 
#include <TObject.h>
 
namespace Belle2 {
 
  class GearDir;

 
  class CDCGeometry: public TObject {
 
  public:
 

    class Rib : public TObject {
 
    public:

      Rib() {}

      Rib(int id, double length, double width, double thick,
          double rotx, double roty, double rotz,
          double x, double y, double z,
          int offset, int ndiv):
        m_id(id), m_length(length), m_width(width), m_thick(thick),
        m_rotx(rotx), m_roty(roty), m_rotz(rotz), m_x(x), m_y(y), m_z(z),
        m_offset(offset), m_ndiv(ndiv)
      {}
 

      ~Rib() {}

      int getId() const {return m_id;}
 

      double getLength() const { return m_length;}

      double getWidth() const { return m_width;}

      double getThick() const { return m_thick;}

      double getRotX() const { return m_rotx;}
      double getRotY() const { return m_roty;}
      double getRotZ() const { return m_rotz;}

      double getX() const { return m_x;}
      double getY() const { return m_y;}
      double getZ() const { return m_z;}

      int getOffset() const { return m_offset;}
      int getNDiv() const { return m_ndiv;}
 
 
    private:
      int m_id; 
      double m_length; 
      double m_width; 
      double m_thick; 
      double m_rotx; 
      double m_roty;
      double m_rotz;
      double m_x; 
      double m_y;
      double m_z;
      int m_offset; 
      int m_ndiv; 
 

      ClassDef(Rib, 2);
    };

    class Rib2 : public TObject {
 
    public:

      Rib2() {}

      Rib2(int id, double length, double width, double thick,
           double width2, double thick2, double rotx, double roty, double rotz,
           double x, double y, double z,
           int ndiv):
        m_id(id), m_length(length), m_width(width), m_thick(thick),
        m_width2(width2), m_thick2(thick2),
        m_rotx(rotx), m_roty(roty), m_rotz(rotz),
        m_x(x), m_y(y), m_z(z), m_ndiv(ndiv)
      {}
 

      ~Rib2() {}

      int getId() const {return m_id;}
 

      double getLength() const { return m_length;}

      double getWidth() const { return m_width;}

      double getThick() const { return m_thick;}

      double getWidth2() const { return m_width2;}

      double getThick2() const { return m_thick2;}

      double getRotX() const { return m_rotx;}

      double getRotY() const { return m_roty;}

      double getRotZ() const { return m_rotz;}

      double getX() const { return m_x;}
      double getY() const { return m_y;}
      double getZ() const { return m_z;}

      int getNDiv() const { return m_ndiv;}
 
 
    private:
      int m_id; 
      double m_length; 
      double m_width; 
      double m_thick; 
      double m_width2; 
      double m_thick2; 
      double m_rotx; 
      double m_roty; 
      double m_rotz; 
      double m_x; 
      double m_y;
      double m_z;
      int m_ndiv; 
 

      ClassDef(Rib2, 1);
    };

    class Rib3 : public TObject {
 
    public:

      Rib3() {}

      Rib3(int id, double length, double width, double thick,
           double r, double x, double y, double z,
           double rx, double ry, double rz, int offset,
           int ndiv):
        m_id(id), m_length(length), m_width(width), m_thick(thick),
        m_r(r), m_x(x), m_y(y), m_z(z),
        m_rx(rx), m_ry(ry), m_rz(rz), m_offset(offset), m_ndiv(ndiv)
      {}
 

      ~Rib3() {}

      int getId() const {return m_id;}
 

      double getLength() const { return m_length;}

      double getWidth() const { return m_width;}

      double getThick() const { return m_thick;}

      double getR() const { return m_r;}

      double getX() const { return m_x;}
      double getY() const { return m_y;}
      double getZ() const { return m_z;}

      double getRx() const { return m_rx;}
      double getRy() const { return m_ry;}
      double getRz() const { return m_rz;}

      int getOffset() const { return m_offset;}

      int getNDiv() const { return m_ndiv;}
 
 
    private:
      int m_id; 
      double m_length; 
      double m_width; 
      double m_thick; 
      double m_r; 
      double m_x; 
      double m_y;
      double m_z;
      double m_rx; 
      double m_ry;
      double m_rz;
      int m_offset; 
      int m_ndiv; 
 

      ClassDef(Rib3, 1);
    };

    class Rib4 : public TObject {
 
    public:

      Rib4() {}

      Rib4(int id, double length, double width, double thick,
           double length2, double width2, double thick2,
           double x, double y, double z,
           double x2, double y2, double z2, int offset,
           int ndiv):
        m_id(id), m_length(length), m_width(width), m_thick(thick),
        m_length2(length2), m_width2(width2), m_thick2(thick2),
        m_x(x), m_y(y), m_z(z),
        m_x2(x2), m_y2(y2), m_z2(z2), m_offset(offset), m_ndiv(ndiv)
      {}
 

      ~Rib4() {}

      int getId() const {return m_id;}
 

      double getLength() const { return m_length;}

      double getWidth() const { return m_width;}

      double getThick() const { return m_thick;}

      double getLength2() const { return m_length2;}

      double getWidth2() const { return m_width2;}

      double getThick2() const { return m_thick2;}

      double getX() const { return m_x;}
      double getY() const { return m_y;}
      double getZ() const { return m_z;}

      double getX2() const { return m_x2;}
      double getY2() const { return m_y2;}
      double getZ2() const { return m_z2;}

      int getOffset() const { return m_offset;}

      int getNDiv() const { return m_ndiv;}
 
 
    private:
      int m_id; 
      double m_length; 
      double m_width; 
      double m_thick; 
      double m_length2; 
      double m_width2; 
      double m_thick2; 
      double m_x; 
      double m_y;
      double m_z;
      double m_x2; 
      double m_y2;
      double m_z2;
      int m_offset; 
      int m_ndiv; 
 

      ClassDef(Rib4, 1);
    };

    class Rib5 : public TObject {
 
    public:

      Rib5() {}

      Rib5(int id, double dr, double dz, double width, double thick, double rin,
           double x, double y, double z,
           double rotx, double roty, double rotz, int offset,
           int ndiv):
        m_id(id), m_dr(dr), m_dz(dz), m_width(width), m_thick(thick),
        m_rin(rin), m_x(x), m_y(y), m_z(z),
        m_rotx(rotx), m_roty(roty), m_rotz(rotz), m_offset(offset), m_ndiv(ndiv)
      {}
 

      ~Rib5() {}

      int getId() const {return m_id;}
 

      double getDr() const { return m_dr;}

      double getDz() const { return m_dz;}

      double getWidth() const { return m_width;}

      double getThick() const { return m_thick;}

      double getRin() const { return m_rin;}

      double getX() const { return m_x;}
      double getY() const { return m_y;}
      double getZ() const { return m_z;}

      double getRotx() const { return m_rotx;}
      double getRoty() const { return m_roty;}
      double getRotz() const { return m_rotz;}

      int getOffset() const { return m_offset;}

      int getNDiv() const { return m_ndiv;}
 
 
    private:
      int m_id; 
      double m_dr; 
      double m_dz; 
      double m_width; 
      double m_thick; 
      double m_rin; 
      double m_x; 
      double m_y;
      double m_z;
      double m_rotx; 
      double m_roty;
      double m_rotz;
      int m_offset; 
      int m_ndiv; 
 

      ClassDef(Rib5, 1);
    };

    class Cover : public TObject {
 
    public:

      Cover() {}

      Cover(int id, double rmin1, double rmin2,
            double rmax1, double rmax2,
            double thick, double z):
        m_id(id), m_rmin1(rmin1), m_rmin2(rmin2), m_rmax1(rmax1),
        m_rmax2(rmax2), m_thick(thick), m_z(z)
      {}
 

      ~Cover() {}

      int getId() const {return m_id;}
      double getRmin1() const { return m_rmin1;}

      double getRmin2() const { return m_rmin2;}

      double getRmax1() const { return m_rmax1;}

      double getRmax2() const { return m_rmax2;}

      double getThick() const { return m_thick;}

      double getZ() const { return m_z;}
 
 
 
    private:
      int m_id; 
      double m_rmin1; 
      double m_rmin2; 
      double m_rmax1; 
      double m_rmax2; 
      double m_thick;
      double m_z;
 

      ClassDef(Cover, 1);
    };

    class Cover2 : public TObject {
 
    public:

      Cover2() {}

      Cover2(int id, double rmin, double rmax,
             double phis, double dphi,
             double thick, double z):
        m_id(id), m_rmin(rmin), m_rmax(rmax), m_phis(phis),
        m_dphi(dphi), m_thick(thick), m_z(z)
      {}
 

      ~Cover2() {}

      int getId() const {return m_id;}
      double getRmin() const { return m_rmin;}

      double getRmax() const { return m_rmax;}

      double getPhis() const { return m_phis;}

      double getDphi() const { return m_dphi;}

      double getThick() const { return m_thick;}

      double getZ() const { return m_z;}
 
 
 
    private:
      int m_id; 
      double m_rmin; 
      double m_rmax; 
      double m_phis; 
      double m_dphi; 
      double m_thick;
      double m_z;
 

      ClassDef(Cover2, 1);
    };

    class NeutronShield : public TObject {
 
    public:

      NeutronShield() {}

      NeutronShield(int id, double rmin1, double rmin2,
                    double rmax1, double rmax2,
                    double thick, double z, int ena):
        m_id(id), m_rmin1(rmin1), m_rmin2(rmin2), m_rmax1(rmax1),
        m_rmax2(rmax2), m_thick(thick), m_z(z), m_ena(ena)
      {}
 

      ~NeutronShield() {}

      int getId() const {return m_id;}
      double getRmin1() const { return m_rmin1;}

      double getRmin2() const { return m_rmin2;}

      double getRmax1() const { return m_rmax1;}

      double getRmax2() const { return m_rmax2;}

      double getThick() const { return m_thick;}

      double getZ() const { return m_z;}

      int getEna() const { return m_ena;}
 
 
    private:
      int m_id; 
      double m_rmin1; 
      double m_rmin2; 
      double m_rmax1; 
      double m_rmax2; 
      double m_thick;
      double m_z;
      int m_ena;
 

      ClassDef(NeutronShield, 1);
    };

    class Frontend : public TObject {
 
    public:

      Frontend() {}

      Frontend(int id, double rmin, double rmax, double zfwd, double zbwd):
        m_id(id), m_rmin(rmin), m_rmax(rmax), m_zfwd(zfwd), m_zbwd(zbwd)
      {}
 

      ~Frontend() {}
      int getId() const {return m_id;}
      double getRmin() const { return m_rmin;}
      double getRmax() const { return m_rmax;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}
 
 
    private:
      int m_id; 
      double m_rmin; 
      double m_rmax; 
      double m_zfwd;
      double m_zbwd;

      ClassDef(Frontend, 1);
    };
 

    class MotherVolume : public TObject {
 
    public:

      MotherVolume() {}

      ~MotherVolume() {}

      void appendNode(double rmin, double rmax, double z);

      int getNNodes() const { return m_rmin.size();}
      std::vector<double> getRmin() const { return m_rmin;}
      std::vector<double> getRmax() const { return m_rmax;}
      std::vector<double> getZ() const { return m_z;}
 
 
    private:
      std::vector<double> m_rmin; 
      std::vector<double> m_rmax; 
      std::vector<double> m_z; 

      ClassDef(MotherVolume, 1);
    };

    class SenseLayer : public TObject {
 
    public:

      SenseLayer()
      {}

      SenseLayer(int id,
                 double r, double zfwd, double zbwd,
                 double dZfwd, double dZbwd,
                 double offset, int nWires, double nShifts) :
        m_id(id), m_r(r),
        m_zfwd(zfwd), m_zbwd(zbwd),
        m_dZfwd(dZfwd), m_dZbwd(dZbwd),
        m_offset(offset),
        m_nWires(nWires),
        m_nShifts(nShifts)
      {
      }

      ~SenseLayer() {}

      int  getId() const { return m_id;}

      double getR() const {return m_r;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}

      void setDZfwd(double dz) {m_dZfwd = dz;}

      double getDZfwd() const {return m_dZfwd;}

      void setDZbwd(double dz) {m_dZbwd = dz;}

      double getDZbwd() const {return m_dZbwd;}

      void setOffset(double offset) {m_offset = offset;}

      double getOffset() const {return m_offset;}

      void setNShifts(int n) {m_nShifts = n;}

      int getNShifts() const {return m_nShifts;}

      void setNWires(int n) {m_nWires = n;}

      int  getNWires() const { return m_nWires;}
 
 
    private:
 
      int m_id; 
      double m_r;     
      double m_zfwd;     
      double m_zbwd;     
      double m_dZfwd;   
      double m_dZbwd;   
      double m_offset;  
      int m_nWires; 
      double m_nShifts; 

      ClassDef(SenseLayer, 1);
    };

    class FieldLayer : public TObject {
 
    public:

      FieldLayer() {}

      FieldLayer(int id,
                 double r, double zfwd, double zbwd):
        m_id(id), m_r(r),
        m_zfwd(zfwd), m_zbwd(zbwd)
      {
      }

      ~FieldLayer() {}

      int  getId() const { return m_id;}

      double getR() const {return m_r;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}
 
    private:
 
      int m_id; 
      double m_r;     
      double m_zfwd;     
      double m_zbwd;     

      ClassDef(FieldLayer, 1);
    };

    class OuterWall : public TObject {
 
    public:

      OuterWall() {}

      OuterWall(const std::string& name, int id,
                double rmin, double rmax, double zfwd, double zbwd):
        m_name(name), m_id(id), m_rmin(rmin),
        m_rmax(rmax), m_zfwd(zfwd), m_zbwd(zbwd)
      {
      }

      ~OuterWall() {}

      std::string getName() const { return m_name;}

      int  getId() const { return m_id;}

      double getRmin() const { return m_rmin;}

      double getRmax() const {return m_rmax;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}
 
    private:
 
      std::string m_name; 
      int m_id; 
      double m_rmin;     
      double m_rmax;     
      double m_zfwd;     
      double m_zbwd;     
 

      ClassDef(OuterWall, 1);
    };

    class InnerWall : public TObject {
 
    public:

      InnerWall() {}

      InnerWall(const std::string& name, int id,
                double rmin, double rmax, double zfwd, double zbwd):
        m_name(name), m_id(id), m_rmin(rmin),
        m_rmax(rmax), m_zfwd(zfwd), m_zbwd(zbwd)
      {
      }

      ~InnerWall() {}

      std::string getName() const { return m_name;}

      int  getId() const { return m_id;}

      double getRmin() const { return m_rmin;}

      double getRmax() const {return m_rmax;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}
 
    private:
 
      std::string m_name; 
      int m_id; 
      double m_rmin;     
      double m_rmax;     
 
      double m_zfwd;     
      double m_zbwd;     
 

      ClassDef(InnerWall, 1);
    };

    class EndPlateLayer : public TObject {
 
    public:

      EndPlateLayer() {}

      EndPlateLayer(const std::string& name, int ilay,
                    double rmin, double rmax, double zfwd, double zbwd):
        m_name(name), m_layer(ilay), m_rmin(rmin),
        m_rmax(rmax), m_zfwd(zfwd), m_zbwd(zbwd)
      {
      }

      ~EndPlateLayer() {}

      std::string getName() const { return m_name;}

      int getILayer() const { return m_layer;}

      double getRmin() const { return m_rmin;}

      double getRmax() const {return m_rmax;}

      double getZfwd() const { return m_zfwd;}

      double getZbwd() const { return m_zbwd;}
 
    private:
 
      std::string m_name; 
      int m_layer; 
      double m_rmin;     
      double m_rmax;     
      double m_zfwd;     
      double m_zbwd;     

      ClassDef(EndPlateLayer, 1);
    };

    class EndPlate : public TObject {
 
    public:
      EndPlate() {}

      explicit EndPlate(int i) :  m_id(i) {};

      ~EndPlate() {}

      void appendNew(const std::string& name, int ilay,
                     double rmin, double rmax, double zfwd, double zbwd)
      {
        EndPlateLayer eplayer(name, ilay, rmin, rmax, zfwd, zbwd);
        m_endPlateLayers.push_back(eplayer);
      }

      int getNEndPlateLayers() const {return m_endPlateLayers.size();}

      int getId() const {return m_id;}

      EndPlateLayer getEndPlateLayer(int i) const {return m_endPlateLayers.at(i);}
      std::vector<EndPlateLayer> getEndPlateLayers() const {return m_endPlateLayers;}
    private:
      int m_id; 
      std::vector<EndPlateLayer> m_endPlateLayers; 

      ClassDef(EndPlate, 1);
    };

    CDCGeometry() {};

    explicit CDCGeometry(const GearDir&);

    ~CDCGeometry();

    void read(const GearDir&);

    double getGlobalPhiRotation() const { return m_globalPhiRotation;}

    double getGlobalOffsetX() const { return m_globalOffsetX;}

    double getGlobalOffsetY() const { return m_globalOffsetY;}

    double getGlobalOffsetZ() const { return m_globalOffsetZ;}

    double getGlobalOffsetA() const { return m_globalOffsetA;}

    double getGlobalOffsetB() const { return m_globalOffsetB;}

    double getGlobalOffsetC() const { return m_globalOffsetC;}

    double getFiducialRmin() const { return m_fiducialRmin;}

    double getFiducialRmax() const { return m_fiducialRmax;}

    MotherVolume getMotherVolume() const { return m_mother;}

    std::vector<OuterWall> getOuterWalls() const { return m_outerWalls;}

    OuterWall getOuterWall(int i) const { return m_outerWalls.at(i);}

    std::vector<InnerWall> getInnerWalls() const { return m_innerWalls;}

    InnerWall getInnerWall(int i) const { return m_innerWalls.at(i);}

    int getNEndplates() const { return m_endplates.size();}

    EndPlate getEndPlate(int i) const {return m_endplates.at(i);};

    std::vector<EndPlate> getEndPlates() const { return m_endplates;}

    int getNFrontends() const {return m_frontends.size();}

    std::vector<Frontend> getFrontends() const {return m_frontends;}

    SenseLayer getSenseLayer(int i) const { return m_senseLayers.at(i);}

    std::vector<SenseLayer> getSenseLayers() const { return m_senseLayers;}

    int getNSenseLayers() const { return m_senseLayers.size();}

    double getSenseDiameter() const { return m_senseWireDiameter;}

    double getSenseTension() const { return m_senseWireTension;}

    int getNSenseWires() const { return m_senseWireNumbers;}

    FieldLayer getFieldLayer(int i) const { return m_fieldLayers.at(i);}

    std::vector<FieldLayer> getFieldLayers() const { return m_fieldLayers;}

    int getNFieldLayers() const { return m_fieldLayers.size();}
 

    std::vector<NeutronShield> getNeutronShields() const { return m_neutronShields;}

    std::vector<Cover> getCovers() const { return m_covers;}

    std::vector<Cover2> getCover2s() const { return m_cover2s;}

    std::vector<Rib> getRibs() const { return m_ribs;}

    std::vector<Rib2> getRib2s() const { return m_rib2s;}

    std::vector<Rib3> getRib3s() const { return m_rib3s;}

    std::vector<Rib4> getRib4s() const { return m_rib4s;}

    std::vector<Rib5> getRib5s() const { return m_rib5s;}
 

    double getFieldDiameter() const { return m_fieldWireDiameter;}

    int getNFieldWires() const { return m_fieldWireNumbers;}

    double getFeedthroughLength() const {return m_feedThroughLength;}

    double getClockFrequency() const { return m_clockFrequency;}

    double getNominalSpaceResolution() const { return m_nominalSpaceResolution;}
 

    ushort getNumberOfSenseLayers() const { return m_maxNSenseLayers;}

    ushort getNumberOfFieldLayers() const { return m_maxNFieldLayers;}

    ushort getMaxNumberOfSuperLayers() const { return m_maxNSuperLayers;}

    ushort getOffsetOfFirstLayer() const { return m_firstLayerOffset;}

    ushort getOffsetOfFirstSuperLayer() const { return m_firstSuperLayerOffset;}

    ushort getMaxNumberOfCellsPerLayer() const { return m_maxNCellsPerLayer;}
 
 
  private:
 
    double m_globalPhiRotation = 0.0; 
    double m_globalOffsetX = 0.0; 
    double m_globalOffsetY = 0.0; 
    double m_globalOffsetZ = 0.0; 
    double m_globalOffsetA = 0.0; 
    double m_globalOffsetB = 0.0; 
    double m_globalOffsetC = 0.0; 
    double m_fiducialRmin; 
    double m_fiducialRmax; 
    MotherVolume m_mother;    
    std::vector<SenseLayer> m_senseLayers; 
    std::vector<FieldLayer> m_fieldLayers; 
    std::vector<InnerWall> m_innerWalls;
    std::vector<OuterWall> m_outerWalls;
    std::vector<EndPlate> m_endplates;
    std::vector<Frontend> m_frontends;
    std::vector<NeutronShield> m_neutronShields;
    std::vector<Cover> m_covers;
    std::vector<Cover2> m_cover2s;
    std::vector<Rib> m_ribs;
    std::vector<Rib2> m_rib2s;
    std::vector<Rib3> m_rib3s;
    std::vector<Rib4> m_rib4s;
    std::vector<Rib5> m_rib5s;
 
    double m_senseWireDiameter = 0.003; 
    int m_senseWireNumbers = 14336;
    double m_senseWireTension = 50.0; 
    double m_fieldWireDiameter = 0.0126; 
    int m_fieldWireNumbers = 42240; 
    double m_feedThroughLength = 1.5; 
    double m_clockFrequency; 
    double m_nominalSpaceResolution;
 
    ushort m_maxNSenseLayers        = 56;   
    ushort m_maxNFieldLayers        = 55;   
    ushort m_maxNSuperLayers        = 9;    
    ushort m_firstLayerOffset       = 0;    
    ushort m_firstSuperLayerOffset  = 0;    
    ushort m_maxNCellsPerLayer      = 384;  
 
    ClassDef(CDCGeometry, 5);  
 
  };

} // end of namespace Belle2