for systematic_label in systematics_label:
for central_or_shift in getattr(systematics, systematic_label):
if central_or_shift not in central_or_shifts:
central_or_shifts.append(central_or_shift)
lumi = get_lumi(era)
if era == "2016":
from tthAnalysis.HiggsToTauTau.samples.tthAnalyzeSamples_2016 import samples_2016 as samples
elif era == "2017":
from tthAnalysis.HiggsToTauTau.samples.tthAnalyzeSamples_2017 import samples_2017 as samples
elif era == "2018":
from tthAnalysis.HiggsToTauTau.samples.tthAnalyzeSamples_2018 import samples_2018 as samples
else:
raise ValueError("Invalid era: %s" % era)
for sample_name, sample_info in samples.items():
if sample_name == 'sum_events': continue
if sample_info["use_it"] == False: continue
if sample_info["type"] == "mc":
sample_info["triggers"] = [ "1e", "2e" ]
if sample_info["process_name_specific"].startswith("DYJetsToLL"):
sample_info["sample_category"] = "DY"
elif sample_info["process_name_specific"].startswith(("TTTo", "TTJets")) and sample_info["sample_category"] == "TT":
sample_info["sample_category"] = "TTbar"
elif sample_info["process_name_specific"].startswith(("WJetsToLNu", "W1JetsToLNu", "W2JetsToLNu", "W3JetsToLNu", "W4JetsToLNu")):
sample_info["sample_category"] = "WJets"
elif sample_info["process_name_specific"].startswith("ST_") and sample_info["sample_category"] == "TT":
sample_info["sample_category"] = "Singletop"
elif sample_info["process_name_specific"].startswith(("WWTo2L2Nu", "WZTo3LNu", "ZZTo4L")):
sample_info["sample_category"] = "Diboson"
elif sample_info["sample_category"] == "data_obs":
from tthAnalysis.HiggsToTauTau.samples.tthAnalyzeSamples_2016 import samples_2016
from tthAnalysis.HiggsToTauTau.samples.tthAnalyzeSamples_2016_BDT import bdt_samples as bdt_samples_common
from tthAnalysis.HiggsToTauTau.samples.stitch import samples_to_stitch_2016
from tthAnalysis.HiggsToTauTau.analysisTools import split_stitched
dy_samples_inclusive, dy_samples_binned = split_stitched(
samples_to_stitch_2016, 'DY')
bdt_samples = bdt_samples_common + dy_samples_inclusive + dy_samples_binned
for sample_name, sample_info in samples_2016.items():
if sample_name == 'sum_events': continue
sample_info["use_it"] = sample_info["process_name_specific"] in bdt_samples
if sample_info["process_name_specific"] in [
"TTJets_DiLept",
"TTJets_DiLept_ext1",
"TTJets_SingleLeptFromT",
"TTJets_SingleLeptFromT_ext1",
"TTJets_SingleLeptFromTbar",
"TTJets_SingleLeptFromTbar_ext1",
]:
sample_info["use_it"] = True
if sample_info["process_name_specific"].startswith("TTTo"):
sample_info["use_it"] = False