AliAOD and AODParticle (T.Kuhr) - Readers, AODStdParticle and Cuts (P.Skowronski...

[u/mrichter/AliRoot.git] / ANALYSIS / AliAODRun.cxx

diff --git a/ANALYSIS/AliAODRun.cxx b/ANALYSIS/AliAODRun.cxx
new file mode 100644 (file)
index 0000000..1f96274
--- /dev/null
+++ b/ANALYSIS/AliAODRun.cxx
@@ -0,0 +1,86 @@
+#include "AliAODRun.h"
+//____________________
+///////////////////////////////////////////////////////
+//                                                   //
+// AliAODRun                                         //
+//                                                   //
+// Class storing and managing events                 //
+//                                                   //
+// Piotr.Skowronski@cern.ch                          //
+// http://alisoft.cern.ch/people/skowron/analyzer    //
+//                                                   //
+///////////////////////////////////////////////////////
+
+#include <TObjArray.h>
+
+ClassImp(AliAODRun)
+/**************************************************************************/ 
+
+AliAODRun::AliAODRun()
+{ 
+ //contructor
+  fEvents = new TObjArray();//create array for AliAODs
+  if(!fEvents) Fatal("AliAODRun::AliAODRun","Can not allocate memory");
+  fEvents->SetOwner(); //array is an owner: when is deleted or cleared it deletes objects that it contains
+}
+/**************************************************************************/
+
+AliAODRun::~AliAODRun()
+{
+  //destructor
+  delete fEvents;//delete array with events
+}
+/**************************************************************************/
+
+void AliAODRun::Reset()
+ { 
+   fEvents->Clear();//clear an array with events. 
+                    //All events are deleted because array is an owner (set in constructor)
+ }
+/**************************************************************************/
+
+void AliAODRun::AddParticle(Int_t event, AliAODParticle* part)
+{
+ //Adds particle to event
+ //if there is no event of this number, crate it and add to the collection
+ if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
+ 
+ GetEvent(event)->AddParticle(part);
+}
+/**************************************************************************/
+
+void AliAODRun::AddParticle(Int_t event, TParticle* part, Int_t idx)
+{
+ //if there is no event of this number, crate it and add to the collection
+ if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
+ GetEvent(event)->AddParticle(part,idx);
+}
+/**************************************************************************/
+
+void AliAODRun::AddParticle(Int_t event, Int_t pdg, Int_t idx,
+                            Double_t px, Double_t py, Double_t pz, Double_t etot,
+                            Double_t vx, Double_t vy, Double_t vz, Double_t time)
+{
+ //if there is no event of this number, crate it and add to the collection
+ if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
+ GetEvent(event)->AddParticle(pdg,idx,px,py,pz,etot,vx,vy,vz,time);
+}
+/**************************************************************************/ 
+
+void AliAODRun::SetEvent(Int_t number, AliAOD* event)
+{
+  //adds an event to the run
+  if (event == 0x0)
+   {
+     delete fEvents->RemoveAt(number);
+     return;
+   }
+  AliAOD* ev = GetEvent(number);
+  if (ev == event) return;
+  
+  delete fEvents->RemoveAt(number);
+  fEvents->AddAtAndExpand(event, number);
+  
+}
+/**************************************************************************/ 
+