Jet analysis using MC information.

[u/mrichter/AliRoot.git] / JETAN / AliJetKineReader.h

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)
 };