ARICHGeoMirrors.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 <framework/gearbox/Unit.h>
 
#include <arich/dbobjects/ARICHGeoBase.h>
#include <string>
#include <TVector3.h>
#include <TRotation.h>
#include <geometry/dbobjects/GeoOpticalSurface.h>
 
namespace Belle2 {
  class ARICHGeoMirrors: public ARICHGeoBase {
 
  public:
 
    ARICHGeoMirrors()
    {}
 
    bool isConsistent() const override;
 
    void print(const std::string& title = "Mirror configuration parameters") const override;
 
    void setMirrorPlate(double thick, double width, double length, const std::string& material, const GeoOpticalSurface& surf)
    {
      m_plateThickness = thick;
      m_plateWidth = width;
      m_plateLength = length;
      m_material = material;
      m_surface = surf;
    };
 
    void setNMirrors(unsigned nmir) {m_nMirrors = nmir;}
 
    void setRadius(double r) {m_radius = r;}
 
    void setStartAngle(double stang) {m_startAngle = stang;}
 
    void setZPosition(double zPos) {m_z = zPos;}
 
    double getZPosition() const {return m_z / s_unit;}
 
    double getRadius() const {return m_radius / s_unit;};
 
    double getStartAngle() const {return m_startAngle;};
 
    double getPlateThickness() const {return m_plateThickness / s_unit;};
 
    double getPlateWidth() const {return m_plateWidth / s_unit;};
 
    double getPlateLength() const {return m_plateLength / s_unit;};
 
    unsigned getNMirrors() const {return m_nMirrors;}
 
    void setMirrorNormVector(unsigned mirID, double x, double y, double z)
    {
      if (mirID < 1 || mirID > m_normVector.size()) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      m_normVector[mirID - 1] = TVector3(x, y, z).Unit();
    }
 
    void setMirrorPoint(unsigned mirID, double x, double y, double z)
    {
      if (mirID < 1 || mirID > m_nMirrors) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      m_point[mirID - 1] = TVector3(x, y, z);
    }
 
    const std::string& getMaterial() const {return m_material;}
 
    const GeoOpticalSurface& getMirrorSurface() const {return m_surface;}
 
    const TVector3& getNormVector(const unsigned mirID) const
    {
      if (mirID < 1 || mirID > m_nMirrors) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      return m_normVector[mirID - 1];
    }
 
    const TVector3& getPoint(const unsigned mirID) const
    {
      if (mirID < 1 || mirID > m_nMirrors) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      return m_point[mirID - 1];
    }
 
    void initializeDefault();
 
    void setInstalled(unsigned mirID, bool installed)
    {
      if (mirID < 1 || mirID > m_nMirrors) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      m_installed[mirID - 1] = installed;
    }
 
    bool isInstalled(unsigned mirID) const
    {
      if (mirID < 1 || mirID > m_nMirrors) B2ERROR("ARICHGeoMirrors: invalid mirror plate ID number");
      return m_installed[mirID - 1];
    }
 
 
  private:
 
    std::vector<TVector3> m_normVector;    
    std::vector<TVector3> m_point;         
    std::vector<bool> m_installed;         
    std::string m_material;                
    double m_z = 0;                            
    double m_radius = 0;                   
    double m_startAngle = 0;               
    double m_plateThickness = 0;           
    double m_plateWidth = 0;               
    double m_plateLength = 0;              
    unsigned m_nMirrors = 0;               
    GeoOpticalSurface m_surface;           
    ClassDefOverride(ARICHGeoMirrors, 1); 
  };
 
} // end namespace Belle2