[u/mrichter/AliRoot.git] / STEER / AliAODEvent.h

#ifndef AliAODEvent_H
#define AliAODEvent_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------------------------
//     AOD base class
//     Author: Markus Oldenburg, CERN
//-------------------------------------------------------------------------

#include <TBuffer.h>
#include <TClonesArray.h>
#include <TList.h>
#include <TTree.h>
#include <TNamed.h>

#include "AliVEvent.h"
#include "AliVParticle.h"
#include "AliAODHeader.h"
#include "AliAODTrack.h"
#include "AliAODVertex.h"
#include "AliAODv0.h"
#include "AliAODTracklets.h"
#include "AliAODJet.h"
#include "AliAODCaloCells.h"
#include "AliAODCaloCluster.h"
#include "AliAODPmdCluster.h"
#include "AliAODFmdCluster.h"

class TTree;
class TFolder;

class AliAODEvent : public AliVEvent {

 public :
  enum AODListIndex_t {kAODHeader,
		       kAODTracks,
		       kAODVertices,
		       kAODv0,
		       kAODTracklets,
		       kAODJets,
		       kAODEmcalCells,
		       kAODPhosCells,
		       kAODCaloClusters,
		       kAODFmdClusters,
		       kAODPmdClusters,
		       kAODListN
  };

  AliAODEvent();
  virtual ~AliAODEvent();

  AliAODEvent(const AliAODEvent& aodevent);             
  AliAODEvent& operator=(const AliAODEvent& aodevent);  

  void          AddObject(TObject *obj);
  void          RemoveObject(TObject *obj);
  TObject      *FindListObject(const char *objName) ;
  TList        *GetList()                const { return fAODObjects; }

  // -- Header
  AliAODHeader *GetHeader()              const { return fHeader; }
  void          AddHeader(const AliAODHeader* hdx)
    {
	delete fHeader; fHeader = new AliAODHeader(*hdx);
	(fAODObjects->FirstLink())->SetObject(fHeader);
    }

  // setters and getters for header information
  void     SetRunNumber(Int_t n) {if (fHeader) fHeader->SetRunNumber(n);}
  void     SetPeriodNumber(UInt_t n){if (fHeader) fHeader->SetPeriodNumber(n);}
  void     SetOrbitNumber(UInt_t n) {if (fHeader) fHeader->SetOrbitNumber(n);}
  void     SetBunchCrossNumber(UShort_t n) {if (fHeader) fHeader->SetBunchCrossNumber(n);}
  void     SetMagneticField(Double_t mf){if (fHeader) fHeader->SetMagneticField(mf);}

  Int_t    GetRunNumber() const {return fHeader ? fHeader->GetRunNumber() : -999;}
  UInt_t   GetPeriodNumber() const {return fHeader ? fHeader->GetPeriodNumber() : 0;}
  UInt_t   GetOrbitNumber() const {return fHeader ? fHeader->GetOrbitNumber() : 0;}
  UShort_t GetBunchCrossNumber() const {return fHeader ? fHeader->GetBunchCrossNumber() : 0;}
  Double_t GetMagneticField() const {return fHeader ? fHeader->GetMagneticField() : -999.;}
  
  void      SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
  void      SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
  void      SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}

  UInt_t    GetEventType()  const { return fHeader ? fHeader->GetEventType() : 0;}
  ULong64_t GetTriggerMask() const {return fHeader ? fHeader->GetTriggerMask() : 0;}
  UChar_t   GetTriggerCluster() const {return fHeader ? fHeader->GetTriggerCluster() : 0;}

  Double_t  GetZDCN1Energy()        const { return fHeader ? fHeader->GetZDCN1Energy() : -999.; }
  Double_t  GetZDCP1Energy()        const { return fHeader ? fHeader->GetZDCP1Energy() : -999.; }
  Double_t  GetZDCN2Energy()        const { return fHeader ? fHeader->GetZDCN2Energy() : -999.; }
  Double_t  GetZDCP2Energy()        const { return fHeader ? fHeader->GetZDCP2Energy() : -999.; }
  Double_t  GetZDCEMEnergy(Int_t i) const { return fHeader ? fHeader->GetZDCEMEnergy(i) : -999.; }

  // -- Tracks
  TClonesArray *GetTracks()              const { return fTracks; }
  Int_t         GetNTracks()             const { return fTracks->GetEntriesFast(); }
  Int_t         GetNumberOfTracks()      const { return GetNTracks(); }
  AliAODTrack  *GetTrack(Int_t nTrack)   const { return (AliAODTrack*)fTracks->UncheckedAt(nTrack); }
  Int_t         AddTrack(const AliAODTrack* trk)
    {new((*fTracks)[fTracks->GetEntriesFast()]) AliAODTrack(*trk); return fTracks->GetEntriesFast()-1;}
  Int_t         GetMuonTracks(TRefArray *muonTracks) const;

  // -- Vertex
  TClonesArray *GetVertices()            const { return fVertices; }
  Int_t         GetNumberOfVertices()    const { return fVertices->GetEntriesFast(); }
  AliAODVertex *GetVertex(Int_t nVertex) const { return (AliAODVertex*)fVertices->UncheckedAt(nVertex); }
  Int_t         AddVertex(const AliAODVertex* vtx)
  {new((*fVertices)[fVertices->GetEntriesFast()]) AliAODVertex(*vtx); return fVertices->GetEntriesFast()-1;}
  
  // primary vertex
  virtual AliAODVertex *GetPrimaryVertex() const { return GetVertex(0); }

  // V0
  TClonesArray *GetV0s()                 const { return fV0s; }
  Int_t         GetNumberOfV0s()         const { return fV0s->GetEntriesFast(); }
  AliAODv0     *GetV0(Int_t nV0)         const { return (AliAODv0*)fV0s->UncheckedAt(nV0); }
  Int_t         AddV0(const AliAODv0* v0)
  {new((*fV0s)[fV0s->GetEntriesFast()]) AliAODv0(*v0); return fV0s->GetEntriesFast()-1;}

  // -- EMCAL and PHOS Cluster
  TClonesArray *GetCaloClusters()        const { return fCaloClusters; }
  Int_t         GetNCaloClusters()       const { return fCaloClusters->GetEntriesFast(); }
  AliAODCaloCluster *GetCaloCluster(Int_t nCluster) const { return (AliAODCaloCluster*)fCaloClusters->UncheckedAt(nCluster); }
  Int_t         AddCaloCluster(const AliAODCaloCluster* clus)
  {new((*fCaloClusters)[fCaloClusters->GetEntriesFast()]) AliAODCaloCluster(*clus); return fCaloClusters->GetEntriesFast()-1;}

  Int_t GetEMCALClusters(TRefArray *clusters) const;
  Int_t GetPHOSClusters(TRefArray *clusters) const;


  // -- FMD Cluster
  TClonesArray *GetFmdClusters()        const { return fFmdClusters; }
  Int_t         GetNFmdClusters()       const { return fFmdClusters->GetEntriesFast(); }
  AliAODFmdCluster *GetFmdCluster(Int_t nCluster) const { return (AliAODFmdCluster*)fFmdClusters->UncheckedAt(nCluster); }
  Int_t         AddFmdCluster(const AliAODFmdCluster* clus)
  {new((*fFmdClusters)[fFmdClusters->GetEntriesFast()]) AliAODFmdCluster(*clus); return fFmdClusters->GetEntriesFast()-1;}

  // -- PMD Cluster
  TClonesArray *GetPmdClusters()        const { return fPmdClusters; }
  Int_t         GetNPmdClusters()       const { return fPmdClusters->GetEntriesFast(); }
  AliAODPmdCluster *GetPmdCluster(Int_t nCluster) const { return (AliAODPmdCluster*)fPmdClusters->UncheckedAt(nCluster); }
  Int_t         AddPmdCluster(const AliAODPmdCluster* clus)
  {new((*fPmdClusters)[fPmdClusters->GetEntriesFast()]) AliAODPmdCluster(*clus); return fPmdClusters->GetEntriesFast()-1;}

  // -- Jet
  TClonesArray *GetJets()            const { return fJets; }
  Int_t         GetNJets()           const { return fJets->GetEntriesFast(); }
  AliAODJet    *GetJet(Int_t nJet) const { return (AliAODJet*)fJets->UncheckedAt(nJet); }
  Int_t         AddJet(const AliAODJet* vtx)
    {new((*fJets)[fJets->GetEntriesFast()]) AliAODJet(*vtx); return fJets->GetEntriesFast()-1;}

  // -- Tracklets
  AliAODTracklets *GetTracklets() const { return fTracklets; }

  // -- Calorimeter Cells
  AliAODCaloCells *GetEMCALCells() const { return fEmcalCells; }
  AliAODCaloCells *GetPHOSCells() const { return fPhosCells; }

  // -- Services
  void    CreateStdContent();
  void    SetStdNames();
  void    GetStdContent();
  void    CreateStdFolders();
  void    ResetStd(Int_t trkArrSize = 0, 
		   Int_t vtxArrSize = 0, 
		   Int_t v0ArrSize = 0, 
		   Int_t jetSize = 0, 
		   Int_t caloClusSize = 0, 
		   Int_t fmdClusSize = 0, 
		   Int_t pmdClusSize = 0);
  void    ClearStd();
  void    ReadFromTree(TTree *tree, Option_t* opt = "");
  const void WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}

  void  Print(Option_t *option="") const;
  void  MakeEntriesReferencable();
  static void AssignIDtoCollection(TCollection* col);
 private :

  TList   *fAODObjects; //  list of AODObjects
  TFolder *fAODFolder;  //  folder structure of branches
  Bool_t   fConnected;  //! flag if leaves are alreday connected 
  // standard content
  AliAODHeader    *fHeader;       //! event information
  TClonesArray    *fTracks;       //! charged tracks
  TClonesArray    *fVertices;     //! vertices
  TClonesArray    *fV0s;          //! V0s
  AliAODTracklets *fTracklets;    //! SPD tracklets
  TClonesArray    *fJets;         //! jets
  AliAODCaloCells *fEmcalCells;   //! EMCAL calorimenter cells
  AliAODCaloCells *fPhosCells;    //! PHOS calorimenter cells
  TClonesArray    *fCaloClusters; //! calorimeter clusters
  TClonesArray    *fFmdClusters;  //! FMDclusters
  TClonesArray    *fPmdClusters;  //! PMDclusters

  static const char* fAODListName[kAODListN]; //!

  ClassDef(AliAODEvent, 4);
};

#endif
Commit	Line	Data
df9db588	1	#ifndef AliAODEvent_H
	2	#define AliAODEvent_H
	3	/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	//-------------------------------------------------------------------------
	9	// AOD base class
	10	// Author: Markus Oldenburg, CERN
	11	//-------------------------------------------------------------------------
	12
	13	#include <TBuffer.h>
	14	#include <TClonesArray.h>
	15	#include <TList.h>
8ff33247	16	#include <TTree.h>
df9db588	17	#include <TNamed.h>
df9db588	18
08e4c457	19	#include "AliVEvent.h"
08e4c457	20	#include "AliVParticle.h"
df9db588	21	#include "AliAODHeader.h"
	22	#include "AliAODTrack.h"
	23	#include "AliAODVertex.h"
3dd9f9e3	24	#include "AliAODv0.h"
821f8f1d	25	#include "AliAODTracklets.h"
3dd9f9e3	26	#include "AliAODJet.h"
	27	#include "AliAODCaloCells.h"
	28	#include "AliAODCaloCluster.h"
	29	#include "AliAODPmdCluster.h"
	30	#include "AliAODFmdCluster.h"
df9db588	31
8ff33247	32	class TTree;
5c1dc41f	33	class TFolder;
ec4af4c1	34
08e4c457	35	class AliAODEvent : public AliVEvent {
df9db588	36
df9db588	37	public :
08e4c457	38	enum AODListIndex_t {kAODHeader,
	39	kAODTracks,
	40	kAODVertices,
3dd9f9e3	41	kAODv0,
08e4c457	42	kAODTracklets,
3dd9f9e3	43	kAODJets,
37792174	44	kAODEmcalCells,
37792174	45	kAODPhosCells,
3dd9f9e3	46	kAODCaloClusters,
	47	kAODFmdClusters,
	48	kAODPmdClusters,
08e4c457	49	kAODListN
	50	};
	51
df9db588	52	AliAODEvent();
	53	virtual ~AliAODEvent();
	54
5c1dc41f	55	AliAODEvent(const AliAODEvent& aodevent);
5c1dc41f	56	AliAODEvent& operator=(const AliAODEvent& aodevent);
df9db588	57
df9db588	58	void AddObject(TObject *obj);
4b3de353	59	void RemoveObject(TObject *obj);
08e4c457	60	TObject FindListObject(const char objName) ;
df9db588	61	TList *GetList() const { return fAODObjects; }
	62
	63	// -- Header
08e4c457	64	AliAODHeader *GetHeader() const { return fHeader; }
df9db588	65	void AddHeader(const AliAODHeader* hdx)
635db10f	66	{
	67	delete fHeader; fHeader = new AliAODHeader(*hdx);
	68	(fAODObjects->FirstLink())->SetObject(fHeader);
	69	}
df9db588	70
08e4c457	71	// setters and getters for header information
	72	void SetRunNumber(Int_t n) {if (fHeader) fHeader->SetRunNumber(n);}
	73	void SetPeriodNumber(UInt_t n){if (fHeader) fHeader->SetPeriodNumber(n);}
	74	void SetOrbitNumber(UInt_t n) {if (fHeader) fHeader->SetOrbitNumber(n);}
	75	void SetBunchCrossNumber(UShort_t n) {if (fHeader) fHeader->SetBunchCrossNumber(n);}
	76	void SetMagneticField(Double_t mf){if (fHeader) fHeader->SetMagneticField(mf);}
	77
	78	Int_t GetRunNumber() const {return fHeader ? fHeader->GetRunNumber() : -999;}
	79	UInt_t GetPeriodNumber() const {return fHeader ? fHeader->GetPeriodNumber() : 0;}
	80	UInt_t GetOrbitNumber() const {return fHeader ? fHeader->GetOrbitNumber() : 0;}
	81	UShort_t GetBunchCrossNumber() const {return fHeader ? fHeader->GetBunchCrossNumber() : 0;}
	82	Double_t GetMagneticField() const {return fHeader ? fHeader->GetMagneticField() : -999.;}
	83
	84	void SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
	85	void SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
	86	void SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
	87
	88	UInt_t GetEventType() const { return fHeader ? fHeader->GetEventType() : 0;}
	89	ULong64_t GetTriggerMask() const {return fHeader ? fHeader->GetTriggerMask() : 0;}
	90	UChar_t GetTriggerCluster() const {return fHeader ? fHeader->GetTriggerCluster() : 0;}
	91
	92	Double_t GetZDCN1Energy() const { return fHeader ? fHeader->GetZDCN1Energy() : -999.; }
	93	Double_t GetZDCP1Energy() const { return fHeader ? fHeader->GetZDCP1Energy() : -999.; }
	94	Double_t GetZDCN2Energy() const { return fHeader ? fHeader->GetZDCN2Energy() : -999.; }
	95	Double_t GetZDCP2Energy() const { return fHeader ? fHeader->GetZDCP2Energy() : -999.; }
a85132e7	96	Double_t GetZDCEMEnergy(Int_t i) const { return fHeader ? fHeader->GetZDCEMEnergy(i) : -999.; }
08e4c457	97
df9db588	98	// -- Tracks
	99	TClonesArray *GetTracks() const { return fTracks; }
	100	Int_t GetNTracks() const { return fTracks->GetEntriesFast(); }
08e4c457	101	Int_t GetNumberOfTracks() const { return GetNTracks(); }
	102	AliAODTrack GetTrack(Int_t nTrack) const { return (AliAODTrack)fTracks->UncheckedAt(nTrack); }
	103	Int_t AddTrack(const AliAODTrack* trk)
	104	{new((fTracks)[fTracks->GetEntriesFast()]) AliAODTrack(trk); return fTracks->GetEntriesFast()-1;}
fb41236c	105	Int_t GetMuonTracks(TRefArray *muonTracks) const;
df9db588	106
	107	// -- Vertex
	108	TClonesArray *GetVertices() const { return fVertices; }
c1b93f02	109	Int_t GetNumberOfVertices() const { return fVertices->GetEntriesFast(); }
08e4c457	110	AliAODVertex GetVertex(Int_t nVertex) const { return (AliAODVertex)fVertices->UncheckedAt(nVertex); }
	111	Int_t AddVertex(const AliAODVertex* vtx)
	112	{new((fVertices)[fVertices->GetEntriesFast()]) AliAODVertex(vtx); return fVertices->GetEntriesFast()-1;}
5c2fdfa2	113
3dd9f9e3	114	// primary vertex
3dd9f9e3	115	virtual AliAODVertex *GetPrimaryVertex() const { return GetVertex(0); }
df9db588	116
3dd9f9e3	117	// V0
3dd9f9e3	118	TClonesArray *GetV0s() const { return fV0s; }
c1b93f02	119	Int_t GetNumberOfV0s() const { return fV0s->GetEntriesFast(); }
3dd9f9e3	120	AliAODv0 GetV0(Int_t nV0) const { return (AliAODv0)fV0s->UncheckedAt(nV0); }
	121	Int_t AddV0(const AliAODv0* v0)
	122	{new((fV0s)[fV0s->GetEntriesFast()]) AliAODv0(v0); return fV0s->GetEntriesFast()-1;}
	123
	124	// -- EMCAL and PHOS Cluster
	125	TClonesArray *GetCaloClusters() const { return fCaloClusters; }
	126	Int_t GetNCaloClusters() const { return fCaloClusters->GetEntriesFast(); }
	127	AliAODCaloCluster GetCaloCluster(Int_t nCluster) const { return (AliAODCaloCluster)fCaloClusters->UncheckedAt(nCluster); }
	128	Int_t AddCaloCluster(const AliAODCaloCluster* clus)
	129	{new((fCaloClusters)[fCaloClusters->GetEntriesFast()]) AliAODCaloCluster(clus); return fCaloClusters->GetEntriesFast()-1;}
	130
78902954	131	Int_t GetEMCALClusters(TRefArray *clusters) const;
	132	Int_t GetPHOSClusters(TRefArray *clusters) const;
	133
	134
3dd9f9e3	135	// -- FMD Cluster
	136	TClonesArray *GetFmdClusters() const { return fFmdClusters; }
	137	Int_t GetNFmdClusters() const { return fFmdClusters->GetEntriesFast(); }
	138	AliAODFmdCluster GetFmdCluster(Int_t nCluster) const { return (AliAODFmdCluster)fFmdClusters->UncheckedAt(nCluster); }
	139	Int_t AddFmdCluster(const AliAODFmdCluster* clus)
	140	{new((fFmdClusters)[fFmdClusters->GetEntriesFast()]) AliAODFmdCluster(clus); return fFmdClusters->GetEntriesFast()-1;}
	141
	142	// -- PMD Cluster
	143	TClonesArray *GetPmdClusters() const { return fPmdClusters; }
	144	Int_t GetNPmdClusters() const { return fPmdClusters->GetEntriesFast(); }
	145	AliAODPmdCluster GetPmdCluster(Int_t nCluster) const { return (AliAODPmdCluster)fPmdClusters->UncheckedAt(nCluster); }
	146	Int_t AddPmdCluster(const AliAODPmdCluster* clus)
	147	{new((fPmdClusters)[fPmdClusters->GetEntriesFast()]) AliAODPmdCluster(clus); return fPmdClusters->GetEntriesFast()-1;}
df9db588	148
	149	// -- Jet
	150	TClonesArray *GetJets() const { return fJets; }
	151	Int_t GetNJets() const { return fJets->GetEntriesFast(); }
08e4c457	152	AliAODJet GetJet(Int_t nJet) const { return (AliAODJet)fJets->UncheckedAt(nJet); }
	153	Int_t AddJet(const AliAODJet* vtx)
	154	{new((fJets)[fJets->GetEntriesFast()]) AliAODJet(vtx); return fJets->GetEntriesFast()-1;}
df9db588	155
821f8f1d	156	// -- Tracklets
	157	AliAODTracklets *GetTracklets() const { return fTracklets; }
	158
3dd9f9e3	159	// -- Calorimeter Cells
e649177a	160	AliAODCaloCells *GetEMCALCells() const { return fEmcalCells; }
e649177a	161	AliAODCaloCells *GetPHOSCells() const { return fPhosCells; }
3dd9f9e3	162
df9db588	163	// -- Services
ec4af4c1	164	void CreateStdContent();
08e4c457	165	void SetStdNames();
ec4af4c1	166	void GetStdContent();
5c1dc41f	167	void CreateStdFolders();
3dd9f9e3	168	void ResetStd(Int_t trkArrSize = 0,
	169	Int_t vtxArrSize = 0,
	170	Int_t v0ArrSize = 0,
	171	Int_t jetSize = 0,
	172	Int_t caloClusSize = 0,
	173	Int_t fmdClusSize = 0,
	174	Int_t pmdClusSize = 0);
ec4af4c1	175	void ClearStd();
3b93c32a	176	void ReadFromTree(TTree tree, Option_t opt = "");
08e4c457	177	const void WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}
	178
	179	void Print(Option_t *option="") const;
c8e4f399	180	void MakeEntriesReferencable();
c8e4f399	181	static void AssignIDtoCollection(TCollection* col);
08e4c457	182	private :
df9db588	183
5c1dc41f	184	TList *fAODObjects; // list of AODObjects
5c1dc41f	185	TFolder *fAODFolder; // folder structure of branches
dd389a30	186	Bool_t fConnected; //! flag if leaves are alreday connected
df9db588	187	// standard content
3dd9f9e3	188	AliAODHeader *fHeader; //! event information
	189	TClonesArray *fTracks; //! charged tracks
	190	TClonesArray *fVertices; //! vertices
	191	TClonesArray *fV0s; //! V0s
	192	AliAODTracklets *fTracklets; //! SPD tracklets
	193	TClonesArray *fJets; //! jets
37792174	194	AliAODCaloCells *fEmcalCells; //! EMCAL calorimenter cells
37792174	195	AliAODCaloCells *fPhosCells; //! PHOS calorimenter cells
3dd9f9e3	196	TClonesArray *fCaloClusters; //! calorimeter clusters
	197	TClonesArray *fFmdClusters; //! FMDclusters
	198	TClonesArray *fPmdClusters; //! PMDclusters
821f8f1d	199
08e4c457	200	static const char* fAODListName[kAODListN]; //!
08e4c457	201
5c1dc41f	202	ClassDef(AliAODEvent, 4);
df9db588	203	};
	204
	205	#endif