ARICHGeoSupport.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>
 
namespace Belle2 {
  class ARICHGeoSupport: public ARICHGeoBase {
  public:
 
    ARICHGeoSupport()
    {}
 
    struct tube {
      double innerR;
      double outerR;
      double length;
      double zPosition;
      std::string material;
      std::string name;
    };
 
    struct wedge {
      int type;
      double r;
      double phi;
      double z;
      std::string material;
    };
 
    struct box {
      double size[3];
      double position[3];
      double rotation[3];
      std::string name;
      std::string material;
    };
 
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "Parameters of ARICH support structure and neutron shield") const override;
 
    void setWedge(unsigned i, const std::vector<double>& par) {if (i > m_nWedgeType || i == 0) { B2ERROR("invalid wedge number"); return;}; m_wedgePar[i - 1] = par;};
 
    void addWedge(int type, double r, double phi, double z, const std::string& material)
    {
      m_wedges.push_back({type, r, phi, z, material});
    }
 
    void setMaterial(const std::string& material) { m_material = material;};
 
    void addTube(double inR, double outR, double length, double zPosition, const std::string& material, const std::string& name)
    {
      m_tubes.push_back({inR, outR, length, zPosition, material, name});
    }
 
    unsigned getNTubes() const {return m_tubes.size();};
 
    double getTubeInnerR(unsigned i) const {return m_tubes.at(i).innerR / s_unit;}
 
    double getTubeOuterR(unsigned i) const {return m_tubes.at(i).outerR / s_unit;}
 
    double getTubeLength(unsigned i) const {return m_tubes.at(i).length / s_unit;}
 
    double getTubeZPosition(unsigned i) const {return m_tubes.at(i).zPosition / s_unit;}
 
    unsigned getNWedges() const {return m_wedges.size();};
 
    int getWedgeType(unsigned i) const {return  m_wedges.at(i).type;}
 
    double getWedgeR(unsigned i) const {return  m_wedges.at(i).r / s_unit;}
 
    double getWedgePhi(unsigned i) const {return  m_wedges.at(i).phi;}
 
    double getWedgeZ(unsigned i) const {return  m_wedges.at(i).z / s_unit;}
 
    const std::string& getTubeMaterial(unsigned i) const {return m_tubes.at(i).material;}
 
    const std::string& getWedgeMaterial(unsigned i) const {return m_wedges.at(i).material;}
 
    const std::string& getTubeName(unsigned i) const {return m_tubes.at(i).name;}
 
    const std::string& getMaterial() const {return m_material;}
 
    const std::vector<double> getWedge(unsigned i) const {std::vector<double> pars; if (i > m_nWedgeType || i == 0) { B2ERROR("invalid wedge number"); return pars;}; for (auto par : m_wedgePar[i - 1]) pars.push_back(par / s_unit); return pars;}
 
    void addBox(const std::string& name, const std::string& material, double size[3], double position[3], double rotation[3])
    {
      m_boxes.push_back({{size[0], size[1], size[2]}, {position[0], position[1], position[2]}, {rotation[0], rotation[1], rotation[2]}, name, material});
    }
 
    void setBox(int i, const std::string& name, const std::string& material, double size[3], double position[3], double rotation[3])
    {
      m_boxes[i] = {{size[0], size[1], size[2]}, {position[0], position[1], position[2]}, {rotation[0], rotation[1], rotation[2]}, name, material};
    }
 
    unsigned getNBoxes() const
    {
      return m_boxes.size();
    }
 
    box getBox(unsigned i) const
    {
      return m_boxes[i];
    }
 
    void clearBoxes()
    {
      m_boxes.clear();
    }
 
 
 
  private:
 
    std::vector<tube> m_tubes;           
    std::string m_material;              
    static const unsigned m_nWedgeType = 3; 
    std::vector<double> m_wedgePar[m_nWedgeType];   
    std::vector<wedge> m_wedges;         
    std::vector<box> m_boxes;            
    ClassDefOverride(ARICHGeoSupport, 1); 
  };
 
} // end namespace Belle2