JETAN/AliJetParticlesReaderKine.cxx

   1 // $Id$
   2
   3 //_______________________________________________________________________
   4 /////////////////////////////////////////////////////////////////////////
   5 //
   6 // class AliJetParticlesReaderKine
   7 //
   8 // Reader for Kinematics
   9 //
  10 // loizides@ikf.uni-frankfurt.de
  11 //
  12 /////////////////////////////////////////////////////////////////////////
  13
  14 #include <Riostream.h>
  15 #include <TString.h>
  16 #include <TParticle.h>
  17 #include <AliRunLoader.h>
  18 #include <AliStack.h>
  19
  20 #include "AliJetParticle.h"
  21 #include "AliJetEventParticles.h"
  22 #include "AliJetParticlesReaderKine.h"
  23
  24 ClassImp(AliJetParticlesReaderKine)
  25
  26
  27 /**********************************************************/
  28
  29 AliJetParticlesReaderKine::AliJetParticlesReaderKine() :
  30   AliJetParticlesReader(),
  31   fFileName("galice.root"),
  32   fRunLoader(0),
  33   fNeutral(kTRUE),fCharged(kTRUE),fEM(kTRUE),
  34   fUseTracks(kFALSE)
  35 {
  36   //constructor
  37 }
  38
  39 /**********************************************************/
  40
  41
  42 AliJetParticlesReaderKine::AliJetParticlesReaderKine(TString& fname) :
  43   AliJetParticlesReader(),
  44   fFileName(fname),
  45   fRunLoader(0),
  46   fNeutral(kTRUE),fCharged(kTRUE),fEM(kTRUE),
  47   fUseTracks(kFALSE)
  48 {
  49   //constructor
  50 }
  51
  52 /**********************************************************/
  53
  54 AliJetParticlesReaderKine::AliJetParticlesReaderKine(TObjArray* dirs, const Char_t *filename):
  55   AliJetParticlesReader(dirs),
  56   fFileName(filename),
  57   fRunLoader(0),
  58   fNeutral(kTRUE),fCharged(kTRUE),fEM(kTRUE),
  59   fUseTracks(kFALSE)
  60 {
  61   //constructor
  62 }
  63
  64 /**********************************************************/
  65
  66 AliJetParticlesReaderKine::~AliJetParticlesReaderKine()
  67 {
  68   //destructor
  69   if(fRunLoader) delete fRunLoader;
  70 }
  71
  72 /**********************************************************/
  73
  74 void AliJetParticlesReaderKine::Rewind()
  75 {
  76   //Rewinds to the beginning
  77   if(fRunLoader) delete fRunLoader;
  78   fRunLoader = 0;
  79   fCurrentDir = 0;
  80   fNEventsRead= 0;
  81 }
  82
  83 /**********************************************************/
  84
  85 Int_t AliJetParticlesReaderKine::ReadNext()
  86 {
  87   //Reads Kinematics Tree
  88
  89   if((!fOwner) || (fEventParticles == 0))
  90     fEventParticles = new AliJetEventParticles();
  91
  92   while(fCurrentDir < GetNumberOfDirs())
  93     {
  94       if (!OpenFile(fCurrentDir))
  95       {
  96         fCurrentDir++;
  97         continue;
  98       }
  99
 100       if (fCurrentEvent == fRunLoader->GetNumberOfEvents())
 101         {
 102           //read next directory
 103           delete fRunLoader; //close current session
 104           fRunLoader = 0;    //assure pointer is null
 105           fCurrentDir++;     //go to next dir
 106           continue;
 107         }
 108
 109       //Info("ReadNext","Reading Event %d",fCurrentEvent);
 110
 111       fRunLoader->GetEvent(fCurrentEvent);
 112       AliStack* stack = fRunLoader->Stack();
 113       if (!stack)
 114         {
 115           Error("ReadNext","Can not get stack for event %d",fCurrentEvent);
 116           continue;
 117         }
 118
 119       //get vertex
 120       const TParticle *kv = stack->Particle(0);
 121       if(kv) {
 122         fEventParticles->SetVertex(kv->Vx(),kv->Vy(),kv->Vz());
 123       }
 124
 125       Int_t npart = stack->GetNprimary();
 126       if(fUseTracks)
 127         npart = stack->GetNtrack();
 128
 129       fEventParticles->Reset(npart);
 130       for (Int_t i = 0;i<npart; i++)
 131         {
 132           TParticle *p = stack->Particle(i);
 133           if(!p) continue;
 134           Int_t child1 = p->GetFirstDaughter();
 135           //Int_t child2 = p->GetLastDaughter();
 136           //Int_t mother = p->GetFirstMother();
 137           if((child1>=0) && (child1<npart)) continue;
 138           //cout << child1 << " " << child2 << " " << mother << endl;
 139
 140           if(IsAcceptedParticle(p)) //put particle in event
 141             fEventParticles->AddParticle(p,i);
 142         }
 143       fCurrentEvent++;
 144       fCurrentDir++;
 145       fNEventsRead++;
 146       return kTRUE;
 147     }
 148       //end of loop over directories specified in fDirs Obj Array
 149   return kFALSE;
 150 }
 151
 152 /**********************************************************/
 153
 154 Int_t AliJetParticlesReaderKine::OpenFile(Int_t n)
 155 {
 156   //opens file with kine tree
 157
 158   const TString& dirname = GetDirName(n);
 159   if (dirname == "")
 160     {
 161       Error("OpenNextFile","Can not get directory name with index %d",n);
 162       return kFALSE;
 163     }
 164
 165   TString filename = dirname +"/"+ fFileName;
 166   if(fRunLoader) delete fRunLoader;
 167   fRunLoader = AliRunLoader::Open(filename.Data());
 168
 169   if (fRunLoader == 0)
 170     {
 171       Error("OpenNextFile","Can't open session from file %s",filename.Data());
 172       return kFALSE;
 173     }
 174
 175   if (fRunLoader->GetNumberOfEvents() <= 0)
 176     {
 177       Error("OpenNextFile","There is no events in this directory.");
 178       delete fRunLoader;
 179       fRunLoader = 0;
 180       return kFALSE;
 181     }
 182
 183   if (fRunLoader->LoadKinematics())
 184     {
 185       Error("OpenNextFile","Error occured while loading kinematics.");
 186       return kFALSE;
 187     }
 188
 189   fCurrentEvent = 0;
 190   return kTRUE;
 191 }