fastlisaresponse 1.1.4__cp312-cp312-manylinux_2_24_aarch64.manylinux_2_28_aarch64.whl

fastlisaresponse/__init__.py

@@ -0,0 +1,63 @@
+"""FastLISAResponse."""
+
+# ruff: noqa: E402
+try:
+    from fastlisaresponse._version import (  # pylint: disable=E0401,E0611
+        __version__,
+        __version_tuple__,
+    )
+
+except ModuleNotFoundError:
+    from importlib.metadata import PackageNotFoundError, version  # pragma: no cover
+
+    try:
+        __version__ = version(__name__)
+        __version_tuple__ = tuple(__version__.split("."))
+    except PackageNotFoundError:  # pragma: no cover
+        __version__ = "unknown"
+        __version_tuple__ = (0, 0, 0, "unknown")
+    finally:
+        del version, PackageNotFoundError
+
+_is_editable: bool
+try:
+    from . import _editable
+
+    _is_editable = True
+    del _editable
+except (ModuleNotFoundError, ImportError):
+    _is_editable = False
+
+from . import cutils, utils
+from .cutils import KNOWN_BACKENDS
+
+from gpubackendtools import Globals
+from .cutils import FastLISAResponseCpuBackend, FastLISAResponseCuda11xBackend, FastLISAResponseCuda12xBackend
+
+
+add_backends = {
+    "fastlisaresponse_cpu": FastLISAResponseCpuBackend,
+    "fastlisaresponse_cuda11x": FastLISAResponseCuda11xBackend,
+    "fastlisaresponse_cuda12x": FastLISAResponseCuda12xBackend,
+}
+
+Globals().backends_manager.add_backends(add_backends)
+
+
+from gpubackendtools import get_backend, has_backend, get_first_backend
+
+from .response import pyResponseTDI, ResponseWrapper
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "_is_editable",
+    "pyResponseTDI",
+    "ResponseWrapper",
+    "get_logger",
+    "get_config",
+    "get_config_setter",
+    "get_backend",
+    "get_file_manager",
+    "has_backend",
+]

fastlisaresponse/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '1.1.4'
+__version_tuple__ = version_tuple = (1, 1, 4)
+
+__commit_id__ = commit_id = None

fastlisaresponse/cutils/__init__.py

@@ -0,0 +1,136 @@
+from __future__ import annotations
+import dataclasses
+import enum
+import types
+import typing
+import abc
+from typing import Optional, Sequence, TypeVar, Union
+from ..utils.exceptions import *
+
+from gpubackendtools.gpubackendtools import BackendMethods, CpuBackend, Cuda11xBackend, Cuda12xBackend
+from gpubackendtools.exceptions import *
+
+@dataclasses.dataclass
+class FastLISAResponseBackendMethods(BackendMethods):
+    get_response_wrap: typing.Callable[(...), None]
+    get_tdi_delays_wrap: typing.Callable[(...), None]
+
+
+class FastLISAResponseBackend:
+    get_response_wrap: typing.Callable[(...), None]
+    get_tdi_delays_wrap: typing.Callable[(...), None]
+
+    def __init__(self, fastlisaresponse_backend_methods):
+
+        # set direct fastlisaresponse methods
+        # pass rest to general backend
+        assert isinstance(fastlisaresponse_backend_methods, FastLISAResponseBackendMethods)
+
+        self.get_response_wrap = fastlisaresponse_backend_methods.get_response_wrap
+        self.get_tdi_delays_wrap = fastlisaresponse_backend_methods.get_tdi_delays_wrap
+
+
+class FastLISAResponseCpuBackend(CpuBackend, FastLISAResponseBackend):
+    """Implementation of the CPU backend"""
+    
+    _backend_name = "fastlisaresponse_backend_cpu"
+    _name = "fastlisaresponse_cpu"
+    def __init__(self, *args, **kwargs):
+        CpuBackend.__init__(self, *args, **kwargs)
+        FastLISAResponseBackend.__init__(self, self.cpu_methods_loader())
+
+    @staticmethod
+    def cpu_methods_loader() -> FastLISAResponseBackendMethods:
+        try:
+            import fastlisaresponse_backend_cpu.responselisa
+            
+        except (ModuleNotFoundError, ImportError) as e:
+            raise BackendUnavailableException(
+                "'cpu' backend could not be imported."
+            ) from e
+
+        numpy = FastLISAResponseCpuBackend.check_numpy()
+
+        return FastLISAResponseBackendMethods(
+            get_response_wrap=fastlisaresponse_backend_cpu.responselisa.get_response_wrap,
+            get_tdi_delays_wrap=fastlisaresponse_backend_cpu.responselisa.get_tdi_delays_wrap,
+            xp=numpy,
+        )
+
+
+class FastLISAResponseCuda11xBackend(Cuda11xBackend, FastLISAResponseBackend):
+
+    """Implementation of CUDA 11.x backend"""
+    _backend_name : str = "fastlisaresponse_backend_cuda11x"
+    _name = "fastlisaresponse_cuda11x"
+
+    def __init__(self, *args, **kwargs):
+        Cuda11xBackend.__init__(self, *args, **kwargs)
+        FastLISAResponseBackend.__init__(self, self.cuda11x_module_loader())
+        
+    @staticmethod
+    def cuda11x_module_loader():
+        try:
+            import fastlisaresponse_backend_cuda11x.responselisa
+
+        except (ModuleNotFoundError, ImportError) as e:
+            raise BackendUnavailableException(
+                "'cuda11x' backend could not be imported."
+            ) from e
+
+        try:
+            import cupy
+        except (ModuleNotFoundError, ImportError) as e:
+            raise MissingDependencies(
+                "'cuda11x' backend requires cupy", pip_deps=["cupy-cuda11x"]
+            ) from e
+
+        return FastLISAResponseBackendMethods(
+            get_response_wrap=fastlisaresponse_backend_cuda11x.responselisa.get_response_wrap,
+            get_tdi_delays_wrap=fastlisaresponse_backend_cuda11x.responselisa.get_tdi_delays_wrap,
+            xp=cupy,
+        )
+
+class FastLISAResponseCuda12xBackend(Cuda12xBackend, FastLISAResponseBackend):
+    """Implementation of CUDA 12.x backend"""
+    _backend_name : str = "fastlisaresponse_backend_cuda12x"
+    _name = "fastlisaresponse_cuda12x"
+    
+    def __init__(self, *args, **kwargs):
+        Cuda12xBackend.__init__(self, *args, **kwargs)
+        FastLISAResponseBackend.__init__(self, self.cuda12x_module_loader())
+        
+    @staticmethod
+    def cuda12x_module_loader():
+        try:
+            import fastlisaresponse_backend_cuda12x.responselisa
+
+        except (ModuleNotFoundError, ImportError) as e:
+            raise BackendUnavailableException(
+                "'cuda12x' backend could not be imported."
+            ) from e
+
+        try:
+            import cupy
+        except (ModuleNotFoundError, ImportError) as e:
+            raise MissingDependencies(
+                "'cuda12x' backend requires cupy", pip_deps=["cupy-cuda12x"]
+            ) from e
+
+        return FastLISAResponseBackendMethods(
+            get_response_wrap=fastlisaresponse_backend_cuda12x.responselisa.get_response_wrap,
+            get_tdi_delays_wrap=fastlisaresponse_backend_cuda12x.responselisa.get_tdi_delays_wrap,
+            xp=cupy,
+        )
+
+
+KNOWN_BACKENDS = {
+    "cuda12x": FastLISAResponseCuda12xBackend,
+    "cuda11x": FastLISAResponseCuda11xBackend,
+    "cpu": FastLISAResponseCpuBackend,
+}
+
+"""List of existing backends, per default order of preference."""
+# TODO: __all__ ?
+
+

fastlisaresponse/git_version.py

@@ -0,0 +1,7 @@
+"""Metadata deduced from git at build time."""
+
+id: str
+short_id: str
+
+id = "fc57dfe9046150a7cfc185abdf4c675e76fd316e"
+short_id = "fc57dfe"

fastlisaresponse/git_version.py.in

@@ -0,0 +1,7 @@
+"""Metadata deduced from git at build time."""
+
+id: str
+short_id: str
+
+id = "@FASTLISA_GIT_COMMIT_ID@"
+short_id = "@FASTLISA_GIT_COMMIT_SHORT_ID@"