Jet analysis using MC information.

diff --git a/JETAN/AliJetESDmcReader.cxx b/JETAN/AliJetESDmcReader.cxx
index dc51b3e21cb3e5f6f61df0468fffcf00867821f0..ad10e5da9e0c994dd1f470abf9d57cb36e80c4a7 100644 (file)
--- a/JETAN/AliJetESDmcReader.cxx
+++ b/JETAN/AliJetESDmcReader.cxx
@@ -29,12 +29,14 @@
 #include "AliJetESDReaderHeader.h"
 #include "AliESD.h"
 #include "AliESDtrack.h"
+#include "AliHeader.h"
 
 ClassImp(AliJetESDmcReader)
 
 AliJetESDmcReader::AliJetESDmcReader():
-  fMass(0),
-  fSign(0)
+    fAliHeader(0),
+    fMass(0),
+    fSign(0)
 {
   // Constructor    
   fReaderHeader = 0x0;
@@ -97,7 +99,7 @@ void AliJetESDmcReader::OpenInputFiles()
 
 //____________________________________________________________________________
 
-void AliJetESDmcReader::FillMomentumArray(Int_t event)
+Bool_t AliJetESDmcReader::FillMomentumArray(Int_t event)
 {
   // Fill momentum array
 
@@ -166,7 +168,8 @@ void AliJetESDmcReader::FillMomentumArray(Int_t event)
   // set the signal flags
   fSignalFlag.Set(goodTrack,sflag);
   fCutFlag.Set(goodTrack,cflag);
-
+  return kTRUE;
+  
 }
 
 

diff --git a/JETAN/AliJetESDmcReader.h b/JETAN/AliJetESDmcReader.h
index 5fc1f4970e12d46bac428d721b0e32adc32af295..83662dfb0df54f2ec48247340f9912aa6b4e7225 100644 (file)
--- a/JETAN/AliJetESDmcReader.h
+++ b/JETAN/AliJetESDmcReader.h
@@ -10,6 +10,8 @@
 
 #include "AliJetReader.h"
 class AliJetESDReaderHeader;
+class AliHeader;
+
 
 
 class AliJetESDmcReader : public AliJetReader
@@ -23,12 +25,13 @@ class AliJetESDmcReader : public AliJetReader
   Int_t   GetTrackSign() const {return fSign;}  // returns sign of the track
 
   // Setters
-  void FillMomentumArray(Int_t event); 
-  void OpenInputFiles();
+  Bool_t FillMomentumArray(Int_t event); 
+  void   OpenInputFiles();
    
  protected:
-  Float_t fMass;    // Particle mass
-  Int_t   fSign;    // Particle sign
+  AliHeader* fAliHeader; //! Event header
+  Float_t    fMass;      //! Particle mass
+  Int_t      fSign;      //! Particle sign
 
   ClassDef(AliJetESDmcReader,1)
 };

diff --git a/JETAN/AliJetKineReader.cxx b/JETAN/AliJetKineReader.cxx
index 6f65b2ad7d47cd465859e2793b97c3a1ec95bcd6..7797e8eb81b1f0e12857ab0dac3f6d5459e076a7 100644 (file)
--- a/JETAN/AliJetKineReader.cxx
+++ b/JETAN/AliJetKineReader.cxx
@@ -30,15 +30,19 @@
 #include <TDatabasePDG.h>
 #include <TRandom.h>
 // From AliRoot ...
+#include "AliJet.h"
 #include "AliJetKineReaderHeader.h"
 #include "AliJetKineReader.h"
 #include "AliRunLoader.h"
 #include "AliStack.h"
+#include "AliHeader.h"
+#include "AliGenPythiaEventHeader.h"
 
 ClassImp(AliJetKineReader)
 
 AliJetKineReader::AliJetKineReader():
   fRunLoader(0x0),
+  fAliHeader(0x0),
   fMass(0),
   fPdgC(0)
 {
@@ -52,6 +56,7 @@ AliJetKineReader::~AliJetKineReader()
 {
   // Destructor
   delete fReaderHeader;
+  delete fAliHeader;
 }
 
 //____________________________________________________________________________
@@ -81,7 +86,7 @@ void AliJetKineReader::OpenInputFiles()
 
 //____________________________________________________________________________
 
-void AliJetKineReader::FillMomentumArray(Int_t event)
+Bool_t AliJetKineReader::FillMomentumArray(Int_t event)
 {
   // Fill momentum array for event
  
@@ -175,7 +180,7 @@ void AliJetKineReader::FillMomentumArray(Int_t event)
   fSignalFlag.Set(goodTrack,sflag);
   fCutFlag.Set(goodTrack,cflag);
   printf("\nIn event %d, number of good tracks %d \n", event, goodTrack);
-
+  return kTRUE;
 }
 
 
@@ -235,3 +240,25 @@ Bool_t AliJetKineReader::Efficiency(Float_t p, Float_t /*eta*/, Float_t phi)
 
 }
 
+Bool_t AliJetKineReader::GetGenJets(AliJet* genJets)
+{
+    // Get generated jets from mc header
+    AliHeader* alih = GetAliHeader(); 
+    if (alih == 0) return kFALSE;
+    AliGenEventHeader * genh = alih->GenEventHeader();
+    if (genh == 0) return kFALSE;
+    Int_t nj =((AliGenPythiaEventHeader*)genh)->NTriggerJets(); 
+    Int_t* m = new Int_t[nj];
+    Int_t* k = new Int_t[nj];
+    for (Int_t i=0; i< nj; i++) {
+       Float_t p[4];
+       ((AliGenPythiaEventHeader*)genh)->TriggerJet(i,p);
+       genJets->AddJet(p[0],p[1],p[2],p[3]);
+       m[i]=1;
+       k[i]=i;
+    }
+    genJets->SetNinput(nj);
+    genJets->SetMultiplicities(m);
+    genJets->SetInJet(k);
+    return kTRUE;
+}

diff --git a/JETAN/AliJetKineReader.h b/JETAN/AliJetKineReader.h
index 1895bb14ed6424cc06b380619a3c0f9ec75664dd..b562d620025cca30a4f69a0755a6bd136d4eff13 100644 (file)
--- a/JETAN/AliJetKineReader.h
+++ b/JETAN/AliJetKineReader.h
@@ -11,6 +11,7 @@
 #include "AliJetReader.h"
 
 class AliRunLoader;
+class AliHeader;
 
 class AliJetKineReader : public AliJetReader
 {
@@ -22,20 +23,24 @@ class AliJetKineReader : public AliJetReader
   Float_t GetParticleMass() const {return fMass;}        // returns mass of the Track
   Int_t   GetParticlePdgCode() const {return fPdgC;}     // returns Pdg code
   // Setters
-  void FillMomentumArray(Int_t event);
-  void OpenInputFiles();
+  Bool_t  FillMomentumArray(Int_t event);
+  void    OpenInputFiles();
   // Fast Simulation
   Float_t SmearMomentum(Int_t ind, Float_t p);
   Bool_t  Efficiency(Float_t pt, Float_t eta, Float_t phi);
-
+  // Others
+  virtual Bool_t GetGenJets(AliJet* /*genJets*/);
+  virtual AliHeader* GetAliHeader() {return fAliHeader;}
+  
  protected:
   AliJetKineReader(const AliJetKineReader& rJetKine);
   AliJetKineReader& operator = (const AliJetKineReader& rkr);
 
-  AliRunLoader *fRunLoader;       // Pointer to the run loader
+  AliRunLoader *fRunLoader;       //! Pointer to the run loader
+  AliHeader    *fAliHeader;       //! Header
   
-  Float_t fMass;                  // Particle mass
-  Int_t   fPdgC;                  // Pdg code
+  Float_t fMass;                  //!Particle mass
+  Int_t   fPdgC;                  //!Pdg code
  
   ClassDef(AliJetKineReader,1)
 };

diff --git a/JETAN/AliJetMCReader.cxx b/JETAN/AliJetMCReader.cxx
index 173170c8587127826bfd9c2589e3ceae5b0fde70..ae1cc235b4abf5d2d46ac26daccdc6e5211e0001 100644 (file)
--- a/JETAN/AliJetMCReader.cxx
+++ b/JETAN/AliJetMCReader.cxx
@@ -52,7 +52,7 @@ AliJetMCReader::~AliJetMCReader()
 //____________________________________________________________________________
 
 
-void AliJetMCReader::FillMomentumArray(Int_t event)
+Bool_t AliJetMCReader::FillMomentumArray(Int_t event)
 {
 // Fill momentum array
   TClonesArray &arrayMC = *fArrayMC;
@@ -92,6 +92,7 @@ void AliJetMCReader::FillMomentumArray(Int_t event)
     goodTrack++;
   }
   printf("\nNumber of good tracks %d \n", goodTrack);
+  return kTRUE;
 }
 
 

diff --git a/JETAN/AliJetMCReader.h b/JETAN/AliJetMCReader.h
index 670d7fccd1ade28444ffc16546942d8f5e808017..9c754ffb119e8f15c6ef282b755c57cc2c1b8361 100644 (file)
--- a/JETAN/AliJetMCReader.h
+++ b/JETAN/AliJetMCReader.h
@@ -15,7 +15,7 @@ class AliJetMCReader : public AliJetESDReader
  public: 
     AliJetMCReader();
     virtual ~AliJetMCReader();
-    void FillMomentumArray(Int_t event);
+    Bool_t FillMomentumArray(Int_t event);
     
  protected:
     Float_t fPdgC;   // Pdg code

diff --git a/JETAN/AliJetMCReaderHeader.cxx b/JETAN/AliJetMCReaderHeader.cxx
index 0c149d11d3c370b637bf5e97de477a1c88377d7f..c0a3801a4d7a76facdd62256add76522172f52d6 100644 (file)
--- a/JETAN/AliJetMCReaderHeader.cxx
+++ b/JETAN/AliJetMCReaderHeader.cxx
@@ -28,7 +28,7 @@ AliJetMCReaderHeader::AliJetMCReaderHeader():
  AliJetReaderHeader("AliJetMCReaderHeader") 
 {
   // Constructor
-  SetPtCut();
+  fDCA = 3.;
 }
 
 //____________________________________________________________________________

diff --git a/JETAN/AliJetMCReaderHeader.h b/JETAN/AliJetMCReaderHeader.h
index e611872f0b9b93c72f945f06e60d079fd2eebea8..d38ab4e4a0a11370a236272705db7f3cdb51fc6a 100644 (file)
--- a/JETAN/AliJetMCReaderHeader.h
+++ b/JETAN/AliJetMCReaderHeader.h
@@ -16,21 +16,7 @@ class AliJetMCReaderHeader : public AliJetReaderHeader
  public:
   AliJetMCReaderHeader();
   virtual ~AliJetMCReaderHeader();
-  
-  // Getters
-  Double_t GetPtCut() const {return fPtCut;}
-  Float_t GetDCA() const {return fDCA;} // not working so far..(always 0)
-  
-  // Setters
-  virtual void SetPtCut(Double_t par=2.0) {fPtCut = par;}
-  virtual void SetDCA(Float_t dca = 3.0) {fDCA = dca;} // this is for the track!!! (recommended by P.H.)
-  
  protected:
-  //parameters set by user
-  Double_t fPtCut;
-  Float_t fDCA;
-
-
   ClassDef(AliJetMCReaderHeader,1);
 };