from os import environ as env
from plotter.objects.plotter import Plotter
from plotter.objects.selections import Selections
from plotter.objects.utils import save_plotter_and_selections
ch = 'mem'
cuts = Selections(ch)
selection = [
cuts.selections['pt_iso'],
cuts.selections['pre_baseline'],
cuts.selections['vetoes_02_OS'],
cuts.selections['fsr_veto'],
'l0_q!=l2_q',
# cuts.selections['sideband'],
# cuts.selections['signal_region'],
'hlt_IsoMu24',
'hnl_2d_disp_sig>5',
'sv_prob>0.0002',
'abs(l1_dxy) > 0.001',
'abs(l2_dxy) > 0.001',
'sv_cos>0.9',
'abs(l1_dz)<10',
'abs(l2_dz)<10',
'hnl_pt_12>10',
'((hnl_w_vis_m<50 | hnl_w_vis_m>80) | nbj>0)',
]
# extra selection to be applied on variables that don't exist
# in the root tree but they're created for the pandas dataset
baseline = 'pre_baseline'
extra_selections = [
'hnl_2d_disp_sig>5',
'sv_prob>0.0002',
'abs(l1_dxy) > 0.001',
'abs(l2_dxy) > 0.001',
'sv_cos>0.9',
'abs(l1_dz)<10',
'abs(l2_dz)<10',
'hnl_pt_12>10',
# 'hnl_2d_disp_sig_alt>20',
]
cuts_mmm = Selections('mmm')
selection_mmm = [
cuts_mmm.selections['pt_iso'],
cuts_mmm.selections[baseline],
cuts_mmm.selections['vetoes_12_OS'],
cuts_mmm.selections['vetoes_01_OS'],
cuts_mmm.selections['vetoes_02_OS'],
cuts_mmm.selections['sideband'],
] + extra_selections
cuts_mem = Selections('mem')
selection_mem = [
cuts_mem.selections['pt_iso'],
cuts_mem.selections[baseline],
cuts_mem.selections['sideband'],
cuts_mem.selections['vetoes_02_OS'],
