Fixes for coverity checks.

[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / qa / DrawNeighbors.C

diff --git a/PWG2/FORWARD/analysis2/qa/DrawNeighbors.C b/PWG2/FORWARD/analysis2/qa/DrawNeighbors.C
index 764d772bbb12fa2a6da46709d6d3b297c983c899..e43e44071d573a0ae0c2c6605d17a7692bf04a70 100644 (file)
--- a/PWG2/FORWARD/analysis2/qa/DrawNeighbors.C
+++ b/PWG2/FORWARD/analysis2/qa/DrawNeighbors.C
@@ -1,3 +1,18 @@
+#ifndef __CINT__
+# include <TH1.h>
+# include <TH2.h>
+# include <TList.h>
+# include <TFile.h>
+# include <TString.h>
+# include <TError.h>
+# include <TPad.h>
+# include <TCanvas.h>
+# include <TLine.h>
+# include <TLatex.h>
+# include <TStyle.h>
+#else
+class TList;
+#endif
 /** 
  * Draw the correlation of neighboring strips before/after merging 
  * 
@@ -65,7 +80,8 @@ DrawRingNeighbors(TList* p, UShort_t d, Char_t r)
  * @ingroup pwg2_forward_scripts_qa
  */
 void
-DrawNeighbors(const char* filename="forward.root")
+DrawNeighbors(const char* filename="forward.root", 
+             const char* folder="ForwardResults")
 {
   gStyle->SetPalette(1);
   gStyle->SetOptFit(0);
@@ -85,9 +101,9 @@ DrawNeighbors(const char* filename="forward.root")
     return;
   }
 
-  TList* forward = static_cast<TList*>(file->Get("Forward"));
+  TList* forward = static_cast<TList*>(file->Get(folder));
   if (!forward) { 
-    Error("DrawNeighbors", "List Forward not found in %s", filename);
+    Error("DrawNeighbors", "List %s not found in %s", folder, filename);
     return;
   }
 
@@ -110,6 +126,7 @@ DrawNeighbors(const char* filename="forward.root")
   c->cd(6); DrawRingNeighbors(sf, 3, 'O');
   c->cd(4)->SetFillColor(0);
   c->cd();
+  c->SaveAs("neighbors.png");
 }