From 6723350c8471fef9ecbf70ceca4e3f7e0d10e817 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 28 May 2021 13:47:06 -0500
Subject: [PATCH] minor

---
 modules/amv/aeolus.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/modules/amv/aeolus.py b/modules/amv/aeolus.py
index 54ea8346..27f8b29c 100644
--- a/modules/amv/aeolus.py
+++ b/modules/amv/aeolus.py
@@ -36,6 +36,7 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
     num_files = len(clvrx_files.flist)
 
     keys = list(aeolus_dict.keys())
+    num_profs = len(keys)
 
     a_lons = [[] for i in range(num_files)]
     a_lats = [[] for i in range(num_files)]
@@ -62,15 +63,14 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
 
         a_lons[f_idx].append(lon)
         a_lats[f_idx].append(lat)
-        a_profs[f_idx].append(layers)
         a_times[f_idx].append(key)
+        a_profs[f_idx].append(layers)
 
     for f_idx in range(num_files):
         a_lons[f_idx] = np.array(a_lons[f_idx])
         a_lats[f_idx] = np.array(a_lats[f_idx])
         a_hgts[f_idx] = np.array(a_hgts[f_idx])
         a_times[f_idx] = np.array(a_times[f_idx])
-        a_profs[f_idx] = np.array(a_profs[f_idx])
 
     grd_hgt = []
     grd_mean = []
@@ -114,9 +114,14 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
         grd_crds = np.unravel_index(nn_idxs, grd_vals.shape)
         a_lons[f_idx] = a_lons[f_idx][on_earth]
         a_lats[f_idx] = a_lats[f_idx][on_earth]
-        a_profs[f_idx] = a_profs[f_idx][on_earth]
         a_times[f_idx] = a_times[f_idx][on_earth]
 
+        layers = []
+        for k in range(len(nn_idxs)):
+            if on_earth[k]:
+                layers.append(a_profs[f_idx][k])
+        a_profs[f_idx] = layers
+
         total += len(nn_idxs)
         # loop over Aeolus profile valid in the file
         for k in range(len(nn_idxs)):
@@ -210,7 +215,9 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
             layer_top.append(tops)
 
         h5f.close()
+        print(f_idx, num_files)
 
+    print('num aeolus profiles: ', num_profs)
     print('Overall: ', num_hits, vld_grd, total)
     print('*')
     hits = np.array(hits)
-- 
GitLab