From 512fce06ba4e2ae803b0937c269382fcb35b7a4f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 12 Nov 2021 11:47:55 -0600
Subject: [PATCH] improve output file
---
modules/util/util.py | 12 ++++++++++--
1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/modules/util/util.py b/modules/util/util.py
index 49f435ae..298ef6ed 100644
--- a/modules/util/util.py
+++ b/modules/util/util.py
@@ -563,6 +563,14 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
# -------------------------------------------------------------------------------------------
+flt_level_ranges_str = {k: None for k in range(5)}
+flt_level_ranges_str[0] = '0_2000'
+flt_level_ranges_str[1] = '2000_4000'
+flt_level_ranges_str[2] = '4000_6000'
+flt_level_ranges_str[3] = '6000_8000'
+flt_level_ranges_str[4] = '8000_15000'
+
+
def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lons, lats):
outfile_name = output_dir + 'icing_prediction_'+clvrx_str_time+'.h5'
h5f_out = h5py.File(outfile_name, 'w')
@@ -573,7 +581,7 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
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 = h5f_out.create_dataset('icing_prediction_level_'+flt_level_ranges_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
@@ -581,7 +589,7 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon
for flvl in flt_lvls:
probs = probs_dct[flvl]
- icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+str(flvl), data=probs, dtype='f4')
+ icing_prob_ds = h5f_out.create_dataset('icing_probability_level_'+flt_level_ranges_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
--
GitLab