add option to save figures as pngs

Author: jalimena

Configuration/python/processingUtilities.py

@@ -108,6 +108,8 @@ def set_commandline_arguments(parser):
                       help="Include the yield of each source in the legend")
     parser.add_option("-p", "--pdfs",  action="store_true", dest="savePDFs", default=False,
                       help="Save pdfs files for all plots made")
+    parser.add_option("--pngs",  action="store_true", dest="savePNGs", default=False,
+                      help="Save pngs files for all plots made")
 
     return parser
 

Configuration/scripts/makePlots.py

@@ -1225,6 +1225,9 @@ def MakeOneDHist(pathToDir,histogramName,integrateDir):
     if arguments.savePDFs:
         pathToDirString = plainTextString(pathToDir)
         Canvas.SaveAs(condor_dir+"/stacked_histograms_pdfs/"+pathToDirString+"/"+histogramName+".pdf")
+    if arguments.savePNGs:
+        pathToDirString = plainTextString(pathToDir)
+        Canvas.SaveAs(condor_dir+"/stacked_histograms_pngs/"+pathToDirString+"/"+histogramName+".png")
 
 
 ##########################################################################################################################################
@@ -1518,6 +1521,9 @@ def MakeTwoDHist(pathToDir,histogramName):
     if arguments.savePDFs:
         pathToDirString = plainTextString(pathToDir)
         Canvas.SaveAs(condor_dir+"/stacked_histograms_pdfs/"+pathToDirString+"/"+histogramName+".pdf")
+    if arguments.savePNGs:
+        pathToDirString = plainTextString(pathToDir)
+        Canvas.SaveAs(condor_dir+"/stacked_histograms_pngs/"+pathToDirString+"/"+histogramName+".png")
 
 
 
@@ -1581,6 +1587,13 @@ def MakeTwoDHist(pathToDir,histogramName):
         pass
     os.mkdir ("%s/stacked_histograms_pdfs" % (condor_dir))
 
+if arguments.savePNGs:
+    try:
+        shutil.rmtree ("%s/stacked_histograms_pngs" % (condor_dir))
+    except OSError:
+        pass
+    os.mkdir ("%s/stacked_histograms_pngs" % (condor_dir))
+
 #### make output file
 outputFileString = ''
 if arguments.rebinFactor:
@@ -1613,6 +1626,8 @@ def MakeTwoDHist(pathToDir,histogramName):
     outputFile.mkdir(rootDirectory)
     if arguments.savePDFs:
         os.mkdir ("%s/stacked_histograms_pdfs/%s" % (condor_dir,plainTextString(rootDirectory)))
+    if arguments.savePNGs:
+        os.mkdir ("%s/stacked_histograms_pngs/%s" % (condor_dir,plainTextString(rootDirectory)))
 
     #cd to root directory and look for histograms
     inputFile.cd(rootDirectory)
@@ -1635,6 +1650,8 @@ def MakeTwoDHist(pathToDir,histogramName):
             gDirectory.mkdir(key2.GetName())
             if arguments.savePDFs:
                 os.mkdir ("%s/stacked_histograms_pdfs/%s" % (condor_dir,plainTextString(level2Directory)))
+            if arguments.savePNGs:
+                os.mkdir ("%s/stacked_histograms_pngs/%s" % (condor_dir,plainTextString(level2Directory)))
 
             #####################################################
             ###  This layer is typically the "channels" layer ###
@@ -1662,6 +1679,8 @@ def MakeTwoDHist(pathToDir,histogramName):
                     gDirectory.mkdir(key3.GetName())
                     if arguments.savePDFs:
                         os.mkdir ("%s/stacked_histograms_pdfs/%s" % (condor_dir,plainTextString(level3Directory)))
+                    if arguments.savePNGs:
+                        os.mkdir ("%s/stacked_histograms_pngs/%s" % (condor_dir,plainTextString(level3Directory)))
 
                     #################################################
                     ###  This layer is typically the "cuts" layer ###