kdedge 0.0.1__py3-none-any.whl

kdedge/__init__.py

@@ -0,0 +1,43 @@
+"""
+Edge bundling algorithms for visualizing graphs.
+
+"""
+
+__version__: str = "0.0.1"
+__author__: str = "cuzi"
+__email__: str = "cuzi@openmail.cc"
+__source__: str = "https://github.com/cvzi/kdedge"
+__license__: str = """
+MIT License
+
+Copyright (c) cuzi 2025
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+"""
+
+__all__ = [
+    "bundle", 
+    "exponential_schedule",
+    "kdeeb",
+    "linear_schedule",
+    "list_backends"
+    ]
+
+from .algo import bundle, exponential_schedule, kdeeb, linear_schedule
+from .impl.registry import list_backends

kdedge/_shared.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+from typing import Literal, TypeAlias
+
+import numpy as np
+from numpy.typing import ArrayLike, NDArray
+
+__all__ = [
+    "ArraySplatKernel",
+    "FloatArray",
+    "GridShape",
+    "SplatKernel",
+    "SplatKernelArrayFn",
+    "SplatKernelFn",
+    "as_float_array",
+]
+
+EPS = 1e-12
+
+FloatArray: TypeAlias = NDArray[np.floating]
+GridShape: TypeAlias = tuple[int, int]
+SplatKernelFn: TypeAlias = Callable[[float, float], float]
+SplatKernelArrayFn: TypeAlias = Callable[[FloatArray, float], FloatArray]
+AdvectEdgesFn: TypeAlias = Callable[
+    [
+        list[FloatArray], # polylines
+        FloatArray, # own_field
+        FloatArray | None, # node_grad
+        float, # attract
+        float, # node_repel
+        float, # smooth_lambda
+        int, # smooth_iterations
+        int, # h
+        int, # w
+    ],
+    list[FloatArray],
+]
+
+
+@dataclass(frozen=True)
+class ArraySplatKernel:
+    fn: SplatKernelArrayFn
+
+SplatKernel: TypeAlias = (
+    Literal["linear", "cone", "gaussian"] | SplatKernelFn | ArraySplatKernel | None
+)
+
+def as_float_array(values: ArrayLike | float | int, *,
+                   copy: bool = False) -> FloatArray:
+    if copy:
+        return np.array(values, dtype=np.float64, copy=True)
+    return np.asarray(values, dtype=np.float64)