def test_ROOT():
a = ROOT.TH1F("a", "a", 10, 0, 1)
assert_true(isinstance(a, _Hist))
b = ROOT.Hist(10, 0, 1, type='F')
assert_equal(a.TYPE, b.TYPE)
def __init__(self,
files=None,
outname=None,
type='MC',
dtag='',
maxevents=-1,
channel=['4mu', '2mu2e'],
ctau=None,
chargedlj=False):
if type.upper() not in ['MC', 'DATA']:
raise ValueError("Argument `type` need to be MC/DATA")
self.OutName = outname
self.Type = type.upper()
self.ChargedLJ = chargedlj
self.MaxEvents = maxevents
self.Channel = channel
self.Dtag = dtag
self.Ctau = ctau
__signal_sample_param = dict(
[substr.split('-') for substr in self.Dtag.split('_')])
self.SignalParam = {
k.upper(): float(v.replace('p', '.'))
for k, v in __signal_sample_param.items()
}
if not files: raise ValueError("Argument `files` need to be non-empty")
if isinstance(files, str): files = [
files,
]
self.Chain = TreeChain('ffNtuplizer/ffNtuple', files)
### register collections ###
# self.Chain.define_collection('pvs', prefix='pv_', size='pv_n')
self.Chain.define_collection('electrons',
prefix='electron_',
size='electron_n')
self.Chain.define_collection('muons', prefix='muon_', size='muon_n')
self.Chain.define_collection('dsamuons',
prefix='dsamuon_',
size='dsamuon_n')
self.Chain.define_collection('photons',
prefix='photon_',
size='photon_n')
self.Chain.define_collection('ak4jets',
prefix='akjet_ak4PFJetsCHS_',
size='akjet_ak4PFJetsCHS_n')
self.Chain.define_collection('hftagscores',
prefix='hftagscore_',
size='hftagscore_n')
self.Chain.define_collection('leptonjets',
prefix='pfjet_',
size='pfjet_n',
mix=LeptonJetMix)
self.Chain.define_collection('ljsources',
prefix='ljsource_',
size='ljsource_n')
self.Chain.define_collection('cosmicmuons',
prefix='cosmicmuon_',
size='cosmicmuon_n')
self.Chain.define_collection('trigobjs',
prefix='trigobj_',
size='trigobj_n')
# self.Chain.define_collection('',)
self.Chain.define_object('hlt', prefix='HLT_')
self.Chain.define_object('metfilters', prefix='metfilters_')
self.Chain.define_object('cosmicveto', prefix='cosmicveto_')
self.Triggers = [
"DoubleL2Mu23NoVtx_2Cha",
"DoubleL2Mu23NoVtx_2Cha_NoL2Matched",
"DoubleL2Mu23NoVtx_2Cha_CosmicSeed",
"DoubleL2Mu23NoVtx_2Cha_CosmicSeed_NoL2Matched",
"DoubleL2Mu25NoVtx_2Cha_Eta2p4",
"DoubleL2Mu25NoVtx_2Cha_CosmicSeed_Eta2p4",
]
#self.addTRG = [
# "DoubleL2Mu23NoVtx_2Cha",]
self.Histos = {}
for chan in channel:
self.Histos['{}/cutflow'.format(chan)] = ROOT.Hist(
20, 0, 20, title='cutflow', drawstyle='hist')
self.KeepCutFlow = False
self.RawCutFlow = False
self.LookupWeight = root_open(
os.path.join(
os.getenv('CMSSW_BASE'),
'src/FireROOT/Analysis/data/PUWeights_2018.root')).Get(
'puWeights')
self.LookupMuonSFLowpT = root_open(
os.path.join(os.getenv('CMSSW_BASE'),
'src/FireROOT/Analysis/data/mu_Loose_pt7.root')).Get(
'ratio_syst')
self.LookupMuonSF = root_open(
os.path.join(os.getenv('CMSSW_BASE'),
'src/FireROOT/Analysis/data/RunABCD_SF_ID.root')).Get(
'NUM_LooseID_DEN_TrackerMuons_pt_abseta_syst')
self.LookupElectronSF = root_open(
os.path.join(
os.getenv('CMSSW_BASE'),
'src/FireROOT/Analysis/data/2018_ElectronLoose.root')).Get(
'EGamma_SF2D')
self.LookupPhotonSF = root_open(
os.path.join(
os.getenv('CMSSW_BASE'),
'src/FireROOT/Analysis/data/2018_PhotonsLoose.root')).Get(
'EGamma_SF2D')
self.Scale = 1.