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 TTree;
  24 class AliJetEventParticles;
  25
  26 class AliJetParticlesReader: public TNamed
  27 {
  28   public:
  29   AliJetParticlesReader();
  30   AliJetParticlesReader(TObjArray* dirs);
  31   virtual ~AliJetParticlesReader();
  32
  33   void SetDirs(TObjArray* dirs){fDirs = dirs;} //sets array directories names
  34   void ReadEventsFromTo(Int_t first,Int_t last){fFirst = first; fLast = last;}
  35
  36   virtual Int_t Next(); //call this if you want the next event
  37   virtual void  Rewind() = 0;
  38
  39   void SetPtCut(Float_t ptmin=0, Float_t ptmax=100)
  40     {fPtMin=ptmin;fPtMax=ptmax;}
  41   void SetPhiCut(Float_t phi=2*TMath::Pi()){SetPhiCut(0,phi);}
  42   void SetPhiCut(Float_t phimin, Float_t phimax)
  43     {fPhiMin=phimin;fPhiMax=phimax;}
  44   void SetEtaCut(Float_t e=1){SetEtaCut(-e,e);}
  45   void SetEtaCut(Float_t emin, Float_t emax)
  46     {fEtaMin=emin;fEtaMax=emax;}
  47   void SetTree(TTree *t){fTree=t;fNewTree=kTRUE;}
  48
  49   const AliJetEventParticles* GetEventParticles() const {return fEventParticles;}
  50   AliJetEventParticles* GetEventParticles(Bool_t o)
  51     {fOwner=o; return fEventParticles;} //return particles and set ownership
  52
  53   Int_t GetNumberOfDirs()   const {return (fDirs)?fDirs->GetEntries():0;}
  54   Int_t GetCurEventNumber() const {return fCurrentEvent;}
  55   Int_t GetCurDirNumber()   const {return fCurrentDir;}
  56   Int_t GetTotEventsRead()  const {return fNEventsRead;}
  57   Int_t GetFirstEvent()     const {return fFirst;}
  58   Int_t GetLastEvent()      const {return fLast;}
  59
  60   protected:
  61
  62   virtual Int_t ReadNext() = 0; //this methods reads next event and
  63                                 //put result in fParticles
  64
  65   TString& GetDirName(Int_t entry);
  66
  67   AliJetEventParticles* fEventParticles; //array with read particles
  68   Bool_t fOwner;              //ownership of particles
  69   TObjArray*   fDirs;         //array with directories to read data from
  70   Int_t        fCurrentEvent; //number of current event in current file
  71   Int_t        fCurrentDir;   //number of current directory in array
  72   Int_t        fNEventsRead;  //total number of processed events
  73   Int_t        fFirst;        //first event to return (all before are skipped)
  74   Int_t        fLast;         //last event to return (relative to total number)
  75
  76   Float_t fPtMin;   //min pt cut
  77   Float_t fPtMax;   //max pt cut
  78   Float_t fEtaMin;  //min eta cut
  79   Float_t fEtaMax;  //max eta cut
  80   Float_t fPhiMin;  //min phi cut
  81   Float_t fPhiMax;  //max phi cut
  82
  83   Bool_t fNewTree;  // signals new tree
  84   TTree *fTree;     //! if non-zero store AliJetEventParticles
  85
  86   ClassDef(AliJetParticlesReader,2) // Basic AliJetParticles Reader class
  87 };
  88
  89 #endif