Merge pull request #7 from zshaheen/more_parameters

More parameters

acme_diags/acme_parameter.py

@@ -7,9 +7,37 @@ def __init__(self):
         self.test_data_path = ''
         self.reference_data_set = ''
         self.test_data_set = ''
+
         self.variables = ''
         self.season = ''
+        self.regrid_tool = 'esmf'
+        self.regrid_method = 'linear'
+        self.output_file = ''
+
+        # VCS Options
+        self.main_title = 'Main Title'
+
+        self.test_name = ''
+        self.test_title = 'Reference'
+        self.test_colormap = ''
+        self.test_levels = []
+        self.test_units = ''
+
+        self.reference_name = ''
+        self.reference_title = 'Observation'
+        self.reference_colormap = ''
+        self.reference_levels = []
+        self.reference_units = ''
+
+        self.diff_name = ''
+        self.diff_title = 'Model - Observation'
+        self.diff_colormap = ''
+        self.diff_levels = []
+        self.diff_units = ''
 
+        self.canvas_size_w = 1212
+        self.canvas_size_h = 1628
+        self.logo = True
 
     def check_values(self):
         # just check if reference_data_path + reference_data_set and

acme_diags/acme_parser.py

@@ -66,3 +66,63 @@ def load_default_args(self):
             dest='output_file',
             help='Name of the output file',
             required=False)
+
+        self.add_argument(
+            '--reference_name',
+            dest='reference_name',
+            help='Name of the reference variable',
+            required=False)
+
+        self.add_argument(
+            '--test_name',
+            dest='test_name',
+            help='Name of the test variable',
+            required=False)
+
+        self.add_argument(
+            '--diff_name',
+            dest='diff_name',
+            help='Name of the difference variable',
+            required=False)
+
+        self.add_argument(
+            '--main_title',
+            dest='main_title',
+            help='The big title that appears on the top of the graph',
+            required=False)
+
+        self.add_argument(
+            '--reference_title',
+            dest='reference_title',
+            help='Title for the middle graph.',
+            required=False)
+
+        self.add_argument(
+            '--test_title',
+            dest='test_title',
+            help='Title for the top graph',
+            required=False)
+
+        self.add_argument(
+            '--diff_title',
+            dest='diff_title',
+            help='Title for the bottom graph',
+            required=False)
+
+        self.add_argument(
+            '--reference_colormap',
+            dest='reference_colormap',
+            help='Colormap for the middle graph.',
+            required=False)
+
+        self.add_argument(
+            '--test_colormap',
+            dest='test_colormap',
+            help='Colormap for the top graph',
+            required=False)
+
+        self.add_argument(
+            '--diff_colormap',
+            dest='diff_colormap',
+            help='Colormap for the bottom graph',
+            required=False)

acme_diags/plotset5/make_plot_set_5_json.py

@@ -0,0 +1,220 @@
+#!/usr/bin/env python
+import os
+import cdms2
+import cdutil
+import vcs
+import genutil.statistics
+import numpy
+import acme_diags.acme_parser
+
+def compute_rmse(model, obs):
+    rmse = -numpy.infty
+    try:
+        rmse = float(genutil.statistics.rms(model, obs, axis='xy', weights='generate'))
+    except Exception, err:
+        print err
+    return rmse
+
+def compute_corr(model, obs):
+    corr = -numpy.infty
+    try:
+        corr = float(genutil.statistics.correlation(model, obs, axis='xy', weights='generate'))
+    except Exception, err:
+        print err
+    return corr
+
+def plot_min_max_mean(canvas, template, variable, plot_name):
+    """canvas is a vcs.Canvas, template is a vcs.Template,
+    variable is a cdms2.tvariable.TransientVariable"""
+    var_min = '%.2f' % float(variable.min())
+    var_max = '%.2f' % float(variable.max())
+
+    # Doesn't work: var_mean = str(round(float(variable.mean()), 2))
+    var_mean = '%.2f' % cdutil.averager(variable, axis='xy', weights='generate')
+
+    # Todo: Just have a textorientation object in the script titled 'min_max_mean'
+    # which will have the height and everything
+    # EX: text_orientation = canvas.gettextorientation("min_max_mean")
+    text_orientation = canvas.gettextorientation(template.mean.textorientation)
+    height = text_orientation.height
+
+    # Draw the "Max", "Mean", and "Min" labels
+    plot_text(canvas, 'Min', template.min.x, template.min.y, height, "left", plot_name + '_min_label')
+    plot_text(canvas, 'Max', template.max.x, template.max.y, height, "left", plot_name + '_max_label')
+    plot_text(canvas, 'Mean', template.mean.x, template.mean.y, height, "left", plot_name + '_mean_label')
+
+    # Draw the actual mean, max, min labels
+    plot_text(canvas, '0.00', template.min.x+0.12, template.min.y, height, "right", plot_name + '_min_value')
+    plot_text(canvas, '0.00', template.max.x+0.12, template.max.y, height, "right", plot_name + '_max_value')
+    plot_text(canvas, '0.00', template.mean.x+0.12, template.mean.y, height, "right", plot_name + '_mean_value')
+
+def plot_rmse_and_corr(canvas, template, model, obs):
+    """canvas is a vcs.Canvas, template is a vcs.Template,
+    model and obs are a cdms2.tvariable.TransientVariable"""
+
+    rmse = '%.2f' % compute_rmse(obs, model)
+    corr = '%.2f' % compute_corr(obs, model)
+
+    text_orientation = canvas.gettextorientation(template.mean.textorientation)
+    height = text_orientation.height
+
+    plot_text(canvas, "RMSE", template.comment1.x, template.comment1.y, height, "left", 'diff_plot_comment1_title')
+    plot_text(canvas, "CORR", template.comment2.x, template.comment2.y, height, "left", 'diff_plot_comment2_title')
+
+    plot_text(canvas, rmse, template.comment1.x+0.12, template.comment1.y, height, "right", 'diff_plot_comment1_value')
+    plot_text(canvas, corr, template.comment2.x+0.12, template.comment2.y, height, "right", 'diff_plot_comment2_value')
+
+def plot_text(canvas, label_string, x, y, height, align, vcs_name):
+    label = canvas.createtextcombined(vcs_name)
+    label.x = x
+    label.y = y
+    label.string = label_string
+    label.height = height
+    label.halign = align
+    label.script('plot_set_5_new.json')
+    canvas.plot(label)
+
+parser = acme_diags.acme_parser.ACMEParser()
+parameter = parser.get_parameter()
+
+reference_data_path = parameter.reference_data_path
+reference_data_set = parameter.reference_data_set # observation
+test_data_path = parameter.test_data_path
+test_data_set = parameter.test_data_set # model
+
+case_id = parameter.case_id
+if not os.path.exists(case_id):
+    os.makedirs(case_id)
+var = parameter.variables
+season = parameter.season
+
+
+# Below should be read from metadata
+mod_name = '1850_alpha6_01 (yrs0070-0099)'
+obs_name = 'GPCP (yrs1979-2009)'
+
+f_obs = cdms2.open(reference_data_path + reference_data_set)
+f_mod = cdms2.open(test_data_path + test_data_set)
+
+obs_pr = f_obs(var, longitude=(-180, 540))
+mod_pr = (f_mod('PRECC', longitude=(-180, 540)) + f_mod('PRECL', longitude=(-180, 540)))*3600.0*24.0*1000.0
+mod_pr.units = 'mm/day'
+
+# For plotting, original grid is plotted for model observation, differece plot is regridded to coaser grid. Need if statement to evaluate grid size. aminusb_2ax from uvcmetrics takes care of this,which also considers complex corner cases.
+axes1 = mod_pr.getAxisList()
+axes2 = obs_pr.getAxisList()
+if len(axes1[1]) <= len(axes2[1]): # use nlat to decide data resolution, higher number means higher data resolution. For the difference plot, regrid toward lower resolution
+    model_grid = mod_pr.getGrid()
+    mod_pr_reg = mod_pr
+    obs_pr_reg = obs_pr.regrid(model_grid, regridTool='esmf', regridMethod='linear')
+else:
+    obs_grid = obs_pr.getGrid()
+    obs_pr_reg = obs_pr
+    mod_pr_reg = mod_pr.regrid(obs_grid, regridTool='esmf', regridMethod='linear')
+dif_pr = mod_pr_reg - obs_pr_reg
+
+# Plotting
+x = vcs.init(bg=True, geometry=(1212,1628))
+
+x.scriptrun('plot_set_5.json')
+#x.scriptrun('plot_set_5_new.json')
+template_0 = x.gettemplate('plotset5_0_x_0')
+template_1 = x.gettemplate('plotset5_0_x_1')
+template_2 = x.gettemplate('plotset5_0_x_2')
+
+mod_pr.long_name = 'model'
+obs_pr.long_name = 'observation'
+dif_pr.long_name = 'model-observation'
+
+template_0.title.priority = 1
+template_1.title.priority = 1
+template_2.title.priority = 1
+
+template_0.units.priority = 1
+template_1.units.priority = 1
+template_2.units.priority = 1
+
+mod_pr.id = mod_name
+obs_pr.id = obs_name
+template_0.dataname.priority = 1
+template_1.dataname.priority = 1
+
+# model and observation graph
+
+isofill = x.createisofill('reference_isofill')
+isofill.datawc_x1 = 0
+isofill.datawc_x2 = 360
+isofill.datawc_y1 = -90
+isofill.datawc_y2 = 90
+isofill.levels = [0, 0.2, 0.5, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14, 17]
+# NOTE: because of the obs and model data files used,
+# there is no 360 degree value, so we use 358 as 0.
+# Same for 0 where we use 2 instead.
+isofill.xticlabels1 = {0: '0', 30: '30E', 60: '60E', 90: '90E',
+                       120: '120E', 150: '150E', 180: '180W', 210: '150W',
+                       240: '120W', 270: '90W', 300: '60W', 330: '30W', 360: '0'}
+isofill.yticlabels1 = {-90: '90S', -80: '80S', -60: '60S', -40: '40S',
+                       -20:'20S', 0: 'Eq', 20: '20N', 40: '40N', 60: '60N',
+                       80: '80N', 90: '90N'}
+
+# ext_1 and ext_2 are arrows
+isofill.ext_1 = True
+isofill.ext_2 = True
+isofill.script('plot_set_5_new.json')
+
+isofill = x.createisofill('test_isofill')
+isofill.datawc_x1 = 0
+isofill.datawc_x2 = 360
+isofill.datawc_y1 = -90
+isofill.datawc_y2 = 90
+isofill.levels = [0, 0.2, 0.5, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 14, 17]
+# NOTE: because of the obs and model data files used,
+# there is no 360 degree value, so we use 358 as 0.
+# Same for 0 where we use 2 instead.
+isofill.xticlabels1 = {0: '0', 30: '30E', 60: '60E', 90: '90E',
+                       120: '120E', 150: '150E', 180: '180W', 210: '150W',
+                       240: '120W', 270: '90W', 300: '60W', 330: '30W', 360: '0'}
+isofill.yticlabels1 = {-90: '90S', -80: '80S', -60: '60S', -40: '40S',
+                       -20:'20S', 0: 'Eq', 20: '20N', 40: '40N', 60: '60N',
+                       80: '80N', 90: '90N'}
+
+# ext_1 and ext_2 are arrows
+isofill.ext_1 = True
+isofill.ext_2 = True
+isofill.script('plot_set_5_new.json')
+
+plot_min_max_mean(x, template_0, mod_pr, 'test')
+plot_min_max_mean(x, template_1, obs_pr, 'reference')
+#x.plot(mod_pr, template_0, vcs.getisofill('reference_isofill'))
+x.plot(mod_pr, template_0, isofill)
+#x.plot(obs_pr, template_1, vcs.getisofill('reference_isofill'))
+x.plot(obs_pr, template_1, isofill)
+
+# Create main title for the 3 plots
+plot_text(x, ' '.join([var, season]), 0.5, 0.98, 18, "center", 'main_title')
+
+
+# difference graph
+isofill = x.createisofill('diff_isofill')
+isofill.datawc_x1 = 0
+isofill.datawc_x2 = 360
+isofill.datawc_y1 = -90
+isofill.datawc_y2 = 90
+
+isofill.levels=[-6, -5, -4, -3, -2, -1, -0.5, 0, 0.5, 1, 2, 3, 4, 5, 6]
+#isofill.levels=vcs.mkscale(dif_pr.min(), dif_pr.max())
+isofill.ext_1 = True
+isofill.ext_2 = True
+
+isofill.colormap = x.getcolormap('bl_to_darkred')
+# do this for only old colormaps
+colors = vcs.getcolors(isofill.levels, colors=range(6, 240))
+isofill.fillareacolors = colors
+isofill.script('plot_set_5_new.json')
+
+plot_min_max_mean(x, template_2, dif_pr, 'diff')
+#x.plot(dif_pr, template_2, isofill)
+x.plot(dif_pr, template_2, vcs.getisofill('diff_isofill'))
+
+plot_rmse_and_corr(x, template_2, mod_pr_reg, obs_pr_reg)
+x.png(case_id + '/' + parameter.output_file)

acme_diags/plotset5/plot_set_5.json

@@ -187,7 +187,7 @@
     "y": 0.9565555391090741
    },
    "title": {
-    "priority": 0,
+    "priority": 1,
     "textorientation": "__textorientation_675849204359008",
     "texttable": "default",
     "x": 0.4915,
@@ -555,7 +555,7 @@
     "y": 0.6432222057757409
    },
    "title": {
-    "priority": 0,
+    "priority": 1,
     "textorientation": "__textorientation_428791170453960",
     "texttable": "default",
     "x": 0.4915,
@@ -820,7 +820,7 @@
     "y2": 0.32333333333333325
    },
    "dataname": {
-    "priority": 0,
+    "priority": 1,
     "textorientation": "__textorientation_937211945085079",
     "texttable": "default",
     "x": 0.12300000000000001,
@@ -923,7 +923,7 @@
     "y": 0.3298888724424074
    },
    "title": {
-    "priority": 0,
+    "priority": 1,
     "textorientation": "__textorientation_916780501350462",
     "texttable": "default",
     "x": 0.4915,