def makePlots(variables, cuts, total_weight, sample_dict, hist_dict, qcd_from_same_sign, w_qcd_mssm_method, mt_cut, friend_func, dc_postfix, make_plots=True, create_trees=False):
ams_dict = {}
sample_names = set()
for cut in cuts:
cfg_main = HistogramCfg(name=cut.name, var=None, cfgs=sample_dict['all_samples'], cut=cut.cut, lumi=int_lumi, weight=total_weight)
cfg_main.vars = variables
HISTS = CreateHists(cfg_main)
plots = HISTS.createHistograms(cfg_main, verbose=False, friend_func=friend_func)
#plots.legendPos = 'right'
for variable in variables:
# for plot in plots.itervalues():
plot = plots[variable.name]
# plot.Group('data_obs', ['data_2017B_e', 'data_2017C_e', 'data_2017D_e', 'data_2017E_e', 'data_2017F_e'])
# plot.Group('single t', ['ST_tW_at_5f_incD_e', 'ST_tW_t_5f_incD_e'])
# plot.Group('Diboson' , ['WZTo3LNu_e', 'ZZTo4L_e', 'WWTo2L2Nu_e'])
# plot.Group('Triboson', ['ZZZ_e', 'WWW_e', 'WGGJets_e'])
# plot.Group('ttV' , ['TTZToLLNuNu_e', 'TTWJetsToLNu_e'])
# plot.Group('DY' , ['DYJets_M5T50_e', 'DYJets_M50_x_e', 'DYJets_M50_e'])
createDefaultGroups(plot)
if make_plots:
HistDrawer.draw(plot, plot_dir = '/eos/user/m/manzoni/HNL/plots/'+cut.name)#plot_dir='plots/'+cut.name)
print '\nOptimisation results:'
all_vals = ams_dict.items()
for sample_name in sample_names:
vals = [v for v in all_vals if sample_name + '_' in v[0]]
vals.sort(key=itemgetter(1))
for key, item in vals:
print item, key
print '\nBy variable'
for variable in variables:
name = variable.name
print '\nResults for variable', name
for key, item in vals:
if key.startswith(name + '__'):
print item, key
def makePlots(variables,
cuts,
total_weight,
sample_dict,
hist_dict,
qcd_from_same_sign,
w_qcd_mssm_method,
mt_cut,
friend_func,
dc_postfix,
make_plots=True,
create_trees=False):
ams_dict = {}
sample_names = set()
for cut in cuts:
cfg_main = HistogramCfg(name=cut.name,
var=None,
cfgs=sample_dict['all_samples'],
cut=cut.cut,
lumi=int_lumi,
weight=total_weight)
# cfg_main = HistogramCfg(name=cut.name, var=None, cfgs=sample_dict['samples_essential'], cut=cut.cut, lumi=int_lumi, weight=total_weight)
cfg_main.vars = variables
HISTS = CreateHists(cfg_main)
plots = HISTS.createHistograms(cfg_main,
verbose=False,
friend_func=friend_func)
#plots.legendPos = 'right'
for variable in variables:
# for plot in plots.itervalues():
plot = plots[variable.name]
plot.Group('data_obs', [
'data_2017B', 'data_2017C', 'data_2017D', 'data_2017E',
'data_2017F'
])
plot.Group('single t', ['ST_tW_at_5f_incD', 'ST_tW_t_5f_incD'])
# plot.Group('Diboson', ['WZTo3LNu', 'ZZTo4L', 'WWTo2L2Nu'])
plot.Group('Diboson', ['WZTo3LNu', 'WWTo2L2Nu'])
plot.Group('Triboson', ['ZZZ', 'WWW', 'WGGJets,WWZ,WZZ'])
plot.Group('ttV', ['TTZToLLNuNu', 'TTWJetsToLNu'])
plot.Group('QCD', [
'QCD_pt_15to20_mu', 'QCD_pt_20to30_mu', 'QCD_pt_30to50_mu',
'QCD_pt_50to80_mu', 'QCD_pt_80to120_mu'
])
plot.Group(
'WJets',
['W1JetsToLNu', 'W2JetsToLNu', 'W3JetsToLNu', 'W4JetsToLNu'])
# plot.Group('DY', ['DYJetsToLL_M10to50', 'DYJets_ext'])
plot.Group('DYJetsToLL_M10to50',
['DYJetsToLL_M10to50', 'DYJetsToLL_M10to50_ext'])
createDefaultGroups(plot)
if make_plots:
HistDrawer.draw(plot,
channel=channel_name,
plot_dir=plotDir +
cut.name) #plot_dir='plots/'+cut.name)
print '\nOptimisation results:'
all_vals = ams_dict.items()
for sample_name in sample_names:
vals = [v for v in all_vals if sample_name + '_' in v[0]]
vals.sort(key=itemgetter(1))
for key, item in vals:
print item, key
print '\nBy variable'
for variable in variables:
name = variable.name
print '\nResults for variable', name
for key, item in vals:
if key.startswith(name + '__'):
print item, key