FlexibleInterpVar.h

Go to the documentation of this file.

// @(#)root/roostats:$Id:  cranmer $
// Author: Kyle Cranmer, Akira Shibata
/*************************************************************************
 * Copyright (C) 1995-2008, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOSTATS_FLEXIBLEINTERPVAR
#define ROOSTATS_FLEXIBLEINTERPVAR
 
#include "RooAbsPdf.h"
#include "RooRealProxy.h"
#include "RooListProxy.h"
#include <vector>
 
class RooRealVar;
class RooArgList ;
 
namespace RooStats{
namespace HistFactory{
 
  class FlexibleInterpVar : public RooAbsReal {
  public:
 
    FlexibleInterpVar() ;
    FlexibleInterpVar(const char *name, const char *title,
            const RooArgList& _paramList,
            double nominal, const RooArgList& low, const RooArgList& high);
 
    FlexibleInterpVar(const char *name, const char *title,
            const RooArgList& _paramList,
            double nominal, std::vector<double> low, std::vector<double> high);
 
    FlexibleInterpVar(const char *name, const char *title,
            const RooArgList& _paramList, double nominal, std::vector<double> low,
            std::vector<double> high,std::vector<int> code);
 
    FlexibleInterpVar(const char *name, const char *title);
    FlexibleInterpVar(const FlexibleInterpVar&, const char*);
 
    void setInterpCode(RooAbsReal& param, int code);
    void setAllInterpCodes(int code);
    void setGlobalBoundary(double boundary) {_interpBoundary = boundary;}
    void setNominal(double newNominal);
    void setLow(RooAbsReal& param, double newLow);
    void setHigh(RooAbsReal& param, double newHigh);
 
    void printAllInterpCodes();
    const std::vector<int>&  interpolationCodes() const { return _interpCode; }    
 
    TObject* clone(const char* newname) const override { return new FlexibleInterpVar(*this, newname); }
    ~FlexibleInterpVar() override ;
 
    void printMultiline(std::ostream& os, Int_t contents, bool verbose = false, TString indent = "") const override;
    virtual void printFlexibleInterpVars(std::ostream& os) const;
 
    const RooListProxy& variables() const;
    double nominal() const;
    const std::vector<double>& low() const;
    const std::vector<double>& high() const;
 
    void computeBatch(cudaStream_t*, double* output, size_t size, RooFit::Detail::DataMap const&) const override;
 
  private:
 
    double PolyInterpValue(int i, double x) const;
 
    void processParam(std::size_t i, double paramVal, double &total) const;
 
  protected:
 
    RooListProxy _paramList ;
    double _nominal;
    std::vector<double> _low;
    std::vector<double> _high;
    std::vector<int> _interpCode;
    double _interpBoundary;
 
    mutable bool         _logInit ;            ///<! flag used for caching polynomial coefficients
    mutable std::vector< double>  _polCoeff;     ///<! cached polynomial coefficients
 
    double evaluate() const override;
 
    ClassDefOverride(RooStats::HistFactory::FlexibleInterpVar,2) // flexible interpolation
  };
}
}
 
#endif