PWGJE
[u/mrichter/AliRoot.git] / PWGJE / UserTasks / AliAnalysisTaskIDFragmentationFunction.h
index b98f210..1a656c1 100644 (file)
@@ -22,8 +22,10 @@ class TProfile;
 class THnSparse; 
 class TRandom3;
 class TArrayS;
+class AliAnalysisUtils;
 class AliAODTrack;
 class AliAODMCParticle;
+class AliAnalysisTaskPID;
 
 #include "AliAnalysisTaskSE.h"
 #include "AliPID.h"
@@ -223,6 +225,7 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   virtual void   SetIDFFMode(Int_t idff = 0)      {fIDFFMode = idff;}
   virtual void   SetEffMode(Int_t eff = 1)    {fEffMode = eff;}
   virtual void   SetJSMode(Int_t js = 1)      {fJSMode = js;}
+  virtual void   SetMCPtHardCut(Float_t ptHardCut)      { fMCPtHardCut = ptHardCut; }
 
   static  void   SetProperties(TH1* h,const char* x, const char* y);
   static  void   SetProperties(TH1* h,const char* x, const char* y,const char* z);
@@ -254,12 +257,12 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
     fQATrackNBinsPhi = nPhi; fQATrackPhiMin = phiMin; fQATrackPhiMax = phiMax; }
   
 
-
   Float_t  GetFFRadius() const { return fFFRadius; }
   Float_t  GetFFMinLTrackPt() const { return fFFMinLTrackPt; }
   Float_t  GetFFMaxTrackPt() const { return fFFMaxTrackPt; }
   Float_t  GetFFMinNTracks() const { return fFFMinnTracks; }
   Float_t  GetFFBckgRadius() const { return fFFBckgRadius; }
+  Float_t  GetMCPtHardCut() const  { return fMCPtHardCut; }
   void    GetJetTracksTrackrefs(TList* l, const AliAODJet* j, Double_t minPtL, Double_t maxPt, Bool_t& isBadPt);
   void    GetJetTracksPointing(TList* in, TList* out, const AliAODJet* j, Double_t r, Double_t& sumPt, Double_t minPtL, Double_t maxPt,
                                 Bool_t& isBadPt);  
@@ -469,6 +472,7 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   TList        *fCommonHistList;         // List of common histos
   
   TH1F  *fh1EvtSelection;         //! event cuts 
+  TH1F  *fh1VtxSelection;         //! type of accepted vertices
   TH1F *fh1VertexNContributors;  //! NContributors to prim vertex
   TH1F *fh1VertexZ;              //! prim vertex z distribution
   TH1F *fh1EvtMult;              //! number of reconstructed tracks after cuts 
@@ -478,6 +482,8 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   TH1F*     fh1Trials;            //! sum of trials
   TH1F*     fh1PtHard;            //! pt hard of the event
   TH1F*     fh1PtHardTrials;      //! pt hard of the event
+  
+  TH1F*     fh1EvtsPtHardCut;     //! Number events before and after the cut on MC pT hard
 
   TH1F  *fh1nRecJetsCuts;         //! number of jets from reconstructed tracks per event 
   TH1F  *fh1nGenJets;             //! number of jets from generated tracks per event
@@ -491,6 +497,10 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   
   TH2F  *fhDCA_XY;                //! DCA XY for all rec. particles
   TH2F  *fhDCA_Z;                 //! DCA Z for all rec. particles
+  
+  TH2F  *fhJetPtRefMultEta5;      //! Jet pT vs. reference multiplicity (|eta|<0.5)
+  TH2F  *fhJetPtRefMultEta8;      //! Jet pT vs. reference multiplicity (|eta|<0.8)
+  TH2F  *fhJetPtMultPercent;      //! Jet pT vs. multiplicity percentile (usually V0M)
 
   TH2F  *fhDCA_XY_prim_MCID[AliPID::kSPECIES];   //! DCA XY for all rec. prim. particles sorted by MC-ID
   TH2F  *fhDCA_Z_prim_MCID[AliPID::kSPECIES];    //! DCA Z for all rec. prim. particles sorted by MC-ID
@@ -582,6 +592,9 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   TRandom3* fRandom;                        //! TRandom3 for background estimation 
   
   Bool_t fOnlyLeadingJets;                  // Flag indicating whether some histos are filled with leading jets only or all jets
+  Float_t fMCPtHardCut;                     // Cut on MC pThard (smaller that threshold), if set to non-negative value
+  
+  AliAnalysisUtils *fAnaUtils;              //! Object to use analysis utils like pile-up rejection
   
   // PID framework
   Int_t fNumInclusivePIDtasks;              // Number of inclusive PID tasks used 
@@ -601,7 +614,7 @@ class AliAnalysisTaskIDFragmentationFunction : public AliAnalysisTaskSE {
   AliFragFuncHistos* fIDFFHistosRecCuts[AliPID::kSPECIES];    //! Identified FF reconstructed tracks after cuts 
   AliFragFuncHistos* fIDFFHistosGen[AliPID::kSPECIES];    //! Identified FF generated tracks after cuts 
 
-  ClassDef(AliAnalysisTaskIDFragmentationFunction, 18);
+  ClassDef(AliAnalysisTaskIDFragmentationFunction, 21);
 };