Logo ROOT  
Reference Guide
RooProdPdf.h
Go to the documentation of this file.
1/*****************************************************************************
2 * Project: RooFit *
3 * Package: RooFitCore *
4 * File: $Id: RooProdPdf.h,v 1.44 2007/07/16 21:04:28 wouter Exp $
5 * Authors: *
6 * WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu *
7 * DK, David Kirkby, UC Irvine, dkirkby@uci.edu *
8 * *
9 * Copyright (c) 2000-2005, Regents of the University of California *
10 * and Stanford University. All rights reserved. *
11 * *
12 * Redistribution and use in source and binary forms, *
13 * with or without modification, are permitted according to the terms *
14 * listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
15 *****************************************************************************/
16#ifndef ROO_PROD_PDF
17#define ROO_PROD_PDF
18
19#include "RooAbsPdf.h"
20#include "RooListProxy.h"
21#include "RooLinkedList.h"
22#include "RooAICRegistry.h"
23#include "RooObjCacheManager.h"
24#include "RooCmdArg.h"
25
26#include <vector>
27#include <list>
28#include <string>
29
32
33class RooProdPdf : public RooAbsPdf {
34public:
35
36 RooProdPdf() ;
37 RooProdPdf(const char *name, const char *title, double cutOff=0);
38 RooProdPdf(const char *name, const char *title,
39 RooAbsPdf& pdf1, RooAbsPdf& pdf2, double cutOff=0) ;
40 RooProdPdf(const char* name, const char* title, const RooArgList& pdfList, double cutOff=0) ;
41 RooProdPdf(const char* name, const char* title, const RooArgSet& fullPdfSet, const RooLinkedList& cmdArgList) ;
42
43 RooProdPdf(const char* name, const char* title, const RooArgSet& fullPdfSet,
44 const RooCmdArg& arg1 , const RooCmdArg& arg2=RooCmdArg(),
45 const RooCmdArg& arg3=RooCmdArg(), const RooCmdArg& arg4=RooCmdArg(),
46 const RooCmdArg& arg5=RooCmdArg(), const RooCmdArg& arg6=RooCmdArg(),
47 const RooCmdArg& arg7=RooCmdArg(), const RooCmdArg& arg8=RooCmdArg()) ;
48
49 RooProdPdf(const char* name, const char* title,
50 const RooCmdArg& arg1, const RooCmdArg& arg2=RooCmdArg(),
51 const RooCmdArg& arg3=RooCmdArg(), const RooCmdArg& arg4=RooCmdArg(),
52 const RooCmdArg& arg5=RooCmdArg(), const RooCmdArg& arg6=RooCmdArg(),
53 const RooCmdArg& arg7=RooCmdArg(), const RooCmdArg& arg8=RooCmdArg()) ;
54
55 RooProdPdf(const RooProdPdf& other, const char* name=0) ;
56 TObject* clone(const char* newname) const override { return new RooProdPdf(*this,newname) ; }
57 ~RooProdPdf() override ;
58
59 bool checkObservables(const RooArgSet* nset) const override ;
60
61 bool forceAnalyticalInt(const RooAbsArg& dep) const override ;
62 Int_t getAnalyticalIntegralWN(RooArgSet& allVars, RooArgSet& numVars, const RooArgSet* normSet, const char* rangeName=0) const override ;
63 double analyticalIntegralWN(Int_t code, const RooArgSet* normSet, const char* rangeName=0) const override ;
64 bool selfNormalized() const override { return _selfNorm ; }
65
66 ExtendMode extendMode() const override ;
67 double expectedEvents(const RooArgSet* nset) const override ;
68
69 const RooArgList& pdfList() const { return _pdfList ; }
70
71 Int_t getGenerator(const RooArgSet& directVars, RooArgSet &generateVars, bool staticInitOK=true) const override;
72 void initGenerator(Int_t code) override ;
73 void generateEvent(Int_t code) override;
74 bool isDirectGenSafe(const RooAbsArg& arg) const override ;
75
76 // Constraint management
77 RooArgSet* getConstraints(const RooArgSet& observables, RooArgSet& constrainedParams, bool stripDisconnected) const override ;
78
79 std::list<double>* plotSamplingHint(RooAbsRealLValue& obs, double xlo, double xhi) const override ;
80 std::list<double>* binBoundaries(RooAbsRealLValue& /*obs*/, double /*xlo*/, double /*xhi*/) const override ;
81 bool isBinnedDistribution(const RooArgSet& obs) const override ;
82
83 void printMetaArgs(std::ostream& os) const override ;
84
85 void selectNormalizationRange(const char* rangeName=0, bool force=false) override ;
86 void fixRefRange(const char* rangeName) ;
87
88 void setSelfNormalized(bool flag) { _selfNorm = flag ; }
90
91
92 bool redirectServersHook(const RooAbsCollection& /*newServerList*/, bool /*mustReplaceAll*/, bool /*nameChange*/, bool /*isRecursive*/) override ;
93
94 RooArgSet* getConnectedParameters(const RooArgSet& observables) const ;
95
96 RooArgSet* findPdfNSet(RooAbsPdf const& pdf) const ;
97
98 void writeCacheToStream(std::ostream& os, RooArgSet const* nset) const;
99
100 void fillNormSetForServer(RooArgSet const& normSet, RooAbsArg const& server, RooArgSet& serverNormSet) const override;
101
102private:
103
104 double evaluate() const override ;
105 void computeBatch(cudaStream_t*, double* output, size_t nEvents, RooFit::Detail::DataMap const&) const override;
106 inline bool canComputeBatchWithCuda() const override { return true; }
107
108 RooAbsReal* makeCondPdfRatioCorr(RooAbsReal& term, const RooArgSet& termNset, const RooArgSet& termImpSet, const char* normRange, const char* refRange) const ;
109
110 void getParametersHook(const RooArgSet* /*nset*/, RooArgSet* /*list*/, bool stripDisconnected) const override ;
111
112 void initializeFromCmdArgList(const RooArgSet& fullPdfSet, const RooLinkedList& l) ;
113
114 void factorizeProduct(const RooArgSet& normSet, const RooArgSet& intSet,
115 RooLinkedList& termList, RooLinkedList& normList,
116 RooLinkedList& impDepList, RooLinkedList& crossDepList,
117 RooLinkedList& intList) const;
118 std::string makeRGPPName(const char* pfx, const RooArgSet& term, const RooArgSet& iset, const RooArgSet& nset, const char* isetRangeName) const ;
119 void groupProductTerms(std::list<std::vector<RooArgSet*>>& groupedTerms, RooArgSet& outerIntDeps,
120 const RooLinkedList& terms, const RooLinkedList& norms,
121 const RooLinkedList& imps, const RooLinkedList& ints, const RooLinkedList& cross) const ;
122
123
124
125 Int_t getPartIntList(const RooArgSet* nset, const RooArgSet* iset, const char* isetRangeName=0) const ;
126
127 std::vector<RooAbsReal*> processProductTerm(const RooArgSet* nset, const RooArgSet* iset, const char* isetRangeName,
128 const RooArgSet* term,const RooArgSet& termNSet, const RooArgSet& termISet,
129 bool& isOwned, bool forceWrap=false) const ;
130
131
132 CacheMode canNodeBeCached() const override { return RooAbsArg::NotAdvised ; } ;
133 void setCacheAndTrackHints(RooArgSet&) override ;
134
135 // The cache object
136 class CacheElem final : public RooAbsCacheElement {
137 public:
139 ~CacheElem() override = default;
140 // Payload
145 std::vector<std::unique_ptr<RooArgSet>> _normList;
147 std::unique_ptr<RooAbsReal> _rearrangedNum{};
148 std::unique_ptr<RooAbsReal> _rearrangedDen{};
149 // Cache management functions
151 void printCompactTreeHook(std::ostream&, const char *, Int_t, Int_t) override ;
152 void writeToStream(std::ostream& os) const ;
153 } ;
154 mutable RooObjCacheManager _cacheMgr ; //! The cache manager
155
156 CacheElem* getCacheElem(RooArgSet const* nset) const ;
157 void rearrangeProduct(CacheElem&) const;
158 RooAbsReal* specializeIntegral(RooAbsReal& orig, const char* targetRangeName) const ;
159 RooAbsReal* specializeRatio(RooFormulaVar& input, const char* targetRangeName) const ;
160 double calculate(const RooProdPdf::CacheElem& cache, bool verbose=false) const ;
161
162
163 friend class RooProdGenContext ;
164 RooAbsGenContext* genContext(const RooArgSet &vars, const RooDataSet *prototype=0,
165 const RooArgSet *auxProto=0, bool verbose= false) const override ;
166
167
168 mutable RooAICRegistry _genCode ; ///<! Registry of composite direct generator codes
169
170 double _cutOff ; ///< Cutoff parameter for running product
171 RooListProxy _pdfList ; ///< List of PDF components
172 std::vector<std::unique_ptr<RooArgSet>> _pdfNSetList ; ///< List of PDF component normalization sets
173 Int_t _extendedIndex ; ///< Index of extended PDF (if any)
174
175 void useDefaultGen(bool flag=true) { _useDefaultGen = flag ; }
176 bool _useDefaultGen ; ///< Use default or distributed event generator
177
178 mutable TNamed* _refRangeName ; ///< Reference range name for interpretation of conditional products
179
180 bool _selfNorm ; ///< Is self-normalized
181 RooArgSet _defNormSet ; ///< Default normalization set
182
183private:
184
185 ClassDefOverride(RooProdPdf,6) // PDF representing a product of PDFs
186};
187
188
189#endif
RooArgList * pRooArgList
Definition: RooProdPdf.h:30
RooLinkedList * pRooLinkedList
Definition: RooProdPdf.h:31
#define ClassDefOverride(name, id)
Definition: Rtypes.h:339
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void input
char name[80]
Definition: TGX11.cxx:110
RooAICRegistry is a utility class for operator p.d.f classes that keeps track of analytical integrati...
RooAbsArg is the common abstract base class for objects that represent a value and a "shape" in RooFi...
Definition: RooAbsArg.h:77
@ NotAdvised
Definition: RooAbsArg.h:412
RooAbsCacheElement is the abstract base class for objects to be stored in RooAbsCache cache manager o...
RooAbsCollection is an abstract container object that can hold multiple RooAbsArg objects.
virtual void removeAll()
Remove all arguments from our set, deleting them if we own them.
virtual RooAbsArg * addClone(const RooAbsArg &var, bool silent=false)
Add a clone of the specified argument to list.
RooAbsGenContext is the abstract base class for generator contexts of RooAbsPdf objects.
const char * normRange() const
Definition: RooAbsPdf.h:292
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:64
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:57
RooCmdArg is a named container for two doubles, two integers two object points and three string point...
Definition: RooCmdArg.h:26
RooDataSet is a container class to hold unbinned data.
Definition: RooDataSet.h:55
A RooFormulaVar is a generic implementation of a real-valued object, which takes a RooArgList of serv...
Definition: RooFormulaVar.h:30
RooLinkedList is an collection class for internal use, storing a collection of RooAbsArg pointers in ...
Definition: RooLinkedList.h:38
Class RooObjCacheManager is an implementation of class RooCacheManager<RooAbsCacheElement> and specia...
RooProdGenContext is an efficient implementation of the generator context specific for RooProdPdf PDF...
RooArgList containedArgs(Action) override
Return RooAbsArg components contained in the cache.
~CacheElem() override=default
std::unique_ptr< RooAbsReal > _rearrangedNum
Definition: RooProdPdf.h:147
void printCompactTreeHook(std::ostream &, const char *, Int_t, Int_t) override
Hook function to print cache contents in tree printing of RooProdPdf.
RooArgList _ownedList
Definition: RooProdPdf.h:144
RooArgList _partList
Definition: RooProdPdf.h:141
std::vector< std::unique_ptr< RooArgSet > > _normList
Definition: RooProdPdf.h:145
std::unique_ptr< RooAbsReal > _rearrangedDen
Definition: RooProdPdf.h:148
void writeToStream(std::ostream &os) const
RooProdPdf is an efficient implementation of a product of PDFs of the form.
Definition: RooProdPdf.h:33
void setCacheAndTrackHints(RooArgSet &) override
Label OK'ed components of a RooProdPdf with cache-and-track, and label all RooProdPdf descendants wit...
Int_t getGenerator(const RooArgSet &directVars, RooArgSet &generateVars, bool staticInitOK=true) const override
Query internal generation capabilities of component p.d.f.s and aggregate capabilities into master co...
void rearrangeProduct(CacheElem &) const
bool selfNormalized() const override
Shows if a PDF is self-normalized, which means that no attempt is made to add a normalization term.
Definition: RooProdPdf.h:64
void factorizeProduct(const RooArgSet &normSet, const RooArgSet &intSet, RooLinkedList &termList, RooLinkedList &normList, RooLinkedList &impDepList, RooLinkedList &crossDepList, RooLinkedList &intList) const
Factorize product in irreducible terms for given choice of integration/normalization.
Definition: RooProdPdf.cxx:548
~RooProdPdf() override
Destructor.
Definition: RooProdPdf.cxx:427
Int_t _extendedIndex
Index of extended PDF (if any)
Definition: RooProdPdf.h:173
Int_t getPartIntList(const RooArgSet *nset, const RooArgSet *iset, const char *isetRangeName=0) const
Return list of (partial) integrals of product terms for integration of p.d.f over observables iset wh...
Definition: RooProdPdf.cxx:732
void computeBatch(cudaStream_t *, double *output, size_t nEvents, RooFit::Detail::DataMap const &) const override
Evaluate product of PDFs in batch mode.
Definition: RooProdPdf.cxx:491
RooProdPdf()
Default constructor.
Definition: RooProdPdf.cxx:82
RooAbsGenContext * genContext(const RooArgSet &vars, const RooDataSet *prototype=0, const RooArgSet *auxProto=0, bool verbose=false) const override
Return generator context optimized for generating events from product p.d.f.s.
bool checkObservables(const RooArgSet *nset) const override
Obsolete.
bool _useDefaultGen
Use default or distributed event generator.
Definition: RooProdPdf.h:176
std::vector< std::unique_ptr< RooArgSet > > _pdfNSetList
List of PDF component normalization sets.
Definition: RooProdPdf.h:172
void useDefaultGen(bool flag=true)
Definition: RooProdPdf.h:175
Int_t getAnalyticalIntegralWN(RooArgSet &allVars, RooArgSet &numVars, const RooArgSet *normSet, const char *rangeName=0) const override
Determine which part (if any) of given integral can be performed analytically.
bool forceAnalyticalInt(const RooAbsArg &dep) const override
Force RooRealIntegral to offer all observables for internal integration.
void writeCacheToStream(std::ostream &os, RooArgSet const *nset) const
void getParametersHook(const RooArgSet *, RooArgSet *, bool stripDisconnected) const override
void setSelfNormalized(bool flag)
Definition: RooProdPdf.h:88
TNamed * _refRangeName
Reference range name for interpretation of conditional products.
Definition: RooProdPdf.h:178
RooAICRegistry _genCode
! Registry of composite direct generator codes
Definition: RooProdPdf.h:168
RooListProxy _pdfList
List of PDF components.
Definition: RooProdPdf.h:171
void selectNormalizationRange(const char *rangeName=0, bool force=false) override
Interface function used by test statistics to freeze choice of range for interpretation of conditiona...
void printMetaArgs(std::ostream &os) const override
Customized printing of arguments of a RooProdPdf to more intuitively reflect the contents of the prod...
std::list< double > * binBoundaries(RooAbsRealLValue &, double, double) const override
Forward the plot sampling hint from the p.d.f. that defines the observable obs.
RooAbsReal * specializeIntegral(RooAbsReal &orig, const char *targetRangeName) const
void fillNormSetForServer(RooArgSet const &normSet, RooAbsArg const &server, RooArgSet &serverNormSet) const override
Fills a RooArgSet to be used as the normalization set for a server, given a normalization set for thi...
CacheMode canNodeBeCached() const override
Definition: RooProdPdf.h:132
RooObjCacheManager _cacheMgr
Definition: RooProdPdf.h:154
std::string makeRGPPName(const char *pfx, const RooArgSet &term, const RooArgSet &iset, const RooArgSet &nset, const char *isetRangeName) const
Make an appropriate automatic name for a RooGenProdProj object in getPartIntList()
bool isDirectGenSafe(const RooAbsArg &arg) const override
Forward determination of safety of internal generator code to component p.d.f that would generate the...
RooArgSet * findPdfNSet(RooAbsPdf const &pdf) const
Look up user specified normalization set for given input PDF component.
RooAbsReal * makeCondPdfRatioCorr(RooAbsReal &term, const RooArgSet &termNset, const RooArgSet &termImpSet, const char *normRange, const char *refRange) const
For single normalization ranges.
ExtendMode extendMode() const override
If this product contains exactly one extendable p.d.f return the extension abilities of that p....
std::list< double > * plotSamplingHint(RooAbsRealLValue &obs, double xlo, double xhi) const override
Forward the plot sampling hint from the p.d.f. that defines the observable obs.
double expectedEvents(const RooArgSet *nset) const override
Return the expected number of events associated with the extendable input PDF in the product.
std::vector< RooAbsReal * > processProductTerm(const RooArgSet *nset, const RooArgSet *iset, const char *isetRangeName, const RooArgSet *term, const RooArgSet &termNSet, const RooArgSet &termISet, bool &isOwned, bool forceWrap=false) const
Calculate integrals of factorized product terms over observables iset while normalized to observables...
TObject * clone(const char *newname) const override
Definition: RooProdPdf.h:56
bool isBinnedDistribution(const RooArgSet &obs) const override
If all components that depend on obs are binned that so is the product.
RooArgSet * getConnectedParameters(const RooArgSet &observables) const
Return all parameter constraint p.d.f.s on parameters listed in constrainedParams.
double calculate(const RooProdPdf::CacheElem &cache, bool verbose=false) const
Calculate running product of pdfs terms, using the supplied normalization set in 'normSetList' for ea...
Definition: RooProdPdf.cxx:460
RooArgSet _defNormSet
Default normalization set.
Definition: RooProdPdf.h:181
RooArgSet * getConstraints(const RooArgSet &observables, RooArgSet &constrainedParams, bool stripDisconnected) const override
Return all parameter constraint p.d.f.s on parameters listed in constrainedParams.
RooAbsReal * specializeRatio(RooFormulaVar &input, const char *targetRangeName) const
CacheElem * getCacheElem(RooArgSet const *nset) const
The cache manager.
Definition: RooProdPdf.cxx:433
bool redirectServersHook(const RooAbsCollection &, bool, bool, bool) override
Implement support for node removal.
void initGenerator(Int_t code) override
Forward one-time initialization call to component generation initialization methods.
void generateEvent(Int_t code) override
Generate a single event with configuration specified by 'code' Defer internal generation to component...
const RooArgList & pdfList() const
Definition: RooProdPdf.h:69
void groupProductTerms(std::list< std::vector< RooArgSet * > > &groupedTerms, RooArgSet &outerIntDeps, const RooLinkedList &terms, const RooLinkedList &norms, const RooLinkedList &imps, const RooLinkedList &ints, const RooLinkedList &cross) const
Group product into terms that can be calculated independently.
void fixRefRange(const char *rangeName)
double analyticalIntegralWN(Int_t code, const RooArgSet *normSet, const char *rangeName=0) const override
Return analytical integral defined by given scenario code.
double evaluate() const override
Calculate current value of object.
Definition: RooProdPdf.cxx:449
void initializeFromCmdArgList(const RooArgSet &fullPdfSet, const RooLinkedList &l)
Initialize RooProdPdf configuration from given list of RooCmdArg configuration arguments and set of '...
Definition: RooProdPdf.cxx:368
bool _selfNorm
Is self-normalized.
Definition: RooProdPdf.h:180
double _cutOff
Cutoff parameter for running product.
Definition: RooProdPdf.h:170
bool canComputeBatchWithCuda() const override
Definition: RooProdPdf.h:106
void setDefNormSet(const RooArgSet &nset)
Definition: RooProdPdf.h:89
The TNamed class is the base class for all named ROOT classes.
Definition: TNamed.h:29
Mother of all ROOT objects.
Definition: TObject.h:37
std::map< DataKey, RooSpan< const double > > DataMap
Definition: DataMap.h:59
auto * l
Definition: textangle.C:4
static void output()