[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 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);  // not implemented
  //AliAODEvent& operator=(const AliAODEvent& aodevent);  // not implemented

  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         GetNVertices()           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         GetNV0s()                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;}

  // -- 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    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);
  const void WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}

  void  Print(Option_t *option="") const;

 private :

  TList *fAODObjects; //  list of AODObjects
  
  // 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,3);
};

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