]>
Commit | Line | Data |
---|---|---|
a5556ea5 | 1 | #ifndef ALIAODRUN_H |
2 | #define ALIAODRUN_H | |
3 | //____________________ | |
4 | /////////////////////////////////////////////////////// | |
5 | // // | |
6 | // AliAODRun // | |
7 | // // | |
8 | // Class storing and managing set events // | |
9 | // designed for fast acces // | |
10 | // // | |
11 | // Piotr.Skowronski@cern.ch // | |
c7ffd78f | 12 | // http://aliweb.cern.ch/people/skowron/analyzer // |
a5556ea5 | 13 | // // |
14 | /////////////////////////////////////////////////////// | |
15 | ||
16 | ||
17 | #include "AliAOD.h" | |
18 | #include <TObjArray.h> | |
19 | ||
afa8b37b | 20 | class AliVAODParticle; |
a5556ea5 | 21 | class TParticle; |
22 | ||
23 | class AliAODRun: public TObject | |
24 | { | |
25 | public: | |
26 | AliAODRun(); | |
27 | virtual ~AliAODRun(); | |
28 | ||
afa8b37b | 29 | void AddParticle(Int_t event, AliVAODParticle* part); //inerface to AliAOD::AddParticle(AliVAODParticle*) |
a5556ea5 | 30 | void AddParticle(Int_t event, TParticle* part, Int_t idx);//inerface to AliAOD::AddParticle(TParticle*) |
31 | ||
32 | //inerface to AliAOD::AddParticle(Int_t.Double_t,Double_t,Double_t,Double_t,Double_t,Double_t,Double_t,Double_t,Double_t) | |
33 | void AddParticle(Int_t event, Int_t pdg, Int_t idx, | |
34 | Double_t px, Double_t py, Double_t pz, Double_t etot, | |
35 | Double_t vx, Double_t vy, Double_t vz, Double_t time); | |
36 | ||
37 | void SetEvent(Int_t number, AliAOD* event); | |
afa8b37b | 38 | AliVAODParticle* GetParticle(Int_t event, Int_t n); //returns nth particle from event |
a5556ea5 | 39 | AliAOD* GetEvent(Int_t event) const; //returns AliAOD number "event" |
40 | ||
41 | Int_t GetNumberOfEvents() const; //returns number of events | |
42 | Int_t GetNumberOfParticlesInEvent(Int_t event) const; //returns number of particles in event number "event" | |
43 | void Reset();//clears all events in the array (deletes) | |
44 | protected: | |
45 | TObjArray* fEvents;//!Array containig AliAODs | |
46 | private: | |
62411bd0 | 47 | |
48 | AliAODRun(const AliAODRun&); // Not implemented | |
49 | AliAODRun& operator=(const AliAODRun&); // Not implemented | |
50 | ||
a5556ea5 | 51 | ClassDef(AliAODRun,1) |
52 | }; | |
53 | ||
54 | ||
55 | /**************************************************************************/ | |
56 | ||
57 | inline | |
58 | AliAOD* AliAODRun::GetEvent(Int_t event) const | |
59 | { | |
60 | //returns pointer to AliAOD number "event" | |
61 | //check if array is enough big - protect from error massage from array "Out Of Bounds" | |
62 | if (event>=fEvents->GetSize()) return 0x0;//WARNING, that line relies | |
63 | //on index of first object in TObjArray is 0 | |
64 | //== LowerBound = 0 | |
65 | return (AliAOD*)fEvents->At(event); | |
66 | } | |
67 | /**************************************************************************/ | |
68 | inline | |
afa8b37b | 69 | AliVAODParticle* AliAODRun::GetParticle(Int_t event, Int_t n) |
a5556ea5 | 70 | { |
71 | //returns nth particle from event number event | |
72 | AliAOD* e = GetEvent(event); | |
73 | return (e)?e->GetParticle(n):0x0; | |
74 | } | |
75 | ||
76 | /**************************************************************************/ | |
77 | ||
78 | inline | |
79 | Int_t AliAODRun::GetNumberOfEvents() const | |
80 | { | |
81 | //returns number of events in collection | |
82 | ||
83 | return fEvents->GetEntriesFast(); //there may be empty slots but we do not care | |
84 | //Analysis checks it if return is not NULL | |
85 | } | |
86 | /**************************************************************************/ | |
87 | ||
88 | inline | |
89 | Int_t AliAODRun::GetNumberOfParticlesInEvent(Int_t event) const | |
90 | { | |
91 | //returns number of Particles in event | |
92 | AliAOD* e = GetEvent(event); | |
93 | return (e)?e->GetNumberOfParticles():0x0; | |
94 | } | |
95 | ||
96 | #endif |