eja: special subalgebra handling for Cartesian product EJAs.

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

diff --git a/mjo/eja/eja_subalgebra.py b/mjo/eja/eja_subalgebra.py
index ca8efa1fd410b8f5f3f6d177b62779b1a24ccaf5..c9abada53cf3fbd3803601fa0e7b430a0b506fd1 100644 (file)
--- a/mjo/eja/eja_subalgebra.py
+++ b/mjo/eja/eja_subalgebra.py
@@ -2,7 +2,8 @@ from sage.matrix.constructor import matrix
 from sage.misc.cachefunc import cached_method
 
 from mjo.eja.eja_algebra import FiniteDimensionalEJA
-from mjo.eja.eja_element import FiniteDimensionalEJAElement
+from mjo.eja.eja_element import (FiniteDimensionalEJAElement,
+                                 CartesianProductParentEJAElement)
 
 class FiniteDimensionalEJASubalgebraElement(FiniteDimensionalEJAElement):
     """
@@ -265,3 +266,41 @@ class FiniteDimensionalEJASubalgebra(FiniteDimensionalEJA):
 
 
     Element = FiniteDimensionalEJASubalgebraElement
+
+
+
+class FiniteDimensionalCartesianProductEJASubalgebraElement(FiniteDimensionalEJASubalgebraElement, CartesianProductParentEJAElement):
+    r"""
+    The class for elements that both belong to a subalgebra and
+    have a Cartesian product algebra as their parent. By inheriting
+    :class:`CartesianProductParentEJAElement` in addition to
+    :class:`FiniteDimensionalEJASubalgebraElement`, we allow the
+    ``to_matrix()`` method to be overridden with the version that
+    works on Cartesian products.
+
+    SETUP::
+
+        sage: from mjo.eja.eja_algebra import (HadamardEJA,
+        ....:                                  RealSymmetricEJA)
+
+    TESTS:
+
+    This used to fail when ``subalgebra_idempotent()`` tried to
+    embed the subalgebra element back into the original EJA::
+
+        sage: J1 = HadamardEJA(0, field=QQ, orthonormalize=False)
+        sage: J2 = RealSymmetricEJA(2, field=QQ, orthonormalize=False)
+        sage: J = cartesian_product([J1,J2])
+        sage: J.one().subalgebra_idempotent() == J.one()
+        True
+
+    """
+    pass
+
+class FiniteDimensionalCartesianProductEJASubalgebra(FiniteDimensionalEJASubalgebra):
+    r"""
+    Subalgebras whose parents are Cartesian products. Exists only
+    to specify a special element class that will (in addition)
+    inherit from ``CartesianProductParentEJAElement``.
+    """
+    Element = FiniteDimensionalCartesianProductEJASubalgebraElement