Skip to content
Snippets Groups Projects
Commit e837f890 authored by Nils Brüggemann's avatar Nils Brüggemann
Browse files

pyic_view: Some changes to allow to use healpix data.

parent 510819da
No related branches found
No related tags found
No related merge requests found
Pipeline #99117 passed
......@@ -85,7 +85,7 @@ def get_fpath_ckdtree(data, res, path_grid, gname='auto', fpath_tgrid='auto'):
return fpath_ckdtree
def get_data(ds, var_name, it, iz, fpath_ckdtree, lon_reg, lat_reg):
def get_data(ds, var_name, it, iz, res, lon_reg, lat_reg):
isel_dict = dict(time=it)
if ds[var_name].ndim==3:
depth_name = pyic.identify_depth_name(ds[var_name])
......@@ -95,7 +95,7 @@ def get_data(ds, var_name, it, iz, fpath_ckdtree, lon_reg, lat_reg):
#else:
# raise ValueError(f"::: Unknown number of dimensions for {var_name}: {ds[var_name].shape}")
dai = pyic.interp_to_rectgrid_xr(
ds[var_name].isel(**isel_dict), fpath_ckdtree,
ds[var_name].isel(**isel_dict), res=res,
lon_reg=lon_reg, lat_reg=lat_reg,
)
dai.attrs["depth_name"] = depth_name
......@@ -307,7 +307,8 @@ class view(object):
"clon_bnds", "clat_bnds", "elon_bnds", "elat_bnds",
"vlon_bnds", "vlat_bnds",
"clon", "clat", "elon", "elat",
"lev"
"lev",
"healpix",
]
for var in delvars:
try:
......@@ -325,7 +326,7 @@ class view(object):
self.fpath_ckdtree = get_fpath_ckdtree(self.ds, self.res, self.path_grid)
self.dai = get_data(
self.ds, self.var_name, self.it, self.iz,
self.fpath_ckdtree, self.lon_reg, self.lat_reg
self.res, self.lon_reg, self.lat_reg
)
self.Lon, self.Lat = np.meshgrid(self.dai.lon.data, self.dai.lat.data)
if self.proj=="None":
......@@ -422,7 +423,7 @@ class view(object):
self.update_lon_lat_reg()
self.dai = get_data(
self.ds, self.var_name, self.it, self.iz,
self.fpath_ckdtree, self.lon_reg, self.lat_reg
self.res, self.lon_reg, self.lat_reg
)
self.Lon, self.Lat = np.meshgrid(self.dai.lon, self.dai.lat)
self.hm[0].set_array(self.dai.data.flatten())
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment