JETAN/AliJetParticlesReader.h

   1 #ifndef ALIJETPARTICLESREADER_H
   2 #define ALIJETPARTICLESREADER_H
   3
   4 /* $Id$ */
   5
   6 //_________________________________________________________________________
   7 ///////////////////////////////////////////////////////////////////////////
   8 //                                                                       //
   9 // class AliJetReader                                                    //
  10 // File reader for ESD                                                   //
  11 //                                                                       //
  12 // This reader reads tracks from Event Summary Data                      //
  13 // taken from Piotr.Skowronski@cern.ch                                   //
  14 // more info: http://alisoft.cern.ch/people/skowron/analyzer/index.html  //
  15 //                                                                       //
  16 // loizides@ikf.uni-frankfurt.de                                         //
  17 ///////////////////////////////////////////////////////////////////////////
  18
  19 #include <TNamed.h>
  20 #include <TObjArray.h>
  21 class TClonesArray;
  22 class TString;
  23 class AliJetEventParticles;
  24
  25 class AliJetParticlesReader: public TNamed
  26 {
  27   public:
  28   AliJetParticlesReader();
  29   AliJetParticlesReader(TObjArray* dirs);
  30   virtual ~AliJetParticlesReader();
  31
  32   void SetDirs(TObjArray* dirs){fDirs = dirs;} //sets array directories names
  33   void ReadEventsFromTo(Int_t first,Int_t last){fFirst = first; fLast = last;}
  34
  35   virtual Int_t Next(); //call this if you want the next event
  36   virtual void  Rewind() = 0;
  37
  38   void SetPtCut(Float_t ptmin=0, Float_t ptmax=1000)
  39     {fPtMin=ptmin;fPtMax=ptmax;}
  40   void SetPhiCut(Float_t phi=2*TMath::Pi()){SetPhiCut(0,phi);}
  41   void SetPhiCut(Float_t phimin, Float_t phimax)
  42     {fPhiMin=phimin;fPhiMax=phimax;}
  43   void SetEtaCut(Float_t e=1){SetEtaCut(-e,e);}
  44   void SetEtaCut(Float_t emin, Float_t emax)
  45     {fEtaMin=emin;fEtaMax=emax;}
  46
  47   const AliJetEventParticles* GetEventParticles() const {return fEventParticles;}
  48   AliJetEventParticles* GetEventParticles(Bool_t o)
  49     {fOwner=o; return fEventParticles;} //return particles and set ownership
  50
  51   Int_t GetNumberOfDirs()   const {return (fDirs)?fDirs->GetEntries():0;}
  52   Int_t GetCurEventNumber() const {return fCurrentEvent;}
  53   Int_t GetCurDirNumber()   const {return fCurrentDir;}
  54   Int_t GetTotEventsRead()  const {return fNEventsRead;}
  55   Int_t GetFirstEvent()     const {return fFirst;}
  56   Int_t GetLastEvent()      const {return fLast;}
  57
  58   protected:
  59
  60   virtual Int_t ReadNext() = 0; //this methods reads next event and
  61                                 //put result in fParticles
  62
  63   TString& GetDirName(Int_t entry);
  64
  65   AliJetEventParticles* fEventParticles; //array with read particles
  66   Bool_t fOwner;              //ownership of particles
  67   TObjArray*   fDirs;         //array with directories to read data from
  68   Int_t        fCurrentEvent; //number of current event in current file
  69   Int_t        fCurrentDir;   //number of current directory in array
  70   Int_t        fNEventsRead;  //total number of processed events
  71   Int_t        fFirst;        //first event to return (all before are skipped)
  72   Int_t        fLast;         //last event to return (relative to total number)
  73
  74   Float_t fPtMin;   //min pt cut
  75   Float_t fPtMax;   //max pt cut
  76   Float_t fEtaMin;  //min eta cut
  77   Float_t fEtaMax;  //max eta cut
  78   Float_t fPhiMin;  //min phi cut
  79   Float_t fPhiMax;  //max phi cut
  80
  81   ClassDef(AliJetParticlesReader,1) // Basic AliJetParticles Reader class
  82 };
  83
  84 #endif