Reconstruction QA by Sylwester

[u/mrichter/AliRoot.git] / PWG0 / dNdEta / makeCorrection2.C

diff --git a/PWG0/dNdEta/makeCorrection2.C b/PWG0/dNdEta/makeCorrection2.C
index 5da9f38d605d03dc1e71f575cbe52885e82d990b..21521ac675570806bc7256779eb915831d53c247 100644 (file)
--- a/PWG0/dNdEta/makeCorrection2.C
+++ b/PWG0/dNdEta/makeCorrection2.C
@@ -8,33 +8,82 @@
 //
 
 #include "../CreateESDChain.C"
+#include "../PWG0Helper.C"
 
-void makeCorrection2(Char_t* dataDir, Int_t nRuns=20, Int_t offset = 0, Bool_t debug = kFALSE)
+void makeCorrection2(Char_t* dataDir, Int_t nRuns=20, Int_t offset = 0, Bool_t debug = kFALSE, Bool_t aProof = kFALSE, const Char_t* option = "", const Char_t* proofConnect = "proof01@lxb6046")
 {
-  gSystem->Load("libPWG0base");
-  gSystem->Load("libPWG0dep");
+  if (aProof)
+    connectProof(proofConnect);
 
-  TChain* chain = CreateESDChainFromDir(dataDir, nRuns, offset);
+  TString libraries("libEG;libGeom;libESD;libPWG0base;libVMC;libMinuit;libSTEER;libPWG0dep;libEVGEN;libFASTSIM;libmicrocern;libpdf;libpythia6;libEGPythia6;libAliPythia6");
+  TString packages("PWG0base;PWG0dep");
 
-  fEsdTrackCuts = new AliESDtrackCuts();
-  fEsdTrackCuts->DefineHistograms(1);
+  if (!prepareQuery(libraries, packages, kTRUE))
+    return;
 
-  fEsdTrackCuts->SetMinNClustersTPC(50);
-  fEsdTrackCuts->SetMaxChi2PerClusterTPC(3.5);
-  fEsdTrackCuts->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);
-  fEsdTrackCuts->SetRequireTPCRefit(kTRUE);
+  gROOT->ProcessLine(".L CreateCuts.C");
 
-  fEsdTrackCuts->SetMinNsigmaToVertex(3);
-  fEsdTrackCuts->SetAcceptKingDaughters(kFALSE);
+  AliESDtrackCuts* esdTrackCuts = CreateTrackCuts();
+  if (!esdTrackCuts)
+  {
+    printf("ERROR: esdTrackCuts could not be created\n");
+    return;
+  }
 
-  chain->GetUserInfo()->Add(fEsdTrackCuts);
+  TList inputList;
+  inputList.Add(esdTrackCuts);
 
-  AliLog::SetClassDebugLevel("AlidNdEtaCorrectionSelector", AliLog::kInfo);
-  AliLog::SetClassDebugLevel("AliSelectorRL", AliLog::kInfo);
+  TChain* chain = CreateESDChain(dataDir, nRuns, offset);
 
-  TString selector("AlidNdEtaCorrectionSelector.cxx+");
+  TString selector("AlidNdEtaCorrectionSelector.cxx++");
   if (debug != kFALSE)
     selector += "g";
 
-  chain->Process(selector);
+  Int_t result = executeQuery(chain, &inputList, selector, option);
+}
+
+void VerifyCorrection(Int_t draw = 0, const char* correctionMapFile = "correction_map.root", const char* correctionMapFolder = "dndeta_correction", const char* analysisDir = "dndetaESD", const char* checkDir = "dndetaMC")
+{
+  gSystem->Load("libPWG0base");
+
+  AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection(correctionMapFolder, correctionMapFolder);
+  dNdEtaCorrection->LoadHistograms(correctionMapFile, correctionMapFolder);
+  dNdEtaCorrection->SetName("dndeta_correction");
+  dNdEtaCorrection->SetTitle("dndeta_correction");
+
+  TFile::Open(correctionMapFile);
+
+  dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis(analysisDir, analysisDir);
+  fdNdEtaAnalysis->LoadHistograms();
+
+  // correct with track2particle
+  TH3F* hist = fdNdEtaAnalysis->GetHistogram();
+  for (Int_t x=1; x<=hist->GetNbinsX(); ++x)
+    for (Int_t y=1; y<=hist->GetNbinsY(); ++y)
+      for (Int_t z=1; z<=hist->GetNbinsZ(); ++z)
+      {
+        Float_t correction = dNdEtaCorrection->GetTrack2ParticleCorrection(hist->GetXaxis()->GetBinCenter(x), hist->GetYaxis()->GetBinCenter(y), hist->GetZaxis()->GetBinCenter(z));
+        Float_t value = hist->GetBinContent(x, y, z);
+
+        hist->SetBinContent(x, y, z, correction * value);
+      }
+
+  fdNdEtaAnalysis->Finish(dNdEtaCorrection, 0.3);
+
+  dNdEtaAnalysis* fdNdEtaAnalysisMC = new dNdEtaAnalysis(checkDir, checkDir);
+  fdNdEtaAnalysisMC->LoadHistograms();
+
+  fdNdEtaAnalysisMC->Finish(0, 0.3);
+
+  if (draw == 0)
+    fdNdEtaAnalysis->DrawHistograms();
+  else if (draw == 1)
+    fdNdEtaAnalysisMC->DrawHistograms();
+  else
+  {
+    for (Int_t i=0; i<dNdEtaAnalysis::kVertexBinning; ++i)
+      fdNdEtaAnalysis->GetdNdEtaHistogram(i)->Divide(fdNdEtaAnalysisMC->GetdNdEtaHistogram(i));
+
+    fdNdEtaAnalysis->DrawHistograms();
+  }
 }