--- UserCode/OSUT3Analysis/Configuration/scripts/makeSinglePlot.py 2013/03/13 09:27:11 1.3 +++ UserCode/OSUT3Analysis/Configuration/scripts/makeSinglePlot.py 2013/03/28 09:06:24 1.10 @@ -10,10 +10,6 @@ from OSUT3Analysis.Configuration.configu from OSUT3Analysis.Configuration.processingUtilities import * - - - - ##set default plotting options line_width = 2 plotting_options = "" @@ -22,17 +18,20 @@ parser = OptionParser() parser = set_commandline_arguments(parser) -parser.add_argument("--hist", action="store_true", dest="plot_hist", default=False, +parser.add_option("--hist", action="store_true", dest="plot_hist", default=False, help="plot as hollow histograms instead of error bar crosses") -parser.add_argument("--line-width", dest="line_width", +parser.add_option("--line-width", dest="line_width", help="set line width (default is 2)") +parser.add_option("--pdf", action="store_true", dest="plot_savePdf", default=False, + help="save plot as pdf in addition") (arguments, args) = parser.parse_args() if arguments.localConfig: - sys.path.append(os.getcwd()) + sys.path.insert(0,os.getcwd()) exec("from " + arguments.localConfig.rstrip('.py') + " import *") + print arguments.localConfig.rstrip('.py') outputFileName = "simple_plot.root" if arguments.outputFileName: @@ -69,7 +68,6 @@ gROOT.ForceStyle() outputFile = TFile(outputFileName, "RECREATE") - datasets_needed = [] for histogram in input_histograms: if histogram['dataset'] not in datasets_needed: @@ -77,6 +75,7 @@ for histogram in input_histograms: weight = intLumi / 10000.0 for dataset in datasets_needed: + dataset_file = "%s/%s.root" % (condor_dir,dataset) if types[dataset] != "data": os.system("mergeTFileServiceHistograms -i %s -o %s -w %g" % (dataset_file, dataset_file + "_tmp", weight)) @@ -103,7 +102,6 @@ for histogram in input_histograms: if inputFile.IsZombie() or not inputFile.GetNkeys(): continue - print "OSUAnalysis/"+histogram['channel']+"/"+histogram['name'] Histogram = inputFile.Get("OSUAnalysis/"+histogram['channel']+"/"+histogram['name']).Clone() Histogram.SetDirectory(0) inputFile.Close() @@ -131,6 +129,7 @@ counter = 0 for Histogram in Histograms: if counter is 0: Histogram.SetMaximum(1.1*finalMax) + Histogram.SetMinimum(0.0001) Histogram.Draw(plotting_options) else: @@ -148,7 +147,11 @@ if arguments.normalizeToUnitArea: outputFile.cd() Canvas.Write() - + +if arguments.plot_savePdf: + pdfFileName = outputFileName.replace(".root", ".pdf") + Canvas.SaveAs(pdfFileName) + print "Saved file: " + pdfFileName for dataset in datasets_needed: @@ -156,3 +159,4 @@ for dataset in datasets_needed: os.remove(dataset_file) outputFile.Close() +print "Saved plot in file: " + outputFileName