First commit of new jet reconstruction and analysis code to be used for the

[u/mrichter/AliRoot.git] / JETAN / AliJetMCReader.cxx

diff --git a/JETAN/AliJetMCReader.cxx b/JETAN/AliJetMCReader.cxx
new file mode 100644 (file)
index 0000000..173170c
--- /dev/null
+++ b/JETAN/AliJetMCReader.cxx
@@ -0,0 +1,97 @@
+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ *                                                                        *
+ * Author: The ALICE Off-line Project.                                    *
+ * Contributors are mentioned in the code where appropriate.              *
+ *                                                                        *
+ * Permission to use, copy, modify and distribute this software and its   *
+ * documentation strictly for non-commercial purposes is hereby granted   *
+ * without fee, provided that the above copyright notice appears in all   *
+ * copies and that both the copyright notice and this permission notice   *
+ * appear in the supporting documentation. The authors make no claims     *
+ * about the suitability of this software for any purpose. It is          *
+ * provided "as is" without express or implied warranty.                  *
+ **************************************************************************/
+ 
+// Jet MC Reader 
+// MC reader for jet analysis
+// Author: Mercedes Lopez Noriega (mercedes.lopez.noriega@cern.ch)
+
+// From root ...
+#include <TClonesArray.h>
+#include <TPDGCode.h>
+#include <TParticle.h>
+#include <TParticlePDG.h>
+#include <TVector3.h>
+#include <TLorentzVector.h>
+#include <TSystem.h>
+// From AliRoot ...
+#include "AliJetMCReader.h"
+#include "AliJetMCReaderHeader.h"
+#include "AliESD.h"
+#include "AliESDtrack.h"
+
+ClassImp(AliJetMCReader)
+
+AliJetMCReader::AliJetMCReader()
+{
+  // Constructor
+  fReaderHeader = 0x0;
+  fMass = 0;
+  fPdgC = 0;
+}
+
+//____________________________________________________________________________
+
+AliJetMCReader::~AliJetMCReader()
+{
+  // Destructor
+  delete fReaderHeader;
+}
+
+//____________________________________________________________________________
+
+
+void AliJetMCReader::FillMomentumArray(Int_t event)
+{
+// Fill momentum array
+  TClonesArray &arrayMC = *fArrayMC;
+  Int_t goodTrack = 0;
+  Int_t nt = 0;
+  Float_t pt, e;
+  TVector3 p;
+
+  // clear array
+  ClearArray();
+  // get event from chains
+  fChain->GetEntry(event);
+  fChainMC->GetEntry(event);
+  // get number of tracks in event (for the loop)
+  nt = fESD->GetNumberOfTracks();
+
+  // get cuts set by user
+  Double_t ptMin = ((AliJetMCReaderHeader*) fReaderHeader)->GetPtCut();
+
+  //loop over particles
+  for (Int_t it = 0; it < nt; it++) {
+    AliESDtrack *track = fESD->GetTrack(it); //track
+    UInt_t status = track->GetStatus();
+    if ((status & AliESDtrack::kITSrefit) == 0) continue; // quality check
+    //    track->GetImpactParameters(dca,z);
+    // if (dca > dcaMax) continue; // check track is reasonable 
+    Int_t label = TMath::Abs(track->GetLabel());
+    TParticle *part = (TParticle*)arrayMC[label]; //particle
+    pt = part->Pt(); // pt of the particle
+    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);
+}
+
+