EVGEN/AliGenReaderTreeK.h

   1 #ifndef ALIGENREADERTREEK_H
   2 #define ALIGENREADERTREEK_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7 //
   8 // Realisation of AliGenReader to be used with AliGenExtFile
   9 // It reads events from a kinematics TreeK.
  10 // Author: andreas.morsch@cern.ch
  11 //
  12 #include "AliGenReader.h"
  13
  14 class TFile;
  15 class AliRunLoader;
  16 class AliStack;
  17 class TString;
  18 class TObjArray;
  19
  20 class AliGenReaderTreeK : public AliGenReader
  21 {
  22  public:
  23     AliGenReaderTreeK();
  24     AliGenReaderTreeK(const AliGenReaderTreeK &reader);
  25     virtual ~AliGenReaderTreeK();
  26     // Initialise
  27     virtual void Init();
  28     // Read
  29     virtual Int_t NextEvent();
  30     virtual TParticle*  NextParticle();
  31     virtual void RewindEvent();
  32     virtual void SetFirstEvent(Int_t evt)    {fNcurrent  = evt;};
  33     virtual void SetOnlyPrimaries(Bool_t flag) {fOnlyPrimaries = flag;}
  34     AliGenReaderTreeK & operator=(const AliGenReaderTreeK & rhs);
  35     void SetDirs(TObjArray* dirs){fDirs = dirs;} //sets array directories names
  36     void AddDir(const char* dirname);
  37     AliRunLoader * GetRunLoader() const {return fInRunLoader;}
  38
  39  protected:
  40     Int_t             fNcurrent;          // points to the next entry
  41     Int_t             fNparticle;         // Next particle in list
  42     Int_t             fNp;                // number of particles
  43     AliRunLoader     *fInRunLoader;       //!Run Loader of the input event
  44     TFile            *fBaseFile;          //! pointer to base file
  45     AliStack         *fStack;             //! Particle stack
  46     Bool_t            fOnlyPrimaries;     // Flag indicating wether only primaries are read
  47     TObjArray        *fDirs;              //arry with directories to read data from
  48     Int_t             fCurrentDir;        //Number of current directory
  49     static const TString fgkEventFolderName;//!name of folder where event to read is mounted
  50
  51     TString&   GetDirName(Int_t entry);
  52     TParticle* GetParticle(Int_t i);
  53  private:
  54     void Copy(TObject&) const;
  55     ClassDef(AliGenReaderTreeK,1) // Read particles from TreeK
  56 };
  57
  58
  59
  60 #endif
  61
  62
  63
  64
  65
  66