#101318: Patch for various problems in AliROOT
[u/mrichter/AliRoot.git] / JETAN / AliJetMCReader.cxx
index 7f2c642..98277a6 100644 (file)
 #include <TVector3.h>
 #include <TLorentzVector.h>
 #include <TSystem.h>
+#include <TChain.h>
 // From AliRoot ...
 #include "AliJetMCReader.h"
 #include "AliJetMCReaderHeader.h"
-#include "AliESD.h"
+#include "AliESDEvent.h"
 #include "AliESDtrack.h"
 
-ClassImp(AliJetMCReader);
+ClassImp(AliJetMCReader)
 
 
 AliJetMCReader::AliJetMCReader():
@@ -83,7 +84,7 @@ Bool_t AliJetMCReader::FillMomentumArray(Int_t event)
     TParticle *part = (TParticle*)arrayMC[label]; //particle
     pt = part->Pt(); // pt of the particle
     if (pt < ptMin) continue; //check  cuts 
-    p = part->P();
+    p.SetXYZ(part->Px(), part->Py(), part->Pz());
     e = part->Energy();
    // fill momentum array
     new ((*fMomentumArray)[goodTrack]) TLorentzVector(p.X(), p.Y(), p.Z(), e);