From 00d7b309bc18194b87f5b9df2c03b312482e1a2b Mon Sep 17 00:00:00 2001 From: Grant Firl Date: Tue, 11 Feb 2020 11:50:17 -0700 Subject: [PATCH] change python plotting script to have usetex=configurable value rather than hard-coded true to remove requirement of docker container having latex installed --- scm/etc/scripts/gmtb_scm_plotting_routines.py | 21 ++++++++++--------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/scm/etc/scripts/gmtb_scm_plotting_routines.py b/scm/etc/scripts/gmtb_scm_plotting_routines.py index 9f2c911ad..22125bd4d 100755 --- a/scm/etc/scripts/gmtb_scm_plotting_routines.py +++ b/scm/etc/scripts/gmtb_scm_plotting_routines.py @@ -14,6 +14,7 @@ mpl.rcParams['savefig.dpi'] = 150 mpl.rcParams['legend.handlelength'] = 5 +latex_labels = False def calc_skill_score(data, experiment, R_0, k): sigma_data = np.std(data) @@ -25,7 +26,7 @@ def calc_skill_score(data, experiment, R_0, k): def plot_profile(z, values, x_label, y_label, filename, xticks=[], yticks=[], y_inverted = False, y_log = False, y_lim = []): #xticks = [x_start, x_end, x_num] - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) fig = plt.figure() @@ -52,7 +53,7 @@ def plot_profile(z, values, x_label, y_label, filename, xticks=[], yticks=[], y_ def plot_profile_multi(z, values, labels, x_label, y_label, filename, obs_z=None, obs_values=None, xticks=[], yticks=[], y_inverted = False, y_log = False, y_lim = [], conversion_factor=1.0, freeze_axis=None, line_type=None, color_index=None, skill_scores=False, zero_line=False): #xticks = [x_start, x_end, x_num] - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) if np.count_nonzero(values) == 0: print 'The plot for {} will not be created due to all zero values'.format(x_label) @@ -226,7 +227,7 @@ def plot_profile_multi(z, values, labels, x_label, y_label, filename, obs_z=None def plot_profile_compare(z, values, LES_values, LES_z, x_label, y_label, filename, xticks=[], yticks=[], y_inverted = False, y_log = False, x_lim = [], y_lim = []): #xticks = [x_start, x_end, x_num] - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) fig = plt.figure() @@ -260,7 +261,7 @@ def plot_profile_compare(z, values, LES_values, LES_z, x_label, y_label, filenam plt.close() def plot_time_series(time, values, x_label, y_label, filename): - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) fig = plt.figure() @@ -272,7 +273,7 @@ def plot_time_series(time, values, x_label, y_label, filename): plt.close() def plot_time_series_compare(time, values, LES_time, LES_values, x_label, y_label, filename): - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) fig = plt.figure() @@ -291,7 +292,7 @@ def plot_time_series_compare(time, values, LES_time, LES_values, x_label, y_labe plt.close() def plot_time_series_multi(time, values, labels, x_label, y_label, filename, obs_time=None, obs_values=None, obs_label=None, line_type=None, color_index=None, skill_scores=False, conversion_factor=1.0): - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) if np.count_nonzero(values) == 0: print 'The plot for {} will not be created due to all zero values'.format(y_label) @@ -405,7 +406,7 @@ def plot_time_series_multi(time, values, labels, x_label, y_label, filename, obs def contour_plot_firl(x_dim, y_dim, values, min_val, max_val, title, x_label, y_label, filename, xticks=[], yticks=[], plot_mean = 0, annotation = 0, y_inverted = 0, y_log = False, y_lim = [], conversion_factor=1.0): #os.environ['PATH'] = os.environ['PATH'] + ':/usr/texbin' #plt.rc('ps', usedistiller='xpdf') - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) if np.count_nonzero(values) == 0: print 'The plot for {} will not be created due to all zero values'.format(title) @@ -485,7 +486,7 @@ def contour_plot_firl(x_dim, y_dim, values, min_val, max_val, title, x_label, y_ def plot_profile_multi_ens(z, values, labels, x_label, y_label, filename, obs_z=None, obs_values=None, xticks=[], yticks=[], y_inverted = False, y_log = False, y_lim = [], freeze_axis=None, line_type=None, color_index=None, skill_scores=False): #xticks = [x_start, x_end, x_num] - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) #ensemble data processing values_ens_processed = [] @@ -661,7 +662,7 @@ def plot_profile_multi_ens(z, values, labels, x_label, y_label, filename, obs_z= plt.close() def plot_time_series_multi_ens(time, values, labels, x_label, y_label, filename, obs_time=None, obs_values=None, obs_label=None, line_type=None, color_index=None, skill_scores=False): - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) fig = plt.figure() @@ -806,7 +807,7 @@ def plot_time_series_multi_ens(time, values, labels, x_label, y_label, filename, plt.close() def plot_scatter_multi(x, y, x_label, y_label, filename, x_lim=[], y_lim=[], color_index=None): - plt.rc('text', usetex=True) + plt.rc('text', usetex=latex_labels) colors = ['#e41a1c','#4daf4a','#377eb8','#984ea3','#ff7f00','#a65628','#f781bf','#ffff33'] if color_index is None: