Logo ROOT   6.18/05
Reference Guide
RooFFTConvPdf.h
Go to the documentation of this file.
1/*****************************************************************************
2 * Project: RooFit *
3 * *
4 * Copyright (c) 2000-2005, Regents of the University of California *
5 * and Stanford University. All rights reserved. *
6 * *
7 * Redistribution and use in source and binary forms, *
8 * with or without modification, are permitted according to the terms *
9 * listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
10 *****************************************************************************/
11
12#ifndef ROOFFTCONVPDF
13#define ROOFFTCONVPDF
14
15#include "RooAbsCachedPdf.h"
16#include "RooRealProxy.h"
17#include "RooSetProxy.h"
18#include "RooAbsReal.h"
19#include "RooHistPdf.h"
20#include "TVirtualFFT.h"
21class RooRealVar ;
22
23#include <map>
24
25///PDF for the numerical (FFT) convolution of two PDFs.
27public:
28
30 // coverity[UNINIT_CTOR]
31 } ;
32 RooFFTConvPdf(const char *name, const char *title, RooRealVar& convVar, RooAbsPdf& pdf1, RooAbsPdf& pdf2, Int_t ipOrder=2);
33 RooFFTConvPdf(const char *name, const char *title, RooAbsReal& pdfConvVar, RooRealVar& convVar, RooAbsPdf& pdf1, RooAbsPdf& pdf2, Int_t ipOrder=2);
34 RooFFTConvPdf(const RooFFTConvPdf& other, const char* name=0) ;
35 virtual TObject* clone(const char* newname) const { return new RooFFTConvPdf(*this,newname); }
36 virtual ~RooFFTConvPdf() ;
37
38 void setShift(Double_t val1, Double_t val2) { _shift1 = val1 ; _shift2 = val2 ; }
40 const RooArgSet& cacheObservables() const { return _cacheObs ; }
41
43 // Return value of buffer fraction applied in FFT calculation array beyond either
44 // end of the observable domain to reduce cyclical effects
45 return _bufFrac ;
46 }
47
48 enum BufStrat { Extend=0, Mirror=1, Flat=2 } ;
50 // Return the strategy currently used to fill the buffer:
51 // 'Extend' means is that the input p.d.f convolution observable range is widened to include the buffer range
52 // 'Flat' means that the buffer is filled with the p.d.f. value at the boundary of the observable range
53 // 'Mirror' means that the buffer is filled with a mirror image of the p.d.f. around the convolution observable boundary
54 return _bufStrat ;
55 }
57 void setBufferFraction(Double_t frac) ;
58
59 void printMetaArgs(std::ostream& os) const ;
60
61 // Propagate maximum value estimate of pdf1 as convolution can only result in lower max values
62 virtual Int_t getMaxVal(const RooArgSet& vars) const { return _pdf1.arg().getMaxVal(vars) ; }
63 virtual Double_t maxVal(Int_t code) const { return _pdf1.arg().maxVal(code) ; }
64
65
66protected:
67
68 RooRealProxy _x ; // Convolution observable
69 RooRealProxy _xprime ; // Input function representing value of convolution observable
70 RooRealProxy _pdf1 ; // First input p.d.f
71 RooRealProxy _pdf2 ; // Second input p.d.f
72 RooSetProxy _params ; // Effective parameters of this p.d.f.
73
74 void calcParams() ;
75 Bool_t redirectServersHook(const RooAbsCollection& newServerList, Bool_t mustReplaceAll, Bool_t nameChange, Bool_t isRecursive) ;
76
77 Double_t* scanPdf(RooRealVar& obs, RooAbsPdf& pdf, const RooDataHist& hist, const RooArgSet& slicePos, Int_t& N, Int_t& N2, Int_t& zeroBin, Double_t shift) const ;
78
79 class FFTCacheElem : public PdfCacheElem {
80 public:
81 FFTCacheElem(const RooFFTConvPdf& self, const RooArgSet* nset) ;
83
85
89
92
95
96 };
97
98 friend class FFTCacheElem ;
99
100 virtual Double_t evaluate() const { RooArgSet dummy(_x.arg()) ; return getVal(&dummy) ; } ; // dummy
101 virtual const char* inputBaseName() const ;
102 virtual RooArgSet* actualObservables(const RooArgSet& nset) const ;
103 virtual RooArgSet* actualParameters(const RooArgSet& nset) const ;
104 virtual RooAbsArg& pdfObservable(RooAbsArg& histObservable) const ;
105 virtual void fillCacheObject(PdfCacheElem& cache) const ;
106 void fillCacheSlice(FFTCacheElem& cache, const RooArgSet& slicePosition) const ;
107
108 virtual PdfCacheElem* createCache(const RooArgSet* nset) const ;
109 virtual TString histNameSuffix() const ;
110
111 // mutable std::map<const RooHistPdf*,CacheAuxInfo*> _cacheAuxInfo ; //! Auxilary Cache information (do not persist)
112 Double_t _bufFrac ; // Sampling buffer size as fraction of domain size
113 BufStrat _bufStrat ; // Strategy to fill the buffer
114
117
118 virtual RooAbsGenContext* genContext(const RooArgSet &vars, const RooDataSet *prototype=0,
119 const RooArgSet* auxProto=0, Bool_t verbose= kFALSE) const ;
120
121 friend class RooConvGenContext ;
122 RooSetProxy _cacheObs ; // Non-convolution observables that are also cached
123
124private:
125
126 void prepareFFTBinning(RooRealVar& convVar) const;
127
128 ClassDef(RooFFTConvPdf,1) // Convolution operator p.d.f based on numeric Fourier transforms
129};
130
131#endif
static RooMathCoreReg dummy
int Int_t
Definition: RtypesCore.h:41
const Bool_t kFALSE
Definition: RtypesCore.h:88
bool Bool_t
Definition: RtypesCore.h:59
double Double_t
Definition: RtypesCore.h:55
#define ClassDef(name, id)
Definition: Rtypes.h:326
#define N
char name[80]
Definition: TGX11.cxx:109
RooAbsArg is the common abstract base class for objects that represent a value (of arbitrary type) an...
Definition: RooAbsArg.h:70
RooAbsBinning is the abstract base class for RooRealVar binning definitions This class defines the in...
Definition: RooAbsBinning.h:26
RooAbsCachedPdf is the abstract base class for p.d.f.s that need or want to cache their evaluate() ou...
friend class PdfCacheElem
RooAbsCollection is an abstract container object that can hold multiple RooAbsArg objects.
RooAbsGenContext is the abstract base class for generator contexts of RooAbsPdf objects.
RooAbsReal is the common abstract base class for objects that represent a real value and implements f...
Definition: RooAbsReal.h:53
Double_t getVal(const RooArgSet *normalisationSet=nullptr) const
Evaluate object.
Definition: RooAbsReal.h:81
virtual Double_t maxVal(Int_t code) const
Return maximum value for set of observables identified by code assigned in getMaxVal.
virtual Int_t getMaxVal(const RooArgSet &vars) const
Advertise capability to determine maximum value of function for given set of observables.
RooArgList is a container object that can hold multiple RooAbsArg objects.
Definition: RooArgList.h:21
RooArgSet is a container object that can hold multiple RooAbsArg objects.
Definition: RooArgSet.h:28
RooConvGenContext is an efficient implementation of the generator context specific for RooAbsAnaConvP...
The RooDataHist is a container class to hold N-dimensional binned data.
Definition: RooDataHist.h:40
RooDataSet is a container class to hold unbinned data.
Definition: RooDataSet.h:31
virtual RooArgList containedArgs(Action)
Returns all RooAbsArg objects contained in the cache element.
RooAbsBinning * histBinning
Definition: RooFFTConvPdf.h:93
FFTCacheElem(const RooFFTConvPdf &self, const RooArgSet *nset)
Clone input pdf and attach to dataset.
RooAbsBinning * scanBinning
Definition: RooFFTConvPdf.h:94
PDF for the numerical (FFT) convolution of two PDFs.
Definition: RooFFTConvPdf.h:26
Bool_t redirectServersHook(const RooAbsCollection &newServerList, Bool_t mustReplaceAll, Bool_t nameChange, Bool_t isRecursive)
calcParams() ;
void printMetaArgs(std::ostream &os) const
Customized printing of arguments of a RooNumConvPdf to more intuitively reflect the contents of the p...
virtual PdfCacheElem * createCache(const RooArgSet *nset) const
Return specialized cache subclass for FFT calculations.
virtual ~RooFFTConvPdf()
Destructor.
RooSetProxy _params
Definition: RooFFTConvPdf.h:72
BufStrat _bufStrat
void calcParams()
(Re)calculate effective parameters of this p.d.f.
Double_t _shift2
virtual void fillCacheObject(PdfCacheElem &cache) const
Fill the contents of the cache the FFT convolution output.
BufStrat bufferStrategy() const
Definition: RooFFTConvPdf.h:49
void setCacheObservables(const RooArgSet &obs)
Definition: RooFFTConvPdf.h:39
void prepareFFTBinning(RooRealVar &convVar) const
Try to improve the binning and inform user if possible.
virtual TString histNameSuffix() const
Suffix for cache histogram (added in addition to suffix for cache name)
Double_t _bufFrac
void fillCacheSlice(FFTCacheElem &cache, const RooArgSet &slicePosition) const
Fill a slice of cachePdf with the output of the FFT convolution calculation.
RooRealProxy _xprime
Definition: RooFFTConvPdf.h:69
virtual Double_t maxVal(Int_t code) const
Return maximum value for set of observables identified by code assigned in getMaxVal.
Definition: RooFFTConvPdf.h:63
Double_t * scanPdf(RooRealVar &obs, RooAbsPdf &pdf, const RooDataHist &hist, const RooArgSet &slicePos, Int_t &N, Int_t &N2, Int_t &zeroBin, Double_t shift) const
Scan the values of 'pdf' in observable 'obs' using the bin values stored in 'hist' at slice position ...
RooRealProxy _pdf1
Definition: RooFFTConvPdf.h:70
RooRealProxy _x
Definition: RooFFTConvPdf.h:68
virtual RooAbsGenContext * genContext(const RooArgSet &vars, const RooDataSet *prototype=0, const RooArgSet *auxProto=0, Bool_t verbose=kFALSE) const
Create appropriate generator context for this convolution.
const RooArgSet & cacheObservables() const
Definition: RooFFTConvPdf.h:40
void setShift(Double_t val1, Double_t val2)
Definition: RooFFTConvPdf.h:38
virtual const char * inputBaseName() const
Return base name component for cache components in this case 'PDF1_CONV_PDF2'.
void setBufferFraction(Double_t frac)
Change the size of the buffer on either side of the observable range to frac times the size of the ra...
virtual RooArgSet * actualObservables(const RooArgSet &nset) const
Return the observables to be cached given the normalization set nset.
RooRealProxy _pdf2
Definition: RooFFTConvPdf.h:71
virtual RooAbsArg & pdfObservable(RooAbsArg &histObservable) const
Return p.d.f.
void setBufferStrategy(BufStrat bs)
Change strategy to fill the overflow buffer on either side of the convolution observable range.
Double_t _shift1
virtual Int_t getMaxVal(const RooArgSet &vars) const
Advertise capability to determine maximum value of function for given set of observables.
Definition: RooFFTConvPdf.h:62
virtual TObject * clone(const char *newname) const
Definition: RooFFTConvPdf.h:35
virtual RooArgSet * actualParameters(const RooArgSet &nset) const
Return the parameters on which the cache depends given normalization set nset.
Double_t bufferFraction() const
Definition: RooFFTConvPdf.h:42
virtual Double_t evaluate() const
Evaluate this PDF / function / constant. Needs to be overridden by all derived classes.
friend class FFTCacheElem
Definition: RooFFTConvPdf.h:98
RooSetProxy _cacheObs
RooRealProxy is the concrete proxy for RooAbsReal objects A RooRealProxy is the general mechanism to ...
Definition: RooRealProxy.h:23
const RooAbsReal & arg() const
Definition: RooRealProxy.h:43
RooRealVar represents a fundamental (non-derived) real valued object.
Definition: RooRealVar.h:36
RooSetProxy is the concrete proxy for RooArgSet objects.
Definition: RooSetProxy.h:24
virtual Bool_t add(const RooAbsArg &var, Bool_t silent=kFALSE)
Overloaded RooArgSet::add() method inserts 'var' into set and registers 'var' as server to owner with...
virtual void removeAll()
Remove all argument inset using remove(const RooAbsArg&).
Mother of all ROOT objects.
Definition: TObject.h:37
Basic string class.
Definition: TString.h:131
TVirtualFFT is an interface class for Fast Fourier Transforms.
Definition: TVirtualFFT.h:88