X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=JETAN%2FAliJetReader.h;h=c74f08a9e2af86b65054bdcd0ef193213edb54de;hb=b8187de48ce76b1cacbdbb128e583e7c34d98fbf;hp=69f17473415939c6dbcc7d4cd6210b7380f1eb80;hpb=1b7d5d7e12bdaf252e04ac315f09a741cfc3cff2;p=u%2Fmrichter%2FAliRoot.git

diff --git a/JETAN/AliJetReader.h b/JETAN/AliJetReader.h
index 69f17473415..c74f08a9e2a 100755
--- a/JETAN/AliJetReader.h
+++ b/JETAN/AliJetReader.h
@@ -6,16 +6,25 @@
  
 // Jet reader base class
 // manages the reading of input for jet algorithms
-// Author: jgcn@mda.cinvestav.mx
-  
+// Authors: jgcn@mda.cinvestav.mx
+//          Magali Estienne <magali.estienne@subatech.in2p3.fr>  
+
 #include <TObject.h>
 #include <TChain.h>
 #include <TArrayI.h>
 
+class TTree;
+class TChain;
+class TTask;
 class TClonesArray;
+class TRefArray;
 class AliJetReaderHeader;
-class AliESD;
+class AliESDEvent;
 class AliHeader;
+class AliJetUnitArray;
+class AliJetHadronCorrection;
+class AliJet;
+class AliJetFillUnitArray;
 
 class AliJetReader : public TObject 
 {
@@ -24,36 +33,69 @@ class AliJetReader : public TObject
   virtual ~AliJetReader();
 
   // Getters
-  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];}
-
+  virtual TClonesArray*       GetMomentumArray()     const {return fMomentumArray;}
+  virtual TRefArray*          GetReferences()        const {return 0;}   
+  virtual TClonesArray        *GetUnitArray() const {return fUnitArray;}  
+  virtual AliJetReaderHeader* GetReaderHeader()      const {return fReaderHeader;}
+  virtual AliHeader           *GetAliHeader() const  {return fAliHeader;}
+  virtual Int_t               GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
+  virtual Int_t               GetCutFlag(Int_t i)    const {return fCutFlag[i];}
+  virtual Int_t               GetArrayInitialised()  const {return fArrayInitialised;}
+  virtual Int_t               GetNumCandidate()      const {return fNumCandidate;} 
+  virtual Int_t               GetNumCandidateCut()   const {return fNumCandidateCut;}
+  
   // Setters
-  virtual void FillMomentumArray(Int_t) {}
-  virtual void SetReaderHeader(AliJetReaderHeader* header) 
-    {fReaderHeader = header;}
-	  
-  // others
-  virtual void OpenInputFiles() {}
+  virtual Bool_t FillMomentumArray() {return kTRUE;}
+  virtual Bool_t ReadEventLoader(Int_t) {return kTRUE;}
+  virtual void   InitUnitArray() {}
+  virtual void   InitParameters() {}
+  virtual void   CreateTasks(TChain* /*tree*/) {}
+  virtual Bool_t ExecTasks(Bool_t /*procid*/, TRefArray* /*refArray*/) {return kFALSE;}
+  // Correction of hadronic energy 
+  virtual void   SetHadronCorrector(AliJetHadronCorrection*) {;} 
+  virtual void   SetApplyMIPCorrection(Bool_t /*val*/){;}
+  virtual void   SetApplyFractionHadronicCorrection(Bool_t /*val*/){;}
+  virtual void   SetFractionHadronicCorrection(Double_t /*val*/){;}
+  virtual void   SetApplyElectronCorrection(Int_t /*flag*/) {;}
+  virtual void   SetReaderHeader(AliJetReaderHeader* header) 
+  {fReaderHeader = header;}
+  virtual void   SetESD(AliESDEvent* esd) { fESD = esd;}
+
+  // Others
+  virtual void   OpenInputFiles() {}
+  virtual void   SetInputEvent(const TObject* /*esd*/, const TObject* /*aod*/, const TObject* /*mc*/) {;}
+  virtual void   ConnectTree(TTree* /*tree*/, TObject* /*data*/) {}
+  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
+  TChain                          *fChain;                 // chain for reconstructed tracks
+  TChain                          *fTree;                  // tree for reconstructed tracks
+  TClonesArray                    *fMomentumArray;         // array of particle momenta
+  TClonesArray                    *fArrayMC;               //! array of mc particles
+  TTask                           *fFillUnitArray;         //! task list for filling the UnitArray
+  AliESDEvent                     *fESD;                   // pointer to esd
+  AliJetReaderHeader              *fReaderHeader;          // pointer to header
+  AliHeader                       *fAliHeader;             // AliHeader
+  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
+  TClonesArray                    *fUnitArray;             // array of digit position and energy 
+  Bool_t                           fArrayInitialised;      // To check that array of units is initialised  
+  AliJetFillUnitArray             *fFillUAFromTracks;      // For charged particle task
+  AliJetFillUnitArray             *fFillUAFromEMCalDigits; // For neutral particle task
+  Int_t                            fNumCandidate;          // Number of entries different from zero in unitarray
+  Int_t                            fNumCandidateCut;       // Number of entries different from zero in unitarray
+                                                           // which pass pt cut
+  AliJetHadronCorrection          *fHadronCorrector;       //! Pointer to hadronic correction 
+  Int_t                            fHCorrection;           //  Hadron correction flag 
+  Int_t                            fECorrection;           //  Electron correction flag 
+  Bool_t                           fEFlag;                 //  Electron correction flag 
+
 
   ClassDef(AliJetReader,1)
 };