JETAN/AliTkHijingAna.h

   1 // $Id$
   2
   3 #ifndef ALITKHIJINGANA_H
   4 #define ALITKHIJINGANA_H
   5
   6 #include <Riostream.h>
   7 #include <TObject.h>
   8 #include "AliTkEtaPhiVector.h"
   9
  10 class AliTkHijingAna : public TObject {
  11  public:
  12   // how to use it: - call setCenter
  13   //                - call setParticles at beginning (not mandatory)
  14   //                - call getParticlesInRadius with maximum radius
  15   //                - call analysis functions
  16   //                - repeat last 2 steps with decreasing radia
  17
  18   // center of cone
  19   void setCenter(Float_t eta, Float_t phi);
  20   Float_t getCenterEta() const;
  21   Float_t getCenterPhi() const;
  22
  23   // helper functions
  24   Bool_t isParticleInRadius(TParticle *particle,Float_t radius);
  25   Bool_t isParticleAccepted(TParticle *particle, Float_t pt=0.1, Bool_t neutral=kFALSE);
  26   Bool_t isParticleAcceptedALICE(TParticle *particle, Float_t pt=0.1, Bool_t neutral=kFALSE);
  27   void clear();
  28   void clearCone();
  29
  30   // functions to create local copies of particles
  31   void setParticles(TClonesArray *particles);
  32   void setParticles(TClonesArray *particles,Float_t pt,Bool_t neutral);
  33   TClonesArray *getParticlesInRadius(Float_t radius);
  34   TClonesArray *getParticlesInRadius(TClonesArray *particles, Float_t radius);
  35
  36   // analysis functions
  37   Int_t getNParticlesInRadius(Float_t ptCut);
  38   Int_t getNParticlesInRadius() { return getNParticlesInRadius(0); }
  39   Float_t getEtInRadius(Float_t ptCut);
  40   Float_t getEtInRadius() { return getEtInRadius(0); }
  41
  42
  43   //--- old stuff, don't use anymore
  44   Float_t getEtInRadius(TClonesArray *particles,Float_t radius);
  45   //--- end of old stuff
  46
  47  private:
  48   AliTkEtaPhiVector center;
  49   TClonesArray *oParticles;
  50   TClonesArray *mParticles;
  51   Float_t mRadius;
  52   Bool_t updated;
  53
  54   ClassDef(AliTkHijingAna,0)
  55 };
  56
  57 #endif