Since it contains fixes of coding rule violations, all classes are involved. Further...
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpriVtx.h
CommitLineData
259c3296 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
75d81601 22#ifndef ALIHFEPRIVTX_H
23#define ALIHFEPRIVTX_H
259c3296 24
25#ifndef ROOT_TObject
75d81601 26//#include <TObject.h>
259c3296 27#endif
28
29class TH1F;
30class TH1I;
31class TH2F;
32class TString;
33class AliESDEvent;
34class AliESDtrack;
35class AliStack;
36
37//________________________________________________________________
38class 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
259c3296 46 void CreateHistograms(TString hnopt=""); // create histograms
259c3296 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();
75d81601 54 void FillNprimVtxContributor() const;
259c3296 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
75d81601 67 struct AliHists{
259c3296 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
259c3296 74
75d81601 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
259c3296 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
dbe3abbe 112 ClassDef(AliHFEpriVtx,0);
259c3296 113};
114
115#endif