Renames and new scripts
[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / MakeELossFits.C
1 /**
2  * @file 
3  * 
4  * @ingroup pwg2_forward_scripts
5  */
6 /** 
7  * Run a pass on ESD data to produce the energ loss fits 
8  * 
9  *
10  * @ingroup pwg2_forward_scripts
11  */
12 void MakeELossFits(const char* esddir, 
13                    Int_t       nEvents=1000, 
14                    Int_t       proof=0,
15                    Bool_t      mc=false)
16 {
17   // --- Libraries to load -------------------------------------------
18   gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
19
20   // --- Check for proof mode, and possibly upload pars --------------
21   if (proof > 0) {
22     gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
23     LoadPars(proof);
24   }
25   
26   // --- Our data chain ----------------------------------------------
27   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/MakeESDChain.C");
28   TChain* chain = MakeESDChain(esddir, true);
29   // If 0 or less events is select, choose all 
30   if (nEvents <= 0) nEvents = chain->GetEntries();
31   Info("MakeELossFits", "Will analyse %d events", nEvents);
32
33   // --- Creating the manager and handlers ---------------------------
34   AliAnalysisManager *mgr  = new AliAnalysisManager("Forward ELoss Train", 
35                                                     "Forward energy loss");
36   AliAnalysisManager::SetCommonFileName("forward_eloss.root");
37
38   AliESDInputHandler *esdHandler = new AliESDInputHandler();
39   esdHandler->SetInactiveBranches("AliESDACORDE "
40                                   "AliRawDataErrorLogs "
41                                   "CaloClusters "
42                                   "Cascades "
43                                   "EMCALCells "
44                                   "EMCALTrigger "
45                                   "Kinks "
46                                   "Cascades "
47                                   "MuonTracks "
48                                   "TrdTracks "
49                                   "CaloClusters "
50                                   "HLTGlobalTrigger");
51   mgr->SetInputEventHandler(esdHandler);      
52   
53   // AOD output handler
54   AliAODHandler* aodHandler   = new AliAODHandler();
55   mgr->SetOutputEventHandler(aodHandler);
56   aodHandler->SetOutputFileName("AliAODs.root");
57
58   // --- Add tasks ---------------------------------------------------
59   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
60   AddTaskPhysicsSelection(mc, kTRUE, kTRUE);
61   AliFMDEnergyFitterTask* task = new AliFMDEnergyFitterTask("fmdEnergyFitter");
62   mgr->AddTask(task);
63
64   // --- Make the output container and connect it --------------------
65   TString outputfile = AliAnalysisManager::GetCommonFileName();
66   AliAnalysisDataContainer* histOut = 
67     mgr->CreateContainer("Forward", TList::Class(), 
68                          AliAnalysisManager::kOutputContainer,outputfile);
69   mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
70   mgr->ConnectOutput(task, 1, histOut);
71
72   // --- Set parameters on the algorithms ----------------------------
73   // Set the number of SPD tracklets for which we consider the event a
74   // low flux event
75   task->GetEventInspector().SetLowFluxCut(1000); 
76   // Set the maximum error on v_z [cm]
77   task->GetEventInspector().SetMaxVzErr(0.2);
78   // Set the eta axis to use - note, this overrides whatever is used
79   // by the rest of the algorithms - but only for the energy fitter
80   // algorithm. 
81   task->GetEnergyFitter().SetEtaAxis(200, -4, 6);
82   // Set maximum energy loss to consider 
83   task->GetEnergyFitter().SetMaxE(15); 
84   // Set number of energy loss bins 
85   task->GetEnergyFitter().SetNEbins(100);
86   // Set whether to use increasing bin sizes 
87   task->GetEnergyFitter().SetUseIncreasingBins(true);
88   // Set whether to do fit the energy distributions 
89   task->GetEnergyFitter().SetDoFits(kTRUE);
90   // Set whether to make the correction object 
91   task->GetEnergyFitter().SetDoMakeObject(kTRUE);
92   // Set the low cut used for energy
93   task->GetEnergyFitter().SetLowCut(0.4);
94   // Set the number of bins to subtract from maximum of distributions
95   // to get the lower bound of the fit range
96   task->GetEnergyFitter().SetFitRangeBinWidth(4);
97   // Set the maximum number of landaus to try to fit (max 5)
98   task->GetEnergyFitter().SetNParticles(5);
99   // Set the minimum number of entries in the distribution before
100   // trying to fit to the data
101   task->GetEnergyFitter().SetMinEntries(1000);
102   // --- Set limits on fits the energy -------------------------------
103   // Maximum relative error on parameters 
104   AliFMDCorrELossFit::ELossFit::fgMaxRelError = .12;
105   // Least weight to use 
106   AliFMDCorrELossFit::ELossFit::fgLeastWeight = 1e-5;
107   // Maximum value of reduced chi^2 
108   AliFMDCorrELossFit::ELossFit::fgMaxChi2nu   = 5;
109   
110   // --- Run the analysis --------------------------------------------
111   TStopwatch t;
112   if (!mgr->InitAnalysis()) {
113     Error("RunManager", "Failed to initialize analysis train!");
114     return;
115   }
116   // Some informative output 
117   mgr->PrintStatus();
118   // mgr->SetDebugLevel(3);
119   if (mgr->GetDebugLevel() < 1 && proof <= 0) mgr->SetUseProgressBar(kTRUE);
120
121   // Run the train 
122   t.Start();
123   Printf("=== RUNNING ANALYSIS ==================================");
124   mgr->StartAnalysis(proof > 0 ? "proof" : "local", chain, nEvents);
125   t.Stop();
126   t.Print();
127 }
128 //
129 // EOF
130 //