cherab.imas.ggd.UnstructGrid3D

class cherab.imas.ggd.UnstructGrid3D(vertices: ArrayLike, cells: NDArray[integer[Any]] | Sequence[Sequence[SupportsIndex] | NDArray[integer[Any]]], name: str = 'Cells')Source

Bases: GGDGrid

Unstructured 3D grid object.

The grid cells are hexahedra (8-vertex cells). Vertices may be shared with neighbouring cells.

To use Raysect’s KDtree accelerator, each hexahedral cell is tetrahedralized.

Parameters:
vertices: ArrayLike

Array-like of shape (N, 3) containing vertex coordinates in (X, Y, Z).

cells: NDArray[integer[Any]] | Sequence[Sequence[SupportsIndex] | NDArray[integer[Any]]]

Array-like of shape (M, 8) containing vertex indices for each hexahedral cell.

name: str = 'Cells'

Name of the grid, by default 'Cells'.

Methods

interpolator(grid_data[, fill_value, ...])

Return an UnstructGridFunction3D interpolator instance for the data defined on this grid.

plot_mesh([data, ax])

Plot the grid geometry to a matplotlib figure.

subset(indices[, name])

Create a subset UnstructGrid3D from this instance.

vector_interpolator(grid_vectors[, ...])

Return an UnstructGridVectorFunction3D interpolator instance for the vector data defined on this grid.

Attributes

cell_area

Cell areas as (num_cell,) array.

cell_centre

Coordinate of cell centres as (num_cell, dimension) array.

cell_to_tetra_map

Array of shape (M, 2) mapping every grid cell index to tetrahedral IDs.

cell_volume

Cell volume as (num_cell,) array.

cells

Mesh cells as (M, 8) array.

coordinate_system

Coordinate system.

dimension

Grid dimension.

mesh_extent

Extent of the mesh.

name

Grid name.

num_cell

Number of grid cells.

tetra_to_cell_map

Array of shape (5M,) mapping every tetrahedral index to a grid cell ID.

tetrahedra

Mesh tetrahedra as (5M, 4) array.

vertices

Mesh vertex coordinates as (N, 3) array in the (X, Y, Z) space.

property vertices : NDArray[float64]Source

Mesh vertex coordinates as (N, 3) array in the (X, Y, Z) space.

property cells : NDArray[int32]Source

Mesh cells as (M, 8) array.

property tetrahedra : NDArray[int32]Source

Mesh tetrahedra as (5M, 4) array.

property tetra_to_cell_map : NDArray[int32]Source

Array of shape (5M,) mapping every tetrahedral index to a grid cell ID.

property cell_to_tetra_map : NDArray[int32]Source

Array of shape (M, 2) mapping every grid cell index to tetrahedral IDs.

The first column is the index of the first tetrahedron forming the cell. The second column is the number of tetrahedra forming the cell.

>>> itet, ntet = mesh.cell_to_tetra_map[icell]
>>> tet_cell = mesh.tetrahedra[itet : itet + ntet]
subset(indices: Sequence[SupportsIndex] | NDArray[integer[Any]], name: str | None = None) UnstructGrid3DSource

Create a subset UnstructGrid3D from this instance.

Parameters:
indices: Sequence[SupportsIndex] | NDArray[integer[Any]]

Indices of the cells of the original grid in the subset.

name: str | None = None

Name of the grid subset. Default is instance.name + " subset".

Returns:

UnstructGrid3D – Subset instance.

interpolator(grid_data: NDArray[floating[Any]] | Sequence[float], fill_value: float = 0, *, interpolator_cache: 'none' | 'memory' | 'disk' = 'memory', interpolator_cache_dir: str | Path | None = None, interpolator_cache_namespace: str = 'ggd') UnstructGridFunction3DSource

Return an UnstructGridFunction3D interpolator instance for the data defined on this grid.

On the second and subsequent calls, the interpolator is created as an instance of the previously created interpolator sharing the same KDtree structure.

Parameters:
grid_data: NDArray[floating[Any]] | Sequence[float]

Array containing data in the grid cells.

fill_value: float = 0

Value returned outside the grid, by default 0.0.

interpolator_cache: 'none' | 'memory' | 'disk' = 'memory'

Cache mode for the interpolator, by default "memory". The cache mode is described in the InterpolatorCacheMode type alias.

interpolator_cache_dir: str | Path | None = None

Directory used when interpolator_cache="disk", by default None (uses the system cache directory, e.g., ~/.cache/cherab/imas/interpolators).

interpolator_cache_namespace: str = 'ggd'

Namespace prefix to avoid cache-key collisions, by default "ggd".

Returns:

UnstructGridFunction3D – Interpolator instance.

vector_interpolator(grid_vectors: NDArray[float64], fill_vector: Vector3D = Vector3D(0.0, 0.0, 0.0), *, interpolator_cache: 'none' | 'memory' | 'disk' = 'memory', interpolator_cache_dir: str | Path | None = None, interpolator_cache_namespace: str = 'ggd') UnstructGridVectorFunction3DSource

Return an UnstructGridVectorFunction3D interpolator instance for the vector data defined on this grid.

On the second and subsequent calls, the interpolator is created as an instance of the previously created interpolator sharing the same KDtree structure.

Parameters:
grid_vectors: NDArray[float64]

(3, L) Array containing 3D vectors in the grid cells.

fill_vector: Vector3D = Vector3D(0.0, 0.0, 0.0)

3D vector returned outside the grid, by default Vector3D(0, 0, 0).

interpolator_cache: 'none' | 'memory' | 'disk' = 'memory'

Cache mode for the interpolator, by default "memory". The cache mode is described in the InterpolatorCacheMode type alias.

interpolator_cache_dir: str | Path | None = None

Directory used when interpolator_cache="disk", by default None (uses the system cache directory, e.g., ~/.cache/cherab/imas/interpolators).

interpolator_cache_namespace: str = 'ggd'

Namespace prefix to avoid cache-key collisions, by default "ggd".

Returns:

UnstructGridVectorFunction3D – Interpolator instance.

property cell_area : NDArray[float64]Source

Cell areas as (num_cell,) array.

property cell_centre : NDArray[float64]Source

Coordinate of cell centres as (num_cell, dimension) array.

property cell_volume : NDArray[float64]Source

Cell volume as (num_cell,) array.

property coordinate_system : strSource

Coordinate system.

property dimension : intSource

Grid dimension.

property mesh_extent : dict[str, float]Source

Extent of the mesh.

A dictionary with xmin, xmax, ymin and ymax, … keys.

property name : strSource

Grid name.

property num_cell : intSource

Number of grid cells.

plot_mesh(data: NDArray[floating[Any]] | Sequence[float] | None = None, ax: Axes | None = None, **grid_styles) AxesSource

Plot the grid geometry to a matplotlib figure.

Parameters:
data: NDArray[floating[Any]] | Sequence[float] | None = None

Data array defined on the grid.

ax: Axes | None = None

Matplotlib axes to plot on. If None, a new figure and axes are created.

**grid_styles

Styles for the grid lines and faces.