Updates to scripts. Mostly documentation and some new functionalities
[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / MakeMCCorr.C
CommitLineData
269cc80d 1/**
56236b95 2 * @file MakeMCCorr.C
269cc80d 3 * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
56236b95 4 * @date Tue Jul 12 10:06:07 2011
269cc80d 5 *
56236b95 6 * @brief Generate MC corrections
269cc80d 7 *
8 * @ingroup pwg2_forward_scripts_makers
9 */
56236b95 10//====================================================================
269cc80d 11/**
56236b95 12 * Generatew MC corrections
269cc80d 13 *
14 * If the ROOT AliEn interface library (libRAliEn) can be loaded,
15 * and the parameter @a name is not empty, then use the plugin to do
16 * the analysis. Note that in this case, the output is placed
17 * in a sub-directory named by @a name after escaping spaces and special
18 * characters
19 *
c6115ede 20 * @param esddir AOD input directory. Any file matching the pattern
269cc80d 21 * *AliAODs*.root are added to the chain
22 * @param nEvents Number of events to process. If 0 or less, then
23 * all events are analysed
269cc80d 24 * @param vzMin Least @f$ v_z@f$ (centimeter)
25 * @param vzMax Largest @f$ v_z@f$ (centimeter)
26 * @param proof If larger then 1, run in PROOF-Lite mode with this
27 * many number of workers.
28 * @param name Name of train - free form. This will be the name
29 * of the output directory if the plug-in is used
30 *
56236b95 31 * @ingroup pwg2_forward_corr
32 * @ingroup pwg2_forward_scripts_makers
269cc80d 33 */
34void MakeMCCorr(const char* esddir = ".",
35 Int_t nEvents = -1,
36 Double_t vzMin = -10,
37 Double_t vzMax = +10,
38 Int_t proof = 0,
39 const char* name = 0)
40{
41 if ((name && name[0] != '\0') && gSystem->Load("libRAliEn") >= 0) {
42 Error("MakeMCCorr", "Plug-in mode not implemented yet!");
43 return;
44
56236b95 45 const char* fwdPath =
46 gSystem->ExpandPathName("$(ALICE_ROOT)/PWG2/FORWARD/analysis2");
47 gROOT->LoadMacro(Form("%s/trains/BuildTrain.C", fwdPath));
48 BuildTrain("MakeMCCorrTrain");
49
269cc80d 50 MakeMCCorrTrain t(name, vzMin, vzMax);
51 t.SetDataDir(esddir);
52 t.SetDataSet("");
53 t.SetAllowOverwrite(true);
54 t.SetProofServer(Form("workers=%d",proof));
55 t.Run(proof > 0 ? "PROOF" : "LOCAL", "FULL", nEvents, proof > 0);
56 return;
57 }
58 // --- Libraries to load -------------------------------------------
59 gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
60
61 // --- Check for proof mode, and possibly upload pars --------------
62 if (proof> 0) {
63 gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
64 LoadPars(proof);
65 }
66
67 // --- Our data chain ----------------------------------------------
68 gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/MakeChain.C");
69 TChain* chain = MakeChain("ESD", esddir,true);
70 // If 0 or less events is select, choose all
71 if (nEvents <= 0) nEvents = chain->GetEntries();
72
73 // --- Set the macro path ------------------------------------------
74 gROOT->SetMacroPath(Form("%s:$(ALICE_ROOT)/PWG2/FORWARD/analysis2:"
75 "$ALICE_ROOT/ANALYSIS/macros",
76 gROOT->GetMacroPath()));
77
78 // --- Creating the manager and handlers ---------------------------
79 AliAnalysisManager *mgr = new AliAnalysisManager(name, "Forward MC corr");
80 AliAnalysisManager::SetCommonFileName("forward_mccorr.root");
81
82 // --- ESD input handler -------------------------------------------
83 AliESDInputHandler *esdHandler = new AliESDInputHandler();
84 mgr->SetInputEventHandler(esdHandler);
85
86 // --- Monte Carlo handler -----------------------------------------
87 AliMCEventHandler* mcHandler = new AliMCEventHandler();
88 mgr->SetMCtruthEventHandler(mcHandler);
89 mcHandler->SetReadTR(true);
90
91 // --- Add Physics Selection ---------------------------------------
92 // Physics selection
93 gROOT->LoadMacro("AddTaskPhysicsSelection.C");
94 AddTaskPhysicsSelection(kTRUE, kTRUE, kFALSE);
95 // --- Fix up physics selection to give proper A,C, and E triggers -
96 AliInputEventHandler* ih =
97 static_cast<AliInputEventHandler*>(mgr->GetInputEventHandler());
98 AliPhysicsSelection* ps =
99 static_cast<AliPhysicsSelection*>(ih->GetEventSelection());
100 // Ignore trigger class when selecting events. This mean that we
101 // get offline+(A,C,E) events too
102 // ps->SetSkipTriggerClassSelection(true);
103
104 // --- Add our task ------------------------------------------------
105 gROOT->LoadMacro("AddTaskForwardMCCorr.C");
106 AddTaskForwardMCCorr();
107
108 gROOT->LoadMacro("AddTaskCentralMCCorr.C");
109 AddTaskCentralMCCorr();
110
111 // --- Run the analysis --------------------------------------------
112 TStopwatch t;
113 if (!mgr->InitAnalysis()) {
114 Error("MakedNdeta", "Failed to initialize analysis train!");
115 return;
116 }
117 // Skip terminate if we're so requested and not in Proof or full mode
118 mgr->SetSkipTerminate(false);
119 // Some informative output
120 mgr->PrintStatus();
121 // mgr->SetDebugLevel(3);
122 if (mgr->GetDebugLevel() < 1 && !proof)
123 mgr->SetUseProgressBar(kTRUE,100);
124
125 // Run the train
126 t.Start();
127 Printf("=== RUNNING ANALYSIS ==================================");
128 mgr->StartAnalysis(proof ? "proof" : "local", chain, nEvents);
129 t.Stop();
130 t.Print();
131}
132//
133// EOF
134//