From 7d618bfd7f43c62ba8b0fbd50947cec118217f28 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 29 Oct 2021 12:04:45 -0500
Subject: [PATCH] snapshot...
---
modules/util/util.py | 17 +++++++++++++----
1 file changed, 13 insertions(+), 4 deletions(-)
diff --git a/modules/util/util.py b/modules/util/util.py
index 86d7f2c3..4ce4560e 100644
--- a/modules/util/util.py
+++ b/modules/util/util.py
@@ -562,13 +562,22 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
# -------------------------------------------------------------------------------------------
-def write_icing_file(clvrx_str_time, preds, x, y, lons, lats):
+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')
- icing_pred_ds = h5f_out.create_dataset('icing_prediction', data=preds, dtype='i2')
- icing_pred_ds.attrs.create('coordinates', data='y x')
- icing_pred_ds.attrs.create('grid_mapping', data='Projection')
+ 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_'+flvl, data=preds, dtype='i2')
+ icing_pred_ds.attrs.create('coordinates', data='y x')
+ icing_pred_ds.attrs.create('grid_mapping', data='Projection')
+
+ for flvl in flt_lvls:
+ probs = probs_dct[flvl]
+ icing_pred_ds = h5f_out.create_dataset('icing_probability_level_'+flvl, data=probs, dtype='f4')
+ icing_pred_ds.attrs.create('coordinates', data='y x')
+ icing_pred_ds.attrs.create('grid_mapping', data='Projection')
lon_ds = h5f_out.create_dataset('longitude', data=lons, dtype='f4')
lon_ds.attrs.create('units', data='degrees_east')
--
GitLab