def gen_card(self, file_name, **kwargs):
doDataDriven = kwargs.pop("doDataDriven", True)
# get the plotter
nl = 3
sigMap = getSigMap(nl)
intLumiMap = getIntLumiMap()
mergeDict = getMergeDict(self.period)
channelBackground = getChannelBackgrounds(self.period)
channels, leptons = getChannels(nl)
saves = "%s_%s_%iTeV" % (self.analysis, self.region, self.period)
plotter = Plotter(
self.region,
ntupleDir=self.ntuple_dir,
saveDir=saves,
period=self.period,
rootName="plots_limits_wz",
mergeDict=mergeDict,
scaleFactor=self.scalefactor,
datadriven=True,
)
plotter.initializeBackgroundSamples(
[sigMap[self.period][x] for x in channelBackground[self.region + "datadriven"]]
)
plotter.initializeDataSamples([sigMap[self.period]["data"]])
plotter.setIntLumi(intLumiMap[self.period])
# set expected and observed yields
sources = (
[
x
for x in channelBackground[self.region + "datadriven"]
if x not in ["TT", "T", "DY", "Z", "Zfiltered", "WW"]
]
+ ["datadriven"]
if doDataDriven
else channelBackground[self.region]
)
for bg in sources:
self.log.info("Processing {0}".format(bg))
val, err = plotter.getNumEntries(self.selection, sigMap[self.period][bg], doError=True)
statname = "stat_{0}_{1}".format(bg, file_name.split(".")[0])
staterrs = {bg: err / val + 1.0 if val else 1.0}
if self.doStat:
self.add_systematics(statname, "lnN", **staterrs)
if bg in ["WZ"]:
self.datacard.add_sig(bg, val)
else:
self.datacard.add_bkg(bg, val)
self.log.info("Processing observed")
self.datacard.set_observed(plotter.getDataEntries(self.selection))
# write out the datacard
self.log.info("Saving card to file")
with open("%s/%s" % (self.out_dir, file_name), "w") as outfile:
outfile.write(self.datacard.dump())
def getPlotter(self,analysis,region,runPeriod,mass,runTau,plotName,doFakes):
nl = 3 if analysis=='Hpp3l' or analysis=='WZ' else 4
ntuples = getNtupleDirectory(analysis,region,runPeriod)
saves = '%s_%s_%iTeV' % (analysis,region,runPeriod)
sigMap = getSigMap(nl,mass)
intLumiMap = getIntLumiMap()
mergeDict = getMergeDict(runPeriod)
regionBackground = getChannelBackgrounds(runPeriod)
channels, leptons = getChannels(nl,runTau=runTau)
plotter = Plotter(region,ntupleDir=ntuples,saveDir=saves,period=runPeriod,rootName=plotName,mergeDict=mergeDict,scaleFactor=self.scalefactor)
if not doFakes: plotter.initializeBackgroundSamples([sigMap[runPeriod][x] for x in regionBackground[analysis]])
if runPeriod==8: plotter.initializeDataSamples([sigMap[runPeriod]['data']])
plotter.setIntLumi(intLumiMap[runPeriod])
return plotter
