updated

[u/mrichter/AliRoot.git] / EVGEN / AliGenReaderTreeK.cxx

diff --git a/EVGEN/AliGenReaderTreeK.cxx b/EVGEN/AliGenReaderTreeK.cxx
index 035a89981e6e9780fbd211fcafb95c81b0470b0c..62614be4b2f845d85cef2d29c906e060fbef4c0e 100644 (file)
--- a/EVGEN/AliGenReaderTreeK.cxx
+++ b/EVGEN/AliGenReaderTreeK.cxx
@@ -30,9 +30,10 @@
 #include "AliGenReaderTreeK.h"
 #include "AliHeader.h"
 #include "AliRun.h"
+#include "AliStack.h"
 #include "AliRunLoader.h"
 
-ClassImp(AliGenReaderTreeK);
+ClassImp(AliGenReaderTreeK)
 
 const TString AliGenReaderTreeK::fgkEventFolderName("GenReaderTreeK");
 
@@ -79,7 +80,7 @@ void AliGenReaderTreeK::Init()
 // Initialization
 // Connect base file and file to read from
 
-    TTree *ali = gAlice->TreeE();
+    TTree *ali = AliRunLoader::Instance()->TreeE();
     if (ali) {
       fBaseFile = ali->GetCurrentFile();
     } else {
@@ -215,3 +216,9 @@ void AliGenReaderTreeK::AddDir(const char* dirname)
   TObjString *odir= new TObjString(dirname);
   fDirs->Add(odir);
 }
+
+TParticle* AliGenReaderTreeK::GetParticle(Int_t i)
+ {
+  if (fStack && i<fNp) return fStack->Particle(i);
+  return 0x0;
+ }