ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/OSUT3Analysis/Configuration/scripts/makeCutFlows.py

Comparing UserCode/OSUT3Analysis/Configuration/scripts/makeCutFlows.py (file contents):
Revision 1.6 by ahart, Wed Feb 6 00:28:41 2013 UTC vs.
Revision 1.15 by ahart, Tue Apr 9 21:42:18 2013 UTC

#	Line 9 | Line 9 | from OSUT3Analysis.Configuration.process
9
10		parser = OptionParser()
11		parser = set_commandline_arguments(parser)
12	+	(arguments, args) = parser.parse_args()
13
14	<	(options, args) = parser.parse_args()
14	<
15	<	if options.localConfig:
14	>	if arguments.localConfig:
15		sys.path.append(os.getcwd())
16	<	exec("from " + options.localConfig.rstrip('.py') + " import *")
16	>	exec("from " + arguments.localConfig.rstrip('.py') + " import *")
17
18	<	condor_dir = set_condor_output_dir(options)
18	>	condor_dir = set_condor_output_dir(arguments)
19
20		from ROOT import TFile, gROOT, gStyle, gDirectory, TKey
21
#	Line 27 | Line 26 | texfile = condor_dir + "/cutFlow.tex"
26		replacements = {
27		">":"$>$",
28		"<":"$<$",
29	<	"eta ":"$\\eta$ ",
29	>	" eta ":" $\\eta$ ",
30	>	" abs(eta) ":" $|\\eta|$ ",
31		"#":"Num",
32
33		"\\rightarrow":"{\\rightarrow}",
#	Line 36 | Line 36 | replacements = {
36		"BCtoE QCD":"BCtoE$ $QCD",
37
38
39	<	"Pt ":"pt ",
40	<	"PT ":"pt ",
41	<	"pT ":"pt ",
42	<	"pt ":"$p_{T}$ ",
43	<
44	<	"Ht ":"HT ",
45	<	"ht ":"HT ",
46	<	"hT ":"HT ",
47	<	"HT ":"$H_{T}$ ",
48	<	"tig$H_{T}$ ":"tight ",
39	>	" pt ":" $p_{T}$ ",
40	>	" ht ":"$H_{T}$ ",
41
50	–	"D0":"d0",
42		"d0":"$d_{0}$",
43
44	<	"MET ":"Met ",
45	<	"MEt ":"Met ",
46	<	"met ":"Met ",
47	<	"Met ":"$\\not\\!\\!{E}_{T}$ ",
48	<
49	<	"MHT ":"Mht ",
50	<	"MHt ":"Mht ",
51	<	"mht ":"Mht ",
52	<	"Mht ":"$\\not\\!\\!{H}_{T}$ ",
44	>	" MET ":" Met ",
45	>	" MEt ":" Met ",
46	>	" met ":" Met ",
47	>	" Met ":"$\\not\\!\\!{E}_{T}$ ",
48	>
49	>	" MHT ":" Mht ",
50	>	" MHt ":" Mht ",
51	>	" mht ":" Mht ",
52	>	" Mht ":"$\\not\\!\\!{H}_{T}$ ",
53
54		"M_Z" : "$M_{Z}$",
55		"M_mumu" : "$M_{\\mu\\mu}$",
56		"M_ee" : "$M_{ee}$",
57		"M_ll" : "$M_{ll}$",
58
68	–	"|" : "$|$"
59		}
60
61		secondary_replacements = {
62		"$$<$":"$<"
63	+
64		}
65
66
#	Line 100 | Line 91 | for key in testFile.GetListOfKeys():
91		channels.append(key2.GetName())
92
93		fout = open (texfile, "w")
94	<	fout.write ("\\documentclass{article}\n\n")
94	>	fout.write ("\\documentclass[a2paper,8pt]{article}\n\n")
95		fout.write ("\\usepackage[landscape,margin=0.15cm]{geometry}\n\n")
96		fout.write ("\\usepackage{multirow}\n\n")
97		fout.write ("\\begin{document}\n\n")
#	Line 109 | Line 100 | fout.close ()
100
101		firstChannel = True
102
103	<	weight = intLumi / 10000.0
104	<	for dataset in processed_datasets:
105	<	dataset_file = "%s/%s.root" % (condor_dir,dataset)
106	<	os.system("mergeTFileServiceHistograms -i %s -o %s -w %g" % (dataset_file, dataset_file + "_tmp", weight))
103	>	#weight = intLumi / 10000.0
104	>	#for dataset in processed_datasets:
105	>	#    dataset_file = "%s/%s.root" % (condor_dir,dataset)
106	>	#    fin = TFile (dataset_file)
107	>	#    flags = fin.Get ("flags")
108	>	#    noWeights = flags and flags.GetBinContent (1)
109	>	#    fin.Close ()
110	>	#    if types[dataset] != "data" and not noWeights:
111	>	#        os.system("mergeTFileServiceHistograms -i %s -o %s -w %g" % (dataset_file, dataset_file + "_tmp", weight))
112	>	#    else:
113	>	#        os.system("mergeTFileServiceHistograms -i %s -o %s -w %g" % (dataset_file, dataset_file + "_tmp", 1.0))
114
115		for channel in channels: # loop over final states, which each have their own directory
116	<	fout = open (texfile, "a")
119	<	if not firstChannel:
120	<	fout.write ("\\pagebreak\n\n")
121	<	firstChannel = False
122	<	fout.write ("\\section{" + channel + " channel}\n\n")
123	<	fout.close ()
124	<
116	>	formatted_channel = channel.replace("_"," ")
117		cutFlowArgs = ""
118		selectionArgs = ""
119		minusOneArgs = ""
120		#print hist
121		for dataset in processed_datasets:
122	<	dataset_file = "%s/%s.root_tmp" % (condor_dir,dataset)
122	>	dataset_file = "%s/%s.root" % (condor_dir,dataset)
123		#print dataset_file
124		cutFlowArgs = cutFlowArgs + " " + dataset_file
125		selectionArgs = selectionArgs + " " + dataset_file
#	Line 147 | Line 139 | for channel in channels: # loop over fin
139
140		#make cutFlowTable objects
141		fout = open (texfile, "a")
142	<	fout.write ("\\subsection{Cut Flow}\n\n")
142	>	if not firstChannel:
143	>	fout.write ("\\pagebreak\n\n")
144	>	firstChannel = False
145	>	fout.write ("\\section*{" + formatted_channel + " channel}\n\n")
146	>	fout.write ("\\subsection*{Cut flow}\n\n")
147		fout.close ()
148	<	os.system("cutFlowTable -l %g %s >> %s" % (intLumi,cutFlowArgs,texfile))
148	>	os.system("cutFlowTable -l %g -m %s >> %s" % (intLumi,cutFlowArgs,texfile))
149		fout = open (texfile, "a")
150		fout.write ("\\pagebreak\n\n")
151	<	fout.write ("\\subsection{Individual Selection}\n\n")
151	>	fout.write ("\\section*{" + formatted_channel + " channel}\n\n")
152	>	fout.write ("\\subsection*{Individual selection}\n\n")
153		fout.close ()
154		os.system("cutFlowTable -l %g %s >> %s" % (intLumi,selectionArgs,texfile))
155		fout = open (texfile, "a")
156		fout.write ("\\pagebreak\n\n")
157	<	fout.write ("\\subsection{Minus One}\n\n")
157	>	fout.write ("\\section*{" + formatted_channel + " channel}\n\n")
158	>	fout.write ("\\subsection*{Minus one}\n\n")
159		fout.close ()
160		os.system("cutFlowTable -l %g %s >> %s" % (intLumi,minusOneArgs,texfile))
161
162	<	for dataset in processed_datasets:
163	<	dataset_file = "%s/%s.root_tmp" % (condor_dir,dataset)
164	<	os.remove(dataset_file)
162	>	#for dataset in processed_datasets:
163	>	#    dataset_file = "%s/%s.root_tmp" % (condor_dir,dataset)
164	>	#    os.remove(dataset_file)
165
166		#reformat tex files
167		for line in fileinput.FileInput(texfile,inplace=1):

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines