musica 0.14.4__cp310-cp310-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl

musica/__init__.py

@@ -0,0 +1,11 @@
+"""
+MUSICA: A Python library for atmospheric chemistry simulations.
+"""
+
+from ._version import version as __version__
+from .micm import MICM, SolverType, State, Conditions
+from .tuvx import TUVX, GridMap, Grid, ProfileMap, Profile, RadiatorMap, Radiator
+from . import mechanism_configuration
+from . import carma
+from . import cuda
+from .examples import Examples

musica/_version.py

@@ -0,0 +1 @@
+version = "0.14.4"

musica/backend.py

@@ -0,0 +1,58 @@
+"""
+Backend selection and symbol loading for MUSICA.
+"""
+import importlib.util
+
+
+def _safe_find_spec(name):
+    try:
+        return importlib.util.find_spec(name)
+    except ModuleNotFoundError:
+        return None
+
+
+def _gpu_deps_installed():
+    return (
+        _safe_find_spec("nvidia.cublas") is not None or
+        _safe_find_spec("nvidia_cuda_runtime") is not None or
+        _safe_find_spec("nvidia-cublas-cu12") is not None or
+        _safe_find_spec("nvidia-cuda-runtime-cu12") is not None
+    )
+
+
+# Global backend instance to ensure it's only loaded once
+_backend_instance = None
+
+
+def get_backend():
+    """Get the appropriate backend module."""
+
+    global _backend_instance
+
+    if _backend_instance is not None:
+        return _backend_instance
+
+    try:
+        if _gpu_deps_installed():
+            import musica._musica_gpu as backend
+        else:
+            import musica._musica as backend
+
+        _backend_instance = backend
+        return _backend_instance
+
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import MUSICA backend. Make sure the package is properly built and installed. Error: {e}")
+
+
+def tuvx_available():
+    """Check if the TUVX backend is available."""
+    _backend = get_backend()
+    return hasattr(_backend._tuvx, "_get_tuvx_version")
+
+
+def carma_available():
+    """Check if the CARMA backend is available."""
+    _backend = get_backend()
+    return hasattr(_backend._carma, "_get_carma_version")

musica/carma/__init__.py

@@ -0,0 +1,20 @@
+from .carma import (
+    # Main classes
+    CARMA, CARMAParameters, CARMAState, CARMASurfaceProperties,
+
+    # Configuration classes
+    CARMAGroupConfig, CARMAElementConfig, CARMASoluteConfig, CARMAGasConfig,
+    CARMAWavelengthBin, CARMACoagulationConfig, CARMAGrowthConfig,
+    CARMANucleationConfig, CARMAInitializationConfig,
+
+    # Enums
+    ParticleShape, ParticleType, ParticleSwellingAlgorithm,
+    ParticleSwellingComposition, ParticleFallVelocityAlgorithm,
+    MieCalculationAlgorithm, OpticsAlgorithm, VaporizationAlgorithm,
+    GasComposition, ParticleComposition, ParticleCollectionAlgorithm,
+    ParticleNucleationAlgorithm, SulfateNucleationMethod, CarmaCoordinates,
+)
+from .. import backend
+_backend = backend.get_backend()
+
+__version__ = _backend._carma._get_carma_version() if backend.carma_available() else None