diff --git a/modules/util/util.py b/modules/util/util.py index 50470704890ae959a080b28c9b1b831e8afbd0fa..fa2d4ce73f5af5bdd813a21128fba15d5d9eaeac 100644 --- a/modules/util/util.py +++ b/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')