args.lumi = samples.default_lumi / 1000.0
else:
log.critical("Invalid era string selected: " + args.era)
sys.exit(1)
args.output_dir = os.path.abspath(os.path.expandvars(args.output_dir))
if args.clear_output_dir and os.path.exists(args.output_dir):
logger.subprocessCall("rm -r " + args.output_dir, shell=True)
#prepare binning
nbins = args.n_bins
nbins1D = int(nbins**(1.0 / len(args.quantities)))
if nbins1D < 2:
nbins1D = 2
# initialisations for plotting
sample_settings = samples.Samples()
binnings_settings = binnings.BinningsDict()
systematics_factory = systematics.SystematicsFactory()
plot_configs = []
output_files = []
merged_output_files = []
hadd_commands = []
datacards = smhttdatacards.SMHttDatacards(higgs_masses=args.higgs_masses,
useRateParam=args.use_rateParam,
year=args.era)
if args.for_dcsync:
datacards = smhttdatacards.SMHttDatacardsForSync(
higgs_masses=args.higgs_masses,
useRateParam=args.use_rateParam,
args.samples = [
sample for sample in args.samples
if hasattr(samples.Samples, sample)
]
list_of_samples = [
getattr(samples.Samples, sample) for sample in args.samples
]
print "list_of_samples", list_of_samples
if args.emb:
if args.run1:
log.critical(
"Embedding --emb only valid for run2. Remove --emb or select run2 samples."
)
sys.exit(1)
sample_settings = samples.Samples(embedding=True)
else:
sample_settings = samples.Samples()
if args.mssm:
bkg_samples = [
sample for sample in args.samples
if sample not in ["data", "htt", "ggh", "bbh"]
]
sig_samples_raw = [
sample for sample in args.samples
if sample in ["htt", "ggh", "bbh"]
]
else:
bkg_samples = [
sample for sample in args.samples
args.use_asimov_dataset = args.use_asimov_dataset.split()
asimov_nicks = copy.deepcopy(args.samples if len(
args.use_asimov_dataset) == 0 else args.use_asimov_dataset)
if "data" in asimov_nicks:
asimov_nicks.remove("data")
if args.polarisation:
args.no_ewkz_as_dy = True
if args.run1 and (args.emb or args.ttbar_retuned):
log.critical(
"Embedding --emb and --ttbar-retuned only valid for run2. Remove --emb and --tbar-retuned or select run2 samples."
)
sys.exit(1)
elif args.run1:
sample_settings = samples.Samples()
else:
sample_settings = samples.Samples(
embedding=args.emb,
embedding_weight=args.embedded_weights,
ttbar_retuned=args.ttbar_retuned)
if args.mssm:
bkg_samples = [
sample for sample in args.samples
if sample not in ["data", "htt", "ggh", "bbh"]
]
sig_samples_raw = [
sample for sample in args.samples
if sample in ["htt", "ggh", "bbh"]
]
else:
args.samples = [
sample for sample in args.samples
if hasattr(samples.Samples, sample)
]
list_of_samples = [
getattr(samples.Samples, sample) for sample in args.samples
]
if args.emb:
if args.run1:
log.critical(
"Embedding --emb only valid for run2. Remove --emb or select run2 samples."
)
sys.exit(1)
sample_settings = samples.Samples(
embedding=True, embedding_weight=args.embedded_weights)
else:
sample_settings = samples.Samples()
if args.mssm:
bkg_samples = [
sample for sample in args.samples
if sample not in ["data", "htt", "ggh", "bbh"]
]
sig_samples_raw = [
sample for sample in args.samples
if sample in ["htt", "ggh", "bbh"]
]
else:
bkg_samples = [
sample for sample in args.samples
if sample not in ["data", "htt", "ggh", "qqh", "vh"]
