TENDER becomes Tender, removing .so
[u/mrichter/AliRoot.git] / PWGCF / FLOW / macros / runTaskFlowOnMC.C
CommitLineData
3ccea5c3 1enum anaModes {mLocal,mPROOF,mGrid};
2//mLocal: Analyze locally files in your computer using aliroot
3//mPROOF: Analyze CAF files with PROOF
4//mGrid: Analyze files on Grid via AliEn plug-in and using precompiled FLOW libraries
5
6// CENTRALITY DEFINITION
7//Int_t binfirst = 4; //where do we start numbering bins
8//Int_t binlast = 6; //where do we stop numbering bins
9//const Int_t numberOfCentralityBins = 9;
10//Int_t binfirst = 1; //where do we start numbering bins
11//Int_t binlast = 2; //where do we stop numbering bins
12//const Int_t numberOfCentralityBins = 9;
13//Float_t centralityArray[numberOfCentralityBins+1] = {0.,5.,10.,20.,30.,40.,50.,60.,70.,80.}; // in centrality percentile
14//Int_t centralityArray[numberOfCentralityBins+1] = {41,80,146,245,384,576,835,1203,1471,10000}; // in terms of TPC only reference multiplicity
15
16TString commonOutputFileName = "output"; // e.g.: result for centrality bin 0 will be in the file "outputCentrality0.root", etc
17
cd049c51 18void runTaskFlowOnMC(Int_t harmonic = 2,
19 Double_t gImpactParameterMin = 0.0,
20 Double_t gImpactParameterMax = 100.0,
21 Int_t gRefMultMin = 0,
22 Int_t gRefMultMax = 100,
23 Double_t etaMax = 1.3,
3ccea5c3 24 Double_t ptMin = 0.05,
25 Double_t ptMax = 20.0,
26 Int_t chargePOI = 0,
cd049c51 27 Int_t mode = mLocal,
3ccea5c3 28 Bool_t useFlowParFiles = kFALSE,
29 Bool_t useTender = kFALSE) {
30 // Time:
31 TStopwatch timer;
32 timer.Start();
33
34 // Load needed libraries:
35 LoadLibraries(mode,useFlowParFiles);
cd049c51 36
37 //Fix the filename
38 commonOutputFileName += ".etaMax.";
39 commonOutputFileName += etaMax;
40 commonOutputFileName += ".harmonics.";
41 commonOutputFileName += harmonic;
3ccea5c3 42
43 // Create analysis manager:
44 AliAnalysisManager *mgr = new AliAnalysisManager("FlowAnalysisManager");
45
46 // Chains:
47 if(mode == mLocal) {
48 TChain *chain = new TChain("TE");
49 chain->Add("galice.root");
50 }
51
52 // Connect plug-in to the analysis manager:
53 if(mode == mGrid) {
54 gROOT->LoadMacro("CreateAlienHandler.C");
55 AliAnalysisGrid *alienHandler = CreateAlienHandler();
56 if(!alienHandler) return;
57 mgr->SetGridHandler(alienHandler);
58 }
59
60 // Event handlers:
61 AliVEventHandler* esdH = new AliESDInputHandler;
62 mgr->SetInputEventHandler(esdH);
63 AliMCEventHandler *mc = new AliMCEventHandler();
64 mc->SetReadTR(kFALSE);
65 mgr->SetMCtruthEventHandler(mc);
66
67 if (useTender) {
68 //Add the tenders
69 gROOT->LoadMacro("$ALICE_ROOT/PWG2/FLOW/macros/AddTaskTenderFlow.C");
70 AddTaskTenderFlow();
71 }
72
73 // Setup analysis per centrality bin:
74 gROOT->LoadMacro("AddTaskFlowOnMC.C");
75 AddTaskFlowOnMC(commonOutputFileName,
76 etaMax,
77 ptMin,
78 ptMax,
79 chargePOI,
cd049c51 80 harmonic,
81 gImpactParameterMin,
82 gImpactParameterMax,
83 gRefMultMin,
84 gRefMultMax);
3ccea5c3 85
86 // Enable debug printouts:
87 mgr->SetDebugLevel(2);
88 // Run the analysis:
89 if(!mgr->InitAnalysis()) return;
90 mgr->PrintStatus();
91 if(mode == mLocal) {
92 mgr->StartAnalysis("local",chain);
93 }
94 else if(mode == mPROOF) {
95 mgr->StartAnalysis("proof",dataDir,nEvents,offset);
96 }
97 else if(mode == mGrid) {
98 mgr->StartAnalysis("grid");
99 }
100
101 // Print real and CPU time used for analysis:
102 timer.Stop();
103 timer.Print();
104
105} // end of void runTaskFlowOnMC(...)
106
107//===============================================================================================
108/*
109void CrossCheckUserSettings(Bool_t bData)
110{
111 // Check in this method if the user settings make sense.
112 if(LYZ1SUM && LYZ2SUM) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
113 if(LYZ1PROD && LYZ2PROD) {cout<<" WARNING: You cannot run LYZ1 and LYZ2 at the same time! LYZ2 needs the output from LYZ1 !!!!"<<endl; exit(0); }
114 if(LYZ2SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ2 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
115 if(LYZ1SUM && LYZEP) {cout<<" WARNING: You cannot run LYZ1 and LYZEP at the same time! LYZEP needs the output from LYZ2 !!!!"<<endl; exit(0); }
116} // end of void CrossCheckUserSettings()
117*/
118//===============================================================================================
119
120void LoadLibraries(const anaModes mode, Bool_t useFlowParFiles ) {
121 //--------------------------------------
122 // Load the needed libraries most of them already loaded by aliroot
123 //--------------------------------------
124
125 gSystem->Load("libCore");
126 gSystem->Load("libTree");
127 gSystem->Load("libGeom");
128 gSystem->Load("libVMC");
129 gSystem->Load("libXMLIO");
130 gSystem->Load("libPhysics");
131 gSystem->Load("libXMLParser");
132 gSystem->Load("libProof");
133 gSystem->Load("libMinuit");
134
135 if (mode==mLocal || mode==mGrid)
136 {
137 gSystem->Load("libSTEERBase");
138 gSystem->Load("libCDB");
139 gSystem->Load("libRAWDatabase");
140 gSystem->Load("libRAWDatarec");
141 gSystem->Load("libESD");
142 gSystem->Load("libAOD");
143 gSystem->Load("libSTEER");
144 gSystem->Load("libANALYSIS");
145 gSystem->Load("libANALYSISalice");
146 gSystem->Load("libTPCbase");
147 gSystem->Load("libTOFbase");
148 gSystem->Load("libTOFsim");
149 gSystem->Load("libTOFrec");
150 gSystem->Load("libTRDbase");
151 gSystem->Load("libVZERObase");
152 gSystem->Load("libVZEROrec");
153 gSystem->Load("libT0base");
154 gSystem->Load("libT0rec");
230b7aff 155 gSystem->Load("libTender");
156 gSystem->Load("libTenderSupplies");
3ccea5c3 157
158 if (useFlowParFiles)
159 {
160 AliAnalysisAlien::SetupPar("PWGflowBase");
161 AliAnalysisAlien::SetupPar("PWGflowTasks");
162 }
163 else
164 {
165 gSystem->Load("libPWGflowBase");
166 gSystem->Load("libPWGflowTasks");
167 }
168 }
169 else if (mode==mPROOF)
170 {
171 TList* list = new TList();
172 list->Add(new TNamed("ALIROOT_MODE", "ALIROOT"));
173 if (useFlowParFiles)
230b7aff 174 list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies"));
3ccea5c3 175 else
230b7aff 176 list->Add(new TNamed("ALIROOT_EXTRA_LIBS", "ANALYSIS:ANALYSISalice:Tender:TenderSupplies:PWG2flowCommon:PWG2flowTasks"));
3ccea5c3 177
178 //list->Add(new TNamed("ALIROOT_EXTRA_INCLUDES","PWG2/FLOW/AliFlowCommon:PWG2/FLOW/AliFlowTasks"));
179
180 // Connect to proof
181 printf("*** Connect to PROOF ***\n");
182 gEnv->SetValue("XSec.GSI.DelegProxy","2");
183 //TProof* proof = TProof::Open("alice-caf.cern.ch");
184 TProof* proof = TProof::Open("skaf.saske.sk");
185
186 // list the data available
187 //gProof->ShowDataSets("/*/*");
188 //gProof->ShowDataSets("/alice/sim/"); //for MC Data
189 //gProof->ShowDataSets("/alice/data/"); //for REAL Data
190
191 proof->ClearPackages();
192 proof->EnablePackage("VO_ALICE@AliRoot::v4-21-14-AN",list);
193
194 if (useFlowParFiles)
195 {
196 gProof->UploadPackage("PWG2flowCommon.par");
197 gProof->UploadPackage("PWG2flowTasks.par");
198 }
199
200 // Show enables Packages
201 gProof->ShowEnabledPackages();
202 }
203} // end of void LoadLibraries(const anaModes mode)
204
205// Helper macros for creating chains
206// from: CreateESDChain.C,v 1.10 jgrosseo Exp
207TChain* CreateESDChain(const char* aDataDir, Int_t aRuns, Int_t offset)
208{
209 // creates chain of files in a given directory or file containing a list.
210 // In case of directory the structure is expected as:
211 // <aDataDir>/<dir0>/AliESDs.root
212 // <aDataDir>/<dir1>/AliESDs.root
213 // ...
214
215 if (!aDataDir)
216 return 0;
217
218 Long_t id, size, flags, modtime;
219 if (gSystem->GetPathInfo(aDataDir, &id, &size, &flags, &modtime))
220 {
221 printf("%s not found.\n", aDataDir);
222 return 0;
223 }
224
225 TChain* chain = new TChain("esdTree");
226 TChain* chaingAlice = 0;
227
228 if (flags & 2)
229 {
230 TString execDir(gSystem->pwd());
231 TSystemDirectory* baseDir = new TSystemDirectory(".", aDataDir);
232 TList* dirList = baseDir->GetListOfFiles();
233 Int_t nDirs = dirList->GetEntries();
234 gSystem->cd(execDir);
235
236 Int_t count = 0;
237
238 for (Int_t iDir=0; iDir<nDirs; ++iDir)
239 {
240 TSystemFile* presentDir = (TSystemFile*) dirList->At(iDir);
241 if (!presentDir || !presentDir->IsDirectory() || strcmp(presentDir->GetName(), ".") == 0 || strcmp(presentDir->GetName(), "..") == 0)
242 continue;
243
244 if (offset > 0)
245 {
246 --offset;
247 continue;
248 }
249
250 if (count++ == aRuns)
251 break;
252
253 TString presentDirName(aDataDir);
254 presentDirName += "/";
255 presentDirName += presentDir->GetName();
256 chain->Add(presentDirName + "/AliESDs.root/esdTree");
257 // cerr<<presentDirName<<endl;
258 }
259
260 }
261 else
262 {
263 // Open the input stream
264 ifstream in;
265 in.open(aDataDir);
266
267 Int_t count = 0;
268
269 // Read the input list of files and add them to the chain
270 TString esdfile;
271 while(in.good())
272 {
273 in >> esdfile;
274 if (!esdfile.Contains("root")) continue; // protection
275
276 if (offset > 0)
277 {
278 --offset;
279 continue;
280 }
281
282 if (count++ == aRuns)
283 break;
284
285 // add esd file
286 chain->Add(esdfile);
287 }
288
289 in.close();
290 }
291
292 return chain;
293
294} // end of TChain* CreateESDChain(const char* aDataDir, Int_t aRuns, Int_t offset)
295
296//===============================================================================================
297
298TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
299{
300 // creates chain of files in a given directory or file containing a list.
301 // In case of directory the structure is expected as:
302 // <aDataDir>/<dir0>/AliAOD.root
303 // <aDataDir>/<dir1>/AliAOD.root
304 // ...
305
306 if (!aDataDir)
307 return 0;
308
309 Long_t id, size, flags, modtime;
310 if (gSystem->GetPathInfo(aDataDir, &id, &size, &flags, &modtime))
311 {
312 printf("%s not found.\n", aDataDir);
313 return 0;
314 }
315
316 TChain* chain = new TChain("aodTree");
317 TChain* chaingAlice = 0;
318
319 if (flags & 2)
320 {
321 TString execDir(gSystem->pwd());
322 TSystemDirectory* baseDir = new TSystemDirectory(".", aDataDir);
323 TList* dirList = baseDir->GetListOfFiles();
324 Int_t nDirs = dirList->GetEntries();
325 gSystem->cd(execDir);
326
327 Int_t count = 0;
328
329 for (Int_t iDir=0; iDir<nDirs; ++iDir)
330 {
331 TSystemFile* presentDir = (TSystemFile*) dirList->At(iDir);
332 if (!presentDir || !presentDir->IsDirectory() || strcmp(presentDir->GetName(), ".") == 0 || strcmp(presentDir->GetName(), "..") == 0)
333 continue;
334
335 if (offset > 0)
336 {
337 --offset;
338 continue;
339 }
340
341 if (count++ == aRuns)
342 break;
343
344 TString presentDirName(aDataDir);
345 presentDirName += "/";
346 presentDirName += presentDir->GetName();
347 chain->Add(presentDirName + "/AliAOD.root/aodTree");
348 // cerr<<presentDirName<<endl;
349 }
350
351 }
352 else
353 {
354 // Open the input stream
355 ifstream in;
356 in.open(aDataDir);
357
358 Int_t count = 0;
359
360 // Read the input list of files and add them to the chain
361 TString aodfile;
362 while(in.good())
363 {
364 in >> aodfile;
365 if (!aodfile.Contains("root")) continue; // protection
366
367 if (offset > 0)
368 {
369 --offset;
370 continue;
371 }
372
373 if (count++ == aRuns)
374 break;
375
376 // add aod file
377 chain->Add(aodfile);
378 }
379
380 in.close();
381 }
382
383 return chain;
384
385} // end of TChain* CreateAODChain(const char* aDataDir, Int_t aRuns, Int_t offset)
386