ARICHGeoDetectorPlane.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Luka Santelj                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/logging/Logger.h>
#include <arich/dbobjects/ARICHGeoBase.h>
#include <string>
#include <TVector3.h>
#include <TRotation.h>
 
 
namespace Belle2 {
  class ARICHGeoDetectorPlane: public ARICHGeoBase {
 
  public:
 
    ARICHGeoDetectorPlane()
    {};
 
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "Detector plane parameters") const override;
 
    void setPlacement(double x, double y, double z, double rx, double ry, double rz) {m_x = x; m_y = y; m_z = z; m_rx = rx; m_ry = ry; m_rz = rz;}
 
    void addSupportPlate(double inR, double outR, double thick, const std::string& material)
    {
      m_innerR = inR;
      m_outerR = outR;
      m_thickness = thick;
      m_supportMaterial = material;
    };
 
    void setModuleHoleSize(double hsize) {m_moduleHoleSize = hsize;}
 
    void setSupportBackWallHeight(double h) {m_supportBackWallHeight = h;}
 
    void setSupportBackWallThickness(double d) {m_supportBackWallThickness = d;}
 
    void setSupportZPosition(double zPosition) { m_supportZPosition = zPosition; }
 
    void setRingPar(const std::vector<std::pair<double, double>>& ringPar);
 
    TVector3 getPosition() const {return TVector3(m_x / s_unit, m_y / s_unit, m_z / s_unit);}
 
    TRotation getRotation() const
    {
      TRotation rot;
      rot.RotateX(m_rx).RotateY(m_ry).RotateZ(m_rz);
      return rot;
    }
 
    double getRotationX() const {return m_rx;}
 
    double getRotationY() const {return m_ry;}
 
    double getRotationZ() const {return m_rz;}
 
    double getSupportInnerR() const {return m_innerR / s_unit;};
 
    double getSupportOuterR() const {return m_outerR / s_unit;};
 
    double getSupportThickness() const {return m_thickness / s_unit;};
 
    double getSupportZPosition() const {return m_supportZPosition / s_unit;};
 
    double getModuleHoleSize() const {return m_moduleHoleSize / s_unit;};
 
    double getSupportBackWallHeight() const {return m_supportBackWallHeight / s_unit;};
 
    double getSupportBackWallThickness() const {return m_supportBackWallThickness / s_unit;};
 
    const std::string& getSupportMaterial() const {return m_supportMaterial;}
 
    double getRingR(unsigned iRing) const { if (iRing == 0 || iRing > m_nRings) B2ERROR("ARICHGeoDetectorPlane: invalid module ring number!"); return m_ringPar[iRing - 1].first / s_unit;}
 
    double getRingDPhi(unsigned iRing) const { if (iRing == 0 || iRing > m_nRings) B2ERROR("ARICHGeoDetectorPlane: invalid module ring number!"); return m_ringPar[iRing - 1].second;}
 
    double getSlotR(unsigned modID) const { if (modID == 0 || modID > m_nSlots) B2ERROR("ARICHGeoDetectorPlane: invalid module ID number!"); return m_slotPar[modID - 1].first / s_unit;};
 
    double getSlotPhi(unsigned modID) const { if (modID == 0 || modID > m_nSlots) B2ERROR("ARICHGeoDetectorPlane: invalid module ID number!"); return m_slotPar[modID - 1].second;};
 
    unsigned getNSlots() const {return m_nSlots;}
 
    unsigned getNRings() const {return m_nRings;}
 
    unsigned getSlotRing(unsigned slotID) const;
 
    unsigned pointSlotID(double x, double y) const;
 
    unsigned getSlotIDFromSRF(unsigned sector, unsigned ring, unsigned azimuth) const;
 
  private:
 
    std::vector<std::pair<double, double>> m_ringPar; 
    std::vector<std::pair<double, double>> m_slotPar; 
    // detector plane position in local ARICH volume
    double m_x = 0; 
    double m_y = 0; 
    double m_z =
      0; 
    // rotations in local ARICH volume
    double m_rx = 0; 
    double m_ry = 0; 
    double m_rz = 0; 
    // support structure parameters
    std::string m_supportMaterial; 
    double m_innerR = 0;        
    double m_outerR = 0;        
    double m_thickness = 0;     
    double m_moduleHoleSize = 0;  
    double m_supportBackWallHeight = 0; 
    double m_supportBackWallThickness = 0;  
    double m_supportZPosition =
      0; 
    unsigned m_nRings = 0;        
    unsigned m_nSlots = 0;        
    ClassDefOverride(ARICHGeoDetectorPlane, 1); 
  };
 
} // end namespace Belle2