Ticket #3485: trac3485-sage_input.patch

a/sage/misc/all.py

@@ -63 +63 @@
-from interpreter import preparser
-
-from sage_eval import sage_eval, sageobj
+
+from sage_input import sage_input
-
-from cython import cython_lambda
-from cython_c import cython

/dev/null

@@ +1 @@
+r"""
+Sage Input Formatting
+
+AUTHORS:
+    -- Carl Witty (2008-04): new file
+
+This module provides a function \function{sage_input}.  This function
+takes an arbitrary \sage value and produces a sequence of commands
+that, if typed at the \code{sage:} prompt, will recreate the value.
+(If this is not implemented for a particular value, then an exception
+is raised instead.)  We attempt to produce commands that are readable 
+and idiomatic.
+
+sage: sage_input(3)
+3
+sage: sage_input((polygen(RR) + RR(pi))^2, verify=True)
+# Verified
+R.<x> = RR[]
+x^2 + 6.2831853071795862*x + 9.8696044010893580
+
+With \code{verify=True}, \function{sage_input} also verifies the
+results, by calling sage_eval on the result and verifying that it is
+equal to the input.
+
+sage: sage_input(GF(2)(1), verify=True)
+# Verified
+GF(2)(1)
+
+We can generate code that works without the preparser, with
+\code{preparse=False}; or we can generate code that will work
+whether or not the preparser is enabled, with \code{preparse=None}.
+
+sage: sage_input(5, verify=True)
+# Verified
+5
+sage: sage_input(5, preparse=False)
+ZZ(5)
+sage: sage_input(5, preparse=None)
+ZZ(5)
+sage: sage_input(5r, verify=True)
+# Verified
+5r
+sage: sage_input(5r, preparse=False)
+5
+sage: sage_input(5r, preparse=None)
+int(5)
+
+Adding \function{sage_input} support to your own classes is
+straightforward.  You need to add a \method{_sage_input_} method which
+returns a \class{SageInputExpression} (henceforth abbreviated as SIE)
+which will reconstruct this instance of your class.
+
+A \method{_sage_input_} method takes two parameters, conventionally
+named \var{sib} and \var{coerced}.  The first argument is a
+\class{SageInputBuilder}; it has methods to build SIEs.  The second
+argument, \var{coerced}, is a boolean.  This is only useful if your
+class is a subclass of \class{Element} (although it is always
+present).  If \var{coerced} is \code{False}, then your method must
+generate an expression which will evaluate to a value of the correct 
+type with the correct parent.  If \var{coerced} is \code{True}, then
+your method may generate an expression of a type that has a canonical
+coercion to your type.
+
+Let's work through some examples.  We'll build a sequence of functions
+that would be acceptable as \method{_sage_input_} methods for the
+\class{Rational} class.
+
+Here's the first and simplest version.
+
+sage: def qq_sage_input_v1(self, sib, coerced):
+...       return sib(self.numerator())/sib(self.denominator())
+
+We see that given a \class{SageInputBuilder} \var{sib}, you can
+construct a SIE for a value \var{v} simply with \code{sib(v)}, and you
+can construct a SIE for a quotient with the division operator.  Of course,
+the other operators also work, and so do function calls, method calls,
+subscripts, etc.
+
+We'll test with the following code, which you don't need to understand.
+(It produces a list of 8 results, showing the formatted versions of
+-5/7 and 3, with the preparser either enabled or disabled and either
+with or without an automatic coercion to QQ.)
+
+sage: from sage.misc.sage_input import SageInputBuilder
+sage: def test_qq_formatter(fmt):
+...       results = []
+...       for v in [-5/7, QQ(3)]:
+...           for pp in [False, True]:
+...               for coerced in [False, True]:
+...                   sib = SageInputBuilder(preparse=pp)
+...                   results.append(sib.result(fmt(v, sib, coerced)))
+...       return results
+
+sage: test_qq_formatter(qq_sage_input_v1)
+[-ZZ(5)/ZZ(7), -ZZ(5)/ZZ(7), -5/7, -5/7, ZZ(3)/ZZ(1), ZZ(3)/ZZ(1), 3/1, 3/1]
+
+Let's try for some shorter, perhaps nicer-looking output.  We'll start
+by getting rid of the \code{ZZ} in the denominators; even without the
+preparser, \code{-ZZ(5)/7 == -ZZ(5)/ZZ(7)}.
+
+sage: def qq_sage_input_v2(self, sib, coerced):
+...       return sib(self.numerator())/sib.int(self.denominator())
+
+The \method{int} method on \class{SageInputBuilder} returns a SIE for
+an integer that is always represented in the simple way, without
+coercions.  (So, depending on the preparser mode, it might read in as an
+\class{Integer}, an \class{int}, or a \class{long}.)
+
+sage: test_qq_formatter(qq_sage_input_v2)
+[-ZZ(5)/7, -ZZ(5)/7, -5/7, -5/7, ZZ(3)/1, ZZ(3)/1, 3/1, 3/1]
+
+Next let's get rid of the divisions by 1.  These are more complicated,
+since if we're not careful we'll get results in \ZZ instead of \QQ.
+
+sage: def qq_sage_input_v3(self, sib, coerced):
+...       if self.denominator() == 1:
+...           if coerced:
+...               return sib.int(self.numerator())
+...           else:
+...               return sib.name('QQ')(sib.int(self.numerator()))
+...       return sib(self.numerator())/sib.int(self.denominator())
+
+We see that the \method{name} method gives an SIE representing a \sage
+constant or function.
+
+sage: test_qq_formatter(qq_sage_input_v3)
+[-ZZ(5)/7, -ZZ(5)/7, -5/7, -5/7, QQ(3), 3, QQ(3), 3]
+
+This is the prettiest output we're going to get, but let's make one
+further refinement.  Other \class{_sage_input_} methods, like the one
+for polynomials, analyze the structure of SIEs; they work better (give
+prettier output) if negations are at the outside.
+
+sage: def qq_sage_input_v4(self, sib, coerced):
+...       num = self.numerator()
+...       neg = (num < 0)
+...       if neg: num = -num
+...       if self.denominator() == 1:
+...           if coerced:
+...               v = sib.int(num)
+...           else:
+...               v = sib.name('QQ')(sib.int(num))
+...       else:
+...           v = sib(num)/sib.int(self.denominator())
+...       if neg: v = -v
+...       return v
+
+sage: test_qq_formatter(qq_sage_input_v4)
+[-ZZ(5)/7, -ZZ(5)/7, -5/7, -5/7, QQ(3), 3, QQ(3), 3]
+
+"""
+
+from sage.misc.functional import parent
+
+##########################################################################
+#
+#       Copyright (C) 2008 Carl Witty <Carl.Witty@gmail.com>
+#
+#  Distributed under the terms of the GNU General Public License (GPL)
+#
+#                  http://www.gnu.org/licenses/
+#
+##########################################################################
+
+def sage_input(x, preparse=True, verify=False, allow_locals=False):
+    r"""
+    INPUTS:
+        x -- the value we want to find an ``input form'' for
+        preparse -- (default \code{True}) Whether to generate code
+            that requires the preparser.  With \code{True}, generated
+            code requires the preparser.  With \code{False}, generated
+            code requires that the preparser not be used.  With \code{None},
+            generated code will work whether or not the preparser is used.
+        verify -- (default \code{False}) If \code{True}, then the 
+            answer will be evaluated with \function{sage_eval}, and 
+            an exception will be raised if the result is not equal to
+            the original value.  (In fact, for \code{verify=True},
+            \function{sage_input} is effectively run three times,
+            with \var{preparse} set to \code{True}, \code{False}, and 
+            \code{None}, and all three results are checked.)  This is
+            particularly useful for doctests.
+        allow_locals -- (default \code{False}) If \code{True}, then
+            values that \function{sage_input} cannot handle are returned
+            in a dictionary, and the returned code assumes that this 
+            dictionary is passed as the \var{locals} parameter of 
+            \function{sage_eval}.  (Otherwise, if \function{sage_input}
+            cannot handle a value, an exception is raised.)
+
+    EXAMPLES:
+        sage: sage_input(GF(2)(1))
+        GF(2)(1)
+        sage: sage_input((GF(2)(0), GF(2)(1)), verify=True)
+        # Verified
+        GF_2 = GF(2)
+        (GF_2(0), GF_2(1))
+
+    When the preparser is enabled, we use the \sage generator syntax.
+
+        sage: K.<x> = GF(5)[]
+        sage: sage_input(x^3 + 2*x, verify=True)
+        # Verified
+        R.<x> = GF(5)[]
+        x^3 + 2*x
+        sage: sage_input(x^3 + 2*x, preparse=False)
+        R = GF(5)['x']
+        x = R.gen()
+        x**3 + 2*x
+
+    The result of \function{sage_input} is actually a pair of strings with
+    a special \method{__repr__} method to print nicely.
+
+        sage: r = sage_input(RealField(20)(pi), verify=True)
+        sage: r
+        # Verified
+        RealField(20)(3.1415939)
+        sage: isinstance(r, tuple)
+        True
+        sage: len(r)
+        2
+        sage: tuple(r)
+        ('# Verified\n', 'RealField(20)(3.1415939)')
+
+    We cannot find an input form for a function.
+
+        sage: sage_input((3, lambda x: x))
+        Traceback (most recent call last):
+        ...
+        ValueError: Can't convert <function <lambda> at 0x...> to sage_input form
+
+    But we can have \function{sage_input} continue anyway, and return
+    an input form for the rest of the expression, with 
+    \code{allow_locals=True}.
+
+        sage: r = sage_input((3, lambda x: x), verify=True, allow_locals=True)
+        sage: r
+        LOCALS:
+            _sil1: <function <lambda> at 0x...>
+        # Verified
+        (3, _sil1)
+        sage: tuple(r)
+        ('# Verified\n', '(3, _sil1)', {'_sil1': <function <lambda> at 0x...>})
+    """
+    if not verify:
+        sib = SageInputBuilder(allow_locals=allow_locals, preparse=preparse)
+        return sib.result(sib(x))
+
+    # In verify mode, we actually compute and verify the answer with
+    # all three settings of preparse.
+
+    for pp in (True, False, None):
+        sib = SageInputBuilder(allow_locals=allow_locals, preparse=pp)
+        ans = sib.result(sib(x))
+        verify_si_answer(x, ans, pp)
+        if pp == preparse:
+            ans_l = list(ans)
+            ans_l[0] = '# Verified\n' + ans_l[0]
+            final_answer = SageInputAnswer(*ans_l)
+
+    return final_answer
+
+class SageInputBuilder:
+    r"""
+    An instance of this class is passed to \method{_sage_input_} methods.
+    It keeps track of the current state of the \method{_sage_input_} process,
+    and contains many utility methods for building \class{SageInputExpression}
+    objects.
+
+    In normal use, instances of \class{SageInputBuilder} are created
+    internally by \function{sage_input}, but it may be useful to create
+    an instance directly for testing or doctesting.
+
+    EXAMPLES:
+        sage: from sage.misc.sage_input import SageInputBuilder
+        
+    We can create a \class{SageInputBuilder}, use it to create some
+    \class{SageInputExpression}s, and get a result.  (As mentioned
+    above, this is only useful for testing or doctesting; normally
+    you would just use \function{sage_input}.)
+
+        sage: sib = SageInputBuilder()
+        sage: sib.result((sib(3) + sib(4)) * (sib(5) + sib(6)))
+        (3 + 4)*(5 + 6)
+    """
+
+    def __init__(self, allow_locals=False, preparse=True):
+        r"""
+        Construct an instance of \class{SageInputBuilder}.
+
+        In normal use, instances of \class{SageInputBuilder} are created
+        internally by \function{sage_input}, but it may be useful to create
+        an instance directly for testing or doctesting.
+
+        INPUTS:
+            allow_locals -- (default \code{False}) If true, then values
+                that cannot be converted to input form will be stored in
+                a dictionary, which must be passed as the \var{locals}
+                when evaluating the result.
+            preparse -- (default \code{True}) If true, then the result
+                will assume that the preparser is enabled.  If false, then
+                the result will assume that the preparser is disabled.
+                If \code{None}, then the result will work whether or
+                not the preparser is enabled.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+            sage: SageInputBuilder().preparse()
+            True
+            sage: SageInputBuilder(preparse=False).preparse()
+            False
+        """
+        self._allow_locals = allow_locals
+        self._preparse = preparse
+        self._cached_types = set()
+        self._cache = {}
+        self._parent_gens = {}
+        self._next_local = 1
+        self._locals = {}
+
+    def __call__(self, x, coerced=False):
+        r"""
+        Tries to convert an arbitrary value \var{x} into a
+        \class{SageInputExpression} (an SIE).
+
+        We first check to see if an SIE has been cached for \var{x};
+        if so, we return it.  If \var{x} is already an SIE, we return
+        it unchanged.
+
+        If \var{x} has a \method{_sage_input_} method, we call that
+        method.
+
+        Otherwise, if \var{x} is a value of some Python type that
+        we know how to deal with, we convert it directly.
+
+        Finally, for values we don't know how to convert, if
+        \code{self._allow_locals} is true, we add it to a
+        ``locals'' dictionary.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib(sib(3)))
+            3
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib(GF(17)(5)))
+            GF(17)(5)
+
+        The argument \code{coerced=True} will get passed to the
+        \method{_sage_input_} method of the argument.
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib(GF(17)(5), True))
+            5
+
+        Since \function{sage_input} directly calls this method, all
+        of the following are indirect doctests.
+            sage: sage_input(True)
+            True
+            sage: sage_input(-5r)
+            -5r
+            sage: sage_input(7r, preparse=False)
+            7
+            sage: sage_input(-11r, preparse=None)
+            -int(11)
+            sage: sage_input(long(-5))
+            -long(5)
+            sage: sage_input(long(-7), preparse=False)
+            -7L
+            sage: sage_input(long(11), preparse=None)
+            long(11)
+            sage: sage_input(long(2^70))
+            1180591620717411303424r
+            sage: sage_input(-long(2^80), preparse=False)
+            -1208925819614629174706176
+            sage: sage_input(long(2^75), preparse=None)
+            long(37778931862957161709568)
+            sage: sage_input("Hello, world\n")
+            'Hello, world\n'
+            sage: sage_input((2, 3.5, 'Hi'))
+            (2, RR(3.5000000000000000), 'Hi')
+            sage: sage_input(lambda x: x)
+            Traceback (most recent call last):
+            ...
+            ValueError: Can't convert <function <lambda> at 0x...> to sage_input form
+            sage: sage_input(lambda x: x, allow_locals=True)
+            LOCALS:
+              _sil1: <function <lambda> at 0x...>
+            _sil1
+        """
+        if type(x) in self._cached_types:
+            v = self._cache.get((parent(x), x))
+            if v is not None: return v
+
+        if isinstance(x, SageInputExpression):
+            return x
+
+        if hasattr(x, '_sage_input_'):
+            return x._sage_input_(self, coerced)
+
+        if isinstance(x, bool):
+            return SIE_literal_stringrep(self, str(x))
+
+        if isinstance(x, int) or \
+                (isinstance(x, long) and isinstance(int(x), long)):
+            # For longs that don't fit in an int, we just use the int
+            # code; it will get extended to long automatically.            
+            if self._preparse == True:
+                if x < 0:
+                    return -SIE_literal_stringrep(self, str(-x) + 'r')
+                else:
+                    return SIE_literal_stringrep(self, str(x) + 'r')
+            elif self._preparse == False:
+                return self.int(x)
+            else:
+                tyname = 'int' if isinstance(x, int) else 'long'
+                if x < 0:
+                    return -self.name(tyname)(self.int(-x))
+                else:
+                    return self.name(tyname)(self.int(x))
+
+        if isinstance(x, long):
+            # This must be a long that does fit in an int, so we need either
+            # long(x) or an 'L' suffix.
+            # With the current preparser, 1Lr does not work.
+            # 1rL does work; but that's just ugly, so I don't use it.
+            if self._preparse == False:
+                if x < 0:
+                    return -SIE_literal_stringrep(self, str(-x) + 'L')
+                else:
+                    return SIE_literal_stringrep(self, str(x) + 'L')
+            else:
+                if x < 0:
+                    return -self.name('long')(self.int(-x))
+                else:
+                    return self.name('long')(self.int(x))            
+
+        if isinstance(x, str):
+            return SIE_literal_stringrep(self, repr(x))
+
+        if isinstance(x, tuple):
+            return SIE_tuple(self, map(self, x))
+
+        if self._allow_locals:
+            loc = self._next_local
+            self._next_local += 1
+            loc_name = '_sil%d' % loc
+            self._locals[loc_name] = x
+            return SIE_literal_stringrep(self, loc_name)
+        else:
+            raise ValueError, "Can't convert %r to sage_input form"%x
+
+    def preparse(self):
+        r"""
+        Checks the preparse status of this \class{SageInputBuilder}.
+        (\code{True} if the preparser will be enabled, \code{False}
+        if it will be disabled, and \code{None} if the result must
+        work whether or not the preparser is enabled.)
+
+        For example, this is useful in the \method{_sage_input_}
+        methods of \class{Integer} and \class{RealNumber}; but most
+        \method{_sage_input_} methods will not need to examine this.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+            sage: SageInputBuilder().preparse()
+            True
+            sage: SageInputBuilder(preparse=False).preparse()
+            False
+        """
+        return self._preparse
+
+    def int(self, n):
+        r"""
+        Given an integer (an \class{Integer}, an \class{int}, or a
+        \class{long}), produce a \class{SageInputExpression} that displays
+        the integer with no marking for what kind of integer it is
+        (so it may read back as an \class{Integer}, an \class{int}, or
+        a \class{long}, depending on its size and whether the preparser
+        is enabled).
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.int(-3^50))
+            -717897987691852588770249
+            
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.int(long(2^65)))
+            36893488147419103232
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.int(-42r))
+            -42
+        """
+        if n < 0:
+            return -SIE_literal_stringrep(self, -n)
+        else:
+            return SIE_literal_stringrep(self, n)
+
+    def float_str(self, n):
+        r"""
+        Given a string representing a floating-point number,
+        produces a \class{SageInputExpression} that formats as that
+        string.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.float_str(repr(RR(e))))
+            2.71828182845905
+        """
+        return SIE_literal_stringrep(self, n)
+
+    def name(self, n):
+        r"""
+        Given a string representing a Python name,
+        produces a \class{SageInputExpression} for that name.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.name('pi') + sib.name('e'))
+            pi + e
+        """
+        return SIE_literal_stringrep(self, n)
+
+    def cache(self, x, sie, name):
+        r"""
+        INPUTS:
+            x -- an arbitrary value
+            sie -- a \class{SageInputExpression}
+            name -- a requested variable name
+
+        Enters \var{x} and \var{sie} in a cache, so that subsequent calls
+        \code{self(x)} will directly return \var{sie}.  Also, marks the
+        requested name of this \var{sie} to be \var{name}.
+
+        This should almost always be called as part of the 
+        \method{_sage_input_} method of a parent.  It may also be called
+        on values of an arbitrary type, which may be useful if the values
+        are both large and likely to be used multiple times in a single
+        expression.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sie42 = sib(GF(101)(42))
+            sage: sib.cache(GF(101)(42), sie42, 'the_ultimate_answer')
+            sage: sib.result(sib(GF(101)(42)) + sib(GF(101)(42)))
+            the_ultimate_answer = GF(101)(42)
+            the_ultimate_answer + the_ultimate_answer
+
+        Note that we don't assign the result to a variable if the value
+        is only used once.
+            sage: sib = SageInputBuilder()
+            sage: sie42 = sib(GF(101)(42))
+            sage: sib.cache(GF(101)(42), sie42, 'the_ultimate_answer')
+            sage: sib.result(sib(GF(101)(42)) + sib(GF(101)(43)))
+            GF_101 = GF(101)
+            GF_101(42) + GF_101(43)
+        """
+        self._cached_types.add(type(x))
+        self._cache[(parent(x), x)] = sie
+        sie._sie_preferred_varname = name
+
+    def empty_subscript(self, parent):
+        r"""
+        Given a \class{SageInputExpression} representing \code{foo},
+        produces a \class{SageInputExpression} representing \code{foo[]}.
+        Since this is not legal Python syntax, it is useful only for
+        producing the \sage generator syntax for a polynomial ring.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.empty_subscript(sib(2) + sib(3)))
+            (2 + 3)[]
+
+        The following calls this method indirectly.
+            sage: sage_input(polygen(ZZ['y']))
+            R.<x> = ZZ['y'][]
+            x
+        """
+        return SIE_subscript(self, parent, None)
+
+    def parent_with_gens(self, parent, sie, gen_names, name, gens_syntax=None):
+        r"""
+        This method is used for parents with generators, to manage the
+        \sage preparser generator syntax (like \code{K.<x> = QQ[]}).
+
+        The \method{_sage_input_} method of a parent class with
+        generators should construct a \class{SageInputExpression} for
+        the parent, and then call this method with the parent itself,
+        the constructed SIE, a sequence containing the names of the
+        generators, and (optionally) another SIE to use if the \sage
+        generator syntax is used; typically this will be the same as
+        the first SIE except omitting a \var{names} parameter.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+
+            sage: def test_setup(use_gens=True, preparse=True):
+            ...       sib = SageInputBuilder(preparse=preparse)
+            ...       gen_names=('foo', 'bar')
+            ...       parent = "some parent"
+            ...       normal_sie = sib.name('make_a_parent')(names=gen_names)
+            ...       if use_gens:
+            ...           gens_sie = sib.name('make_a_parent')()
+            ...       else:
+            ...           gens_sie = None
+            ...       name = 'the_thing'
+            ...       result = sib.parent_with_gens(parent, normal_sie, 
+            ...                                     gen_names, name,
+            ...                                     gens_syntax=gens_sie)
+            ...       return sib, result
+
+            sage: sib, par_sie = test_setup()
+            sage: sib.result(par_sie)
+            make_a_parent(names=('foo', 'bar'))
+
+            sage: sib, par_sie = test_setup()
+            sage: sib.result(sib(3) * sib.gen("some parent", 0))
+            the_thing.<foo,bar> = make_a_parent()
+            3*foo
+
+            sage: sib, par_sie = test_setup(preparse=False)
+            sage: sib.result(par_sie)
+            make_a_parent(names=('foo', 'bar'))
+
+            sage: sib, par_sie = test_setup(preparse=False)
+            sage: sib.result(sib(3) * sib.gen("some parent", 0))
+            the_thing = make_a_parent(names=('foo', 'bar'))
+            foo,bar = the_thing.gens()
+            ZZ(3)*foo
+
+            sage: sib, par_sie = test_setup(use_gens=False)
+            sage: sib.result(par_sie)
+            make_a_parent(names=('foo', 'bar'))
+
+            sage: sib, par_sie = test_setup(use_gens=False)
+            sage: sib.result(sib(3) * sib.gen("some parent", 0))
+            the_thing = make_a_parent(names=('foo', 'bar'))
+            foo,bar = the_thing.gens()
+            3*foo
+
+            sage: sib, par_sie = test_setup()
+            sage: sib.result(par_sie - sib.gen("some parent", 1))
+            the_thing.<foo,bar> = make_a_parent()
+            the_thing - bar            
+        """
+        v = SIE_gens_constructor(self, sie, gen_names, gens_syntax=gens_syntax)
+        self.cache(parent, v, name)
+        gens = [SIE_gen(self, v, n) for n in gen_names]
+        self._parent_gens[parent] = gens
+        v._sie_gens = gens
+        return v
+
+    def gen(self, parent, n=0):
+        r"""
+        Given a parent, returns a \class{SageInputExpression} for
+        the $n$th (default 0) generator of the parent.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.gen(ZZ['y']))
+            R.<y> = ZZ[]
+            y
+        """
+        if not parent in self._parent_gens:
+            self(parent)
+            if not parent in self._parent_gens:
+                raise ValueError, "%s did not register generators for sage_input" % parent
+
+        gens = self._parent_gens[parent]
+
+        if n > len(gens):
+            raise ValueError, "%s registered only %d generators for sage_input" % (parent, len(gens))
+
+        return gens[n]
+        
+    def prod(self, factors, simplify=False):
+        r"""
+        Given a sequence, returns a \class{SageInputExpression}
+        for the product of the elements.
+
+        With \code{simplify=True}, performs some simplifications
+        first.  If any element is formatted as a string \code{'0'},
+        then that element is returned directly.  If any element is
+        formatted as a string \code{'1'}, then it is removed
+        from the sequence (unless it is the only element in the sequence).
+        And any negations are removed from the elements and moved to the
+        outside of the product.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.prod([-1, 0, 1, -2]))
+            -1*0*1*-2
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.prod([-1, 0, 1, 2], simplify=True))
+            0
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.prod([-1, 2, -3, -4], simplify=True))
+            -2*3*4
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.prod([-1, 1, -1, -1], simplify=True))
+            -1
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.prod([1, 1, 1], simplify=True))
+            1
+        """
+        neg = False
+        factors = [self(factor) for factor in factors]
+        if simplify:
+            i = 0
+            while i < len(factors):
+                factor = factors[i]
+                while isinstance(factor, SIE_unary) and factor._sie_op == '-':
+                    neg = not neg
+                    factor = factor._sie_operand
+                    factors[i] = factor
+                if isinstance(factor, SIE_literal_stringrep) and factor._sie_value == '0':
+                    factors = [factor]
+                    neg = False
+                    break
+                if isinstance(factor, SIE_literal_stringrep) and factor._sie_value == '1':
+                    factors[i:i+1] = []
+                else:
+                    i += 1
+            if len(factors) == 0:
+                factors.append(SIE_literal_stringrep(self, '1'))
+
+        prod = factors[0]
+        for factor in factors[1:]:
+            prod = prod * factor
+        if neg:
+            prod = -prod
+        return prod
+
+    def sum(self, terms, simplify=False):
+        r"""
+        Given a sequence, returns a \class{SageInputExpression}
+        for the product of the elements.
+
+        With \code{simplify=True}, performs some simplifications
+        first.  If any element is formatted as a string \code{'1'},
+        then it is removed from the sequence (unless it is the only
+        element in the sequence); and any instances of \code{a + -b} 
+        are changed to \code{a - b}.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.sum([-1, 0, 1, 0, -1]))
+            -1 + 0 + 1 + 0 + -1
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.sum([-1, 0, 1, 0, -1], simplify=True))
+            -1 + 1 - 1
+
+            sage: sib = SageInputBuilder()
+            sage: sib.result(sib.sum([0, 0, 0], simplify=True))
+            0
+        """
+        terms = [self(term) for term in terms]
+        if simplify:
+            i = 0
+            while i < len(terms):
+                term = terms[i]
+                if isinstance(term, SIE_literal_stringrep) and term._sie_value == '0':
+                    terms[i:i+1] = []
+                else:
+                    i += 1
+            if len(terms) == 0:
+                terms.append(SIE_literal_stringrep(self, '0'))
+
+        sum = terms[0]
+        for term in terms[1:]:
+            if simplify and isinstance(term, SIE_unary) and term._sie_op == '-':
+                sum = sum - term._sie_operand
+            else:
+                sum = sum + term
+        return sum
+
+    def result(self, e):
+        r"""
+        Given a \class{SageInputExpression} constructed using \code{self},
+        returns a tuple of a list of commands and an expression
+        (and possibly a dictionary of local variables) suitable for
+        \function{sage_eval}.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: r = sib.result(sib(6) * sib(7)); r
+            6*7
+            sage: tuple(r)
+            ('', '6*7')
+        """
+        sif = SageInputFormatter()
+
+        e._sie_prepare(sif)
+
+        s = sif.format(e, 0)
+
+        locals = self._locals
+        if len(locals):
+            return SageInputAnswer(sif._commands, sif.format(e, 0), locals)
+        else:
+            return SageInputAnswer(sif._commands, sif.format(e, 0))
+
+# Python's precedence levels.  Hand-transcribed from section 5.14 of 
+# the Python reference manual.
+_prec_lambda = 2
+_prec_or = 4
+_prec_and = 6
+_prec_not = 8
+_prec_membership = 10
+_prec_identity = 12
+_prec_comparison = 14
+_prec_bitor = 16
+_prec_bitxor = 18
+_prec_bitand = 20
+_prec_shift = 22
+_prec_addsub = 24
+_prec_muldiv = 26
+_prec_negate = 28
+_prec_not = 30
+_prec_exponent = 32
+_prec_attribute = 34
+_prec_subscript = 36
+_prec_slicing = 38
+_prec_funcall = 40
+_prec_atomic = 42
+
+class SageInputExpression(object):
+    r"""
+    Subclasses of this class represent expressions for \function{sage_input}.
+    \sage classes should define a \method{_sage_input_} method, which
+    will return an instance of \class{SageInputExpression}, created using
+    methods of \class{SageInputBuilder}.
+
+    To the extent possible, operations on \class{SageInputExpression} objects
+    construct a new \class{SageInputExpression} representing that operation.
+    That is, if \var{a} is a \class{SageInputExpression}, then \code{a + b}
+    constructs a \class{SageInputExpression} representing this sum.
+    This also works for attribute access, function calls, subscripts, etc.
+    Since arbitrary attribute accesses might be used to construct a new
+    attribte-access expression, all internal attributes and methods
+    have names that begin with \code{_sie_} to reduce the chance of
+    collisions.
+
+    It is expected that instances of this class will not be directly
+    created outside this module; instead, instances will be created
+    using methods of \class{SageInputBuilder} and \class{SageInputExpression}.
+    """
+
+    def __init__(self, sib):
+        r"""
+        Initialize a \class{SageInputExpression}.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sie = sib(3) # indirect doctest
+            sage: sie._sie_builder is sib
+            True
+        """
+        self._sie_refcount = 0
+        self._sie_builder = sib
+        self._sie_context = None
+        self._sie_preferred_varname = None
+        self._sie_varname = None
+        self._sie_use_var = False
+        self._sie_requested_varname = False
+
+    def _sie_is_simple(self):
+        r"""
+        Returns \code{True} if this \class{SageInputExpression} is simple
+        enough that duplicate uses are not worth caching.  Normally
+        this will be true if the expression represents a single token.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: sib.name('QQ')._sie_is_simple()
+            True
+            sage: sib(GF(2))._sie_is_simple()
+            False
+        """
+        return False
+
+    def _sie_referenced(self):
+        r"""
+        Returns a list of the immediate subexpressions of this
+        \class{SageInputExpression}.
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder
+
+            sage: sib = SageInputBuilder()
+            sage: len(sib(GF(2))._sie_referenced())
+            2
+            sage: sib(5)._sie_referenced()
+            []
+        """
+        return []
+
+    def _sie_prepare(self, sif):
+        r"""
+        We traverse the entire expression DAG to prepare for printing.
+        Here, we notice nodes with more than one parent, and mark them
+        to replace with a variable (rather than generating the value
+        multiple times).
+
+        EXAMPLES:
+            sage: from sage.misc.sage_input import SageInputBuilder, SageInputFormatter
+
+            sage: sib = SageInputBuilder()
+            sage: sif = SageInputFormatter()
+            sage: pair = sib((GF(2), GF(2)))
+            sage: single = sib(GF(2))
+            sage: single._sie_refcount
+            0
+            sage: single._sie_use_var
+            False
+            sage: sib((GF(2), GF(2)))._sie_prepare(sif)
+            sage: single._sie_refcount
+            2
+            sage: single._sie_use_var
+            True
+        """
+        if self._sie_context is not sif:
+            self._sie_context = sif
+            self._sie_refcount = 0
+        self._sie_refcount += 1
+        if not self._sie_is_simple():
+            if self._sie_refcount == 2:
+                self._sie_require_varname(sif)
+                self._sie_use_var = True
+        if self._sie_refcount == 1:
+            for r in self._sie_referenced():
+                r._sie_prepare(sif)
+
+    def _sie_require_varname(self, sif):
+        if not self._sie_requested_varname:
+            sif.register_name(self._sie_preferred_varname)
+            self._sie_requested_varname = True
+            self._sie_generated = False
+
+    def _sie_get_varname(self, sif):
+        if self._sie_varname is None:
+            self._sie_varname = sif.get_name(self._sie_preferred_varname)
+
+        return self._sie_varname
+
+    def __call__(self, *args, **kwargs):
+        args = map(self._sie_builder, args)
+        for k in kwargs:
+            kwargs[k] = self._sie_builder(kwargs[k])
+        return SIE_call(self._sie_builder, self, args, kwargs)
+            
+    def __getitem__(self, key):
+        skey = self._sie_builder(key)
+        return SIE_subscript(self._sie_builder, self, skey)
+
+    def __pow__(self, other):
+        return self._sie_binop('**', other)
+
+    def __mul__(self, other):
+        return self._sie_binop('*', other)
+
+    def __div__(self, other):
+        return self._sie_binop('/', other)
+
+    def __add__(self, other):
+        return self._sie_binop('+', other)
+
+    def __sub__(self, other):
+        return self._sie_binop('-', other)