metcones 0.0.1__py3-none-any.whl

metcones/__init__.py

@@ -0,0 +1,2 @@
+from .metric import *
+from .cut import *

metcones/cut.py

@@ -0,0 +1,51 @@
+import numpy as np
+from itertools import combinations
+from sage.geometry.cone import Cone
+from sage.geometry.polyhedron.constructor import Polyhedron
+from sage.all import QQ
+from metcones.utils import link_obj_methods
+
+
+def get_cut_rays_vectorized(n):
+    inc_matrix = np.arange(2 ** (n - 1))[:, None]
+    inc_steps = inc_matrix & (1 << np.arange(n - 1))
+    inc_steps = (inc_steps > 0).astype(int)
+    nodes = np.hstack([np.ones((2 ** (n - 1), 1)), inc_steps])
+    edges = np.array(list(combinations(range(n), 2)))
+    semi_metrics = (nodes[:, edges[:, 0]] != nodes[:, edges[:, 1]]).astype(int)
+    return semi_metrics[np.any(semi_metrics, axis=1)]
+
+
+def CutPoly(n, **kwargs):
+    if "backend" in kwargs and "base_ring" not in kwargs:
+        kwargs["base_ring"] = QQ
+    elif "backend" not in kwargs and "base_ring" not in kwargs:
+        kwargs["backend"] = "normaliz"
+        kwargs["base_ring"] = QQ
+    cut_rays = get_cut_rays_vectorized(n)
+    return Polyhedron(rays=cut_rays, **kwargs)
+
+
+def CutCone(n, **kwargs):
+    cut_rays = get_cut_rays_vectorized(n)
+    poly = Polyhedron(rays=cut_rays, **kwargs)
+    c = Cone(poly)
+    methods = [
+        "n_Hrepresentation",
+        "n_Vrepresentation",
+        "n_equations",
+        "n_inequalities",
+        "n_vertices",
+        "n_rays",
+        "n_lines",
+        "n_facets",
+        "f_vector",
+        "triangulate",
+    ]
+    link_obj_methods(poly, c, methods, src_name="poly")
+    return c
+
+
+if __name__ == "__main__":
+    print(get_cut_rays_vectorized(3))
+    print(get_cut_rays_vectorized(4))

metcones/metric.py

@@ -0,0 +1,145 @@
+import numpy as np
+from scipy.sparse import csr_matrix
+from itertools import combinations
+
+from sage.all import QQ
+from sage.geometry.polyhedron.constructor import Polyhedron
+from sage.geometry.cone import Cone
+from metcones.utils import link_obj_methods
+
+
+def MetricPoly(n_pts, include_pos: bool = False, no_limit: bool = False, **kwargs):
+    if n_pts < 3:
+        raise ValueError(
+            "Invalid metric poly definition. (Triangle inequality definition requires a minimum of three points)"
+        )
+    elif not no_limit and n_pts > 7:
+        raise ValueError(
+            "Value exceeds calculation limit. (Combinatorial explosion on metric cones n >= 7)"
+        )
+    ieqs = _calc_ieqs(n_pts, include_pos=include_pos).astype(int)
+    ieqs = _ieqs_to_sage_ieqs(ieqs)
+    if "backend" in kwargs and "base_ring" not in kwargs:
+        kwargs["base_ring"] = QQ
+    elif "backend" not in kwargs and "base_ring" not in kwargs:
+        kwargs["backend"] = "normaliz"
+        kwargs["base_ring"] = QQ
+    poly = Polyhedron(ieqs=ieqs, **kwargs)
+    return poly
+
+
+def MetricCone(n_pts, include_pos=False, no_limit: bool = False, **kwargs):
+    poly = MetricPoly(n_pts, include_pos=include_pos, no_limit=no_limit, **kwargs)
+    c = Cone(poly)
+    methods = [
+        "n_Hrepresentation",
+        "n_Vrepresentation",
+        "n_equations",
+        "n_inequalities",
+        "n_vertices",
+        "n_rays",
+        "n_lines",
+        "n_facets",
+        "f_vector",
+        "triangulate",
+    ]
+    link_obj_methods(poly, c, methods, src_name="poly")
+    return c
+
+
+def UMetPoly(n_pts, include_pos: bool = False, no_limit: bool = False, **kwargs):
+    if n_pts < 3:
+        raise ValueError(
+            "Invalid metric poly definition. (Triangle inequality definition requires a minimum of three points)"
+        )
+    elif not no_limit and n_pts > 7:
+        raise ValueError(
+            "Value exceeds calculation limit. (Combinatorial explosion on metric cones n >= 7)"
+        )
+    ieqs = _calc_ultra_ieqs(n_pts, include_pos=include_pos).astype(int)
+    ieqs = _ieqs_to_sage_ieqs(ieqs)
+    if "backend" in kwargs and "base_ring" not in kwargs:
+        kwargs["base_ring"] = QQ
+    elif "backend" not in kwargs and "base_ring" not in kwargs:
+        kwargs["backend"] = "normaliz"
+        kwargs["base_ring"] = QQ
+    poly = Polyhedron(ieqs=ieqs, **kwargs)
+    return poly
+
+
+def UMetCone(n_pts, include_pos=False, no_limit: bool = False, **kwargs):
+    poly = UMetPoly(n_pts, include_pos=include_pos, no_limit=no_limit, **kwargs)
+    c = Cone(poly)
+    methods = [
+        "n_Hrepresentation",
+        "n_Vrepresentation",
+        "n_equations",
+        "n_inequalities",
+        "n_vertices",
+        "n_rays",
+        "n_lines",
+        "n_facets",
+        "f_vector",
+        "triangulate",
+    ]
+    link_obj_methods(poly, c, methods, src_name="poly")
+    return c
+
+
+def _calc_ieqs(n_pts: int, include_pos: bool = False):
+    pts = range(n_pts)
+    edges = list(combinations(pts, 2))
+    num_edges = len(edges)
+    edge_map = {edge: i for i, edge in enumerate(edges)}
+    trips = np.array(list(combinations(pts, 3)))
+    num_trips = len(trips)
+    idx = np.array(
+        [
+            [edge_map[(t[0], t[1])], edge_map[(t[1], t[2])], edge_map[(t[0], t[2])]]
+            for t in trips
+        ]
+    )
+    cols = idx[:, [[0, 1, 2], [0, 2, 1], [1, 2, 0]]].reshape(-1, 3)
+    rows = np.repeat(np.arange(num_trips * 3), 3)
+    data = np.tile([1, 1, -1], len(trips) * 3)
+    ieqs = csr_matrix(
+        (data, (rows, cols.ravel())), shape=(num_trips * 3, num_edges)
+    ).toarray()
+    if include_pos:
+        ieqs = np.vstack([ieqs, np.eye(num_edges)])
+    return ieqs
+
+
+def _calc_ultra_ieqs(n_pts: int, include_pos: bool = False):
+    pts = range(n_pts)
+    edges = list(combinations(pts, 2))
+    num_edges = len(edges)
+    edge_map = {edge: i for i, edge in enumerate(edges)}
+    trips = np.array(list(combinations(pts, 3)))
+    num_trips = len(trips)
+    idx = np.array(
+        [
+            [edge_map[(t[0], t[1])], edge_map[(t[1], t[2])], edge_map[(t[0], t[2])]]
+            for t in trips
+        ]
+    )
+    cols = idx[:, [[0, 1], [0, 2], [1, 2], [1, 0], [2, 0], [2, 1]]].reshape(-1, 2)
+    rows = np.repeat(np.arange(num_trips * 6), 2)
+    data = np.tile([1, -1], num_trips * 6)
+    ieqs = csr_matrix(
+        (data, (rows, cols.ravel())), shape=(num_trips * 6, num_edges)
+    ).toarray()
+    if include_pos:
+        ieqs = np.vstack([ieqs, np.eye(num_edges)])
+    return ieqs
+
+
+def _ieqs_to_sage_ieqs(ieqs: np.ndarray) -> np.ndarray:
+    assert ieqs is not None, "ieqs must be a valid sparse matrix"
+    rows, _ = ieqs.shape
+    zeros = np.zeros((rows, 1))
+    return np.hstack((zeros, ieqs))
+
+
+if __name__ == "__main__":
+    print(MetricPoly(3).f_vector())

metcones/utils.py

@@ -0,0 +1,20 @@
+from sage.geometry.polyhedron.parent import Polyhedra
+
+
+def link_obj_methods(src, dst, method_names, src_name: str = "__src_obj"):
+    setattr(dst, src_name, src)
+
+    for name in method_names:
+
+        def make_method(n, s_name):
+            return lambda *args, **kwargs: getattr(getattr(dst, s_name), n)(
+                *args, **kwargs
+            )
+
+        setattr(dst, name, make_method(name, src_name))
+
+
+def compare_rays(C1, C2):
+    rays1 = set(tuple(ray) for ray in C1.rays())
+    rays2 = set(tuple(ray) for ray in C2.rays())
+    return rays1 == rays2

metcones-0.0.1.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.4
+Name: metcones
+Version: 0.0.1
+Summary: Library adding Metric and Cut Cone generation utilizing sage math.
+Author-email: Daniel Castro <Danielaca1818@gmail.com>
+Description-Content-Type: text/markdown
+Requires-Dist: sage
+
+# MetCones
+
+A library to generate metric and cut cones as SageMath objects.
+
+## Installation
+
+```bash
+pip install metcones
+```
+
+## Usage
+
+Generating the cones as SageMath objects:
+
+```python
+from metcones import CutCone, MetricCone, CutPoly, MetricPoly
+
+cut = CutCone(3)
+metric = MetricCone(3)
+cut_poly = CutPoly(3)
+met_poly = MetricPoly(3)
+```
+
+## Requirements
+
+- SageMath
+- PyNormaliz
+
+## License
+
+MIT

metcones-0.0.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+metcones/__init__.py,sha256=G8Ms2PUa-qlOpZRajl6AflkTK8bvvT3FiYDan01cl6E,41
+metcones/cut.py,sha256=6PfZ-noM4jSmt1UUvE1jObBnh4JFfxFqP6l5SIwx0Zo,1555
+metcones/metric.py,sha256=TfMvSj8vcytUuUo-ZI6nyCFQWxQKknbU_IAywKllwpg,4776
+metcones/utils.py,sha256=LKzH9kteULeL0Y7DNrodJA0sX3hnrgxOHvG5XDBA-S8,563
+metcones-0.0.1.dist-info/METADATA,sha256=XTS_PK7ENNHua19uc-eaUrLuSCfpPsFA3JmLA-euRbI,647
+metcones-0.0.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+metcones-0.0.1.dist-info/top_level.txt,sha256=cc7eGxtws6cj_Qb5RllvEabLUaJ8KQ_1l9_1flPEOdg,9
+metcones-0.0.1.dist-info/RECORD,,

metcones-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

metcones-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+metcones