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'],