snapshot...

modules/util/util.py

@@ -566,26 +566,37 @@ def write_icing_file(clvrx_str_time, preds_dct, probs_dct, x, y, lons, lats):
     outfile_name = homedir + 'icing_prediction_'+clvrx_str_time+'.h5'
     h5f_out = h5py.File(outfile_name, 'w')
 
+    dim_0_name = 'pixel_elements'
+    dim_1_name = 'scan_lines'
+
     flt_lvls = list(preds_dct.keys())
     for flvl in flt_lvls:
         preds = preds_dct[flvl]
         icing_pred_ds = h5f_out.create_dataset('icing_prediction_level_'+str(flvl), data=preds, dtype='i2')
         icing_pred_ds.attrs.create('coordinates', data='y x')
         icing_pred_ds.attrs.create('grid_mapping', data='Projection')
+        icing_pred_ds.dims[0].label = dim_0_name
+        icing_pred_ds.dims[1].label = dim_1_name
 
     for flvl in flt_lvls:
         probs = probs_dct[flvl]
-        icing_pred_ds = h5f_out.create_dataset('icing_probability_level_'+str(flvl), data=probs, dtype='f4')
-        icing_pred_ds.attrs.create('coordinates', data='y x')
-        icing_pred_ds.attrs.create('grid_mapping', data='Projection')
+        icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+str(flvl), data=probs, dtype='f4')
+        icing_prob_ds.attrs.create('coordinates', data='y x')
+        icing_prob_ds.attrs.create('grid_mapping', data='Projection')
+        icing_prob_ds.dims[0].label = dim_0_name
+        icing_prob_ds.dims[1].label = dim_1_name
 
     lon_ds = h5f_out.create_dataset('longitude', data=lons, dtype='f4')
     lon_ds.attrs.create('units', data='degrees_east')
     lon_ds.attrs.create('long_name', data='icing prediction longitude')
+    lon_ds.dims[0] = dim_0_name
+    lon_ds.dims[1] = dim_1_name
 
     lat_ds = h5f_out.create_dataset('latitude', data=lats, dtype='f4')
     lat_ds.attrs.create('units', data='degrees_north')
     lat_ds.attrs.create('long_name', data='icing prediction latitude')
+    lat_ds.dims[0] = dim_0_name
+    lat_ds.dims[1] = dim_1_name
 
     proj_ds = h5f_out.create_dataset('Projection', data=0, dtype='b')
     proj_ds.attrs.create('long_name', data='Himawari Imagery Projection')