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):
        m_id(id), m_rmin1(rmin1), m_rmin2(rmin2), m_rmax1(rmax1),
        m_rmax2(rmax2), m_thick(thick), m_z(z)
      {}
 
 
      ~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;}
 
 
 
    private:
      int m_id; 
      double m_rmin1; 
      double m_rmin2; 
      double m_rmax1; 
      double m_rmax2; 
      double m_thick;
      double m_z;
      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