
    gX,                     b    d dl mZ d dlmZ d dlmZ  G d de          Z G d de          ZdS )	    )ExprWithLimits)S)Eqc                   "     e Zd ZdZ fdZ xZS )ReorderErrorzC
    Exception raised when trying to reorder dependent limits.
    c                 V    t                                          |d|d           d S )Nz could not be reordered: .)super__init__)selfexprmsg	__class__s      ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/concrete/expr_with_intlimits.pyr   zReorderError.__init__	   s9    04ccc:	< 	< 	< 	< 	<    )__name__
__module____qualname____doc__r   __classcell__)r   s   @r   r   r      sB         < < < < < < < < <r   r   c                   F    e Zd ZdZdZd	dZd Zd Zd Ze	d             Z
dS )
ExprWithIntLimitsz
    Superclass for Product and Sum.

    See Also
    ========

    sympy.concrete.expr_with_limits.ExprWithLimits
    sympy.concrete.products.Product
    sympy.concrete.summations.Sum
     Nc                 j   ||}g }| j         D ]`}|d         |k    r;|                    |          }|                                dk    rt          d          |                    |          }|                    t
          j                  }|j        r|t
          j        k    r1|                    |||d         z  |z   ||d         z  |z   f           |t
          j	        k    r2|                    |||d         z  |z   ||d         z  |z   f           
t          d          |                    |||d         z  |z   ||d         z  |z   f           K|                    |           b| j
                            |||z
  |z            }	|	                    ||          }	 | j        |	g|R  S )a  
        Change index of a Sum or Product.

        Perform a linear transformation `x \mapsto a x + b` on the index variable
        `x`. For `a` the only values allowed are `\pm 1`. A new variable to be used
        after the change of index can also be specified.

        Explanation
        ===========

        ``change_index(expr, var, trafo, newvar=None)`` where ``var`` specifies the
        index variable `x` to transform. The transformation ``trafo`` must be linear
        and given in terms of ``var``. If the optional argument ``newvar`` is
        provided then ``var`` gets replaced by ``newvar`` in the final expression.

        Examples
        ========

        >>> from sympy import Sum, Product, simplify
        >>> from sympy.abc import x, y, a, b, c, d, u, v, i, j, k, l

        >>> S = Sum(x, (x, a, b))
        >>> S.doit()
        -a**2/2 + a/2 + b**2/2 + b/2

        >>> Sn = S.change_index(x, x + 1, y)
        >>> Sn
        Sum(y - 1, (y, a + 1, b + 1))
        >>> Sn.doit()
        -a**2/2 + a/2 + b**2/2 + b/2

        >>> Sn = S.change_index(x, -x, y)
        >>> Sn
        Sum(-y, (y, -b, -a))
        >>> Sn.doit()
        -a**2/2 + a/2 + b**2/2 + b/2

        >>> Sn = S.change_index(x, x+u)
        >>> Sn
        Sum(-u + x, (x, a + u, b + u))
        >>> Sn.doit()
        -a**2/2 - a*u + a/2 + b**2/2 + b*u + b/2 - u*(-a + b + 1) + u
        >>> simplify(Sn.doit())
        -a**2/2 + a/2 + b**2/2 + b/2

        >>> Sn = S.change_index(x, -x - u, y)
        >>> Sn
        Sum(-u - y, (y, -b - u, -a - u))
        >>> Sn.doit()
        -a**2/2 - a*u + a/2 + b**2/2 + b*u + b/2 - u*(-a + b + 1) + u
        >>> simplify(Sn.doit())
        -a**2/2 + a/2 + b**2/2 + b/2

        >>> P = Product(i*j**2, (i, a, b), (j, c, d))
        >>> P
        Product(i*j**2, (i, a, b), (j, c, d))
        >>> P2 = P.change_index(i, i+3, k)
        >>> P2
        Product(j**2*(k - 3), (k, a + 3, b + 3), (j, c, d))
        >>> P3 = P2.change_index(j, -j, l)
        >>> P3
        Product(l**2*(k - 3), (k, a + 3, b + 3), (l, -d, -c))

        When dealing with symbols only, we can make a
        general linear transformation:

        >>> Sn = S.change_index(x, u*x+v, y)
        >>> Sn
        Sum((-v + y)/u, (y, b*u + v, a*u + v))
        >>> Sn.doit()
        -v*(a*u - b*u + 1)/u + (a**2*u**2/2 + a*u*v + a*u/2 - b**2*u**2/2 - b*u*v + b*u/2 + v)/u
        >>> simplify(Sn.doit())
        a**2*u/2 + a/2 - b**2*u/2 + b/2

        However, the last result can be inconsistent with usual
        summation where the index increment is always 1. This is
        obvious as we get back the original value only for ``u``
        equal +1 or -1.

        See Also
        ========

        sympy.concrete.expr_with_intlimits.ExprWithIntLimits.index,
        reorder_limit,
        sympy.concrete.expr_with_intlimits.ExprWithIntLimits.reorder,
        sympy.concrete.summations.Sum.reverse_order,
        sympy.concrete.products.Product.reverse_order
        Nr      z"Index transformation is not linear   z>Linear transformation results in non-linear summation stepsize)limitsas_polydegree
ValueErrorcoeff_monomialr   One	is_numberappendNegativeOnefunctionsubsfunc)
r   vartrafonewvarr   limitpalphabetar&   s
             r   change_indexzExprWithIntLimits.change_index   s   r >F[ 	% 	%EQx3MM#&&88::??$%IJJJ((--''..? 	Z~~vuU1X~/DeERSHnW[F[&\]]]]!-//vuU1X~/DeERSHnW[F[&\]]]]()ijjj MM65q>D+@%a.SWBW"XYYYYe$$$$=%%cC$J+=>>==f--ty+F++++r   c                     d | j         D             }|                    |          dk    rt          | d          |                    |          S )aX  
        Return the index of a dummy variable in the list of limits.

        Explanation
        ===========

        ``index(expr, x)``  returns the index of the dummy variable ``x`` in the
        limits of ``expr``. Note that we start counting with 0 at the inner-most
        limits tuple.

        Examples
        ========

        >>> from sympy.abc import x, y, a, b, c, d
        >>> from sympy import Sum, Product
        >>> Sum(x*y, (x, a, b), (y, c, d)).index(x)
        0
        >>> Sum(x*y, (x, a, b), (y, c, d)).index(y)
        1
        >>> Product(x*y, (x, a, b), (y, c, d)).index(x)
        0
        >>> Product(x*y, (x, a, b), (y, c, d)).index(y)
        1

        See Also
        ========

        reorder_limit, reorder, sympy.concrete.summations.Sum.reverse_order,
        sympy.concrete.products.Product.reverse_order
        c                     g | ]
}|d          S r   r   .0r,   s     r   
<listcomp>z+ExprWithIntLimits.index.<locals>.<listcomp>   s    777%U1X777r   r   z0Number of instances of variable not equal to one)r   countr    index)r   x	variabless      r   r8   zExprWithIntLimits.index   sS    > 874;777	??1""T#UVVV??1%%%r   c                 ~   | }|D ]}t          |          dk    rt          |d          |d         }|d         }t          |d         t                    s|                     |d                   }t          |d         t                    s|                     |d                   }|                    ||          }|S )a  
        Reorder limits in a expression containing a Sum or a Product.

        Explanation
        ===========

        ``expr.reorder(*arg)`` reorders the limits in the expression ``expr``
        according to the list of tuples given by ``arg``. These tuples can
        contain numerical indices or index variable names or involve both.

        Examples
        ========

        >>> from sympy import Sum, Product
        >>> from sympy.abc import x, y, z, a, b, c, d, e, f

        >>> Sum(x*y, (x, a, b), (y, c, d)).reorder((x, y))
        Sum(x*y, (y, c, d), (x, a, b))

        >>> Sum(x*y*z, (x, a, b), (y, c, d), (z, e, f)).reorder((x, y), (x, z), (y, z))
        Sum(x*y*z, (z, e, f), (y, c, d), (x, a, b))

        >>> P = Product(x*y*z, (x, a, b), (y, c, d), (z, e, f))
        >>> P.reorder((x, y), (x, z), (y, z))
        Product(x*y*z, (z, e, f), (y, c, d), (x, a, b))

        We can also select the index variables by counting them, starting
        with the inner-most one:

        >>> Sum(x**2, (x, a, b), (x, c, d)).reorder((0, 1))
        Sum(x**2, (x, c, d), (x, a, b))

        And of course we can mix both schemes:

        >>> Sum(x*y, (x, a, b), (y, c, d)).reorder((y, x))
        Sum(x*y, (y, c, d), (x, a, b))
        >>> Sum(x*y, (x, a, b), (y, c, d)).reorder((y, 0))
        Sum(x*y, (y, c, d), (x, a, b))

        See Also
        ========

        reorder_limit, index, sympy.concrete.summations.Sum.reverse_order,
        sympy.concrete.products.Product.reverse_order
        r   zInvalid number of argumentsr   r   )lenr    
isinstanceintr8   reorder_limit)r   argnew_exprrindex1index2s         r   reorderzExprWithIntLimits.reorder   s    \  	> 	>A1vv{{ $ABBBqTFqTFadC(( *AaD))adC(( *AaD))--ff==HHr   c                 t   d | j         D             }| j         |         }| j         |         }t          t          |d         j                                      |                    dk    r?t          t          |d         j                                      |                    dk    r t          t          |d         j                                      |                    dk    rt          t          |d         j                                      |                    dk    rg }t          | j                   D ]R\  }}||k    r|                    |           !||k    r|                    |           =|                    |           S t          |           | j        g|R  S t          | d          )a-  
        Interchange two limit tuples of a Sum or Product expression.

        Explanation
        ===========

        ``expr.reorder_limit(x, y)`` interchanges two limit tuples. The
        arguments ``x`` and ``y`` are integers corresponding to the index
        variables of the two limits which are to be interchanged. The
        expression ``expr`` has to be either a Sum or a Product.

        Examples
        ========

        >>> from sympy.abc import x, y, z, a, b, c, d, e, f
        >>> from sympy import Sum, Product

        >>> Sum(x*y*z, (x, a, b), (y, c, d), (z, e, f)).reorder_limit(0, 2)
        Sum(x*y*z, (z, e, f), (y, c, d), (x, a, b))
        >>> Sum(x**2, (x, a, b), (x, c, d)).reorder_limit(1, 0)
        Sum(x**2, (x, c, d), (x, a, b))

        >>> Product(x*y*z, (x, a, b), (y, c, d), (z, e, f)).reorder_limit(0, 2)
        Product(x*y*z, (z, e, f), (y, c, d), (x, a, b))

        See Also
        ========

        index, reorder, sympy.concrete.summations.Sum.reverse_order,
        sympy.concrete.products.Product.reverse_order
        c                     h | ]
}|d          S r3   r   r4   s     r   	<setcomp>z2ExprWithIntLimits.reorder_limit.<locals>.<setcomp>  s    111EuQx111r   r   r   r   z.could not interchange the two limits specified)
r   r<   setfree_symbolsintersection	enumerater$   typer&   r   )	r   r9   yr)   limit_xlimit_yr   ir,   s	            r   r?   zExprWithIntLimits.reorder_limit   s   @ 21T[111+a.+a.GAJ+,,99#>>??1DDGAJ+,,99#>>??1DDGAJ+,,99#>>??1DDGAJ+,,99#>>??1DDF%dk22 ) )566MM'****!VVMM'****MM%((((4::dm5f5555t%UVVVr   c                     d}| j         D ]5}|d         |d         z
  }t          |d          }|dk    r dS |dk    r3d}6|rdS dS )a  
        Returns True if the Sum or Product is computed for an empty sequence.

        Examples
        ========

        >>> from sympy import Sum, Product, Symbol
        >>> m = Symbol('m')
        >>> Sum(m, (m, 1, 0)).has_empty_sequence
        True

        >>> Sum(m, (m, 1, 1)).has_empty_sequence
        False

        >>> M = Symbol('M', integer=True, positive=True)
        >>> Product(m, (m, 1, M)).has_empty_sequence
        False

        >>> Product(m, (m, 2, M)).has_empty_sequence

        >>> Product(m, (m, M + 1, M)).has_empty_sequence
        True

        >>> N = Symbol('N', integer=True, positive=True)
        >>> Sum(m, (m, N, M)).has_empty_sequence

        >>> N = Symbol('N', integer=True, negative=True)
        >>> Sum(m, (m, N, M)).has_empty_sequence
        False

        See Also
        ========

        has_reversed_limits
        has_finite_limits

        Fr   r   TN)r   r   )r   ret_Nonelimdifeqs        r   has_empty_sequencez$ExprWithIntLimits.has_empty_sequence.  sn    N ; 	  	 Ca&3q6/CCBTzzttu 	4ur   )N)r   r   r   r   	__slots__r0   r8   rE   r?   propertyrW   r   r   r   r   r      s        	 	 It, t, t, t,n$& $& $&L> > >B4W 4W 4Wl 3 3 X3 3 3r   r   N)	sympy.concrete.expr_with_limitsr   sympy.core.singletonr   sympy.core.relationalr   NotImplementedErrorr   r   r   r   r   <module>r^      s    : : : : : : " " " " " " $ $ $ $ $ $< < < < <& < < <U U U U U U U U U Ur   