diff --git a/modules/icing/util.py b/modules/icing/util.py index 6dedb02d7f26bd3772b50a704107fb572086658b..644a242ba1ede6d7f49ef33422941d916e699e17 100644 --- a/modules/icing/util.py +++ b/modules/icing/util.py @@ -90,13 +90,15 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h lons_2d, lats_2d, x_rad, y_rad = get_lon_lat_2d_mesh(nav, ll, cc, offset=int(8 / res_fac)) ancil_data_dct, _, _ = make_for_full_domain_predict(h5f, name_list= - ['solar_zenith_angle', 'sensor_zenith_angle', 'cld_height_acha', 'cld_geo_thick'], + ['solar_zenith_angle', 'sensor_zenith_angle', 'cld_height_acha', 'cld_geo_thick', 'temp_10_4um_nom'], satellite=satellite, domain=domain, res_fac=res_fac) satzen = ancil_data_dct['sensor_zenith_angle'] solzen = ancil_data_dct['solar_zenith_angle'] + bt_10_4 = ancil_data_dct['temp_10_4um_nom'] day_idxs = [] nght_idxs = [] + avg_bts = [] for j in range(num_lines): for i in range(num_elems): k = i + j*num_elems @@ -106,6 +108,7 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h day_idxs.append(k) else: nght_idxs.append(k) + avg_bts.append(bt_10_4[k]) num_tiles = num_lines * num_elems num_day_tiles = len(day_idxs) @@ -176,7 +179,9 @@ def run_icing_predict(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output_dir=h preds_2d_dct[flvl] = fd_preds.reshape((num_lines, num_elems)) probs_2d_dct[flvl] = fd_probs.reshape((num_lines, num_elems)) - write_icing_file_nc4(clvrx_str_time, output_dir, preds_2d_dct, probs_2d_dct, + bt_10_4_2d = bt_10_4.reshape((num_lines, num_elems)) + + write_icing_file_nc4(clvrx_str_time, output_dir, preds_2d_dct, probs_2d_dct, bt_10_4_2d, x_rad, y_rad, lons_2d, lats_2d, cc, ll, satellite=satellite, domain=domain, use_nan=use_nan, has_time=has_time, prob_thresh=prob_thresh)