PWG4/JetTasks/AliAnalysisTaskFragFuncBB.h

   1 #ifndef AliAnalysisTaskFragFuncBB_h
   2 #define AliAnalysisTaskFragFuncBB_h
   3
   4 class AliESDEvent;
   5 class AliAODEvent;
   6 class TList;
   7 class TH1F;
   8 class TH2F;
   9 class TH3F;
  10
  11
  12 #include "AliAnalysisTaskSE.h"
  13
  14 class AliAnalysisTaskFragFuncBB : public AliAnalysisTaskSE {
  15
  16         public:
  17
  18         AliAnalysisTaskFragFuncBB();
  19         AliAnalysisTaskFragFuncBB(const char *name);
  20         virtual ~AliAnalysisTaskFragFuncBB();
  21
  22         virtual void   UserCreateOutputObjects();
  23         virtual void   Init();
  24         virtual void   LocalInit() {Init();}
  25         virtual void   UserExec(Option_t *option);
  26         virtual void   Terminate(Option_t* );
  27
  28         virtual void   SetFilterMask(UInt_t i) {fFilterMask = i;}
  29         virtual void   SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
  30         virtual void   SetAODJetInput(Bool_t b){fUseAODJetInput = b;}
  31         virtual void   SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
  32         virtual void   SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
  33         virtual void   SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
  34         virtual void   SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
  35         virtual void   SetBranchGenJets(const char* c){fBranchGenJets = c;}
  36         virtual void   SetBranchRecJets(const char* c){fBranchRecJets = c;}
  37
  38
  39         private:
  40
  41         // Consts
  42         enum{ fgkMaxJets=4 }; // max. nb. of stored jets
  43
  44         enum {kTrackUndef=0, kTrackAOD, kTrackKineAll, kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
  45
  46         //
  47         Int_t   GetListOfTracks(TList *list, Int_t type);
  48
  49         //
  50         AliESDEvent* fESD;
  51         AliAODEvent* fAOD;
  52         AliMCEvent*  fMCEvent;
  53
  54         TString fBranchRecJets; // branch name for reconstructed jets
  55         TString fBranchGenJets; // branch name for generated jets
  56
  57         Int_t   fLeadingRecJet; //
  58         Int_t   fLeadingGenJet; //
  59
  60         Int_t   fTrackTypeRec;       // type of reconstructed tracks
  61         Int_t   fTrackTypeGen;       // type of generated tracks
  62         UInt_t  fFilterMask;         // filter bit for selected tracks
  63         Bool_t  fUseGlobalSelection; // use selection of JetHelperTask
  64         Bool_t  fUseAODJetInput;         // take jet from input AOD not from ouptut AOD
  65         Bool_t  fUseAODTrackInput;       // take track from input AOD not from ouptut AOD
  66         Bool_t  fUseAODMCInput;          // take MC from input AOD not from ouptut AOD
  67
  68         Float_t fJetRadius;  // jet radius, used for fragmentation function
  69         Float_t fEtaMaxPart; // eta acceptance range for particles
  70         Float_t fEtaMaxJets; // eta acceptance range for jets
  71
  72
  73         // Histograms
  74         TList           *fHistList;               // List of histos
  75
  76         TH1F            *fh1_eventSelection;
  77         TH1F            *fh1_vertexNContributors;
  78         TH1F            *fh1_vertexZ;
  79
  80         // all jets
  81         TH1F            *fh1_recJets_Et;
  82         TH1F            *fh1_genJets_Et;
  83         TH2F            *fh2_recJets_EtaPhi;
  84         TH2F            *fh2_genJets_EtaPhi;
  85
  86         // jets, without acceptance cuts
  87         TH1F            *fh1_recJetsWoC_Et;
  88         TH1F            *fh1_genJetsWoC_Et;
  89         TH2F            *fh2_recJetsWoC_EtaPhi;
  90         TH2F            *fh2_genJetsWoC_EtaPhi;
  91
  92         // leading jets
  93         TH1F            *fh1_recLJets_Et;
  94         TH1F            *fh1_genLJets_Et;
  95         TH2F            *fh2_recLJets_EtaPhi;
  96         TH2F            *fh2_genLJets_EtaPhi;
  97
  98         // FF, all jets
  99         TH2F            *fh2_recFF_JetEt;
 100         TH2F            *fh2_genFF_JetEt;
 101         TH2F            *fh2_recHumpBacked_JetEt;
 102         TH2F            *fh2_genHumpBacked_JetEt;
 103
 104
 105         // FF, all jets without acceptance cuts
 106         TH2F            *fh2_recFF_JetWoCEt;
 107         TH2F            *fh2_genFF_JetWoCEt;
 108         TH2F            *fh2_recHumpBacked_JetWoCEt;
 109         TH2F            *fh2_genHumpBacked_JetWoCEt;
 110
 111         // FF, leading jets
 112         TH2F            *fh2_recFF_LJetEt;
 113         TH2F            *fh2_genFF_LJetEt;
 114         TH2F            *fh2_recHumpBacked_LJetEt;
 115         TH2F            *fh2_genHumpBacked_LJetEt;
 116
 117
 118         // particles
 119         TH1F            *fh1_recPart_Pt;
 120         TH1F            *fh1_genPart_Pt;
 121         TH2F            *fh2_recPart_EtaPhi;
 122         TH2F            *fh2_genPart_EtaPhi;
 123
 124         TH1F            *fh1_recJetPart_Pt;
 125         TH1F            *fh1_genJetPart_Pt;
 126         TH2F            *fh2_recJetPart_EtaPhi;
 127         TH2F            *fh2_genJetPart_EtaPhi;
 128
 129         TH2F            *fh2_recJetPart_RJetPt;
 130         TH2F            *fh2_genJetPart_RJetPt;
 131
 132         TH1F            *fh1_recJetWoCPart_Pt;
 133         TH1F            *fh1_genJetWoCPart_Pt;
 134         TH2F            *fh2_recJetWoCPart_EtaPhi;
 135         TH2F            *fh2_genJetWoCPart_EtaPhi;
 136
 137         TH2F            *fh2_recJetWoCPart_RJetPt;
 138         TH2F            *fh2_genJetWoCPart_RJetPt;
 139
 140         TH1F            *fh1_recLJetPart_Pt;
 141         TH1F            *fh1_genLJetPart_Pt;
 142         TH2F            *fh2_recLJetPart_EtaPhi;
 143         TH2F            *fh2_genLJetPart_EtaPhi;
 144
 145         TH2F            *fh2_recLJetPart_RJetPt;
 146         TH2F            *fh2_genLJetPart_RJetPt;
 147
 148         TH3F            *fh3_recPart_EtaPhiPt;
 149
 150         ClassDef(AliAnalysisTaskFragFuncBB, 1);
 151 };
 152
 153 #endif