4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
10 //-------------------------------------------------------------------------
13 // Origin: Andreas.Morsch, CERN, andreas.morsch@cern.ch
14 //-------------------------------------------------------------------------
18 #include <TRefArray.h>
20 #include <AliVEvent.h>
21 #include "AliVHeader.h"
22 #include "AliMCParticle.h"
26 class AliGenEventHeader;
31 class AliMCEvent : public AliVEvent {
36 virtual ~AliMCEvent() {;}
37 AliMCEvent(const AliMCEvent& mcEvnt);
38 AliMCEvent& operator=(const AliMCEvent& mcEvnt);
40 // Methods implementing the interface
43 virtual void AddObject(TObject* /*obj*/) {;}
44 virtual TObject* FindListObject(const char */*name*/) {return 0;}
45 virtual TList* GetList() const {return 0;}
46 virtual void CreateStdContent() {;}
47 virtual void GetStdContent() {;}
48 virtual void ReadFromTree(TTree * /*tree*/, Option_t* /*opt*/) {;}
49 virtual void WriteToTree(TTree* /*tree*/) const {;}
51 virtual void SetStdNames() {;}
52 virtual void Print(Option_t */*option=""*/) const {;}
56 virtual AliVHeader* GetHeader() const {return 0;}
58 // Delegated methods for fESDRun or AODHeader
60 virtual void SetRunNumber(Int_t /*n*/) {;}
61 virtual void SetPeriodNumber(UInt_t /*n*/) {;}
62 virtual void SetMagneticField(Double_t /*mf*/) {;}
65 virtual Int_t GetRunNumber() const {return 0;}
66 virtual UInt_t GetPeriodNumber() const {return 0;}
67 virtual Double_t GetMagneticField() const {return 0.;}
70 virtual void SetOrbitNumber(UInt_t /*n*/) {;}
71 virtual void SetBunchCrossNumber(UShort_t /*n*/) {;}
72 virtual void SetEventType(UInt_t /*eventType*/) {;}
73 virtual void SetTriggerMask(ULong64_t /*n*/) {;}
74 virtual void SetTriggerCluster(UChar_t /*n*/) {;}
76 virtual UInt_t GetOrbitNumber() const {return 0;}
77 virtual UShort_t GetBunchCrossNumber() const {return 0;}
79 virtual UInt_t GetEventType() const {return 0;}
81 virtual ULong64_t GetTriggerMask() const {return 0;}
82 virtual UChar_t GetTriggerCluster() const {return 0;}
83 virtual Double_t GetZDCN1Energy() const {return 0.;}
84 virtual Double_t GetZDCP1Energy() const {return 0.;}
85 virtual Double_t GetZDCN2Energy() const {return 0.;}
86 virtual Double_t GetZDCP2Energy() const {return 0.;}
87 virtual Double_t GetZDCEMEnergy(Int_t /*i*/)
90 virtual AliMCParticle *GetTrack(Int_t i) const;
91 virtual Int_t GetNumberOfTracks() const {return fNparticles;}
92 virtual Int_t GetNumberOfV0s() const {return -1;}
93 virtual Int_t GetNumberOfCascades() const {return -1;}
96 // MC Specific methods
99 AliStack* Stack() {return fStack;}
100 AliHeader* Header() {return fHeader;}
101 AliGenEventHeader* GenEventHeader();
103 virtual void ConnectTreeE (TTree* tree);
104 virtual void ConnectTreeK (TTree* tree);
105 virtual void ConnectTreeTR(TTree* tree);
106 virtual void Clean();
107 virtual void InitEvent();
108 virtual void FinishEvent();
109 virtual Int_t GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
110 virtual void DrawCheck(Int_t i, Int_t search);
111 virtual void AddSubsidiaryEvent(AliMCEvent* event);
112 virtual Int_t GetNumberOfPrimaries() {return fNprimaries;}
113 virtual Int_t GetPrimaryOffset() const {return fPrimaryOffset;}
114 virtual Int_t GetSecondaryOffset() const {return fSecondaryOffset;}
115 virtual void SetPrimaryOffset(Int_t ioff) {fPrimaryOffset = ioff;}
116 virtual void SetSecondaryOffset(Int_t ioff) {fSecondaryOffset = ioff;}
117 virtual Bool_t IsPhysicalPrimary(Int_t i);
118 virtual Int_t BgLabelToIndex(Int_t label);
119 static Int_t BgLabelOffset() {return fgkBgLabelOffset;}
122 virtual void ReorderAndExpandTreeTR();
123 virtual Int_t FindIndexAndEvent(Int_t oldidx, AliMCEvent*& event) const;
125 AliStack *fStack; // Current pointer to stack
126 TClonesArray *fMCParticles; // Pointer to list of particles
127 TRefArray *fMCParticleMap; // Map of MC Particles
128 AliHeader *fHeader; // Current pointer to header
129 TClonesArray *fTRBuffer; // Track reference buffer
130 TClonesArray *fTrackReferences; // Array of track references
131 TTree *fTreeTR; // Pointer to Track Reference Tree
132 TTree *fTmpTreeTR; // Temporary tree TR to read old format
133 TFile *fTmpFileTR; // Temporary file with TreeTR to read old format
134 Int_t fNprimaries; // Number of primaries
135 Int_t fNparticles; // Number of particles
136 TList *fSubsidiaryEvents; // List of possible subsidiary events (for example merged underlying event)
137 Int_t fPrimaryOffset; // Offset for primaries
138 Int_t fSecondaryOffset; // Offset for secondaries
139 static Int_t fgkBgLabelOffset; // Standard branch name
140 ClassDef(AliMCEvent, 1) // AliVEvent realisation for MC data