more fixes

Author: pancetta

projects/parallelSDC/newton_vs_sdc.py

@@ -136,7 +136,7 @@ def plot_graphs(cwd=''):
 
     plt_helper.setup_mpl()
 
-    plt_helper.newfig(textwidth=338.0, scale=1.0)
+    plt_helper.newfig(textwidth=238.96, scale=0.89)
 
     for sweeper, color, marker, label in setups:
         plt_helper.plt.loglog(dt_list, results[sweeper], lw=1, ls='-', color=color, marker=marker,

projects/parallelSDC/preconditioner_playground.py

@@ -193,7 +193,7 @@ def plot_iterations():
     # loop over setups and Q-delta types: one figure per setup, all Qds in one plot
     for setup in setup_list:
 
-        plt_helper.newfig(textwidth=338.0, scale=0.475)
+        plt_helper.newfig(textwidth=238.96, scale=0.89)
 
         for qd_type, marker, color in zip(qd_type_list, marker_list, color_list):
             niter = np.zeros(len(results[setup][1]))

pySDC/helpers/plot_helper.py

@@ -57,7 +57,10 @@ def newfig(textwidth, scale):
     return fig, ax
 
 
-def savefig(filename):
-    plt.savefig('{}.pgf'.format(filename), rasterized=True, bbox_inches='tight')
-    plt.savefig('{}.pdf'.format(filename), rasterized=True, bbox_inches='tight')
-    plt.savefig('{}.png'.format(filename), rasterized=True, bbox_inches='tight')
+def savefig(filename, save_pdf=True, save_pgf=True, save_png=True):
+    if save_pgf:
+        plt.savefig('{}.pgf'.format(filename), rasterized=True, bbox_inches='tight')
+    if save_pdf:
+        plt.savefig('{}.pdf'.format(filename), rasterized=True, bbox_inches='tight')
+    if save_png:
+        plt.savefig('{}.png'.format(filename), rasterized=True, bbox_inches='tight')