make_grid_info - Part 1 #79
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -3,6 +3,7 @@ | |
| import numpy as np | ||
| from numpy.typing import NDArray | ||
| import xarray as xr | ||
| import logging | ||
|
|
||
| from fmsgridtools.shared.gridtools_utils import get_provenance_attrs | ||
| from fmsgridtools.shared.gridobj import GridObj | ||
|
|
@@ -57,7 +58,46 @@ def fill_cubic_grid_halo( | |
| data[(nyp+1)*nxph+i] = data2_all[ln*nxp*nyp+(nxp-i)*nyp+joff] # north halo | ||
|
|
||
| class HGridObj(): | ||
| """ | ||
| Holds metadata for a computational grid tile. | ||
|
|
||
| Attributes: | ||
| tile (str): | ||
| x (np.ndarray): 1D array of float64 values specifying longitudes of grid cell corners. | ||
| y (np.ndarray): 1D array of float64 values specifying latitudes of grid cell corners. | ||
| dx (np.ndarray): 1D array of float64 values specifying distances between grid points in the x-direction. | ||
| dy (np.ndarray): 1D array of float64 values specifying distances between grid points in the y-direction. | ||
| area (np.ndarray): 1D array of float64 values specifying area of each grid cell. | ||
| angle_dx (np.ndarray): 1D array of float64 values specifying angles between the local x-direction and geographic east. | ||
| angle_dy (np.ndarray): 1D array of float64 values specifying angles between the local y-direction and geographic north. | ||
| arcx (str): Describes the arc (edge) types in the x-direction for each element along the given dimension. | ||
| nx (int): Number of grid points at cell centers in the x-direction. | ||
| ny (int): Number of grid points at cell centers in the y-direction. | ||
| nxp (int): Number of grid points at cell edges in the x-direction. | ||
| nyp (int): Number of grid points at cell edges in the y-direction. | ||
| nxl (np.ndarray): 1D array of int32 values specifying the number of x-direction grid points per tile. | ||
| nyl (np.ndarray): 1D array of int32 values specifying the number of y-direction grid points per tile. | ||
| isc (int): Starting index of the x-axis for the compute domain (excluding halos). | ||
| iec (int): Ending index of the x-axis for the compute domain (excluding halos). | ||
| jsc (int): Starting index of the y-axis for the compute domain (excluding halos). | ||
| jec (int): Ending index of the y-axis for the compute domain (excluding halos). | ||
| """ | ||
|
|
||
| def __init__(self, verbose:bool): | ||
|
|
||
| self.logger = logging.getLogger("HGridObj") | ||
| self.logger.setLevel(logging.DEBUG) | ||
| if not self.logger.handlers: | ||
| handler = logging.StreamHandler() | ||
| formatter = logging.Formatter('%(name)s - %(levelname)s - %(message)s') | ||
| handler.setFormatter(formatter) | ||
| self.logger.addHandler(handler) | ||
|
|
||
| if verbose: | ||
| self.logger.setLevel(logging.DEBUG) | ||
| else: | ||
| self.logger.setLevel(logging.WARNING) | ||
|
|
||
| self.tile = "" | ||
| self.x = None | ||
| self.y = None | ||
|
|
@@ -78,6 +118,48 @@ def __init__(self): | |
| self.jsc = None | ||
| self.jec = None | ||
|
|
||
| def __init_numpy_arrays__(self, nsuper, narea, ndx, ndy): | ||
|
Owner
There was a problem hiding this comment. do the arrays need to be initialized beforehand in python?
Collaborator
Author
There was a problem hiding this comment. ok init is a bad name, This is allocating memory for all of the 1D members of They are set in
Owner
There was a problem hiding this comment. I think the allocation can be abstracted into pyfrenctools.make_hgrid_wrappers.create_* (do the allocation there and return the appropriate arrays)
Collaborator
Author
There was a problem hiding this comment. The way I see this turning out is
Let's talk about this tomorrow :) |
||
|
|
||
| self.logger.debug(f"make_grid_info: allocating x and y to {nsuper}") | ||
| self.logger.debug(f"make_grid_info: allocating angle_dx and angle_dy {nsuper}") | ||
| self.logger.debug(f"make_grid_info: allocating dx to {ndx} and dy to {ndy}") | ||
|
|
||
| self.x = np.empty(shape=nsuper, dtype=np.float64) | ||
| self.y = np.empty(shape=nsuper, dtype=np.float64) | ||
| self.area = np.empty(shape=narea, dtype=np.float64) | ||
| self.dx = np.empty(shape=ndx, dtype=np.float64) | ||
| self.dy = np.empty(shape=ndy, dtype=np.float64) | ||
| self.angle_dx = np.empty(shape=nsuper, dtype=np.float64) | ||
| self.angle_dy = np.empty(shape=nsuper, dtype=np.float64) | ||
|
|
||
| def make_grid_info_lat_lon(self, nlon: int=None, nlat: int=None): | ||
|
|
||
| self.logger.debug(f"make_grid_info: Creating a lat-lon with nlon={nlon} and nlat={nlat}") | ||
|
|
||
| self.nxl = np.array(nlon, dtype=np.int32) | ||
|
Owner
There was a problem hiding this comment. is nxl, nyl just scalars? They probably can just be self.nxl = nlon...
Collaborator
Author
There was a problem hiding this comment. They are 1D integers of size equal to the number of tiles. In this lat/lon case they are just one. The code that writes the output is currently expecting them to be arrays. FMSgridtools/fmsgridtools/make_hgrid/hgridobj.py Lines 456 to 457 in b2cfa66
Owner
There was a problem hiding this comment. ooo.... that should be changed...
Collaborator
Author
There was a problem hiding this comment. Yes, there is a lot of unnecessarily complicated code. |
||
| self.nyl = np.array(nlat, dtype=np.int32) | ||
| self.arcx = "small_circle" | ||
|
|
||
| self.nx = self.nxl[0] | ||
| self.ny = self.nyl[0] | ||
| self.nxp = self.nx + 1 | ||
| self.nyp = self.ny + 1 | ||
|
|
||
| # Number of grid points including the edges and centers (super!) | ||
| nsuper_grid = self.nxp * self.nyp | ||
| narea = self.nx * self.ny | ||
| ndx = self.nx * self.nyp | ||
| ndy = self.nxp * self.ny | ||
|
|
||
| self.__init_numpy_arrays__(nsuper_grid, narea, ndx, ndy) | ||
|
|
||
| self.isc = 0 | ||
| self.iec = self.nx - 1 | ||
| self.jsc = 0 | ||
| self.jec = self.ny - 1 | ||
|
|
||
| self.logger.debug(f"make_grid_info: Finished allocating the hgrid_object") | ||
|
|
||
| def make_grid_info( | ||
| self, | ||
| nxbnds: int=None, | ||
|
|
@@ -99,11 +181,9 @@ def make_grid_info( | |
| output_length_angle: bool=True, | ||
| verbose: bool=False, | ||
| ): | ||
|
|
||
| self.logger.debug(f"make_grid_info: Number of total tiles - including nests (ntiles): {ntiles}") | ||
| self.logger.debug(f"make_grid_info: Number of total parent tiles (ntiles_global): {ntiles_global}") | ||
|
|
||
| self.nxl = np.empty(shape=ntiles, dtype=np.int32) | ||
| self.nyl = np.empty(shape=ntiles, dtype=np.int32) | ||
|
|
@@ -133,18 +213,6 @@ def make_grid_info( | |
| self.nxl[n] = (iend_nest[nn] - istart_nest[nn] + 1) * refine_ratio[nn] | ||
| self.nyl[n] = (jend_nest[nn] - jstart_nest[nn] + 1) * refine_ratio[nn] | ||
|
|
||
| self.nx = self.nxl[0] | ||
| self.ny = self.nyl[0] | ||
| self.nxp = self.nx + 1 | ||
|
|
@@ -160,21 +228,10 @@ def make_grid_info( | |
| for n_nest in range(ntiles): | ||
| print(f"[INFO] tile: {n_nest}, nxl[{self.nxl[n_nest]}], nyl[{self.nyl[n_nest]}], ntiles: {ntiles}\n") | ||
|
|
||
| size1 = ctypes.c_ulong(self.nxp * self.nyp * ntiles_global) | ||
| size2 = ctypes.c_ulong(self.nx * self.nyp * ntiles_global) | ||
| size3 = ctypes.c_ulong(self.nxp * self.ny * ntiles_global) | ||
| size4 = ctypes.c_ulong(self.nx * self.ny * ntiles_global) | ||
|
|
||
| if not (nest_grids == 1 and parent_tile[0] == 0): | ||
| for n_nest in range(ntiles_global, ntiles_global+nest_grids): | ||
|
|
||
Uh oh!
There was an error while loading. Please reload this page.