visiter-native 0.1.0__tar.gz

visiter_native-0.1.0/Cargo.lock

@@ -0,0 +1,180 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 4
+
+[[package]]
+name = "autocfg"
+version = "1.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9330f8b2ff13f34540b44e946ef35111825727b38d33286ef986142615121801"
+
+[[package]]
+name = "heck"
+version = "0.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
+
+[[package]]
+name = "indoc"
+version = "2.0.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "79cf5c93f93228cf8efb3ba362535fb11199ac548a09ce117c9b1adc3030d706"
+dependencies = [
+ "rustversion",
+]
+
+[[package]]
+name = "libc"
+version = "0.2.186"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68ab91017fe16c622486840e4c83c9a37afeff978bd239b5293d61ece587de66"
+
+[[package]]
+name = "memoffset"
+version = "0.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "488016bfae457b036d996092f6cb448677611ce4449e970ceaf42695203f218a"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "once_cell"
+version = "1.21.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9f7c3e4beb33f85d45ae3e3a1792185706c8e16d043238c593331cc7cd313b50"
+
+[[package]]
+name = "portable-atomic"
+version = "1.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c33a9471896f1c69cecef8d20cbe2f7accd12527ce60845ff44c153bb2a21b49"
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.106"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8fd00f0bb2e90d81d1044c2b32617f68fcb9fa3bb7640c23e9c748e53fb30934"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "pyo3"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7778bffd85cf38175ac1f545509665d0b9b92a198ca7941f131f85f7a4f9a872"
+dependencies = [
+ "cfg-if",
+ "indoc",
+ "libc",
+ "memoffset",
+ "once_cell",
+ "portable-atomic",
+ "pyo3-build-config",
+ "pyo3-ffi",
+ "pyo3-macros",
+ "unindent",
+]
+
+[[package]]
+name = "pyo3-build-config"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94f6cbe86ef3bf18998d9df6e0f3fc1050a8c5efa409bf712e661a4366e010fb"
+dependencies = [
+ "once_cell",
+ "target-lexicon",
+]
+
+[[package]]
+name = "pyo3-ffi"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e9f1b4c431c0bb1c8fb0a338709859eed0d030ff6daa34368d3b152a63dfdd8d"
+dependencies = [
+ "libc",
+ "pyo3-build-config",
+]
+
+[[package]]
+name = "pyo3-macros"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fbc2201328f63c4710f68abdf653c89d8dbc2858b88c5d88b0ff38a75288a9da"
+dependencies = [
+ "proc-macro2",
+ "pyo3-macros-backend",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "pyo3-macros-backend"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fca6726ad0f3da9c9de093d6f116a93c1a38e417ed73bf138472cf4064f72028"
+dependencies = [
+ "heck",
+ "proc-macro2",
+ "pyo3-build-config",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "41f2619966050689382d2b44f664f4bc593e129785a36d6ee376ddf37259b924"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rustversion"
+version = "1.0.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
+
+[[package]]
+name = "syn"
+version = "2.0.117"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e665b8803e7b1d2a727f4023456bbbbe74da67099c585258af0ad9c5013b9b99"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "target-lexicon"
+version = "0.12.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "61c41af27dd6d1e27b1b16b489db798443478cef1f06a660c96db617ba5de3b1"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.24"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6e4313cd5fcd3dad5cafa179702e2b244f760991f45397d14d4ebf38247da75"
+
+[[package]]
+name = "unindent"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7264e107f553ccae879d21fbea1d6724ac785e8c3bfc762137959b5802826ef3"
+
+[[package]]
+name = "visiter_native"
+version = "0.1.0"
+dependencies = [
+ "pyo3",
+]

visiter_native-0.1.0/Cargo.toml

@@ -0,0 +1,21 @@
+[package]
+name = "visiter_native"
+version = "0.1.0"
+edition = "2021"
+description = "Optional native BFS engine for visiter (Path A: Python callbacks via PyO3)."
+license = "MIT"
+readme = "README.md"
+
+[lib]
+name = "visiter_native"
+crate-type = ["cdylib"]
+
+[dependencies]
+# abi3-py311: one stable-ABI wheel per platform works on CPython 3.11+,
+# so distribution doesn't need a wheel per Python version.
+pyo3 = { version = "0.23", features = ["abi3-py311"] }
+
+[profile.release]
+opt-level = 3
+lto = "fat"
+codegen-units = 1

visiter_native-0.1.0/PKG-INFO

@@ -0,0 +1,36 @@
+Metadata-Version: 2.4
+Name: visiter_native
+Version: 0.1.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Rust
+Classifier: Topic :: Scientific/Engineering :: Visualization
+Summary: Optional native BFS engine for visiter (engine='native').
+Keywords: visiter,graph,bfs,native,pyo3
+Author-email: Jakob Stemberger <jakob.stemberger@gmail.com>
+License: MIT
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: Homepage, https://github.com/yaccob/visiter
+Project-URL: Issues, https://github.com/yaccob/visiter/issues
+
+# visiter_native
+
+The optional native BFS engine for [visiter](https://pypi.org/project/visiter/)
+(Path A): a PyO3 extension that runs visiter's graph build natively while keeping
+your Python callbacks.
+
+You normally don't install this directly — use the extra:
+
+```bash
+pip install "visiter[native]"
+```
+
+Once `visiter_native` is importable, `viter(...).build()` uses it automatically
+for unbounded builds (`engine="auto"`, the default) and produces a graph
+byte-identical to the pure-Python build. visiter works without it — pure Python
+is the always-available baseline.
+
+Distributed as `abi3` wheels (one per platform, CPython 3.11+). MIT licensed.
+See the [visiter manual](https://github.com/yaccob/visiter/blob/main/docs/manual.md#8-optional-native-acceleration-and-columnar-storage).
+

visiter_native-0.1.0/README.md

@@ -0,0 +1,19 @@
+# visiter_native
+
+The optional native BFS engine for [visiter](https://pypi.org/project/visiter/)
+(Path A): a PyO3 extension that runs visiter's graph build natively while keeping
+your Python callbacks.
+
+You normally don't install this directly — use the extra:
+
+```bash
+pip install "visiter[native]"
+```
+
+Once `visiter_native` is importable, `viter(...).build()` uses it automatically
+for unbounded builds (`engine="auto"`, the default) and produces a graph
+byte-identical to the pure-Python build. visiter works without it — pure Python
+is the always-available baseline.
+
+Distributed as `abi3` wheels (one per platform, CPython 3.11+). MIT licensed.
+See the [visiter manual](https://github.com/yaccob/visiter/blob/main/docs/manual.md#8-optional-native-acceleration-and-columnar-storage).

visiter_native-0.1.0/pyproject.toml

@@ -0,0 +1,26 @@
+[build-system]
+requires = ["maturin>=1.0,<2.0"]
+build-backend = "maturin"
+
+[project]
+name = "visiter_native"
+version = "0.1.0"
+description = "Optional native BFS engine for visiter (engine='native')."
+readme = "README.md"
+license = { text = "MIT" }
+authors = [{ name = "Jakob Stemberger", email = "jakob.stemberger@gmail.com" }]
+requires-python = ">=3.11"
+keywords = ["visiter", "graph", "bfs", "native", "pyo3"]
+classifiers = [
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Rust",
+    "Topic :: Scientific/Engineering :: Visualization",
+]
+
+[project.urls]
+Homepage = "https://github.com/yaccob/visiter"
+Issues = "https://github.com/yaccob/visiter/issues"
+
+[tool.maturin]
+features = ["pyo3/extension-module"]

visiter_native-0.1.0/src/lib.rs

@@ -0,0 +1,149 @@
+//! Optional native BFS engine for visiter (Path A / ①).
+//!
+//! `build_raw` mirrors the pure-Python `visiter.iteration.build` BFS for the
+//! **unbounded** subset (no max_depth / max_nodes / time_limit / bound). It
+//! calls the Python condition/op callables per node, so the callbacks stay
+//! Python — only the BFS expansion, deduplication and interning run natively.
+//!
+//! It returns the raw discovered structure (node objects in BFS order, their
+//! depths, and deduplicated edges as `(from_idx, to_idx, op_idx, label?)`).
+//! The Python shim assembles the final Graph dict (string keys, key_type,
+//! tags, op labels) from this, so the output is byte-identical to the
+//! pure-Python build for the supported subset.
+//!
+//! Node order, edge order, depth assignment and dedup semantics are kept
+//! faithful to the Python loop so a Graph-level parity test passes exactly.
+
+use pyo3::prelude::*;
+use std::collections::{HashMap, HashSet};
+
+/// A Python object used as a node key: hashed by the value's Python `hash()`
+/// (computed once), equality resolved by Python `==` only on hash collision.
+struct Key {
+    obj: Py<PyAny>,
+    hash: isize,
+}
+
+impl std::hash::Hash for Key {
+    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
+        state.write_isize(self.hash);
+    }
+}
+
+impl PartialEq for Key {
+    fn eq(&self, other: &Self) -> bool {
+        if self.hash != other.hash {
+            return false;
+        }
+        Python::with_gil(|py| {
+            self.obj.bind(py).eq(other.obj.bind(py)).unwrap_or(false)
+        })
+    }
+}
+impl Eq for Key {}
+
+type Edge = (u32, u32, usize, Option<String>);
+
+/// Run the unbounded BFS natively. `op_result_type` is the Python `OpResult`
+/// class, used to detect per-call label overrides.
+#[pyfunction]
+#[pyo3(signature = (starts, conditions, ops, exclusive, default_op, op_result_type))]
+fn build_raw(
+    py: Python<'_>,
+    starts: Vec<Py<PyAny>>,
+    conditions: Vec<Py<PyAny>>,
+    ops: Vec<Py<PyAny>>,
+    exclusive: Vec<bool>,
+    default_op: Option<Py<PyAny>>,
+    op_result_type: Py<PyAny>,
+) -> PyResult<(Vec<Py<PyAny>>, Vec<u32>, Vec<Edge>)> {
+    let n_rules = conditions.len();
+    let default_idx = n_rules; // op index reserved for the default op
+    let ort = op_result_type.bind(py);
+
+    let mut id_of: HashMap<Key, u32> = HashMap::new();
+    let mut values: Vec<Py<PyAny>> = Vec::new();
+    let mut depths: Vec<u32> = Vec::new();
+    let mut edges: Vec<Edge> = Vec::new();
+    let mut seen_edges: HashSet<(u32, u32)> = HashSet::new();
+
+    // `fire` is a macro (not a fn) so it can mutate the locals in place without
+    // fighting the borrow checker over a dozen &mut parameters.
+    macro_rules! fire {
+        ($xid:expr, $xb:expr, $op_idx:expr, $op_func:expr, $next:expr) => {{
+            let result = $op_func.bind(py).call1(($xb,))?;
+            let (nv, label): (Py<PyAny>, Option<String>) =
+                if result.is_instance(ort)? {
+                    let v = result.getattr("value")?.unbind();
+                    let l: Option<String> = result.getattr("label")?.extract()?;
+                    (v, l)
+                } else {
+                    (result.unbind(), None)
+                };
+            let h = nv.bind(py).hash()?;
+            let key = Key { obj: nv.clone_ref(py), hash: h };
+            let (nid, is_new) = match id_of.get(&key) {
+                Some(&i) => (i, false),
+                None => {
+                    let i = values.len() as u32;
+                    values.push(nv);
+                    depths.push(depths[$xid as usize] + 1);
+                    id_of.insert(key, i);
+                    (i, true)
+                }
+            };
+            if seen_edges.insert(($xid, nid)) {
+                edges.push(($xid, nid, $op_idx, label));
+            }
+            if is_new {
+                $next.push(nid);
+            }
+        }};
+    }
+
+    let mut frontier: Vec<u32> = Vec::new();
+    for s in starts {
+        let h = s.bind(py).hash()?;
+        let key = Key { obj: s.clone_ref(py), hash: h };
+        if !id_of.contains_key(&key) {
+            let id = values.len() as u32;
+            values.push(s);
+            depths.push(0);
+            id_of.insert(key, id);
+            frontier.push(id);
+        }
+    }
+
+    while !frontier.is_empty() {
+        let mut next: Vec<u32> = Vec::new();
+        for &xid in &frontier {
+            let x = values[xid as usize].clone_ref(py);
+            let xb = x.bind(py);
+            let mut any_matched = false;
+            for ri in 0..n_rules {
+                if conditions[ri].bind(py).call1((xb,))?.is_truthy()? {
+                    any_matched = true;
+                    fire!(xid, xb, ri, ops[ri], next);
+                    if exclusive[ri] {
+                        break;
+                    }
+                }
+            }
+            if !any_matched {
+                if let Some(ref dop) = default_op {
+                    fire!(xid, xb, default_idx, dop, next);
+                }
+            }
+        }
+        frontier = next;
+    }
+
+    Ok((values, depths, edges))
+}
+
+#[pymodule]
+fn visiter_native(m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(build_raw, m)?)?;
+    m.add("__version__", "0.0.0")?;
+    Ok(())
+}