|
16 | 16 | import pandas as pd |
17 | 17 | import geopandas as gpd |
18 | 18 | import xarray as xr |
19 | | -from netCDF4 import Dataset |
20 | 19 | from siem.proxy import create_wrf_grid, configure_grid_spatial |
21 | 20 | import pyproj |
22 | 21 |
|
@@ -101,22 +100,35 @@ def retrive_proj_from(geogrid_path: str): |
101 | 100 | a = 6370000.0 |
102 | 101 | b = 6370000.0 |
103 | 102 |
|
104 | | - lcc = pyproj.Proj(proj='lcc', lat_1=geo_ds.TRUELAT1, lat_2=geo_ds.TRUELAT2, |
105 | | - lat_0=geo_ds.MOAD_CEN_LAT, lon_0=geo_ds.STAND_LON, |
106 | | - a=a, b=b) |
107 | | - merc = pyproj.Proj(proj='merc', lon_0=geo_ds.STAND_LON, lat_ts=geo_ds.TRUELAT1, |
108 | | - a=a, b=b) |
109 | | - stere = pyproj.Proj(proj='stere', lat_0=geo_ds.TRUELAT1, lon_0=geo_ds.STAND_LON, |
110 | | - lat_ts=geo_ds.TRUELAT1, a=a, b=b) |
111 | | - latlon = pyproj.Proj(proj='longlat', lon_0=geo_ds.STAND_LON, a=a, b=b) |
| 103 | + lcc = pyproj.Proj( |
| 104 | + proj="lcc", |
| 105 | + lat_1=geo_ds.TRUELAT1, |
| 106 | + lat_2=geo_ds.TRUELAT2, |
| 107 | + lat_0=geo_ds.MOAD_CEN_LAT, |
| 108 | + lon_0=geo_ds.STAND_LON, |
| 109 | + a=a, |
| 110 | + b=b, |
| 111 | + ) |
| 112 | + merc = pyproj.Proj( |
| 113 | + proj="merc", lon_0=geo_ds.STAND_LON, lat_ts=geo_ds.TRUELAT1, a=a, b=b |
| 114 | + ) |
| 115 | + stere = pyproj.Proj( |
| 116 | + proj="stere", |
| 117 | + lat_0=geo_ds.TRUELAT1, |
| 118 | + lon_0=geo_ds.STAND_LON, |
| 119 | + lat_ts=geo_ds.TRUELAT1, |
| 120 | + a=a, |
| 121 | + b=b, |
| 122 | + ) |
| 123 | + latlon = pyproj.Proj(proj="longlat", lon_0=geo_ds.STAND_LON, a=a, b=b) |
112 | 124 |
|
113 | 125 | proj_codes = { |
114 | | - 1: lcc, # lambert |
115 | | - 2: stere, # polar |
116 | | - 3: merc, # merc |
117 | | - 6: latlon # latlon |
| 126 | + 1: lcc, # lambert |
| 127 | + 2: stere, # polar |
| 128 | + 3: merc, # merc |
| 129 | + 6: latlon, # latlon |
118 | 130 | } |
119 | | - |
| 131 | + |
120 | 132 | wrf_proj = proj_codes[geo_ds.MAP_PROJ] |
121 | 133 | wrf_crs = pyproj.CRS.from_proj4(str(wrf_proj)) |
122 | 134 | return wrf_crs |
|
0 commit comments