c5b807e46bbc026585e482f7e421aac90bf4fe2b
[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AliSpectraAODHistoManager.h
1
2 #ifndef ALISPECTRAAODHISTOMANAGER_H
3 #define ALISPECTRAAODHISTOMANAGER_H
4
5 /*  See cxx source for full Copyright notice */
6
7 //-------------------------------------------------------------------------
8 //                      AliSpectraAODHistoManager
9 //
10 //
11 //
12 //
13 // Authors: Michele Floris, CERN, Philip Versteeg, UU, Redmer Bertens, UU
14 //-------------------------------------------------------------------------
15
16 class AliAODEvent;
17 class TH1F;
18 class TH2F;
19 class TList;
20 #include "TNamed.h"
21 #include "Histograms.h" // Change this file if you want to add an histogram
22 #include "HistogramNames.h"
23
24 namespace AliSpectraNameSpace
25 {
26
27    
28    enum AODParticleSpecies_t
29    {
30      kProton = 0,
31      kKaon,
32      kPion,
33      kNSpecies = kPion,
34    }; // Particle species used in plotting
35
36    const char * kParticleSpecies[] =
37    {
38        "ProtonPlus",
39        "ProtonMinus",
40        "KaonPlus",
41        "KaonMinus",
42        "PionPlus",
43        "PionMinus",
44    };
45    enum AODHistoType_t
46    {
47        kHistSpectraRec = 0,
48        kHistSpectraGen,
49        kHNHistoTypes = kHistSpectraGen,
50    }; // Types of histos
51
52 }
53
54 using namespace AliSpectraNameSpace;
55
56 class AliSpectraAODHistoManager : public TNamed
57 {
58 public:
59    AliSpectraAODHistoManager() :  TNamed(), fOutputList(0) {}
60    AliSpectraAODHistoManager(const char *name);
61    virtual  ~AliSpectraAODHistoManager() {}
62
63
64    TH2F*   BookPtGenHistogram(const char * name);
65    TH2F*   BookPtRecHistogram(const char * name);
66    TH2F*   BookPIDHistogram(const char * name);
67    TH2F*   BookNSigHistogram(const char * name);
68    TH2F*   BookqVecHistogram(const char * name);
69    TH1F*   GetPtHistogram1D(const char * name,Double_t minDCA,Double_t maxDCA);
70    TH1F*   GetDCAHistogram1D(const char * name,Double_t minPt,Double_t maxPt);
71    TH2*     GetHistogram(UInt_t id)      {      return (TH2*) fOutputList->At(id);   }
72    TH2*     GetPtHistogram(UInt_t id)    {      return (TH2*) fOutputList->At(id);   }
73    TH2*     GetPtHistogram(const char * name)   {      return (TH2*) fOutputList->FindObject(name);   }
74    TH2*     GetPtHistogramByName(UInt_t id)     {      return (TH2*) fOutputList->FindObject(kHistName[id]); }  // Use this if you want to read a file saved with a different histo list   
75    TH2*     GetPIDHistogram(UInt_t id)   {      return (TH2*) fOutputList->At(id);   }
76    TH2*     GetPIDHistogram(const char * name)  {      return (TH2*) fOutputList->FindObject(name);   }
77    TH2*     GetPIDHistogramByName(UInt_t id)    {      return (TH2*) fOutputList->FindObject(kHistName[id]);  }// Use this if you want to read a file saved with a different histo list
78    TH2*     GetNSigHistogram(UInt_t id)   {      return (TH2*) fOutputList->At(id);   }
79    TH2*     GetNSigHistogram(const char * name)  {      return (TH2*) fOutputList->FindObject(name);   }
80    TH2*     GetNSigHistogramByName(UInt_t id)    {      return (TH2*) fOutputList->FindObject(kHistName[id]);  }// Use this if you want to read a file saved with a different histo list
81    TH2*     GetqVecHistogram(UInt_t id)   {      return (TH2*) fOutputList->At(id);   }
82    TH2*     GetqVecHistogram(const char * name)  {      return (TH2*) fOutputList->FindObject(name);   }
83    TH2*     GetqVecHistogramByName(UInt_t id)    {      return (TH2*) fOutputList->FindObject(kHistName[id]);  }// Use this if you want to read a file saved with a different histo list
84   
85    //TH1F*   GetTH1F(UInt_t id)            {      return (TH1F*) GetPtHistogram(id);   }
86    //TH2F*   GetTH2F(UInt_t id)            {      return (TH2F*) GetPIDHistogram(id);   }
87
88   TList * GetOutputList() {return fOutputList;}
89
90   Long64_t Merge(TCollection* list);
91
92
93 private:
94    TList     *fOutputList;  // List of Pt Histo's
95
96    AliSpectraAODHistoManager(const AliSpectraAODHistoManager&);
97    AliSpectraAODHistoManager& operator=(const AliSpectraAODHistoManager&);
98
99    ClassDef(AliSpectraAODHistoManager, 1);
100
101 };
102 #endif
103