From 018b45a915c509d6fcf8cb07415ad60b4620fce8 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 11 Dec 2021 13:24:58 -0600
Subject: [PATCH] snapshot...
---
modules/aeolus/aeolus_amv.py | 66 ++++++++++++++++++++++++++++++++++++
1 file changed, 66 insertions(+)
diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py
index a4b4d06a..a033436c 100644
--- a/modules/aeolus/aeolus_amv.py
+++ b/modules/aeolus/aeolus_amv.py
@@ -1731,6 +1731,72 @@ def match_amvs_to_aeolus_fast(aeolus_dict, amv_files_path, amv_source='OPS', ban
return match_dict
+
+def match_calipso_clavrx_to_amvs(calipso_clavrx_file, amv_files_path, amv_source='OPS', band='14', amv_files=None):
+ if amv_files is None:
+ amv_files = get_datasource(amv_files_path, amv_source, band=band)
+ nav = amv_files.get_navigation()
+ amv_params = amv_files.get_parameters()
+ all_params = [amv_files.lon_name, amv_files.lat_name, amv_files.elem_name, amv_files.line_name] + amv_params
+ match_dict = {}
+ coords = {'num_params': all_params}
+ dims = ['num_params', 'num_amvs']
+
+ calipso_clavrx_ds = None
+
+ nom_time = calipso_clavrx_ds.get_datetime(calipso_clavrx_file).timestamp()
+
+ fname, ftime, f_idx = amv_files.get_file_containing_time(nom_time)
+ if f_idx is None:
+ return
+
+ match_dict[nom_time] = []
+
+ amv_ds = Dataset(fname)
+ amv_lons = amv_ds[amv_files.lon_name][:]
+ amv_lats = amv_ds[amv_files.lat_name][:]
+ amv_cc = amv_ds[amv_files.elem_name][:]
+ amv_ll = amv_ds[amv_files.line_name][:]
+
+ param_s = []
+ param_s.append(amv_lons)
+ param_s.append(amv_lats)
+ param_s.append(cc)
+ param_s.append(ll)
+ for param in amv_params:
+ if param == 'V_3D':
+ param_s.append(amv_ds[param][:, 0])
+ param_s.append(amv_ds[param][:, 1])
+ else:
+ param_s.append(amv_ds[param][:])
+
+ amv_ds.close()
+
+ xs = calipso_clavrx_ds['x'][:]
+ ys = calipso_clavrx_ds['y'][:]
+ clvr_xy_s = zip(xs, ys)
+
+ for clvr_xy in clvr_xy_s:
+ cc, ll = clvr_xy
+
+ c_rng, l_rng = get_search_box(cc, ll)
+
+ in_cc = np.logical_and(amv_cc > c_rng[0], amv_cc < c_rng[1])
+ in_ll = np.logical_and(amv_ll > l_rng[0], amv_ll < l_rng[1])
+ in_box = np.logical_and(in_cc, in_ll)
+
+ num_amvs = np.sum(in_box)
+ if num_amvs == 0:
+ continue
+ # dist = haversine_np(lon, lat, amv_lons[in_box], amv_lats[in_box])
+ param_nd = np.vstack(param_s)
+ param_nd = param_nd[:, in_box]
+ amvs_da = xr.DataArray(param_nd, coords=coords, dims=dims)
+ match_dict[nom_time].append((cc, ll, f_idx, None, amvs_da))
+
+ return match_dict
+
+
# aeolus_dict: time -> profiles
# amv_files_path: directory containing AMVs, '/home/user/amvdir/'
# return dict: aeolus time -> tuple (amv_lon, amv_lat, amv_pres, amv_spd, amv_dir)
--
GitLab