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

/* $Id$ */

Int_t gMax = 5;

void drawPlots()
{
  drawPlots(5);
  drawPlots(2);
}

void drawPlots(Int_t max)
{
  gMax = max;

  ptCutoff();
  TriggerBias();
  VtxRecon();
  Track2Particle2D();
  Track2Particle3D();
  ptSpectrum();
  dNdEta();
}

void dNdEta()
{
  TFile* file = TFile::Open("analysis_esd.root");
  TH1* histESD = file->Get("dndeta/dndeta_dNdEta_corrected_2");
  TH1* histESDNoPt = file->Get("dndeta/dndeta_dNdEta_2");
  TH1* histESDMB = file->Get("dndeta_mb/dndeta_mb_dNdEta_corrected_2");
  TH1* histESDMBVtx = file->Get("dndeta_mbvtx/dndeta_mbvtx_dNdEta_corrected_2");

  TCanvas* canvas = new TCanvas("dNdEta1", "dNdEta1", 500, 500);

  Prepare1DPlot(histESD);
  Prepare1DPlot(histESDNoPt);
  Prepare1DPlot(histESDMB);
  Prepare1DPlot(histESDMBVtx);

  histESD->SetLineColor(kRed);
  histESDMB->SetLineColor(kBlue);
  histESDMBVtx->SetLineColor(103);

  TH2F* dummy = new TH2F("dummy", "", 100, -1.5, 1.5, 100, 0, histESDMBVtx->GetMaximum() * 1.1);
  dummy->SetStats(kFALSE);
  dummy->SetXTitle("#eta");
  dummy->SetYTitle("dN/d#eta");

  histESDMBVtx->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
  histESDMB->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
  histESD->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
  histESDNoPt->GetXaxis()->SetRangeUser(-0.7999, 0.7999);

  dummy->Draw();
  histESDMBVtx->Draw("SAME");
  histESDMB->Draw("SAME");
  histESD->Draw("SAME");

  canvas->SaveAs("dNdEta1.gif");

  TFile* file2 = TFile::Open("analysis_mc.root");
  TH1* histMC = file2->Get("dndeta/dndeta_dNdEta_corrected_2")->Clone("cloned");

  gSystem->Load("libPWG0base");
  fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
  fdNdEtaAnalysis->LoadHistograms();
  fdNdEtaAnalysis->Finish(0, 0.3);
  TH1* histMCPtCut = fdNdEtaAnalysis->GetdNdEtaHistogram(2);

  TCanvas* canvas2 = new TCanvas("dNdEta2", "dNdEta2", 500, 500);

  Prepare1DPlot(histMC);
  Prepare1DPlot(histMCPtCut);

  histMC->SetLineColor(kBlue);
  histMCPtCut->SetLineColor(104);
  histESDNoPt->SetLineColor(102);

  TH2* dummy2 = dummy->Clone("dummy2");
  dummy2->GetYaxis()->SetRangeUser(0, histESD->GetMaximum() * 1.1);

  dummy2->Draw();
  histMC->Draw("SAME");
//  histMC->Draw();
  histESD->Draw("SAME");
  histESDNoPt->Draw("SAME");
  histMCPtCut->Draw("SAME");

  canvas2->SaveAs("dNdEta2.gif");
}

void ptSpectrum()
{
  TFile* file = TFile::Open("analysis_esd.root");
  TH1* histESD = file->Get("dndeta/dndeta_pt");

  TFile* file2 = TFile::Open("analysis_mc.root");
  TH1* histMC = file2->Get("dndeta/dndeta_pt");

  TCanvas* canvas = new TCanvas("ptSpectrum", "ptSpectrum", 500, 500);
  InitPad();
  gPad->SetLogy();

  Prepare1DPlot(histMC);
  Prepare1DPlot(histESD);

  histESD->SetTitle("");
  histESD->GetXaxis()->SetTitle("p_{T} [GeV/c]");
  histESD->GetYaxis()->SetTitle("#frac{dN}{d#eta dp_{T}} [c/GeV]");

  histMC->SetLineColor(kBlue);
  histESD->SetLineColor(kRed);

  histESD->GetYaxis()->SetTitleOffset(1.5);
  histESD->GetXaxis()->SetRangeUser(0, 4.9999);

  histESD->SetMaximum(TMath::Max(histESD->GetMaximum(), histMC->GetMaximum()) * 2);

  histESD->Draw();
  histMC->Draw("SAME");

  canvas->SaveAs("ptSpectrum.gif");
}

void ptCutoff()
{
  gSystem->Load("libPWG0base");

  AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
  dNdEtaCorrection->LoadHistograms("correction_map.root","dndeta_correction");

  dNdEtaCorrection->GetMeasuredFraction(0.3, -1, kTRUE);

  TH1* hist = dynamic_cast<TH1*> (gROOT->FindObject("gene_nTrackToNPart_pt"));

  hist->GetXaxis()->SetRangeUser(0, 0.9999);
  hist->SetMinimum(0);

  hist->SetTitle("Generated Particles");
  Prepare1DPlot(hist);

  TCanvas* canvas = new TCanvas("ptCutoff", "ptCutoff", 500, 500);
  hist->Draw();

  TLine* line = new TLine(0.3, 0 - hist->GetMaximum() * 0, 0.3, hist->GetMaximum() * 1.1);
  line->SetLineWidth(3);
  line->SetLineColor(kRed);
  line->Draw();

  canvas->SaveAs("ptCutoff.gif");
}

void TriggerBias()
{
  TFile* file = TFile::Open("correction_map.root");

  TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_trigger"));

  Prepare2DPlot(corr);
  corr->SetTitle("Trigger bias correction");

  TCanvas* canvas = new TCanvas("TriggerBias", "TriggerBias", 500, 500);
  InitPadCOLZ();
  corr->DrawCopy("COLZ");

  canvas->SaveAs(Form("TriggerBias_%d.gif", gMax));

  corr->GetYaxis()->SetRangeUser(0, 5);

  canvas = new TCanvas("TriggerBiasZoom", "TriggerBiasZoom", 500, 500);
  InitPadCOLZ();
  corr->DrawCopy("COLZ");

  canvas->SaveAs(Form("TriggerBiasZoom_%d.gif", gMax));
}

void VtxRecon()
{
  TFile* file = TFile::Open("correction_map.root");

  TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_vtxReco"));

  Prepare2DPlot(corr);
  corr->SetTitle("Vertex reconstruction correction");

  TCanvas* canvas = new TCanvas("VtxRecon", "VtxRecon", 500, 500);
  InitPadCOLZ();
  corr->DrawCopy("COLZ");

  canvas->SaveAs(Form("VtxRecon_%d.gif", gMax));


  corr->GetYaxis()->SetRangeUser(0, 5);

  TCanvas* canvas = new TCanvas("VtxReconZoom", "VtxReconZoom", 500, 500);
  InitPadCOLZ();
  corr->DrawCopy("COLZ");

  canvas->SaveAs(Form("VtxReconZoom_%d.gif", gMax));
}

void Track2Particle2D()
{
  gSystem->Load("libPWG0base");

  TFile* file = TFile::Open("correction_map.root");
  AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
  dNdEtaCorrection->LoadHistograms("correction_map.root","dndeta_correction");

  TH3* gene = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetGeneratedHistogram();
  TH3* meas = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetMeasuredHistogram();

  gene->GetZaxis()->SetRangeUser(0.3, 10);
  meas->GetZaxis()->SetRangeUser(0.3, 10);
  AliPWG0Helper::CreateDividedProjections(gene, meas, "yx");
  gene->GetZaxis()->UnZoom();
  meas->GetZaxis()->UnZoom();

  gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
  meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
  AliPWG0Helper::CreateDividedProjections(gene, meas, "zx");
  gene->GetYaxis()->UnZoom();
  meas->GetYaxis()->UnZoom();

  gene->GetXaxis()->SetRangeUser(-10, 10);
  meas->GetXaxis()->SetRangeUser(-10, 10);
  AliPWG0Helper::CreateDividedProjections(gene, meas, "zy");
  gene->GetXaxis()->UnZoom();
  meas->GetXaxis()->UnZoom();

  TH2* corrYX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx"));
  TH2* corrZX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx"));
  TH2* corrZY = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy"));

  /* this reads them from the file
  TH2* corrYX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx"));
  TH2* corrZX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx"));
  TH2* corrZY = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy"));*/

  Prepare2DPlot(corrYX);
  Prepare2DPlot(corrZX);
  Prepare2DPlot(corrZY);

  const char* title = "Track2Particle Correction";
  corrYX->SetTitle(title);
  corrZX->SetTitle(title);
  corrZY->SetTitle(title);

  TCanvas* canvas = new TCanvas("Track2Particle2D", "Track2Particle2D", 1200, 400);
  canvas->Divide(3, 1);

  canvas->cd(1);
  InitPadCOLZ();
  corrYX->Draw("COLZ");

  canvas->cd(2);
  InitPadCOLZ();
  corrZX->Draw("COLZ");

  canvas->cd(3);
  InitPadCOLZ();
  corrZY->Draw("COLZ");

  canvas->SaveAs(Form("Track2Particle2D_%d.gif", gMax));
}

void Track2Particle3D()
{
  // get left margin proper

  TFile* file = TFile::Open("correction_map.root");

  TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));

  corr->SetTitle("Correction Factor");
  SetRanges(corr->GetZaxis());

  Prepare3DPlot(corr);

  TCanvas* canvas = new TCanvas("Track2Particle3D", "Track2Particle3D", 500, 500);
  canvas->SetTheta(29.428);
  canvas->SetPhi(16.5726);

  corr->Draw();

  canvas->SaveAs("Track2Particle3D.gif");
}

void Track2Particle3DAll()
{
  TFile* file = TFile::Open("correction_map.root");

  TH3* gene = dynamic_cast<TH3*> (file->Get("dndeta_correction/gene_nTrackToNPart"));
  TH3* meas = dynamic_cast<TH3*> (file->Get("dndeta_correction/meas_nTrackToNPart"));
  TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));

  gene->SetTitle("Generated Particles");
  meas->SetTitle("Measured Tracks");
  corr->SetTitle("Correction Factor");

  Prepare3DPlot(gene);
  Prepare3DPlot(meas);
  Prepare3DPlot(corr);

  TCanvas* canvas = new TCanvas("Track2Particle3DAll", "Track2Particle3DAll", 1200, 400);
  canvas->Divide(3, 1);

  canvas->cd(1);
  InitPad();
  gene->Draw();

  canvas->cd(2);
  meas->Draw();

  canvas->cd(3);
  corr->Draw();

  canvas->SaveAs("Track2Particle3DAll.gif");
}

void SetRanges(TH1* hist)
{
  SetRanges(hist->GetXaxis());
  SetRanges(hist->GetYaxis());
  SetRanges(hist->GetZaxis());
}

void SetRanges(TAxis* axis)
{
  if (strcmp(axis->GetTitle(), "#eta") == 0)
    axis->SetRangeUser(-1.7999, 1.7999);
  if (strcmp(axis->GetTitle(), "p_{T} [GeV/c]") == 0)
    axis->SetRangeUser(0, 9.9999);
  if (strcmp(axis->GetTitle(), "vtx z [cm]") == 0)
    axis->SetRangeUser(-15, 14.9999);
  if (strcmp(axis->GetTitle(), "Ntracks") == 0)
    axis->SetRangeUser(0, 99.9999);
}

void Prepare3DPlot(TH3* hist)
{
  hist->GetXaxis()->SetTitleOffset(1.5);
  hist->GetYaxis()->SetTitleOffset(1.5);
  hist->GetZaxis()->SetTitleOffset(1.5);

  hist->SetStats(kFALSE);
}

void Prepare2DPlot(TH2* hist)
{
  hist->SetStats(kFALSE);
  hist->GetYaxis()->SetTitleOffset(1.4);

  hist->SetMinimum(0);
  hist->SetMaximum(gMax);

  SetRanges(hist);
}

void Prepare1DPlot(TH1* hist)
{
  hist->SetLineWidth(2);
  hist->SetStats(kFALSE);
}

void InitPad()
{
  gPad->Range(0, 0, 1, 1);
  gPad->SetLeftMargin(0.15);
  //gPad->SetRightMargin(0.05);
  //gPad->SetTopMargin(0.13);
  //gPad->SetBottomMargin(0.1);

  //gPad->SetGridx();
  //gPad->SetGridy();
}

void InitPadCOLZ()
{
  gPad->Range(0, 0, 1, 1);
  gPad->SetRightMargin(0.15);
  gPad->SetLeftMargin(0.12);

  gPad->SetGridx();
  gPad->SetGridy();
}
Commit	Line	Data
1afae8ff	1	/* $Id$ */
1afae8ff	2
25db2d85	3	Int_t gMax = 5;
25db2d85	4
1afae8ff	5	void drawPlots()
92d2d8ad	6	{
25db2d85	7	drawPlots(5);
	8	drawPlots(2);
	9	}
	10
	11	void drawPlots(Int_t max)
	12	{
	13	gMax = max;
	14
d09fb536	15	ptCutoff();
	16	TriggerBias();
	17	VtxRecon();
	18	Track2Particle2D();
	19	Track2Particle3D();
	20	ptSpectrum();
	21	dNdEta();
	22	}
	23
	24	void dNdEta()
	25	{
	26	TFile* file = TFile::Open("analysis_esd.root");
	27	TH1* histESD = file->Get("dndeta/dndeta_dNdEta_corrected_2");
	28	TH1* histESDNoPt = file->Get("dndeta/dndeta_dNdEta_2");
	29	TH1* histESDMB = file->Get("dndeta_mb/dndeta_mb_dNdEta_corrected_2");
	30	TH1* histESDMBVtx = file->Get("dndeta_mbvtx/dndeta_mbvtx_dNdEta_corrected_2");
	31
	32	TCanvas* canvas = new TCanvas("dNdEta1", "dNdEta1", 500, 500);
	33
	34	Prepare1DPlot(histESD);
	35	Prepare1DPlot(histESDNoPt);
	36	Prepare1DPlot(histESDMB);
	37	Prepare1DPlot(histESDMBVtx);
	38
	39	histESD->SetLineColor(kRed);
	40	histESDMB->SetLineColor(kBlue);
	41	histESDMBVtx->SetLineColor(103);
	42
	43	TH2F* dummy = new TH2F("dummy", "", 100, -1.5, 1.5, 100, 0, histESDMBVtx->GetMaximum() * 1.1);
	44	dummy->SetStats(kFALSE);
	45	dummy->SetXTitle("#eta");
	46	dummy->SetYTitle("dN/d#eta");
	47
	48	histESDMBVtx->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
	49	histESDMB->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
	50	histESD->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
	51	histESDNoPt->GetXaxis()->SetRangeUser(-0.7999, 0.7999);
	52
	53	dummy->Draw();
	54	histESDMBVtx->Draw("SAME");
	55	histESDMB->Draw("SAME");
	56	histESD->Draw("SAME");
	57
	58	canvas->SaveAs("dNdEta1.gif");
	59
	60	TFile* file2 = TFile::Open("analysis_mc.root");
	61	TH1* histMC = file2->Get("dndeta/dndeta_dNdEta_corrected_2")->Clone("cloned");
	62
	63	gSystem->Load("libPWG0base");
	64	fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
	65	fdNdEtaAnalysis->LoadHistograms();
	66	fdNdEtaAnalysis->Finish(0, 0.3);
	67	TH1* histMCPtCut = fdNdEtaAnalysis->GetdNdEtaHistogram(2);
	68
	69	TCanvas* canvas2 = new TCanvas("dNdEta2", "dNdEta2", 500, 500);
	70
	71	Prepare1DPlot(histMC);
	72	Prepare1DPlot(histMCPtCut);
	73
	74	histMC->SetLineColor(kBlue);
	75	histMCPtCut->SetLineColor(104);
	76	histESDNoPt->SetLineColor(102);
	77
	78	TH2* dummy2 = dummy->Clone("dummy2");
79	dummy2->GetYaxis()->SetRangeUser(0, histESD->GetMaximum() * 1.1);
80
81	dummy2->Draw();
82	histMC->Draw("SAME");
83	// histMC->Draw();
84	histESD->Draw("SAME");
85	histESDNoPt->Draw("SAME");
86	histMCPtCut->Draw("SAME");
87
88	canvas2->SaveAs("dNdEta2.gif");
89	}
90
91	void ptSpectrum()
92	{
93	TFile* file = TFile::Open("analysis_esd.root");
94	TH1* histESD = file->Get("dndeta/dndeta_pt");
95
96	TFile* file2 = TFile::Open("analysis_mc.root");
97	TH1* histMC = file2->Get("dndeta/dndeta_pt");
98
99	TCanvas* canvas = new TCanvas("ptSpectrum", "ptSpectrum", 500, 500);
100	InitPad();
101	gPad->SetLogy();
102
103	Prepare1DPlot(histMC);
104	Prepare1DPlot(histESD);
105
106	histESD->SetTitle("");
25db2d85	107	histESD->GetXaxis()->SetTitle("p_{T} [GeV/c]");
25db2d85	108	histESD->GetYaxis()->SetTitle("#frac{dN}{d#eta dp_{T}} [c/GeV]");
d09fb536	109
	110	histMC->SetLineColor(kBlue);
	111	histESD->SetLineColor(kRed);
	112
	113	histESD->GetYaxis()->SetTitleOffset(1.5);
	114	histESD->GetXaxis()->SetRangeUser(0, 4.9999);
	115
	116	histESD->SetMaximum(TMath::Max(histESD->GetMaximum(), histMC->GetMaximum()) * 2);
	117
	118	histESD->Draw();
	119	histMC->Draw("SAME");
	120
	121	canvas->SaveAs("ptSpectrum.gif");
92d2d8ad	122	}
	123
	124	void ptCutoff()
	125	{
	126	gSystem->Load("libPWG0base");
	127
8b3563f4	128	AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
92d2d8ad	129	dNdEtaCorrection->LoadHistograms("correction_map.root","dndeta_correction");
	130
	131	dNdEtaCorrection->GetMeasuredFraction(0.3, -1, kTRUE);
	132
	133	TH1* hist = dynamic_cast<TH1*> (gROOT->FindObject("gene_nTrackToNPart_pt"));
	134
	135	hist->GetXaxis()->SetRangeUser(0, 0.9999);
	136	hist->SetMinimum(0);
92d2d8ad	137
92d2d8ad	138	hist->SetTitle("Generated Particles");
d09fb536	139	Prepare1DPlot(hist);
92d2d8ad	140
	141	TCanvas* canvas = new TCanvas("ptCutoff", "ptCutoff", 500, 500);
	142	hist->Draw();
	143
d09fb536	144	TLine* line = new TLine(0.3, 0 - hist->GetMaximum() * 0, 0.3, hist->GetMaximum() * 1.1);
92d2d8ad	145	line->SetLineWidth(3);
d09fb536	146	line->SetLineColor(kRed);
92d2d8ad	147	line->Draw();
	148
	149	canvas->SaveAs("ptCutoff.gif");
	150	}
	151
	152	void TriggerBias()
	153	{
	154	TFile* file = TFile::Open("correction_map.root");
	155
	156	TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_trigger"));
	157
	158	Prepare2DPlot(corr);
	159	corr->SetTitle("Trigger bias correction");
	160
	161	TCanvas* canvas = new TCanvas("TriggerBias", "TriggerBias", 500, 500);
	162	InitPadCOLZ();
	163	corr->DrawCopy("COLZ");
	164
25db2d85	165	canvas->SaveAs(Form("TriggerBias_%d.gif", gMax));
92d2d8ad	166
	167	corr->GetYaxis()->SetRangeUser(0, 5);
	168
	169	canvas = new TCanvas("TriggerBiasZoom", "TriggerBiasZoom", 500, 500);
	170	InitPadCOLZ();
	171	corr->DrawCopy("COLZ");
	172
25db2d85	173	canvas->SaveAs(Form("TriggerBiasZoom_%d.gif", gMax));
92d2d8ad	174	}
	175
	176	void VtxRecon()
	177	{
	178	TFile* file = TFile::Open("correction_map.root");
	179
	180	TH2* corr = dynamic_cast<TH2*> (file->Get("dndeta_correction/corr_vtxReco"));
	181
	182	Prepare2DPlot(corr);
	183	corr->SetTitle("Vertex reconstruction correction");
	184
	185	TCanvas* canvas = new TCanvas("VtxRecon", "VtxRecon", 500, 500);
	186	InitPadCOLZ();
25db2d85	187	corr->DrawCopy("COLZ");
	188
	189	canvas->SaveAs(Form("VtxRecon_%d.gif", gMax));
	190
	191
	192	corr->GetYaxis()->SetRangeUser(0, 5);
	193
	194	TCanvas* canvas = new TCanvas("VtxReconZoom", "VtxReconZoom", 500, 500);
	195	InitPadCOLZ();
	196	corr->DrawCopy("COLZ");
92d2d8ad	197
25db2d85	198	canvas->SaveAs(Form("VtxReconZoom_%d.gif", gMax));
92d2d8ad	199	}
	200
	201	void Track2Particle2D()
1afae8ff	202	{
25db2d85	203	gSystem->Load("libPWG0base");
25db2d85	204
1afae8ff	205	TFile* file = TFile::Open("correction_map.root");
8b3563f4	206	AlidNdEtaCorrection* dNdEtaCorrection = new AlidNdEtaCorrection("dndeta_correction", "dndeta_correction");
25db2d85	207	dNdEtaCorrection->LoadHistograms("correction_map.root","dndeta_correction");
	208
	209	TH3* gene = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetGeneratedHistogram();
	210	TH3* meas = dNdEtaCorrection->GetTrack2ParticleCorrection()->GetMeasuredHistogram();
1afae8ff	211
25db2d85	212	gene->GetZaxis()->SetRangeUser(0.3, 10);
	213	meas->GetZaxis()->SetRangeUser(0.3, 10);
	214	AliPWG0Helper::CreateDividedProjections(gene, meas, "yx");
	215	gene->GetZaxis()->UnZoom();
	216	meas->GetZaxis()->UnZoom();
	217
	218	gene->GetYaxis()->SetRangeUser(-0.8, 0.8);
	219	meas->GetYaxis()->SetRangeUser(-0.8, 0.8);
	220	AliPWG0Helper::CreateDividedProjections(gene, meas, "zx");
	221	gene->GetYaxis()->UnZoom();
	222	meas->GetYaxis()->UnZoom();
	223
	224	gene->GetXaxis()->SetRangeUser(-10, 10);
	225	meas->GetXaxis()->SetRangeUser(-10, 10);
	226	AliPWG0Helper::CreateDividedProjections(gene, meas, "zy");
	227	gene->GetXaxis()->UnZoom();
	228	meas->GetXaxis()->UnZoom();
	229
	230	TH2* corrYX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx"));
	231	TH2* corrZX = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx"));
	232	TH2* corrZY = dynamic_cast<TH2*> (gROOT->FindObject("gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy"));
	233
	234	/* this reads them from the file
d09fb536	235	TH2* corrYX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_yx_div_meas_nTrackToNPart_yx"));
d09fb536	236	TH2* corrZX = dynamic_cast<TH2*> (file->Get("dndeta_correction/gene_nTrackToNPart_zx_div_meas_nTrackToNPart_zx"));
25db2d85	237	TH2* corrZY = dynamic_cast<TH2> (file->Get("dndeta_correction/gene_nTrackToNPart_zy_div_meas_nTrackToNPart_zy"));/
1afae8ff	238
	239	Prepare2DPlot(corrYX);
	240	Prepare2DPlot(corrZX);
	241	Prepare2DPlot(corrZY);
	242
92d2d8ad	243	const char* title = "Track2Particle Correction";
	244	corrYX->SetTitle(title);
	245	corrZX->SetTitle(title);
	246	corrZY->SetTitle(title);
1afae8ff	247
d09fb536	248	TCanvas* canvas = new TCanvas("Track2Particle2D", "Track2Particle2D", 1200, 400);
1afae8ff	249	canvas->Divide(3, 1);
	250
	251	canvas->cd(1);
	252	InitPadCOLZ();
	253	corrYX->Draw("COLZ");
	254
	255	canvas->cd(2);
	256	InitPadCOLZ();
	257	corrZX->Draw("COLZ");
	258
	259	canvas->cd(3);
	260	InitPadCOLZ();
	261	corrZY->Draw("COLZ");
92d2d8ad	262
25db2d85	263	canvas->SaveAs(Form("Track2Particle2D_%d.gif", gMax));
1afae8ff	264	}
	265
	266	void Track2Particle3D()
	267	{
	268	// get left margin proper
	269
	270	TFile* file = TFile::Open("correction_map.root");
	271
d09fb536	272	TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));
	273
	274	corr->SetTitle("Correction Factor");
	275	SetRanges(corr->GetZaxis());
	276
	277	Prepare3DPlot(corr);
	278
	279	TCanvas* canvas = new TCanvas("Track2Particle3D", "Track2Particle3D", 500, 500);
	280	canvas->SetTheta(29.428);
	281	canvas->SetPhi(16.5726);
	282
	283	corr->Draw();
	284
	285	canvas->SaveAs("Track2Particle3D.gif");
	286	}
	287
	288	void Track2Particle3DAll()
	289	{
d09fb536	290	TFile* file = TFile::Open("correction_map.root");
d09fb536	291
1afae8ff	292	TH3* gene = dynamic_cast<TH3*> (file->Get("dndeta_correction/gene_nTrackToNPart"));
	293	TH3* meas = dynamic_cast<TH3*> (file->Get("dndeta_correction/meas_nTrackToNPart"));
	294	TH3* corr = dynamic_cast<TH3*> (file->Get("dndeta_correction/corr_nTrackToNPart"));
	295
	296	gene->SetTitle("Generated Particles");
	297	meas->SetTitle("Measured Tracks");
	298	corr->SetTitle("Correction Factor");
	299
	300	Prepare3DPlot(gene);
	301	Prepare3DPlot(meas);
	302	Prepare3DPlot(corr);
	303
d09fb536	304	TCanvas* canvas = new TCanvas("Track2Particle3DAll", "Track2Particle3DAll", 1200, 400);
1afae8ff	305	canvas->Divide(3, 1);
	306
	307	canvas->cd(1);
	308	InitPad();
	309	gene->Draw();
	310
	311	canvas->cd(2);
	312	meas->Draw();
	313
	314	canvas->cd(3);
	315	corr->Draw();
d09fb536	316
d09fb536	317	canvas->SaveAs("Track2Particle3DAll.gif");
1afae8ff	318	}
	319
	320	void SetRanges(TH1* hist)
	321	{
	322	SetRanges(hist->GetXaxis());
	323	SetRanges(hist->GetYaxis());
	324	SetRanges(hist->GetZaxis());
	325	}
	326
	327	void SetRanges(TAxis* axis)
	328	{
	329	if (strcmp(axis->GetTitle(), "#eta") == 0)
d09fb536	330	axis->SetRangeUser(-1.7999, 1.7999);
92d2d8ad	331	if (strcmp(axis->GetTitle(), "p_{T} [GeV/c]") == 0)
1afae8ff	332	axis->SetRangeUser(0, 9.9999);
1afae8ff	333	if (strcmp(axis->GetTitle(), "vtx z [cm]") == 0)
d09fb536	334	axis->SetRangeUser(-15, 14.9999);
92d2d8ad	335	if (strcmp(axis->GetTitle(), "Ntracks") == 0)
92d2d8ad	336	axis->SetRangeUser(0, 99.9999);
1afae8ff	337	}
	338
	339	void Prepare3DPlot(TH3* hist)
	340	{
	341	hist->GetXaxis()->SetTitleOffset(1.5);
	342	hist->GetYaxis()->SetTitleOffset(1.5);
	343	hist->GetZaxis()->SetTitleOffset(1.5);
d09fb536	344
d09fb536	345	hist->SetStats(kFALSE);
1afae8ff	346	}
	347
	348	void Prepare2DPlot(TH2* hist)
	349	{
	350	hist->SetStats(kFALSE);
92d2d8ad	351	hist->GetYaxis()->SetTitleOffset(1.4);
	352
	353	hist->SetMinimum(0);
25db2d85	354	hist->SetMaximum(gMax);
92d2d8ad	355
92d2d8ad	356	SetRanges(hist);
1afae8ff	357	}
1afae8ff	358
d09fb536	359	void Prepare1DPlot(TH1* hist)
	360	{
	361	hist->SetLineWidth(2);
	362	hist->SetStats(kFALSE);
	363	}
	364
1afae8ff	365	void InitPad()
	366	{
	367	gPad->Range(0, 0, 1, 1);
d09fb536	368	gPad->SetLeftMargin(0.15);
	369	//gPad->SetRightMargin(0.05);
	370	//gPad->SetTopMargin(0.13);
	371	//gPad->SetBottomMargin(0.1);
1afae8ff	372
d09fb536	373	//gPad->SetGridx();
d09fb536	374	//gPad->SetGridy();
1afae8ff	375	}
	376
	377	void InitPadCOLZ()
d09fb536	378	{
1afae8ff	379	gPad->Range(0, 0, 1, 1);
	380	gPad->SetRightMargin(0.15);
	381	gPad->SetLeftMargin(0.12);
d09fb536	382
	383	gPad->SetGridx();
	384	gPad->SetGridy();
1afae8ff	385	}