Fix for the problem during PbPb run of Nov 2010 (Indra)

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

diff --git a/JETAN/AliJetKineReader.h b/JETAN/AliJetKineReader.h
index f543ca8bd7f133843117b9a610a7e30e4e6136c0..852551af0efafe05521dd9e8c0b2ef143ed5a909 100644 (file)
--- a/JETAN/AliJetKineReader.h
+++ b/JETAN/AliJetKineReader.h
@@ -11,30 +11,34 @@
 #include "AliJetReader.h"
 
 class AliRunLoader;
+class AliHeader;
+class AliMCEvent;
+class TClonesArray;
+
 
 class AliJetKineReader : public AliJetReader
 {
  public: 
   AliJetKineReader();
   virtual ~AliJetKineReader();
-
-  Bool_t Efficiency(Float_t p, Float_t /*eta*/, Float_t phi);
-  Float_t SmearMomentum(Int_t ind, Float_t p);
-
-  // Getters
-  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();
-
- protected:
-  AliRunLoader           *fRunLoader; // Pointer to the run loader
+  Bool_t  FillMomentumArray();
+  void    OpenInputFiles();
+  void    SetInputEvent(const TObject* esd, const TObject* aod, const TObject* mc);
+  // Fast Simulation
+  Float_t SmearMomentum(Int_t ind, Float_t p);
+  Bool_t  Efficiency(Float_t pt, Float_t eta, Float_t phi);
+  // Others
+  TClonesArray*      GetGeneratedJets();
+  virtual AliHeader* GetAliHeader() const {return fAliHeader;}
   
-  Float_t fMass;                      // Particle mass
-  Int_t   fPdgC;                      // Pdg code
- 
+ protected:
+  AliJetKineReader(const AliJetKineReader& rJetKine);
+  AliJetKineReader& operator = (const AliJetKineReader& rkr);
+
+  AliHeader          *fAliHeader;       //! Header
+  AliMCEvent         *fMCEvent;  //! Monte Carlo Event Handler
+  TClonesArray       *fGenJets;         //! List of generated jets
   ClassDef(AliJetKineReader,1)
 };