[u/mrichter/AliRoot.git] / PWGLF / FORWARD / analysis2 / qa / DrawRecAnaEloss.C

/**
 * @file   DrawRecAnaEloss.C
 * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
 * @date   Thu Jul  7 10:58:50 2011
 * 
 * @brief  Draw energ-loss before/after merging and used in the
 * density calculations 
 *
 * @deprecated Use QATrender instead
 * @ingroup pwglf_forward_scripts_qa
 */

#ifndef __CINT__
#include <TList.h>
#include <TH1.h>
#include <TH2.h>
#include <TCanvas.h>
#include <TLatex.h>
#include <TLine.h>
#include <TFile.h>
#include <TError.h>
#include <TParameter.h>
#include <TStyle.h>
#else
class TLatex;
#endif

/** 
 * Draw some text
 * 
 * @param l    LaTeX object
 * @param x    x coordinate
 * @param y    y coordinate (incremented on return)
 * @param c1   First string
 * @param c2   Second string 
 *
 * @deprecated Use QATrender instead
 * @ingroup pwglf_forward_scripts_qa
 */
void 
DrawText(TLatex* l, Double_t x, Double_t& y, const char* c1, const char* c2)
{
  y -= 1.2*l->GetTextSize();
  l->DrawLatex(x,    y, c1);
  l->DrawLatex(x+.4, y, c2);
}
/** 
 * Draw the energy loss before/after mergin for a single ring
 * 
 * @param p      List 1 
 * @param p2     List 2
 * @param lowCut Low cut
 * @param d      Detector
 * @param r      Ring 
 *
 * @deprecated Use QATrender instead
 * @ingroup pwglf_forward_scripts_qa
 */
void
DrawRingRecAnaEloss(TList* p, TList* p2, Double_t lowCut, UShort_t d, Char_t r)
{
  if (!p) return;

  TList* ring = static_cast<TList*>(p->FindObject(Form("FMD%d%c",d,r)));
  if (!ring) { 
    Error("DrawRecAnaEloss", "List FMD%d%c not found in %s",d,r,p->GetName());
    return;
  }
  TList* ring2 = static_cast<TList*>(p2->FindObject(Form("FMD%d%c",d,r)));
  if (!ring2){
    Error("DrawRecAnaEloss","List FMD%d%c not found in %s",d,r,p2->GetName());
    return;
  }

  TH1* before = static_cast<TH1D*>(ring->FindObject("esdEloss"));
  if (!before) { 
    Error("DrawRingRecAnaEloss", "Histogram esdEloss not found in FMD%d%c",
	  d, r);
    return;
  }
  TH1* after = static_cast<TH1D*>(ring->FindObject("anaEloss"));
  if (!after) { 
    Error("DrawRingRecAnaEloss", "Histogram anaEloss not found in FMD%d%c",
	  d, r);
    return;
  }
  TH1* presented = static_cast<TH1D*>(ring2->FindObject("eloss"));
  if (!presented) {
    Error("DrawRingRecAnaEloss", "Histogram eloss not found in FMD%d%c",
	  d, r);
    return;
  }
  TH1* used = static_cast<TH1D*>(ring2->FindObject("elossUsed"));
  if (!used) {
    Error("DrawRingRecAnaEloss", "Histogram elossUsed not found in FMD%d%c",
	  d, r);
    return;
  }


  Int_t low = before->GetXaxis()->FindBin(lowCut);
  Int_t ib  = Int_t(before->Integral(low,before->GetNbinsX()));
  Int_t ia  = Int_t(after->Integral(low,after->GetNbinsX()));
  Int_t ip  = Int_t(presented->Integral(low,presented->GetNbinsX()));
  Int_t iu  = Int_t(used->Integral(low,used->GetNbinsX()));
  // before->Scale(1. / ib);
  // after->Scale(1. / ib);
  // presented->Scale(1. / ib);
  // used->Scale(1. / ib);

  gPad->SetLogy();
  gPad->SetFillColor(0);
  before->SetTitle(Form("FMD%d%c",d,r));
  before->Draw("");
  after->Draw("same");
  presented->Draw("same");
  used->Draw("same");
  
  // ib           = before->Integral(low,before->GetNbinsX());
  // ia           = after->Integral(low,after->GetNbinsX());
  // ip           = presented->Integral(low,presented->GetNbinsX());
  // iu           = used->Integral(low,used->GetNbinsX());
  Double_t ts  = 0.03;
  Double_t x   = gPad->GetLeftMargin() + .25;
  Double_t y   = 1-gPad->GetTopMargin()-gStyle->GetTitleH()+ts;
  TLatex*  ltx = new TLatex(x, y, Form("FMD%d%c", d, r));
  ltx->SetNDC();
  ltx->SetTextAlign(13);
  ltx->SetTextSize(ts);
  // ltx->Draw();
  // ltx->SetTextSize(.05);
  TString inte(Form("Integral [%4.2f,#infty]", lowCut));
  DrawText(ltx, x, y, Form("%s before:", inte.Data()), Form("%9d", ib));
  DrawText(ltx, x, y, Form("%s after:",  inte.Data()), Form("%9d", ia));
  DrawText(ltx, x, y, Form("%s input:",  inte.Data()), Form("%9d", ip));
  DrawText(ltx, x, y, Form("%s user:",   inte.Data()), Form("%9d", iu));
  TLine* l = new TLine;
  l->SetLineWidth(1);
  l->DrawLineNDC(x, y-0.9*ts, 1-gPad->GetRightMargin()-0.01, y-0.9*ts);
  if (ib != 0 && ia != 0) {
    DrawText(ltx, x, y, "Change (merging)", Form("%5.1f%%", (100.*(ia-ib))/ib));
    DrawText(ltx, x, y, "Change (input)",   Form("%5.1f%% (%5.1f%%)", 
						 (100.*(ip-ia))/ia,
						 (100.*(ip-ib))/ib));
    DrawText(ltx, x, y, "Change (use)",     Form("%5.1f%% (%5.1f%%)", 
						 (100.*(iu-ip))/ip,
						 (100.*(iu-ib))/ib));
  }
  before->GetXaxis()->SetRangeUser(0, 4);
  gPad->cd();
}

/** 
 * Draw energy loss before/after merging 
 * 
 * @param filename Input file name
 * @param folder   Input folder name in file 
 *
 * @deprecated Use QATrender instead
 * @ingroup pwglf_forward_scripts_qa
 */
void
DrawRecAnaEloss(const char* filename="forward.root",
		const char* folder="ForwardResults")
{
  gStyle->SetPalette(1);
  gStyle->SetOptFit(0);
  gStyle->SetOptStat(0);
  gStyle->SetOptTitle(1);
  gStyle->SetTitleW(.4);
  gStyle->SetTitleH(.1);
  gStyle->SetTitleColor(0);
  gStyle->SetTitleStyle(0);
  gStyle->SetTitleBorderSize(0);
  gStyle->SetTitleX(.6);
  
  TFile* file = TFile::Open(filename, "READ");
  if (!file) { 
    Error("DrawRecAnaEloss", "failed to open %s", filename);
    return;
  }

  TList* forward = static_cast<TList*>(file->Get(folder));
  if (!forward) { 
    Error("DrawRecAnaEloss", "List %s not found in %s", folder, filename);
    return;
  }

  TList* sf = static_cast<TList*>(forward->FindObject("fmdSharingFilter"));
  if (!sf) { 
    Error("DrawRecAnaEloss", "List fmdSharingFilter not found in Forward");
    return;
  }
  TList* dc = static_cast<TList*>(forward->FindObject("fmdDensityCalculator"));
  if (!dc) {
    Error("DrawRecAnaEloss","List fmdDensityCalculator not found in Forward");
    return;
  }

  TParameter<double>* lowCut = 
    static_cast<TParameter<double>*>(sf->FindObject("lowCut"));
  Double_t low = (lowCut ? lowCut->GetVal() : 0.15);
  if (!lowCut)
    Warning("DrawRecAnaEloss", "Low cut not found in %s, assuming %f",
	    sf->GetName(), low);
  TCanvas* c = new TCanvas("recAnaELoss", 
			   "Reconstructed and Analysed energy loss", 900, 700);
  c->SetFillColor(0);
  c->SetBorderSize(0);
  c->SetLeftMargin(0.15);
  c->SetRightMargin(0.02);
  c->SetTopMargin(0.02);
  c->Divide(3, 2, 0, 0);
  
  c->cd(1); DrawRingRecAnaEloss(sf, dc, low, 1, 'I');
  c->cd(2); DrawRingRecAnaEloss(sf, dc, low, 2, 'I');
  c->cd(5); DrawRingRecAnaEloss(sf, dc, low, 2, 'O');
  c->cd(3); DrawRingRecAnaEloss(sf, dc, low, 3, 'I');
  c->cd(6); DrawRingRecAnaEloss(sf, dc, low, 3, 'O');
  TVirtualPad* p = c->cd(4);
  // p->SetTopMargin(0.05);
  p->SetRightMargin(0.15);
  p->SetFillColor(0);
  TH2D* highCuts = static_cast<TH2D*>(sf->FindObject("highCuts"));
  if (highCuts) highCuts->Draw("colz");
  c->cd();
  c->SaveAs("recAnaELoss.png");
}
 
//
// EOF
//
Commit	Line	Data
56199f2b	1	/**
	2	* @file DrawRecAnaEloss.C
	3	* @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
	4	* @date Thu Jul 7 10:58:50 2011
	5	*
	6	* @brief Draw energ-loss before/after merging and used in the
	7	* density calculations
	8	*
2dbde04b	9	* @deprecated Use QATrender instead
bd6f5206	10	* @ingroup pwglf_forward_scripts_qa
56199f2b	11	*/
56199f2b	12
b5c9a732	13	#ifndef __CINT__
	14	#include <TList.h>
	15	#include <TH1.h>
	16	#include <TH2.h>
	17	#include <TCanvas.h>
	18	#include <TLatex.h>
	19	#include <TLine.h>
	20	#include <TFile.h>
	21	#include <TError.h>
	22	#include <TParameter.h>
	23	#include <TStyle.h>
	24	#else
	25	class TLatex;
	26	#endif
	27
56199f2b	28	/**
	29	* Draw some text
	30	*
2dbde04b	31	* @param l LaTeX object
	32	* @param x x coordinate
	33	* @param y y coordinate (incremented on return)
	34	* @param c1 First string
	35	* @param c2 Second string
56199f2b	36	*
2dbde04b	37	* @deprecated Use QATrender instead
bd6f5206	38	* @ingroup pwglf_forward_scripts_qa
56199f2b	39	*/
b5c9a732	40	void
	41	DrawText(TLatex* l, Double_t x, Double_t& y, const char* c1, const char* c2)
	42	{
	43	y -= 1.2*l->GetTextSize();
	44	l->DrawLatex(x, y, c1);
	45	l->DrawLatex(x+.4, y, c2);
	46	}
56199f2b	47	/**
	48	* Draw the energy loss before/after mergin for a single ring
	49	*
	50	* @param p List 1
	51	* @param p2 List 2
	52	* @param lowCut Low cut
	53	* @param d Detector
	54	* @param r Ring
	55	*
2dbde04b	56	* @deprecated Use QATrender instead
bd6f5206	57	* @ingroup pwglf_forward_scripts_qa
56199f2b	58	*/
0919aa56	59	void
b5c9a732	60	DrawRingRecAnaEloss(TList* p, TList* p2, Double_t lowCut, UShort_t d, Char_t r)
0919aa56	61	{
	62	if (!p) return;
	63
	64	TList* ring = static_cast<TList*>(p->FindObject(Form("FMD%d%c",d,r)));
	65	if (!ring) {
	66	Error("DrawRecAnaEloss", "List FMD%d%c not found in %s",d,r,p->GetName());
	67	return;
	68	}
b5c9a732	69	TList* ring2 = static_cast<TList*>(p2->FindObject(Form("FMD%d%c",d,r)));
	70	if (!ring2){
	71	Error("DrawRecAnaEloss","List FMD%d%c not found in %s",d,r,p2->GetName());
	72	return;
	73	}
	74
	75	TH1* before = static_cast<TH1D*>(ring->FindObject("esdEloss"));
0919aa56	76	if (!before) {
	77	Error("DrawRingRecAnaEloss", "Histogram esdEloss not found in FMD%d%c",
	78	d, r);
	79	return;
	80	}
b5c9a732	81	TH1* after = static_cast<TH1D*>(ring->FindObject("anaEloss"));
0919aa56	82	if (!after) {
	83	Error("DrawRingRecAnaEloss", "Histogram anaEloss not found in FMD%d%c",
	84	d, r);
	85	return;
	86	}
b5c9a732	87	TH1* presented = static_cast<TH1D*>(ring2->FindObject("eloss"));
	88	if (!presented) {
	89	Error("DrawRingRecAnaEloss", "Histogram eloss not found in FMD%d%c",
	90	d, r);
	91	return;
	92	}
	93	TH1* used = static_cast<TH1D*>(ring2->FindObject("elossUsed"));
	94	if (!used) {
	95	Error("DrawRingRecAnaEloss", "Histogram elossUsed not found in FMD%d%c",
	96	d, r);
	97	return;
	98	}
	99
	100
	101	Int_t low = before->GetXaxis()->FindBin(lowCut);
	102	Int_t ib = Int_t(before->Integral(low,before->GetNbinsX()));
	103	Int_t ia = Int_t(after->Integral(low,after->GetNbinsX()));
	104	Int_t ip = Int_t(presented->Integral(low,presented->GetNbinsX()));
	105	Int_t iu = Int_t(used->Integral(low,used->GetNbinsX()));
	106	// before->Scale(1. / ib);
	107	// after->Scale(1. / ib);
	108	// presented->Scale(1. / ib);
	109	// used->Scale(1. / ib);
d015ecfe	110
0919aa56	111	gPad->SetLogy();
	112	gPad->SetFillColor(0);
	113	before->SetTitle(Form("FMD%d%c",d,r));
	114	before->Draw("");
	115	after->Draw("same");
b5c9a732	116	presented->Draw("same");
	117	used->Draw("same");
	118
	119	// ib = before->Integral(low,before->GetNbinsX());
	120	// ia = after->Integral(low,after->GetNbinsX());
	121	// ip = presented->Integral(low,presented->GetNbinsX());
	122	// iu = used->Integral(low,used->GetNbinsX());
	123	Double_t ts = 0.03;
	124	Double_t x = gPad->GetLeftMargin() + .25;
	125	Double_t y = 1-gPad->GetTopMargin()-gStyle->GetTitleH()+ts;
	126	TLatex* ltx = new TLatex(x, y, Form("FMD%d%c", d, r));
	127	ltx->SetNDC();
	128	ltx->SetTextAlign(13);
	129	ltx->SetTextSize(ts);
	130	// ltx->Draw();
	131	// ltx->SetTextSize(.05);
	132	TString inte(Form("Integral [%4.2f,#infty]", lowCut));
	133	DrawText(ltx, x, y, Form("%s before:", inte.Data()), Form("%9d", ib));
	134	DrawText(ltx, x, y, Form("%s after:", inte.Data()), Form("%9d", ia));
	135	DrawText(ltx, x, y, Form("%s input:", inte.Data()), Form("%9d", ip));
	136	DrawText(ltx, x, y, Form("%s user:", inte.Data()), Form("%9d", iu));
	137	TLine* l = new TLine;
	138	l->SetLineWidth(1);
	139	l->DrawLineNDC(x, y-0.9ts, 1-gPad->GetRightMargin()-0.01, y-0.9ts);
d015ecfe	140	if (ib != 0 && ia != 0) {
	141	DrawText(ltx, x, y, "Change (merging)", Form("%5.1f%%", (100.*(ia-ib))/ib));
	142	DrawText(ltx, x, y, "Change (input)", Form("%5.1f%% (%5.1f%%)",
	143	(100.*(ip-ia))/ia,
	144	(100.*(ip-ib))/ib));
	145	DrawText(ltx, x, y, "Change (use)", Form("%5.1f%% (%5.1f%%)",
	146	(100.*(iu-ip))/ip,
	147	(100.*(iu-ib))/ib));
	148	}
b5c9a732	149	before->GetXaxis()->SetRangeUser(0, 4);
0919aa56	150	gPad->cd();
	151	}
	152
56199f2b	153	/**
	154	* Draw energy loss before/after merging
	155	*
2dbde04b	156	* @param filename Input file name
2dbde04b	157	* @param folder Input folder name in file
56199f2b	158	*
2dbde04b	159	* @deprecated Use QATrender instead
bd6f5206	160	* @ingroup pwglf_forward_scripts_qa
56199f2b	161	*/
0919aa56	162	void
d015ecfe	163	DrawRecAnaEloss(const char* filename="forward.root",
d015ecfe	164	const char* folder="ForwardResults")
0919aa56	165	{
	166	gStyle->SetPalette(1);
	167	gStyle->SetOptFit(0);
	168	gStyle->SetOptStat(0);
b5c9a732	169	gStyle->SetOptTitle(1);
0919aa56	170	gStyle->SetTitleW(.4);
	171	gStyle->SetTitleH(.1);
	172	gStyle->SetTitleColor(0);
	173	gStyle->SetTitleStyle(0);
	174	gStyle->SetTitleBorderSize(0);
	175	gStyle->SetTitleX(.6);
	176
	177	TFile* file = TFile::Open(filename, "READ");
	178	if (!file) {
	179	Error("DrawRecAnaEloss", "failed to open %s", filename);
	180	return;
	181	}
	182
d015ecfe	183	TList* forward = static_cast<TList*>(file->Get(folder));
0919aa56	184	if (!forward) {
d015ecfe	185	Error("DrawRecAnaEloss", "List %s not found in %s", folder, filename);
0919aa56	186	return;
	187	}
	188
	189	TList* sf = static_cast<TList*>(forward->FindObject("fmdSharingFilter"));
	190	if (!sf) {
	191	Error("DrawRecAnaEloss", "List fmdSharingFilter not found in Forward");
	192	return;
	193	}
b5c9a732	194	TList* dc = static_cast<TList*>(forward->FindObject("fmdDensityCalculator"));
	195	if (!dc) {
	196	Error("DrawRecAnaEloss","List fmdDensityCalculator not found in Forward");
	197	return;
	198	}
	199
	200	TParameter<double>* lowCut =
	201	static_cast<TParameter<double>*>(sf->FindObject("lowCut"));
	202	Double_t low = (lowCut ? lowCut->GetVal() : 0.15);
	203	if (!lowCut)
	204	Warning("DrawRecAnaEloss", "Low cut not found in %s, assuming %f",
	205	sf->GetName(), low);
0919aa56	206	TCanvas* c = new TCanvas("recAnaELoss",
	207	"Reconstructed and Analysed energy loss", 900, 700);
	208	c->SetFillColor(0);
	209	c->SetBorderSize(0);
b5c9a732	210	c->SetLeftMargin(0.15);
	211	c->SetRightMargin(0.02);
	212	c->SetTopMargin(0.02);
0919aa56	213	c->Divide(3, 2, 0, 0);
0919aa56	214
b5c9a732	215	c->cd(1); DrawRingRecAnaEloss(sf, dc, low, 1, 'I');
	216	c->cd(2); DrawRingRecAnaEloss(sf, dc, low, 2, 'I');
	217	c->cd(5); DrawRingRecAnaEloss(sf, dc, low, 2, 'O');
	218	c->cd(3); DrawRingRecAnaEloss(sf, dc, low, 3, 'I');
	219	c->cd(6); DrawRingRecAnaEloss(sf, dc, low, 3, 'O');
0919aa56	220	TVirtualPad* p = c->cd(4);
	221	// p->SetTopMargin(0.05);
	222	p->SetRightMargin(0.15);
	223	p->SetFillColor(0);
	224	TH2D* highCuts = static_cast<TH2D*>(sf->FindObject("highCuts"));
	225	if (highCuts) highCuts->Draw("colz");
	226	c->cd();
d015ecfe	227	c->SaveAs("recAnaELoss.png");
0919aa56	228	}
0919aa56	229
56199f2b	230	//
	231	// EOF
	232	//