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"]