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