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