Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ParamHistFunc.h
Go to the documentation of this file.
1// @(#)root/roostats:$Id: cranmer $
2// Author: George Lewis, Kyle Cranmer
3/*************************************************************************
4 * Copyright (C) 1995-2008, Rene Brun and Fons Rademakers. *
5 * All rights reserved. *
6 * *
7 * For the licensing terms see $ROOTSYS/LICENSE. *
8 * For the list of contributors see $ROOTSYS/README/CREDITS. *
9 *************************************************************************/
10
11
12#ifndef ROO_PARAMHISTFUNC
13#define ROO_PARAMHISTFUNC
14
15#include "RooAbsReal.h"
16#include "RooListProxy.h"
17#include "RooObjCacheManager.h"
18#include "RooDataHist.h"
19
20// Forward Declarations
21class RooRealVar;
22class RooWorkspace;
23
24class ParamHistFunc : public RooAbsReal {
25public:
26
28 ParamHistFunc(const char *name, const char *title, const RooArgList& vars, const RooArgList& paramSet );
29 ParamHistFunc(const char *name, const char *title, const RooArgList& vars, const RooArgList& paramSet, const TH1* hist );
30
31 ParamHistFunc(const ParamHistFunc& other, const char *name = nullptr);
32 TObject* clone(const char* newname) const override { return new ParamHistFunc(*this, newname); }
33
34 const RooArgList& paramList() const { return _paramSet ; }
35
36 Int_t numBins() const { return _dataSet.numEntries(); } // Number of bins (called numEntries in RooDataHist)
37
38 void setParamConst( Int_t, bool=true );
39 void setConstant(bool constant);
40
41 void setShape(TH1* shape);
42
43 RooAbsReal& getParameter() const ;
44 RooAbsReal& getParameter( Int_t masterIdx ) const ;
45
46 const RooArgSet* get(Int_t masterIdx) const { return _dataSet.get( masterIdx ) ; }
47 const RooArgSet* get(const RooArgSet& coord) const { return _dataSet.get( coord ) ; }
48
49 double binVolume() const { return _dataSet.binVolume(); }
50
51 bool forceAnalyticalInt(const RooAbsArg&) const override { return true ; }
52
53 Int_t getAnalyticalIntegralWN(RooArgSet& allVars, RooArgSet& analVars, const RooArgSet* normSet,const char* rangeName=nullptr) const override;
54 double analyticalIntegralWN(Int_t code, const RooArgSet* normSet, const char* rangeName=nullptr) const override;
55
56 static RooArgList createParamSet(RooWorkspace& w, const std::string&, const RooArgList& Vars);
57 static RooArgList createParamSet(RooWorkspace& w, const std::string&, const RooArgList& Vars, double, double);
58 static RooArgList createParamSet(const std::string&, Int_t, double, double);
59
60 std::list<double>* binBoundaries(RooAbsRealLValue& /*obs*/, double /*xlo*/, double /*xhi*/) const override;
61 std::list<double>* plotSamplingHint(RooAbsRealLValue& obs, double xlo, double xhi) const override;
62 bool isBinnedDistribution(const RooArgSet& obs) const override { return _dataVars.overlaps(obs); }
63
64
65protected:
66
68 public:
71 ret.add(_lowIntList);
72 ret.add(_highIntList);
73 return ret ;
74 }
78 // will want std::vector<RooRealVar*> for low and high also
79 } ;
80 mutable RooObjCacheManager _normIntMgr ; ///<! The integration cache manager
81
82 RooListProxy _dataVars; ///< The RooRealVars
83 RooListProxy _paramSet ; ///< interpolation parameters
84
86 struct NumBins {
88 NumBins(int nx, int ny, int nz) : x{nx}, y{ny}, z{nz}, xy{x*y}, xz{x*z}, yz{y*z}, xyz{xy*z} {}
89 int x = 0;
90 int y = 0;
91 int z = 0;
92 int xy = 0;
93 int xz = 0;
94 int yz = 0;
95 int xyz = 0;
96 };
99
100 Int_t getCurrentBin() const;
101 Int_t addVarSet( const RooArgList& vars );
102 Int_t addParamSet( const RooArgList& params );
103 static Int_t GetNumBins( const RooArgSet& vars );
104 double evaluate() const override;
105 void computeBatch(cudaStream_t*, double* output, size_t size, RooFit::Detail::DataMap const&) const override;
106
107private:
108 static NumBins getNumBinsPerDim(RooArgSet const& vars);
109
111};
112
113#endif
size_t size(const MatrixT &matrix)
retrieve the size of a square matrix
#define ClassDefOverride(name, id)
Definition Rtypes.h:341
char name[80]
Definition TGX11.cxx:110
RooArgList containedArgs(Action) override
A class which maps the current values of a RooRealVar (or a set of RooRealVars) to one of a number of...
static NumBins getNumBinsPerDim(RooArgSet const &vars)
const RooArgSet * get(Int_t masterIdx) const
std::list< double > * binBoundaries(RooAbsRealLValue &, double, double) const override
Return sampling hint for making curves of (projections) of this function as the recursive division st...
RooDataHist _dataSet
static Int_t GetNumBins(const RooArgSet &vars)
TObject * clone(const char *newname) const override
double analyticalIntegralWN(Int_t code, const RooArgSet *normSet, const char *rangeName=nullptr) const override
Implement analytical integrations by doing appropriate weighting from component integrals functions t...
void setConstant(bool constant)
const RooArgList & paramList() const
Int_t getCurrentBin() const
Get the index of the gamma parameter associated with the current bin.
Int_t getAnalyticalIntegralWN(RooArgSet &allVars, RooArgSet &analVars, const RooArgSet *normSet, const char *rangeName=nullptr) const override
Advertise that all integrals can be handled internally.
RooObjCacheManager _normIntMgr
! The integration cache manager
Int_t numBins() const
double evaluate() const override
Find the bin corresponding to the current value of the observable, and evaluate the associated parame...
const RooArgSet * get(const RooArgSet &coord) const
Int_t addVarSet(const RooArgList &vars)
return 0 for success return 1 for failure Check that the elements are actually RooRealVar's If so,...
RooAbsReal & getParameter() const
bool forceAnalyticalInt(const RooAbsArg &) const override
Int_t addParamSet(const RooArgList &params)
NumBins _numBinsPerDim
static RooArgList createParamSet(RooWorkspace &w, const std::string &, const RooArgList &Vars)
Create the list of RooRealVar parameters which represent the height of the histogram bins.
void setParamConst(Int_t, bool=true)
std::list< double > * plotSamplingHint(RooAbsRealLValue &obs, double xlo, double xhi) const override
Return sampling hint for making curves of (projections) of this function as the recursive division st...
RooListProxy _paramSet
interpolation parameters
void setShape(TH1 *shape)
double binVolume() const
RooListProxy _dataVars
The RooRealVars.
bool isBinnedDistribution(const RooArgSet &obs) const override
Tests if the distribution is binned. Unless overridden by derived classes, this always returns false.
void computeBatch(cudaStream_t *, double *output, size_t size, RooFit::Detail::DataMap const &) const override
Find all bins corresponding to the values of the observables in evalData, and evaluate the associated...
RooAbsArg is the common abstract base class for objects that represent a value and a "shape" in RooFi...
Definition RooAbsArg.h:74
RooAbsCacheElement is the abstract base class for objects to be stored in RooAbsCache cache manager o...
virtual bool add(const RooAbsArg &var, bool silent=false)
Add the specified argument to list.
bool overlaps(Iterator_t otherCollBegin, Iterator_t otherCollEnd) const
virtual Int_t numEntries() const
Return number of entries in dataset, i.e., count unweighted entries.
RooAbsRealLValue is the common abstract base class for objects that represent a real value that may a...
RooAbsReal is the common abstract base class for objects that represent a real value and implements f...
Definition RooAbsReal.h:62
RooArgList is a container object that can hold multiple RooAbsArg objects.
Definition RooArgList.h:22
RooArgSet is a container object that can hold multiple RooAbsArg objects.
Definition RooArgSet.h:55
The RooDataHist is a container class to hold N-dimensional binned data.
Definition RooDataHist.h:39
double binVolume(std::size_t i) const
Return bin volume of i-th bin.
const RooArgSet * get() const override
Get bin centre of current bin.
Definition RooDataHist.h:76
Class RooObjCacheManager is an implementation of class RooCacheManager<RooAbsCacheElement> and specia...
RooRealVar represents a variable that can be changed from the outside.
Definition RooRealVar.h:40
The RooWorkspace is a persistable container for RooFit projects.
TH1 is the base class of all histogram classes in ROOT.
Definition TH1.h:58
Mother of all ROOT objects.
Definition TObject.h:41
NumBins(int nx, int ny, int nz)
static void output()