]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGLF/FORWARD/analysis2/sim/AOD.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
.so cleanup: no .so ext to GetLibraries()
[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / sim / AOD.C
CommitLineData
f8b7a926 1/**
2 * @file AOD.C
3 * @author Christian Holm Christensen <cholm@nbi.dk>
4 * @date Wed Sep 24 15:02:00 2014
5 *
6 * @brief Master script for AOD production.
7 *
8 * @note Do not modify this script.
9 *
421f877c 10 * This script reads in 4 other scripts
f8b7a926 11 *
12 * - GRP.C to load the global run parameters for the selected run,
13 * such as collision system, energy, etc.
14 *
15 * - AODConfig.C which defines a number of functions that return
16 * either true or false. The tasks added depends on these functions
421f877c 17 *
18 * - BaseConfig.C which defines some base classes
19 *
20 * - DetConfig.C which defines which detectors are active and on.
21 *
22 * Users can customize QAConfig.C and DetConfig.C according to their
23 * needs
f8b7a926 24 */
25/** Path to CDB */
26const char *cdbPath = "raw://";
ee275c29 27Bool_t cholm = false;
f8b7a926 28/**
29 * Interface (pure virtual) that all configuration classes must
30 * implement.
31 */
32struct VirtualAODCfg
33{
34 /**
35 * @{
36 * @name Plug-in settings
37 * Settings that make sense when using the Alien plugin
38 */
39 /** @return Connect to CDB */
40 virtual Bool_t UseCDBconnect() const = 0;
41 /** @return use physics selection */
42 virtual Bool_t UsePhysicsSelection() const = 0;
43 /** @return use tender wagon */
44 virtual Bool_t UseTender() const = 0;
45 /** @return centrality */
46 virtual Bool_t UseCentrality() const = 0;
47 /** @return use V0 correction in tender */
48 virtual Bool_t UseV0tender() const = 0;
49 /** @return activate debugging */
50 virtual Bool_t UseDBG() const = 0;
51 /** @return use MC info */
52 virtual Bool_t UseMC() const = 0;
53 /** @return use Kinematics filter */
54 virtual Bool_t UseKFILTER() const = 0;
55 /** @return use track references */
56 virtual Bool_t UseTR() const = 0;
57 /** @return do not change */
58 virtual Bool_t UseCORRFW() const = 0;
59 /** @return use AOD tags */
60 virtual Bool_t UseAODTAGS() const = 0;
61 /** @return use sys info */
62 virtual Bool_t UseSysInfo() const = 0;
63 /* @} */
64
65 /**
66 * @{
67 * @name Modules
68 * Analysis modules to be included. Some may not be yet fully implemented.
69 */
70 /** @return Analysis produces an AOD or dAOD's */
71 virtual Bool_t UseAODhandler() const = 0;
72 /** @return ESD to AOD filter (barrel + muon tracks) */
73 virtual Bool_t UseESDfilter() const = 0;
74 /** @return Use Muon train */
75 virtual Bool_t UsePWGMuonTrain() const = 0;
76 /** @return Task that copies only muon events */
77 virtual Bool_t UseMUONcopyAOD() const = 0;
78 /** @return Jet analysis (PWG4) */
79 virtual Bool_t UseJETAN() const = 0;
80 /** @return Jet delta AODs */
81 virtual Bool_t UseJETANdelta() const = 0;
82 /** @return Vertexing HF task (PWG3) */
83 virtual Bool_t UsePWGHFvertexing() const = 0;
84 /** @return JPSI filtering (PWG3) */
85 virtual Bool_t UsePWGDQJPSIfilter() const = 0;
86 /** @return D0->2 hadrons (PWG3) */
87 virtual Bool_t UsePWGHFd2h() const = 0;
88 /** @return PID response */
89 virtual Bool_t UsePIDResponse() const = 0;
90 /** @return Forward mult task (PWGLF) */
91 virtual Bool_t UsePWGLFForward() const = 0;
92 /* @} */
ee275c29 93 /**
94 * Print one flag
95 *
96 * @param title Title
97 * @param use Use or not
98 */
99 virtual void PrintOne(const char* title, Bool_t use) const
100 {
101 Printf("%-30s : %3s", title, use ? "yes" : "no");
102 }
103 /**
104 * Print settings
105 *
106 */
107 virtual void Print() const {
108 PrintOne("Connect to CDB", UseCDBconnect());
109 PrintOne("Use physics selection", UsePhysicsSelection());
110 PrintOne("Use tender wagon", UseTender());
111 PrintOne("Use centrality", UseCentrality());
112 PrintOne("Use V0 correction in tender", UseV0tender());
113 PrintOne("Activate debugging", UseDBG());
114 PrintOne("Use MC info", UseMC());
115 PrintOne("Use Kinematics filter", UseKFILTER());
116 PrintOne("Use track references", UseTR());
117 PrintOne("Use correction framework", UseCORRFW());
118 PrintOne("Use AOD tags", UseAODTAGS());
119 PrintOne("Use sys info", UseSysInfo());
120 PrintOne("Produces an AOD or dAOD's", UseAODhandler());
121 PrintOne("ESD to AOD filter", UseESDfilter());
122 PrintOne("Use Muon train ", UsePWGMuonTrain());
123 PrintOne("Copy muon events", UseMUONcopyAOD());
124 PrintOne("Jet analysis (PWG4)", UseJETAN());
125 PrintOne("Jet delta AODs", UseJETANdelta());
126 PrintOne("Vertexing HF task (PWG3)", UsePWGHFvertexing());
127 PrintOne("JPSI filtering (PWG3)", UsePWGDQJPSIfilter());
128 PrintOne("D0->2 hadrons (PWG3)", UsePWGHFd2h());
129 PrintOne("PID response", UsePIDResponse());
130 PrintOne("Forward mult task (PWGLF)", UsePWGLFForward());
131 }
f8b7a926 132};
133
134VirtualAODCfg* aodCfg = 0;
135
136//====================================================================
137/**
138 * Load a library/module
139 *
140 * @param module Library/module name
141 *
142 * @return true on success
143 */
144Bool_t LoadLibrary(const char *module)
145{
146 // Load a module library in a given mode. Reports success.
147 Int_t result = 0;
148 TString mod(module);
149 ::Info("LoadLibrary", "Loading %s", module);
150 gROOT->IncreaseDirLevel();
151
152 if (mod.IsNull()) {
153 ::Error("AnalysisTrainNew.C::LoadLibrary", "Empty module name");
154 gROOT->DecreaseDirLevel();
155 return kFALSE;
156 }
157
158 // If a library is specified, just load it
159 if (mod.EndsWith(".so")) {
160 mod.Remove(mod.Index(".so"));
161 ::Info("LoadLibrary", "Loading .so: %s", mod.Data());
162 result = gSystem->Load(mod);
163 if (result < 0) {
164 ::Error("AnalysisTrainNew.C::LoadLibrary",
165 "Could not load library %s", module);
166 }
167 gROOT->DecreaseDirLevel();
168 return (result >= 0);
169 }
170 // Check if the library is already loaded
9a3036c4 171 if (strlen(gSystem->GetLibraries(module, "", kFALSE)) > 0) {
f8b7a926 172 ::Info("LoadLibrary", "Module %s.so already loaded", module);
173 gROOT->DecreaseDirLevel();
174 return kTRUE;
175 }
176
177 ::Info("LoadLibrary", "Trying to load lib%s.so", module);
b0635849 178 result = gSystem->Load(Form("lib%s", module));
f8b7a926 179 if (result < 0)
180 ::Error("AnalysisTrainNew.C::LoadLibrary",
181 "Could not load module %s", module);
182 ::Info("LoadLibrary", "Module %s, successfully loaded", module);
183 gROOT->DecreaseDirLevel();
184 return (result >= 0);
185}
186
187//====================================================================
188/**
189 * Load common libraries
190 *
191 * @return true on sucess
192 */
193Bool_t LoadCommonLibraries()
194{
195 // Load common analysis libraries.
196 if (!gSystem->Getenv("ALICE_ROOT")) {
197 ::Error("AnalysisTrainNew.C::LoadCommonLibraries",
198 "Analysis train requires that analysis libraries are "
199 "compiled with a local AliRoot");
200 return kFALSE;
201 }
202
203 Bool_t success = kTRUE;
204 // Load framework classes. Par option ignored here.
205 success &= LoadLibrary("libSTEERBase.so");
206 success &= LoadLibrary("libESD.so");
207 success &= LoadLibrary("libAOD.so");
208 success &= LoadLibrary("libANALYSIS.so");
209 success &= LoadLibrary("libOADB.so");
210 success &= LoadLibrary("libANALYSISalice.so");
211 success &= LoadLibrary("libESDfilter.so");
212 success &= LoadLibrary("libCORRFW.so");
213 gROOT->ProcessLine(".include $ALICE_ROOT/include");
214 if (success) {
215 ::Info("AnalysisTrainNew.C::LoadCommodLibraries",
216 "Load common libraries: SUCCESS");
217 ::Info("AnalysisTrainNew.C::LoadCommodLibraries",
218 "Include path for Aclic compilation:\n%s",
219 gSystem->GetIncludePath());
220 } else {
221 ::Info("AnalysisTrainNew.C::LoadCommodLibraries",
222 "Load common libraries: FAILED");
223 }
224 return success;
225}
226
227//====================================================================
228/**
229 * Load libraries needed by the train
230 *
231 * @param useTender
232 * @param doCDBconnect
233 * @param iESDfilter
234 * @param iPWGMuonTrain
235 * @param iJETAN
236 * @param iJETANdelta
237 * @param iPWGHFvertexing
238 * @param iPWGHFd2h
239 * @param iPWGDQJPSIfilter
240 *
241 * @return true on success
242 */
243Bool_t LoadAnalysisLibraries()
244{
245 // Load common analysis libraries.
246 if (aodCfg->UseTender() || aodCfg->UseCDBconnect()) {
247 if (!LoadLibrary("TENDER") ||!LoadLibrary("TENDERSupplies")) return kFALSE;
248 }
249 // CDBconnect
250 if ((aodCfg->UseCDBconnect() && !aodCfg->UseTender())
251 && !LoadLibrary("PWGPP")) return false;
ee275c29 252 if ((aodCfg->UseESDfilter() ||
253 (aodCfg->UsePWGMuonTrain() && detCfg->UseMUON())))
254 if (!LoadLibrary("PWGmuon")) return kFALSE;
f8b7a926 255 // JETAN
ee275c29 256 if ((aodCfg->UseJETAN() || aodCfg->UseJETANdelta()))
257 if (!LoadLibrary("JETAN")) return kFALSE;
f8b7a926 258 if (aodCfg->UseJETANdelta()) { // CINT doesn't like long '||' chains
259 if (!LoadLibrary("CGAL")) return false;
260 if (!LoadLibrary("fastjet")) return false;
261 if (!LoadLibrary("siscone")) return false;
262 if (!LoadLibrary("SISConePlugin")) return false;
263 if (!LoadLibrary("FASTJETAN")) return false;
264 }
265
266 // PWG3 Vertexing HF
ee275c29 267 if (aodCfg->UsePWGHFvertexing() || aodCfg->UsePWGHFd2h())) {
f8b7a926 268 // CINT doesn't like long '||' chains
269 if (!LoadLibrary("PWGflowBase")) return false;
270 if (!LoadLibrary("PWGflowTasks")) return false;
ee275c29 271 if (cholm) if (!LoadLibrary("PWGTRD")) return false;
f8b7a926 272 if (!LoadLibrary("PWGHFvertexingHF")) return false;
273 }
274
275 // PWG3 dielectron
276 if (aodCfg->UsePWGDQJPSIfilter() &&
277 !LoadLibrary("PWGDQdielectron")) return kFALSE;
278
279 ::Info("AnalysisTrainNew.C::LoadAnalysisLibraries",
280 "Load other libraries: SUCCESS");
281 return kTRUE;
282}
283
284//====================================================================
285/**
286 * Add tasks to the train
287 *
288 * @param cdb_location
289 */
290void AddAnalysisTasks(const char *cdb_location)
291{
292 // === Add all analysis task wagons to the train ===================
293 //
294 // --- Some constants ----------------------------------------------
295 TString ali = "$(ALICE_ROOT)";
296 TString ana = ali + "/ANALYSIS";
297 TString pwghf = ali + "/PWGHF";
298 TString pwglf = ali + "/PWGLF";
299 TString pwgje = ali + "/PWGJE";
300 TString pwgdq = ali + "/PWGDQ";
301 TString pwgpp = ali + "/PWGPP";
302
303 // --- Get the analysis manager ------------------------------------
304 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
305 AliAnalysisManager::SetCommonFileName("AODQA.root");
306
307 // --- PIDResponse(JENS) -------------------------------------------
308 if (aodCfg->UsePIDResponse()) {
309 gROOT->LoadMacro(ana+"/macros/AddTaskPIDResponse.C");
310 AliAnalysisTaskPIDResponse *PIDResponse = AddTaskPIDResponse(kTRUE);
311 }
312
313 // --- CDB connection ----------------------------------------------
314 if (aodCfg->UseCDBconnect() && !aodCfg->UseTender()) {
315 gROOT->LoadMacro(pwgpp+"/PilotTrain/AddTaskCDBconnect.C");
316 AliTaskCDBconnect *taskCDB = AddTaskCDBconnect(cdb_location, grp->run);
317 if (!taskCDB) return;
318
319 AliCDBManager *cdb = AliCDBManager::Instance();
320 // cdb->SetDefaultStorage(cdb_location);
321 cdb->SetDefaultStorageFromRun(grp->run);
322 }
323 if (aodCfg->UseTender()) {
324 gROOT->LoadMacro(ana+"/TenderSupplies/AddTaskTender.C");
325 AliAnalysisTaskSE *tender = AddTaskTender(aodCfg->UseV0tender());
326 }
327
328 // --- Physics selection -------------------------------------------
329 if (aodCfg->UsePhysicsSelection()) {
330 // Physics selection task
331 gROOT->LoadMacro(ana+"/macros/AddTaskPhysicsSelection.C");
332 mgr->RegisterExtraFile("event_stat.root");
333 AliPhysicsSelectionTask *physSelTask =
334 AddTaskPhysicsSelection(aodCfg->UseMC());
335 }
336
337 // --- Centrality (only Pb-Pb) -------------------------------------
338 if (aodCfg->UseCentrality()) {
339 gROOT->LoadMacro(ana+"/macros/AddTaskCentrality.C");
340 AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();
341 taskCentrality->SetMCInput();
342 }
343
344 // --- PWGLF - Forward (cholm@nbi.dk) -----------------------------
ee275c29 345 if (aodCfg->UsePWGLFForward() &&
346 aodCfg->UsePhysicsSelection() &&
347 detCfg->UseFMD()) {
f8b7a926 348 gROOT->LoadMacro(pwglf+"/FORWARD/analysis2/AddTaskForwardMult.C");
349 // Arguments are
350 // mc Assume MC input
351 // runNo Run number to do local initialization - not used
352 // sys Collision system (1:pp,2:PbPb,3:pPb/Pbp) - not used
353 // sNN Collision energy in GeV - not used
354 // field L3 magnetic field strength - not used
355 // config Configuration script
356 // corrdir Possible directory containing custom OADB corrections
357 // HACK load custom corrections
ee275c29 358 Info("", "Adding forward AOD task with mc=%d",
359 aodCfg->UseMC() && aodCfg->UseTR());
f8b7a926 360 AddTaskForwardMult(aodCfg->UseMC() && aodCfg->UseTR(),0,0,0,0,
361 "ForwardAODConfig.C",".");
362 gROOT->LoadMacro(pwglf+"/FORWARD/analysis2/AddTaskCentralMult.C");
363 AddTaskCentralMult(aodCfg->UseMC() && aodCfg->UseTR(),0,0,0,0,
364 "CentralAODConfig.C");
365 const char* hack2="AliForwardCorrectionManager::Instance().Print(\"R\")";
366 gROOT->ProcessLine(hack2);
367 }
368
369
370 // --- ESD filter --------------------------------------------------
371 if (aodCfg->UseESDfilter()) {
372 // ESD filter task configuration.
373 gROOT->LoadMacro(ana+"/ESDfilter/macros/AddTaskESDFilter.C");
ee275c29 374 if (aodCfg->UseMUONcopyAOD() && detCfg->UseMUON()) {
f8b7a926 375 printf("Registering delta AOD file\n");
376 mgr->RegisterExtraFile("AliAOD.Muons.root");
377 mgr->RegisterExtraFile("AliAOD.Dimuons.root");
378 }
379 UInt_t runFlag = (grp->Year()%100)*100;
380 AliAnalysisTaskESDfilter *taskesdfilter =
381 AddTaskESDFilter(aodCfg->UseKFILTER(),
382 aodCfg->UseMUONcopyAOD(), // write Muon AOD
383 kFALSE, // write dimuon AOD
384 kFALSE, // usePhysicsSelection
385 kFALSE, // centrality OBSOLETE
386 kTRUE, // enable TPS only tracks
387 kFALSE, // disable cascades
388 kFALSE, // disable kinks
389 runFlag); // run flag (YY00)
390 }
391
392 TString configPWGHFd2h =
393 (grp->IsPP() ?
394 pwghf+"/vertexingHF/ConfigVertexingHF.C" :
395 pwghf+"/vertexingHF/ConfigVertexingHF_Pb_AllCent.C");
396
397 // --- PWG3 vertexing ----------------------------------------------
398 if (aodCfg->UsePWGHFvertexing()) {
399 gROOT->LoadMacro(pwghf+"/vertexingHF/macros/AddTaskVertexingHF.C");
400
401 if (!aodCfg->UsePWGHFd2h())
402 TFile::Cp(gSystem->ExpandPathName(configPWG3d2h.Data()),
403 "file:ConfigVertexingHF.C");
404
405 AliAnalysisTaskSEVertexingHF *taskvertexingHF = AddTaskVertexingHF();
406 if (!taskvertexingHF)
407 ::Warning("AnalysisTrainNew",
408 "AliAnalysisTaskSEVertexingHF cannot run for this train "
409 "conditions - EXCLUDED");
410 else
411 mgr->RegisterExtraFile("AliAOD.VertexingHF.root");
412
413 taskvertexingHF->SelectCollisionCandidates(0);
414 }
415
416 // ---- PWG3 JPSI filtering (only pp) ------------------------------
a0bdf9a4 417 if (aodCfg->UsePWGDQJPSIfilter()) {
f8b7a926 418 gROOT->LoadMacro(pwgdq+"/dielectron/macros/AddTaskJPSIFilter.C");
419 AliAnalysisTaskSE *taskJPSIfilter = AddTaskJPSIFilter();
420
421 if (!taskJPSIfilter)
422 ::Warning("AnalysisTrainNew",
423 "AliAnalysisTaskDielectronFilter cannot run for this train "
424 "conditions - EXCLUDED");
425 else
426 mgr->RegisterExtraFile("AliAOD.Dielectron.root");
427
428 taskJPSIfilter->SelectCollisionCandidates(0);
429 }
430
431 // --- PWG3 D2h ----------------------------------------------------
ee275c29 432 if (aodCfg->UsePWGHFd2h() && aodCfg->UsePWGHFvertexing()) {
f8b7a926 433 gROOT->LoadMacro(pwghf+"/vertexingHF/AddD2HTrain.C");
434 TFile::Cp(gSystem->ExpandPathName(configPWGHFd2h.Data()),
435 "file:ConfigVertexingHF.C");
436 AddD2HTrain(kFALSE, 1,0,0,0,0,0,0,0,0,0,0);
437 }
438
439 // --- Jet analysis ------------------------------------------------
440
441 // Configurations flags, move up?
442 if (aodCfg->UseJETAN()) {
a0bdf9a4 443#if 0
f8b7a926 444 Warning("", "JET analysis disabled - major restructuring ofg JETAN");
445#else
446 TString jetAOD = "AliAOD.Jets.root";
447 UInt_t highPtMask = 768;// from esd filter
448 TString subtractBranches = "";
449 UInt_t psFlag = 0;
450
451 Info("", "Loading macro %s/macros/AddTaskJets.C", pwgje.Data());
452 gROOT->LoadMacro(pwgje+"/macros/AddTaskJets.C");
453 // Default jet reconstructor running on ESD's
454 // no background subtraction
455 AliAnalysisTask* task = AddTaskJets("AOD","UA1",0.4F,highPtMask,1.F,0);
456 if (!task)
457 ::Fatal("AnalysisTrainNew",
458 "AliAnalysisTaskJets cannot run for this train "
459 "conditions - EXCLUDED");
460
461 AliAnalysisTaskJets* taskjets = static_cast<AliAnalysisTaskJets*>(task);
462 if(!jetAOD.IsNull()) taskjets->SetNonStdOutputFile(jetAOD);
463
464 if (aodCfg->UseJETANdelta()) {
465 // need to modify this accordingly in the add task jets
466 // AddTaskJetsDelta("AliAOD.Jets.root");
467 mgr->RegisterExtraFile(jetAOD);
468 TString cTmp("");
469
470 if(grp->IsAA()){
471 // UA1 intrinsic background subtraction
472 // background subtraction
473 taskjets = AddTaskJets("AOD","UA1",0.4,highPtMask,1.,2);
474 if(!jetAOD.IsNull()) taskjets->SetNonStdOutputFile(jetAOD);
475 }
476 // SICONE
477 //no background subtration to be done later....
478 taskjets = AddTaskJets("AOD","SISCONE",0.4,highPtMask,0.15,0);
479 if(!jetAOD.IsNull()) taskjets->SetNonStdOutputFile(jetAOD.Data());
480 cTmp = taskjets->GetNonStdBranch();
481 if(!cTmp.IsNull()) subtractBranches += Form("%s ",cTmp.Data());
482
483 // Add the clusters..
484 gROOT->LoadMacro(pwgje+"/macros/AddTaskJetCluster.C");
485 AliAnalysisTaskJetCluster *taskCl = 0;
486 Float_t fCenUp = 0;
487 Float_t fCenLo = 0;
488 Float_t fTrackEtaWindow = 0.9;
489 // this one is for the background and random jets, random cones
490 // with no skip
491 taskCl = AddTaskJetCluster("AOD","",highPtMask,
492 psFlag,"KT",0.4,0,1,
493 jetAOD,0.15,
494 fTrackEtaWindow,0);
495 taskCl->SetBackgroundCalc(kTRUE);
496 taskCl->SetNRandomCones(10);
497 taskCl->SetCentralityCut(fCenLo,fCenUp);
498 taskCl->SetGhostEtamax(fTrackEtaWindow);
499 TString bkgBranch = Form("%s_%s",
500 AliAODJetEventBackground::StdBranchName(),
501 taskCl->GetJetOutputBranch());
502
503 taskCl = AddTaskJetCluster("AOD", "", highPtMask, psFlag, "ANTIKT",
504 0.4, 2, 1, jetAOD, 0.15);
505 taskCl->SetCentralityCut(fCenLo,fCenUp);
506 if(grp->IsAA()) taskCl->SetBackgroundBranch(bkgBranch.Data());
507
508 taskCl->SetNRandomCones(10);
509 subtractBranches += Form("%s ",taskCl->GetJetOutputBranch());
510
511 taskCl = AddTaskJetCluster("AOD", "", highPtMask, psFlag, "ANTIKT",
512 0.2, 0 , 1, jetAOD, 0.15);
513 taskCl->SetCentralityCut(fCenLo,fCenUp);
514 if(grp->IsAA())taskCl->SetBackgroundBranch(bkgBranch);
515
516 subtractBranches += Form("%s ",taskCl->GetJetOutputBranch());
517
518 // DO THE BACKGROUND SUBTRACTION
519 if(grp->IsAA() && !subtractBranches.IsNull()) {
520 gROOT->LoadMacro(pwgje+"/macros/AddTaskJetBackgroundSubtract.C");
521 AliAnalysisTaskJetBackgroundSubtract *taskSubtract = 0;
522 taskSubtract = AddTaskJetBackgroundSubtract(subtractBranches,1,
523 "B0","B%d");
524 taskSubtract->SetBackgroundBranch(bkgBranch);
525 if(!jetAOD.IsNull()) taskSubtract->SetNonStdOutputFile(jetAOD.Data());
526 }
527 }
528#endif
529 }
530}
531
532
533
534//====================================================================
535/**
536 * Create the input chain
537 *
538 *
539 * @return Pointer to newly allocated train
540 */
541TChain *CreateChain()
542{
543 // Create the input chain
544 chain = new TChain("esdTree");
545 if (gSystem->AccessPathName("AliESDs.root"))
546 ::Error("AnalysisTrainNew.C::CreateChain",
547 "File: AliESDs.root not in ./data dir");
548 else
549 chain->Add("AliESDs.root");
550 if (chain->GetNtrees()) return chain;
551 return NULL;
552}
553
b2a78539 554/**
555 * Helper function to make @c outputs_valid file
556 *
557 */
558void ValidateOutput()
559{
560 std::ofstream out;
561 out.open("outputs_valid", ios::out);
562 out.close();
563}
f8b7a926 564//====================================================================
565/**
566 * Merge AOD output
567 *
b2a78539 568 * @param dir Directory
569 * @param stage The merging stage
f8b7a926 570 */
b2a78539 571void AODMerge(const char* dir, Int_t stage)
f8b7a926 572{
573 // Merging method. No staging and no terminate phase.
574 TStopwatch timer; timer.Start();
b2a78539 575 TString outputDir = dir;
a0bdf9a4 576 TObjArray outputFiles;
b2a78539 577 // outputFiles.Add(new TObjString("EventStat_temp.root"));
578 outputFiles.Add(new TObjString("AODQA.root"));
a0bdf9a4 579 outputFiles.Add(new TObjString("pyxsec_hists.root"));
b2a78539 580
581 Bool_t mergeTrees = stage <= 1;
582 if (mergeTrees) {
583 outputFiles.Add(new TObjString("AliAOD.root"));
584 if (aodCfg->UsePWGHFvertexing())
585 outputFiles.Add(new TObjString("AliAOD.VertexingHF.root"));
586 if (aodCfg->UseESDfilter() &&
587 aodCfg->UseMUONcopyAOD() &&
588 detCfg->UseMUON())
589 outputFiles.Add(new TObjString("AliAOD.Muons.root"));
590 if (aodCfg->UseJETAN())
591 outputFiles.Add(new TObjString("AliAOD.Jets.root"));
592 if (aodCfg->UsePWGDQJPSIfilter())
593 outputFiles.Add(new TObjString("AliAOD.Dielectron.root"));
594 }
595
f8b7a926 596 TString mergeExcludes = "";
ee275c29 597 TIter iter(&outputFiles);
f8b7a926 598 TObjString* str = 0;
a0bdf9a4 599 Bool_t merged = kTRUE;
f8b7a926 600 while ((str = static_cast<TObjString*>(iter()))) {
a0bdf9a4 601 TString& outputFile = str->GetString();
f8b7a926 602 // Skip already merged outputs
603 if (!gSystem->AccessPathName(outputFile)) {
b2a78539 604 ::Warning("Merge","Output file <%s> found. Not merging again.",
605 outputFile.Data());
f8b7a926 606 continue;
607 }
608 if (mergeExcludes.Contains(outputFile.Data())) continue;
b2a78539 609 merged = AliAnalysisAlien::MergeOutput(outputFile,
610 outputDir,
611 10,
612 stage);
f8b7a926 613 if (!merged) {
b2a78539 614 ::Error("Merge", "Cannot merge %s\n", outputFile.Data());
f8b7a926 615 continue;
616 }
617 }
b2a78539 618
f8b7a926 619 // all outputs merged, validate
b2a78539 620 if (!outputDir.Contains("stage")) {
621 ValidateOutput();
622 timer.Print();
623 return;
624 }
625
626 // --- set up to run terminate -------------------------------------
627 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
628 mgr->SetSkipTerminate(kFALSE);
629 if (!mgr->InitAnalysis()) return;
630
631 mgr->PrintStatus();
632 mgr->SetGridHandler(new AliAnalysisAlien);
633 mgr->StartAnalysis("gridterminate",0);
634 ValidateOutput();
f8b7a926 635 timer.Print();
b2a78539 636
f8b7a926 637}
638
639//====================================================================
640/**
641 * Set-up and run AOD train
642 *
643 * @param run Run number
644 * @param xmlfile Collection
645 * @param stage Stage
646 */
647void AOD(UInt_t run, const char* xmlfile=0, Int_t stage=0)
648{
ee275c29 649 TString host(gSystem->HostName());
650 cholm = host.BeginsWith("hehi");
651 if (cholm) TGrid::Connect("alien:");
652
f8b7a926 653 // -----------------------------------------------------------------
654 //
655 // Get GRP parameters. Defines global "grp" as a pointer to GRPData
656 //
657 gROOT->Macro(Form("GRP.C(%d)", run));
658 gROOT->Macro("AODConfig.C");
421f877c 659 gROOT->Macro("BaseConfig.C");
ee275c29 660 gROOT->Macro("DetConfig.C");
f8b7a926 661
662 // --- Some settings -----------------------------------------------
663 // Set temporary merging directory to current one
664 gSystem->Setenv("TMPDIR", gSystem->pwd());
665 // Set temporary compilation directory to current one
666 gSystem->SetBuildDir(gSystem->pwd(), kTRUE);
667
668 // --- Friendly message --------------------------------------------
669 printf("===================================================\n");
670 printf("=========== RUNNING FILTERING TRAIN ==========\n");
671 printf("===================================================\n");
672 printf("= Configuring analysis train for:\n");
ee275c29 673 aodCfg->Print();
f8b7a926 674
675 // Load common libraries and set include path
676 if (!LoadCommonLibraries()) {
677 ::Error("AnalysisTrain", "Could not load common libraries");
678 return;
679 }
680
681 // === Make the analysis manager and connect event handlers ========
682 //
683 // --- Analysis manager and load libraries -------------------------
684 AliAnalysisManager *mgr = new AliAnalysisManager("Filter","Production train");
b2a78539 685 mgr->SetRunFromPath(grp->run);
f8b7a926 686 if (aodCfg->UseSysInfo()) mgr->SetNSysInfo(100);
687 if (!LoadAnalysisLibraries()) {
688 ::Error("AnalysisTrain", "Could not load analysis libraries");
689 return;
690 }
691
692 // --- Create ESD input handler ------------------------------------
693 AliESDInputHandler *esdHandler = new AliESDInputHandler();
694 mgr->SetInputEventHandler(esdHandler);
695
696 // --- Monte Carlo handler -----------------------------------------
697 if (aodCfg->UseMC()) {
698 AliMCEventHandler* mcHandler = new AliMCEventHandler();
699 mgr->SetMCtruthEventHandler(mcHandler);
700 mcHandler->SetPreReadMode(1);
701 mcHandler->SetReadTR(aodCfg->UseTR());
702 }
703
704 // --- AOD output handler ------------------------------------------
705 if (aodCfg->UseAODhandler()) {
706 AliAODHandler* aodHandler = new AliAODHandler();
707 aodHandler->SetOutputFileName("AliAOD.root");
708 mgr->SetOutputEventHandler(aodHandler);
709 }
710
f8b7a926 711 // === Set up tasks ================================================
712 //
713 // --- Create tasks ------------------------------------------------
714 AddAnalysisTasks(cdbPath);
715
b2a78539 716 // --- Debugging if needed -----------------------------------------
717 if (aodCfg->UseDBG()) mgr->SetDebugLevel(3);
718
719
f8b7a926 720 // --- If merging, do so here and exit -----------------------------
721 if (stage > 0) {
b2a78539 722 AODMerge(xmlfile, stage);
f8b7a926 723 return;
724 }
725 // === Run the analysis ============================================
726 //
727 // --- Make our chain ----------------------------------------------
728 TChain *chain = CreateChain();
729 if (!chain) return;
730
731 // --- Run the thing -----------------------------------------------
732 TStopwatch timer;
733 timer.Start();
b2a78539 734 if (!mgr->InitAnalysis()) return;
735
736
737 mgr->PrintStatus();
f8b7a926 738 mgr->SetSkipTerminate(kTRUE);
b2a78539 739 mgr->StartAnalysis("local", chain);
f8b7a926 740 timer.Print();
741}
742
743//
744// EOF
745//
746