DataFormats/Math/interface/deltaPhi.h - master - githubmirror/cmssw

DataFormats/Math/interface/deltaPhi.h
86 строк3 KB
Manos Vourliotis
Rename reduceRange to reducePhiRange
21 янв 2025, 01:04
21 янв 2025, 01:043139210
#ifndef DataFormats_Math_deltaPhi_h
#define DataFormats_Math_deltaPhi_h

/* function to compute deltaPhi
 *
 * Ported from original code in RecoJets 
 * by Fedor Ratnikov, FNAL
 * stabilize range reduction
 */

#include <cmath>
#include "DataFormats/Math/interface/angle_units.h"

namespace reco {

  // reduce to [-pi,pi]
  template <typename T>
  constexpr T reducePhiRange(T x) {
    constexpr T o2pi = 1. / (2. * M_PI);
    if (std::abs(x) <= T(M_PI))
      return x;
    T n = std::round(x * o2pi);
    return x - n * T(2. * M_PI);
  }

  constexpr double deltaPhi(double phi1, double phi2) { return reducePhiRange(phi1 - phi2); }

  constexpr double deltaPhi(float phi1, double phi2) { return deltaPhi(static_cast<double>(phi1), phi2); }

  constexpr double deltaPhi(double phi1, float phi2) { return deltaPhi(phi1, static_cast<double>(phi2)); }

  constexpr float deltaPhi(float phi1, float phi2) { return reducePhiRange(phi1 - phi2); }

  template <typename T1, typename T2>
  constexpr auto deltaPhi(T1 const& t1, T2 const& t2) -> decltype(deltaPhi(t1.phi(), t2.phi())) {
    return deltaPhi(t1.phi(), t2.phi());
  }

  template <typename T>
  constexpr T deltaPhi(T phi1, T phi2) {
    return reducePhiRange(phi1 - phi2);
  }
}  // namespace reco

// lovely!  VI
using reco::deltaPhi;

template <typename T1, typename T2 = T1>
struct DeltaPhi {
  constexpr auto operator()(const T1& t1, const T2& t2) -> decltype(reco::deltaPhi(t1, t2)) const {
    return reco::deltaPhi(t1, t2);
  }
};

namespace angle0to2pi {

  using angle_units::operators::operator""_pi;

  // make0To2pi constrains an angle to be >= 0 and < 2pi.
  // This function is a faster version of reco::reducePhiRange.
  // In timing tests, it is almost always faster than reco::reduceRange.
  // It also protects against floating-point value drift over repeated calculations.
  // This implementation uses multiplication instead of division and avoids
  // calling fmod to improve performance.

  template <class valType>
  inline constexpr valType make0To2pi(valType angle) {
    constexpr valType twoPi = 2. * M_PI;
    constexpr valType oneOverTwoPi = 1. / (2. * M_PI);
    constexpr valType epsilon = 1.e-13;

    if ((std::abs(angle) <= epsilon) || (std::abs(twoPi - std::abs(angle)) <= epsilon))
      return (0.);
    if (std::abs(angle) > twoPi) {
      valType nFac = trunc(angle * oneOverTwoPi);
      angle -= (nFac * twoPi);
      if (std::abs(angle) <= epsilon)
        return (0.);
    }
    if (angle < 0.)
      angle += twoPi;
    return (angle);
  }
}  // namespace angle0to2pi

#endif