[u/mrichter/AliRoot.git] / PWGPP / analysisQA / processDrawUDQA.C

void processDrawUDQA(const char *infile="AnalysisResults.root", 
		     TString suffix = "eps", 
		     const char *outfile="DrawUDQA_outfile.root") {

TFile *f = TFile::Open(infile, "read");

if(!f) {
 Printf("FATAL: File doesn't exist");
 return;
}

TList *flistQA = (TList*)f->Get("UpcTree/ListHist");

if(!flistQA) {
 Printf("FATAL: TList - UpcTree/ListHist -  doesn't exist");
 return;
}


TH1D *fHistNeventsJPsi = dynamic_cast<TH1D*> (flistQA->At(0));
fHistNeventsJPsi->GetYaxis()->SetTitle("Counts (Events)");

TH2D *fHistTPCsignalJPsi = dynamic_cast<TH2D*> (flistQA->At(1));
fHistTPCsignalJPsi->GetXaxis()->SetTitle("d#it{E}/d#it{x}^{TPC} (#it{l}^{+}) (a.u.)");
fHistTPCsignalJPsi->GetYaxis()->SetTitle("d#it{E}/d#it{x}^{TPC} (#it{l}^{-}) (a.u.)");

TH2D *fHistDiLeptonPtJPsi = dynamic_cast<TH2D*> (flistQA->At(2));
fHistDiLeptonPtJPsi->GetXaxis()->SetTitle("#it{p}_{T} (#it{l}^{+}) (GeV/#it{c})");
fHistDiLeptonPtJPsi->GetYaxis()->SetTitle("#it{p}_{T} (#it{l}^{-}) (GeV/#it{c})");

TH1D *fHistDiElectronMass = dynamic_cast<TH1D*> (flistQA->At(3));
fHistDiElectronMass->GetYaxis()->SetTitle("dN/dM");

TH1D *fHistDiMuonMass = dynamic_cast<TH1D*> (flistQA->At(4));
fHistDiMuonMass->GetYaxis()->SetTitle("dN/dM");

//___________________________________
  // Added by sjena
TFile *fout = TFile::Open(outfile,"UPDATE");
  fout->ls();
  
  TDirectoryFile *cdd = NULL;
  cdd = (TDirectoryFile*)fout->Get("UD");
  if(!cdd) {
    Printf("Warning: UD <dir> doesn't exist, creating a new one");
    cdd = (TDirectoryFile*)fout->mkdir("UD");
  }
  cdd->cd();
  cdd->ls();

  fHistNeventsJPsi->Write();
  fHistTPCsignalJPsi->Write();
  fHistDiLeptonPtJPsi->Write();
  fHistDiElectronMass->Write();
  fHistDiMuonMass->Write();
  fout->Close();
//___________________________________


  myOptions();
  gROOT->ForceStyle();

  TDatime now;
  int iDate = now.GetDate();
  int iYear=iDate/10000;
  int iMonth=(iDate%10000)/100;
  int iDay=iDate%100;
  char* cMonth[12]={"Jan","Feb","Mar","Apr","May","Jun",
		    "Jul","Aug","Sep","Oct","Nov","Dec"};
  char cStamp1[25],cStamp2[25];
  sprintf(cStamp1,"%i %s %i",iDay, cMonth[iMonth-1], iYear);
  sprintf(cStamp2,"%i/%.2d/%i",iDay, iMonth, iYear);

myHistSetUp(fHistNeventsJPsi);
fHistNeventsJPsi->SetLineWidth(3.0);
myHistSetUp(fHistTPCsignalJPsi);
myHistSetUp(fHistDiLeptonPtJPsi);
myHistSetUp(fHistDiElectronMass);
fHistDiElectronMass->SetLineWidth(2.0);
myHistSetUp(fHistDiMuonMass);
fHistDiMuonMass->SetLineWidth(2.0);

TCanvas *c1 = new TCanvas("fig_ud_NeventsJPsi"," ",800,400);
c1->Draw();
c1->cd();
TPad *myPad1 = new TPad("myPad1", "The pad",0,0,1,1);
myPadSetUp(myPad1,0.15,0.1,0.04,0.15);
myPad1->SetLogy();
myPad1->Draw();
myPad1->cd();

fHistNeventsJPsi->Draw();
 c1->SaveAs(Form("fig_ud_NeventsJPsi.%s",suffix.Data()));


TCanvas *c21 = new TCanvas("fig_ud_TPCsignalJPsi"," ",500,500);
c21->Draw();
c21->cd();
TPad *myPad21 = new TPad("myPad21", "The pad",0,0,1,1);
myPadSetUp(myPad21,0.15,0.15,0.15,0.15);
myPad21->Draw();
myPad21->cd();
fHistTPCsignalJPsi->Draw("COLZ");
 c21->SaveAs(Form("fig_ud_TPCsignalJPsi.%s",suffix.Data()));


TCanvas *c22 = new TCanvas("fig_ud_DiLeptonPtJPsi"," ",500,500);
c22->Draw();
c22->cd();
TPad *myPad22 = new TPad("myPad22", "The pad",0,0,1,1);
myPadSetUp(myPad22,0.15,0.15,0.15,0.15);
myPad22->Draw();
myPad22->cd();
fHistDiLeptonPtJPsi->Draw("COLZ");
 c22->SaveAs(Form("fig_ud_DiLeptonPtJPsi.%s",suffix.Data()));
 

TCanvas *c31 = new TCanvas("fig_ud_DiElectronMassJPsi"," ",800,500);
c31->Draw();
c31->cd();
TPad *myPad31 = new TPad("myPad31", "The pad",0,0,1,1);
myPadSetUp(myPad31,0.15,0.1,0.04,0.15);
myPad31->Draw();
myPad31->cd();
fHistDiElectronMass->Draw();
 c31->SaveAs(Form("fig_ud_DiElectronMassJPsi.%s",suffix.Data()));
 

TCanvas *c32 = new TCanvas("fig_ud_DiMuonMassJPsi"," ",800,500);
c32->Draw();
c32->cd();
TPad *myPad32 = new TPad("myPad32", "The pad",0,0,1,1);
myPadSetUp(myPad32,0.15,0.1,0.04,0.15);
myPad32->Draw();
myPad32->cd();
fHistDiMuonMass->Draw();
 c32->SaveAs(Form("fig_ud_DiMuonMassJPsi.%s",suffix.Data()));
 

/*
   TFile* outfile = new TFile("fig_ud.root","recreate");
   c1->Write();
   c21->Write();
   c31->Write();
   c22->Write();
   c32->Write();
 */

}

void myPadSetUp(TPad *currentPad, float currentLeft=0.11, float currentTop=0.04, float currentRight=0.04, float currentBottom=0.15){
  currentPad->SetLeftMargin(currentLeft);
  currentPad->SetTopMargin(currentTop);
  currentPad->SetRightMargin(currentRight);
  currentPad->SetBottomMargin(currentBottom);
  return;
}


void myHistSetUp(TH1 *currentGraph=0){
 
  currentGraph->SetLabelSize(0.05,"xyz");
  currentGraph->SetLabelFont(42,"xyz"); 
  currentGraph->SetLabelOffset(0.01,"xyz");
  currentGraph->SetTitleFont(42,"xyz");   
  currentGraph->GetXaxis()->SetTitleOffset(1.1);
  currentGraph->GetYaxis()->SetTitleOffset(1.1);  
  currentGraph->SetTitleSize(0.06,"xyz");
  currentGraph->SetStats(kFALSE); 
  currentGraph->SetTitle("");
  return;
}
void myHistSetUp(TH2 *currentGraph=0){
 
  currentGraph->SetLabelSize(0.05,"xyz");
  currentGraph->SetLabelFont(42,"xyz"); 
  currentGraph->SetLabelOffset(0.01,"xyz");
  currentGraph->SetTitleFont(42,"xyz"); 
  currentGraph->GetXaxis()->SetTitleOffset(1.3);
  currentGraph->GetYaxis()->SetTitleOffset(1.3);  
  currentGraph->SetTitleSize(0.05,"xyz");
  currentGraph->SetStats(kFALSE);
  currentGraph->SetTitle(""); 
  return;
}

void myOptions(Int_t lStat=0){
  // Set gStyle
  int font = 42;
  // From plain
  gStyle->SetFrameBorderMode(0);
  gStyle->SetFrameFillColor(0);
  gStyle->SetCanvasBorderMode(0);
  gStyle->SetPadBorderMode(0);
  gStyle->SetPadColor(10);
  gStyle->SetCanvasColor(10);
  gStyle->SetTitleFillColor(10);
  gStyle->SetTitleBorderSize(1);
  gStyle->SetStatColor(10);
  gStyle->SetStatBorderSize(1);
  gStyle->SetLegendBorderSize(1);
  //
  gStyle->SetDrawBorder(0);
  gStyle->SetTextFont(font);
  gStyle->SetStatFont(font);
  gStyle->SetStatFontSize(0.05);
  gStyle->SetStatX(0.97);
  gStyle->SetStatY(0.98);
  gStyle->SetStatH(0.03);
  gStyle->SetStatW(0.3);
  gStyle->SetTickLength(0.02,"y");
  gStyle->SetEndErrorSize(3);
  gStyle->SetLabelSize(0.05,"xyz");
  gStyle->SetLabelFont(font,"xyz"); 
  gStyle->SetLabelOffset(0.01,"xyz");
  gStyle->SetTitleFont(font,"xyz");  
  gStyle->SetTitleOffset(1.0,"xyz");  
  gStyle->SetTitleSize(0.06,"xyz");  
  gStyle->SetMarkerSize(1); 
  gStyle->SetPalette(1,0); 
  if (lStat){
    gStyle->SetOptTitle(1);
    gStyle->SetOptStat(1111);
    gStyle->SetOptFit(1111);
    }
  else {
    gStyle->SetOptTitle(0);
    gStyle->SetOptStat(0);
    gStyle->SetOptFit(0);
  }
}
Commit	Line	Data
948b38b2	1	void processDrawUDQA(const char *infile="AnalysisResults.root",
	2	TString suffix = "eps",
	3	const char *outfile="DrawUDQA_outfile.root") {
	4
	5	TFile *f = TFile::Open(infile, "read");
	6
	7	if(!f) {
	8	Printf("FATAL: File doesn't exist");
	9	return;
	10	}
	11
	12	TList flistQA = (TList)f->Get("UpcTree/ListHist");
	13
	14	if(!flistQA) {
	15	Printf("FATAL: TList - UpcTree/ListHist - doesn't exist");
	16	return;
	17	}
	18
	19
	20
	21	TH1D fHistNeventsJPsi = dynamic_cast<TH1D> (flistQA->At(0));
	22	fHistNeventsJPsi->GetYaxis()->SetTitle("Counts (Events)");
	23
	24	TH2D fHistTPCsignalJPsi = dynamic_cast<TH2D> (flistQA->At(1));
	25	fHistTPCsignalJPsi->GetXaxis()->SetTitle("d#it{E}/d#it{x}^{TPC} (#it{l}^{+}) (a.u.)");
	26	fHistTPCsignalJPsi->GetYaxis()->SetTitle("d#it{E}/d#it{x}^{TPC} (#it{l}^{-}) (a.u.)");
	27
	28	TH2D fHistDiLeptonPtJPsi = dynamic_cast<TH2D> (flistQA->At(2));
	29	fHistDiLeptonPtJPsi->GetXaxis()->SetTitle("#it{p}_{T} (#it{l}^{+}) (GeV/#it{c})");
	30	fHistDiLeptonPtJPsi->GetYaxis()->SetTitle("#it{p}_{T} (#it{l}^{-}) (GeV/#it{c})");
	31
	32	TH1D fHistDiElectronMass = dynamic_cast<TH1D> (flistQA->At(3));
	33	fHistDiElectronMass->GetYaxis()->SetTitle("dN/dM");
	34
	35	TH1D fHistDiMuonMass = dynamic_cast<TH1D> (flistQA->At(4));
	36	fHistDiMuonMass->GetYaxis()->SetTitle("dN/dM");
	37
	38	//___________________________________
	39	// Added by sjena
	40	TFile *fout = TFile::Open(outfile,"UPDATE");
	41	fout->ls();
	42
	43	TDirectoryFile *cdd = NULL;
	44	cdd = (TDirectoryFile*)fout->Get("UD");
	45	if(!cdd) {
	46	Printf("Warning: UD <dir> doesn't exist, creating a new one");
	47	cdd = (TDirectoryFile*)fout->mkdir("UD");
	48	}
	49	cdd->cd();
	50	cdd->ls();
	51
	52	fHistNeventsJPsi->Write();
	53	fHistTPCsignalJPsi->Write();
	54	fHistDiLeptonPtJPsi->Write();
	55	fHistDiElectronMass->Write();
	56	fHistDiMuonMass->Write();
	57	fout->Close();
	58	//___________________________________
	59
	60
	61
	62
	63	myOptions();
	64	gROOT->ForceStyle();
65
66	TDatime now;
67	int iDate = now.GetDate();
68	int iYear=iDate/10000;
69	int iMonth=(iDate%10000)/100;
70	int iDay=iDate%100;
71	char* cMonth[12]={"Jan","Feb","Mar","Apr","May","Jun",
72	"Jul","Aug","Sep","Oct","Nov","Dec"};
73	char cStamp1[25],cStamp2[25];
74	sprintf(cStamp1,"%i %s %i",iDay, cMonth[iMonth-1], iYear);
75	sprintf(cStamp2,"%i/%.2d/%i",iDay, iMonth, iYear);
76
77	myHistSetUp(fHistNeventsJPsi);
78	fHistNeventsJPsi->SetLineWidth(3.0);
79	myHistSetUp(fHistTPCsignalJPsi);
80	myHistSetUp(fHistDiLeptonPtJPsi);
81	myHistSetUp(fHistDiElectronMass);
82	fHistDiElectronMass->SetLineWidth(2.0);
83	myHistSetUp(fHistDiMuonMass);
84	fHistDiMuonMass->SetLineWidth(2.0);
85
86	TCanvas *c1 = new TCanvas("fig_ud_NeventsJPsi"," ",800,400);
87	c1->Draw();
88	c1->cd();
89	TPad *myPad1 = new TPad("myPad1", "The pad",0,0,1,1);
90	myPadSetUp(myPad1,0.15,0.1,0.04,0.15);
91	myPad1->SetLogy();
92	myPad1->Draw();
93	myPad1->cd();
94
95	fHistNeventsJPsi->Draw();
96	c1->SaveAs(Form("fig_ud_NeventsJPsi.%s",suffix.Data()));
97
98
99	TCanvas *c21 = new TCanvas("fig_ud_TPCsignalJPsi"," ",500,500);
100	c21->Draw();
101	c21->cd();
102	TPad *myPad21 = new TPad("myPad21", "The pad",0,0,1,1);
103	myPadSetUp(myPad21,0.15,0.15,0.15,0.15);
104	myPad21->Draw();
105	myPad21->cd();
106	fHistTPCsignalJPsi->Draw("COLZ");
107	c21->SaveAs(Form("fig_ud_TPCsignalJPsi.%s",suffix.Data()));
108
109
110	TCanvas *c22 = new TCanvas("fig_ud_DiLeptonPtJPsi"," ",500,500);
111	c22->Draw();
112	c22->cd();
113	TPad *myPad22 = new TPad("myPad22", "The pad",0,0,1,1);
114	myPadSetUp(myPad22,0.15,0.15,0.15,0.15);
115	myPad22->Draw();
116	myPad22->cd();
117	fHistDiLeptonPtJPsi->Draw("COLZ");
118	c22->SaveAs(Form("fig_ud_DiLeptonPtJPsi.%s",suffix.Data()));
119
120
121	TCanvas *c31 = new TCanvas("fig_ud_DiElectronMassJPsi"," ",800,500);
122	c31->Draw();
123	c31->cd();
124	TPad *myPad31 = new TPad("myPad31", "The pad",0,0,1,1);
125	myPadSetUp(myPad31,0.15,0.1,0.04,0.15);
126	myPad31->Draw();
127	myPad31->cd();
128	fHistDiElectronMass->Draw();
129	c31->SaveAs(Form("fig_ud_DiElectronMassJPsi.%s",suffix.Data()));
130
131
132	TCanvas *c32 = new TCanvas("fig_ud_DiMuonMassJPsi"," ",800,500);
133	c32->Draw();
134	c32->cd();
135	TPad *myPad32 = new TPad("myPad32", "The pad",0,0,1,1);
136	myPadSetUp(myPad32,0.15,0.1,0.04,0.15);
137	myPad32->Draw();
138	myPad32->cd();
139	fHistDiMuonMass->Draw();
140	c32->SaveAs(Form("fig_ud_DiMuonMassJPsi.%s",suffix.Data()));
141
142
143	/*
144	TFile* outfile = new TFile("fig_ud.root","recreate");
145	c1->Write();
146	c21->Write();
147	c31->Write();
148	c22->Write();
149	c32->Write();
150	*/
151
152	}
153
154	void myPadSetUp(TPad *currentPad, float currentLeft=0.11, float currentTop=0.04, float currentRight=0.04, float currentBottom=0.15){
155	currentPad->SetLeftMargin(currentLeft);
156	currentPad->SetTopMargin(currentTop);
157	currentPad->SetRightMargin(currentRight);
158	currentPad->SetBottomMargin(currentBottom);
159	return;
160	}
161
162
163	void myHistSetUp(TH1 *currentGraph=0){
164
165	currentGraph->SetLabelSize(0.05,"xyz");
166	currentGraph->SetLabelFont(42,"xyz");
167	currentGraph->SetLabelOffset(0.01,"xyz");
168	currentGraph->SetTitleFont(42,"xyz");
169	currentGraph->GetXaxis()->SetTitleOffset(1.1);
170	currentGraph->GetYaxis()->SetTitleOffset(1.1);
171	currentGraph->SetTitleSize(0.06,"xyz");
172	currentGraph->SetStats(kFALSE);
173	currentGraph->SetTitle("");
174	return;
175	}
176	void myHistSetUp(TH2 *currentGraph=0){
177
178	currentGraph->SetLabelSize(0.05,"xyz");
179	currentGraph->SetLabelFont(42,"xyz");
180	currentGraph->SetLabelOffset(0.01,"xyz");
181	currentGraph->SetTitleFont(42,"xyz");
182	currentGraph->GetXaxis()->SetTitleOffset(1.3);
183	currentGraph->GetYaxis()->SetTitleOffset(1.3);
184	currentGraph->SetTitleSize(0.05,"xyz");
185	currentGraph->SetStats(kFALSE);
186	currentGraph->SetTitle("");
187	return;
188	}
189
190	void myOptions(Int_t lStat=0){
191	// Set gStyle
192	int font = 42;
193	// From plain
194	gStyle->SetFrameBorderMode(0);
195	gStyle->SetFrameFillColor(0);
196	gStyle->SetCanvasBorderMode(0);
197	gStyle->SetPadBorderMode(0);
198	gStyle->SetPadColor(10);
199	gStyle->SetCanvasColor(10);
200	gStyle->SetTitleFillColor(10);
201	gStyle->SetTitleBorderSize(1);
202	gStyle->SetStatColor(10);
203	gStyle->SetStatBorderSize(1);
204	gStyle->SetLegendBorderSize(1);
205	//
206	gStyle->SetDrawBorder(0);
207	gStyle->SetTextFont(font);
208	gStyle->SetStatFont(font);
209	gStyle->SetStatFontSize(0.05);
210	gStyle->SetStatX(0.97);
211	gStyle->SetStatY(0.98);
212	gStyle->SetStatH(0.03);
213	gStyle->SetStatW(0.3);
214	gStyle->SetTickLength(0.02,"y");
215	gStyle->SetEndErrorSize(3);
216	gStyle->SetLabelSize(0.05,"xyz");
217	gStyle->SetLabelFont(font,"xyz");
218	gStyle->SetLabelOffset(0.01,"xyz");
219	gStyle->SetTitleFont(font,"xyz");
220	gStyle->SetTitleOffset(1.0,"xyz");
221	gStyle->SetTitleSize(0.06,"xyz");
222	gStyle->SetMarkerSize(1);
223	gStyle->SetPalette(1,0);
224	if (lStat){
225	gStyle->SetOptTitle(1);
226	gStyle->SetOptStat(1111);
227	gStyle->SetOptFit(1111);
228	}
229	else {
230	gStyle->SetOptTitle(0);
231	gStyle->SetOptStat(0);
232	gStyle->SetOptFit(0);
233	}
234	}
235