JETAN/AliJetReader.h

   1 #ifndef ALIJETREADER_H
   2 #define ALIJETREADER_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //---------------------------------------------------------------------
   8 // Jet reader base class
   9 // manages the reading of input for jet algorithms
  10 // Author: jgcn@mda.cinvestav.mx
  11 //---------------------------------------------------------------------
  12
  13 #include <TObject.h>
  14 #include <TChain.h>
  15 #include <TArrayI.h>
  16
  17 class TClonesArray;
  18 class AliJetReaderHeader;
  19 class AliESD;
  20 class AliHeader;
  21
  22
  23 class AliJetReader : public TObject
  24 {
  25  public:
  26   AliJetReader();
  27   virtual ~AliJetReader();
  28
  29   // Getters
  30   virtual TClonesArray *GetMomentumArray() {return fMomentumArray;}
  31   virtual Int_t GetChainEntries() {return fChain->GetEntries();}
  32   virtual AliJetReaderHeader* GetReaderHeader() { return fReaderHeader;}
  33   virtual AliHeader* GetAliHeader() { return fAliHeader;}
  34   virtual Int_t GetSignalFlag(Int_t i) const {return fSignalFlag[i];}
  35
  36   // Setters
  37   virtual void FillMomentumArray(Int_t) {}
  38   virtual void SetReaderHeader(AliJetReaderHeader* header)
  39     {fReaderHeader = header;}
  40
  41   // others
  42   virtual void OpenInputFiles() {}
  43   void ClearArray();
  44
  45  protected:
  46   TChain                  *fChain;         // chain for reconstructed tracks
  47   TChain                  *fChainMC;       // chain for mc information
  48   TClonesArray            *fMomentumArray; // array of particle momenta
  49   TClonesArray            *fArrayMC;       // array of mc particles
  50   AliESD                  *fESD;           // pointer to esd
  51   AliJetReaderHeader      *fReaderHeader;  // pointer to header
  52   AliHeader               *fAliHeader;     // pointer to event header
  53   TArrayI fSignalFlag;   // to flag if a particle comes from pythia or
  54                         // from the underlying event
  55
  56   ClassDef(AliJetReader,1)
  57 };
  58
  59 #endif