snapshot...

modules/util/plot.py

@@ -238,7 +238,8 @@ def plot_image3(image, bt, cmap='Greys'):
     fig.savefig('/Users/tomrink/image')
 
 
-def make_icing_image(h5f, probs, ice_lons, ice_lats, clvrx_str_time, satellite, domain, ice_lons_vld=None, ice_lats_vld=None, imagename='icing', extent=[-110, -60, 10, 55]):
+def make_icing_image(h5f, probs, ice_lons, ice_lats, clvrx_str_time, satellite, domain, ice_lons_vld=None, ice_lats_vld=None,
+                     ice_intensity=None, imagename='icing', extent=[-110, -60, 10, 55]):
     geos, xlen, xmin, xmax, ylen, ymin, ymax = get_cartopy_crs(satellite, domain)
     bg_image = None
     if h5f is not None:
@@ -266,9 +267,20 @@ def make_icing_image(h5f, probs, ice_lons, ice_lats, clvrx_str_time, satellite,
     if ice_lons is not None:
         ax.scatter(ice_lons, ice_lats, s=40.0, marker='o', color='blue', transform=ccrs.PlateCarree())
     if ice_lats_vld is not None:
-        # ax.scatter(ice_lons_vld, ice_lats_vld, s=180.0, marker='o', color='red', transform=ccrs.PlateCarree())
-        ax.scatter(ice_lons_vld, ice_lats_vld, s=500.0, marker='o', color='white', transform=ccrs.PlateCarree())
-        ax.scatter(ice_lons_vld, ice_lats_vld, s=200.0, marker='o', color='black', transform=ccrs.PlateCarree())
+        if ice_intensity is None:
+            # ax.scatter(ice_lons_vld, ice_lats_vld, s=180.0, marker='o', color='red', transform=ccrs.PlateCarree())
+            ax.scatter(ice_lons_vld, ice_lats_vld, s=500.0, marker='o', color='white', transform=ccrs.PlateCarree())
+            ax.scatter(ice_lons_vld, ice_lats_vld, s=200.0, marker='o', color='black', transform=ccrs.PlateCarree())
+        else:
+            lons = ice_lons_vld[ice_intensity == 1]
+            lats = ice_lats_vld[ice_intensity == 1]
+            ax.scatter(lons, lats, s=500.0, marker='o', color='white', transform=ccrs.PlateCarree())
+            ax.scatter(lons, lats, s=200.0, marker='o', color='black', transform=ccrs.PlateCarree())
+            lons = ice_lons_vld[ice_intensity > 1]
+            lats = ice_lats_vld[ice_intensity > 1]
+            ax.scatter(lons, lats, s=500.0, marker='^', color='white', transform=ccrs.PlateCarree())
+            ax.scatter(lons, lats, s=200.0, marker='^', color='black', transform=ccrs.PlateCarree())
+
 
     title = satellite+', '+clvrx_str_time+'  >50% Probability Icing'
     plt.title(title, loc='left', fontweight='bold', fontsize=24)