From 6aea0cda178339e87295a93a70a4aafc6e8047b7 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 15 Sep 2022 12:43:41 -0500
Subject: [PATCH] snaphot...
---
modules/amv/caliop_clavrx_amv.py | 33 ++++++++++++++++++++++++++++++++
1 file changed, 33 insertions(+)
diff --git a/modules/amv/caliop_clavrx_amv.py b/modules/amv/caliop_clavrx_amv.py
index 8f24fe91..e52c1ddd 100644
--- a/modules/amv/caliop_clavrx_amv.py
+++ b/modules/amv/caliop_clavrx_amv.py
@@ -350,3 +350,36 @@ def analyze(filename):
amv_press_avg = np.nanmean(amv_press, axis=1)
return cc_ctp, acha_ctp, amv_press_avg, lats
+
+
+def compute_and_add_geo_hgt(path_to_amvs, path_to_gfs, amv_source, band='14'):
+ gfs_ds = get_datasource(path_to_gfs, 'GFS')
+ amv_ds = get_datasource(path_to_amvs, amv_source, file_time_span=20, band=band)
+ amv_filenames = amv_ds.flist
+
+ for fname in amv_filenames:
+ print(fname)
+ nom_time = amv_ds.get_datetime(fname).timestamp()
+ amvs_nd = get_amv_nd(fname, delimiter=',')
+ amvs_nd = filter_amvs(amvs_nd)
+ amv_lons = amvs_nd[:, amv_lon_idx]
+ amv_lats = amvs_nd[:, amv_lat_idx]
+ num_amvs = amvs_nd.shape[0]
+
+ gfs_fname, _, _ = gfs_ds.get_file(nom_time, window=180.0)
+ if gfs_fname is None:
+ return None
+ print(gfs_fname)
+ gfs_xr = xr.open_dataset(gfs_fname)
+ gfs_press = gfs_xr['pressure levels']
+ gfs_press = gfs_press.values
+
+ temp_prof = get_vert_profile_s(gfs_xr, ['temperature'], amv_lons, amv_lats, method='nearest')
+ temp_prof = temp_prof.values
+ temp_prof_s = temp_prof[0, :, :]
+
+ alt_s = []
+ for k in range(num_amvs):
+ alt = pressure_to_altitude(amvs_nd[k, amv_pres_idx], None, gfs_press, temp_prof_s[k, :])
+ alt_s.append(alt.magnitude)
+ alt_s = np.array(alt_s)
\ No newline at end of file
--
GitLab