Coverity 16571

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

diff --git a/STEER/AliAODEvent.h b/STEER/AliAODEvent.h
index bbafd3339dcf2bcecd20c3f71f0329926f1f558f..6c5aa4af96241d4618495e9792b98cd89e3c113e 100644 (file)
--- a/STEER/AliAODEvent.h
+++ b/STEER/AliAODEvent.h
@@ -30,9 +30,12 @@
 #include "AliAODPmdCluster.h"
 #include "AliAODFmdCluster.h"
 #include "AliAODDimuon.h"
+#include "AliAODVZERO.h"
 
 class TTree;
 class TFolder;
+class AliCentrality;
+class AliEventplane;
 
 class AliAODEvent : public AliVEvent {
 
@@ -50,6 +53,7 @@ class AliAODEvent : public AliVEvent {
                       kAODFmdClusters,
                       kAODPmdClusters,
                       kAODDimuons,
+                      kAODVZERO,
                       kAODListN
   };
 
@@ -63,6 +67,8 @@ class AliAODEvent : public AliVEvent {
   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; }
@@ -111,9 +117,9 @@ class AliAODEvent : public AliVEvent {
 
   // -- Tracks
   TClonesArray *GetTracks()              const { return fTracks; }
-  Int_t         GetNTracks()             const { return fTracks->GetEntriesFast(); }
+  Int_t         GetNTracks()             const { return fTracks? fTracks->GetEntriesFast() : 0; }
   Int_t         GetNumberOfTracks()      const { return GetNTracks(); }
-  AliAODTrack  *GetTrack(Int_t nTrack)   const { return (AliAODTrack*)fTracks->UncheckedAt(nTrack); }
+  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;
@@ -121,8 +127,8 @@ class AliAODEvent : public AliVEvent {
 
   // -- Vertex
   TClonesArray *GetVertices()            const { return fVertices; }
-  Int_t         GetNumberOfVertices()    const { return fVertices->GetEntriesFast(); }
-  AliAODVertex *GetVertex(Int_t nVertex) const { return (AliAODVertex*)fVertices->At(nVertex); }
+  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;}
   
@@ -180,8 +186,8 @@ class AliAODEvent : public AliVEvent {
 
   // -- 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         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;}
 
@@ -225,14 +231,19 @@ class AliAODEvent : public AliVEvent {
 
   void  Print(Option_t *option="") const;
   void  MakeEntriesReferencable();
-  static void AssignIDtoCollection(TCollection* col);
+  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();}
+
+  // VZERO 
+  AliAODVZERO *GetVZEROData() const { return fAODVZERO; }
+
   private :
 
   TList   *fAODObjects; //  list of AODObjects
@@ -252,10 +263,11 @@ class AliAODEvent : public AliVEvent {
   TClonesArray    *fFmdClusters;  //! FMDclusters
   TClonesArray    *fPmdClusters;  //! PMDclusters
   TClonesArray    *fDimuons;      //! dimuons
+  AliAODVZERO     *fAODVZERO;     //! VZERO AOD
 
   static const char* fAODListName[kAODListN]; //!
 
-  ClassDef(AliAODEvent, 6);
+  ClassDef(AliAODEvent, 7);
 };
 
 #endif