[u/mrichter/AliRoot.git] / STEER / AOD / 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 "AliAODcascade.h"
#include "AliAODTracklets.h"
#include "AliAODJet.h"
#include "AliAODCaloCells.h"
#include "AliAODCaloCluster.h"
#include "AliAODCaloTrigger.h"
#include "AliAODPmdCluster.h"
#include "AliAODFmdCluster.h"
#include "AliAODDimuon.h"
#include "AliAODTZERO.h"
#include "AliAODVZERO.h"
#include "AliAODZDC.h"
#ifdef MFT_UPGRADE
#include "AliAODMFT.h"
#endif

class TTree;
class TFolder;
class AliCentrality;
class AliEventplane;

class AliAODEvent : public AliVEvent {

 public :
  enum AODListIndex_t {kAODHeader,
		       kAODTracks,
		       kAODVertices,
		       kAODv0,
		       kAODcascade,
		       kAODTracklets,
		       kAODJets,
		       kAODEmcalCells,
		       kAODPhosCells,
		       kAODCaloClusters,
	           kAODEMCALTrigger,
	           kAODPHOSTrigger,
		       kAODFmdClusters,
		       kAODPmdClusters,
		       kAODDimuons,
		       kAODTZERO,
		       kAODVZERO,
		       kAODZDC,
		       kAODListN
	           #ifdef MFT_UPGRADE
	           ,kAODVZERO
			   #endif
  };

  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) const;
  TList        *GetList()                const { return fAODObjects; }
  void          SetConnected(Bool_t conn=kTRUE) {fConnected=conn;}
  Bool_t        GetConnected() const {return fConnected;}

  // -- 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);}
  void     SetMuonMagFieldScale(Double_t mf){if (fHeader) fHeader->SetMuonMagFieldScale(mf);}
  void     SetDiamond(Float_t xy[2],Float_t cov[3]){if (fHeader) fHeader->SetDiamond(xy,cov);}
  void     SetDiamondZ(Float_t z, Float_t sig2z){if (fHeader) fHeader->SetDiamondZ(z,sig2z);}
  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.;}
  Double_t GetMuonMagFieldScale() const {return fHeader ? fHeader->GetMuonMagFieldScale() : -999.;}
  Double_t GetDiamondX() const {return fHeader ? fHeader->GetDiamondX() : -999.;}
  Double_t GetDiamondY() const {return fHeader ? fHeader->GetDiamondY() : -999.;}
  Double_t GetDiamondZ() const {return fHeader ? fHeader->GetDiamondZ() : -999.;}
  void     GetDiamondCovXY(Float_t cov[3]) const {cov[0]=-999.; if(fHeader) fHeader->GetDiamondCovXY(cov);}
  Double_t GetSigma2DiamondX() const {return fHeader ? fHeader->GetSigma2DiamondX() : -999.;}
  Double_t GetSigma2DiamondY() const {return fHeader ? fHeader->GetSigma2DiamondY() : -999.;}
  Double_t GetSigma2DiamondZ() const {return fHeader ? fHeader->GetSigma2DiamondZ() : -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;}
  TString   GetFiredTriggerClasses()const { return fHeader->GetFiredTriggerClasses();};
  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? fTracks->GetEntriesFast() : 0; }
  Int_t         GetNumberOfTracks()      const { return GetNTracks(); }
  AliAODTrack  *GetTrack(Int_t nTrack)   const { return fTracks ? (AliAODTrack*)fTracks->UncheckedAt(nTrack):0; }
  Int_t         AddTrack(const AliAODTrack* trk)
    {new((*fTracks)[fTracks->GetEntriesFast()]) AliAODTrack(*trk); return fTracks->GetEntriesFast()-1;}
  Int_t         GetMuonTracks(TRefArray *muonTracks) const;
  Int_t         GetNumberOfMuonTracks() const;

  // -- Vertex
  TClonesArray *GetVertices()            const { return fVertices; }
  Int_t         GetNumberOfVertices()    const { return fVertices?fVertices->GetEntriesFast():0; }
  AliAODVertex *GetVertex(Int_t nVertex) const { return fVertices?(AliAODVertex*)fVertices->At(nVertex):0; }
  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); }
  virtual AliAODVertex *GetPrimaryVertexSPD() const;

  // -- Pileup vertices 
  Int_t         GetNumberOfPileupVerticesTracks()   const;
  Int_t         GetNumberOfPileupVerticesSPD()    const;
  virtual AliAODVertex *GetPileupVertexSPD(Int_t iV=0) const;
  virtual AliAODVertex *GetPileupVertexTracks(Int_t iV=0) const;
  virtual Bool_t  IsPileupFromSPD(Int_t minContributors=3, Double_t minZdist=0.8, Double_t nSigmaZdist=3., Double_t nSigmaDiamXY=2., Double_t nSigmaDiamZ=5.) const;
  virtual Bool_t IsPileupFromSPDInMultBins() const;


  // 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;}

  // Cascades
  TClonesArray  *GetCascades()            const { return fCascades; }
  Int_t          GetNumberOfCascades()    const { return fCascades->GetEntriesFast(); }
  AliAODcascade *GetCascade(Int_t nCasc)  const { return (AliAODcascade*)fCascades->UncheckedAt(nCasc); }
  Int_t          AddCascade(const AliAODcascade* cascade)
  {new((*fCascades)[fCascades->GetEntriesFast()]) AliAODcascade(*cascade); return fCascades->GetEntriesFast()-1;}

  // -- EMCAL and PHOS Cluster
  TClonesArray *GetCaloClusters()          const { return fCaloClusters; }
  Int_t         GetNumberOfCaloClusters()  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;}
  AliAODCaloTrigger *GetCaloTrigger(TString calo) const 
  { 	  
     if (calo.Contains("EMCAL")) return fEMCALTrigger;
     else
     return fPHOSTrigger; 
  }	
	
  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?fJets->GetEntriesFast():0; }
  AliAODJet    *GetJet(Int_t nJet) const { return fJets?(AliAODJet*)fJets->UncheckedAt(nJet):0; }
  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; }
  const TGeoHMatrix* GetPHOSMatrix(Int_t /*i*/) const { return NULL; }
  const TGeoHMatrix* GetEMCALMatrix(Int_t /*i*/)const { return NULL; }


  // -- Dimuons
  TClonesArray *GetDimuons()              const { return fDimuons; }
  Int_t         GetNDimuons()             const { return fDimuons->GetEntriesFast(); }
  Int_t         GetNumberOfDimuons()      const { return GetNDimuons(); }
  AliAODDimuon *GetDimuon(Int_t nDimu)    const { return (AliAODDimuon*)fDimuons->UncheckedAt(nDimu); }
  Int_t         AddDimuon(const AliAODDimuon* dimu)
    {new((*fDimuons)[fDimuons->GetEntriesFast()]) AliAODDimuon(*dimu); return fDimuons->GetEntriesFast()-1;}
  
  // -- 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 cascadeArrSize = 0,
		   Int_t jetSize = 0, 
		   Int_t caloClusSize = 0, 
		   Int_t fmdClusSize = 0, 
		   Int_t pmdClusSize = 0,
		   Int_t dimuonArrsize =0
		   );
  void    ClearStd();
  void    Reset() {ClearStd();} 
  void    ReadFromTree(TTree *tree, Option_t* opt = "");
  void    WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}

  void  Print(Option_t *option="") const;
  void  MakeEntriesReferencable();
  static void AssignIDtoCollection(const TCollection* col);
  
    //Following needed only for mixed event
  virtual Int_t        EventIndex(Int_t)       const {return 0;}
  virtual Int_t        EventIndexForCaloCluster(Int_t) const {return 0;}
  virtual Int_t        EventIndexForPHOSCell(Int_t)    const {return 0;}
  virtual Int_t        EventIndexForEMCALCell(Int_t)   const {return 0;} 
  AliCentrality*       GetCentrality() {return fHeader->GetCentralityP();} 
  AliEventplane*       GetEventplane() {return fHeader->GetEventplaneP();}

  // TZERO 
  AliAODTZERO *GetTZEROData() const { return fAODTZERO; }
 
  // VZERO 
  AliAODVZERO *GetVZEROData() const { return fAODVZERO; }
  virtual const Float_t* GetVZEROEqFactors() const {return fHeader?fHeader->GetVZEROEqFactors():0x0;}
  virtual Float_t        GetVZEROEqMultiplicity(Int_t i) const;
  void           SetVZEROEqFactors(const Float_t *factors) const {
    if(fHeader && factors)
      fHeader->SetVZEROEqFactors(factors);}

  //ZDC
  AliAODZDC   *GetZDCData() const { return fAODZDC; }

#ifdef MFT_UPGRADE
  // MFT 
  AliAODMFT *GetMFTData() const { return fAODMFT; }
#endif

  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
  TClonesArray    *fCascades;     //! Cascades
  AliAODTracklets *fTracklets;    //! SPD tracklets
  TClonesArray    *fJets;         //! jets
  AliAODCaloCells *fEmcalCells;   //! EMCAL calorimenter cells
  AliAODCaloCells *fPhosCells;    //! PHOS calorimenter cells
  TClonesArray    *fCaloClusters; //! calorimeter clusters
  AliAODCaloTrigger *fEMCALTrigger; //! EMCAL Trigger information
  AliAODCaloTrigger *fPHOSTrigger;  //! PHOS Trigger information
  TClonesArray    *fFmdClusters;  //! FMDclusters
  TClonesArray    *fPmdClusters;  //! PMDclusters
  TClonesArray    *fDimuons;      //! dimuons
  AliAODTZERO     *fAODTZERO;     //! TZERO AOD
  AliAODVZERO     *fAODVZERO;     //! VZERO AOD
  AliAODZDC       *fAODZDC;       //! ZDC AOD
#ifdef MFT_UPGRADE
  AliAODMFT       *fAODMFT;       //! VZERO AOD
#endif
  
  static const char* fAODListName[kAODListN]; //!

  ClassDef(AliAODEvent,89);
};

#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"
9a244178	25	#include "AliAODcascade.h"
821f8f1d	26	#include "AliAODTracklets.h"
3dd9f9e3	27	#include "AliAODJet.h"
	28	#include "AliAODCaloCells.h"
	29	#include "AliAODCaloCluster.h"
720f7306	30	#include "AliAODCaloTrigger.h"
3dd9f9e3	31	#include "AliAODPmdCluster.h"
3dd9f9e3	32	#include "AliAODFmdCluster.h"
866d8d78	33	#include "AliAODDimuon.h"
a1f96974	34	#include "AliAODTZERO.h"
ae2bccf1	35	#include "AliAODVZERO.h"
16d867eb	36	#include "AliAODZDC.h"
820b4d9e	37	#ifdef MFT_UPGRADE
	38	#include "AliAODMFT.h"
	39	#endif
df9db588	40
8ff33247	41	class TTree;
5c1dc41f	42	class TFolder;
befa9db4	43	class AliCentrality;
ce7adfe9	44	class AliEventplane;
ec4af4c1	45
08e4c457	46	class AliAODEvent : public AliVEvent {
df9db588	47
df9db588	48	public :
08e4c457	49	enum AODListIndex_t {kAODHeader,
	50	kAODTracks,
	51	kAODVertices,
3dd9f9e3	52	kAODv0,
9a244178	53	kAODcascade,
08e4c457	54	kAODTracklets,
3dd9f9e3	55	kAODJets,
37792174	56	kAODEmcalCells,
37792174	57	kAODPhosCells,
3dd9f9e3	58	kAODCaloClusters,
720f7306	59	kAODEMCALTrigger,
720f7306	60	kAODPHOSTrigger,
3dd9f9e3	61	kAODFmdClusters,
3dd9f9e3	62	kAODPmdClusters,
866d8d78	63	kAODDimuons,
a1f96974	64	kAODTZERO,
ae2bccf1	65	kAODVZERO,
16d867eb	66	kAODZDC,
08e4c457	67	kAODListN
820b4d9e	68	#ifdef MFT_UPGRADE
	69	,kAODVZERO
	70	#endif
08e4c457	71	};
08e4c457	72
df9db588	73	AliAODEvent();
	74	virtual ~AliAODEvent();
	75
9a244178	76	AliAODEvent(const AliAODEvent& aodevent);
9a244178	77	AliAODEvent& operator=(const AliAODEvent& aodevent);
df9db588	78
df9db588	79	void AddObject(TObject *obj);
4b3de353	80	void RemoveObject(TObject *obj);
2811495d	81	TObject FindListObject(const char objName) const;
df9db588	82	TList *GetList() const { return fAODObjects; }
788578f5	83	void SetConnected(Bool_t conn=kTRUE) {fConnected=conn;}
788578f5	84	Bool_t GetConnected() const {return fConnected;}
df9db588	85
df9db588	86	// -- Header
08e4c457	87	AliAODHeader *GetHeader() const { return fHeader; }
df9db588	88	void AddHeader(const AliAODHeader* hdx)
635db10f	89	{
	90	delete fHeader; fHeader = new AliAODHeader(*hdx);
	91	(fAODObjects->FirstLink())->SetObject(fHeader);
	92	}
df9db588	93
08e4c457	94	// setters and getters for header information
	95	void SetRunNumber(Int_t n) {if (fHeader) fHeader->SetRunNumber(n);}
	96	void SetPeriodNumber(UInt_t n){if (fHeader) fHeader->SetPeriodNumber(n);}
	97	void SetOrbitNumber(UInt_t n) {if (fHeader) fHeader->SetOrbitNumber(n);}
	98	void SetBunchCrossNumber(UShort_t n) {if (fHeader) fHeader->SetBunchCrossNumber(n);}
	99	void SetMagneticField(Double_t mf){if (fHeader) fHeader->SetMagneticField(mf);}
33fe5eb1	100	void SetMuonMagFieldScale(Double_t mf){if (fHeader) fHeader->SetMuonMagFieldScale(mf);}
3bfd99da	101	void SetDiamond(Float_t xy[2],Float_t cov[3]){if (fHeader) fHeader->SetDiamond(xy,cov);}
a98c78e5	102	void SetDiamondZ(Float_t z, Float_t sig2z){if (fHeader) fHeader->SetDiamondZ(z,sig2z);}
08e4c457	103	Int_t GetRunNumber() const {return fHeader ? fHeader->GetRunNumber() : -999;}
	104	UInt_t GetPeriodNumber() const {return fHeader ? fHeader->GetPeriodNumber() : 0;}
	105	UInt_t GetOrbitNumber() const {return fHeader ? fHeader->GetOrbitNumber() : 0;}
	106	UShort_t GetBunchCrossNumber() const {return fHeader ? fHeader->GetBunchCrossNumber() : 0;}
	107	Double_t GetMagneticField() const {return fHeader ? fHeader->GetMagneticField() : -999.;}
33fe5eb1	108	Double_t GetMuonMagFieldScale() const {return fHeader ? fHeader->GetMuonMagFieldScale() : -999.;}
3bfd99da	109	Double_t GetDiamondX() const {return fHeader ? fHeader->GetDiamondX() : -999.;}
3bfd99da	110	Double_t GetDiamondY() const {return fHeader ? fHeader->GetDiamondY() : -999.;}
a98c78e5	111	Double_t GetDiamondZ() const {return fHeader ? fHeader->GetDiamondZ() : -999.;}
3bfd99da	112	void GetDiamondCovXY(Float_t cov[3]) const {cov[0]=-999.; if(fHeader) fHeader->GetDiamondCovXY(cov);}
a98c78e5	113	Double_t GetSigma2DiamondX() const {return fHeader ? fHeader->GetSigma2DiamondX() : -999.;}
	114	Double_t GetSigma2DiamondY() const {return fHeader ? fHeader->GetSigma2DiamondY() : -999.;}
	115	Double_t GetSigma2DiamondZ() const {return fHeader ? fHeader->GetSigma2DiamondZ() : -999.;}
08e4c457	116
	117	void SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
	118	void SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
	119	void SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
	120
27346f69	121	UInt_t GetEventType() const { return fHeader ? fHeader->GetEventType() : 0;}
	122	ULong64_t GetTriggerMask() const { return fHeader ? fHeader->GetTriggerMask() : 0;}
	123	UChar_t GetTriggerCluster() const { return fHeader ? fHeader->GetTriggerCluster() : 0;}
	124	TString GetFiredTriggerClasses()const { return fHeader->GetFiredTriggerClasses();};
08e4c457	125	Double_t GetZDCN1Energy() const { return fHeader ? fHeader->GetZDCN1Energy() : -999.; }
	126	Double_t GetZDCP1Energy() const { return fHeader ? fHeader->GetZDCP1Energy() : -999.; }
	127	Double_t GetZDCN2Energy() const { return fHeader ? fHeader->GetZDCN2Energy() : -999.; }
	128	Double_t GetZDCP2Energy() const { return fHeader ? fHeader->GetZDCP2Energy() : -999.; }
a85132e7	129	Double_t GetZDCEMEnergy(Int_t i) const { return fHeader ? fHeader->GetZDCEMEnergy(i) : -999.; }
08e4c457	130
df9db588	131	// -- Tracks
df9db588	132	TClonesArray *GetTracks() const { return fTracks; }
5d691d96	133	Int_t GetNTracks() const { return fTracks? fTracks->GetEntriesFast() : 0; }
08e4c457	134	Int_t GetNumberOfTracks() const { return GetNTracks(); }
5d691d96	135	AliAODTrack GetTrack(Int_t nTrack) const { return fTracks ? (AliAODTrack)fTracks->UncheckedAt(nTrack):0; }
08e4c457	136	Int_t AddTrack(const AliAODTrack* trk)
08e4c457	137	{new((fTracks)[fTracks->GetEntriesFast()]) AliAODTrack(trk); return fTracks->GetEntriesFast()-1;}
fb41236c	138	Int_t GetMuonTracks(TRefArray *muonTracks) const;
fa8b0e56	139	Int_t GetNumberOfMuonTracks() const;
df9db588	140
	141	// -- Vertex
	142	TClonesArray *GetVertices() const { return fVertices; }
5d691d96	143	Int_t GetNumberOfVertices() const { return fVertices?fVertices->GetEntriesFast():0; }
5d691d96	144	AliAODVertex GetVertex(Int_t nVertex) const { return fVertices?(AliAODVertex)fVertices->At(nVertex):0; }
08e4c457	145	Int_t AddVertex(const AliAODVertex* vtx)
08e4c457	146	{new((fVertices)[fVertices->GetEntriesFast()]) AliAODVertex(vtx); return fVertices->GetEntriesFast()-1;}
5c2fdfa2	147
3dd9f9e3	148	// primary vertex
3dd9f9e3	149	virtual AliAODVertex *GetPrimaryVertex() const { return GetVertex(0); }
a98c78e5	150	virtual AliAODVertex *GetPrimaryVertexSPD() const;
	151
	152	// -- Pileup vertices
	153	Int_t GetNumberOfPileupVerticesTracks() const;
	154	Int_t GetNumberOfPileupVerticesSPD() const;
	155	virtual AliAODVertex *GetPileupVertexSPD(Int_t iV=0) const;
	156	virtual AliAODVertex *GetPileupVertexTracks(Int_t iV=0) const;
	157	virtual Bool_t IsPileupFromSPD(Int_t minContributors=3, Double_t minZdist=0.8, Double_t nSigmaZdist=3., Double_t nSigmaDiamXY=2., Double_t nSigmaDiamZ=5.) const;
b46ff4b0	158	virtual Bool_t IsPileupFromSPDInMultBins() const;
a98c78e5	159
df9db588	160
3dd9f9e3	161	// V0
3dd9f9e3	162	TClonesArray *GetV0s() const { return fV0s; }
c1b93f02	163	Int_t GetNumberOfV0s() const { return fV0s->GetEntriesFast(); }
3dd9f9e3	164	AliAODv0 GetV0(Int_t nV0) const { return (AliAODv0)fV0s->UncheckedAt(nV0); }
	165	Int_t AddV0(const AliAODv0* v0)
	166	{new((fV0s)[fV0s->GetEntriesFast()]) AliAODv0(v0); return fV0s->GetEntriesFast()-1;}
	167
9a244178	168	// Cascades
	169	TClonesArray *GetCascades() const { return fCascades; }
	170	Int_t GetNumberOfCascades() const { return fCascades->GetEntriesFast(); }
	171	AliAODcascade GetCascade(Int_t nCasc) const { return (AliAODcascade)fCascades->UncheckedAt(nCasc); }
	172	Int_t AddCascade(const AliAODcascade* cascade)
	173	{new((fCascades)[fCascades->GetEntriesFast()]) AliAODcascade(cascade); return fCascades->GetEntriesFast()-1;}
	174
3dd9f9e3	175	// -- EMCAL and PHOS Cluster
c8fe2783	176	TClonesArray *GetCaloClusters() const { return fCaloClusters; }
c8fe2783	177	Int_t GetNumberOfCaloClusters() const { return fCaloClusters->GetEntriesFast(); }
3dd9f9e3	178	AliAODCaloCluster GetCaloCluster(Int_t nCluster) const { return (AliAODCaloCluster)fCaloClusters->UncheckedAt(nCluster); }
	179	Int_t AddCaloCluster(const AliAODCaloCluster* clus)
	180	{new((fCaloClusters)[fCaloClusters->GetEntriesFast()]) AliAODCaloCluster(clus); return fCaloClusters->GetEntriesFast()-1;}
720f7306	181	AliAODCaloTrigger *GetCaloTrigger(TString calo) const
	182	{
	183	if (calo.Contains("EMCAL")) return fEMCALTrigger;
	184	else
	185	return fPHOSTrigger;
	186	}
	187
78902954	188	Int_t GetEMCALClusters(TRefArray *clusters) const;
	189	Int_t GetPHOSClusters(TRefArray *clusters) const;
	190
	191
3dd9f9e3	192	// -- FMD Cluster
	193	TClonesArray *GetFmdClusters() const { return fFmdClusters; }
	194	Int_t GetNFmdClusters() const { return fFmdClusters->GetEntriesFast(); }
	195	AliAODFmdCluster GetFmdCluster(Int_t nCluster) const { return (AliAODFmdCluster)fFmdClusters->UncheckedAt(nCluster); }
	196	Int_t AddFmdCluster(const AliAODFmdCluster* clus)
	197	{new((fFmdClusters)[fFmdClusters->GetEntriesFast()]) AliAODFmdCluster(clus); return fFmdClusters->GetEntriesFast()-1;}
	198
	199	// -- PMD Cluster
	200	TClonesArray *GetPmdClusters() const { return fPmdClusters; }
	201	Int_t GetNPmdClusters() const { return fPmdClusters->GetEntriesFast(); }
	202	AliAODPmdCluster GetPmdCluster(Int_t nCluster) const { return (AliAODPmdCluster)fPmdClusters->UncheckedAt(nCluster); }
	203	Int_t AddPmdCluster(const AliAODPmdCluster* clus)
	204	{new((fPmdClusters)[fPmdClusters->GetEntriesFast()]) AliAODPmdCluster(clus); return fPmdClusters->GetEntriesFast()-1;}
df9db588	205
	206	// -- Jet
	207	TClonesArray *GetJets() const { return fJets; }
5d691d96	208	Int_t GetNJets() const { return fJets?fJets->GetEntriesFast():0; }
5d691d96	209	AliAODJet GetJet(Int_t nJet) const { return fJets?(AliAODJet)fJets->UncheckedAt(nJet):0; }
08e4c457	210	Int_t AddJet(const AliAODJet* vtx)
08e4c457	211	{new((fJets)[fJets->GetEntriesFast()]) AliAODJet(vtx); return fJets->GetEntriesFast()-1;}
df9db588	212
821f8f1d	213	// -- Tracklets
	214	AliAODTracklets *GetTracklets() const { return fTracklets; }
	215
3dd9f9e3	216	// -- Calorimeter Cells
e649177a	217	AliAODCaloCells *GetEMCALCells() const { return fEmcalCells; }
e649177a	218	AliAODCaloCells *GetPHOSCells() const { return fPhosCells; }
c8fe2783	219	const TGeoHMatrix* GetPHOSMatrix(Int_t /i/) const { return NULL; }
	220	const TGeoHMatrix* GetEMCALMatrix(Int_t /i/)const { return NULL; }
	221
3dd9f9e3	222
866d8d78	223	// -- Dimuons
	224	TClonesArray *GetDimuons() const { return fDimuons; }
	225	Int_t GetNDimuons() const { return fDimuons->GetEntriesFast(); }
	226	Int_t GetNumberOfDimuons() const { return GetNDimuons(); }
	227	AliAODDimuon GetDimuon(Int_t nDimu) const { return (AliAODDimuon)fDimuons->UncheckedAt(nDimu); }
	228	Int_t AddDimuon(const AliAODDimuon* dimu)
	229	{new((fDimuons)[fDimuons->GetEntriesFast()]) AliAODDimuon(dimu); return fDimuons->GetEntriesFast()-1;}
	230
df9db588	231	// -- Services
ec4af4c1	232	void CreateStdContent();
08e4c457	233	void SetStdNames();
ec4af4c1	234	void GetStdContent();
5c1dc41f	235	void CreateStdFolders();
3dd9f9e3	236	void ResetStd(Int_t trkArrSize = 0,
	237	Int_t vtxArrSize = 0,
	238	Int_t v0ArrSize = 0,
9a244178	239	Int_t cascadeArrSize = 0,
3dd9f9e3	240	Int_t jetSize = 0,
	241	Int_t caloClusSize = 0,
	242	Int_t fmdClusSize = 0,
866d8d78	243	Int_t pmdClusSize = 0,
866d8d78	244	Int_t dimuonArrsize =0
9a244178	245	);
ec4af4c1	246	void ClearStd();
ec8aded8	247	void Reset() {ClearStd();}
3b93c32a	248	void ReadFromTree(TTree tree, Option_t opt = "");
f12d42ce	249	void WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}
08e4c457	250
08e4c457	251	void Print(Option_t *option="") const;
c8e4f399	252	void MakeEntriesReferencable();
5e6a3170	253	static void AssignIDtoCollection(const TCollection* col);
c8fe2783	254
	255	//Following needed only for mixed event
	256	virtual Int_t EventIndex(Int_t) const {return 0;}
	257	virtual Int_t EventIndexForCaloCluster(Int_t) const {return 0;}
	258	virtual Int_t EventIndexForPHOSCell(Int_t) const {return 0;}
	259	virtual Int_t EventIndexForEMCALCell(Int_t) const {return 0;}
befa9db4	260	AliCentrality* GetCentrality() {return fHeader->GetCentralityP();}
ce7adfe9	261	AliEventplane* GetEventplane() {return fHeader->GetEventplaneP();}
ae2bccf1	262
a1f96974	263	// TZERO
	264	AliAODTZERO *GetTZEROData() const { return fAODTZERO; }
	265
ae2bccf1	266	// VZERO
ae2bccf1	267	AliAODVZERO *GetVZEROData() const { return fAODVZERO; }
5e14e698	268	virtual const Float_t* GetVZEROEqFactors() const {return fHeader?fHeader->GetVZEROEqFactors():0x0;}
	269	virtual Float_t GetVZEROEqMultiplicity(Int_t i) const;
	270	void SetVZEROEqFactors(const Float_t *factors) const {
	271	if(fHeader && factors)
	272	fHeader->SetVZEROEqFactors(factors);}
ae2bccf1	273
16d867eb	274	//ZDC
	275	AliAODZDC *GetZDCData() const { return fAODZDC; }
	276
820b4d9e	277	#ifdef MFT_UPGRADE
	278	// MFT
	279	AliAODMFT *GetMFTData() const { return fAODMFT; }
	280	#endif
	281
c8fe2783	282	private :
df9db588	283
5c1dc41f	284	TList *fAODObjects; // list of AODObjects
5c1dc41f	285	TFolder *fAODFolder; // folder structure of branches
dd389a30	286	Bool_t fConnected; //! flag if leaves are alreday connected
df9db588	287	// standard content
3dd9f9e3	288	AliAODHeader *fHeader; //! event information
	289	TClonesArray *fTracks; //! charged tracks
	290	TClonesArray *fVertices; //! vertices
	291	TClonesArray *fV0s; //! V0s
9a244178	292	TClonesArray *fCascades; //! Cascades
3dd9f9e3	293	AliAODTracklets *fTracklets; //! SPD tracklets
3dd9f9e3	294	TClonesArray *fJets; //! jets
37792174	295	AliAODCaloCells *fEmcalCells; //! EMCAL calorimenter cells
37792174	296	AliAODCaloCells *fPhosCells; //! PHOS calorimenter cells
3dd9f9e3	297	TClonesArray *fCaloClusters; //! calorimeter clusters
720f7306	298	AliAODCaloTrigger *fEMCALTrigger; //! EMCAL Trigger information
720f7306	299	AliAODCaloTrigger *fPHOSTrigger; //! PHOS Trigger information
3dd9f9e3	300	TClonesArray *fFmdClusters; //! FMDclusters
3dd9f9e3	301	TClonesArray *fPmdClusters; //! PMDclusters
866d8d78	302	TClonesArray *fDimuons; //! dimuons
a1f96974	303	AliAODTZERO *fAODTZERO; //! TZERO AOD
ae2bccf1	304	AliAODVZERO *fAODVZERO; //! VZERO AOD
16d867eb	305	AliAODZDC *fAODZDC; //! ZDC AOD
820b4d9e	306	#ifdef MFT_UPGRADE
	307	AliAODMFT *fAODMFT; //! VZERO AOD
	308	#endif
16d867eb	309
08e4c457	310	static const char* fAODListName[kAODListN]; //!
08e4c457	311
a1f96974	312	ClassDef(AliAODEvent,89);
df9db588	313	};
	314
	315	#endif