MELANCSpline_2D_fast.h

Go to the documentation of this file.

#ifndef MELANCSPLINE_2D_FAST
#define MELANCSPLINE_2D_FAST
 
#include <vector>
#include "RooAbsPdf.h"
#include "RooRealProxy.h"
#include "RooRealVar.h"
#include "RooAbsReal.h"
#include "MELANCSplineCore.h"
 
class MELANCSpline_2D_fast : public MELANCSplineCore{
protected:
  T rangeYmin;
  T rangeYmax;
 
  BoundaryCondition const bcBeginX;
  BoundaryCondition const bcEndX;
  BoundaryCondition const bcBeginY;
  BoundaryCondition const bcEndY;
 
  RooRealProxy theYVar;
  std::vector<T> YList;
 
  std::vector<std::vector<T>> FcnList;
 
  std::vector<T> kappaX;
  std::vector<T> kappaY;
  std::vector<std::vector<std::vector<std::vector<T>>>> coefficients; // [ix][A_x,B_x,C_x,D_x][iy][A_x_y,B_x_y,C_x_y,D_x_y]
 
public:
  MELANCSpline_2D_fast();
  MELANCSpline_2D_fast(
    const char* name,
    const char* title
    );
  MELANCSpline_2D_fast(
    const char* name,
    const char* title,
    RooAbsReal& inXVar,
    RooAbsReal& inYVar,
    const std::vector<T>& inXList,
    const std::vector<T>& inYList,
    const std::vector<std::vector<T>>& inFcnList,
    MELANCSplineCore::BoundaryCondition const bcBeginX_=MELANCSplineCore::bcNaturalSpline,
    MELANCSplineCore::BoundaryCondition const bcEndX_=MELANCSplineCore::bcNaturalSpline,
    MELANCSplineCore::BoundaryCondition const bcBeginY_=MELANCSplineCore::bcNaturalSpline,
    MELANCSplineCore::BoundaryCondition const bcEndY_=MELANCSplineCore::bcNaturalSpline,
    Bool_t inUseFloor=true,
    T inFloorEval=0,
    T inFloorInt=0
    );
  MELANCSpline_2D_fast(const MELANCSpline_2D_fast& other, const char* name=0);
    virtual TObject* clone(const char* newname)const { return new MELANCSpline_2D_fast(*this, newname); }
    inline virtual ~MELANCSpline_2D_fast(){}
 
  void setRangeValidity(const T valmin, const T valmax, const Int_t whichDirection);
 
  virtual Int_t getAnalyticalIntegral(RooArgSet& allVars, RooArgSet& analVars, const char* rangeName=0) const;
  virtual Double_t analyticalIntegral(Int_t code, const char* rangeName=0)const;
 
protected:
  virtual void emptyFcnList(){ std::vector<std::vector<T>> tmp; FcnList.swap(tmp); }
 
  unsigned int npointsY()const{ return YList.size(); }
 
  virtual Int_t getWhichBin(const T& val, const Int_t whichDirection)const;
  virtual T getTVar(const std::vector<T>& kappas, const T& val, const Int_t& bin, const Int_t whichDirection)const;
  virtual void getKappas(std::vector<T>& kappas, const Int_t whichDirection);
 
  Bool_t testRangeValidity(const T& val, const Int_t whichDirection)const;
  void cropValueForRange(T& val, const Int_t whichDirection)const;
 
  virtual std::vector<std::vector<T>> getCoefficientsPerY(const std::vector<T>& kappaX, const TMatrix_t& xAinv, const Int_t& ybin, MELANCSplineCore::BoundaryCondition const& bcBegin, MELANCSplineCore::BoundaryCondition const& bcEnd, const Int_t xbin)const; // xbin can be -1, which means push all of them
 
  virtual T interpolateFcn(Int_t code, const char* rangeName=0)const;
 
  virtual Double_t evaluate()const;
 
 
  ClassDef(MELANCSpline_2D_fast, 2)
 
};
 
#endif