Logo ROOT  
Reference Guide
RooPoisson.h
Go to the documentation of this file.
1 /*****************************************************************************
2 * Project: RooFit *
3 * *
4 * Simple Poisson PDF
5 * author: Kyle Cranmer <cranmer@cern.ch>
6 * *
7 *****************************************************************************/
8
9#ifndef ROOPOISSON
10#define ROOPOISSON
11
12#include "RooAbsPdf.h"
13#include "RooRealProxy.h"
14#include "RooCategoryProxy.h"
15#include "RooAbsReal.h"
16#include "RooAbsCategory.h"
17#include "RooTrace.h"
18
19class RooPoisson : public RooAbsPdf {
20public:
22 RooPoisson(const char *name, const char *title, RooAbsReal& _x, RooAbsReal& _mean, Bool_t noRounding=kFALSE);
23 RooPoisson(const RooPoisson& other, const char* name=0) ;
24 virtual TObject* clone(const char* newname) const override { return new RooPoisson(*this,newname); }
25 inline virtual ~RooPoisson() { }
26
27 Int_t getAnalyticalIntegral(RooArgSet& allVars, RooArgSet& analVars, const char* rangeName=0) const override;
28 Double_t analyticalIntegral(Int_t code, const char* rangeName=0) const override;
29
30 Int_t getGenerator(const RooArgSet& directVars, RooArgSet &generateVars, Bool_t staticInitOK=kTRUE) const override;
31 void generateEvent(Int_t code) override;
32
33 /// Switch off/on rounding of `x` to the nearest integer.
34 void setNoRounding(bool flag = kTRUE) {_noRounding = flag;}
35 /// Switch on or off protection against negative means.
36 void protectNegativeMean(bool flag = kTRUE) {_protectNegative = flag;}
37
38 /// Get the x variable.
39 RooAbsReal const& getX() const { return x.arg(); }
40
41 /// Get the mean parameter.
42 RooAbsReal const& getMean() const { return mean.arg(); }
43
44protected:
45
50
51 Double_t evaluate() const override;
52 void computeBatch(cudaStream_t*, double* output, size_t nEvents, RooBatchCompute::DataMap&) const override;
53 inline bool canComputeBatchWithCuda() const override { return true; }
54
55 ClassDefOverride(RooPoisson,3) // A Poisson PDF
56};
57
58#endif
int Int_t
Definition: RtypesCore.h:45
const Bool_t kFALSE
Definition: RtypesCore.h:101
bool Bool_t
Definition: RtypesCore.h:63
double Double_t
Definition: RtypesCore.h:59
const Bool_t kTRUE
Definition: RtypesCore.h:100
#define ClassDefOverride(name, id)
Definition: Rtypes.h:329
char name[80]
Definition: TGX11.cxx:110
RooAbsReal is the common abstract base class for objects that represent a real value and implements f...
Definition: RooAbsReal.h:63
RooArgSet is a container object that can hold multiple RooAbsArg objects.
Definition: RooArgSet.h:35
Poisson pdf.
Definition: RooPoisson.h:19
RooRealProxy x
Definition: RooPoisson.h:46
Int_t getAnalyticalIntegral(RooArgSet &allVars, RooArgSet &analVars, const char *rangeName=0) const override
Interface function getAnalyticalIntergral advertises the analytical integrals that are supported.
Definition: RooPoisson.cxx:76
virtual ~RooPoisson()
Definition: RooPoisson.h:25
bool canComputeBatchWithCuda() const override
Definition: RooPoisson.h:53
RooAbsReal const & getX() const
Get the x variable.
Definition: RooPoisson.h:39
void protectNegativeMean(bool flag=kTRUE)
Switch on or off protection against negative means.
Definition: RooPoisson.h:36
Bool_t _protectNegative
Definition: RooPoisson.h:49
void generateEvent(Int_t code) override
Implement internal generator using TRandom::Poisson.
Definition: RooPoisson.cxx:156
Int_t getGenerator(const RooArgSet &directVars, RooArgSet &generateVars, Bool_t staticInitOK=kTRUE) const override
Advertise internal generator in x.
Definition: RooPoisson.cxx:147
Double_t analyticalIntegral(Int_t code, const char *rangeName=0) const override
Implements the actual analytical integral(s) advertised by getAnalyticalIntegral.
Definition: RooPoisson.cxx:85
virtual TObject * clone(const char *newname) const override
Definition: RooPoisson.h:24
Double_t evaluate() const override
Implementation in terms of the TMath::Poisson() function.
Definition: RooPoisson.cxx:54
void computeBatch(cudaStream_t *, double *output, size_t nEvents, RooBatchCompute::DataMap &) const override
Compute multiple values of the Poisson distribution.
Definition: RooPoisson.cxx:67
Bool_t _noRounding
Definition: RooPoisson.h:48
RooRealProxy mean
Definition: RooPoisson.h:47
void setNoRounding(bool flag=kTRUE)
Switch off/on rounding of x to the nearest integer.
Definition: RooPoisson.h:34
RooAbsReal const & getMean() const
Get the mean parameter.
Definition: RooPoisson.h:42
const T & arg() const
Return reference to object held in proxy.
Mother of all ROOT objects.
Definition: TObject.h:37
std::map< DataKey, RooSpan< const double > > DataMap
static void output(int code)
Definition: gifencode.c:226