minor changes to improve serializability

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

diff --git a/PWG0/dNdEta/CreatedNdEta.C b/PWG0/dNdEta/CreatedNdEta.C
index d176a8399595c6365c0ffb7eed2e22528ca44c7a..ec23af438a827721fd3ffecc0652a91fc33a7d5d 100644 (file)
--- a/PWG0/dNdEta/CreatedNdEta.C
+++ b/PWG0/dNdEta/CreatedNdEta.C
@@ -1,23 +1,28 @@
 // this macro combines the correction and the analysis and draws them
 
-void CreatedNdEta(Bool_t correct = kTRUE)
+void CreatedNdEta(Bool_t correct = kTRUE, const Char_t* filename = "analysis_esd.root", const char* object = "dndeta")
 {
   gSystem->Load("libPWG0base");
 
-  dNdEtaCorrection* dNdEtaCorrection = 0;
+  AlidNdEtaCorrection* dNdEtaCorrection = 0;
   if (correct)
   {
-    dNdEtaCorrection = new dNdEtaCorrection();
+    dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
     dNdEtaCorrection->LoadHistograms("correction_map.root","dndeta_correction");
-    dNdEtaCorrection->RemoveEdges(2, 0, 2);
+    //dNdEtaCorrection->RemoveEdges(2, 0, 2);
   }
 
-  fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
+  fdNdEtaAnalysis = new dNdEtaAnalysis(object, object);
 
-  TFile::Open("out.root");
+  TFile* file = TFile::Open(filename);
+  if (!file)
+  {
+    cout << "Error. File out.root not found" << endl;
+    return;
+  }
   fdNdEtaAnalysis->LoadHistograms();
 
-  fdNdEtaAnalysis->Finish(dNdEtaCorrection);
+  fdNdEtaAnalysis->Finish(dNdEtaCorrection, (correct) ? 0.3 : -1);
 
   fdNdEtaAnalysis->DrawHistograms();
 }