CDCBFieldUtil.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
#include <tracking/trackingUtilities/eventdata/trajectories/CDCBFieldUtil.h>
 
#include <tracking/trackingUtilities/geometry/VectorUtil.h>
 
#include <tracking/trackingUtilities/numerics/ERotation.h>
#include <tracking/trackingUtilities/numerics/ESign.h>
 
#include <framework/geometry/BFieldManager.h>
 
#include <framework/geometry/B2Vector3.h>
 
#include <Math/Vector3D.h>
#include <Math/Vector2D.h>
#include <TMath.h>
#include <cmath>
 
using namespace Belle2;
using namespace TrackingUtilities;
 
bool CDCBFieldUtil::isOff()
{
  double b = BFieldManager::getField(0, 0, 0).R();
  double c_EarthMagneticField = 3.2e-5 * Unit::T;
  return not(b > 5 * c_EarthMagneticField);
}
 
ESign CDCBFieldUtil::getBFieldZSign()
{
  return sign(CDCBFieldUtil::getBFieldZ());
}
 
double CDCBFieldUtil::getBFieldZ()
{
  return getBFieldZ(ROOT::Math::XYZVector(0, 0, 0));
}
 
double CDCBFieldUtil::getBFieldZ(const ROOT::Math::XYVector& pos2D)
{
  return getBFieldZ(ROOT::Math::XYZVector(pos2D.X(), pos2D.Y(), 0));
}
 
double CDCBFieldUtil::getBFieldZ(const ROOT::Math::XYZVector& pos3D)
{
  // The BFieldMap can not handle positions with not a number coordinates
  // which can occur if fits fail.
  // Return NAN to the caller and let him decide what to do next.
  if (VectorUtil::hasNAN(pos3D)) return NAN;
  auto mag3D = BFieldManager::getField(pos3D.x(), pos3D.y(), pos3D.z());
  return mag3D.Z() / Unit::T;
}
 
 
 
double CDCBFieldUtil::getAlphaFromBField(double bField)
{
  return 1.0 / (bField * TMath::C()) * 1E11;
}
 
double CDCBFieldUtil::getAlphaZ(const ROOT::Math::XYVector& pos2D)
{
  return getAlphaFromBField(getBFieldZ(pos2D));
}
 
double CDCBFieldUtil::getAlphaZ(const ROOT::Math::XYZVector& pos3D)
{
  return getAlphaFromBField(getBFieldZ(pos3D));
}
 
ESign CDCBFieldUtil::ccwInfoToChargeSign(ERotation ccwInfo)
{
  return static_cast<ESign>(- ccwInfo * getBFieldZSign());
}
 
ERotation CDCBFieldUtil::chargeSignToERotation(ESign chargeSign)
{
  return static_cast<ERotation>(- chargeSign * getBFieldZSign());
}
 
ERotation CDCBFieldUtil::chargeToERotation(double charge)
{
  return chargeSignToERotation(sign(charge));
}
 
double CDCBFieldUtil::absMom2DToBendRadius(double absMom2D,
                                           double bZ)
{
  // In case of zero magnetic field return something large
  return std::fmin(4440, absMom2D / (bZ * 0.00299792458));
}
 
double CDCBFieldUtil::absMom2DToBendRadius(double absMom2D,
                                           const ROOT::Math::XYVector& pos2D)
{
  return absMom2DToBendRadius(absMom2D, getBFieldZ(pos2D));
}
 
double CDCBFieldUtil::absMom2DToBendRadius(double absMom2D,
                                           const ROOT::Math::XYZVector& pos3D)
{
  return absMom2DToBendRadius(absMom2D, getBFieldZ(pos3D));
}
 
double CDCBFieldUtil::absMom2DToCurvature(double absMom2D,
                                          double charge,
                                          double bZ)
{
  return - charge * bZ * 0.00299792458 * std::fmax(0, 1 / absMom2D);
}
 
double CDCBFieldUtil::absMom2DToCurvature(double absMom2D,
                                          double charge,
                                          const ROOT::Math::XYVector& pos2D)
{
  return absMom2DToCurvature(absMom2D, charge, getBFieldZ(pos2D));
}
 
double CDCBFieldUtil::absMom2DToCurvature(double absMom2D,
                                          double charge,
                                          const ROOT::Math::XYZVector& pos3D)
{
  return absMom2DToCurvature(absMom2D, charge, getBFieldZ(pos3D));
}
 
double CDCBFieldUtil::curvatureToAbsMom2D(double curvature,
                                          double bZ)
{
  return std::fmin(20, std::fabs(bZ * 0.00299792458 / curvature));
}
 
double CDCBFieldUtil::curvatureToAbsMom2D(double curvature,
                                          const ROOT::Math::XYVector& pos2D)
{
  return curvatureToAbsMom2D(curvature, getBFieldZ(pos2D));
}
 
double CDCBFieldUtil::curvatureToAbsMom2D(double curvature,
                                          const ROOT::Math::XYZVector& pos3D)
{
  return curvatureToAbsMom2D(curvature, getBFieldZ(pos3D));
}