Exemple #1
0
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':
            hist.Hist('Counts', dataset_axis, iso_axis, bin_axis),
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


"""Leptonjet isolation, pt, """
Exemple #3
0
    description="find isolation cut for EGM-type leptonjet")
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('2mu2e')

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


def root_filling(output, datatype):
Exemple #4
0
from FireHydrant.Tools.metfilter import MetFilters
from FireHydrant.Tools.trigger import Triggers

np.seterr(divide='ignore', invalid='ignore', over='ignore')
plt.rcParams["savefig.dpi"] = 120
plt.rcParams["savefig.bbox"] = "tight"

parser = argparse.ArgumentParser(description="[AN] leptonjet pair delta phi")
parser.add_argument(
    "--sync",
    action='store_true',
    help="issue rsync command to sync plots folder to lxplus web server")
args = parser.parse_args()

sdml = SigDatasetMapLoader()
sigDS, sigSCALE = sdml.fetch('all')


class MuLJVtxProcessor(processor.ProcessorABC):
    def __init__(self):
        dataset_axis = hist.Cat('dataset', 'dataset')
        reco_axis = hist.Cat('reco', 'reco type')
        vxy_axis = hist.Bin('vxy', 'vxy [cm]', [0, 5, 70, 700])
        lxy_axis = hist.Bin('lxy', 'lxy [cm]', 100, 0, 700)
        bool_axis = hist.Bin('boolean', 'true/false', 2, 0, 2)
        reso_axis = hist.Bin('reso', '(vxy(reco)-vxy(gen)) [cm]', 100, -25, 25)
        error_axis = hist.Bin('error', '$\sigma_{lxy}$', 100, 0, 100)
        cos_axis = hist.Bin('cos', r'$|cos(\theta)|$', 50, -1, 1)
        sig_axis = hist.Bin('sig', 'lxy/$\sigma_{lxy}$', 50, 0, 50)
        self._accumulator = processor.dict_accumulator({
            'vertexgood':
plt.rcParams["savefig.bbox"] = "tight"

parser = argparse.ArgumentParser(
    description="produce 2D leptonjet isolation map")
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()
bkgDS, bkgMAP, bkgSCALE = dml.fetch('bkg')
dataDS, dataMAP = dml.fetch('data')

sdml = SigDatasetMapLoader()
sigDS, sigSCALE = sdml.fetch('simple')
"""Leptonjet Isolation."""


class LeptonJetIsoProcessor(processor.ProcessorABC):
    def __init__(self, dphi_control=False, data_type='bkg'):
        self.dphi_control = dphi_control
        self.data_type = data_type
        dataset_axis = hist.Cat('dataset', 'dataset')
        lj0iso_axis = hist.Bin('lj0iso', 'iso value', 20, 0, 1)
        lj1iso_axis = hist.Bin('lj1iso', 'iso value', 20, 0, 1)
        channel_axis = hist.Cat('channel', 'channel')
        type_axis = hist.Cat('isotype', 'isotype')
        njet_axis = hist.Bin('njet', 'njet', 10, 0, 10)

        self._accumulator = processor.dict_accumulator({