From 7d7ed5f1cef10d913c6da4e1ce434a37d13aa5a1 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Thu, 10 Dec 2020 17:16:11 -0600 Subject: [PATCH] snapshot... --- modules/aeolus/aeolus_amv.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py index 2377e1ad..c5bb1771 100644 --- a/modules/aeolus/aeolus_amv.py +++ b/modules/aeolus/aeolus_amv.py @@ -523,6 +523,42 @@ def get_product_at_locs(raob_to_amv_dct, ts, files, filepath=None): return m_dct +def get_product_at_locs1x1(raob_to_amv_dct, ts, files, filepath=None): + keys = list(raob_to_amv_dct.keys()) + m_dct = {} + + nav = files.get_navigation() + the_params = files.get_parameters() + num_params = len(the_params) + + if filepath is None: + filepath, ftime, f_idx = files.get_file(ts) + ds = Dataset(filepath) + + var_s = [] + for pstr in the_params: + var = ds[pstr] + var_s.append(var) + + for key in keys: + amvs = raob_to_amv_dct.get(key) + num_amvs = amvs.shape[1] + alons = amvs[0, :] + alats = amvs[1, :] + + cc, ll = nav.earth_to_lc_s(alons, alats) + + aaa = np.zeros((num_params, num_amvs), dtype=np.float) + for vidx, var in enumerate(var_s): + for k in range(num_amvs): + aaa[vidx, k] = var[ll[k], cc[k]].data + + m_dct[key] = aaa + + ds.close() + + return m_dct + def run_best_fit(raob_to_amv_dct, raob_dct, gfs_filename=None): keys = list(raob_to_amv_dct.keys()) -- GitLab