ANALYSIS/AliAODRun.cxx

   1 #include "AliAODRun.h"
   2 //____________________
   3 ///////////////////////////////////////////////////////
   4 //                                                   //
   5 // AliAODRun                                         //
   6 //                                                   //
   7 // Class storing and managing events                 //
   8 //                                                   //
   9 // Piotr.Skowronski@cern.ch                          //
  10 // http://alisoft.cern.ch/people/skowron/analyzer    //
  11 //                                                   //
  12 ///////////////////////////////////////////////////////
  13
  14 #include <TObjArray.h>
  15
  16 ClassImp(AliAODRun)
  17 /**************************************************************************/
  18
  19 AliAODRun::AliAODRun()
  20 {
  21  //contructor
  22   fEvents = new TObjArray();//create array for AliAODs
  23   if(!fEvents) Fatal("AliAODRun::AliAODRun","Can not allocate memory");
  24   fEvents->SetOwner(); //array is an owner: when is deleted or cleared it deletes objects that it contains
  25 }
  26 /**************************************************************************/
  27
  28 AliAODRun::~AliAODRun()
  29 {
  30   //destructor
  31   delete fEvents;//delete array with events
  32 }
  33 /**************************************************************************/
  34
  35 void AliAODRun::Reset()
  36  {
  37    fEvents->Clear();//clear an array with events.
  38                     //All events are deleted because array is an owner (set in constructor)
  39  }
  40 /**************************************************************************/
  41
  42 void AliAODRun::AddParticle(Int_t event, AliAODParticle* part)
  43 {
  44  //Adds particle to event
  45  //if there is no event of this number, crate it and add to the collection
  46  if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
  47
  48  GetEvent(event)->AddParticle(part);
  49 }
  50 /**************************************************************************/
  51
  52 void AliAODRun::AddParticle(Int_t event, TParticle* part, Int_t idx)
  53 {
  54  //if there is no event of this number, crate it and add to the collection
  55  if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
  56  GetEvent(event)->AddParticle(part,idx);
  57 }
  58 /**************************************************************************/
  59
  60 void AliAODRun::AddParticle(Int_t event, Int_t pdg, Int_t idx,
  61                             Double_t px, Double_t py, Double_t pz, Double_t etot,
  62                             Double_t vx, Double_t vy, Double_t vz, Double_t time)
  63 {
  64  //if there is no event of this number, crate it and add to the collection
  65  if(!GetEvent(event))  fEvents->AddAtAndExpand(new AliAOD, event);
  66  GetEvent(event)->AddParticle(pdg,idx,px,py,pz,etot,vx,vy,vz,time);
  67 }
  68 /**************************************************************************/
  69
  70 void AliAODRun::SetEvent(Int_t number, AliAOD* event)
  71 {
  72   //adds an event to the run
  73   if (event == 0x0)
  74    {
  75      delete fEvents->RemoveAt(number);
  76      return;
  77    }
  78   AliAOD* ev = GetEvent(number);
  79   if (ev == event) return;
  80
  81   delete fEvents->RemoveAt(number);
  82   fEvents->AddAtAndExpand(event, number);
  83
  84 }
  85 /**************************************************************************/
  86