def __plot(self, successful):
"""summary plots
"""
load_root_style()
for name in [
"histoSigmaCorr", "hDirectEffpt", "hFeedDownEffpt", "hRECpt"
]:
for ibin2 in range(self.nominal_analyzer_merged.p_nbin2):
self.__make_single_plot(name, ibin2, successful)
for ibin2 in range(self.nominal_analyzer_merged.p_nbin2):
self.__make_summary_plot("histoSigmaCorr", ibin2, successful)
def plot_hfptspectrum_ratios_comb(case_num, case_den, arraytype):
load_root_style()
with open("../data/database_ml_parameters_%s.yml" % case_num,
'r') as param_config_num:
data_param_num = yaml.load(param_config_num, Loader=yaml.FullLoader)
with open("../data/database_ml_parameters_%s.yml" % case_den,
'r') as param_config_den:
data_param_den = yaml.load(param_config_den, Loader=yaml.FullLoader)
folder_plots_num = data_param_num[case_num]["analysis"][
"dir_general_plots"]
folder_plots_den = data_param_den[case_den]["analysis"][
"dir_general_plots"]
if not os.path.exists(folder_plots_num):
print("creating folder ", folder_plots_num)
os.makedirs(folder_plots_num)
if not os.path.exists(folder_plots_den):
print("creating folder ", folder_plots_den)
os.makedirs(folder_plots_den)
folder_num_allperiods = \
data_param_num[case_num]["analysis"][arraytype[0]]["data"]["resultsallp"]
folder_den_allperiods = \
data_param_den[case_den]["analysis"][arraytype[0]]["data"]["resultsallp"]
folder_num_triggered = \
data_param_num[case_num]["analysis"][arraytype[1]]["data"]["resultsallp"]
folder_den_triggered = \
data_param_den[case_den]["analysis"][arraytype[1]]["data"]["resultsallp"]
binsmin_num = data_param_num[case_num]["analysis"][
arraytype[0]]["sel_binmin2"]
binsmax_num = data_param_num[case_num]["analysis"][
arraytype[0]]["sel_binmax2"]
name_num = data_param_num[case_num]["analysis"][
arraytype[0]]["latexnamemeson"]
name_den = data_param_den[case_den]["analysis"][
arraytype[0]]["latexnamemeson"]
latexbin2var = data_param_num[case_num]["analysis"][
arraytype[0]]["latexbin2var"]
plotbinMB = data_param_num[case_num]["analysis"][arraytype[0]]["plotbin"]
plotbinHM = data_param_num[case_num]["analysis"][arraytype[1]]["plotbin"]
br_num = data_param_num[case_num]["ml"]["opt"]["BR"]
br_den = data_param_den[case_den]["ml"]["opt"]["BR"]
sigmav0_num = data_param_num[case_num]["analysis"]["sigmav0"]
sigmav0_den = data_param_den[case_den]["analysis"]["sigmav0"]
file_num_allperiods = TFile.Open("%s/finalcross%s%smulttot.root" % \
(folder_num_allperiods, case_num, arraytype[0]))
file_den_allperiods = TFile.Open("%s/finalcross%s%smulttot.root" % \
(folder_den_allperiods, case_den, arraytype[0]))
file_num_triggered = TFile.Open("%s/finalcross%s%smulttot.root" % \
(folder_num_triggered, case_num, arraytype[1]))
file_den_triggered = TFile.Open("%s/finalcross%s%smulttot.root" % \
(folder_den_triggered, case_den, arraytype[1]))
if not file_num_allperiods or not file_num_triggered:
print("---Warning: Issue with %s merged files. Meson ratio plot skipped (%s, %s)---" % \
(case_num, arraytype[0], arraytype[1]))
return
if not file_den_allperiods or not file_den_triggered:
print("---Warning: Issue with %s merged files. Meson ratio plot skipped (%s, %s)---" % \
(case_den, arraytype[0], arraytype[1]))
return
rootfilename = "%s/ComparisonRatios_%s%s_%scombined%s.root" % \
(folder_plots_num, case_num, case_den, arraytype[0], arraytype[1])
fileoutput = TFile.Open(rootfilename, "recreate")
ccross = TCanvas('cRatioCross', 'The Fit Canvas')
ccross.SetCanvasSize(1500, 1500)
ccross.SetWindowSize(500, 500)
maxplot = 1.0
if case_num == "Dspp":
maxplot = 0.5
ccross.cd(1).DrawFrame(0.9, 0, 30, maxplot, ";#it{p}_{T} (GeV/#it{c});%s / %s" % \
(name_num, name_den))
ccross.cd(1).SetLogx()
legyield = TLegend(.4, .68, .8, .88)
legyield.SetBorderSize(0)
legyield.SetFillColor(0)
legyield.SetFillStyle(0)
legyield.SetTextFont(42)
legyield.SetTextSize(0.025)
colors = [
kBlack, kRed, kGreen + 2, kBlue, kViolet - 1, kOrange + 2, kAzure + 1,
kOrange - 7
]
for imult, iplot in enumerate(plotbinMB):
if not iplot:
continue
hratio = file_num_allperiods.Get("histoSigmaCorr%d" % (imult))
hratio.Scale(1. / (br_num * sigmav0_num * 1e12))
hcross_den = file_den_allperiods.Get("histoSigmaCorr%d" % (imult))
hcross_den.Scale(1. / (br_den * sigmav0_den * 1e12))
hratio.Divide(hcross_den)
hratio.SetLineColor(colors[imult % len(colors)])
hratio.SetMarkerColor(colors[imult % len(colors)])
hratio.SetMarkerStyle(21)
hratio.SetTitle(";#it{p}_{T} (GeV/#it{c});%s / %s" %
(name_num, name_den))
hratio.Draw("same")
legyieldstring = "%.1f #leq %s < %.1f (MB)" % \
(binsmin_num[imult], latexbin2var, binsmax_num[imult])
legyield.AddEntry(hratio, legyieldstring, "LEP")
fileoutput.cd()
hratio.Write("hratio_fromMB_%.1f_%s_%.1f" % \
(binsmin_num[imult], latexbin2var, binsmax_num[imult]))
for imult, iplot in enumerate(plotbinHM):
if not iplot:
continue
hratioHM = file_num_triggered.Get("histoSigmaCorr%d" % (imult))
hratioHM.Scale(1. / (br_num * sigmav0_num * 1e12))
hcrossHM_den = file_den_triggered.Get("histoSigmaCorr%d" % (imult))
hcrossHM_den.Scale(1. / (br_den * sigmav0_den * 1e12))
hratioHM.Divide(hcrossHM_den)
hratioHM.SetLineColor(colors[imult % len(colors)])
hratioHM.SetMarkerColor(colors[imult % len(colors)])
hratioHM.SetTitle(";#it{p}_{T} (GeV/#it{c});%s / %s" %
(name_num, name_den))
hratioHM.Draw("same")
legyieldstring = "%.1f #leq %s < %.1f (HM)" % \
(binsmin_num[imult], latexbin2var, binsmax_num[imult])
legyield.AddEntry(hratioHM, legyieldstring, "LEP")
fileoutput.cd()
hratioHM.Write("hratio_fromHM_%.1f_%s_%.1f" % \
(binsmin_num[imult], latexbin2var, binsmax_num[imult]))
legyield.Draw()
ccross.SaveAs("%s/PtSpec_ComparisonRatios_%s%s_%scombined%s_logx.eps" % \
(folder_plots_num, case_num, case_den, arraytype[0], arraytype[1]))
ccross.SaveAs("%s/PtSpec_ComparisonRatios_%s%s_%scombined%s_logx.eps" % \
(folder_plots_den, case_num, case_den, arraytype[0], arraytype[1]))
ccross.cd(1).SetLogx(0)
ccross.SaveAs("%s/PtSpec_ComparisonRatios_%s%s_%scombined%s.eps" % \
(folder_plots_num, case_num, case_den, arraytype[0], arraytype[1]))
ccross.SaveAs("%s/PtSpec_ComparisonRatios_%s%s_%scombined%s.eps" % \
(folder_plots_den, case_num, case_den, arraytype[0], arraytype[1]))
fileoutput.cd()
ccross.Write()
fileoutput.Close()
rootfilenameden = "%s/ComparisonRatios_%s%s_%scombined%s.root" % \
(folder_plots_den, case_num, case_den, arraytype[0], arraytype[1])
copyfile(rootfilename, rootfilenameden)
print("---Output stored in:", rootfilename, "and", rootfilenameden, "---")
save_path = f"{folder_plots}/ratio_{case_ml_1}_{case_ml_2}_{hn_ml}_ml_std_mult_" \
f"{mult_bin}_period_{period_number}{suffix}.eps"
plot_histograms([histo_ratio_std, histo_ratio_ml], True, True,
["STD", "ML"], "Ratio", "#it{p}_{T} (GeV/#it{c}",
f"{name_1} / {name_2}", "ML / STD", save_path)
# pylint: disable=import-error, no-name-in-module, unused-import
# pylint: disable=too-many-statements
# pylint: disable=too-many-branches
# pylint: disable=too-many-locals
def plot_hfptspectrum_comb(case, arraytype):
load_root_style()
with open("../data/database_ml_parameters_%s.yml" % case,
'r') as param_config:
data_param = yaml.load(param_config, Loader=yaml.FullLoader)
folder_plots = data_param[case]["analysis"]["dir_general_plots"]
if not os.path.exists(folder_plots):
print("creating folder ", folder_plots)
os.makedirs(folder_plots)
folder_MB_allperiods = data_param[case]["analysis"][
arraytype[0]]["data"]["resultsallp"]
folder_triggered = data_param[case]["analysis"][
arraytype[1]]["data"]["resultsallp"]