"--sync", action='store_true', help="issue rsync command to sync plots folder to lxplus web server") parser.add_argument("--preserve", action='store_true', help="preserve plots in ROOT file") args = parser.parse_args() import ROOT ROOT.gROOT.SetBatch() sdml = SigDatasetMapLoader() sigDS, sigSCALE = sdml.fetch() dml = DatasetMapLoader() bkgDS, bkgMAP, bkgSCALE = dml.fetch('bkg') dataDS, dataMAP = dml.fetch('data') @contextmanager def _setIgnoreLevel(level): originalLevel = ROOT.gErrorIgnoreLevel ROOT.gErrorIgnoreLevel = level yield ROOT.gErrorIgnoreLevel = originalLevel """Leptonjet isolation, pt, """ class LeptonjetIsoProcessor(processor.ProcessorABC):
from FireHydrant.Tools.metfilter import MetFilters from FireHydrant.Tools.trigger import Triggers from FireHydrant.Tools.uproothelpers import fromNestNestIndexArray np.seterr(divide='ignore', invalid='ignore', over='ignore') parser = argparse.ArgumentParser(description="leptonjet ABCD") parser.add_argument( "--sync", action='store_true', help="issue rsync command to sync plots folder to lxplus web server") args = parser.parse_args() dml = DatasetMapLoader() # dataDS, dataMAP = dml.fetch('data') bkgDS, bkgMAP, bkgSCALE = dml.fetch('bkg') sdml = SigDatasetMapLoader() sigDS_2mu2e, sigSCALE_2mu2e = sdml.fetch('2mu2e') sigDS_4mu, sigSCALE_4mu = sdml.fetch('4mu') class LjABCDProcessor(processor.ProcessorABC): def __init__(self, data_type='bkg'): self.data_type = data_type dataset_axis = hist.Cat('dataset', 'dataset') iso_axis = hist.Bin('iso', 'min pfIso', 50, 0, 0.5) bin_axis = hist.Bin('val', 'binary value', 3, 0, 3) self._accumulator = processor.dict_accumulator({ 'chan-4mu':
import awkward import coffea.processor as processor import numpy as np from coffea import hist from coffea.analysis_objects import JaggedCandidateArray from FireHydrant.Analysis.DatasetMapLoader import DatasetMapLoader from FireHydrant.Tools.correction import (get_nlo_weight_function, get_pu_weights_function, get_ttbar_weight) from FireHydrant.Tools.metfilter import MetFilters from FireHydrant.Tools.trigger import Triggers from FireHydrant.Tools.uproothelpers import fromNestNestIndexArray dml = DatasetMapLoader() # bkgDS, bkgMAP, bkgSCALE = dml.fetch('bkg') dataDS, dataMAP = dml.fetch('data') """list events""" class LeptonjetEventDrawer(processor.ProcessorABC): def __init__(self, data_type='data'): self.data_type = data_type self._accumulator = processor.dict_accumulator({ 'run_1': processor.column_accumulator(np.zeros(shape=(0, ))), 'lumi_1': processor.column_accumulator(np.zeros(shape=(0, ))), 'event_1': processor.column_accumulator(np.zeros(shape=(0, ))), 'run_2': processor.column_accumulator(np.zeros(shape=(0, ))),