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
  73 void FinishAnalysisAll(const char* dataInput = "analysis_esd_raw.root", const char* dataOutput = "analysis_esd.root", const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction")
  74 {
  75   gSystem->Load("libPWG0base");
  76
  77   AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
  78   TFile::Open(correctionMapFile);
  79   dNdEtaCorrection->LoadHistograms();
  80
  81   TFile* file = TFile::Open(dataInput);
  82
  83   if (!file)
  84   {
  85     cout << "Error. File not found" << endl;
  86     return;
  87   }
  88
  89   dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
  90   fdNdEtaAnalysis->LoadHistograms("dndeta");
  91   fdNdEtaAnalysis->Finish(dNdEtaCorrection, 0.3, AlidNdEtaCorrection::kINEL);
  92   fdNdEtaAnalysis->DrawHistograms(kTRUE);
  93   TFile* file2 = TFile::Open(dataOutput, "RECREATE");
  94   fdNdEtaAnalysis->SaveHistograms();
  95
  96   file->cd();
  97   fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTr", "dndetaTr");
  98   fdNdEtaAnalysis->LoadHistograms("dndeta");
  99   fdNdEtaAnalysis->Finish(dNdEtaCorrection, 0.3, AlidNdEtaCorrection::kVertexReco);
 100   fdNdEtaAnalysis->DrawHistograms(kTRUE);
 101   file2->cd();
 102   fdNdEtaAnalysis->SaveHistograms();
 103
 104   file->cd();
 105   fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTrVtx", "dndetaTrVtx");
 106   fdNdEtaAnalysis->LoadHistograms("dndeta");
 107   fdNdEtaAnalysis->Finish(dNdEtaCorrection, 0.3, AlidNdEtaCorrection::kTrack2Particle);
 108   fdNdEtaAnalysis->DrawHistograms(kTRUE);
 109   file2->cd();
 110   fdNdEtaAnalysis->SaveHistograms();
 111
 112   file->cd();
 113   fdNdEtaAnalysis = new dNdEtaAnalysis("dndetaTracks", "dndetaTracks");
 114   fdNdEtaAnalysis->LoadHistograms("dndeta");
 115   fdNdEtaAnalysis->Finish(0, 0.3, AlidNdEtaCorrection::kNone);
 116   fdNdEtaAnalysis->DrawHistograms(kTRUE);
 117   file2->cd();
 118   fdNdEtaAnalysis->SaveHistograms();
 119 }
 120
 121 void FinishAnalysis(const char* analysisFile = "analysis_esd.root", const char* analysisDir = "dndeta", const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction", Bool_t useUncorrected = kFALSE, Bool_t simple = kFALSE)
 122 {
 123   gSystem->Load("libPWG0base");
 124
 125   TFile* file = TFile::Open(analysisFile);
 126
 127   dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis(analysisDir, analysisDir);
 128   fdNdEtaAnalysis->LoadHistograms();
 129
 130   if (correctionMapFile)
 131   {
 132     AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
 133     TFile::Open(correctionMapFile);
 134     dNdEtaCorrection->LoadHistograms();
 135
 136     fdNdEtaAnalysis->Finish(dNdEtaCorrection, 0.3);
 137   }
 138
 139   fdNdEtaAnalysis->DrawHistograms(simple);
 140
 141   TH1* hist = fdNdEtaAnalysis->GetdNdEtaHistogram(1);
 142   Int_t binLeft = hist->GetXaxis()->FindBin(-0.5);
 143   Int_t binRight = hist->GetXaxis()->FindBin(0.5);
 144   Float_t value1 = hist->Integral(binLeft, binRight);
 145
 146   hist = fdNdEtaAnalysis->GetdNdEtaHistogram(2);
 147   Float_t value2 = hist->Integral(binLeft, binRight);
 148
 149   printf("Ratio is %f, values are %f %f\n", value1 / value2, value1, value2);
 150 }