a
    RG5dµ  ã                   @   sV   d Z ddlmZmZmZ ddlmZmZmZ ddl	m
Z
 ddd„Zdd	„ Zd
d„ ZdS )z1Gosper's algorithm for hypergeometric summation. é    )ÚSÚDummyÚsymbols)ÚPolyÚparallel_poly_from_exprÚfactor)Úis_sequenceTc                 C   sH  t | |f|ddd\\}}}| ¡ | ¡  }}| ¡ | ¡  }	}
|j||	  }}tdƒ}t|| |||jd}| |
 |¡¡}t	| 
¡  ¡ ƒ}t	|ƒD ]}|jr¨|dk r–| |¡ q–t|ƒD ]X}| |
 |
 ¡¡}| |¡}|
 | | ¡¡}
td|d ƒD ]}|| | ¡9 } qüq¼| |¡}|s>| ¡ }|
 ¡ }
| ¡ }||
|fS )a`  
    Compute the Gosper's normal form of ``f`` and ``g``.

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

    Given relatively prime univariate polynomials ``f`` and ``g``,
    rewrite their quotient to a normal form defined as follows:

    .. math::
        \frac{f(n)}{g(n)} = Z \cdot \frac{A(n) C(n+1)}{B(n) C(n)}

    where ``Z`` is an arbitrary constant and ``A``, ``B``, ``C`` are
    monic polynomials in ``n`` with the following properties:

    1. `\gcd(A(n), B(n+h)) = 1 \forall h \in \mathbb{N}`
    2. `\gcd(B(n), C(n+1)) = 1`
    3. `\gcd(A(n), C(n)) = 1`

    This normal form, or rational factorization in other words, is a
    crucial step in Gosper's algorithm and in solving of difference
    equations. It can be also used to decide if two hypergeometric
    terms are similar or not.

    This procedure will return a tuple containing elements of this
    factorization in the form ``(Z*A, B, C)``.

    Examples
    ========

    >>> from sympy.concrete.gosper import gosper_normal
    >>> from sympy.abc import n

    >>> gosper_normal(4*n+5, 2*(4*n+1)*(2*n+3), n, polys=False)
    (1/4, n + 3/2, n + 1/4)

    T)ÚfieldÚ	extensionÚh©Údomainr   é   )r   ÚLCÚmonicÚoner   r   r   Ú	resultantÚcomposeÚsetÚground_rootsÚkeysÚ
is_IntegerÚremoveÚsortedÚgcdÚshiftÚquoÚrangeÚ
mul_groundÚas_expr)ÚfÚgÚnÚpolysÚpÚqÚoptÚaÚAÚbÚBÚCÚZr   ÚDÚRÚrootsÚrÚiÚdÚj© r4   úQ/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/concrete/gosper.pyÚgosper_normal   s2    &ÿ

r6   c                 C   sÞ  ddl m} || |ƒ}|du r"dS | ¡ \}}t|||ƒ\}}}| d¡}t| ¡ ƒ}	t| ¡ ƒ}
t| ¡ ƒ}|	|
ks†| ¡ | ¡ kr˜|t|	|
ƒ h}nH|	s°||	 d tj	h}n0||	 d | 
|	d ¡| 
|	d ¡ | ¡  h}t|ƒD ]}|jrú|dk rè| |¡ qè|sdS t|ƒ}td|d  td}| ¡ j|Ž }t|||d}|| d¡ ||  | }dd	lm} || ¡ |ƒ}|du rŠdS | ¡  |¡}|D ]}||vrœ| |d¡}qœ|jrÆdS | ¡ | | ¡  S dS )
a&  
    Compute Gosper's hypergeometric term for ``f``.

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

    Suppose ``f`` is a hypergeometric term such that:

    .. math::
        s_n = \sum_{k=0}^{n-1} f_k

    and `f_k` does not depend on `n`. Returns a hypergeometric
    term `g_n` such that `g_{n+1} - g_n = f_n`.

    Examples
    ========

    >>> from sympy.concrete.gosper import gosper_term
    >>> from sympy import factorial
    >>> from sympy.abc import n

    >>> gosper_term((4*n + 1)*factorial(n)/factorial(2*n + 1), n)
    (-n - 1/2)/(n + 1/4)

    r   )Ú	hypersimpNéÿÿÿÿr   zc:%s)Úclsr   )Úsolve)Úsympy.simplifyr7   Úas_numer_denomr6   r   r   Údegreer   ÚmaxÚZeroÚnthr   r   r   r   r   Ú
get_domainÚinjectr   Úsympy.solvers.solversr:   Úcoeffsr   ÚsubsÚis_zero)r    r"   r7   r0   r$   r%   r(   r*   r+   ÚNÚMÚKr-   r2   rD   r   ÚxÚHr:   ÚsolutionÚcoeffr4   r4   r5   Úgosper_termS   sH    

0

rN   c                 C   s¶   d}t |ƒr|\}}}nd}t| |ƒ}|du r2dS |r@| | }nn| |d   ||¡| |  ||¡ }|tju r®z(| |d   ||¡| |  ||¡ }W n ty¬   d}Y n0 t|ƒS )aB  
    Gosper's hypergeometric summation algorithm.

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

    Given a hypergeometric term ``f`` such that:

    .. math ::
        s_n = \sum_{k=0}^{n-1} f_k

    and `f(n)` does not depend on `n`, returns `g_{n} - g(0)` where
    `g_{n+1} - g_n = f_n`, or ``None`` if `s_n` cannot be expressed
    in closed form as a sum of hypergeometric terms.

    Examples
    ========

    >>> from sympy.concrete.gosper import gosper_sum
    >>> from sympy import factorial
    >>> from sympy.abc import n, k

    >>> f = (4*k + 1)*factorial(k)/factorial(2*k + 1)
    >>> gosper_sum(f, (k, 0, n))
    (-factorial(n) + 2*factorial(2*n + 1))/factorial(2*n + 1)
    >>> _.subs(n, 2) == sum(f.subs(k, i) for i in [0, 1, 2])
    True
    >>> gosper_sum(f, (k, 3, n))
    (-60*factorial(n) + factorial(2*n + 1))/(60*factorial(2*n + 1))
    >>> _.subs(n, 5) == sum(f.subs(k, i) for i in [3, 4, 5])
    True

    References
    ==========

    .. [1] Marko Petkovsek, Herbert S. Wilf, Doron Zeilberger, A = B,
           AK Peters, Ltd., Wellesley, MA, USA, 1997, pp. 73--100

    FTNr   )r   rN   rE   r   ÚNaNÚlimitÚNotImplementedErrorr   )r    ÚkZ
indefiniter'   r)   r!   Úresultr4   r4   r5   Ú
gosper_sum¤   s     (

$
(
rT   N)T)Ú__doc__Ú
sympy.corer   r   r   Úsympy.polysr   r   r   Úsympy.utilities.iterablesr   r6   rN   rT   r4   r4   r4   r5   Ú<module>   s   
KQ