From 5fd90aa9dd1ddb56f0854d2c4a470024341acb82 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 14 Jul 2022 16:28:37 -0500
Subject: [PATCH] snapshot...
---
modules/aeolus/aeolus_amv.py | 8 ++++++++
1 file changed, 8 insertions(+)
diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py
index d947a46e..25cd5ba8 100644
--- a/modules/aeolus/aeolus_amv.py
+++ b/modules/aeolus/aeolus_amv.py
@@ -2118,6 +2118,8 @@ def create_bestfit_file(filename, match_dct, raob_dct, raob_ts, gfs_at_raob_dct,
dist_to_raob = rootgrp.createVariable('dist_to_raob', 'f4', ['time', 'num_raob_profs', 'max_num_amvs'])
dist_to_raob.units = 'km'
+ amv_source_index = rootgrp.createVariable('amv_source_index', 'f4', ['time', 'num_raob_profs', 'max_num_amvs'])
+
var_s = rg_exmpl.variables
for pidx, param in enumerate(amv_params):
v = var_s[param]
@@ -2225,6 +2227,12 @@ def create_bestfit_file(filename, match_dct, raob_dct, raob_ts, gfs_at_raob_dct,
dist_to_raob[0, idx, 0:num_amvs] = haversine_np(key.lon, key.lat, amvs[:, 0], amvs[:, 1])
+ param_s = match_dct.get(key)
+ idx_idx = param_s.shape[0] - 1
+ amv_idxs = param_s[idx_idx, ]
+
+ amv_source_index[0, idx, 0:num_amvs] = amv_idxs[:]
+
rg_exmpl.close()
rootgrp.close()
--
GitLab