mirror of
https://github.com/openmc-dev/openmc.git
synced 2026-07-22 15:05:28 -04:00
610a5d8c34
Co-authored-by: Jonathan Shimwell <drshimwell@gmail.com> Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
"""
|
|
This module provides bindings to C/C++ functions defined by OpenMC shared
|
|
library. When the :mod:`openmc.lib` package is imported, the OpenMC shared
|
|
library is automatically loaded. Calls to the OpenMC library can then be via
|
|
functions or objects in :mod:`openmc.lib`, for example:
|
|
|
|
.. code-block:: python
|
|
|
|
openmc.lib.init()
|
|
openmc.lib.run()
|
|
openmc.lib.finalize()
|
|
|
|
"""
|
|
|
|
from ctypes import CDLL, c_bool, c_int
|
|
import os
|
|
import sys
|
|
|
|
import pkg_resources
|
|
|
|
|
|
# Determine shared-library suffix
|
|
if sys.platform == 'darwin':
|
|
_suffix = 'dylib'
|
|
else:
|
|
_suffix = 'so'
|
|
|
|
if os.environ.get('READTHEDOCS', None) != 'True':
|
|
# Open shared library
|
|
_filename = pkg_resources.resource_filename(
|
|
__name__, 'libopenmc.{}'.format(_suffix))
|
|
_dll = CDLL(_filename)
|
|
else:
|
|
# For documentation builds, we don't actually have the shared library
|
|
# available. Instead, we create a mock object so that when the modules
|
|
# within the openmc.lib package try to configure arguments and return
|
|
# values for symbols, no errors occur
|
|
from unittest.mock import Mock
|
|
_dll = Mock()
|
|
|
|
|
|
def _dagmc_enabled():
|
|
return c_bool.in_dll(_dll, "DAGMC_ENABLED").value
|
|
|
|
def _ncrystal_enabled():
|
|
return c_bool.in_dll(_dll, "NCRYSTAL_ENABLED").value
|
|
|
|
def _coord_levels():
|
|
return c_int.in_dll(_dll, "n_coord_levels").value
|
|
|
|
def _libmesh_enabled():
|
|
return c_bool.in_dll(_dll, "LIBMESH_ENABLED").value
|
|
|
|
def _mcpl_enabled():
|
|
return c_bool.in_dll(_dll, "MCPL_ENABLED").value
|
|
|
|
|
|
from .error import *
|
|
from .core import *
|
|
from .nuclide import *
|
|
from .material import *
|
|
from .cell import *
|
|
from .mesh import *
|
|
from .filter import *
|
|
from .tally import *
|
|
from .settings import settings
|
|
from .math import *
|
|
from .plot import *
|
|
from .weight_windows import *
|
|
|
|
# Flag to denote whether or not openmc.lib.init has been called
|
|
# TODO: Establish and use a flag in the C++ code to represent the status of the
|
|
# openmc_init and openmc_finalize methods
|
|
is_initialized = False
|