eja: implement my own algebra subclass.

[sage.d.git] / mjo / eja / euclidean_jordan_algebra.py

diff --git a/mjo/eja/euclidean_jordan_algebra.py b/mjo/eja/euclidean_jordan_algebra.py
index 5ccf2f29fd480f4a499d55fc3d9dad85942dc6f4..624806fddac93c9292b229d6afd7954224827e87 100644 (file)
--- a/mjo/eja/euclidean_jordan_algebra.py
+++ b/mjo/eja/euclidean_jordan_algebra.py
@@ -5,12 +5,56 @@ are used in optimization, and have some additional nice methods beyond
 what can be supported in a general Jordan Algebra.
 """
 
-from sage.all import *
+from sage.structure.unique_representation import UniqueRepresentation
+from sage.algebras.finite_dimensional_algebras.finite_dimensional_algebra import FiniteDimensionalAlgebra
+
+class FiniteDimensionalEuclideanJordanAlgebra(FiniteDimensionalAlgebra):
+    @staticmethod
+    def __classcall__(cls, field, mult_table, names='e', category=None):
+        fda = super(FiniteDimensionalEuclideanJordanAlgebra, cls)
+        return fda.__classcall_private__(cls,
+                                         field,
+                                         mult_table,
+                                         names,
+                                         category)
+
+    def __init__(self, field, mult_table, names='e', category=None):
+        fda = super(FiniteDimensionalEuclideanJordanAlgebra, self)
+        fda.__init__(field, mult_table, names, category)
+
+
+    def _repr_(self):
+        """
+        Return a string representation of ``self``.
+        """
+        return "Euclidean Jordan algebra of degree {} over {}".format(self.degree(), self.base_ring())
+
+
 
 def eja_rn(dimension, field=QQ):
     """
     Return the Euclidean Jordan Algebra corresponding to the set
     `R^n` under the Hadamard product.
+
+    EXAMPLES:
+
+    This multiplication table can be verified by hand::
+
+        sage: J = eja_rn(3)
+        sage: e0,e1,e2 = J.gens()
+        sage: e0*e0
+        e0
+        sage: e0*e1
+        0
+        sage: e0*e2
+        0
+        sage: e1*e1
+        e1
+        sage: e1*e2
+        0
+        sage: e2*e2
+        e2
+
     """
     # The FiniteDimensionalAlgebra constructor takes a list of
     # matrices, the ith representing right multiplication by the ith
@@ -19,5 +63,58 @@ def eja_rn(dimension, field=QQ):
     # component of x; and likewise for the ith basis element e_i.
     Qs = [ matrix(field, dimension, dimension, lambda k,j: 1*(k == j == i))
            for i in xrange(dimension) ]
-    A = FiniteDimensionalAlgebra(QQ,Qs,assume_associative=True)
-    return JordanAlgebra(A)
+
+    # Assuming associativity is wrong here, but it works to
+    # temporarily trick the Jordan algebra constructor into using the
+    # multiplication table.
+    return FiniteDimensionalEuclideanJordanAlgebra(field,Qs)
+
+
+def eja_ln(dimension, field=QQ):
+    """
+    Return the Jordan algebra corresponding to the Lorentz "ice cream"
+    cone of the given ``dimension``.
+
+    EXAMPLES:
+
+    This multiplication table can be verified by hand::
+
+        sage: J = eja_ln(4)
+        sage: e0,e1,e2,e3 = J.gens()
+        sage: e0*e0
+        e0
+        sage: e0*e1
+        e1
+        sage: e0*e2
+        e2
+        sage: e0*e3
+        e3
+        sage: e1*e2
+        0
+        sage: e1*e3
+        0
+        sage: e2*e3
+        0
+
+    In one dimension, this is the reals under multiplication::
+
+      sage: J1 = eja_ln(1)
+      sage: J2 = eja_rn(1)
+      sage: J1 == J2
+      True
+
+    """
+    Qs = []
+    id_matrix = identity_matrix(field,dimension)
+    for i in xrange(dimension):
+        ei = id_matrix.column(i)
+        Qi = zero_matrix(field,dimension)
+        Qi.set_row(0, ei)
+        Qi.set_column(0, ei)
+        Qi += diagonal_matrix(dimension, [ei[0]]*dimension)
+        # The addition of the diagonal matrix adds an extra ei[0] in the
+        # upper-left corner of the matrix.
+        Qi[0,0] = Qi[0,0] * ~field(2)
+        Qs.append(Qi)
+
+    return FiniteDimensionalEuclideanJordanAlgebra(field,Qs)