Updated version of the geometry

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

diff --git a/JETAN/AliJetReader.h b/JETAN/AliJetReader.h
index 1dc85a02a54b9967cec9608a7f6441146dd00d97..58ffc08811c0e30dc02e51e531d3fa4395844631 100755 (executable)
--- a/JETAN/AliJetReader.h
+++ b/JETAN/AliJetReader.h
@@ -4,20 +4,18 @@
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
  
-//---------------------------------------------------------------------
 // Jet reader base class
 // manages the reading of input for jet algorithms
 // Author: jgcn@mda.cinvestav.mx
-//---------------------------------------------------------------------
   
 #include <TObject.h>
 #include <TChain.h>
-
+#include <TArrayI.h>
+class TTree;
 class TClonesArray;
 class AliJetReaderHeader;
 class AliESD;
-class AliHeader;
-
+class AliJet;
 
 class AliJetReader : public TObject 
 {
@@ -29,24 +27,35 @@ class AliJetReader : public TObject
   virtual TClonesArray *GetMomentumArray() {return fMomentumArray;}
   virtual Int_t GetChainEntries() {return fChain->GetEntries();} 
   virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;}
-  virtual AliHeader* GetAliHeader() { return fAliHeader;}
-
+  virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
+  virtual Int_t GetCutFlag(Int_t i) const {return fCutFlag[i];}
+  
   // Setters
-  virtual void FillMomentumArray(Int_t) {}
-  virtual void SetReaderHeader(AliJetReaderHeader* header) {fReaderHeader = header;}
+  virtual Bool_t FillMomentumArray(Int_t) {return kTRUE;}
+  virtual void   SetReaderHeader(AliJetReaderHeader* header) 
+    {fReaderHeader = header;}
          
-  // others
-  virtual void OpenInputFiles() {}
+  // Others
+  virtual void   OpenInputFiles() {}
+  virtual void   ConnectTree(TTree* /*tree*/) {}
+  virtual Bool_t GetGenJets(AliJet* /*genJets*/) {return kFALSE;}
+  
   void ClearArray();
  
  protected:
+  AliJetReader(const AliJetReader& rJetReader);
+  AliJetReader& operator = (const AliJetReader& rhsr);
+
   TChain                  *fChain;         // chain for reconstructed tracks
   TChain                  *fChainMC;       // chain for mc information
   TClonesArray            *fMomentumArray; // array of particle momenta
   TClonesArray            *fArrayMC;       // array of mc particles
   AliESD                  *fESD;           // pointer to esd
   AliJetReaderHeader      *fReaderHeader;  // pointer to header
-  AliHeader               *fAliHeader;     // pointer to event header
+  TArrayI fSignalFlag;   // to flag if a particle comes from pythia or 
+                         // from the underlying event
+  TArrayI fCutFlag;      // to flag if a particle passed the pt cut or not
+
   ClassDef(AliJetReader,1)
 };