X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=mjo%2Feja%2Feja_utils.py;h=1cf13bc640fc39d6840bd745128a8bfe510ccf74;hb=86bf0d59519eef1259fa9fb82457148b28c2fa2e;hp=7c6c581329dd64aaa8f3d958e00b2f9eb3221fcb;hpb=f38e111cd026f1ecdab6dd2d2ed194a8745252a8;p=sage.d.git

diff --git a/mjo/eja/eja_utils.py b/mjo/eja/eja_utils.py
index 7c6c581..1cf13bc 100644
--- a/mjo/eja/eja_utils.py
+++ b/mjo/eja/eja_utils.py
@@ -2,6 +2,42 @@ from sage.functions.other import sqrt
 from sage.matrix.constructor import matrix
 from sage.modules.free_module_element import vector
 
+def _charpoly_sage_input(s):
+    r"""
+    Helper function that you can use on the string output from sage
+    to convert a charpoly coefficient into the corresponding input
+    to be cached.
+
+    SETUP::
+
+        sage: from mjo.eja.eja_algebra import JordanSpinEJA
+        sage: from mjo.eja.eja_utils import _charpoly_sage_input
+
+    EXAMPLES::
+
+        sage: J = JordanSpinEJA(4,QQ)
+        sage: a = J._charpoly_coefficients()
+        sage: a[0]
+        X1^2 - X2^2 - X3^2 - X4^2
+        sage: _charpoly_sage_input(str(a[0]))
+        'X[0]**2 - X[1]**2 - X[2]**2 - X[3]**2'
+
+    """
+    import re
+
+    exponent_out = r"\^"
+    exponent_in = r"**"
+
+    digit_out = r"X([0-9]+)"
+
+    def replace_digit(m):
+        # m is a match object
+        return "X[" + str(int(m.group(1)) - 1) + "]"
+
+    s = re.sub(exponent_out, exponent_in, s)
+    return re.sub(digit_out, replace_digit, s)
+
+
 def _scale(x, alpha):
     r"""
     Scale the vector, matrix, or cartesian-product-of-those-things