Create the rec-point branch even in the case of no digits. Please review and fix...

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

diff --git a/JETAN/AliJetKineReader.h b/JETAN/AliJetKineReader.h
index 364fbb35398f5ae856e173e5f6a91c597f4e9b65..df7ce1c0a35724c0b61d486257c9d5dac07c647d 100644 (file)
--- a/JETAN/AliJetKineReader.h
+++ b/JETAN/AliJetKineReader.h
@@ -12,6 +12,9 @@
 
 class AliRunLoader;
 class AliHeader;
+class AliMCEvent;
+class TClonesArray;
+
 
 class AliJetKineReader : public AliJetReader
 {
@@ -19,21 +22,23 @@ class AliJetKineReader : public AliJetReader
   AliJetKineReader();
   virtual ~AliJetKineReader();
   // Setters
-  Bool_t  FillMomentumArray(Int_t event);
+  Bool_t  FillMomentumArray();
   void    OpenInputFiles();
+  void    SetInputEvent(TObject* esd, TObject* aod, 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
-  virtual Bool_t GetGenJets(AliJet* /*genJets*/);
-  virtual AliHeader* GetAliHeader() {return fAliHeader;}
+  TClonesArray*      GetGeneratedJets();
+  virtual AliHeader* GetAliHeader() const {return fAliHeader;}
   
  protected:
   AliJetKineReader(const AliJetKineReader& rJetKine);
   AliJetKineReader& operator = (const AliJetKineReader& rkr);
 
-  AliRunLoader *fRunLoader;       //! Pointer to the run loader
-  AliHeader    *fAliHeader;       //! Header
+  AliHeader          *fAliHeader;       //! Header
+  AliMCEvent         *fMCEvent;  //! Monte Carlo Event Handler
+  TClonesArray       *fGenJets;         //! List of generated jets
   ClassDef(AliJetKineReader,1)
 };