diff --git a/modules/util/util.py b/modules/util/util.py index b3122b0f20d8f29eff30b6fa5fa3272ba74e8a51..eac7bb3a62256d03a61353f836c64c128b35075f 100644 --- a/modules/util/util.py +++ b/modules/util/util.py @@ -598,7 +598,7 @@ 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): +def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lons, lats, elems, lines): outfile_name = output_dir + 'icing_prediction_'+clvrx_str_time+'.h5' h5f_out = h5py.File(outfile_name, 'w') @@ -670,4 +670,9 @@ def write_icing_file(clvrx_str_time, output_dir, preds_dct, probs_dct, x, y, lon y_ds.attrs.create('LFAC', data=20466275) y_ds.attrs.create('LOFF', data=2750.5) + if elems is not None: + elem_ds = h5f_out.create_dataset('elems', data=elems, dtype='i2') + line_ds = h5f_out.create_dataset('lines', data=lines, dtype='i2') + pass + h5f_out.close()