PeriodicBinFinderInPhi.h

Go to the documentation of this file.

#ifndef PeriodicBinFinderInPhi_H
#define PeriodicBinFinderInPhi_H
 
#include "Utilities/BinningTools/interface/BaseBinFinder.h"
 
#include <algorithm>
#include <cmath>
 
template <class T>
class PeriodicBinFinderInPhi final : public BaseBinFinder<T> {
public:
  PeriodicBinFinderInPhi() {}
 
  PeriodicBinFinderInPhi(T firstPhi, int nbins)
      : theNbins(nbins),
        thePhiStep(twoPiC / T(nbins)),
        theInvPhiStep(T(1) / thePhiStep),
        thePhiOffset(firstPhi - T(0.5) * thePhiStep) {}
 
  int binIndex(T phi) const override {
    T tmp = std::fmod((phi - thePhiOffset), twoPiC) * theInvPhiStep;
    if (tmp < 0)
      tmp += theNbins;
    return std::min(int(tmp), theNbins - 1);
  }
 
  int binIndex(int i) const override {
    int ind = i % theNbins;
    return ind < 0 ? ind + theNbins : ind;
  }
 
  T binPosition(int ind) const override { return thePhiOffset + thePhiStep * (T(ind) + T(0.5)); }
 
  static constexpr T pi() { return piC; }
  static constexpr T twoPi() { return twoPiC; }
 
private:
  static constexpr T piC = 3.141592653589793238;
  static constexpr T twoPiC = 2 * piC;
 
  int theNbins = 0;
  T thePhiStep = 0;
  T theInvPhiStep = 0;
  T thePhiOffset = 0;
};
#endif