Bastian Bathen:
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskFragmentationFunction.C
1
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.
7
8 ** Parameters **
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
21
22 ***************************************************************************************************/
23
24
25
26
27 AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(UInt_t iFlag=1, UInt_t filterMask=16){
28         
29         AliAnalysisTaskFragmentationFunction *ff=0;
30
31         // only reconstructed (default)
32         if(iFlag&(1<<0)) ff = AddTaskFragmentationFunction("jets", "", "", "", filterMask);
33         // charged MC tracks and jets
34         if(iFlag&(1<<1)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA04", "AODMC", "AODMC2", filterMask);
35         // charged MC tracks and jets with acceptance cuts
36         if(iFlag&(1<<2)) ff = AddTaskFragmentationFunction("jets", "jetsAODMC2_UA04", "AODMCb", "AODMC2b", filterMask);
37         // kine tracks in acceptance, pythia jets in acceptance
38         if(iFlag&(1<<3)) ff = AddTaskFragmentationFunction("jets", "", "KINEb", "KINEb", filterMask);
39         // reconstructed charged tracks after cuts, MC jets in acceptance 
40         if(iFlag&(1<<4)) ff = AddTaskFragmentationFunction("jets", "jetsMC2b", "AODMCb", "AOD2b", filterMask);
41         
42         return ff;
43 }
44
45 // _______________________________________________________________________________________
46
47 AliAnalysisTaskFragmentationFunction *AddTaskFragmentationFunction(
48         const char* recJetsBranch,
49         const char* genJetsBranch,
50         const char* jetType,
51         const char* trackType,
52         UInt_t filterMask)
53 {
54    // Creates a fragmentation function task,
55    // configures it and adds it to the analysis manager.
56
57    //******************************************************************************
58    //*** Configuration Parameter **************************************************
59    //******************************************************************************
60
61    // space for configuration parameter: histo bin, cuts, ...
62    // so far only default parameter used
63
64    Int_t debug = -1; // debug level, -1: not set here
65
66    //******************************************************************************
67
68
69    
70    // Get the pointer to the existing analysis manager via the static access method.
71    //==============================================================================
72    AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
73    if (!mgr) {
74           ::Error("AddTaskFragmentationFunction", "No analysis manager to connect to.");
75           return NULL;
76    }
77    
78    // Check the analysis type using the event handlers connected to the analysis manager.
79    //==============================================================================
80    if (!mgr->GetInputEventHandler()) {
81          ::Error("AddTaskFragmentationFunction", "This task requires an input event handler");
82           return NULL;
83    }
84
85    TString type = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
86    Printf("Data Type: %s", type.Data());
87
88    TString branchRecJets(recJetsBranch);
89    TString branchGenJets(genJetsBranch);
90    TString typeJets(jetType);
91    TString typeTracks(trackType);
92
93    if(branchRecJets.Length()==0) branchRecJets = "noRecJets";
94    if(branchGenJets.Length()==0) branchGenJets = "noGenJets";
95    if(typeTracks.Length()==0) typeTracks = "trackTypeUndef";
96    if(typeJets.Length()==0)   typeJets   = "jetTypeUndef";
97    
98    // Create the task and configure it.
99    //===========================================================================
100
101    AliAnalysisTaskFragmentationFunction *task = new AliAnalysisTaskFragmentationFunction(
102         Form("Fragmenation Function %s %s %s %s", branchRecJets.Data(), branchGenJets.Data(), typeJets.Data(), typeTracks.Data()));
103    
104    if(debug>=0) task->SetDebugLevel(debug);
105    
106    Printf("Rec Jets %s", branchRecJets.Data());
107    Printf("Gen Jets %s", branchGenJets.Data());
108    Printf("Jet Type %s", typeJets.Data());
109    Printf("Track Type %s", typeTracks.Data());
110    
111    if(!branchRecJets.Contains("noRecJets")) task->SetBranchRecJets(branchRecJets);
112    if(!branchGenJets.Contains("noGenJets")) task->SetBranchGenJets(branchGenJets);
113
114
115    if(typeTracks.Contains("AODMC2b"))      task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCChargedAcceptance);
116    else if(typeTracks.Contains("AODMC2"))  task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCCharged);
117    else if(typeTracks.Contains("AODMC"))   task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODMCAll);
118    else if(typeTracks.Contains("KINE2b"))  task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineChargedAcceptance);
119    else if(typeTracks.Contains("KINE2"))   task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineCharged);
120    else if(typeTracks.Contains("KINE"))    task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackKineAll);
121    else if(typeTracks.Contains("AODb"))    task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAODCuts);
122    else if(typeTracks.Contains("AOD"))     task->SetTrackTypeGen(AliAnalysisTaskFragmentationFunction::kTrackAOD);
123    else if(typeTracks.Contains("trackTypeUndef")) task->SetTrackTypeGen(0); // undefined
124    else Printf("trackType %s not found", typeTracks.Data());
125
126    if(typeJets.Contains("AODMCb"))         task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance);
127    else if(typeJets.Contains("AODMC"))     task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsGen);
128    else if(typeJets.Contains("KINEb"))     task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKineAcceptance);
129    else if(typeJets.Contains("KINE"))      task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsKine);
130    else if(typeJets.Contains("AODb"))      task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance);
131    else if(typeJets.Contains("AOD"))       task->SetJetTypeGen(AliAnalysisTaskFragmentationFunction::kJetsRec);
132    else if(typeJets.Contains("jetTypeUndef")) task->SetJetTypeGen(0); // undefined
133    else Printf("jetType %s not found", typeJets.Data());
134    
135    if(typeJets.Contains("AODMCb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsGenAcceptance); // kJetsRecAcceptance
136    else if(typeJets.Contains("AODb")) task->SetJetTypeRecEff(AliAnalysisTaskFragmentationFunction::kJetsRecAcceptance); 
137    else task->SetJetTypeRecEff(0);
138
139    task->SetFilterMask(filterMask);
140   
141    // Set default parameters 
142    // Cut selection 
143    task->SetTrackCuts();       // default : pt > 0.150 GeV, |eta|<0.9, full phi acc
144    task->SetJetCuts();         // default: jet pt > 5 GeV, |eta|<0.5, full phi acc
145    task->SetDiJetCuts();       // default: type of cut = 1 (cut in deltaPhi), deltaPhi = 0., cdf = 0.5, fraction of pt = 0.6
146    task->SetKindSlices();      // default: kindSlice = 1 (inv mass)
147    task->SetFFRadius();        // default: R = 0.4
148    task->SetHighPtThreshold(); // default: pt > 5 Gev
149    // Define histo bins
150    task->SetFFHistoBins();
151    task->SetQAJetHistoBins();
152    task->SetQATrackHistoBins();
153    task->SetIJHistoBins();
154    task->SetDiJetHistoBins();
155    task->SetQADiJetHistoBins();
156
157    mgr->AddTask(task);
158
159    // Create ONLY the output containers for the data produced by the task.
160    // Get and connect other common input/output containers via the manager as below
161    //==============================================================================
162
163    AliAnalysisDataContainer *coutput_FragFunc = mgr->CreateContainer(
164       Form("fracfunc_%s_%s_%s_%s", branchRecJets.Data(), branchGenJets.Data(), typeTracks.Data(), typeJets.Data()),
165       TList::Class(),
166       AliAnalysisManager::kOutputContainer,
167       Form("%s:PWG4_FragmentationFunction_%s_%s_%s_%s", 
168          AliAnalysisManager::GetCommonFileName(), branchRecJets.Data(), branchGenJets. Data(), typeTracks.Data(), typeJets.Data()));
169
170    mgr->ConnectInput  (task, 0, mgr->GetCommonInputContainer());
171    mgr->ConnectOutput (task, 0, mgr->GetCommonOutputContainer());
172    mgr->ConnectOutput (task, 1, coutput_FragFunc);
173    
174    return task;
175 }