minor...

modules/util/plot.py

@@ -13,6 +13,8 @@ import matplotlib.pyplot as plt
 import cartopy.crs as ccrs
 import xarray as xr
 import os
+import h5py
+from util.util import get_grid_values_all
 
 from util.setup import home_dir
 
@@ -217,3 +219,26 @@ def plot_image3(image, cmap='Greys'):
     ax.coastlines(resolution='50m', color='black', linewidth=0.25)
     ax.add_feature(ccrs.cartopy.feature.STATES, linewidth=0.25)
     plt.title('GOES-16', loc='left', fontweight='bold', fontsize=15)
+
+
+def make_icing_image(clvrx_file, ice_lons, ice_lats, ice_lons_vld=None, ice_lats_vld=None, filename='testimage'):
+    h5f = h5py.File(clvrx_file, 'r')
+    image = get_grid_values_all(h5f, 'temp_10_4um_nom')
+
+    exmpl_ds = xr.open_dataset(exmp_file)
+    mdat = exmpl_ds.metpy.parse_cf('Rad')
+    geos = mdat.metpy.cartopy_crs
+    x = mdat.x
+    y = mdat.y
+
+    fig = plt.figure(figsize=(15, 24))
+    ax = fig.add_subplot(1, 1, 1, projection=geos)
+    ax.imshow(image, origin='upper', extent=(x.min(), x.max(), y.min(), y.max()), cmap='Greys', transform=geos)
+    ax.coastlines(resolution='50m', color='black', linewidth=0.25)
+    ax.add_feature(ccrs.cartopy.feature.STATES, linewidth=0.25)
+    ax.scatter(ice_lons, ice_lats, s=0.6, marker='o', color='orange', transform=ccrs.PlateCarree())
+    if ice_lats_vld is not None:
+        ax.scatter(ice_lons_vld, ice_lats_vld, s=20.0, marker='o', color='blue', transform=ccrs.PlateCarree())
+    plt.title('GOES-16, 2019-04-30_14:26Z  >60% Probability Icing', loc='left', fontweight='bold', fontsize=15)
+    ImageDirAndName = os.path.join('/Users/tomrink', filename)
+    fig.savefig(ImageDirAndName)
\ No newline at end of file