]>
Commit | Line | Data |
---|---|---|
76ce4b5b | 1 | //////////////////////////////////////////////////////////////////////////////// |
2 | // // | |
3 | // AliFemtoEventReaderAOD - the reader class for the Alice AOD // | |
4 | // Reads in AOD information and converts it into internal AliFemtoEvent // | |
5 | // Authors: Adam Kisiel kisiel@mps.ohio-state.edu // | |
6 | // // | |
7 | //////////////////////////////////////////////////////////////////////////////// | |
8 | ||
9 | #ifndef ALIFEMTOEVENTREADERAOD_H | |
10 | #define ALIFEMTOEVENTREADERAOD_H | |
11 | #include "AliFemtoEventReader.h" | |
12 | #include "AliFemtoEnumeration.h" | |
13 | ||
14 | #include <string> | |
15 | #include <vector> | |
16 | #include "TTree.h" | |
17 | #include "TChain.h" | |
18 | #include "TBits.h" | |
19 | #include "AliAODEvent.h" | |
20 | #include <list> | |
21 | //#include "AliPWG2AODTrack.h" | |
22 | #include "AliAODMCParticle.h" | |
973a91f8 | 23 | #include "AliFemtoV0.h" |
76ce4b5b | 24 | #include "AliAODpidUtil.h" |
1777446b | 25 | #include "AliAODHeader.h" |
76ce4b5b | 26 | |
27 | class AliFemtoEvent; | |
28 | class AliFemtoTrack; | |
29 | ||
30 | class AliFemtoEventReaderAOD : public AliFemtoEventReader | |
31 | { | |
32 | public: | |
1777446b | 33 | enum EventMult {kCentrality=0, kGlobalCount=1, kReference=2, kTPCOnlyRef=3}; |
34 | typedef enum EventMult EstEventMult; | |
35 | ||
76ce4b5b | 36 | AliFemtoEventReaderAOD(); |
37 | AliFemtoEventReaderAOD(const AliFemtoEventReaderAOD &aReader); | |
38 | virtual ~AliFemtoEventReaderAOD(); | |
39 | ||
40 | AliFemtoEventReaderAOD& operator=(const AliFemtoEventReaderAOD& aReader); | |
41 | ||
42 | virtual AliFemtoEvent* ReturnHbtEvent(); | |
43 | AliFemtoString Report(); | |
ce7b3d98 | 44 | void SetInputFile(const char* inputfile); |
76ce4b5b | 45 | void SetFilterBit(UInt_t ibit); |
46 | void SetReadMC(unsigned char a); | |
973a91f8 | 47 | void SetReadV0(unsigned char a); |
76ce4b5b | 48 | void SetCentralityPreSelection(double min, double max); |
973a91f8 | 49 | void SetNoCentrality(bool anocent); |
76ce4b5b | 50 | void SetAODpidUtil(AliAODpidUtil *aAODpidUtil); |
1777446b | 51 | void SetAODheader(AliAODHeader *aAODheader); |
ba3c23a4 | 52 | void SetMagneticFieldSign(int s); |
ce7b3d98 | 53 | void GetGlobalPositionAtGlobalRadiiThroughTPC(AliAODTrack *track, Float_t bfield, Float_t globalPositionsAtRadii[9][3]); |
1777446b | 54 | void SetUseMultiplicity(EstEventMult aType); |
76ce4b5b | 55 | |
56 | protected: | |
57 | virtual void CopyAODtoFemtoEvent(AliFemtoEvent *tEvent); | |
58 | virtual void CopyAODtoFemtoTrack( AliAODTrack *tAodTrack, | |
59 | AliFemtoTrack *tFemtoTrack | |
60 | // AliPWG2AODTrack *tPWG2AODTrack | |
61 | ); | |
973a91f8 | 62 | virtual void CopyAODtoFemtoV0(AliAODv0 *tAODv0, AliFemtoV0 *tFemtoV0); |
76ce4b5b | 63 | virtual void CopyPIDtoFemtoTrack( AliAODTrack *tAodTrack, |
64 | AliFemtoTrack *tFemtoTrack); | |
65 | ||
66 | int fNumberofEvent; // number of Events in AOD file | |
67 | int fCurEvent; // number of current event | |
68 | AliAODEvent* fEvent; // AOD event | |
69 | TBits fAllTrue; // Bit set with all true bits | |
70 | TBits fAllFalse; // Bit set with all false bits | |
71 | UInt_t fFilterBit; // Bitmap bit for AOD filters | |
72 | // TClonesArray* fPWG2AODTracks; // Link to PWG2 specific AOD information (if it exists) | |
73 | ||
74 | unsigned char fReadMC; // Attempt to read the MC information from the AOD | |
973a91f8 | 75 | unsigned char fReadV0; // Read V0 information from the AOD and put it into V0Collection |
76ce4b5b | 76 | unsigned char fUsePreCent; // Use centrality pre-selection to speed up analysis |
1777446b | 77 | EstEventMult fEstEventMult; // Type of the event multiplicity estimator |
76ce4b5b | 78 | double fCentRange[2]; // Centrality pre-selection range |
79 | AliAODpidUtil* fAODpidUtil; | |
1777446b | 80 | AliAODHeader* fAODheader; |
76ce4b5b | 81 | |
82 | private: | |
83 | ||
84 | AliAODMCParticle* GetParticleWithLabel(TClonesArray *mcP, Int_t aLabel); | |
85 | ||
86 | string fInputFile; // name of input file with AOD filenames | |
87 | string fFileName; // name of current AOD file | |
88 | TChain* fTree; // AOD tree | |
89 | TFile* fAodFile; // AOD file | |
ba3c23a4 | 90 | int fMagFieldSign; // Magnetic field sign |
76ce4b5b | 91 | |
1777446b | 92 | |
76ce4b5b | 93 | #ifdef __ROOT__ |
94 | ClassDef(AliFemtoEventReaderAOD, 11) | |
95 | #endif | |
96 | ||
97 | }; | |
98 | ||
99 | #endif | |
100 | ||
101 |