Logo ROOT   6.16/01
Reference Guide
RooCFunction2Binding.h
Go to the documentation of this file.
1/*****************************************************************************
2 * Project: RooFit *
3 * Package: RooFitCore *
4 * File: $Id$
5 * Authors: *
6 * WV, Wouter Verkerke, NIKHEF, verkerke@nikhef.nl *
7 * *
8 * Copyright (c) 2000-2008, NIKHEF, Regents of the University of California *
9 * and Stanford University. All rights reserved. *
10 * *
11 *****************************************************************************/
12
13#ifndef ROOCFUNCTION2BINDING
14#define ROOCFUNCTION2BINDING
15
16#include "RooAbsReal.h"
17#include "RooAbsPdf.h"
18#include "RooRealProxy.h"
19#include "RooMsgService.h"
20
21#include "TBuffer.h"
22#include "TString.h"
23
24#include <string>
25#include <map>
26#include <vector>
27
28namespace RooFit {
29
35
36
42RooAbsPdf* bindPdf(const char* name,CFUNCD2DD func,RooAbsReal& x, RooAbsReal& y) ;
43RooAbsPdf* bindPdf(const char* name,CFUNCD2ID func,RooAbsReal& x, RooAbsReal& y) ;
44RooAbsPdf* bindPdf(const char* name,CFUNCD2UD func,RooAbsReal& x, RooAbsReal& y) ;
45RooAbsPdf* bindPdf(const char* name,CFUNCD2DI func,RooAbsReal& x, RooAbsReal& y) ;
46RooAbsPdf* bindPdf(const char* name,CFUNCD2II func,RooAbsReal& x, RooAbsReal& y) ;
47
48}
49
50template<class VO, class VI1, class VI2>
52 public:
54
55 void add(const char* name, VO (*ptr)(VI1,VI2), const char* arg1name="x", const char* arg2name="y") {
56 // Register function with given name and argument name
57 _ptrmap[name] = ptr ;
58 _namemap[ptr] = name ;
59 _argnamemap[ptr].push_back(arg1name) ;
60 _argnamemap[ptr].push_back(arg2name) ;
61 }
62
63
64 const char* lookupName(VO (*ptr)(VI1,VI2)) {
65 // Return name of function given by pointer
66 return _namemap[ptr].c_str() ;
67 }
68
69 VO (*lookupPtr(const char* name))(VI1,VI2) {
70 // Return pointer of function given by name
71 return _ptrmap[name] ;
72 }
73
74 const char* lookupArgName(VO (*ptr)(VI1,VI2), UInt_t iarg) {
75 // Return name of i-th argument of function. If function is
76 // not registered, argument names 0,1,2 are x,y,z
77 if (iarg<_argnamemap[ptr].size()) {
78 return (_argnamemap[ptr])[iarg].c_str() ;
79 }
80 switch (iarg) {
81 case 0: return "x" ;
82 case 1: return "y" ;
83 case 2: return "z" ;
84 }
85 return "w" ;
86 }
87
88 private:
89
90#ifndef __CINT__
91 std::map<std::string,VO (*)(VI1,VI2)> _ptrmap ; // Pointer-to-name map
92 std::map<VO (*)(VI1,VI2),std::string> _namemap ; // Name-to-pointer map
93 std::map<VO (*)(VI1,VI2),std::vector<std::string> > _argnamemap ; // Pointer-to-argnamelist map
94#endif
95} ;
96
97
98
99template<class VO, class VI1, class VI2>
100class RooCFunction2Ref : public TObject {
101 public:
102 RooCFunction2Ref(VO (*ptr)(VI1,VI2)=0) : _ptr(ptr) {
103 // Constructor of persistable function reference
104 } ;
106
107 VO operator()(VI1 x,VI2 y) const {
108 // Evaluate embedded function
109 return (*_ptr)(x,y) ;
110 }
111
112 const char* name() const {
113 // Return registered name of embedded function. If function
114 // is not registered return string with hex presentation
115 // of function pointer value
116 const char* result = fmap().lookupName(_ptr) ;
117 if (result && strlen(result)) {
118 return result ;
119 }
120 // This union is to avoid a warning message:
121 union {
122 void *_ptr;
123 func_t _funcptr;
124 } temp;
125 temp._funcptr = _ptr;
126 return Form("(%p)",temp._ptr) ;
127 }
128
129 const char* argName(Int_t iarg) {
130 // Return suggested name for i-th argument
131 return fmap().lookupArgName(_ptr,iarg) ;
132 }
133
135 // Return reference to function pointer-to-name mapping service
136 if (!_fmap) {
138 }
139 return *_fmap ;
140 }
141
142 private:
143
144 static VO dummyFunction(VI1,VI2) {
145 // Dummy function used when registered function was not
146 // found in un-persisting object
147 return 0 ;
148 }
149
150 typedef VO (*func_t)(VI1,VI2);
151 func_t _ptr; //! Pointer to embedded function
152
153 static RooCFunction2Map<VO,VI1,VI2>* _fmap ; // Pointer to mapping service object
154
155 ClassDef(RooCFunction2Ref,1) // Persistable reference to C function pointer
156} ;
157
158// Define static member
159template<class VO, class VI1, class VI2>
161
162
163template<class VO, class VI1, class VI2>
165{
166 // Custom streamer for function pointer reference object. When writing,
167 // the function pointer is substituted by its registerd name. When function
168 // is unregistered name 'UNKNOWN' is written and a warning is issues. When
169 // reading back, the embedded name is converted back to a function pointer
170 // using the mapping service. When name UNKNOWN is encountered a warning is
171 // issues and a dummy null function is substituted. When the registered function
172 // name can not be mapped to a function pointer an ERROR is issued and a pointer
173 // to the dummy null function is substituted
174
175 typedef ::RooCFunction2Ref<VO,VI1,VI2> thisClass;
176
177 // Stream an object of class RooCFunction2Ref
178 if (R__b.IsReading()) {
179
180 UInt_t R__s, R__c;
181 Version_t R__v = R__b.ReadVersion(&R__s, &R__c);
182
183 // Read name from file
184 TString tmpName ;
185 tmpName.Streamer(R__b) ;
186
187 if (tmpName=="UNKNOWN" && R__v>0) {
188
189 coutW(ObjectHandling) << "WARNING: Objected embeds function pointer to unknown function, object will not be functional" << std::endl ;
190 _ptr = dummyFunction ;
191
192 } else {
193
194 // Lookup pointer to C function with given name
195 _ptr = fmap().lookupPtr(tmpName.Data()) ;
196
197 if (_ptr==0) {
198 coutW(ObjectHandling) << "ERROR: Objected embeds pointer to function named " << tmpName
199 << " but no such function is registered, object will not be functional" << std::endl ;
200 }
201 }
202
203
204 R__b.CheckByteCount(R__s, R__c, thisClass::IsA());
205
206 } else {
207
208 UInt_t R__c;
209 R__c = R__b.WriteVersion(thisClass::IsA(), kTRUE);
210
211 // Lookup name of reference C function
212 TString tmpName = fmap().lookupName(_ptr) ;
213 if (tmpName.Length()==0) {
214 coutW(ObjectHandling) << "WARNING: Cannot persist unknown function pointer " << Form("0x%lx", (ULong_t)_ptr)
215 << " written object will not be functional when read back" << std::endl ;
216 tmpName="UNKNOWN" ;
217 }
218
219 // Persist the name
220 tmpName.Streamer(R__b) ;
221
222 R__b.SetByteCount(R__c, kTRUE);
223
224 }
225}
226
227
228
229template<class VO,class VI1, class VI2>
231public:
233 // Default constructor
234 } ;
235 RooCFunction2Binding(const char *name, const char *title, VO (*_func)(VI1,VI2), RooAbsReal& _x, RooAbsReal& _y);
236 RooCFunction2Binding(const RooCFunction2Binding& other, const char* name=0) ;
237 virtual TObject* clone(const char* newname) const { return new RooCFunction2Binding(*this,newname); }
238 inline virtual ~RooCFunction2Binding() { }
239
240 void printArgs(std::ostream& os) const {
241 // Print object arguments and name/address of function pointer
242 os << "[ function=" << func.name() << " " ;
243 for (Int_t i=0 ; i<numProxies() ; i++) {
244 RooAbsProxy* p = getProxy(i) ;
245 if (!TString(p->name()).BeginsWith("!")) {
246 p->print(os) ;
247 os << " " ;
248 }
249 }
250 os << "]" ;
251 }
252
253protected:
254
255 RooCFunction2Ref<VO,VI1,VI2> func ; // Function pointer reference
256 RooRealProxy x ; // Argument reference
257 RooRealProxy y ; // Argument reference
258
260 // Return value of embedded function using value of referenced variable x
261 return func(x,y) ;
262 }
263
264private:
265
266 ClassDef(RooCFunction2Binding,1) // RooAbsReal binding to external C functions
267};
268
269template<class VO,class VI1, class VI2>
270RooCFunction2Binding<VO,VI1,VI2>::RooCFunction2Binding(const char *name, const char *title, VO (*_func)(VI1,VI2),
271 RooAbsReal& _x, RooAbsReal& _y) :
272 RooAbsReal(name,title),
273 func(_func),
274 x(func.argName(0),func.argName(0),this,_x),
275 y(func.argName(1),func.argName(1),this,_y)
276{
277 // Constructor of C function binding object given a pointer to a function and a RooRealVar to which the function
278 // argument should be bound. This object is fully functional as a RooFit function object. The only restriction is
279 // if the referenced function is _not_ a standard ROOT TMath or MathCore function it can not be persisted in a
280 // a RooWorkspace
281}
282
283
284template<class VO,class VI1, class VI2>
286 RooAbsReal(other,name),
287 func(other.func),
288 x("x",this,other.x),
289 y("y",this,other.y)
290{
291 // Copy constructor
292}
293
294
295
296
297template<class VO,class VI1, class VI2>
299public:
301 // Default constructor
302 } ;
303 RooCFunction2PdfBinding(const char *name, const char *title, VO (*_func)(VI1,VI2), RooAbsReal& _x, RooAbsReal& _y);
304 RooCFunction2PdfBinding(const RooCFunction2PdfBinding& other, const char* name=0) ;
305 virtual TObject* clone(const char* newname) const { return new RooCFunction2PdfBinding(*this,newname); }
306 inline virtual ~RooCFunction2PdfBinding() { }
307
308 void printArgs(std::ostream& os) const {
309 // Print object arguments and name/address of function pointer
310 os << "[ function=" << func.name() << " " ;
311 for (Int_t i=0 ; i<numProxies() ; i++) {
312 RooAbsProxy* p = getProxy(i) ;
313 if (!TString(p->name()).BeginsWith("!")) {
314 p->print(os) ;
315 os << " " ;
316 }
317 }
318 os << "]" ;
319 }
320
321protected:
322
323 RooCFunction2Ref<VO,VI1,VI2> func ; // Function pointer reference
324 RooRealProxy x ; // Argument reference
325 RooRealProxy y ; // Argument reference
326
328 // Return value of embedded function using value of referenced variable x
329 return func(x,y) ;
330 }
331
332private:
333
334 ClassDef(RooCFunction2PdfBinding,1) // RooAbsReal binding to external C functions
335};
336
337template<class VO,class VI1, class VI2>
338RooCFunction2PdfBinding<VO,VI1,VI2>::RooCFunction2PdfBinding(const char *name, const char *title, VO (*_func)(VI1,VI2),
339 RooAbsReal& _x, RooAbsReal& _y) :
340 RooAbsPdf(name,title),
341 func(_func),
342 x(func.argName(0),func.argName(0),this,_x),
343 y(func.argName(1),func.argName(1),this,_y)
344{
345 // Constructor of C function binding object given a pointer to a function and a RooRealVar to which the function
346 // argument should be bound. This object is fully functional as a RooFit function object. The only restriction is
347 // if the referenced function is _not_ a standard ROOT TMath or MathCore function it can not be persisted in a
348 // a RooWorkspace
349}
350
351
352template<class VO,class VI1, class VI2>
354 RooAbsPdf(other,name),
355 func(other.func),
356 x("x",this,other.x),
357 y("y",this,other.y)
358{
359 // Copy constructor
360}
361
362#endif
#define coutW(a)
Definition: RooMsgService.h:33
int Int_t
Definition: RtypesCore.h:41
short Version_t
Definition: RtypesCore.h:61
unsigned int UInt_t
Definition: RtypesCore.h:42
unsigned long ULong_t
Definition: RtypesCore.h:51
double Double_t
Definition: RtypesCore.h:55
const Bool_t kTRUE
Definition: RtypesCore.h:87
#define ClassDef(name, id)
Definition: Rtypes.h:324
char * Form(const char *fmt,...)
Int_t numProxies() const
Return the number of registered proxies.
Definition: RooAbsArg.cxx:1259
RooAbsProxy * getProxy(Int_t index) const
Return the nth proxy from the proxy list.
Definition: RooAbsArg.cxx:1246
RooAbsPdf is the abstract interface for all probability density functions The class provides hybrid a...
Definition: RooAbsPdf.h:41
RooAbsProxy is the abstact interface for proxy classes.
Definition: RooAbsProxy.h:31
virtual void print(std::ostream &os, Bool_t addContents=kFALSE) const
Print proxy name.
Definition: RooAbsProxy.cxx:75
virtual const char * name() const
Definition: RooAbsProxy.h:41
RooAbsReal is the common abstract base class for objects that represent a real value and implements f...
Definition: RooAbsReal.h:53
RooCFunction2Binding is a templated implementation of class RooAbsReal that binds generic C(++) funct...
void printArgs(std::ostream &os) const
Print object arguments, ie its proxies.
RooCFunction2Ref< VO, VI1, VI2 > func
Double_t evaluate() const
Evaluate this PDF / function / constant. Needs to be overridden by all derived classes.
virtual TObject * clone(const char *newname) const
std::map< VO(*)(VI1, VI2), std::vector< std::string > > _argnamemap
std::map< std::string, VO(*)(VI1, VI2)> _ptrmap
std::map< VO(*)(VI1, VI2), std::string > _namemap
void add(const char *name, VO(*ptr)(VI1, VI2), const char *arg1name="x", const char *arg2name="y")
VO(*)(VI1, VI2) lookupPtr(const char *name)
const char * lookupArgName(VO(*ptr)(VI1, VI2), UInt_t iarg)
const char * lookupName(VO(*ptr)(VI1, VI2))
virtual TObject * clone(const char *newname) const
void printArgs(std::ostream &os) const
Print object arguments, ie its proxies.
RooCFunction2Ref< VO, VI1, VI2 > func
Double_t evaluate() const
Evaluate this PDF / function / constant. Needs to be overridden by all derived classes.
const char * argName(Int_t iarg)
VO operator()(VI1 x, VI2 y) const
static RooCFunction2Map< VO, VI1, VI2 > & fmap()
static VO dummyFunction(VI1, VI2)
RooCFunction2Ref(VO(*ptr)(VI1, VI2)=0)
const char * name() const
VO(* func_t)(VI1, VI2)
static RooCFunction2Map< VO, VI1, VI2 > * _fmap
Pointer to embedded function.
RooRealProxy is the concrete proxy for RooAbsReal objects A RooRealProxy is the general mechanism to ...
Definition: RooRealProxy.h:23
Buffer base class used for serializing objects.
Definition: TBuffer.h:40
virtual void SetByteCount(UInt_t cntpos, Bool_t packInVersion=kFALSE)=0
virtual Version_t ReadVersion(UInt_t *start=0, UInt_t *bcnt=0, const TClass *cl=0)=0
virtual Int_t CheckByteCount(UInt_t startpos, UInt_t bcnt, const TClass *clss)=0
Bool_t IsReading() const
Definition: TBuffer.h:83
virtual UInt_t WriteVersion(const TClass *cl, Bool_t useBcnt=kFALSE)=0
Mother of all ROOT objects.
Definition: TObject.h:37
Basic string class.
Definition: TString.h:131
Ssiz_t Length() const
Definition: TString.h:405
const char * Data() const
Definition: TString.h:364
Bool_t BeginsWith(const char *s, ECaseCompare cmp=kExact) const
Definition: TString.h:610
Double_t y[n]
Definition: legend1.C:17
Double_t x[n]
Definition: legend1.C:17
Double_t(* CFUNCD2UD)(UInt_t, Double_t)
Double_t(* CFUNCD2DI)(Double_t, Int_t)
RooAbsPdf * bindPdf(const char *name, CFUNCD1D func, RooAbsReal &x)
Double_t(* CFUNCD2II)(Int_t, Int_t)
Double_t(* CFUNCD2DD)(Double_t, Double_t)
@ ObjectHandling
Definition: RooGlobalFunc.h:58
RooAbsReal * bindFunction(const char *name, CFUNCD1D func, RooAbsReal &x)
Double_t(* CFUNCD2ID)(Int_t, Double_t)