From 4a525bf65590677d384e4298b6790ff4877c3638 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Tue, 12 Oct 2021 10:00:15 -0500 Subject: [PATCH] bug fix --- modules/util/util.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/modules/util/util.py b/modules/util/util.py index 56d5abe9..67d21071 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]) -- GitLab