Second ZDC module. More ZDC information included in the ESD and AOD classes (Chiara)
[u/mrichter/AliRoot.git] / STEER / AliMCEvent.h
1 // -*- mode: C++ -*- 
2 #ifndef ALIMCEVENT_H
3 #define ALIMCEVENT_H
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7
8 /* $Id$ */
9
10 //-------------------------------------------------------------------------
11 //                          Class AliMCEvent
12 //      
13 // Origin: Andreas.Morsch, CERN, andreas.morsch@cern.ch 
14 //-------------------------------------------------------------------------
15
16
17 #include <TTree.h>
18 #include <TRefArray.h>
19
20 #include <AliVEvent.h>
21 #include "AliVHeader.h"
22 #include "AliMCParticle.h"
23
24 class AliStack;
25 class AliHeader;
26 class AliGenEventHeader;
27
28 class TClonesArray;
29
30 class AliMCEvent : public AliVEvent {
31
32 public:
33
34     AliMCEvent();
35     virtual ~AliMCEvent() {;} 
36     AliMCEvent(const AliMCEvent& mcEvnt); 
37     AliMCEvent& operator=(const AliMCEvent& mcEvnt);
38     //
39     // Methods implementing the interface
40     //
41     // Services
42     virtual void AddObject(TObject* /*obj*/)               {;}
43     virtual TObject* FindListObject(const char */*name*/)  {return 0;}
44     virtual TList* GetList() const                         {return 0;}
45     virtual void CreateStdContent()                        {;} 
46     virtual void GetStdContent()                           {;}
47     virtual void ReadFromTree(TTree * /*tree*/)            {;}
48     virtual const void WriteToTree(TTree* /*tree*/)  const {;}
49
50     virtual void SetStdNames()                             {;}
51     virtual void Print(Option_t */*option=""*/)  const     {;}
52     
53
54     // Header
55     virtual AliVHeader* GetHeader()          const         {return 0;}
56
57     // Delegated methods for fESDRun or AODHeader
58   
59     virtual void     SetRunNumber(Int_t /*n*/)             {;}
60     virtual void     SetPeriodNumber(UInt_t /*n*/)         {;}
61     virtual void     SetMagneticField(Double_t /*mf*/)     {;}
62     
63   
64     virtual Int_t    GetRunNumber()          const         {return 0;}
65     virtual UInt_t   GetPeriodNumber()       const {return 0;}
66     virtual Double_t GetMagneticField()      const         {return 0.;}
67
68     // Setters not needed
69     virtual void      SetOrbitNumber(UInt_t /*n*/)         {;}
70     virtual void      SetBunchCrossNumber(UShort_t /*n*/)  {;}
71     virtual void      SetEventType(UInt_t /*eventType*/)   {;}
72     virtual void      SetTriggerMask(ULong64_t /*n*/)      {;}
73     virtual void      SetTriggerCluster(UChar_t /*n*/)     {;} 
74
75     virtual UInt_t    GetOrbitNumber()       const {return 0;}
76     virtual UShort_t  GetBunchCrossNumber()  const {return 0;}
77     
78     virtual UInt_t    GetEventType()         const         {return 0;}
79
80     virtual ULong64_t GetTriggerMask()        const {return 0;}
81     virtual UChar_t   GetTriggerCluster()     const {return 0;}
82     virtual Double_t  GetZDCN1Energy()        const {return 0.;}
83     virtual Double_t  GetZDCP1Energy()        const {return 0.;}
84     virtual Double_t  GetZDCN2Energy()        const {return 0.;}
85     virtual Double_t  GetZDCP2Energy()        const {return 0.;}
86     virtual Double_t  GetZDCEMEnergy(Int_t i) const {return 0.;}
87     // Tracks
88     virtual AliMCParticle *GetTrack(Int_t i) const;
89     virtual Int_t     GetNumberOfTracks()    const {return fNparticles;}
90
91     //
92     // MC Specific methods
93     //
94     // Getters
95     AliStack*    Stack()   {return fStack;}
96     AliHeader*   Header()  {return fHeader;}
97     AliGenEventHeader* GenEventHeader();
98     // Services
99     virtual void      ConnectTreeE (TTree* tree);
100     virtual void      ConnectTreeK (TTree* tree);
101     virtual void      ConnectTreeTR(TTree* tree);
102     virtual void      Clean();
103     virtual void      FinishEvent();
104     virtual Int_t     GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
105     virtual void      DrawCheck(Int_t i, Int_t search);
106 private:
107     virtual void      ReorderAndExpandTreeTR();
108     
109 private:
110     AliStack         *fStack;           //! Current pointer to stack
111     TClonesArray     *fMCParticles;     //! Pointer to list of particles
112     TRefArray        *fMCParticleMap;   //! Map of MC Particles
113     AliHeader        *fHeader;          //! Current pointer to header
114     TClonesArray     *fTRBuffer;        //! Track reference buffer    
115     TClonesArray     *fTrackReferences; //! Array of track references
116     TTree            *fTreeTR;          //! Pointer to Track Reference Tree
117     TTree            *fTmpTreeTR;       //! Temporary tree TR to read old format
118     TFile            *fTmpFileTR;       //! Temporary file with TreeTR to read old format
119     Int_t             fNprimaries;      //! Number of primaries
120     Int_t             fNparticles;      //! Number of particles
121     ClassDef(AliMCEvent, 0)  // AliVEvent realisation for MC data
122 };
123
124
125 #endif 
126