Commit the good (?) version of cone.py.

[sage.d.git] / mjo / cone / cone.py

diff --git a/mjo/cone/cone.py b/mjo/cone/cone.py
index 4b0193692edd7655f2880408d143bf141e6d567c..3a1e190cb2ebe41f57810f04726f8123294c55cd 100644 (file)
--- a/mjo/cone/cone.py
+++ b/mjo/cone/cone.py
@@ -8,6 +8,58 @@ addsitedir(abspath('../../'))
 from sage.all import *
 
 
+def project_span(K, K2 = None):
+    r"""
+    Return a "copy" of ``K`` embeded in a lower-dimensional space.
+
+    By default, we will project ``K`` into the subspace spanned by its
+    rays. However, if ``K2`` is not ``None``, we will project into the
+    space spanned by the rays of ``K2`` instead.
+
+    EXAMPLES::
+
+        sage: K = Cone([(1,0,0), (0,1,0)])
+        sage: project_span(K)
+        2-d cone in 2-d lattice N
+        sage: project_span(K).rays()
+        N(1, 0),
+        N(0, 1)
+        in 2-d lattice N
+
+        sage: K = Cone([(1,0,0), (0,1,0)])
+        sage: K2 = Cone([(0,1)])
+        sage: project_span(K, K2).rays()
+        N(1)
+        in 1-d lattice N
+
+    """
+    # Allow us to use a second cone to generate the subspace into
+    # which we're "projecting."
+    if K2 is None:
+        K2 = K
+
+    # Use these to generate the new cone.
+    cs1 = K.rays().matrix().columns()
+
+    # And use these to figure out which indices to drop.
+    cs2 = K2.rays().matrix().columns()
+
+    perp_idxs = []
+
+    for idx in range(0, len(cs2)):
+        if cs2[idx].is_zero():
+            perp_idxs.append(idx)
+
+    solid_cols = [ cs1[idx] for idx in range(0,len(cs1))
+                            if not idx in perp_idxs
+                            and not idx >= len(cs2) ]
+
+    m = matrix(solid_cols)
+    L = ToricLattice(len(m.rows()))
+    J = Cone(m.transpose(), lattice=L)
+    return J
+
+
 def discrete_complementarity_set(K):
     r"""
     Compute the discrete complementarity set of this cone.
@@ -152,6 +204,23 @@ def LL(K):
         sage: sum(map(abs, l))
         0
 
+    Try the formula in my paper::
+
+        sage: K = random_cone(max_dim=15, max_rays=25)
+        sage: actual = lyapunov_rank(K)
+        sage: K_S = project_span(K)
+        sage: J_T1 = project_span(K, K_S.dual())
+        sage: J_T2 = project_span(K_S.dual()).dual()
+        sage: J_T2 = Cone(J_T2.rays(), lattice=J_T1.lattice())
+        sage: J_T1 == J_T2
+        True
+        sage: J_T = J_T1
+        sage: l = K.linear_subspace().dimension()
+        sage: codim = K.lattice_dim() - K.dim()
+        sage: expected = lyapunov_rank(J_T) + K.dim()*(l + codim) + codim**2
+        sage: actual == expected
+        True
+
     """
     V = K.lattice().vector_space()