wnet 0.9.0__tar.gz

wnet-0.9.0/CMakeLists.txt

@@ -0,0 +1,27 @@
+cmake_minimum_required(VERSION 3.15)
+
+execute_process(
+    COMMAND python -m pylmcf --include
+    OUTPUT_VARIABLE PYLMCF_INCLUDE_DIRS
+    OUTPUT_STRIP_TRAILING_WHITESPACE
+)
+
+
+project(wnet LANGUAGES CXX)
+find_package(Python 3.8
+  REQUIRED COMPONENTS Interpreter Development.Module
+  OPTIONAL_COMPONENTS Development.SABIModule)
+find_package(nanobind REQUIRED CONFIG)
+
+set(CMAKE_CXX_STANDARD 20)
+
+nanobind_add_module(wnet_cpp
+    NB_STATIC NOMINSIZE
+    src/wnet/cpp/wnet/wnet.cpp)
+
+target_include_directories(wnet_cpp PRIVATE ${PYLMCF_INCLUDE_DIRS})
+
+install(TARGETS wnet_cpp LIBRARY DESTINATION wnet)
+
+# Uncomment the following line to enable detailed debug prints in the C++ code
+# target_compile_definitions(wnet_cpp PRIVATE DO_TONS_OF_PRINTS)

wnet-0.9.0/LICENCE

@@ -0,0 +1,7 @@
+This software is Copyright 2025 Michał Startek, and provided under terms of MIT licence, below.
+
+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.

wnet-0.9.0/PKG-INFO

@@ -0,0 +1,72 @@
+Metadata-Version: 2.4
+Name: wnet
+Version: 0.9.0
+Summary: Tools for calculation of Wasserstein metric between distributions based on Network Flow algorithm
+Keywords: Wasserstein,Optimal Transport,Network Flow,Earth Mover's Distance
+Author-Email: =?utf-8?q?Micha=C5=82_Startek?= <michal.startek@mimuw.edu.pl>
+Maintainer-Email: =?utf-8?q?Micha=C5=82_Startek?= <michal.startek@mimuw.edu.pl>
+License-Expression: MIT
+License-File: LICENCE
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Development Status :: 4 - Beta
+Project-URL: Homepage, https://github.com/michalsta/wnet
+Project-URL: Repository, https://github.com/michalsta/wnet.git
+Requires-Dist: pylmcf
+Requires-Dist: numpy
+Provides-Extra: extras
+Requires-Dist: networkx; extra == "extras"
+Requires-Dist: matplotlib; extra == "extras"
+Description-Content-Type: text/markdown
+
+# wnet
+
+Wasserstein Network (wnet) is a Python/C++ library for working with Wasserstein distances. It uses the Min Cost Flow algorithm as implemented by the [LEMON library](https://lemon.cs.elte.hu/trac/lemon), and exposes this functionality to Python via the [pylmcf module](https://github.com/michalsta/pylmcf), enabling efficient computation and manipulation of Wasserstein distances between multidimensional distributions.
+
+## Features
+- Wasserstein and Truncated Wasserstein distance calculations
+- Calculation of der
+- Python and C++ integration
+
+## Installation
+
+You can install the Python package using pip:
+
+```bash
+pip install .
+```
+
+## Usage
+
+Simple usage:
+```python
+import numpy as np
+from wnet import WassersteinDistance, Distribution
+from wnet.distances import L1Distance
+
+positions1 = np.array(
+    [[0, 1, 5, 10],
+     [0, 0, 0, 3]]
+)
+intensities1 = np.array([10, 5, 5, 5])
+
+positions2 = np.array(
+    [[1,10],
+    [0, 0]])
+intensities2 = np.array([20, 5])
+
+S1 = Distribution(positions1, intensities1)
+S2 = Distribution(positions2, intensities2)
+
+print(WassersteinDistance(S1, S2, L1Distance()))
+# 45
+```
+
+## Licence
+MIT Licence
+
+## Related Projects
+
+- [pylmcf](https://github.com/cheind/pylmcf) - Python bindings for Min Cost Flow algorithms from LEMON library.
+- [LEMON Graph Library](https://lemon.cs.elte.hu/trac/lemon) - C++ library for efficient graph algorithms.

wnet-0.9.0/README.md

@@ -0,0 +1,50 @@
+# wnet
+
+Wasserstein Network (wnet) is a Python/C++ library for working with Wasserstein distances. It uses the Min Cost Flow algorithm as implemented by the [LEMON library](https://lemon.cs.elte.hu/trac/lemon), and exposes this functionality to Python via the [pylmcf module](https://github.com/michalsta/pylmcf), enabling efficient computation and manipulation of Wasserstein distances between multidimensional distributions.
+
+## Features
+- Wasserstein and Truncated Wasserstein distance calculations
+- Calculation of der
+- Python and C++ integration
+
+## Installation
+
+You can install the Python package using pip:
+
+```bash
+pip install .
+```
+
+## Usage
+
+Simple usage:
+```python
+import numpy as np
+from wnet import WassersteinDistance, Distribution
+from wnet.distances import L1Distance
+
+positions1 = np.array(
+    [[0, 1, 5, 10],
+     [0, 0, 0, 3]]
+)
+intensities1 = np.array([10, 5, 5, 5])
+
+positions2 = np.array(
+    [[1,10],
+    [0, 0]])
+intensities2 = np.array([20, 5])
+
+S1 = Distribution(positions1, intensities1)
+S2 = Distribution(positions2, intensities2)
+
+print(WassersteinDistance(S1, S2, L1Distance()))
+# 45
+```
+
+## Licence
+MIT Licence
+
+## Related Projects
+
+- [pylmcf](https://github.com/cheind/pylmcf) - Python bindings for Min Cost Flow algorithms from LEMON library.
+- [LEMON Graph Library](https://lemon.cs.elte.hu/trac/lemon) - C++ library for efficient graph algorithms.

wnet-0.9.0/example.py

@@ -0,0 +1,26 @@
+from wnet import Distribution, WassersteinNetwork
+from wnet.distances import L1Distance, wrap_distance_function
+import numpy as np
+
+positions1 = np.array(
+    [[0, 1, 5, 10],
+     [0, 0, 0, 3]]
+)
+intensities1 = np.array([10, 5, 5, 5])
+
+positions2 = np.array(
+    [[1,10],
+    [0, 0]])
+intensities2 = np.array([20, 5])
+
+S1 = Distribution(positions1, intensities1)
+S2 = Distribution(positions2, intensities2)
+
+W = WassersteinNetwork(S1, [S2], L1Distance(), 10)
+W.add_simple_trash(15)
+W.build()
+W.solve()
+print("Total cost:", W.total_cost())
+print(W.subgraphs()[0].as_netowkrx())
+print(W.flows_for_target(0))
+W.subgraphs()[0].show()

wnet-0.9.0/pyproject.toml

@@ -0,0 +1,32 @@
+[build-system]
+requires = ["scikit-build-core", "nanobind", "pylmcf"]
+build-backend = "scikit_build_core.build"
+
+[project]
+name = "wnet"
+version = "0.9.0"
+description = "Tools for calculation of Wasserstein metric between distributions based on Network Flow algorithm"
+authors = [{ name="Michał Startek", email="michal.startek@mimuw.edu.pl" }]
+maintainers = [{ name="Michał Startek", email="michal.startek@mimuw.edu.pl" }]
+dependencies = ["pylmcf", "numpy"]
+license = "MIT"
+license-files = ["LICENCE"]
+readme = "README.md"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Topic :: Scientific/Engineering :: Mathematics",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Development Status :: 4 - Beta",
+
+]
+keywords = ["Wasserstein", "Optimal Transport", "Network Flow", "Earth Mover's Distance"]
+
+[project.urls]
+"Homepage" = "https://github.com/michalsta/wnet"
+"Repository" = "https://github.com/michalsta/wnet.git"
+
+[project.optional-dependencies]
+extras = ["networkx", "matplotlib"]
+
+[tool.scikit-build]
+sdist.exclude = [".github", ".gitignore", "tests"]

wnet-0.9.0/reinstall.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+
+pip uninstall -y wnet
+VERBOSE=1 pip install -v -e .

wnet-0.9.0/src/wnet/__init__.py

@@ -0,0 +1,8 @@
+#! /usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from .distribution import Distribution_1D, Distribution
+from . import wnet_cpp
+from .wasserstein_network import WassersteinNetwork
+from .wasserstein import WassersteinDistance, TruncatedWassersteinDistance
+from .distribution import Distribution_1D, Distribution

wnet-0.9.0/src/wnet/__main__.py

@@ -0,0 +1,21 @@
+import importlib.metadata
+from pathlib import Path
+
+
+__version__ = importlib.metadata.version("wnet")
+
+def module_main():
+    import argparse
+    parser = argparse.ArgumentParser(description="WNet: A module for calculation of Wasserstein metric between distributions using network flow-based approach.")
+    parser.add_argument("--version", "-v",action="version", version=__version__)
+    parser.add_argument("--include", "-i", help="Print include path for C++ headers", action="store_true")
+
+    args = parser.parse_args()
+
+    if args.include:
+        print(Path(__file__).parent / "cpp")
+    else:
+        parser.print_help()
+
+if __name__ == "__main__":
+    module_main()