| 1 |
|
#!/usr/bin/env python |
| 2 |
|
import os |
| 3 |
+ |
import re |
| 4 |
|
import sys |
| 5 |
+ |
import math |
| 6 |
|
import datetime |
| 7 |
+ |
import copy |
| 8 |
|
from optparse import OptionParser |
| 9 |
+ |
import OSUT3Analysis.DBTools.osusub_cfg as osusub |
| 10 |
+ |
import FWCore.ParameterSet.Config as cms |
| 11 |
|
|
| 12 |
< |
from OSUT3Analysis.Configuration.configurationOptions import * |
| 8 |
< |
|
| 9 |
< |
|
| 10 |
< |
def split_composite_datasets(datasets): |
| 12 |
> |
def split_composite_datasets(datasets, composite_dataset_definitions): |
| 13 |
|
for dataset in datasets: |
| 14 |
|
if dataset in composite_dataset_definitions: |
| 15 |
|
for component_dataset in composite_dataset_definitions[dataset]: |
| 17 |
|
datasets.remove(dataset) |
| 18 |
|
return datasets |
| 19 |
|
|
| 20 |
< |
def get_composite_datasets(datasets): |
| 20 |
> |
def get_composite_datasets(datasets, composite_dataset_definitions): |
| 21 |
|
composite_datasets = [] |
| 22 |
|
for dataset in datasets: |
| 23 |
|
if dataset in composite_dataset_definitions: |
| 24 |
|
composite_datasets.append(dataset) |
| 25 |
|
return composite_datasets |
| 26 |
|
|
| 27 |
< |
def set_condor_submit_dir(options): |
| 28 |
< |
if options.condorDir: |
| 29 |
< |
condor_dir = "condor/%s" % options.condorDir |
| 27 |
> |
def set_condor_submit_dir(arguments): |
| 28 |
> |
if arguments.condorDir: |
| 29 |
> |
condor_dir = "condor/%s" % arguments.condorDir |
| 30 |
|
else: |
| 31 |
|
now = datetime.datetime.now() |
| 32 |
|
date_hash = now.strftime("%Y_%m_%d_%H:%M:%S") |
| 34 |
|
#print "Condor submit directory set to ",condor_dir |
| 35 |
|
return condor_dir |
| 36 |
|
|
| 37 |
< |
def set_condor_output_dir(options): |
| 38 |
< |
if options.condorDir: |
| 39 |
< |
condor_dir = "condor/%s" % options.condorDir |
| 37 |
> |
def set_condor_output_dir(arguments): |
| 38 |
> |
if arguments.condorDir: |
| 39 |
> |
condor_dir = "condor/%s" % arguments.condorDir |
| 40 |
|
else: #get most recent condor submission directory |
| 41 |
|
dir_list = [] |
| 42 |
|
for directory in os.listdir("./condor/"): |
| 50 |
|
return condor_dir |
| 51 |
|
|
| 52 |
|
def set_commandline_arguments(parser): |
| 53 |
+ |
#### Configuration-related Options |
| 54 |
|
parser.add_option("-l", "--localConfig", dest="localConfig", |
| 55 |
|
help="local configuration file") |
| 53 |
– |
|
| 56 |
|
parser.add_option("-c", "--condorDir", dest="condorDir", |
| 57 |
|
help="condor output directory") |
| 58 |
+ |
parser.add_option("-o", "--output-file", dest="outputFileName", |
| 59 |
+ |
help="specify an output file name for the histogram file, default is 'stacked_histograms.root'") |
| 60 |
+ |
|
| 61 |
+ |
|
| 62 |
+ |
#### Histogram Formatting Options |
| 63 |
|
parser.add_option("-n", "--normalize", action="store_true", dest="normalizeToData", default=False, |
| 64 |
|
help="normalize total background MC yield to the data") |
| 65 |
|
parser.add_option("-u", "--unit-area", action="store_true", dest="normalizeToUnitArea", default=False, |
| 66 |
|
help="normalize all samples to unit area (useful to compare shapes)") |
| 67 |
|
parser.add_option("-e", "--empty", action="store_true", dest="noStack", default=False, |
| 68 |
|
help="don't stack the background samples, draw them as empty histograms instead") |
| 69 |
< |
|
| 69 |
> |
|
| 70 |
> |
parser.add_option("-r", "--ratio", action="store_true", dest="makeRatioPlots", default=False, |
| 71 |
> |
help="draw (data-MC)/MC plots below all 1D histograms") |
| 72 |
> |
parser.add_option("-R", "--ratioYRange", dest="ratioYRange", |
| 73 |
> |
help="maximum of range of vertical scale for ratio plots") |
| 74 |
> |
parser.add_option("-d", "--diff", action="store_true", dest="makeDiffPlots", default=False, |
| 75 |
> |
help="draw data-MC plots below all 1D histograms") |
| 76 |
> |
parser.add_option("-b", "--rebin", dest="rebinFactor", |
| 77 |
> |
help="Rebin all the histograms which will have at least 10 bins after rebinning") |
| 78 |
> |
parser.add_option("--2D", action="store_true", dest="draw2DPlots", default=False, |
| 79 |
> |
help="draw stacked 2D histograms") |
| 80 |
> |
parser.add_option("-y", "--yields", action="store_true", dest="printYields", default=False, |
| 81 |
> |
help="Include the yield of each source in the legend") |
| 82 |
> |
parser.add_option("-p", "--pdfs", action="store_true", dest="savePDFs", default=False, |
| 83 |
> |
help="Save pdfs files for all plots made") |
| 84 |
|
|
| 85 |
|
return parser |
| 86 |
|
|
| 87 |
< |
def get_short_name(dataset): |
| 87 |
> |
def get_short_name(dataset, dataset_names): |
| 88 |
|
for key in dataset_names: |
| 89 |
|
if dataset_names[key] == dataset: |
| 90 |
< |
return key; |
| 90 |
> |
return key |
| 91 |
|
return "Unknown" |
| 92 |
+ |
|
| 93 |
+ |
def stop_ctau (dataset): |
| 94 |
+ |
if not re.match (r"stop[^_]*to[^_]*_[^_]*mm.*", dataset): |
| 95 |
+ |
return 0.0 |
| 96 |
+ |
return float (re.sub (r"stop[^_]*to[^_]*_([^_]*)mm.*", r"\1", dataset)) |
| 97 |
+ |
|
| 98 |
+ |
def source_stop_ctau (ctau): |
| 99 |
+ |
return int (math.pow (10.0, math.ceil (math.log10 (ctau)))) |
| 100 |
+ |
|
| 101 |
+ |
def add_stops (options, masses, ctaus, bottomBranchingRatios = []): |
| 102 |
+ |
if not bottomBranchingRatios: |
| 103 |
+ |
bottomBranchingRatios.append (50.0) |
| 104 |
+ |
for mass in masses: |
| 105 |
+ |
for ctau in ctaus: |
| 106 |
+ |
for bottomBranchingRatio in bottomBranchingRatios: |
| 107 |
+ |
datasetName = 'stop' + str (mass) + "_" + str (ctau) + "mm_br" + str (int (bottomBranchingRatio)) |
| 108 |
+ |
bottomDatasetName = 'stop' + str (mass) + "toBl_" + str (ctau) + "mm" |
| 109 |
+ |
sourceBottomDatasetName = 'stop' + str (mass) + "toBl_" + str (source_stop_ctau (ctau)) + "mm" |
| 110 |
+ |
topDatasetName = 'stop' + str (mass) + "toTnu_" + str (ctau) + "mm" |
| 111 |
+ |
sourceTopDatasetName = 'stop' + str (mass) + "toTnu_" + str (source_stop_ctau (ctau)) + "mm" |
| 112 |
+ |
mixedDatasetName = 'stop' + str (mass) + "toBT_" + str (ctau) + "mm" |
| 113 |
+ |
sourceMixedDatasetName = 'stop' + str (mass) + "toBT_" + str (source_stop_ctau (ctau)) + "mm" |
| 114 |
+ |
|
| 115 |
+ |
options['datasets'].append (datasetName) |
| 116 |
+ |
bottomBranchingRatio /= 100.0 |
| 117 |
+ |
options['composite_dataset_definitions'][datasetName] = { |
| 118 |
+ |
bottomDatasetName : bottomBranchingRatio * bottomBranchingRatio, |
| 119 |
+ |
topDatasetName : (1 - bottomBranchingRatio) * (1 - bottomBranchingRatio), |
| 120 |
+ |
mixedDatasetName : (1 - bottomBranchingRatio * bottomBranchingRatio - (1 - bottomBranchingRatio) * (1 - bottomBranchingRatio)) |
| 121 |
+ |
} |
| 122 |
+ |
options['dataset_names'][bottomDatasetName] = options['dataset_names'][sourceBottomDatasetName] |
| 123 |
+ |
options['dataset_names'][topDatasetName] = options['dataset_names'][sourceTopDatasetName] |
| 124 |
+ |
options['dataset_names'][mixedDatasetName] = options['dataset_names'][sourceMixedDatasetName] |
| 125 |
+ |
options['nJobs'][bottomDatasetName] = 1 |
| 126 |
+ |
options['nJobs'][topDatasetName] = 1 |
| 127 |
+ |
options['nJobs'][mixedDatasetName] = 1 |
| 128 |
+ |
options['maxEvents'][bottomDatasetName] = -1 |
| 129 |
+ |
options['maxEvents'][topDatasetName] = -1 |
| 130 |
+ |
options['maxEvents'][mixedDatasetName] = -1 |
| 131 |
+ |
options['types'][datasetName] = "signalMC" |
| 132 |
+ |
options['types'][bottomDatasetName] = "signalMC" |
| 133 |
+ |
options['types'][topDatasetName] = "signalMC" |
| 134 |
+ |
options['types'][mixedDatasetName] = "signalMC" |
| 135 |
+ |
options['labels'][datasetName] = str (mass) + " GeV stop (#LTc#tau#GT = " + str (ctau) + " mm)" |
| 136 |
+ |
options['labels'][bottomDatasetName] = "#tilde{t}#tilde{t}#rightarrowbbll (#LTc#tau#GT = " + str (ctau) + " mm)" |
| 137 |
+ |
options['labels'][topDatasetName] = "#tilde{t}#tilde{t}#rightarrowtt#nu#nu (#LTc#tau#GT = " + str (ctau) + " mm)" |
| 138 |
+ |
options['labels'][mixedDatasetName] = "#tilde{t}#tilde{t}#rightarrowbtl#nu (#LTc#tau#GT = " + str (ctau) + " mm)" |
| 139 |
+ |
|
| 140 |
+ |
def add_channels (process, channels, outputCommands = ["keep *"]): |
| 141 |
+ |
suffix = "" |
| 142 |
+ |
if osusub.batchMode: |
| 143 |
+ |
suffix = "_" + str (osusub.jobNumber) |
| 144 |
+ |
i = 0 |
| 145 |
+ |
for channel in channels: |
| 146 |
+ |
channelName = channel.name.pythonValue () |
| 147 |
+ |
channelName = channelName[1:-1] |
| 148 |
+ |
try: |
| 149 |
+ |
os.mkdir (channelName) |
| 150 |
+ |
except OSError: |
| 151 |
+ |
pass |
| 152 |
+ |
out = cms.OutputModule ("PoolOutputModule", |
| 153 |
+ |
fileName = cms.untracked.string (channelName + "/bean" + suffix + ".root"), |
| 154 |
+ |
SelectEvents = cms.untracked.PSet ( |
| 155 |
+ |
SelectEvents = cms.vstring ("myFilterPath" + str (i)) |
| 156 |
+ |
), |
| 157 |
+ |
outputCommands = cms.untracked.vstring ( |
| 158 |
+ |
outputCommands |
| 159 |
+ |
), |
| 160 |
+ |
dropMetaData = cms.untracked.string ("ALL") |
| 161 |
+ |
) |
| 162 |
+ |
myFilter = cms.EDFilter ("FilterOnChannelDecision", |
| 163 |
+ |
src = cms.InputTag ("OSUAnalysis", "channelDecisions"), |
| 164 |
+ |
channelName = cms.string (channelName) |
| 165 |
+ |
) |
| 166 |
+ |
myFilterPath = cms.Path (myFilter) |
| 167 |
+ |
myEndPath = cms.EndPath (out) |
| 168 |
+ |
process.__setattr__ ("out" + str (i), out) |
| 169 |
+ |
process.OSUAnalysis.channels.append (channel) |
| 170 |
+ |
process.OSUAnalysis.useEDMFormat = cms.bool (True) |
| 171 |
+ |
process.__setattr__ ("myFilter" + str (i), myFilter) |
| 172 |
+ |
process.__setattr__ ("myFilterPath" + str (i), myFilterPath) |
| 173 |
+ |
process.__setattr__ ("myEndPath" + str (i), myEndPath) |
| 174 |
+ |
i += 1 |
