From bdb233403efd9273ee338b67ab42f1b4b5d9f3bc Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 29 Mar 2022 15:35:10 -0500
Subject: [PATCH] snapshot...

---
 modules/aeolus/aeolus_amv.py | 41 ++++++++++--------------------------
 1 file changed, 11 insertions(+), 30 deletions(-)

diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py
index 7623866d..7f1e1bb5 100644
--- a/modules/aeolus/aeolus_amv.py
+++ b/modules/aeolus/aeolus_amv.py
@@ -2015,7 +2015,7 @@ def create_file(filename, aeolus_to_amv_dct, aeolus_dct, amv_files, cld_lyr=Fals
     rootgrp.close()
 
 def create_file_new2(filename, match_dct, raob_dct, gfs_at_raob_dct, bf_dct, prd_dct, raob_filename, amv_file_name):
-    num_aparams = 7
+    num_aparams = 4
     num_aprofs = 0
     max_num_alevels = 0
     max_num_amvs = 0
@@ -2032,44 +2032,25 @@ def create_file_new2(filename, match_dct, raob_dct, gfs_at_raob_dct, bf_dct, prd
     # scan to get max num levels, amvs
     keys = list(match_dct.keys())
     for key in keys:
-        print(key)
         raob = raob_dct.get(key)
-        print(raob.shape)
-        bf_tup = bf_dct.get(key)
-        amvs = bf_tup[0]
-        num_amvs = amvs.shape[0]
         nlevs = raob.shape[0]
         lat = key[0]
         lon = key[1]
+
+        alons.append(lon)
+        alats.append(lat)
+
+        bf_tup = bf_dct.get(key)
+
+        amvs = bf_tup[0]
+        num_amvs = amvs.shape[0]
         if nlevs > max_num_alevels:
             max_num_alevels = nlevs
         if num_amvs > max_num_amvs:
             max_num_amvs = num_amvs
 
-    #     tup_s = match_dct.get(key)
-    #     for tup in tup_s:
-    #         num_aprofs += 1
-    #         prof = tup[3]
-    #         lat = prof[0, 0]
-    #         lon = prof[0, 1]
-    #         alons.append(lon)
-    #         alats.append(lat)
-    #         atimes.append(key)
-    #         elems.append(tup[0])
-    #         lines.append(tup[1])
-    #         fidxs.append(tup[2])
-    #
-    #         nlevs = prof.shape[0]
-    #         if nlevs > max_num_alevels:
-    #             max_num_alevels = nlevs
-    #
-    #         amvs = tup[4]
-    #         num_amvs = amvs.shape[1]
-    #         if num_amvs > max_num_amvs:
-    #             max_num_amvs = num_amvs
-    #
-    # alons = np.array(alons)
-    # alats = np.array(alats)
+    alons = np.array(alons)
+    alats = np.array(alats)
     # atimes = np.array(atimes)
     # elems = np.array(elems)
     # lines = np.array(lines)
-- 
GitLab