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