RotationTools.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2020  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Radek Zlebcik
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* External headers. */
#include <TMatrixDSym.h>
#include <TMatrixD.h>
#include <TVectorD.h>
#include <TRotation.h>
 
namespace Belle2 {
  namespace RotationTools {
 
    inline TMatrixD toMatrix(TRotation r)
    {
      TMatrixD rM(3, 3);
      for (int i = 0; i < 3; ++i)
        for (int j = 0; j < 3; ++j)
          rM(i, j) = r(i, j);
      return rM;
    }
 
 
    inline TMatrixD getRotationMatrixYZ(double angleY, double angleZ)
    {
      TRotation r;
      r.RotateY(angleY);
      r.RotateZ(angleZ);
      return toMatrix(r);
    }
 
 
    inline TMatrixD getRotationMatrixXY(double angleX, double angleY)
    {
      TRotation r;
      r.RotateX(angleX);
      r.RotateY(angleY);
      return toMatrix(r);
    }
 
 
 
    inline TMatrixD rotateTensor(const TVector3& vTo, const TMatrixD& orgMat)
    {
      TMatrixD r = getRotationMatrixYZ(vTo.Theta(), vTo.Phi());
      TMatrixD rT = r; rT.T();
      return r * orgMat * rT;
    }
 
    inline TMatrixD rotateTensorInv(const TVector3& vTo, const TMatrixD& orgMat)
    {
      TMatrixD r = getRotationMatrixYZ(vTo.Theta(), vTo.Phi());
      TMatrixD rT = r; rT.T();
      return rT * orgMat * r;
    }
 
 
    inline TMatrixDSym toSymMatrix(const TMatrixD& m)
    {
      TMatrixDSym mS(m.GetNrows());
      for (int i = 0; i < m.GetNrows(); ++i)
        for (int j = 0; j < m.GetNcols(); ++j) {
          mS(i, j) = (m(i, j) + m(j, i)) / 2;
        }
      return mS;
    }
 
 
    inline TVectorD toVec(TVector3 v)
    {
      return TVectorD(0, 2, v.X(), v.Y(), v.Z(), "END");
    }
 
    TVector3 getUnitOrthogonal(TVector3 v)
    {
      return TVector3(v.Z(), 0, -v.X()).Unit();
    }
 
  }
 
}