9b7536dd0addfa72d1113e0a1780c4a8ca70ab89
[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / MakedNdeta.C
1 /**
2  * @file 
3  * 
4  * @ingroup pwg2_forward_scripts
5  */
6
7 /** 
8  * Run first pass of the analysis - that is read in ESD and produce AOD
9  * 
10  * @param esddir    ESD input directory. Any file matching the pattern 
11  *                  *AliESDs*.root are added to the chain 
12  * @param nEvents   Number of events to process.  If 0 or less, then 
13  *                  all events are analysed
14  * @param proof     Run in proof mode 
15  * @param mc        Run over MC data
16  *
17  * If PROOF mode is selected, then Terminate will be run on the master node 
18  * in any case. 
19  * 
20  *
21  * @ingroup pwg2_forward_scripts
22  */
23 void MakedNdeta(const char* aoddir=".", 
24                 Int_t       nEvents=-1, 
25                 const char* trig="INEL",
26                 Float_t     centLow = 0,
27                 Float_t     centHigh = 100,
28                 Double_t    vzMin=-10,
29                 Double_t    vzMax=10,
30                 Int_t       proof=0)
31 {
32   // --- Libraries to load -------------------------------------------
33   gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
34
35   // --- Check for proof mode, and possibly upload pars --------------
36   if (proof> 0) { 
37     gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
38     LoadPars(proof);
39   }
40   
41   // --- Our data chain ----------------------------------------------
42   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/MakeChain.C");
43   TChain* chain = MakeChain("AOD", aoddir,true);
44   // If 0 or less events is select, choose all 
45   if (nEvents <= 0) nEvents = chain->GetEntries();
46
47   // --- Set the macro path ------------------------------------------
48   gROOT->SetMacroPath(Form("%s:$(ALICE_ROOT)/PWG2/FORWARD/analysis2:"
49                            "$ALICE_ROOT/ANALYSIS/macros",
50                            gROOT->GetMacroPath()));
51
52   // --- Creating the manager and handlers ---------------------------
53   AliAnalysisManager *mgr  = new AliAnalysisManager("Forward Train", 
54                                                     "Forward dN/deta");
55   AliAnalysisManager::SetCommonFileName("forward_dndeta.root");
56
57   // --- ESD input handler -------------------------------------------
58   AliAODInputHandler *aodInputHandler = new AliAODInputHandler();
59   mgr->SetInputEventHandler(aodInputHandler);      
60        
61   // --- Add tasks ---------------------------------------------------
62   // Forward 
63   gROOT->LoadMacro("AddTaskForwarddNdeta.C");
64   AddTaskForwarddNdeta(trig, vzMin, vzMax, centLow, centHigh, true);
65   // Central
66   gROOT->LoadMacro("AddTaskCentraldNdeta.C");
67   AddTaskCentraldNdeta(trig, vzMin, vzMax, centLow, centHigh);
68
69   
70   // --- Run the analysis --------------------------------------------
71   TStopwatch t;
72   if (!mgr->InitAnalysis()) {
73     Error("MakedNdeta", "Failed to initialize analysis train!");
74     return;
75   }
76   // Skip terminate if we're so requested and not in Proof or full mode
77   mgr->SetSkipTerminate(false);
78   // Some informative output 
79   mgr->PrintStatus();
80   // mgr->SetDebugLevel(3);
81   if (mgr->GetDebugLevel() < 1 && !proof) 
82     mgr->SetUseProgressBar(kTRUE,100);
83   
84   // Run the train 
85   t.Start();
86   Printf("=== RUNNING ANALYSIS ==================================");
87   mgr->StartAnalysis(proof ? "proof" : "local", chain, nEvents);
88   t.Stop();
89   t.Print();
90 }
91 //
92 // EOF
93 //