Loading libEventMixing in macros
[u/mrichter/AliRoot.git] / PWGCF / EBYE / macros / runBalanceFunction.C
1 // run.C\r
2 //\r
3 // Template run macro for AliBasicTask.cxx/.h with example layout of\r
4 // physics selections and options, in macro and task.\r
5 //\r
6 // Author: Arvinder Palaha\r
7 //\r
8 class AliAnalysisGrid;\r
9 class AliAnalysisTaskBF;\r
10 class AliBalance;\r
11 \r
12 //Centrality stuff\r
13 Int_t binfirst = 0;  //where do we start numbering bins\r
14 Int_t binlast = 8;  //where do we stop numbering bins\r
15 const Int_t numberOfCentralityBins = 9;\r
16 Float_t centralityArray[numberOfCentralityBins+1] = {0.,5.,10.,20.,30.,40.,50.,60.,70.,80.}; // in centrality percentile\r
17 \r
18 //Systematic studies\r
19 const Int_t numberOfSyst = 13;\r
20 Float_t vZ[numberOfSyst]     = {10.,12.,6.,8.,10.,10.,10.,10.,10.,10.,10.,10.,10.};     // global Vertex Z cut\r
21 Float_t DCAxy[numberOfSyst]  = {-1.,2.4,2.4,2.4,2.2,2.0,1.8,2.4,2.4,2.4,2.4,2.4,2.4};   // DCA xy cut (afterburner, -1 = w/o additional cut)\r
22 Float_t DCAz[numberOfSyst]   = {-1.,3.2,3.2,3.2,3.0,2.8,2.6,3.2,3.2,3.2,3.2,3.2,3.2};   // DCA z cut (afterburner, -1 = w/o additional cut)\r
23 Float_t ptMin[numberOfSyst]  = {0.3,0.3,0.3,0.3,0.3,0.3,0.3,1.5,5.0,0.3,0.3,0.3,0.3};   // pt cuts\r
24 Float_t ptMax[numberOfSyst]  = {1.5,1.5,1.5,1.5,1.5,1.5,1.5,5.0,10.0,10.0,1.5,1.5,1.5}; // pt cuts\r
25 Float_t etaMin[numberOfSyst] = {-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-0.8,-1.0,-0.6,-0.4}; // eta cuts\r
26 Float_t etaMax[numberOfSyst] = {0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,0.8,1.0,0.6,0.4};   // eta cuts\r
27 \r
28 Bool_t kUsePID = kFALSE;\r
29 \r
30 //______________________________________________________________________________\r
31 void runBalanceFunction(\r
32          const char* runtype = "local", // local, proof or grid\r
33          const char *gridmode = "test", // Set the run mode (can be "full", "test", "offline", "submit" or "terminate"). Full & Test work for proof\r
34          const Int_t bunchN = 0,\r
35          const bool bAOD = 1, // 1 = AOD ANALYSIS, 0 = ESD ANALYSIS\r
36          const bool bMCtruth = 0, // 1 = MCEvent handler is on (MC truth), 0 = MCEvent handler is off (MC reconstructed/real data)\r
37          const bool bMCphyssel = 0, // 1 = looking at MC truth or reconstructed, 0 = looking at real data\r
38          const Long64_t nentries = 50000, // for local and proof mode, ignored in grid mode. Set to 1234567890 for all events.\r
39          const Long64_t firstentry = 0, // for local and proof mode, ignored in grid mode\r
40          TString proofdataset = "bunchPROOF", // path to dataset on proof cluster, for proof analysis\r
41          const char *proofcluster = "miweber@alice-caf.cern.ch", // which proof cluster to use in proof mode\r
42          const char *taskname = "BF_Syst_Test" // sets name of grid generated macros\r
43          )\r
44 {\r
45     // check run type\r
46     if(runtype != "local" && runtype != "proof" && runtype != "grid"){\r
47         Printf("\n\tIncorrect run option, check first argument of run macro");\r
48         Printf("\tint runtype = local, proof or grid\n");\r
49         return;\r
50     }\r
51     Printf("%s analysis chosen",runtype);\r
52   \r
53     // load libraries\r
54     gSystem->Load("libCore.so");        \r
55     gSystem->Load("libGeom.so");\r
56     gSystem->Load("libVMC.so");\r
57     gSystem->Load("libPhysics.so");\r
58     gSystem->Load("libTree.so");\r
59     gSystem->Load("libSTEERBase.so");\r
60     gSystem->Load("libESD.so");\r
61     gSystem->Load("libAOD.so");\r
62     gSystem->Load("libANALYSIS.so");\r
63     gSystem->Load("libANALYSISalice.so");\r
64     gSystem->Load("libEventMixing.so");\r
65     gSystem->Load("libPWGCFebye.so");\r
66 \r
67     // additional\r
68 \r
69     // compile standalone stuff\r
70     //gROOT->LoadMacro("AliBalance.cxx++g");\r
71     //gROOT->LoadMacro("AliAnalysisTaskBF.cxx++g");\r
72 \r
73     // add aliroot indlude path\r
74     //gROOT->ProcessLine(".include $PWD/.");\r
75     //gROOT->ProcessLine(Form(".include %s/include",gSystem->ExpandPathName("$ALICE_ROOT")));\r
76 \r
77     gROOT->SetStyle("Plain");\r
78 \r
79     // analysis manager\r
80     AliAnalysisManager* mgr = new AliAnalysisManager(Form("%s%i",taskname,bunchN));\r
81     \r
82     // create the alien handler and attach it to the manager\r
83     AliAnalysisGrid *plugin = CreateAlienHandler(bAOD,bunchN,Form("%s%i",taskname,bunchN), gridmode, proofcluster, Form("%s_%d.txt",proofdataset.Data(),bunchN)); \r
84     mgr->SetGridHandler(plugin);\r
85     \r
86 \r
87     // input handler (ESD or AOD)\r
88     AliVEventHandler* inputH = NULL;\r
89     if(!bAOD){\r
90       inputH = new AliESDInputHandler();\r
91     }\r
92     else{\r
93       inputH = new AliAODInputHandler();\r
94     }\r
95     mgr->SetInputEventHandler(inputH);\r
96     \r
97     // mc event handler\r
98     if(bMCtruth) {\r
99         AliMCEventHandler* mchandler = new AliMCEventHandler();\r
100         // Not reading track references\r
101         mchandler->SetReadTR(kFALSE);\r
102         mgr->SetMCtruthEventHandler(mchandler);\r
103     }   \r
104 \r
105     // AOD output handler\r
106     //AliAODHandler* aodoutHandler = new AliAODHandler();\r
107     //aodoutHandler->SetOutputFileName("aod.root");\r
108     //mgr->SetOutputEventHandler(aodoutHandler); \r
109     \r
110     // === Physics Selection Task ===\r
111     //\r
112     // In SelectCollisionCandidate(), default is kMB, so the task UserExec() \r
113     // function is only called for these events.\r
114     // Options are:\r
115     //    kMB             Minimum Bias trigger\r
116     //    kMBNoTRD        Minimum bias trigger where the TRD is not read out\r
117     //    kMUON           Muon trigger\r
118     //    kHighMult       High-Multiplicity Trigger\r
119     //    kUserDefined    For manually defined trigger selection\r
120     //\r
121     // Multiple options possible with the standard AND/OR operators && and ||\r
122     // These all have the usual offline SPD or V0 selections performed.\r
123     //\r
124     // With a pointer to the physics selection object using physSelTask->GetPhysicsSelection(),\r
125     // one can manually set the selected and background classes using:\r
126     //    AddCollisionTriggerClass("+CINT1B-ABCE-NOPF-ALL")\r
127     //    AddBGTriggerClass("+CINT1A-ABCE-NOPF-ALL");\r
128     //\r
129     // One can also specify multiple classes at once, or require a class to NOT\r
130     // trigger, for e.g.\r
131     //    AddBGTriggerClass("+CSMBA-ABCE-NOPF-ALL -CSMBB-ABCE-NOPF-ALL");\r
132     //\r
133     // NOTE that manually setting the physics selection overrides the standard\r
134     // selection, so it must be done in completeness.\r
135     //\r
136     // ALTERNATIVELY, one can make the physics selection inside the task\r
137     // UserExec().\r
138     // For this case, comment out the task->SelectCol.... line, \r
139     // and see AliBasicTask.cxx UserExec() function for details on this.\r
140 \r
141     //gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");\r
142     //AliPhysicsSelectionTask *physSelTask = AddTaskPhysicsSelection(bMCphyssel);\r
143     //if(!physSelTask) { Printf("no physSelTask"); return; }\r
144     //AliPhysicsSelection *physSel = physSelTask->GetPhysicsSelection();\r
145     //physSel->AddCollisionTriggerClass("+CINT1B-ABCE-NOPF-ALL");// #3119 #769");\r
146                 \r
147     // create task\r
148 \r
149     //Add the centrality determination task and the physics selection \r
150     // (only on ESD level, in AODs centrality is already in header and events are selected)\r
151     if(!bAOD){\r
152       gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");\r
153       AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();\r
154 \r
155       // Add physics selection task (NOT needed for AODs)\r
156       gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");\r
157       AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection(bMCphyssel);\r
158 \r
159       //Add the PID response\r
160       if(kUsePID) {\r
161         gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");\r
162         AddTaskPIDResponse(bMCphyssel); \r
163       }\r
164     }\r
165 \r
166     //Add the BF task (all centralities)\r
167     gROOT->LoadMacro("AddTaskBalanceCentralityTrain.C"); \r
168     AliAnalysisTaskBF *task = AddTaskBalanceCentralityTrain(0,100,0,"V0M",vZ[0],DCAxy[0],DCAz[0],ptMin[0],ptMax[0],etaMin[0],etaMax[0],-1,-1,kUsePID);\r
169     \r
170     // enable debug printouts\r
171     //mgr->SetDebugLevel(2);\r
172     //mgr->SetUseProgressBar(1,100);\r
173     if (!mgr->InitAnalysis()) return;\r
174     mgr->PrintStatus();\r
175   \r
176     // start analysis\r
177     Printf("Starting Analysis....");\r
178     mgr->StartAnalysis(runtype,nentries,firstentry);\r
179 }\r
180 \r
181 //______________________________________________________________________________\r
182 AliAnalysisGrid* CreateAlienHandler(Bool_t bAOD, Int_t bunchN, const char *taskname, const char *gridmode, const char *proofcluster, const char *proofdataset)\r
183 {\r
184     AliAnalysisAlien *plugin = new AliAnalysisAlien();\r
185     // Set the run mode (can be "full", "test", "offline", "submit" or "terminate")\r
186     plugin->SetRunMode(gridmode);\r
187 \r
188     // Set versions of used packages\r
189     plugin->SetAPIVersion("V1.1x");\r
190     plugin->SetROOTVersion("v5-28-00d");\r
191     plugin->SetAliROOTVersion("v5-02-05-AN");\r
192 \r
193     // Declare input data to be processed.\r
194 \r
195     // Method 1: Create automatically XML collections using alien 'find' command.\r
196     // Define production directory LFN\r
197     plugin->SetGridDataDir("/alice/data/2010/LHC10h/");\r
198     // On real reconstructed data:\r
199     // plugin->SetGridDataDir("/alice/data/2009/LHC09d");\r
200 \r
201     // Set data search pattern\r
202     //plugin->SetDataPattern("*ESDs.root"); // THIS CHOOSES ALL PASSES\r
203     // Data pattern for reconstructed data\r
204     if(!bAOD){\r
205       plugin->SetDataPattern("*ESDs/pass2/*ESDs.root"); // CHECK LATEST PASS OF DATA SET IN ALIENSH\r
206     } \r
207     else{\r
208       plugin->SetDataPattern("*ESDs/pass2/AOD049/*/AliAOD.root");\r
209     }\r
210 \r
211     plugin->SetRunPrefix("000");   // real data\r
212     // ...then add run numbers to be considered\r
213     //plugin->SetRunRange(114917,115322);\r
214 \r
215     if(bunchN==0){\r
216       plugin->AddRunNumber(137366);\r
217     }\r
218     \r
219     //bunch1\r
220     else if(bunchN == 1){\r
221       plugin->AddRunNumber(139510);\r
222       plugin->AddRunNumber(139507);\r
223       plugin->AddRunNumber(139505);\r
224       plugin->AddRunNumber(139503); \r
225       plugin->AddRunNumber(139465); \r
226       plugin->AddRunNumber(139438);\r
227       plugin->AddRunNumber(139437);\r
228       plugin->AddRunNumber(139360); \r
229       plugin->AddRunNumber(139329);\r
230       plugin->AddRunNumber(139328); \r
231     }\r
232 \r
233     //bunch2\r
234     else if(bunchN == 2){\r
235       plugin->AddRunNumber(139314); \r
236       plugin->AddRunNumber(139310);\r
237       plugin->AddRunNumber(139309); \r
238       plugin->AddRunNumber(139173); \r
239       plugin->AddRunNumber(139107); \r
240       plugin->AddRunNumber(139105); \r
241       plugin->AddRunNumber(139038); \r
242       plugin->AddRunNumber(139037); \r
243       plugin->AddRunNumber(139036); \r
244       plugin->AddRunNumber(139029); \r
245       plugin->AddRunNumber(139028); \r
246       plugin->AddRunNumber(138872); \r
247       plugin->AddRunNumber(138871); \r
248       plugin->AddRunNumber(138870); \r
249       plugin->AddRunNumber(138837); \r
250       plugin->AddRunNumber(138732); \r
251       plugin->AddRunNumber(138730);\r
252       plugin->AddRunNumber(138666);\r
253       plugin->AddRunNumber(138662); \r
254       plugin->AddRunNumber(138653); \r
255     }\r
256 \r
257     else if(bunchN == 3){\r
258       plugin->AddRunNumber(138652);\r
259       plugin->AddRunNumber(138638);\r
260       plugin->AddRunNumber(138624); \r
261       plugin->AddRunNumber(138621); \r
262       plugin->AddRunNumber(138583); \r
263       plugin->AddRunNumber(138582); \r
264       plugin->AddRunNumber(138579); \r
265       plugin->AddRunNumber(138578);\r
266       plugin->AddRunNumber(138534);\r
267       plugin->AddRunNumber(138469); \r
268     }\r
269 \r
270     else if(bunchN == 4){\r
271       \r
272       plugin->AddRunNumber(138442);\r
273       plugin->AddRunNumber(138439);\r
274       plugin->AddRunNumber(138438);\r
275       plugin->AddRunNumber(138396); \r
276       plugin->AddRunNumber(138364); \r
277       plugin->AddRunNumber(138275); \r
278       plugin->AddRunNumber(138225); \r
279       plugin->AddRunNumber(138201);\r
280       plugin->AddRunNumber(138197); \r
281       plugin->AddRunNumber(138192); \r
282     }\r
283 \r
284     else if(bunchN == 5){\r
285 \r
286       plugin->AddRunNumber(138190);\r
287       plugin->AddRunNumber(137848); \r
288       plugin->AddRunNumber(137844); \r
289       plugin->AddRunNumber(137752); \r
290       plugin->AddRunNumber(137751); \r
291       plugin->AddRunNumber(137724); \r
292       plugin->AddRunNumber(137722); \r
293       plugin->AddRunNumber(137718); \r
294       plugin->AddRunNumber(137704); \r
295       plugin->AddRunNumber(137693);\r
296     }\r
297 \r
298     else if(bunchN == 6){\r
299 \r
300       plugin->AddRunNumber(137692); \r
301       plugin->AddRunNumber(137691); \r
302       plugin->AddRunNumber(137686); \r
303       plugin->AddRunNumber(137685); \r
304       plugin->AddRunNumber(137639); \r
305       plugin->AddRunNumber(137638);\r
306       plugin->AddRunNumber(137608); \r
307       plugin->AddRunNumber(137595);\r
308       plugin->AddRunNumber(137549);\r
309       plugin->AddRunNumber(137546); \r
310 \r
311     }\r
312 \r
313     else if(bunchN == 7){\r
314 \r
315       plugin->AddRunNumber(137544); \r
316       plugin->AddRunNumber(137541); \r
317       plugin->AddRunNumber(137539); \r
318       plugin->AddRunNumber(137531); \r
319       plugin->AddRunNumber(137530); \r
320       plugin->AddRunNumber(137443); \r
321       plugin->AddRunNumber(137441); \r
322       plugin->AddRunNumber(137440); \r
323       plugin->AddRunNumber(137439); \r
324       plugin->AddRunNumber(137434); \r
325 \r
326     }\r
327 \r
328     else if(bunchN == 8){\r
329 \r
330       plugin->AddRunNumber(137432); \r
331       plugin->AddRunNumber(137431); \r
332       plugin->AddRunNumber(137430); \r
333       plugin->AddRunNumber(137366); \r
334       plugin->AddRunNumber(137243); \r
335       plugin->AddRunNumber(137236);\r
336       plugin->AddRunNumber(137235);\r
337       plugin->AddRunNumber(137232); \r
338       plugin->AddRunNumber(137231); \r
339       plugin->AddRunNumber(137162); \r
340       plugin->AddRunNumber(137161);\r
341     }\r
342 \r
343     else{\r
344 \r
345       stderr<<"BUNCH NOT THERE"<<endl;\r
346       return NULL;\r
347 \r
348     }\r
349 \r
350 \r
351     //plugin->AddRunList("139510, 139507, 139505, 139503, 139465, 139438, 139437, 139360, 139329, 139328, 139314, 139310, 139309, 139173, 139107, 139105, 139038, 139037, 139036, 139029, 139028, 138872, 138871, 138870, 138837, 138732, 138730, 138666, 138662, 138653, 138652, 138638, 138624, 138621, 138583, 138582, 138579, 138578, 138534, 138469, 138442, 138439, 138438, 138396, 138364, 138275, 138225, 138201, 138197, 138192, 138190, 137848, 137844, 137752, 137751, 137724, 137722, 137718, 137704, 137693, 137692, 137691, 137686, 137685, 137639, 137638, 137608, 137595, 137549, 137546, 137544, 137541, 137539, 137531, 137530, 137443, 137441, 137440, 137439, 137434, 137432, 137431, 137430, 137366, 137243, 137236, 137235, 137232, 137231, 137162, 137161");\r
352 \r
353 \r
354 \r
355 \r
356 \r
357     plugin->SetNrunsPerMaster(1);\r
358     plugin->SetOutputToRunNo();\r
359     // comment out the next line when using the "terminate" option, unless\r
360     // you want separate merged files for each run\r
361     plugin->SetMergeViaJDL();\r
362 \r
363     // Method 2: Declare existing data files (raw collections, xml collections, root file)\r
364     // If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())\r
365     // XML collections added via this method can be combined with the first method if\r
366     // the content is compatible (using or not tags)\r
367     //   plugin->AddDataFile("tag.xml");\r
368     //   plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");\r
369 \r
370     // Define alien work directory where all files will be copied. Relative to alien $HOME.\r
371     plugin->SetGridWorkingDir(taskname);\r
372 \r
373     // Declare alien output directory. Relative to working directory.\r
374     plugin->SetGridOutputDir("out"); // In this case will be $HOME/taskname/out\r
375 \r
376    // Declare the analysis source files names separated by blancs. To be compiled runtime\r
377     // using ACLiC on the worker nodes.\r
378     plugin->SetAnalysisSource("AliBalance.cxx AliAnalysisTaskBF.cxx");\r
379 \r
380     // Declare all libraries (other than the default ones for the framework. These will be\r
381     // loaded by the generated analysis macro. Add all extra files (task .cxx/.h) here.\r
382     //plugin->AddIncludePath("-I.");\r
383     //plugin->SetAdditionalLibs("libPWGCFebye.so");\r
384     plugin->SetAdditionalLibs("AliBalance.cxx AliBalance.h AliAnalysisTaskBF.cxx AliAnalysisTaskBF.h");\r
385 \r
386      // Declare the output file names separated by blancs.\r
387     // (can be like: file.root or file.root@ALICE::Niham::File)\r
388     // To only save certain files, use SetDefaultOutputs(kFALSE), and then\r
389     // SetOutputFiles("list.root other.filename") to choose which files to save\r
390     plugin->SetDefaultOutputs();\r
391     //plugin->SetOutputFiles("list.root");\r
392 \r
393     // Optionally set a name for the generated analysis macro (default MyAnalysis.C)\r
394     plugin->SetAnalysisMacro(Form("%s.C",taskname));\r
395 \r
396     // Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)\r
397     plugin->SetSplitMaxInputFileNumber(100);\r
398 \r
399     // Optionally modify the executable name (default analysis.sh)\r
400     plugin->SetExecutable(Form("%s.sh",taskname));\r
401 \r
402     // set number of test files to use in "test" mode\r
403     plugin->SetNtestFiles(1);\r
404 \r
405     // Optionally resubmit threshold.\r
406     plugin->SetMasterResubmitThreshold(90);\r
407 \r
408     // Optionally set time to live (default 30000 sec)\r
409     plugin->SetTTL(90000);\r
410 \r
411     // Optionally set input format (default xml-single)\r
412     plugin->SetInputFormat("xml-single");\r
413 \r
414     // Optionally modify the name of the generated JDL (default analysis.jdl)\r
415     plugin->SetJDLName(Form("%s.jdl",taskname));\r
416 \r
417     // Optionally modify job price (default 1)\r
418     plugin->SetPrice(1);      \r
419 \r
420     // Optionally modify split mode (default 'se')    \r
421     plugin->SetSplitMode("se");\r
422 \r
423     //plugin->SetUseSubmitPolicy();\r
424     //plugin->SetKeepLogs();\r
425     \r
426     //----------------------------------------------------------\r
427     //---      PROOF MODE SPECIFIC SETTINGS         ------------\r
428     //---------------------------------------------------------- \r
429     // Proof cluster\r
430     plugin->SetProofCluster(proofcluster);\r
431     // Dataset to be used   \r
432     plugin->SetProofDataSet(proofdataset);\r
433     // May need to reset proof. Supported modes: 0-no reset, 1-soft, 2-hard\r
434     plugin->SetProofReset(0);\r
435     // May limit number of workers\r
436     plugin->SetNproofWorkers(0);\r
437     // May limit the number of workers per slave\r
438     plugin->SetNproofWorkersPerSlave(1);   \r
439     // May use a specific version of root installed in proof\r
440     plugin->SetRootVersionForProof("current");\r
441     // May set the aliroot mode. Check http://aaf.cern.ch/node/83 \r
442     plugin->SetAliRootMode("default"); // Loads AF libs by default\r
443     // May request ClearPackages (individual ClearPackage not supported)\r
444     plugin->SetClearPackages(kFALSE);\r
445     // Plugin test mode works only providing a file containing test file locations, used in "local" mode also\r
446     plugin->SetFileForTestMode("files.txt"); // file should contain path name to a local directory containg *ESDs.root etc\r
447     // Request connection to alien upon connection to grid\r
448     plugin->SetProofConnectGrid(kFALSE);\r
449 \r
450     plugin->Print();\r
451 \r
452     return plugin;\r
453 }\r
454 \r