Sola/model_info_2d
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/SolaProject/model_info_2d

Browse Source
This commit is contained in:
sola_108zx
2023-12-28 15:51:53 +08:00
parent 6c8279bbd2 1ac7632716
commit 6d28267008
1 changed files with 22 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
model_info_2d.py
View File

@ -291,11 +291,11 @@ class model_info_2d(object):
def get_extent( def get_extent(
self, self,
cx : int, cx : float,
cy : int, cy : float,
dx : int, dx : float,
dy : int, dy : float,
ratio : float = 0.8 ratio : float = 1
) -> list: ) -> list:
""" """
用于获取指定数据范围的经纬度坐标 用于获取指定数据范围的经纬度坐标
@ -306,10 +306,22 @@ class model_info_2d(object):
dy: 中心点周围y网格数 dy: 中心点周围y网格数
""" """
XLON, XLAT = self.get_grid() XLON, XLAT = self.get_grid()
XLON, XLAT = XLON[cy-dy:cy+dy, cx-dx:cx+dx], XLAT[cy-dy:cy+dy, cx-dx:cx+dx] # ys, ye, xs, xe = np.floor(cy-dy), np.ceil(cy+dy), np.floor(cx-dx), np.ceil(cx+dx)
clon, clat = np.mean(XLON), np.mean(XLAT) lon_start, _ = self.grid_lonlat(cx-dx*ratio, cy)
dlon, dlat = (np.max(XLON) - np.min(XLON))/2*ratio, (np.max(XLAT) - np.min(XLAT))/2*ratio lon_end, _ = self.grid_lonlat(cx+dx*ratio, cy)
extent = [clon-dlon, clon+dlon, clat-dlat, clat+dlat] _, lat_start = self.grid_lonlat(cx, cy-dy*ratio)
_, lat_end = self.grid_lonlat(cx, cy+dy*ratio)
# if lon_start > lon_end:
# lon_end += 360
# XLON, XLAT = XLON[cy-dy:cy+dy, cx-dx:cx+dx], XLAT[cy-dy:cy+dy, cx-dx:cx+dx]
# clon, clat = np.mean(XLON), np.mean(XLAT)
# dlon, dlat = (np.max(XLON) - np.min(XLON))/2*ratio, (np.max(XLAT) - np.min(XLAT))/2*ratio
# clon, clat = (lon_end + lon_start)/2, (lat_end + lat_start)/2
# dlon, dlat = (lon_end - lon_start)/2*ratio, (lat_end - lat_start)/2*ratio
# extent = [(clon-dlon+180)%360-180, (clon+dlon+180)%360-180, clat-dlat if clat-dlat>=-90 else -90, clat+dlat if clat+dlat<=90 else 90]
constrain_lon = lambda x: (x+180)%360-180
constrain_lat = lambda x: min(abs(x), 90) * (1 if x > 0 else -1)
extent = [constrain_lon(lon_start), constrain_lon(lon_end), constrain_lat(lat_start), constrain_lat(lat_end)]
return extent return extent
def flat_array( def flat_array(