Since it contains fixes of coding rule violations, all classes are involved. Further...
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpriVtx.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 /**************************************************************************
17  *                                                                        *
18  * QA class of primary vertex study for Heavy Flavor electrons            * 
19  *                                                                        *
20  **************************************************************************/
21
22 #ifndef ALIHFEPRIVTX_H
23 #define ALIHFEPRIVTX_H
24
25 #ifndef ROOT_TObject
26 //#include <TObject.h>
27 #endif
28
29 class TH1F;
30 class TH1I;
31 class TH2F;
32 class TString;
33 class AliESDEvent;
34 class AliESDtrack;
35 class AliStack;
36
37 //________________________________________________________________
38 class AliHFEpriVtx : public TObject {
39
40         public: 
41                 AliHFEpriVtx();
42                 AliHFEpriVtx(const AliHFEpriVtx &p); // copy constructor
43                 AliHFEpriVtx &operator=(const AliHFEpriVtx &); // assignment operator
44                 virtual ~AliHFEpriVtx();
45
46                 void CreateHistograms(TString hnopt=""); // create histograms
47                 void Init();
48                 void SetEvent(AliESDEvent* ESD){fESD1=ESD;}; // set ESD pointer
49                 void SetStack(AliStack* stack){fStack=stack;} // set stack pointer
50                 void CountNtracks(Int_t sourcePart, Int_t recpid, Double_t recprob); // count number of tracks passed certain cut
51                 void FillNtracks(); // fill counted number of tracks
52                 void CountPriVxtElecContributor(AliESDtrack *ESDelectron, Int_t sourcePart, Int_t recpid, Double_t recprob); 
53                 void GetNPriVxtContributor();
54                 void FillNprimVtxContributor() const;
55
56                 Int_t GetMCPID(AliESDtrack *track); // return mc pid
57
58         private:
59
60                 AliESDEvent* fESD1; // ESD event 
61                 AliStack* fStack; // MC Stack
62
63                 TString fkSourceLabel[10]; // storing source label
64
65                 enum kSources {kAll, kDirectCharm, kDirectBeauty, kBeautyCharm, kGamma, kPi0, kElse, kBeautyGamma, kBeautyPi0, kBeautyElse};
66
67                 struct AliHists{
68                         TH1F *fNtracks; // histogram to fill number of counted tracks for different sources
69                         TH1F *fNprimVtxContributor; // histogram to fill number of tracks contributing primary vertex 
70                         TH1F *fPtElec; // histogram to fill pt of electron tracks
71                         TH1F *fPtElecContributor; // histogram to fill pt of electron tracks contributing primary vertex
72                         Int_t fNtrackCount; // number of counted track
73                         Int_t fNprimVtxContributorCount; // number of tracks contributing primary vertex
74
75                         AliHists()
76                         : fNtracks()
77                         , fNprimVtxContributor()
78                         , fPtElec()
79                         , fPtElecContributor()
80                         , fNtrackCount(0)
81                         , fNprimVtxContributorCount(0)
82                         {
83                           // default constructor
84                         }
85
86                         AliHists(const AliHists & p)
87                         : fNtracks(p.fNtracks)
88                         , fNprimVtxContributor(p.fNprimVtxContributor)
89                         , fPtElec(p.fPtElec)
90                         , fPtElecContributor(p.fPtElecContributor)
91                         , fNtrackCount(p.fNtrackCount)
92                         , fNprimVtxContributorCount(p.fNprimVtxContributorCount)
93                         {
94                           // copy constructor
95                         }
96                         AliHists &operator=(const AliHists &)
97                         {
98                           // assignment operator, not yet implemented
99                           return *this;
100                         }
101                 };
102                 AliHists fPrimVtx[10]; // define structure of histograms
103
104                 Int_t fNtrackswoPid; //  number of track counted
105                 TH1F *fHNtrackswoPid; // histogram to fill number of track counted
106                 TH1I *fNESDprimVtxContributor; // histogram to fill number of primary vertex contributor for given event 
107                 TH1I *fNESDprimVtxIndices; // histogram to fill number of primary vertex indices for given event
108                 TH2F *fDiffDCAvsPt; // histogram to fill DCA difference as a function of pT
109                 TH2F *fDiffDCAvsNt; // histogram to fill DCA difference as a function of pT
110
111
112         ClassDef(AliHFEpriVtx,0);
113 };
114
115 #endif