2 #ifndef ALICFUNFOLDING_H
3 #define ALICFUNFOLDING_H
5 //--------------------------------------------------------------------//
7 // AliCFUnfolding Class //
8 // Class to handle general unfolding procedure //
9 // For the moment only bayesian unfolding is supported //
10 // The next steps are to add chi2 minimisation and weighting methods //
12 // Author : renaud.vernet@cern.ch //
13 //--------------------------------------------------------------------//
16 #include "THnSparse.h"
20 class AliCFUnfolding : public TNamed {
24 AliCFUnfolding(const Char_t* name, const Char_t* title, const Int_t nVar,
25 const THnSparse* response, const THnSparse* efficiency, const THnSparse* measured, const THnSparse* prior=0x0);
26 AliCFUnfolding(const AliCFUnfolding& c);
27 AliCFUnfolding& operator= (const AliCFUnfolding& c);
30 void SetMaxNumberOfIterations(Int_t n) {fMaxNumIterations=n;}
31 void SetMaxChi2(Double_t val) {fMaxChi2=val;}
32 void SetMaxChi2PerDOF(Double_t val);
33 void UseSmoothing(TF1* fcn=0x0, Option_t* opt="iremn") { // if fcn=0x0 then smooth using neighbouring bins
34 fUseSmoothing=kTRUE; // this function must NOT be used if fNVariables > 3
35 fSmoothFunction=fcn; // the option "opt" is used if "fcn" is specified
41 THnSparse* GetResponse() const {return fResponse;}
42 THnSparse* GetInverseResponse() const {return fInverseResponse;}
43 THnSparse* GetPrior() const {return fPrior;}
44 THnSparse* GetOriginalPrior() const {return fOriginalPrior;}
45 THnSparse* GetEfficiency() const {return fEfficiency;}
46 THnSparse* GetUnfolded() const {return fUnfolded;}
47 THnSparse* GetEstMeasured() const {return fMeasuredEstimate;}
48 THnSparse* GetMeasured() const {return fMeasured;}
49 THnSparse* GetConditional() const {return fConditional;}
50 TF1* GetSmoothFunction() const {return fSmoothFunction;}
54 // user-related settings
55 THnSparse *fResponse; // Response matrix : dimensions must be 2N = 2 x (number of variables)
56 // dimensions 0 -> N-1 must be filled with reconstructed values
57 // dimensions N -> 2N-1 must be filled with generated values
58 THnSparse *fPrior; // This is the assumed generated distribution : dimensions must be N = number of variables
59 // it will be used at the first step
60 // then will be updated automatically at each iteration
61 THnSparse *fEfficiency; // Efficiency map : dimensions must be N = number of variables
62 // this map must be filled only with "true" values of the variables (should not include resolution effects)
63 THnSparse *fMeasured; // Measured spectrum to be unfolded : dimensions must be N = number of variables
64 Int_t fMaxNumIterations; // Maximum number of iterations to be performed
65 Int_t fNVariables; // Number of variables used in analysis spectra (pt, y, ...)
66 Double_t fMaxChi2; // Maximum Chi2 between unfolded and prior distributions.
67 Bool_t fUseSmoothing; // Smooth the unfolded sectrum at each iteration
68 TF1 *fSmoothFunction; // Function used to smooth the unfolded spectrum
69 Option_t *fSmoothOption; // Option to use during the fit (with fSmoothFunction) ; default is "iremn"
72 THnSparse *fOriginalPrior; // This is the original prior distribution : will not be modified
73 THnSparse *fInverseResponse; // Inverse response matrix
74 THnSparse *fMeasuredEstimate; // Estimation of the measured (M) spectrum given the a priori (T) distribution
75 THnSparse *fConditional; // Matrix holding the conditional probabilities P(M|T)
76 THnSparse *fProjResponseInT; // Projection of the response matrix on TRUE axis
77 THnSparse *fUnfolded; // Unfolded spectrum
78 Int_t *fCoordinates2N; // Coordinates in 2N (measured,true) space
79 Int_t *fCoordinatesN_M; // Coordinates in measured space
80 Int_t *fCoordinatesN_T; // Coordinates in true space
84 void Init(); // initialisation of the internal settings
85 void GetCoordinates(); // gets a cell coordinates in Measured and True space
86 void CreateConditional(); // creates the conditional matrix from the response matrix
87 void CreateEstMeasured(); // creates the measured spectrum estimation from the conditional matrix and the prior distribution
88 void CreateInvResponse(); // creates the inverse response function (Bayes Theorem) from the conditional matrix and the prior distribution
89 void CreateUnfolded(); // creates the unfolded spectrum from the inverse response matrix and the measured distribution
90 void CreateFlatPrior(); // creates a flat a priori distribution in case the one given in the constructor is null
91 Double_t GetChi2(); // returns the chi2 between unfolded and prior spectra
92 Short_t Smooth(); // function calling smoothing methods
93 Short_t SmoothUsingNeighbours(); // smoothes the unfolded spectrum using the neighbouring cells
94 Short_t SmoothUsingFunction(); // smoothes the unfolded spectrum using a fit function
96 ClassDef(AliCFUnfolding,0);