tnfr 4.5.1__py3-none-any.whl → 6.0.0__py3-none-any.whl

tnfr/constants_glyphs.pyi

@@ -0,0 +1,12 @@
+from typing import Any
+
+__all__: Any
+
+def __getattr__(name: str) -> Any: ...
+
+ANGLE_MAP: Any
+DISRUPTORS: Any
+STABILIZERS: Any
+GLYPHS_CANONICAL: Any
+GLYPHS_CANONICAL_SET: Any
+GLYPH_GROUPS: Any

tnfr/dynamics/__init__.py

@@ -0,0 +1,136 @@
+"""Facade for TNFR dynamics submodules."""
+
+from __future__ import annotations
+
+from concurrent.futures import ProcessPoolExecutor
+
+from . import coordination, dnfr, integrators
+from .adaptation import adapt_vf_by_coherence
+from .aliases import (
+    ALIAS_D2EPI,
+    ALIAS_DNFR,
+    ALIAS_DSI,
+    ALIAS_EPI,
+    ALIAS_SI,
+    ALIAS_VF,
+)
+from .coordination import coordinate_global_local_phase
+from .dnfr import (
+    _compute_dnfr,
+    _compute_neighbor_means,
+    _init_dnfr_cache,
+    _prepare_dnfr_data,
+    _refresh_dnfr_vectors,
+    default_compute_delta_nfr,
+    dnfr_epi_vf_mixed,
+    dnfr_laplacian,
+    dnfr_phase_only,
+    set_delta_nfr_hook,
+)
+from .integrators import (
+    AbstractIntegrator,
+    DefaultIntegrator,
+    prepare_integration_params,
+    update_epi_via_nodal_equation,
+)
+from .runtime import (
+    _maybe_remesh,
+    _normalize_job_overrides,
+    _prepare_dnfr,
+    _resolve_jobs_override,
+    _run_after_callbacks,
+    _run_before_callbacks,
+    _run_validators,
+    _update_epi_hist,
+    _update_nodes,
+    apply_canonical_clamps,
+    run,
+    step,
+    validate_canon,
+)
+from .sampling import update_node_sample as _update_node_sample
+from .selectors import (
+    AbstractSelector,
+    DefaultGlyphSelector,
+    GlyphCode,
+    ParametricGlyphSelector,
+    _SelectorPreselection,
+    _apply_glyphs,
+    _apply_selector,
+    _choose_glyph,
+    _collect_selector_metrics,
+    _configure_selector_weights,
+    _prepare_selector_preselection,
+    _resolve_preselected_glyph,
+    _selector_parallel_jobs,
+    default_glyph_selector,
+    parametric_glyph_selector,
+)
+from ..operators import apply_glyph
+from ..metrics.sense_index import compute_Si
+from ..utils import get_numpy
+from ..validation.grammar import enforce_canonical_grammar, on_applied_glyph
+
+__all__ = (
+    "coordination",
+    "dnfr",
+    "integrators",
+    "ALIAS_D2EPI",
+    "ALIAS_DNFR",
+    "ALIAS_DSI",
+    "ALIAS_EPI",
+    "ALIAS_SI",
+    "ALIAS_VF",
+    "AbstractSelector",
+    "DefaultGlyphSelector",
+    "ParametricGlyphSelector",
+    "GlyphCode",
+    "_SelectorPreselection",
+    "_apply_glyphs",
+    "_apply_selector",
+    "_choose_glyph",
+    "_collect_selector_metrics",
+    "_configure_selector_weights",
+    "ProcessPoolExecutor",
+    "_maybe_remesh",
+    "_normalize_job_overrides",
+    "_prepare_dnfr",
+    "_prepare_dnfr_data",
+    "_prepare_selector_preselection",
+    "_resolve_jobs_override",
+    "_resolve_preselected_glyph",
+    "_run_after_callbacks",
+    "_run_before_callbacks",
+    "_run_validators",
+    "_selector_parallel_jobs",
+    "_update_epi_hist",
+    "_update_node_sample",
+    "_update_nodes",
+    "_compute_dnfr",
+    "_compute_neighbor_means",
+    "_init_dnfr_cache",
+    "_refresh_dnfr_vectors",
+    "adapt_vf_by_coherence",
+    "apply_canonical_clamps",
+    "coordinate_global_local_phase",
+    "compute_Si",
+    "default_compute_delta_nfr",
+    "default_glyph_selector",
+    "dnfr_epi_vf_mixed",
+    "dnfr_laplacian",
+    "dnfr_phase_only",
+    "enforce_canonical_grammar",
+    "get_numpy",
+    "on_applied_glyph",
+    "apply_glyph",
+    "parametric_glyph_selector",
+    "AbstractIntegrator",
+    "DefaultIntegrator",
+    "prepare_integration_params",
+    "run",
+    "set_delta_nfr_hook",
+    "step",
+    "update_epi_via_nodal_equation",
+    "validate_canon",
+)
+

tnfr/dynamics/__init__.pyi

@@ -0,0 +1,83 @@
+from typing import Any, Literal, Sequence
+
+from tnfr.types import TNFRGraph
+
+__all__: tuple[str, ...]
+
+dnfr: Any
+integrators: Any
+
+ALIAS_D2EPI: Sequence[str]
+ALIAS_DNFR: Sequence[str]
+ALIAS_DSI: Sequence[str]
+ALIAS_EPI: Sequence[str]
+ALIAS_SI: Sequence[str]
+ALIAS_VF: Sequence[str]
+
+AbstractSelector: Any
+DefaultGlyphSelector: Any
+GlyphCode: Any
+ParametricGlyphSelector: Any
+_SelectorPreselection: Any
+_apply_glyphs: Any
+_apply_selector: Any
+_choose_glyph: Any
+_configure_selector_weights: Any
+ProcessPoolExecutor: Any
+_maybe_remesh: Any
+_normalize_job_overrides: Any
+_prepare_dnfr: Any
+_prepare_dnfr_data: Any
+_prepare_selector_preselection: Any
+_resolve_jobs_override: Any
+_resolve_preselected_glyph: Any
+_run_after_callbacks: Any
+_run_before_callbacks: Any
+_run_validators: Any
+_selector_parallel_jobs: Any
+_update_epi_hist: Any
+_update_node_sample: Any
+_update_nodes: Any
+_compute_dnfr: Any
+_compute_neighbor_means: Any
+_init_dnfr_cache: Any
+_refresh_dnfr_vectors: Any
+adapt_vf_by_coherence: Any
+apply_canonical_clamps: Any
+coordinate_global_local_phase: Any
+default_compute_delta_nfr: Any
+default_glyph_selector: Any
+dnfr_epi_vf_mixed: Any
+dnfr_laplacian: Any
+dnfr_phase_only: Any
+enforce_canonical_grammar: Any
+get_numpy: Any
+on_applied_glyph: Any
+apply_glyph: Any
+parametric_glyph_selector: Any
+
+AbstractIntegrator: Any
+DefaultIntegrator: Any
+
+def prepare_integration_params(
+    G: TNFRGraph,
+    dt: float | None = ...,
+    t: float | None = ...,
+    method: Literal["euler", "rk4"] | None = ...,
+) -> tuple[float, int, float, Literal["euler", "rk4"]]: ...
+
+run: Any
+set_delta_nfr_hook: Any
+step: Any
+
+def update_epi_via_nodal_equation(
+    G: TNFRGraph,
+    *,
+    dt: float | None = ...,
+    t: float | None = ...,
+    method: Literal["euler", "rk4"] | None = ...,
+    n_jobs: int | None = ...,
+) -> None: ...
+
+validate_canon: Any
+

tnfr/dynamics/adaptation.py

@@ -0,0 +1,201 @@
+"""νf adaptation routines for TNFR dynamics."""
+
+from __future__ import annotations
+
+import math
+from concurrent.futures import ProcessPoolExecutor
+from typing import Any, cast
+
+from ..alias import collect_attr, set_vf
+from ..constants import get_graph_param
+from ..helpers.numeric import clamp
+from ..metrics.common import ensure_neighbors_map
+from ..types import CoherenceMetric, DeltaNFR, NodeId, TNFRGraph
+from ..utils import get_numpy
+from .aliases import ALIAS_DNFR, ALIAS_SI, ALIAS_VF
+
+__all__ = ("adapt_vf_by_coherence",)
+
+
+def _vf_adapt_chunk(
+    args: tuple[list[tuple[Any, int, tuple[int, ...]]], tuple[float, ...], float]
+) -> list[tuple[Any, float]]:
+    """Return proposed νf updates for ``chunk`` of stable nodes."""
+
+    chunk, vf_values, mu = args
+    updates: list[tuple[Any, float]] = []
+    for node, idx, neighbor_idx in chunk:
+        vf = vf_values[idx]
+        if neighbor_idx:
+            mean = math.fsum(vf_values[j] for j in neighbor_idx) / len(neighbor_idx)
+        else:
+            mean = vf
+        updates.append((node, vf + mu * (mean - vf)))
+    return updates
+
+
+def adapt_vf_by_coherence(G: TNFRGraph, n_jobs: int | None = None) -> None:
+    """Adjust νf toward neighbour mean in nodes with sustained stability."""
+
+    tau = get_graph_param(G, "VF_ADAPT_TAU", int)
+    mu = float(get_graph_param(G, "VF_ADAPT_MU"))
+    eps_dnfr = cast(DeltaNFR, get_graph_param(G, "EPS_DNFR_STABLE"))
+    thr_sel = get_graph_param(G, "SELECTOR_THRESHOLDS", dict)
+    thr_def = get_graph_param(G, "GLYPH_THRESHOLDS", dict)
+    si_hi = cast(
+        CoherenceMetric,
+        float(thr_sel.get("si_hi", thr_def.get("hi", 0.66))),
+    )
+    vf_min = float(get_graph_param(G, "VF_MIN"))
+    vf_max = float(get_graph_param(G, "VF_MAX"))
+
+    nodes = list(G.nodes)
+    if not nodes:
+        return
+
+    neighbors_map = ensure_neighbors_map(G)
+    node_count = len(nodes)
+    node_index = {node: idx for idx, node in enumerate(nodes)}
+
+    jobs: int | None
+    if n_jobs is None:
+        jobs = None
+    else:
+        try:
+            jobs = int(n_jobs)
+        except (TypeError, ValueError):
+            jobs = None
+        else:
+            if jobs <= 1:
+                jobs = None
+
+    np_mod = get_numpy()
+    use_np = np_mod is not None
+
+    si_values = collect_attr(G, nodes, ALIAS_SI, 0.0, np=np_mod if use_np else None)
+    dnfr_values = collect_attr(G, nodes, ALIAS_DNFR, 0.0, np=np_mod if use_np else None)
+    vf_values = collect_attr(G, nodes, ALIAS_VF, 0.0, np=np_mod if use_np else None)
+
+    if use_np:
+        np = np_mod  # type: ignore[assignment]
+        assert np is not None
+        si_arr = si_values.astype(float, copy=False)
+        dnfr_arr = np.abs(dnfr_values.astype(float, copy=False))
+        vf_arr = vf_values.astype(float, copy=False)
+
+        prev_counts = np.fromiter(
+            (int(G.nodes[node].get("stable_count", 0)) for node in nodes),
+            dtype=int,
+            count=node_count,
+        )
+        stable_mask = (si_arr >= si_hi) & (dnfr_arr <= eps_dnfr)
+        new_counts = np.where(stable_mask, prev_counts + 1, 0)
+
+        for node, count in zip(nodes, new_counts.tolist()):
+            G.nodes[node]["stable_count"] = int(count)
+
+        eligible_mask = new_counts >= tau
+        if not bool(eligible_mask.any()):
+            return
+
+        max_degree = 0
+        if node_count:
+            degree_counts = np.fromiter(
+                (len(neighbors_map.get(node, ())) for node in nodes),
+                dtype=int,
+                count=node_count,
+            )
+            if degree_counts.size:
+                max_degree = int(degree_counts.max())
+
+        if max_degree > 0:
+            neighbor_indices = np.zeros((node_count, max_degree), dtype=int)
+            mask = np.zeros((node_count, max_degree), dtype=bool)
+            for idx, node in enumerate(nodes):
+                neigh = neighbors_map.get(node, ())
+                if not neigh:
+                    continue
+                idxs = [node_index[nbr] for nbr in neigh if nbr in node_index]
+                if not idxs:
+                    continue
+                length = len(idxs)
+                neighbor_indices[idx, :length] = idxs
+                mask[idx, :length] = True
+            neighbor_values = vf_arr[neighbor_indices]
+            sums = (neighbor_values * mask).sum(axis=1)
+            counts = mask.sum(axis=1)
+            neighbor_means = np.where(counts > 0, sums / counts, vf_arr)
+        else:
+            neighbor_means = vf_arr
+
+        vf_updates = vf_arr + mu * (neighbor_means - vf_arr)
+        for idx in np.nonzero(eligible_mask)[0]:
+            node = nodes[int(idx)]
+            vf_new = clamp(float(vf_updates[int(idx)]), vf_min, vf_max)
+            set_vf(G, node, vf_new)
+        return
+
+    si_list = [float(val) for val in si_values]
+    dnfr_list = [abs(float(val)) for val in dnfr_values]
+    vf_list = [float(val) for val in vf_values]
+
+    prev_counts = [int(G.nodes[node].get("stable_count", 0)) for node in nodes]
+    stable_flags = [
+        si >= si_hi and dnfr <= eps_dnfr
+        for si, dnfr in zip(si_list, dnfr_list)
+    ]
+    new_counts = [prev + 1 if flag else 0 for prev, flag in zip(prev_counts, stable_flags)]
+
+    for node, count in zip(nodes, new_counts):
+        G.nodes[node]["stable_count"] = int(count)
+
+    eligible_nodes = [node for node, count in zip(nodes, new_counts) if count >= tau]
+    if not eligible_nodes:
+        return
+
+    if jobs is None:
+        for node in eligible_nodes:
+            idx = node_index[node]
+            neigh_indices = [
+                node_index[nbr]
+                for nbr in neighbors_map.get(node, ())
+                if nbr in node_index
+            ]
+            if neigh_indices:
+                total = math.fsum(vf_list[i] for i in neigh_indices)
+                mean = total / len(neigh_indices)
+            else:
+                mean = vf_list[idx]
+            vf_new = vf_list[idx] + mu * (mean - vf_list[idx])
+            set_vf(G, node, clamp(float(vf_new), vf_min, vf_max))
+        return
+
+    work_items: list[tuple[Any, int, tuple[int, ...]]] = []
+    for node in eligible_nodes:
+        idx = node_index[node]
+        neigh_indices = tuple(
+            node_index[nbr]
+            for nbr in neighbors_map.get(node, ())
+            if nbr in node_index
+        )
+        work_items.append((node, idx, neigh_indices))
+
+    chunk_size = max(1, math.ceil(len(work_items) / jobs))
+    chunks = [
+        work_items[i : i + chunk_size]
+        for i in range(0, len(work_items), chunk_size)
+    ]
+    vf_tuple = tuple(vf_list)
+    updates: dict[Any, float] = {}
+    with ProcessPoolExecutor(max_workers=jobs) as executor:
+        args = ((chunk, vf_tuple, mu) for chunk in chunks)
+        for chunk_updates in executor.map(_vf_adapt_chunk, args):
+            for node, value in chunk_updates:
+                updates[node] = float(value)
+
+    for node in eligible_nodes:
+        vf_new = updates.get(node)
+        if vf_new is None:
+            continue
+        set_vf(G, node, clamp(float(vf_new), vf_min, vf_max))
+

tnfr/dynamics/aliases.py

@@ -0,0 +1,22 @@
+"""Shared alias tokens used across TNFR dynamics submodules."""
+
+from __future__ import annotations
+
+from ..constants import get_aliases
+
+ALIAS_VF = get_aliases("VF")
+ALIAS_DNFR = get_aliases("DNFR")
+ALIAS_EPI = get_aliases("EPI")
+ALIAS_SI = get_aliases("SI")
+ALIAS_D2EPI = get_aliases("D2EPI")
+ALIAS_DSI = get_aliases("DSI")
+
+__all__ = (
+    "ALIAS_VF",
+    "ALIAS_DNFR",
+    "ALIAS_EPI",
+    "ALIAS_SI",
+    "ALIAS_D2EPI",
+    "ALIAS_DSI",
+)
+