#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)
-AliJetMCReader::AliJetMCReader()
+
+AliJetMCReader::AliJetMCReader():
+ AliJetESDReader(),
+ fChainMC(0x0)
{
// Constructor
- fReaderHeader = 0x0;
- fMass = 0;
- fPdgC = 0;
}
//____________________________________________________________________________
AliJetMCReader::~AliJetMCReader()
{
// Destructor
- delete fReaderHeader;
+ delete fChainMC;
}
//____________________________________________________________________________
-void AliJetMCReader::FillMomentumArray(Int_t event)
+Bool_t AliJetMCReader::FillMomentumArray(Int_t event)
{
// Fill momentum array
TClonesArray &arrayMC = *fArrayMC;
if (pt < ptMin) continue; //check cuts
p = part->P();
e = part->Energy();
- fMass = part->GetCalcMass();
- fPdgC = part->GetPdgCode();
// fill momentum array
new ((*fMomentumArray)[goodTrack]) TLorentzVector(p.X(), p.Y(), p.Z(), e);
goodTrack++;
}
printf("\nNumber of good tracks %d \n", goodTrack);
+ return kTRUE;
}