LabToCms.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 <framework/utilities/MathHelpers.h>
#include <Math/Boost.h>
#include <Math/AxisAngle.h>
#include <Math/LorentzRotation.h>
#include <Math/Vector3D.h>
 
namespace Belle2 {

  class LabToCms {
 
  public:
 

    static ROOT::Math::LorentzRotation  rotateLabToCms(const ROOT::Math::XYZVector& boostVector,
                                                       double cmsAngleXZ, double  cmsAngleYZ)
    {
      // Boost to CM frame
      ROOT::Math::LorentzRotation boost(ROOT::Math::Boost(-1.*boostVector));
 
      // Target pHER orientation (unit 3D vector)
      ROOT::Math::XYZVector zaxis(0., 0., 1.);
 
      // The current pHER orientation (unit 3D vector)
      double tanAngleXZ = tan(cmsAngleXZ);
      double tanAngleYZ = tan(cmsAngleYZ);
      double Norm   = 1 / sqrt(1 + square(tanAngleXZ) + square(tanAngleYZ));
      ROOT::Math::XYZVector electronCMS(Norm * tanAngleXZ, Norm * tanAngleYZ, Norm);
 
      // Axis to rotate current pHER direction to the target one
      ROOT::Math::XYZVector rotAxis = zaxis.Cross(electronCMS);
      double rotangle = asin(rotAxis.R()); // rotation angle
 
      // Rotation that makes pHER align with z-axis
      ROOT::Math::LorentzRotation rotation(ROOT::Math::AxisAngle(rotAxis, -rotangle));
 
      // First boost, then rotation
      ROOT::Math::LorentzRotation trans = rotation * boost;
      return trans;
    }
 
  };

}