PWGJE/EMCALJetTasks/AliJetContainer.h

   1 #ifndef AliJetContainer_H
   2 #define AliJetContainer_H
   3
   4 //
   5 // container with name, TClonesArray and cuts for jets
   6 //
   7
   8 class TClonesArray;
   9 class TString;
  10
  11 class AliEMCALGeometry;
  12 class AliEmcalJet;
  13 class AliVEvent;
  14 class AliParticleContainer;
  15 class AliClusterContainer;
  16 class AliLocalRhoParameter;
  17
  18 #include "AliRhoParameter.h"
  19
  20 #include "AliEmcalContainer.h"
  21
  22 class AliJetContainer : public AliEmcalContainer {
  23  public:
  24
  25   enum JetAcceptanceType {
  26     kTPC       = 0,     // TPC acceptance
  27     kEMCAL     = 1,     // EMCal acceptance
  28     kUser      = 2      // User defined acceptance
  29   };
  30
  31
  32   AliJetContainer();
  33   AliJetContainer(const char *name);
  34   virtual ~AliJetContainer() {;}
  35
  36   void SetEMCALGeometry();
  37   void SetEMCALGeometry(AliEMCALGeometry *p) {fGeom = p;}
  38   void LoadRho(AliVEvent *event);
  39   void LoadLocalRho(AliVEvent *event);
  40
  41   void                        SetJetAcceptanceType(JetAcceptanceType type)         { fJetAcceptanceType          = type ; }
  42
  43   void                        ResetCuts();
  44
  45   void                        SetJetEtaPhiEMCAL() ;
  46   void                        SetJetEtaPhiTPC()   ;
  47   void                        SetRunNumber(Int_t r)                                { fRunNumber = r;                      }
  48
  49   void                        SetJetEtaLimits(Float_t min, Float_t max)            { fJetMinEta = min, fJetMaxEta = max ; }
  50   void                        SetJetPhiLimits(Float_t min, Float_t max)            { fJetMinPhi = min, fJetMaxPhi = max ; }
  51   void                        SetJetAreaCut(Float_t cut)                           { fJetAreaCut     = cut              ; }
  52   void                        SetPercAreaCut(Float_t p)                            { if(fJetRadius==0.) AliWarning("JetRadius not set. Area cut will be 0"); fJetAreaCut = p*TMath::Pi()*fJetRadius*fJetRadius; }
  53   void                        SetZLeadingCut(Float_t zemc, Float_t zch)            { fZLeadingEmcCut = zemc; fZLeadingChCut = zch ; }
  54   void                        SetNEFCut(Float_t min = 0., Float_t max = 1.)        { fNEFMinCut = min; fNEFMaxCut = max;  }
  55
  56   void                        SetAreaEmcCut(Double_t a = 0.99)                     { fAreaEmcCut     = a                ; }
  57   void                        SetJetPtCut(Float_t cut)                             { fJetPtCut       = cut              ; }
  58   void                        SetJetRadius(Float_t r)                              { fJetRadius      = r                ; }
  59   virtual void                SetRhoName(const char *n)                            { fRhoName        = n                ; }
  60   virtual void                SetLocalRhoName(const char *n)                       { fLocalRhoName   = n                ; }
  61   void                        SetMaxClusterPt(Float_t b)                           { fMaxClusterPt   = b                ; }
  62   void                        SetMaxTrackPt(Float_t b)                             { fMaxTrackPt     = b                ; }
  63   void                        SetPtBiasJetClus(Float_t b)                          { fPtBiasJetClus  = b                ; }
  64   void                        SetNLeadingJets(Int_t t)                             { fNLeadingJets   = t                ; }
  65   void                        SetPtBiasJetTrack(Float_t b)                         { fPtBiasJetTrack = b                ; }
  66
  67   void                        SetLeadingHadronType(Int_t t)                        { fLeadingHadronType = t             ; }
  68   void                        SetJetBitMap(UInt_t m)                               { fJetBitMap      = m                ; }
  69   void                        SetJetTrigger(UInt_t t=AliVEvent::kEMCEJE)           { fJetTrigger     = t                ; }
  70
  71   void                        ConnectParticleContainer(AliParticleContainer *c)    { fParticleContainer = c             ; }
  72   void                        ConnectClusterContainer(AliClusterContainer *c)      { fClusterContainer  = c             ; }
  73
  74   AliEmcalJet                *GetLeadingJet(const char* opt="")          ;
  75   AliEmcalJet                *GetJet(Int_t i)                       const;
  76   AliEmcalJet                *GetAcceptJet(Int_t i)                 const;
  77   AliEmcalJet                *GetJetWithLabel(Int_t lab)            const;
  78   AliEmcalJet                *GetAcceptJetWithLabel(Int_t lab)      const;
  79   AliEmcalJet                *GetNextAcceptJet(Int_t i=-1)               ;
  80   AliEmcalJet                *GetNextJet(Int_t i=-1)                     ;
  81   void                        GetMomentum(TLorentzVector &mom, Int_t i) const;
  82   Bool_t                      AcceptJet(AliEmcalJet* jet)           const;
  83   Bool_t                      AcceptBiasJet(AliEmcalJet* jet)       const;
  84   Int_t                       GetNJets()                            const    {return GetNEntries();}
  85   Double_t                    GetLeadingHadronPt(AliEmcalJet* jet)  const;
  86   void                        GetLeadingHadronMomentum(TLorentzVector &mom, AliEmcalJet* jet)  const;
  87   Double_t                    GetZ(AliEmcalJet *jet, TLorentzVector mom) const;
  88   Double_t                    GetZLeadingEmc(AliEmcalJet *jet)      const;
  89   Double_t                    GetZLeadingCharged(AliEmcalJet *jet)  const;
  90   AliRhoParameter            *GetRhoParameter()                              {return fRho;}
  91   Double_t                    GetRhoVal()                           const    { if (fRho) return fRho->GetVal(); else return 0;}
  92   const TString&              GetRhoName()                          const    {return fRhoName;}
  93   AliLocalRhoParameter       *GetLocalRhoParameter()                const    {return fLocalRho;}
  94   const TString&              GetLocalRhoName()                     const    {return fLocalRhoName;}
  95   Double_t                    GetJetPtCorr(Int_t i)                 const;
  96   Double_t                    GetJetPtCorrLocal(Int_t i)            const;
  97   Float_t                     GetJetRadius()                        const    {return fJetRadius;}
  98   Float_t                     GetJetEtaMin()                        const    {return fJetMinEta;}
  99   Float_t                     GetJetEtaMax()                        const    {return fJetMaxEta;}
 100   Float_t                     GetJetPhiMin()                        const    {return fJetMinPhi;}
 101   Float_t                     GetJetPhiMax()                        const    {return fJetMaxPhi;}
 102   void                        SetClassName(const char *clname);
 103   void                        SetArray(AliVEvent *event);
 104   AliParticleContainer       *GetParticleContainer()                         {return fParticleContainer;}
 105   AliClusterContainer        *GetClusterContainer()                          {return fClusterContainer;}
 106
 107  protected:
 108   JetAcceptanceType           fJetAcceptanceType;    //  acceptance type
 109   Float_t                     fJetRadius;            //  jet radius
 110   TString                     fRhoName;              //  Name of rho object
 111   TString                     fLocalRhoName;         //  Name of local rho object
 112   Float_t                     fPtBiasJetTrack;       //  select jets with a minimum pt track
 113   Float_t                     fPtBiasJetClus;        //  select jets with a minimum pt cluster
 114   Float_t                     fJetPtCut;             //  cut on jet pt
 115   Float_t                     fJetAreaCut;           //  cut on jet area
 116   Float_t                     fAreaEmcCut;           //  minimum cut on jet emcal area
 117   Float_t                     fJetMinEta;            //  minimum eta jet acceptance
 118   Float_t                     fJetMaxEta;            //  maximum eta jet acceptance
 119   Float_t                     fJetMinPhi;            //  minimum phi jet acceptance
 120   Float_t                     fJetMaxPhi;            //  maximum phi jet acceptance
 121   Float_t                     fMaxClusterPt;         //  maximum cluster constituent pt to accept the jet
 122   Float_t                     fMaxTrackPt;           //  maximum track constituent pt to accept the jet
 123   Float_t                     fZLeadingEmcCut;       //  maximum z,leading neutral
 124   Float_t                     fZLeadingChCut;        //  maximum z,leading charged
 125   Float_t                     fNEFMinCut;            //  minimum NEF in a jet
 126   Float_t                     fNEFMaxCut;            //  maximum NEF in a jet
 127   Int_t                       fLeadingHadronType;    //  0 = charged, 1 = neutral, 2 = both
 128   Int_t                       fNLeadingJets;         //  how many jets are to be considered the leading jet(s)
 129   UInt_t                      fJetBitMap;            //  bit map of accepted jets
 130   UInt_t                      fJetTrigger;           //  jet trigger
 131   AliParticleContainer       *fParticleContainer;    //  particle container (jet constituents)
 132   AliClusterContainer        *fClusterContainer;     //  cluster container (jet constituents)
 133
 134   AliRhoParameter            *fRho;                  //! event rho for these jets
 135   AliLocalRhoParameter       *fLocalRho;             //! event local rho for these jets
 136   AliEMCALGeometry           *fGeom;                 //! emcal geometry
 137   Int_t                       fRunNumber;            //! run number
 138
 139  private:
 140   AliJetContainer(const AliJetContainer& obj); // copy constructor
 141   AliJetContainer& operator=(const AliJetContainer& other); // assignment
 142
 143   ClassDef(AliJetContainer,5);
 144
 145 };
 146
 147 #endif
 148
 149