mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-21 14:35:27 -04:00
5ad1a4aa1e
Co-authored-by: Baptiste Mouginot <bam.git.not.reply@gmail.com> Co-authored-by: Patrick Shriwise <pshriwise@gmail.com> Co-authored-by: azimG <azimgivron@yahoo.com> Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from ctypes import c_int, c_int32, POINTER, c_size_t
|
|
|
|
import numpy as np
|
|
|
|
from . import _dll
|
|
from .error import _error_handler
|
|
|
|
|
|
__all__ = [
|
|
'dagmc_universe_cell_ids'
|
|
]
|
|
|
|
# DAGMC functions
|
|
_dll.openmc_dagmc_universe_get_cell_ids.argtypes = [c_int32, POINTER(c_int32), POINTER(c_size_t)]
|
|
_dll.openmc_dagmc_universe_get_cell_ids.restype = c_int
|
|
_dll.openmc_dagmc_universe_get_cell_ids.errcheck = _error_handler
|
|
_dll.openmc_dagmc_universe_get_num_cells.argtypes = [c_int32, POINTER(c_size_t)]
|
|
_dll.openmc_dagmc_universe_get_num_cells.restype = c_int
|
|
_dll.openmc_dagmc_universe_get_num_cells.errcheck = _error_handler
|
|
|
|
|
|
def dagmc_universe_cell_ids(universe_id: int) -> np.ndarray:
|
|
"""Return an array of cell IDs for a DAGMC universe.
|
|
|
|
Parameters
|
|
----------
|
|
dagmc_id : int
|
|
ID of the DAGMC universe to get cell IDs from.
|
|
|
|
Returns
|
|
-------
|
|
numpy.ndarray
|
|
DAGMC cell IDs for the universe.
|
|
|
|
"""
|
|
n = c_size_t()
|
|
_dll.openmc_dagmc_universe_get_num_cells(universe_id, n)
|
|
cell_ids = np.empty(n.value, dtype=np.int32)
|
|
|
|
_dll.openmc_dagmc_universe_get_cell_ids(
|
|
universe_id, cell_ids.ctypes.data_as(POINTER(c_int32)), n
|
|
)
|
|
return cell_ids
|