Added parsing for p-A or A-p beams. Same rigidity is assumed

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

diff --git a/JETAN/AliJetKineReader.h b/JETAN/AliJetKineReader.h
index 1895bb14ed6424cc06b380619a3c0f9ec75664dd..852551af0efafe05521dd9e8c0b2ef143ed5a909 100644 (file)
--- a/JETAN/AliJetKineReader.h
+++ b/JETAN/AliJetKineReader.h
@@ -11,32 +11,34 @@
 #include "AliJetReader.h"
 
 class AliRunLoader;
+class AliHeader;
+class AliMCEvent;
+class TClonesArray;
+
 
 class AliJetKineReader : public AliJetReader
 {
  public: 
   AliJetKineReader();
   virtual ~AliJetKineReader();
-
-  // 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();
+  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;}
+  
  protected:
   AliJetKineReader(const AliJetKineReader& rJetKine);
   AliJetKineReader& operator = (const AliJetKineReader& rkr);
 
-  AliRunLoader *fRunLoader;       // Pointer to the run loader
-  
-  Float_t fMass;                  // Particle mass
-  Int_t   fPdgC;                  // Pdg code
- 
+  AliHeader          *fAliHeader;       //! Header
+  AliMCEvent         *fMCEvent;  //! Monte Carlo Event Handler
+  TClonesArray       *fGenJets;         //! List of generated jets
   ClassDef(AliJetKineReader,1)
 };