MELANCSplineCore.h

Go to the documentation of this file.

#ifndef MELANCSPLINECORE
#define MELANCSPLINECORE  
 
#include <vector>
#include "MELAAccumulators.h"
#include "TMatrixD.h"
#include "TVectorD.h"
#include "RooAbsReal.h"
#include "RooRealVar.h"
#include "RooRealProxy.h"
#include "RooListProxy.h"
#include "RooConstVar.h"
#include "RooArgList.h"
#include "RooMsgService.h"
 
class MELANCSplineCore : public RooAbsReal{
public:
  typedef Float_t T;
  typedef TMatrixT<T> TMatrix_t;
  typedef TVectorT<T> TVector_t;
 
  enum VerbosityLevel{
    kSilent,
    kError,
    kVerbose
  };
 
  enum BoundaryCondition{
    bcApproximatedSlope, // D1 is the same as deltaY/deltaX in the first/last segment
    bcClamped, // D1=0 at endpoint
    bcApproximatedSecondDerivative, // D2 is approximated
    bcNaturalSpline, // D2=0 at endpoint
    bcQuadratic, // Coefficient D=0
    bcQuadraticWithNullSlope, // Coefficient D=0 and D1=0 at endpoint
    NBoundaryConditions
  };
 
  MELANCSplineCore();
  MELANCSplineCore(
    const char* name,
    const char* title
    );
  MELANCSplineCore(
    const char* name,
    const char* title,
    RooAbsReal& inXVar,
    const std::vector<T>& inXList,
    Bool_t inUseFloor=true,
    T inFloorEval=1e-15,
    T inFloorInt=1e-10
    );
  MELANCSplineCore(const MELANCSplineCore& other, const char* name=0);
  virtual TObject* clone(const char* newname)const = 0;
  inline virtual ~MELANCSplineCore(){}
 
  virtual void setVerbosity(VerbosityLevel flag);
  void setEvalFloor(T val);
  void setIntFloor(T val);
  void doFloor(Bool_t flag);
 
  virtual void setRangeValidity(const T valmin, const T valmax, const Int_t whichDirection) = 0;
 
  virtual Int_t getAnalyticalIntegral(RooArgSet& allVars, RooArgSet& analVars, const char* rangeName=0)const = 0;
  virtual Double_t analyticalIntegral(Int_t code, const char* rangeName=0)const = 0;
 
protected:
  VerbosityLevel verbosity;
  Bool_t useFloor;
  T floorEval;
  T floorInt;
 
  T rangeXmin;
  T rangeXmax;
 
  RooRealProxy theXVar;
  RooListProxy leafDepsList;
  std::vector<T> XList;
 
  virtual void emptyFcnList() = 0;
 
  void getLeafDependents(RooRealProxy& proxy, RooArgSet& set);
  void addLeafDependents(RooArgSet& set);
 
  unsigned int npointsX()const{ return XList.size(); }
 
  virtual Int_t getWhichBin(const T& val, const Int_t whichDirection)const = 0;
  virtual void getKappas(std::vector<T>& kappas, const Int_t whichDirection) = 0;
  virtual T getTVar(const std::vector<T>& kappas, const T& val, const Int_t& bin, const Int_t whichDirection)const = 0;
 
  virtual Bool_t testRangeValidity(const T& val, const Int_t whichDirection)const = 0;
  virtual void cropValueForRange(T& val, const Int_t whichDirection)const = 0;
 
  virtual T interpolateFcn(Int_t code, const char* rangeName=0)const = 0;
  virtual Double_t evaluate()const = 0;
 
  virtual void getBArray(const std::vector<T>& kappas, const std::vector<T>& fcnList, std::vector<T>& BArray, BoundaryCondition const& bcBegin, BoundaryCondition const& bcEnd)const;
  virtual void getAArray(const std::vector<T>& kappas, std::vector<std::vector<T>>& AArray, BoundaryCondition const& bcBegin, BoundaryCondition const& bcEnd)const;
  virtual std::vector<std::vector<T>> getCoefficientsAlongDirection(const std::vector<T>& kappas, const TMatrix_t& Ainv, const std::vector<T>& fcnList, BoundaryCondition const& bcBegin, BoundaryCondition const& bcEnd, const Int_t pickBin)const;
  virtual std::vector<T> getCoefficients(const TVector_t& S, const std::vector<T>& kappas, const std::vector<T>& fcnList, const Int_t& bin)const;
 
  virtual T evalSplineSegment(const std::vector<T>& coefs, const T& kappa, const T& tup, const T& tdn, Bool_t doIntegrate=false)const;
 
 
  ClassDef(MELANCSplineCore, 0)
 
};
 
#endif