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':
hist.Hist('Counts', dataset_axis, iso_axis, bin_axis),
'chan-2mu2e':
parser = argparse.ArgumentParser(
description="leptonjet isolation profile of pt, eta")
parser.add_argument(
"--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