merging post-CVS changes into SVN:
[u/mrichter/AliRoot.git] / PWG0 / dNdEta / rundNdEtaAnalysis.C
1 /* $Id$ */
2
3 //
4 // Script to test the dN/dEta analysis using the dNdEtaAnalysis and
5 // dNdEtaCorrection classes. Note that there is a cut on the events,
6 // so the measurement will be biassed.
7 //
8 // implementation with TSelector
9 //
10
11 #include "../CreateESDChain.C"
12 #include "../PWG0Helper.C"
13
14 void rundNdEtaAnalysis(Char_t* data, Int_t nRuns=20, Int_t offset=0, Bool_t aMC = kFALSE, Bool_t aDebug = kFALSE, Bool_t aProof = kFALSE, const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction", const char* option = "", const char* proofServer = "lxb6046")
15 {
16   if (aProof)
17     connectProof(proofServer);
18
19   TString libraries("libEG;libGeom;libESD;libPWG0base");
20   TString packages("PWG0base");
21
22   if (!prepareQuery(libraries, packages, kTRUE))
23     return;
24
25   gROOT->ProcessLine(".L CreateCuts.C");
26   gROOT->ProcessLine(".L drawPlots.C");
27
28   TChain* chain = CreateESDChain(data, nRuns, offset);
29
30   TList inputList;
31
32   // selection of esd tracks
33   AliESDtrackCuts* esdTrackCuts = CreateTrackCuts();
34   if (!esdTrackCuts)
35   {
36     printf("ERROR: esdTrackCuts could not be created\n");
37     return;
38   }
39
40   inputList.Add(esdTrackCuts);
41
42   TString selectorName = ((aMC == kFALSE) ? "AlidNdEtaAnalysisESDSelector" : "AlidNdEtaAnalysisMCSelector");
43   AliLog::SetClassDebugLevel(selectorName, AliLog::kInfo);
44
45   selectorName += ".cxx+";
46
47   if (aDebug != kFALSE)
48     selectorName += "+g";
49
50   Int_t result = executeQuery(chain, &inputList, selectorName, option);
51
52   if (result >= 0)
53   {
54     if (aMC)
55     {
56       dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
57
58       TFile* file = TFile::Open("analysis_mc.root");
59
60       if (!file)
61       {
62         cout << "Error. File not found" << endl;
63         return;
64       }
65       fdNdEtaAnalysis->LoadHistograms();
66       fdNdEtaAnalysis->DrawHistograms(kTRUE);
67     }
68     else
69       FinishAnalysisAll("analysis_esd_raw.root", "analysis_esd.root", correctionMapFile, correctionMapFolder);
70   }
71 }
72