Ticket #6949: trac_6949-symbolic_min_max.patch

sage/functions/all.py

@@ -62 +62 @@
 
 from generalized import (dirac_delta, heaviside, unit_step, sgn, 
                          kronecker_delta)
+
+from min_max import max_symbolic, min_symbolic

new file sage/functions/min_max.py

@@ +1 @@
+###############################################################################
+#   Sage: Open Source Mathematical Software
+#       Copyright (C) 2010 Burcin Erocal <burcin@erocal.org>
+#  Distributed under the terms of the GNU General Public License (GPL),
+#  version 2 or any later version.  The full text of the GPL is available at:
+#                  http://www.gnu.org/licenses/
+###############################################################################
+
+from sage.symbolic.function import BuiltinFunction
+from sage.symbolic.expression import Expression
+from sage.symbolic.ring import SR
+
+from __builtin__ import max as builtin_max, min as builtin_min
+
+class MaxSymbolic(BuiltinFunction):
+    def __init__(self):
+        r"""
+        Symbolic `max` function.
+
+        The Python builtin `max` function doesn't work as expected when symbolic
+        expressions are given as arguments. This function delays evaluation
+        until all symbolic arguments are substituted with values.
+
+        EXAMPLES::
+
+            sage: max_symbolic(3, x)
+            max(3, x)
+            sage: max_symbolic(3, x).subs(x=5)
+            5
+            sage: max_symbolic(3, 5, x)
+            max(x, 5)
+            sage: max_symbolic([3,5,x])
+            max(x, 5)
+
+        TESTS::
+
+            sage: loads(dumps(max_symbolic(x,5)))
+            max(x, 5)
+            sage: latex(max_symbolic(x,5))
+            \max\left(x, 5\right)
+        """
+        BuiltinFunction.__init__(self, 'max', nargs=0, latex_name="\max")
+
+    def _eval_(self, *args):
+        """
+        EXAMPLES::
+
+            sage: t = max_symbolic(x, 5); t
+            max(x, 5)
+            sage: t.subs(x=3) # indirect doctest
+            5
+            sage: max_symbolic(5,3)
+            5
+            sage: u = max_symbolic(*(range(10)+[x])); u
+            max(x, 9)
+            sage: u.subs(x=-1)
+            9
+            sage: u.subs(x=10)
+            10
+            sage: max_symbolic([0,x])
+            max(x, 0)
+
+        TESTS::
+
+            sage: max_symbolic()
+            Traceback (most recent call last):
+            ...
+            ValueError: number of arguments must be > 0
+        """
+        if len(args) == 0:
+            raise ValueError("number of arguments must be > 0")
+
+        # __call__ ensures that if args is a singleton, the element is iterable
+        arg_is_iter = False
+        if len(args) == 1:
+            arg_is_iter = True
+            args = args[0]
+
+        symb_args = []
+        res = None
+        num_non_symbolic_args = 0
+        for x in args:
+            if isinstance(x, Expression):
+                symb_args.append(x)
+            else:
+                num_non_symbolic_args += 1
+                res = builtin_max(res, x)
+
+        # if no symbolic arguments, return the result
+        if len(symb_args) == 0:
+            return res
+
+        # if all arguments were symbolic return 
+        if num_non_symbolic_args <= 1 and not arg_is_iter:
+            return None
+
+        symb_args.append(res)
+        return max_symbolic(*symb_args)
+
+    def __call__(self, *args, **kwds):
+        """
+        EXAMPLES::
+
+            sage: max_symbolic(3,5,x)
+            max(x, 5)
+            sage: max_symbolic(3,5,x, hold=True)
+            max(3, 5, x)
+            sage: max_symbolic([3,5,x])
+            max(x, 5)
+        """
+        if len(args) == 1:
+            try:
+                args=(SR._force_pyobject(iter(args[0])),)
+            except TypeError:
+                return args
+
+        return BuiltinFunction.__call__(self, *args, **kwds)
+
+max_symbolic = MaxSymbolic()
+
+
+class MinSymbolic(BuiltinFunction):
+    def __init__(self):
+        r"""
+        Symbolic `min` function.
+
+        The Python builtin `min` function doesn't work as expected when symbolic
+        expressions are given as arguments. This function delays evaluation
+        until all symbolic arguments are substituted with values.
+
+        EXAMPLES::
+
+            sage: min_symbolic(3, x)
+            min(3, x)
+            sage: min_symbolic(3, x).subs(x=5)
+            3
+            sage: min_symbolic(3, 5, x)
+            min(x, 3)
+            sage: min_symbolic([3,5,x])
+            min(x, 3)
+
+        TESTS::
+
+            sage: loads(dumps(min_symbolic(x,5)))
+            min(x, 5)
+            sage: latex(min_symbolic(x,5))
+            \min\left(x, 5\right)
+        """
+        BuiltinFunction.__init__(self, 'min', nargs=0, latex_name="\min")
+
+    def _eval_(self, *args):
+        """
+        EXAMPLES::
+
+            sage: t = min_symbolic(x, 5); t
+            min(x, 5)
+            sage: t.subs(x=3) # indirect doctest
+            3
+            sage: min_symbolic(5,3)
+            3
+            sage: u = min_symbolic(*(range(10)+[x])); u
+            min(x, 0)
+            sage: u.subs(x=-1)
+            -1 
+            sage: u.subs(x=10)
+            0
+            sage: min_symbolic([3,x])
+            min(x, 3)
+
+        TESTS::
+
+
+            sage: min_symbolic()
+            Traceback (most recent call last):
+            ...
+            ValueError: number of arguments must be > 0
+        """
+        if len(args) == 0:
+            raise ValueError("number of arguments must be > 0")
+
+        # __call__ ensures that if args is a singleton, the element is iterable
+        arg_is_iter = False
+        if len(args) == 1:
+            arg_is_iter = True
+            args = args[0]
+
+        symb_args = []
+        # set res to a value greater than anything
+        res = float('inf')
+        num_non_symbolic_args = 0
+        for x in args:
+            if isinstance(x, Expression):
+                symb_args.append(x)
+            else:
+                num_non_symbolic_args += 1
+                res = builtin_min(res, x)
+
+        # if no symbolic arguments, return the result
+        if len(symb_args) == 0:
+            return res
+
+        # if all arguments were symbolic return 
+        if num_non_symbolic_args <= 1 and not arg_is_iter:
+            return None
+
+        symb_args.append(res)
+        return min_symbolic(*symb_args)
+
+    def __call__(self, *args, **kwds):
+        """
+        EXAMPLES::
+
+            sage: min_symbolic(3,5,x)
+            min(x, 3)
+            sage: min_symbolic(3,5,x, hold=True)
+            min(3, 5, x)
+            sage: min_symbolic([3,5,x])
+            min(x, 3)
+        """
+        if len(args) == 1:
+            try:
+                args=(SR._force_pyobject(iter(args[0])),)
+            except TypeError:
+                return args
+
+        return BuiltinFunction.__call__(self, *args, **kwds)
+
+min_symbolic = MinSymbolic()

sage/symbolic/ring.pyx

@@ -271 +271 @@
 
         return new_Expression_from_GEx(self, exp)
 
+    def _force_pyobject(self, x):
+        """
+        Wrap the given Python object in a symbolic expression even if it
+        cannot be coerced to the Symbolic Ring.
+
+        EXAMPLES::
+
+            sage: t = SR._force_pyobject([3,4,5]); t
+            [3, 4, 5]
+            sage: type(t)
+            <type 'sage.symbolic.expression.Expression'>
+        """
+        cdef GEx exp
+        GEx_construct_pyobject(exp, x)
+        return new_Expression_from_GEx(self, exp)
+
     def wild(self, unsigned int n=0):
         """
         Return the n-th wild-card for pattern matching and substitution.
@@ -441 +457 @@
             t0
 
             sage: SR.symbol() # temporary variable
-            symbol160
+            symbol...
         """
         cdef GSymbol symb
         cdef Expression e