added some control histos, adapted the eta range and removed dependcies on generated...
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliJetSpectrumUnfolding.h
CommitLineData
6d75bdb8 1#ifndef ALIJETSPECTRUMUNFOLDING_H
2#define ALIJETSPECTRUMUNFOLDING_H
6d75bdb8 3//
4// class that contains the correction matrix and the functions for
5// correction the jet spectrum
6// implements 2-dim bayesian method
7//
8
8de9091a 9
10
11
6d75bdb8 12class TH1;
13class TH2;
14class TH3;
15class TH1F;
6d75bdb8 16class TH3F;
17class TF1;
18class TF2;
19class TCollection;
20
8de9091a 21#include "TNamed.h"
22#include <TH2F.h>
6d75bdb8 23#include <THnSparse.h>
24
25class AliJetSpectrumUnfolding : public TNamed {
26 public:
27 AliJetSpectrumUnfolding();
28 AliJetSpectrumUnfolding(const Char_t* name, const Char_t* title);
29 virtual ~AliJetSpectrumUnfolding();
30
31 virtual Long64_t Merge(TCollection* list);
32
33 Bool_t LoadHistograms(const Char_t* dir = 0);
34 void SaveHistograms();
35 void DrawHistograms();
8de9091a 36 void DrawComparison(const char* name, TH2* const genSpec);
6d75bdb8 37
38 void SetBayesianParameters(Float_t smoothing, Int_t nIterations);
39 void ApplyBayesianMethod(Float_t regPar = 1, Int_t nIterations = 100, TH2* initialConditions = 0, Bool_t determineError = kTRUE);
40
8de9091a 41 TH2F* GetRecSpectrum() const { return fRecSpectrum; }
42 TH2F* GetGenSpectrum() const { return fGenSpectrum; }
43 TH2F* GetUnfSpectrum() const { return fUnfSpectrum; }
44 THnSparseF* GetCorrelation() const { return fCorrelation; }
6d75bdb8 45
8de9091a 46 void SetRecSpectrum(TH2F* const hist) { if(fRecSpectrum) delete fRecSpectrum;
47 fRecSpectrum = hist; }
48 void SetGenSpectrum(TH2F* const hist) { if(fGenSpectrum)delete fGenSpectrum;
49 fGenSpectrum = hist; }
50 void SetUnfSpectrum(TH2F* const hist) { if(fUnfSpectrum)delete fUnfSpectrum;
51 fUnfSpectrum = hist; }
52 void SetCorrelation(THnSparseF* const hist){ if(fCorrelation)delete fCorrelation;
53 fCorrelation = hist; }
6d75bdb8 54
55 void SetGenRecFromFunc(TF2* inputGen);
56 TH2F* CalculateRecSpectrum(TH2* inputGen);
57
8de9091a 58 static void NormalizeToBinWidth(TH2* const hist);
6d75bdb8 59
8de9091a 60 void GetComparisonResults(Float_t* const gen = 0, Int_t* const genLimit = 0, Float_t* const residuals = 0, Float_t* const ratioAverage = 0) const;
6d75bdb8 61
62 protected:
63 void SetupCurrentHists(Bool_t createBigBin);
64
8de9091a 65 static Double_t BayesCov(THnSparseF* const M, THnSparseF* const correlation, Int_t* const binTM, Int_t* const binTM1);
66 static Double_t BayesUncertaintyTerms(THnSparseF* const M, THnSparseF *const C, Int_t* const binTM, Int_t* const binTM1, Double_t nt);
67 static Int_t UnfoldWithBayesian(THnSparseF* const correlation, TH2* const measured, TH2* const initialConditions, TH2* const aResult, Float_t regPar, Int_t nIterations, Bool_t calculateErrors = kFALSE);
6d75bdb8 68
69 static Float_t fgBayesianSmoothing; //! smoothing parameter (0 = no smoothing)
70 static Int_t fgBayesianIterations; //! number of iterations in Bayesian method
71
72 TH2F* fCurrentRec;
73 THnSparseF* fCurrentCorrelation;
74
75 TH2F* fRecSpectrum;
76 TH2F* fGenSpectrum;
77 TH2F* fUnfSpectrum;
78 THnSparseF* fCorrelation;
79
80 Float_t fLastChi2MC; //! last Chi2 between MC and unfolded ESD (calculated in DrawComparison)
81 Int_t fLastChi2MCLimit; //! bin where the last chi2 breached a certain threshold
82 Float_t fLastChi2Residuals; //! last Chi2 of the ESD and the folded unfolded ESD (calculated in DrawComparison)
83 Float_t fRatioAverage; //! last average of |ratio-1| where ratio = unfolded / mc (bin 2..150)
84
85 private:
8de9091a 86
87 static const Int_t fgkNBINSE;
88 static const Int_t fgkNBINSZ;
89 static const Int_t fgkNEVENTS;
90 static const Double_t fgkaxisLowerLimitE;
91 static const Double_t fgkaxisLowerLimitZ;
92 static const Double_t fgkaxisUpperLimitE;
93 static const Double_t fgkaxisUpperLimitZ;
94
95
96
6d75bdb8 97 AliJetSpectrumUnfolding(const AliJetSpectrumUnfolding&);
98 AliJetSpectrumUnfolding& operator=(const AliJetSpectrumUnfolding&);
99
100 ClassDef(AliJetSpectrumUnfolding, 2);
101};
102
103#endif
104