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