2 /*************************************************************************************************
3 *** Add Fragmentation Function Task ***
4 **************************************************************************************************
5 The fragmenation function task expects an ESD filter and jet finder running before this task.
6 Or it runs on delta-AODs filled with filtered tracks and jets before.
9 (char) recJetsBranch: branch in AOD for (reconstructed) jets
10 (char) genJetsBranch: branch in AOD for (generated) jets
11 (char) jetType: "AOD" jets from recJetsBranch
12 "AODMC" jets from genJetsBranch
13 "KINE" jets from PYCELL
14 +"b" (e.g. "AODb") jets with acceptance cuts
15 (char) trackType: "AOD" reconstructed tracks from AOD filled by ESD filter (choose filter mask!)
16 "AODMC" MC tracks from AOD filled by kine filter
17 "KINE" kine particles from MC event
18 +"2" (e.g. "AOD2") charged tracks only
19 +"b" (e.g. "AOD2b") with acceptance cuts
20 (UInt_t) filterMask: select filter bit of ESD filter task
22 ***************************************************************************************************/
27 AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=1, UInt_t filterMask=32){
29 AliAnalysisTaskFragmentationFunction *ff=0;
32 // only reconstructed (default)
33 if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jets", "", "", "", filterMask);
34 // charged MC tracks and jets
35 if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMC", "AODMC2", filterMask);
36 // charged MC tracks and jets with acceptance cuts
37 if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODMCb", "AODMC2b", filterMask);
38 // kine tracks in acceptance, pythia jets in acceptance
39 if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
40 // reconstructed charged tracks after cuts, MC jets in acceptance
41 if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jets", "jetsMC2b", "AODMCb", "AOD2b", filterMask);
42 // reconstruction efficiency: pointing with rec jet axis into gen tracks
43 if(iFlag&(1<<5)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA104", "AODb", "AODMC2b", filterMask);
47 if(iFlag&(1<<10)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "", "", "", filterMask);
48 // charged MC tracks and jets
49 if(iFlag&(1<<11)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMC", "AODMC2", filterMask);
50 // charged MC tracks and jets with acceptance cuts
51 if(iFlag&(1<<12)) ff = AddTaskFragmentationFunction("jetsAOD_FASTKT04", "jetsAODMC2_FASTKT04", "AODMCb", "AODMC2b", filterMask);
55 if(iFlag&(1<<20)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "", "", "", filterMask);
56 // charged MC tracks and jets
57 if(iFlag&(1<<21)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMC", "AODMC2", filterMask);
58 // charged MC tracks and jets with acceptance cuts
59 if(iFlag&(1<<22)) ff = AddTaskFragmentationFunction("jetsAOD_FASTJET04", "jetsAODMC2_FASTJET04", "AODMCb", "AODMC2b", filterMask);
66 // _______________________________________________________________________________________
68 AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(
69 const char* recJetsBranch,
70 const char* genJetsBranch,
72 const char* trackType,
75 // Creates a fragmentation function task,
76 // configures it and adds it to the analysis manager.
78 //******************************************************************************
79 //*** Configuration Parameter **************************************************
80 //******************************************************************************
82 // space for configuration parameter: histo bin, cuts, ...
83 // so far only default parameter used
85 Int_t debug = -1; // debug level, -1: not set here
87 //******************************************************************************
91 // Get the pointer to the existing analysis manager via the static access method.
92 //==============================================================================
93 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
95 ::Error("AddTaskFragmentationFunction", "No analysis manager to connect to.");
99 // Check the analysis type using the event handlers connected to the analysis manager.
100 //==============================================================================
101 if (!mgr->GetInputEventHandler()) {
102 ::Error("AddTaskFragmentationFunction", "This task requires an input event handler");
106 TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
107 Printf("Data Type: %s", type.Data());
109 TString branchRecJets(recJetsBranch);
110 TString branchGenJets(genJetsBranch);
111 TString typeJets(jetType);
112 TString typeTracks(trackType);
114 if(branchRecJets.Length()==0) branchRecJets = "noRecJets";
115 if(branchGenJets.Length()==0) branchGenJets = "noGenJets";
116 if(typeTracks.Length()==0) typeTracks = "trackTypeUndef";
117 if(typeJets.Length()==0) typeJets = "jetTypeUndef";
119 // Create the task and configure it.
120 //===========================================================================
122 AliAnalysisTaskFragmentationFunction *task = new AliAnalysisTaskFragmentationFunction(
123 Form("Fragmenation Function %s %s %s %s", branchRecJets.Data(), branchGenJets.Data(), typeJets.Data(), typeTracks.Data()));
125 if(debug>=0) task->SetDebugLevel(debug);
127 Printf("Rec Jets %s", branchRecJets.Data());
128 Printf("Gen Jets %s", branchGenJets.Data());
129 Printf("Jet Type %s", typeJets.Data());
130 Printf("Track Type %s", typeTracks.Data());
132 if(!branchRecJets.Contains("noRecJets")) task->SetBranchRecJets(branchRecJets);
133 if(!branchGenJets.Contains("noGenJets")) task->SetBranchGenJets(branchGenJets);
136 if(typeTracks.Contains("AODMC2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCChargedAcceptance);
137 else if(typeTracks.Contains("AODMC2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCCharged);
138 else if(typeTracks.Contains("AODMC")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCAll);
139 else if(typeTracks.Contains("KINE2b")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineChargedAcceptance);
140 else if(typeTracks.Contains("KINE2")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineCharged);
141 else if(typeTracks.Contains("KINE")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineAll);
142 else if(typeTracks.Contains("AODb")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODCuts);
143 else if(typeTracks.Contains("AOD")) task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAOD);
144 else if(typeTracks.Contains("trackTypeUndef")) task->SetTrackTypeGen(0); // undefined
145 else Printf("trackType %s not found", typeTracks.Data());
147 if(typeJets.Contains("AODMCb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance);
148 else if(typeJets.Contains("AODMC")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGen);
149 else if(typeJets.Contains("KINEb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKineAcceptance);
150 else if(typeJets.Contains("KINE")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKine);
151 else if(typeJets.Contains("AODb")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
152 else if(typeJets.Contains("AOD")) task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRec);
153 else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined
154 else Printf("jetType %s not found", typeJets.Data());
156 if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance
157 else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
158 else task->SetJetTypeRecEff(0);
160 task->SetFilterMask(filterMask);
162 // Set default parameters
164 task->SetTrackCuts(); // default : pt > 0.150 GeV, |eta|<0.9, full phi acc
165 task->SetJetCuts(); // default: jet pt > 5 GeV, |eta|<0.5, full phi acc
166 task->SetDiJetCuts(); // default: type of cut = 1 (cut in deltaPhi), deltaPhi = 0., cdf = 0.5, fraction of pt = 0.6
167 task->SetKindSlices(); // default: kindSlice = 1 (inv mass)
168 task->SetFFRadius(); // default: R = 0.4
169 task->SetFFBckgRadius(); // default: R = 0.7
170 task->SetBckgMode(); // default: bgMode = 1
171 task->SetHighPtThreshold(); // default: pt > 5 Gev
172 task->UseRecEffRecJetPtBins(); // efficiency in bins of rec/gen jet pt - default: kTRUE
175 task->SetFFHistoBins();
176 task->SetQAJetHistoBins();
177 task->SetQATrackHistoBins();
178 task->SetIJHistoBins();
179 task->SetDiJetHistoBins();
180 task->SetQADiJetHistoBins();
184 // Create ONLY the output containers for the data produced by the task.
185 // Get and connect other common input/output containers via the manager as below
186 //==============================================================================
188 AliAnalysisDataContainer *coutput_FragFunc = mgr->CreateContainer(
189 Form("fracfunc_%s_%s_%s_%s", branchRecJets.Data(), branchGenJets.Data(), typeTracks.Data(), typeJets.Data()),
191 AliAnalysisManager::kOutputContainer,
192 Form("%s:PWG4_FragmentationFunction_%s_%s_%s_%s",
193 AliAnalysisManager::GetCommonFileName(), branchRecJets.Data(), branchGenJets. Data(), typeTracks.Data(), typeJets.Data()));
195 mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer());
196 mgr->ConnectOutput (task, 0, mgr->GetCommonOutputContainer());
197 mgr->ConnectOutput (task, 1, coutput_FragFunc);