snappy 3.2__cp313-cp313-macosx_11_0_arm64.whl

snappy/snap/t3mlite/homology.py

@@ -0,0 +1,53 @@
+from .simplex import *
+from .linalg import Matrix
+
+
+def boundary_three(manifold):
+    F, T = len(manifold.Faces), len(manifold.Tetrahedra)
+    ans = Matrix(F, T)
+    for F in manifold.Faces:
+        t0, t1 = (C.Tetrahedron.Index for C in F.Corners)
+        ans[F.Index, t0] += 1
+        ans[F.Index, t1] += -1
+    return ans
+
+
+def boundary_two(manifold):
+    VerticesOfFace = { F0 : (V1, V2, V3), F1 : (V0, V3, V2), F2 : (V0, V1, V3), F3 :
+                       (V0, V2, V1) }
+
+    E, F = len(manifold.Edges), len(manifold.Faces)
+    ans = Matrix(E, F)
+    for F in manifold.Faces:
+        C = F.Corners[0]
+        tet = C.Tetrahedron
+        vertices = VerticesOfFace[C.Subsimplex]
+        for i in range(3):
+            a, b = vertices[i], vertices[(i + 1) % 3]
+            e = tet.Class[a | b]
+            ans[e.index(), F.Index] += e.orientation_with_respect_to(tet, a, b)
+    return ans
+
+
+def boundary_one(manifold):
+    V, E = len(manifold.Vertices), len(manifold.Edges)
+    ans = Matrix(V, E)
+    for e in manifold.Edges:
+        v_init, v_term = (v.Index for v in e.Vertices)
+        ans[v_term, e.Index] += 1
+        ans[v_init, e.Index] += -1
+    return ans
+
+
+def boundary_maps(manifold):
+    """
+    The boundary maps in the homology chain complex of the
+    underlying cell-complex of a Mcomplex.
+
+    >>> M = Mcomplex('o9_12345')
+    >>> len(M.boundary_maps()) == 3
+    True
+    """
+    B1, B2, B3 = boundary_one(manifold), boundary_two(manifold), boundary_three(manifold)
+    assert B1*B2 == 0 and B2*B3 == 0
+    return B1, B2, B3

snappy/snap/t3mlite/linalg.py

@@ -0,0 +1,419 @@
+"""
+The linear algebra needed for t3m.  Build on top of PARI.
+"""
+
+from snappy.pari import pari
+
+PariGen = type(pari(0))
+
+
+def is_iterable(obj):
+    try:
+        iter(obj)
+        return True
+    except TypeError:
+        return False
+
+
+def is_pari_col_vector(obj):
+    return isinstance(obj, PariGen) and obj.type() == 't_COL'
+
+
+def is_pari_row_vector(obj):
+    return isinstance(obj, PariGen) and obj.type() == 't_VEC'
+
+
+def is_pari_matrix(obj):
+    return isinstance(obj, PariGen) and obj.type() == 't_MAT'
+
+
+class Vector:
+    """
+    >>> v = Vector(3, range(3, 6)); v
+    [3, 4, 5]
+    >>> len(v)
+    3
+    >>> v[1]
+    4
+    >>> v[2] = 6; v
+    [3, 4, 6]
+    >>> v[1:3] = [7, 8]; v
+    [3, 7, 8]
+    >>> w = Vector([-2, 2, -6])
+    >>> v + w
+    [1, 9, 2]
+    >>> 2*v
+    [6, 14, 16]
+    >>> 3*v - w
+    [11, 19, 30]
+    >>> v += w; v
+    [1, 9, 2]
+    >>> v*w
+    4
+    >>> abs(w)
+    [2, 2, 6]
+
+    Internally, a Vector is a PARI *column* vector:
+
+    >>> w.pari.type()
+    't_COL'
+    """
+    def __init__(self, n=None, entries=None, pari_vector=None):
+        if n is not None:
+            if entries is None:
+                if is_iterable(n):
+                    entries = n
+                    n = len(entries)
+                else:
+                    entries = n*[0]
+            assert n == len(entries)
+            pari_vector = pari.vector(n, entries).Col()
+        self.pari = pari_vector
+
+    def __getitem__(self, i):
+        """
+        >>> v = Vector([1, 2, 3, 4])
+        >>> v[1]
+        2
+        >>> v[-1]
+        4
+        >>> v[1:-1]
+        [2, 3]
+        """
+        if isinstance(i, slice):
+            return self.__class__(list(self.pari)[i])
+        if i < 0:
+            i = i % len(self)
+        return self.pari[i]
+
+    def __setitem__(self, i, value):
+        self.pari[i] = value
+
+    def __len__(self):
+        return int(self.pari.length())
+
+    def __repr__(self):
+        return repr(self.pari)[:-1]
+
+    def __eq__(self, other):
+        """
+        >>> Vector([1,2,3]) == 0
+        False
+        >>> Vector([0, 0, 0]) == 0
+        True
+        >>> Vector([0, 1]) == Vector([0, 1])
+        True
+        """
+        if isinstance(other, Vector):
+            return self.pari == other.pari
+        if other == 0:
+            return all(e == 0 for e in self)
+        raise NotImplementedError
+
+    def __ne__(self, other):
+        """
+        >>> Vector([1,2,3]) != 0
+        True
+        >>> Vector([0, 0, 0]) != 0
+        False
+        >>> Vector([0, 1]) != Vector([0, 1])
+        False
+        """
+        return not self.__eq__(other)
+
+    def __add__(self, other):
+        if isinstance(other, Vector):
+            return self.__class__(pari_vector=self.pari + other.pari)
+        raise NotImplementedError
+
+    def __rmul__(self, other):
+        return self.__class__(pari_vector=other * self.pari)
+
+    def __truediv__(self, other):
+        return self.__class__(pari_vector=self.pari / other)
+
+    def __mul__(self, other):
+        if isinstance(other, Vector):
+            if self.pari.length() != other.pari.length():
+                raise ValueError("Vectors have different lengths")
+            return self.pari.Vec() * other.pari
+        raise NotImplementedError
+
+    def __sub__(self, other):
+        if isinstance(other, Vector):
+            return self.__class__(pari_vector=self.pari - other.pari)
+        raise NotImplementedError
+
+    def __abs__(self):
+        return self.__class__([abs(s) for s in self])
+
+    def __neg__(self):
+        return (-1)*self
+
+
+class Matrix:
+    """
+    >>> A = Matrix(2, 3, range(6))
+    >>> A
+    [0, 1, 2; 3, 4, 5]
+    >>> (A.nrows(), A.ncols())
+    (2, 3)
+    >>> A[1, 2]
+    5
+    >>> A[0, 2] = 6; A
+    [0, 1, 6; 3, 4, 5]
+    >>> A[0]
+    [0, 1, 6]
+    >>> A.column(2)
+    [6, 5]
+    >>> A.entries()
+    [0, 1, 6, 3, 4, 5]
+    >>> Matrix([[1,0,1], [2,3,4]])
+    [1, 0, 1; 2, 3, 4]
+    """
+
+    _vector_class = Vector
+
+    def __init__(self, nrows=None, ncols=None, entries=None, pari_matrix=None):
+        if nrows is not None:
+            if ncols is None:
+                nice_entries = nrows
+                try:
+                    ncols = len(nrows[0])
+                    nrows = len(nrows)
+                except TypeError:
+                    ncols = nrows.ncols()
+                    nrows = nrows.nrows()
+                entries = [e for row in nice_entries for e in row]
+            if entries is None:
+                entries = (nrows*ncols)*[0]
+            assert len(entries) == nrows*ncols
+            pari_matrix = pari.matrix(nrows, ncols, entries)
+        self.pari = pari_matrix
+
+    def nrows(self):
+        return self.pari.nrows()
+
+    def ncols(self):
+        return self.pari.ncols()
+
+    def column(self, j):
+        pari_col = self.pari[j]
+        return self._vector_class(pari_vector=pari_col)
+
+    def row(self, i):
+        """
+        >>> A = Matrix(3, 2, range(6))
+        >>> A.row(1)
+        [2, 3]
+        """
+        return self._vector_class([self[i, j] for j in range(self.ncols())])
+
+    def columns(self):
+        """
+        >>> A = Matrix(3, 2, range(6))
+        >>> A.columns()
+        [[0, 2, 4], [1, 3, 5]]
+        """
+        return [self.column(j) for j in range(self.ncols())]
+
+    def rows(self):
+        """
+        >>> A = Matrix(3, 2, range(6))
+        >>> A.rows()
+        [[0, 1], [2, 3], [4, 5]]
+        """
+        return [self.row(i) for i in range(self.nrows())]
+
+    def entries(self):
+        ans = []
+        for i in range(self.nrows()):
+            for j in range(self.ncols()):
+                ans.append(self[i, j])
+        return ans
+
+    def dot(self, vec):
+        """
+        >>> A = Matrix(3, 4, range(12))
+        >>> A.dot(range(4, 8))
+        [38, 126, 214]
+        """
+        if not isinstance(vec, Vector):
+            vec = self._vector_class(vec)
+        if vec.pari.length() != self.pari.ncols():
+            raise ValueError('Vector size does not match matrix')
+        ans = self.pari * vec.pari
+        return self._vector_class(pari_vector=ans)
+
+    def solve(self, b):
+        """
+        Return a vector v for which A v = b.
+
+        >>> A = Matrix(2, 2, range(4))
+        >>> A.solve([6, 8])
+        [-5, 6]
+        >>> B = Matrix(4, 2, range(0, 8))
+        >>> B.solve([5, 23, 41, 59])
+        [4, 5]
+        """
+        ncols, nrows = self.ncols(), self.nrows()
+        if not isinstance(b, Vector):
+            b = self._vector_class(b)
+
+        if nrows == ncols:
+            v = self.pari.matsolve(b.pari)
+            ans = self._vector_class(pari_vector=v)
+        elif nrows > ncols:
+            if self.rank() != ncols:
+                raise ValueError
+            ker = self.pari.mattranspose().matker()
+            M = self.pari.concat(ker)
+            v = M.matsolve(b.pari)
+            ans = self._vector_class(list(v)[:ncols])
+        assert self.dot(ans) == b
+        return ans
+
+    def solve_right(self, b):
+        """
+        Return a vector v for which A v = b.
+
+        >>> A = Matrix(2, 2, range(4))
+        >>> A.solve_right([6, 8])
+        [-5, 6]
+        """
+        return self.solve(b)
+
+    def rank(self):
+        """
+        >>> A = Matrix(2, 3, range(6))
+        >>> B = Matrix(3, 2, range(6))
+        >>> A.rank() == B.rank() == 2
+        True
+        """
+        try:
+            return self.pari.matrank()
+        except AttributeError:
+            m, n = self.nrows(), self.ncols()
+            result = [int(x) for x in self.pari.matsnf()]
+            # PARI views the input to matsnf0 as square.
+            if m < n:
+                result = result + [0]*(n-m)
+            if m > n:
+                for i in range(m - n):
+                    result.remove(0)
+            return len([r for r in result if r > 0])
+
+    def list(self):
+        """
+        >>> A = Matrix(4, 5, range(20))
+        >>> A.list() == list(range(20))
+        True
+        """
+        a, b = self.nrows(), self.ncols()
+        return [self[i, j] for i in range(a) for j in range(b)]
+
+    def transpose(self):
+        """
+        >>> A = Matrix(2, 3, range(6))
+        >>> B = Matrix([[0, 3], [1, 4], [2, 5]])
+        >>> A.transpose() == B and B.transpose() == A
+        True
+        """
+        return self.__class__(pari_matrix=self.pari.mattranspose())
+
+    def inverse(self):
+        """
+        >>> A = Matrix([[1, 2], [3, 4]])
+        >>> A * A.inverse()
+        [1, 0; 0, 1]
+        """
+        a, b = self.nrows(), self.ncols()
+        if a != b:
+            raise ValueError('An invertible matrix must be square')
+
+        return self.__class__(pari_matrix=self.pari.matsolve(pari.matid(a)))
+
+    def det(self):
+        return self.pari.matdet()
+
+    def __repr__(self):
+        return repr(self.pari)
+
+    def __setitem__(self, ij, value):
+        self.pari[ij] = value
+
+    def __getitem__(self, ij):
+        try:
+            i, j = ij
+            return self.pari[ij]
+        except TypeError:
+            pari_row = self.pari.mattranspose()[ij]
+            return self._vector_class(pari_row)
+
+    def __mul__(self, other):
+        """
+        >>> A = Matrix(2, 3, range(6))
+        >>> B = Matrix(3, 3, range(9))
+        >>> A * B
+        [15, 18, 21; 42, 54, 66]
+        >>> A * (5, 6, 7)
+        [20, 74]
+        """
+        if isinstance(other, Matrix):
+            if other.pari.nrows() != self.pari.ncols():
+                raise ValueError('Matrix sizes do not allow for multiplication')
+            return self.__class__(pari_matrix=self.pari * other.pari)
+        if is_iterable(other):
+            return self.dot(other)
+
+    def __eq__(self, other):
+        """
+        >>> A = Matrix(2, 2, range(4))
+        >>> B = Matrix(2, 1)
+        >>> A == B
+        False
+        >>> C = Matrix(2, 2, range(4))
+        >>> A == C
+        True
+        >>> B == 0
+        True
+        >>> 0 == B
+        True
+        """
+        if isinstance(other, Matrix):
+            return self.pari == other.pari
+        elif other == 0:
+            return all(e == 0 for e in self.list())
+
+    def __ne__(self, other):
+        """
+        >>> A = Matrix(2, 2, range(4))
+        >>> B = Matrix(2, 1)
+        >>> A != B
+        True
+        >>> C = Matrix(2, 2, range(4))
+        >>> (A != C, B != 0, 0 != B)
+        (False, False, False)
+        """
+        return not self.__eq__(other)
+
+
+def gcd(a, b):
+    a, b = abs(a), abs(b)
+    if a == 0:
+        if b == 0:
+            raise ValueError("gcd(0,0) undefined.")
+        return b
+
+    while True:
+        b = b % a
+        if (b == 0):
+            return a
+        a = a % b
+        if (a == 0):
+            return b
+
+
+if __name__ == '__main__':
+    import doctest
+    print(doctest.testmod())