diff --git a/modules/util/util.py b/modules/util/util.py index 56d5abe9ae0f7dec048a73409520a2808632018f..67d21071a01d5ae1728dda8f7be0283dc641eeab 100644 --- a/modules/util/util.py +++ b/modules/util/util.py @@ -542,8 +542,8 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain n_x = int(xlen/w_x) n_y = int(ylen/w_y) - ll = [j_0 + j*w_y for j in range(n_y-1)] - cc = [i_0 + i*w_x for i in range(n_x-1)] + # ll = [j_0 + j*w_y for j in range(n_y-1)] + # cc = [i_0 + i*w_x for i in range(n_x-1)] for ds_name in name_list: for j in range(n_y-1): @@ -552,6 +552,16 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain i_ul = i * w_x grd_dct_n[ds_name].append(grd_dct[ds_name][j_ul:j_ul+w_y, i_ul:i_ul+w_x]) + # TODO: replace this with numpy meshgrid and flatten + ll = [] + cc = [] + for j in range(n_y-1): + j_y = j_0 + j*w_y + for i in range(n_x-1): + i_x = i_0 + i*w_x + ll.append(j_y) + cc.append(i_x) + grd_dct = {name: None for name in name_list} for didx, ds_name in enumerate(name_list): grd_dct[ds_name] = np.stack(grd_dct_n[ds_name])