Updates + addition of EMCal
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEspectrum.h
1         /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 *                                                                        *
4 * Author: The ALICE Off-line Project.                                    *
5 * Contributors are mentioned in the code where appropriate.              *
6 *                                                                        *
7 * Permission to use, copy, modify and distribute this software and its   *
8 * documentation strictly for non-commercial purposes is hereby granted   *
9 * without fee, provided that the above copyright notice appears in all   *
10 * copies and that both the copyright notice and this permission notice   *
11 * appear in the supporting documentation. The authors make no claims     *
12 * about the suitability of this software for any purpose. It is          *
13 * provided "as is" without express or implied warranty.                  *
14 **************************************************************************/
15 //
16 // Class for spectrum correction
17 // Subtraction of hadronic background, Unfolding of the data and
18 // Renormalization done here
19 // For more information see the implementation file
20 //
21 #ifndef ALIHFESPECTRUM_H
22 #define ALIHFESPECTRUM_H
23
24 #ifndef ROOT_TNamed
25 #include <TNamed.h>
26 #endif
27
28 class TGraphErrors;
29 class TObject;
30 class TH1;
31 class TF1;
32 class TList;
33 class AliCFContainer;
34 class AliHFEcontainer;
35 class AliCFDataGrid;
36 class AliCFEffGrid;
37
38 class AliHFEspectrum : public TNamed{
39   public:
40     enum CFContainer_t{
41       kDataContainer  = 0,
42       kBackgroundData = 1,
43       kMCContainerMC = 2,
44       kMCContainerESD = 3,
45       kDataContainerV0 = 4,
46       kNonHFEBackgroundData = 5, 
47       kMCContainerCharmMC = 6,
48       kMCContainerConversionMC = 7,
49       kMCContainerNonHFEMC = 8
50     };
51    
52     AliHFEspectrum(const char* name);
53     ~AliHFEspectrum();
54     
55
56     Bool_t Init(const AliHFEcontainer *datahfecontainer, const AliHFEcontainer *mchfecontainer, const AliHFEcontainer *v0hfecontainer=0x0, const AliHFEcontainer *bghfecontainer=0x0);
57     Bool_t Correct(Bool_t subtractcontamination=kTRUE);
58     Bool_t CorrectBeauty(Bool_t subtractcontamination=kTRUE);
59    
60     AliCFDataGrid *SubtractBackground(Bool_t setBackground = kFALSE);
61     
62     AliCFDataGrid *CorrectV0Efficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
63     AliCFDataGrid *CorrectParametrizedEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
64    
65     TList *Unfold(AliCFDataGrid* const bgsubpectrum = 0x0);
66     AliCFDataGrid *CorrectForEfficiency(AliCFDataGrid* const bgsubpectrum = 0x0);
67    
68     TGraphErrors *Normalize(THnSparse * const spectrum,Int_t i = 0) const;
69     TGraphErrors *Normalize(AliCFDataGrid * const spectrum,Int_t i = 0) const;
70     TGraphErrors *NormalizeTH1N(TH1 *input,Int_t normalization) const;
71     void CorrectFromTheWidth(TH1D *h1) const;
72     
73     void SetCorrelation(THnSparseF * const correlation) {fCorrelation = correlation; };
74     void SetContainer(AliCFContainer *cont, AliHFEspectrum::CFContainer_t type);
75     void SetEfficiencyFunction(TF1 *efficiencyFunction) { fEfficiencyFunction = efficiencyFunction; };
76     void SetPbPbAnalysis(Bool_t isPbPb = kFALSE) { fBeamType=(Char_t) isPbPb; };
77     
78     void SetNumberOfEvents(Int_t nEvents,Int_t i = 0) { fNEvents[i] = nEvents; };
79     void SetNumberOfMCEvents(Int_t nEvents) { fNMCEvents = nEvents; };
80     void SetNumberOfMC2Events(Int_t nEvents) { fNMCbgEvents = nEvents; }; 
81     void SetMCEffStep(Int_t step) { fStepMC = step; };
82     void SetMCTruthStep(Int_t step) { fStepTrue = step; };
83     void SetStepToCorrect(Int_t step) { fStepData = step; };
84     void SetStepBeforeCutsV0(Int_t step) { fStepBeforeCutsV0 = step; };
85     void SetStepAfterCutsV0(Int_t step) { fStepAfterCutsV0 = step; };
86     void SetNbDimensions(Int_t nbDimensions) { fNbDimensions = nbDimensions; };
87     void SetChargeChoosen(Int_t chargechoosen) {fChargeChoosen = chargechoosen; };
88     void SetUnSetCorrelatedErrors(Bool_t unsetcorrelatederrors) {fUnSetCorrelatedErrors = unsetcorrelatederrors;};
89
90     void SetNCentralityBinAtTheEnd(Int_t nCentralityBinAtTheEnd) {fNCentralityBinAtTheEnd = nCentralityBinAtTheEnd; };
91     void SetLowHighBoundaryCentralityBinAtTheEnd(Int_t low, Int_t high, Int_t i) { fLowBoundaryCentralityBinAtTheEnd[i] = low; fHighBoundaryCentralityBinAtTheEnd[i] = high;};
92
93     void SetBeautyAnalysis() { fInclusiveSpectrum = kFALSE; };
94     void SetHadronEffbyIPcut(THnSparseF* hsHadronEffbyIPcut) { fHadronEffbyIPcut = hsHadronEffbyIPcut;};
95     void SetNonHFEBackground2ndMethod() { fNonHFEbgMethod2 = kTRUE; };
96
97     void SetStepGuessedUnfolding(Int_t stepGuessedUnfolding) { fStepGuessedUnfolding = stepGuessedUnfolding; };
98     void SetNumberOfIteration(Int_t numberOfIteration) { fNumberOfIterations = numberOfIteration; };
99     
100     void SetDumpToFile(Bool_t dumpToFile) { fDumpToFile=dumpToFile; }; 
101   
102     void SetDebugLevel(Int_t debugLevel) { fDebugLevel = debugLevel; };
103
104     AliCFDataGrid* GetCharmBackground();
105     AliCFDataGrid* GetConversionBackground();
106     AliCFDataGrid* GetNonHFEBackground();
107     THnSparse* GetWeights();
108
109     void EnableIPanaHadronBgSubtract() { fIPanaHadronBgSubtract = kTRUE; };
110     void EnableIPanaCharmBgSubtract() { fIPanaCharmBgSubtract = kTRUE; };
111     void EnableIPanaConversionBgSubtract() { fIPanaConversionBgSubtract = kTRUE; };
112     void EnableIPanaNonHFEBgSubtract() { fIPanaNonHFEBgSubtract = kTRUE; };
113
114   protected:
115        
116     AliCFContainer *GetContainer(AliHFEspectrum::CFContainer_t contt);
117     AliCFContainer *GetSlicedContainer(AliCFContainer *cont, Int_t ndim, Int_t *dimensions,Int_t source=-1,Int_t positivenegative=-1);
118     THnSparseF *GetSlicedCorrelation(THnSparseF *correlationmatrix,Int_t nDim, Int_t *dimensions) const;
119     TObject* GetSpectrum(const AliCFContainer * const c, Int_t step);
120     TObject* GetEfficiency(const AliCFContainer * const c, Int_t step, Int_t step0);
121  
122     void AddTemporaryObject(TObject *cont);
123     void ClearObject(TObject *o);
124     
125     TGraphErrors *NormalizeTH1(TH1 *input,Int_t i = 0) const;
126
127
128   private:
129     AliHFEspectrum(const AliHFEspectrum &);
130     AliHFEspectrum &operator=(const AliHFEspectrum &);
131  
132     TList *fCFContainers;         // List of Correction Framework Containers
133     TList *fTemporaryObjects;     // Emulate garbage collection
134     THnSparseF *fCorrelation;     // Correlation Matrices
135     AliCFDataGrid *fBackground;   // Background Grid
136     TF1 *fEfficiencyFunction;     // Efficiency Function
137
138     THnSparseF *fWeightCharm;     // Weight for charm bg
139
140     Bool_t fInclusiveSpectrum;     // Inclusive Spectrum
141     Bool_t fDumpToFile;           // Write Result in a file
142
143     Bool_t fUnSetCorrelatedErrors;    // Unset correlated errors
144
145     Bool_t fIPanaHadronBgSubtract;     // Hadron background subtraction
146     Bool_t fIPanaCharmBgSubtract;      // Charm background subtraction 
147     Bool_t fIPanaConversionBgSubtract; // Conversion background subtraction
148     Bool_t fIPanaNonHFEBgSubtract;     // nonHFE except for conversion background subtraction
149     Bool_t fNonHFEbgMethod2;           // switch for 2nd method to subtract non HFE background
150
151     Int_t fNbDimensions;          // Number of dimensions for the correction
152     Int_t fNEvents[20];           // Number of Events
153     Int_t fNMCEvents;             // Number of MC Events
154     Int_t fNMCbgEvents;           // Number of BG MC Events
155     Int_t fStepMC;                // MC step (for unfolding)
156     Int_t fStepTrue;              // MC step of the final spectrum
157     Int_t fStepData;              // Data Step (various applications)
158     Int_t fStepBeforeCutsV0;      // Before cuts V0
159     Int_t fStepAfterCutsV0;       // After cuts V0
160     Int_t fStepGuessedUnfolding;  // Step for first guessed unfolding
161     Int_t fNumberOfIterations;    // Number of iterations
162     Int_t fChargeChoosen;         // Select positive or negative electrons
163
164     Int_t fNCentralityBinAtTheEnd; // Number of centrality class at the end
165     Int_t fLowBoundaryCentralityBinAtTheEnd[20];  // Boundary of the bins
166     Int_t fHighBoundaryCentralityBinAtTheEnd[20];  // Boundary of the bins
167
168     THnSparseF *fHadronEffbyIPcut;// container for hadron efficiency by IP cut
169     Char_t fBeamType;              // beamtype; default -1; pp =0; PbPb=1
170
171
172     Int_t fDebugLevel;            // Debug Level
173
174     ClassDef(AliHFEspectrum, 1) 
175 };
176 #endif
177