ARICHPositionElement.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 <arich/dbobjects/ARICHGeoBase.h>
#include <framework/gearbox/Unit.h>
#include <TVector3.h>
#include <TRotation.h>
#include <iostream>
 
namespace Belle2 {
  class ARICHPositionElement: public ARICHGeoBase {
  public:
 
    ARICHPositionElement()
    {}
 
    ARICHPositionElement(double x, double y, double z,
                         double alpha, double beta, double gamma, const std::string& name = "ARICHAlignment"):
      ARICHGeoBase(name),
      m_x(x), m_y(y), m_z(z),
      m_alpha(alpha), m_beta(beta), m_gamma(gamma)
    {}
 
    ~ARICHPositionElement()
    {}
 
    void setX(double x) {m_x = x;}
 
    void setY(double y) {m_y = y;}
 
    void setZ(double z) {m_z = z;}
 
    void setRPhi(double r, double phi)
    {
      m_x = r * cos(phi / Unit::rad);
      m_y = r * sin(phi / Unit::rad);
    }
 
    void setAlpha(double alpha) {m_alpha = alpha;}
 
 
    void setBeta(double beta) {m_beta = beta;}
 
 
    void setGamma(double gamma) {m_gamma = gamma;}
 
    void addShift(double dx, double dy, double dz, double dalpha, double dbeta, double dgamma)
    {
      m_x += dx;
      m_y += dy;
      m_z += dz;
      m_alpha += dalpha;
      m_beta += dbeta;
      m_gamma += dgamma;
    };
 
    double getX() const {return m_x / s_unit;}
 
    double getY() const {return m_y / s_unit;}
 
    double getZ() const {return m_z / s_unit;}
 
    double getAlpha() const {return  m_alpha;}
 
    double getBeta() const {return m_beta;}
 
    double getGamma() const {return m_gamma;}
 
    TRotation getRotation() const
    {
      TRotation rot;
      rot.RotateX(m_alpha).RotateY(m_beta).RotateZ(m_gamma);
      return rot;
    }
 
    TVector3 getTranslation() const {return TVector3(m_x, m_y, m_z);}
 
    bool isConsistent() const override {return true;}
 
    void print(const std::string& title = "Parmeters of position element") const override
    {
      std::cout << title << std::endl;
      std::cout << "translations (x,y,z): " << m_x << " " << m_y << " " << m_z << std::endl;
      std::cout << "rotations (x,y,z axis): " << m_alpha << " " << m_beta << " " << m_gamma << std::endl;
 
    }
 
  private:
 
    float m_x = 0; 
    float m_y = 0; 
    float m_z = 0; 
    float m_alpha = 0; 
    float m_beta = 0;  
    float m_gamma = 0; 
    ClassDefOverride(ARICHPositionElement, 1); 
  };
 
} // end namespace Belle2