From Marta and Redmer:

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliAnalysisTaskEmcalJetDev.h

diff --git a/PWGJE/EMCALJetTasks/AliAnalysisTaskEmcalJetDev.h b/PWGJE/EMCALJetTasks/AliAnalysisTaskEmcalJetDev.h
index 1b6ae402de88a6d03dd8be07291f30ed866300c8..865b6816eb4c4b2bed6ef4b384ea4ed893e6e27b 100644 (file)
--- a/PWGJE/EMCALJetTasks/AliAnalysisTaskEmcalJetDev.h
+++ b/PWGJE/EMCALJetTasks/AliAnalysisTaskEmcalJetDev.h
@@ -8,6 +8,7 @@ class TList;
 class TString;
 class AliEmcalJet;
 class AliRhoParameter;
+class AliLocalRhoParameter;
 class AliVCluster;
 class AliVParticle;
 class AliJetContainer;
@@ -21,12 +22,15 @@ class AliAnalysisTaskEmcalJetDev : public AliAnalysisTaskEmcalDev {
   virtual ~AliAnalysisTaskEmcalJetDev();
 
   //these should all point to the jet container
-  void                SetAnaType(UInt_t t, Int_t c = 0);
+  void                SetAnaType(UInt_t t, Int_t c = 0) { SetJetAcceptanceType(t,c); }
+  void                SetJetAcceptanceType(UInt_t t, Int_t c = 0);
   void                SetJetAcceptanceType(TString cutType, Int_t c = 0);
   void                SetJetEtaLimits(Float_t min, Float_t max, Int_t c = 0);
   void                SetJetPhiLimits(Float_t min, Float_t max, Int_t c = 0);
   void                SetJetAreaCut(Float_t cut, Int_t c = 0);
   void                SetPercAreaCut(Float_t p, Int_t c = 0);
+  void                SetZLeadingCut(Float_t zemc, Float_t zch, Int_t c = 0);
+  void                SetNEFCut(Float_t min, Float_t max, Int_t c = 0);
   void                SetAreaEmcCut(Double_t a = 0.99, Int_t c = 0);
   void                SetJetPtCut(Float_t cut, Int_t c = 0);
   void                SetJetRadius(Float_t r, Int_t c = 0);
@@ -38,9 +42,13 @@ class AliAnalysisTaskEmcalJetDev : public AliAnalysisTaskEmcalDev {
   void                SetNLeadingJets(Int_t t, Int_t c = 0);
   void                SetJetBitMap(UInt_t m, Int_t c = 0);
   void                SetJetTrigger(UInt_t t, Int_t c = 0);
+  void                SetIsParticleLevel(Bool_t b, Int_t c = 0);
 
   void                SetJetsName(const char *n)                   { AddJetContainer(n); }
   virtual void        SetRhoName(const char *n, Int_t c = 0);
+  virtual void        SetLocalRhoName(const char *n)               { fLocalRhoName   = n; }
+
+  const TString&      GetRhoName(Int_t c = 0) const;
 
   AliJetContainer    *AddJetContainer(const char *n, TString defaultCutType = "", Float_t jetRadius = 0.4);
   void                RemoveJetContainer(Int_t i)                  { fJetCollArray.RemoveAt(i);} 
@@ -53,9 +61,11 @@ class AliAnalysisTaskEmcalJetDev : public AliAnalysisTaskEmcalDev {
   void                        ExecOnce()                                                                    ;
 
   AliRhoParameter            *GetRhoFromEvent(const char *name)                                             ;
+  AliLocalRhoParameter       *GetLocalRhoFromEvent(const char *name)                                        ;
   Bool_t                      IsJetTrack(AliEmcalJet* jet, Int_t itrack, Bool_t sorted = kFALSE)       const;
   Bool_t                      IsJetCluster(AliEmcalJet* jet, Int_t iclus, Bool_t sorted = kFALSE)      const;
   Bool_t                      RetrieveEventObjects()                                                        ;
+  Double_t                    GetJetRadius(Int_t i=0)                                                  const;
   AliJetContainer            *GetJetContainer(Int_t i=0)                                               const;
   AliJetContainer            *GetJetContainer(const char* name)                                        const;
   TClonesArray               *GetJetArray(Int_t i=0)                                                   const;
@@ -65,16 +75,18 @@ class AliAnalysisTaskEmcalJetDev : public AliAnalysisTaskEmcalDev {
   Double_t                    GetRhoVal(Int_t i=0)                                                     const;
 
   TString                     fRhoName;                    // rho name
+  TString                     fLocalRhoName;               // name for local rho
   TObjArray                   fJetCollArray;               // jet collection array
 
   TClonesArray               *fJets;                       //!jets
   AliRhoParameter            *fRho;                        //!event rho
-  Double_t                    fRhoVal;                     //!event rho value
+  AliLocalRhoParameter       *fLocalRho;                   //!local event rho
+  Double_t                    fRhoVal;                     //!event rho value, same for local rho
 
  private:
   AliAnalysisTaskEmcalJetDev(const AliAnalysisTaskEmcalJetDev&);            // not implemented
   AliAnalysisTaskEmcalJetDev &operator=(const AliAnalysisTaskEmcalJetDev&); // not implemented
 
-  ClassDef(AliAnalysisTaskEmcalJetDev, 2) // EMCAL Jet base analysis task
+  ClassDef(AliAnalysisTaskEmcalJetDev, 3) // EMCAL Jet base analysis task
 };
 #endif