ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/OSUT3Analysis/Configuration/python/processingUtilities.py
Revision: 1.18
Committed: Fri May 17 08:34:38 2013 UTC (11 years, 11 months ago) by wulsin
Content type: text/x-python
Branch: MAIN
CVS Tags: V02-01-01
Changes since 1.17: +2 -0 lines
Log Message: 
add -q, --quickMerge option to merge output without running makeCutFlows.py and makePlots.py

File Contents

# User Rev Content
1 lantonel 1.1 #!/usr/bin/env python
2     import os
3 ahart 1.15 import re
4 lantonel 1.1 import sys
5 ahart 1.15 import math
6 lantonel 1.1 import datetime
7 lantonel 1.8 from optparse import OptionParser
8 ahart 1.15 #from OSUT3Analysis.Configuration.configurationOptions import *
9 lantonel 1.2
10 ahart 1.15 def split_composite_datasets(datasets, composite_dataset_definitions):
11 lantonel 1.1 for dataset in datasets:
12     if dataset in composite_dataset_definitions:
13     for component_dataset in composite_dataset_definitions[dataset]:
14     datasets.insert(datasets.index(dataset),component_dataset)
15     datasets.remove(dataset)
16     return datasets
17    
18 ahart 1.15 def get_composite_datasets(datasets, composite_dataset_definitions):
19 lantonel 1.1 composite_datasets = []
20     for dataset in datasets:
21     if dataset in composite_dataset_definitions:
22     composite_datasets.append(dataset)
23     return composite_datasets
24    
25 lantonel 1.7 def set_condor_submit_dir(arguments):
26     if arguments.condorDir:
27     condor_dir = "condor/%s" % arguments.condorDir
28 lantonel 1.1 else:
29     now = datetime.datetime.now()
30     date_hash = now.strftime("%Y_%m_%d_%H:%M:%S")
31     condor_dir = "condor/condor_%s" % date_hash
32 lantonel 1.3 #print "Condor submit directory set to ",condor_dir
33 lantonel 1.1 return condor_dir
34    
35 lantonel 1.7 def set_condor_output_dir(arguments):
36     if arguments.condorDir:
37     condor_dir = "condor/%s" % arguments.condorDir
38 lantonel 1.1 else: #get most recent condor submission directory
39     dir_list = []
40     for directory in os.listdir("./condor/"):
41     if directory.find("condor_") is not -1:
42     dir_list.append(directory)
43     if len(dir_list) is 0:
44     sys.exit("Cannot find last condor working directory")
45     dir_list.sort(reverse=True)
46     condor_dir = "condor/%s" % dir_list[0]
47 lantonel 1.3 #print "Condor output directory set to ",condor_dir
48 lantonel 1.1 return condor_dir
49    
50     def set_commandline_arguments(parser):
51 lantonel 1.12 #### Configuration-related Options
52 lantonel 1.8 parser.add_option("-l", "--localConfig", dest="localConfig",
53 lantonel 1.1 help="local configuration file")
54 lantonel 1.8 parser.add_option("-c", "--condorDir", dest="condorDir",
55 lantonel 1.5 help="condor output directory")
56 lantonel 1.12 parser.add_option("-o", "--output-file", dest="outputFileName",
57     help="specify an output file name for the histogram file, default is 'stacked_histograms.root'")
58     parser.add_option("-t", "--no-weights", action="store_true", dest="noWeights", default=False,
59     help="do not apply cross section weights")
60 wulsin 1.18 parser.add_option("-q", "--quickMerge", action="store_true", dest="quickMerge", default=False,
61     help="do merge without making cutflow or plots")
62 lantonel 1.12
63    
64     #### Histogram Formatting Options
65 lantonel 1.8 parser.add_option("-n", "--normalize", action="store_true", dest="normalizeToData", default=False,
66 lantonel 1.5 help="normalize total background MC yield to the data")
67 lantonel 1.8 parser.add_option("-u", "--unit-area", action="store_true", dest="normalizeToUnitArea", default=False,
68 lantonel 1.5 help="normalize all samples to unit area (useful to compare shapes)")
69 lantonel 1.8 parser.add_option("-e", "--empty", action="store_true", dest="noStack", default=False,
70 lantonel 1.5 help="don't stack the background samples, draw them as empty histograms instead")
71 lantonel 1.12
72 lantonel 1.8 parser.add_option("-r", "--ratio", action="store_true", dest="makeRatioPlots", default=False,
73 lantonel 1.6 help="draw (data-MC)/MC plots below all 1D histograms")
74 lantonel 1.12 parser.add_option("-d", "--diff", action="store_true", dest="makeDiffPlots", default=False,
75     help="draw data-MC plots below all 1D histograms")
76 ahart 1.15 parser.add_option("-b", "--rebin", dest="rebinFactor",
77 lantonel 1.12 help="Rebin all the histograms which will have at least 10 bins after rebinning")
78 lantonel 1.13 parser.add_option("--2D", action="store_true", dest="draw2DPlots", default=False,
79     help="draw stacked 2D histograms")
80 ahart 1.15 parser.add_option("-y", "--yields", action="store_true", dest="printYields", default=False,
81     help="Include the yield of each source in the legend")
82 lantonel 1.13
83 lantonel 1.12
84 lantonel 1.1 return parser
85    
86 ahart 1.15 def get_short_name(dataset, dataset_names):
87 lantonel 1.1 for key in dataset_names:
88     if dataset_names[key] == dataset:
89 ahart 1.15 return key
90 lantonel 1.1 return "Unknown"
91 ahart 1.15
92     def stop_ctau (dataset):
93     if not re.match (r"stop[^_]*to[^_]*_[^_]*mm.*", dataset):
94     return 0.0
95     return float (re.sub (r"stop[^_]*to[^_]*_([^_]*)mm.*", r"\1", dataset))
96    
97     def source_stop_ctau (ctau):
98     return int (math.pow (10.0, math.ceil (math.log10 (ctau))))
99    
100     def add_stops (options, masses, ctaus, bottomBranchingRatios = []):
101     if not bottomBranchingRatios:
102 ahart 1.16 bottomBranchingRatios.append (50.0)
103 ahart 1.15 for mass in masses:
104     for ctau in ctaus:
105     for bottomBranchingRatio in bottomBranchingRatios:
106 ahart 1.16 datasetName = 'stop' + str (mass) + "_" + str (ctau) + "mm_br" + str (int (bottomBranchingRatio))
107 ahart 1.15 bottomDatasetName = 'stop' + str (mass) + "toBl_" + str (ctau) + "mm"
108     sourceBottomDatasetName = 'stop' + str (mass) + "toBl_" + str (source_stop_ctau (ctau)) + "mm"
109     topDatasetName = 'stop' + str (mass) + "toTnu_" + str (ctau) + "mm"
110     sourceTopDatasetName = 'stop' + str (mass) + "toTnu_" + str (source_stop_ctau (ctau)) + "mm"
111     mixedDatasetName = 'stop' + str (mass) + "toBT_" + str (ctau) + "mm"
112     sourceMixedDatasetName = 'stop' + str (mass) + "toBT_" + str (source_stop_ctau (ctau)) + "mm"
113    
114     options['datasets'].append (datasetName)
115 ahart 1.16 bottomBranchingRatio /= 100.0
116 ahart 1.15 options['composite_dataset_definitions'][datasetName] = {
117     bottomDatasetName : bottomBranchingRatio * bottomBranchingRatio,
118     topDatasetName : (1 - bottomBranchingRatio) * (1 - bottomBranchingRatio),
119     mixedDatasetName : (1 - bottomBranchingRatio * bottomBranchingRatio - (1 - bottomBranchingRatio) * (1 - bottomBranchingRatio))
120     }
121     options['dataset_names'][bottomDatasetName] = options['dataset_names'][sourceBottomDatasetName]
122     options['dataset_names'][topDatasetName] = options['dataset_names'][sourceTopDatasetName]
123     options['dataset_names'][mixedDatasetName] = options['dataset_names'][sourceMixedDatasetName]
124     options['nJobs'][bottomDatasetName] = 1
125     options['nJobs'][topDatasetName] = 1
126     options['nJobs'][mixedDatasetName] = 1
127     options['maxEvents'][bottomDatasetName] = -1
128     options['maxEvents'][topDatasetName] = -1
129     options['maxEvents'][mixedDatasetName] = -1
130     options['types'][datasetName] = "signalMC"
131     options['types'][bottomDatasetName] = "signalMC"
132     options['types'][topDatasetName] = "signalMC"
133     options['types'][mixedDatasetName] = "signalMC"
134 ahart 1.17 options['labels'][datasetName] = str (mass) + " GeV stop (#LTc#tau#GT = " + str (ctau) + " mm)"
135     options['labels'][bottomDatasetName] = "#tilde{t}#tilde{t}#rightarrowbbll (#LTc#tau#GT = " + str (ctau) + " mm)"
136     options['labels'][topDatasetName] = "#tilde{t}#tilde{t}#rightarrowtt#nu#nu (#LTc#tau#GT = " + str (ctau) + " mm)"
137     options['labels'][mixedDatasetName] = "#tilde{t}#tilde{t}#rightarrowbtl#nu (#LTc#tau#GT = " + str (ctau) + " mm)"