ReferenceFrame.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 <Math/Rotation3D.h>
#include <Math/Boost.h>
#include <Math/Vector3D.h>
#include <Math/Vector4D.h>
#include <analysis/dataobjects/Particle.h>
#include <analysis/utility/PCmsLabTransform.h>
 
#include <stack>
 
namespace Belle2 {
  class ReferenceFrame {
  public:
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const = 0;
 
    virtual ROOT::Math::XYZVector getVertex(const Particle* particle) const
    {
      return getVertex(particle->getVertex());
    }
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const = 0;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const Particle* particle) const
    {
      return getMomentum(particle->get4Vector());
    }
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const = 0;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const Particle* particle) const
    {
      return getMomentumErrorMatrix(particle->getMomentumErrorMatrix());
    }
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const = 0;
 
    virtual TMatrixFSym getVertexErrorMatrix(const Particle* particle) const
    {
      return getVertexErrorMatrix(particle->getVertexErrorMatrix());
    }
 
    static const ReferenceFrame& GetCurrent();
 
  private:
    static void Push(const ReferenceFrame* frame)
    {
      m_reference_frames.push(frame);
    }
 
    static void Pop()
    {
      m_reference_frames.pop();
    }
 
    static std::stack<const ReferenceFrame*> m_reference_frames; 
    template<class T>
    friend class UseReferenceFrame;
  };
 
  class RestFrame : public ReferenceFrame {
 
  public:
    explicit RestFrame(const Particle* particle);
 
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const override;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const override;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const override;
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const override;
 
  private:
    ROOT::Math::PxPyPzEVector m_momentum; 
    ROOT::Math::XYZVector m_displacement; 
    ROOT::Math::XYZVector m_boost;        
    ROOT::Math::Boost m_lab2restframe;    
  };
 
  class LabFrame : public ReferenceFrame {
 
  public:
 
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const override;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const override;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const override;
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const override;
 
  };
 
  class CMSFrame : public ReferenceFrame {
 
  public:
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const override;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const override;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const override;
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const override;
 
  private:
    PCmsLabTransform m_transform; 
  };
 
  class RotationFrame : public ReferenceFrame {
 
  public:
    explicit RotationFrame(const ROOT::Math::XYZVector& newX, const ROOT::Math::XYZVector& newY, const ROOT::Math::XYZVector& newZ);
 
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const override;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const override;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const override;
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const override;
 
  private:
    ROOT::Math::Rotation3D m_rotation; 
  };
 
  class CMSRotationFrame : public ReferenceFrame {
 
  public:
    explicit CMSRotationFrame(const ROOT::Math::XYZVector& newX, const ROOT::Math::XYZVector& newY, const ROOT::Math::XYZVector& newZ);
 
    virtual ROOT::Math::XYZVector getVertex(const ROOT::Math::XYZVector& vector) const override;
 
    virtual ROOT::Math::PxPyPzEVector getMomentum(const ROOT::Math::PxPyPzEVector& vector) const override;
 
    virtual TMatrixFSym getMomentumErrorMatrix(const TMatrixFSym& matrix) const override;
 
    virtual TMatrixFSym getVertexErrorMatrix(const TMatrixFSym& matrix) const override;
 
  private:
    CMSFrame cmsframe; 
    RotationFrame rotationframe; 
  };
 
  template<class T>
  class UseReferenceFrame {
  public:
    template<class ...Args>
    explicit UseReferenceFrame(Args&& ... params) : m_frame(std::forward<Args>(params)...)
    {
      ReferenceFrame::Push(&m_frame);
    }
 
    ~UseReferenceFrame()
    {
      ReferenceFrame::Pop();
    }
  private:
    T m_frame; 
  };
 
} // Belle2 namespace