1 const Int_t numberOfCentralityBins = 8;
2 TString centralityArray[numberOfCentralityBins] = {"0-10","10-20","20-30","30-40","40-50","50-60","60-70","70-80"};
4 void drawBalanceFunctionPsiSummarySummary(const char* lhcPeriod = "LHC11h",
6 Double_t psiMin = -0.5,
8 TString eventClass = "Centrality"){
10 TGraphErrors *gPar[3][18];
12 Int_t iCentrality[3] = {1,2,3};
13 Double_t fCentralityMin[3] = {0.,20.,40.};
14 Double_t fCentralityMax[3] = {10.,30.,50.};
17 for(Int_t iCent = 0 ; iCent < 3; iCent++){
20 if(eventClass == "Centrality"){
21 fPar[iCent] = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%.1fTo%.1f_FitParameters.root",lhcPeriod,gTrainID,fCentralityMin[iCent],fCentralityMax[iCent]),"READ");
23 else if(eventClass == "Multiplicity"){
24 fPar[iCent] = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Mult%.1fTo%.1f_FitParameters.root",lhcPeriod,gTrainID,fCentralityMin[iCent],fCentralityMax[iCent]),"RED");
27 fPar[iCent] = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%d_FitParameters.root",lhcPeriod,gTrainID,iCentrality[iCent]),"READ");
31 cerr<<"FILE "<<Form("PbPb/%s/Train%d/figs/correlationFunctionFit_Cent%d_FitParameters.root",lhcPeriod,gTrainID,iCentrality[iCent])<<" not found!"<<endl;
36 for(Int_t iPar = 0 ; iPar < 18; iPar++){
37 gPar[iCent][iPar] = (TGraphErrors*)fPar[iCent]->Get(Form("gPar%d",iPar));
38 if(!gPar[iCent][iPar]){
39 cerr<<"Graph for parameter "<<iPar<<" in centrality "<< iCent << " not found!"<<endl;
47 TCanvas *cSummary[18];
48 for(Int_t iPar = 0 ; iPar < 18; iPar++){
50 cSummary[iPar] = new TCanvas(Form("cSummary%d",iPar),Form("Summary %d",iPar));
52 // compare centralities
54 gPar[0][iPar]->SetMarkerColor(1);
55 gPar[0][iPar]->SetLineColor(1);
56 gPar[0][iPar]->Draw("AP");
57 gPar[1][iPar]->SetMarkerColor(2);
58 gPar[1][iPar]->SetLineColor(2);
59 gPar[1][iPar]->Draw("P");
60 gPar[2][iPar]->SetMarkerColor(4);
61 gPar[2][iPar]->SetLineColor(4);
62 gPar[2][iPar]->Draw("P");
65 TLegend *legend2 = new TLegend(0.2,0.6,0.85,0.88,"","brNDC");
66 setupLegend(legend2,0.065);
67 for(Int_t iCent = 0 ; iCent < 3; iCent++){
68 legend2->AddEntry(gPar[iCent][iPar],Form("%s \%",centralityArray[iCentrality[iCent]-1].Data()),"lp");
72 cSummary[iPar]->SaveAs(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_FitParameter%d_Summary.eps",lhcPeriod,gTrainID,iPar));
79 void drawBalanceFunctionPsiSummary(const char* lhcPeriod = "LHC11h",
81 Int_t gCentrality = 9,
84 TString eventClass = "Centrality") {
85 // Macro that draws the fit results for the
86 // correlation functions from the balance function analysis
87 // Author: m.weber@cern.ch
89 gROOT->LoadMacro("~/SetPlotStyle.C");
91 gStyle->SetPalette(1,0);
93 //Load the PWG2ebye library
94 gSystem->Load("libANALYSIS.so");
95 gSystem->Load("libANALYSISalice.so");
96 gSystem->Load("libEventMixing.so");
97 gSystem->Load("libCORRFW.so");
98 gSystem->Load("libPWGTools.so");
99 gSystem->Load("libPWGCFebye.so");
102 // this could also be retrieved directly from AliBalancePsi
103 //const Int_t kNPtBins = 16;
104 //Double_t ptBins[kNPtBins+1] = {0.2,0.6,1.0,1.5,2.0,2.5,3.0,3.5,4.0,5.0,6.0,7.0,8.0,10.,12.,15.,20.};
105 const Int_t kNPtBins = 5;
106 Double_t ptBins[kNPtBins+1] = {0.2,1.0,2.0,3.0,4.0,8.0};
107 //const Int_t kNPtBins = 4;
108 //Double_t ptBins[kNPtBins+1] = {1.0,2.0,3.0,4.0,8.0};
109 //const Int_t kNPtBins = 1;
110 //Double_t ptBins[kNPtBins+1] = {0.2,2.0};
112 Double_t pt[kNPtBins*kNPtBins];
113 Double_t ptE[kNPtBins*kNPtBins];
114 for(Int_t i = 0; i < kNPtBins; i++){
115 for(Int_t j = 0; j < kNPtBins; j++){
116 pt[i*kNPtBins+j] = 10*i + (ptBins[j]+ptBins[j+1])/2.;
117 ptE[i*kNPtBins+j] = 0.2;
123 TCanvas *cQA[kNPtBins][kNPtBins];
124 for(Int_t i = 0; i < kNPtBins; i++){
125 for(Int_t j = 0; j <= i; j++){
126 cQA[i][j] = new TCanvas(Form("cQA%d%d",i,j),Form("Fitting QA for bin %d %d",i,j),1200,900);
127 cQA[i][j]->Divide(3,3);
133 Double_t ptTriggerMin = 0.0;
134 Double_t ptTriggerMax = 0.;
135 Double_t ptAssociatedMin = 0.0;
136 Double_t ptAssociatedMax = 0.0;
137 TString inFileName = "";
140 Double_t p[18][kNPtBins*kNPtBins];
141 Double_t pE[18][kNPtBins*kNPtBins];
143 for(Int_t iPar = 0; iPar < 18; iPar++){
144 for(Int_t i = 0; i < kNPtBins; i++){
145 for(Int_t j = 0; j < kNPtBins; j++){
146 p[iPar][i*kNPtBins+j] = -1.;
147 pE[iPar][i*kNPtBins+j] = 0.;
152 TFile *inFile = NULL;
153 TH2D *hTMPData = NULL;
154 TH2D *hTMPRes = NULL;
155 TH2D *hTMPFit = NULL;
159 for(Int_t i = 0; i < kNPtBins; i++){
160 for(Int_t j = 0; j <= i; j++){
162 cout<<" PROCESSING PT BIN "<<i<<" "<<j<<endl;
164 ptTriggerMin = ptBins[i];
165 ptTriggerMax = ptBins[i+1];
166 ptAssociatedMin = ptBins[j];
167 ptAssociatedMax = ptBins[j+1];
171 TString centralityLatex = "Centrality: ";
172 centralityLatex += centralityArray[gCentrality-1];
173 centralityLatex += "%";
176 if((psiMin == -0.5)&&(psiMax == 0.5))
177 psiLatex = " -7.5^{o} < #varphi - #Psi_{2} < 7.5^{o}";
178 else if((psiMin == 0.5)&&(psiMax == 1.5))
179 psiLatex = " 37.5^{o} < #varphi - #Psi_{2} < 52.5^{o}";
180 else if((psiMin == 1.5)&&(psiMax == 2.5))
181 psiLatex = " 82.5^{o} < #varphi - #Psi_{2} < 97.5^{o}";
183 psiLatex = " 0^{o} < #varphi - #Psi_{2} < 180^{o}";
185 TString pttLatex = Form("%.1f",ptTriggerMin);
186 pttLatex += " < p_{T,trig} < "; pttLatex += Form("%.1f",ptTriggerMax);
187 pttLatex += " GeV/c";
189 TString ptaLatex = Form("%.1f",ptAssociatedMin);
190 ptaLatex += " < p_{T,assoc} < "; ptaLatex += Form("%.1f",ptAssociatedMax);
191 ptaLatex += " GeV/c";
193 TLatex *latexInfo1 = new TLatex();
194 latexInfo1->SetNDC();
195 latexInfo1->SetTextSize(0.045);
196 latexInfo1->SetTextColor(1);
199 inFileName = Form("PbPb/%s/Train%d/Fits/balanceFunctionFit2D",lhcPeriod,gTrainID);
200 if(eventClass == "Centrality"){
201 inFileName += Form(".Centrality%.1fTo%.1f",psiMin,psiMax);
202 inFileName += ".PsiAll.PttFrom";
204 else if(eventClass == "Multiplicity"){
205 inFileName += Form(".Multiplicity%.0fTo%.0f",psiMin,psiMax);
206 inFileName += ".PsiAll.PttFrom";
208 else{ // "EventPlane" (default)
209 inFileName += ".Centrality";
210 inFileName += gCentrality; inFileName += ".Psi";
211 if((psiMin == -0.5)&&(psiMax == 0.5)) inFileName += "InPlane.Ptt";
212 else if((psiMin == 0.5)&&(psiMax == 1.5)) inFileName += "Intermediate.Ptt";
213 else if((psiMin == 1.5)&&(psiMax == 2.5)) inFileName += "OutOfPlane.Ptt";
214 else if((psiMin == 2.5)&&(psiMax == 3.5)) inFileName += "Rest.PttFrom";
215 else inFileName += "All.PttFrom";
217 inFileName += Form("%.1f",ptTriggerMin); inFileName += "To";
218 inFileName += Form("%.1f",ptTriggerMax); inFileName += "PtaFrom";
219 inFileName += Form("%.1f",ptAssociatedMin); inFileName += "To";
220 inFileName += Form("%.1f",ptAssociatedMax);
221 inFileName += "_2pMethod.root";
222 inFile = TFile::Open(inFileName.Data(),"read");
224 hTMPData = (TH2D*)inFile->Get(Form("gHistBalanceFunctionSubtracted"));
225 hTMPRes = (TH2D*)inFile->Get("gHistResidual");
226 hTMPFit = (TH2D*)inFile->Get(Form("gHistBalanceFunctionSubtractedFit"));
229 hTMPData->DrawCopy("surf1fb");
230 latexInfo1->DrawLatex(0.2,0.95,"Data");
232 latexInfo1->DrawLatex(0.44,0.88,centralityLatex.Data());
233 //latexInfo1->DrawLatex(0.44,0.82,psiLatex.Data());
234 latexInfo1->DrawLatex(0.44,0.82,pttLatex.Data());
235 latexInfo1->DrawLatex(0.44,0.76,ptaLatex.Data());
238 hTMPFit->DrawCopy("surf1fb");
239 latexInfo1->DrawLatex(0.2,0.95,"Fit");
242 hTMPRes->DrawCopy("surf1fb");
243 latexInfo1->DrawLatex(0.2,0.95,"Residual");
246 hTMPData->ProjectionX()->DrawCopy("");
247 latexInfo1->DrawLatex(0.2,0.95,"Data");
250 hTMPFit->ProjectionX()->DrawCopy("");
251 latexInfo1->DrawLatex(0.2,0.95,"Fit");
254 hTMPRes->ProjectionX()->DrawCopy("");
255 latexInfo1->DrawLatex(0.2,0.95,"Residual");
258 hTMPData->ProjectionY()->DrawCopy("");
259 latexInfo1->DrawLatex(0.2,0.95,"Data");
262 hTMPFit->ProjectionY()->DrawCopy("");
263 latexInfo1->DrawLatex(0.2,0.95,"Fit");
266 hTMPRes->ProjectionY()->DrawCopy("");
267 latexInfo1->DrawLatex(0.2,0.95,"Residual");
270 cQA[i][j]->SaveAs(Form(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_PttFrom%.1fTo%.1fPtaFrom%.1fTo%.1f.eps",lhcPeriod,gTrainID,ptTriggerMin,ptTriggerMax,ptAssociatedMin,ptAssociatedMax)));
271 cQA[i][j]->SaveAs(Form(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_PttFrom%.1fTo%.1fPtaFrom%.1fTo%.1f.pdf",lhcPeriod,gTrainID,ptTriggerMin,ptTriggerMax,ptAssociatedMin,ptAssociatedMax)));
274 fFit = (TF2*)inFile->Get("gFitFunction");
275 for(Int_t iPar = 0; iPar < 18; iPar++){
276 p[iPar][i*kNPtBins+j] = fFit->GetParameter(iPar);
277 pE[iPar][i*kNPtBins+j] = fFit->GetParError(iPar);
278 cout<<iPar<<") Parameter "<<fFit->GetParName(iPar)<<" : "<<p[iPar][i*kNPtBins+j]<<" +- "<<pE[iPar][i*kNPtBins+j]<<endl;
285 TGraphErrors *gPar[18];
286 for(Int_t iPar = 0; iPar < 18; iPar++){
287 gPar[iPar] = new TGraphErrors(kNPtBins*kNPtBins,pt,p[iPar],ptE,pE[iPar]);
288 gPar[iPar]->SetName(Form("gPar%d",iPar));
289 gPar[iPar]->SetTitle(fFit->GetParName(iPar));
290 gPar[iPar]->GetXaxis()->SetTitle("p_{T}");
291 gPar[iPar]->GetYaxis()->SetTitle(fFit->GetParName(iPar));
292 gPar[iPar]->SetMinimum(0.01);
293 gPar[iPar]->SetMaximum(2);
294 gPar[iPar]->SetMarkerStyle(20);
295 gPar[iPar]->SetMarkerColor(2);
296 gPar[iPar]->SetLineColor(2);
299 TLatex *latexInfo2 = new TLatex();
300 latexInfo2->SetNDC();
301 latexInfo2->SetTextSize(0.045);
302 latexInfo2->SetTextColor(1);
304 TCanvas *cPar = new TCanvas("cPar","parameters",1200,900);
323 gPar[17]->Draw("AP");
326 gPar[12]->Draw("AP");
329 gPar[14]->Draw("AP");
332 gPar[15]->Draw("AP");
334 cPar->SaveAs(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%d_FitParameters.eps",lhcPeriod,gTrainID,gCentrality));
335 cPar->SaveAs(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%d_FitParameters.pdf",lhcPeriod,gTrainID,gCentrality));
338 if(eventClass == "Centrality"){
339 fOut = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%.1fTo%.1f_FitParameters.root",lhcPeriod,gTrainID,psiMin,psiMax),"RECREATE");
341 else if(eventClass == "Multiplicity"){
342 fOut = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Mult%.1fTo%.1f_FitParameters.root",lhcPeriod,gTrainID,psiMin,psiMax),"RECREATE");
345 fOut = TFile::Open(Form("PbPb/%s/Train%d/figs/balanceFunctionFit_Cent%d_FitParameters.root",lhcPeriod,gTrainID,gCentrality),"RECREATE")
348 for(Int_t iPar = 0; iPar < 18; iPar++){
353 for(Int_t i = 0; i < kNPtBins; i++){
354 for(Int_t j = 0; j <= i; j++){