a
    <b$                     @   s  d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZ d d	lmZ d d
lmZmZ d dlmZ d dlmZmZmZ d dlmZ d dlmZmZ d dl m!Z!m"Z" d dl#m$Z$ d dl%m&Z& d dl'm(Z(m)Z)m*Z* d dl
m+Z+ G dd deZ,G dd de,edZ-G dd de,Z.G dd de.Z/G dd de.Z0G dd  d e,Z1d*d"d#Z2G d$d% d%e,Z3G d&d' d'e3Z4G d(d) d)e3Z5d!S )+    )Basic)cacheit)Tuple)call_highest_priority)global_parameters)AppliedUndefMul)Integer)Eq)S	Singleton)ordered)DummySymbolWildsympify)lcmfactor)IntervalIntersection)simplify)Idx)flattenis_sequenceiterable)expandc                   @   s  e Zd ZdZdZdZedd Zdd Ze	dd	 Z
e	d
d Ze	dd Ze	dd Ze	dd Ze	dd Ze	dd Zedd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zed$d%d& Zd'd( Zed)d*d+ Zd,d- Zd.d/ Zed0d1d2 Zd3d4 Z d5d6 Z!d:d8d9Z"d7S );SeqBasezBase class for sequencesT   c              
   C   s.   z
| j }W n tttfy(   tj}Y n0 |S )z[Return start (if possible) else S.Infinity.

        adapted from Set._infimum_key
        )startNotImplementedErrorAttributeError
ValueErrorr   Infinity)exprr     r&   f/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/series/sequences.py
_start_key!   s    

zSeqBase._start_keyc                 C   s   t | j|j}|j|jfS )zTReturns start and stop.

        Takes intersection over the two intervals.
        )r   intervalinfsup)selfotherr)   r&   r&   r'   _intersect_interval.   s    zSeqBase._intersect_intervalc                 C   s   t d|  dS )z&Returns the generator for the sequencez(%s).genNr!   r,   r&   r&   r'   gen6   s    zSeqBase.genc                 C   s   t d|  dS )z-The interval on which the sequence is definedz(%s).intervalNr/   r0   r&   r&   r'   r)   ;   s    zSeqBase.intervalc                 C   s   t d|  dS ):The starting point of the sequence. This point is includedz
(%s).startNr/   r0   r&   r&   r'   r    @   s    zSeqBase.startc                 C   s   t d|  dS )z8The ending point of the sequence. This point is includedz	(%s).stopNr/   r0   r&   r&   r'   stopE   s    zSeqBase.stopc                 C   s   t d|  dS )zLength of the sequencez(%s).lengthNr/   r0   r&   r&   r'   lengthJ   s    zSeqBase.lengthc                 C   s   dS )z-Returns a tuple of variables that are boundedr&   r&   r0   r&   r&   r'   	variablesO   s    zSeqBase.variablesc                    s    fdd j D S )aG  
        This method returns the symbols in the object, excluding those
        that take on a specific value (i.e. the dummy symbols).

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n, m
        >>> SeqFormula(m*n**2, (n, 0, 5)).free_symbols
        {m}
        c                    s$   h | ]}|j  jD ]}|qqS r&   )free_symbols
differencer5   ).0ijr0   r&   r'   	<setcomp>b   s   z'SeqBase.free_symbols.<locals>.<setcomp>argsr0   r&   r0   r'   r6   T   s    zSeqBase.free_symbolsc                 C   s0   || j k s|| jkr&td|| jf | |S )z#Returns the coefficient at point ptzIndex %s out of bounds %s)r    r3   
IndexErrorr)   _eval_coeffr,   ptr&   r&   r'   coeffe   s    zSeqBase.coeffc                 C   s   t d| j d S )NzhThe _eval_coeff method should be added to%s to return coefficient so it is availablewhen coeff calls it.)r!   funcr@   r&   r&   r'   r?   l   s    zSeqBase._eval_coeffc                 C   s<   | j tju r| j}n| j }| j tju r,d}nd}|||  S )a  Returns the i'th point of a sequence.

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

        If start point is negative infinity, point is returned from the end.
        Assumes the first point to be indexed zero.

        Examples
        =========

        >>> from sympy import oo
        >>> from sympy.series.sequences import SeqPer

        bounded

        >>> SeqPer((1, 2, 3), (-10, 10))._ith_point(0)
        -10
        >>> SeqPer((1, 2, 3), (-10, 10))._ith_point(5)
        -5

        End is at infinity

        >>> SeqPer((1, 2, 3), (0, oo))._ith_point(5)
        5

        Starts at negative infinity

        >>> SeqPer((1, 2, 3), (-oo, 0))._ith_point(5)
        -5
           )r    r   NegativeInfinityr3   )r,   r9   initialstepr&   r&   r'   
_ith_pointr   s     zSeqBase._ith_pointc                 C   s   dS )aI  
        Should only be used internally.

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

        self._add(other) returns a new, term-wise added sequence if self
        knows how to add with other, otherwise it returns ``None``.

        ``other`` should only be a sequence object.

        Used within :class:`SeqAdd` class.
        Nr&   r,   r-   r&   r&   r'   _add   s    zSeqBase._addc                 C   s   dS )aS  
        Should only be used internally.

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

        self._mul(other) returns a new, term-wise multiplied sequence if self
        knows how to multiply with other, otherwise it returns ``None``.

        ``other`` should only be a sequence object.

        Used within :class:`SeqMul` class.
        Nr&   rJ   r&   r&   r'   _mul   s    zSeqBase._mulc                 C   s
   t | |S )a  
        Should be used when ``other`` is not a sequence. Should be
        defined to define custom behaviour.

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n
        >>> SeqFormula(n**2).coeff_mul(2)
        SeqFormula(2*n**2, (n, 0, oo))

        Notes
        =====

        '*' defines multiplication of sequences with sequences only.
        r   rJ   r&   r&   r'   	coeff_mul   s    zSeqBase.coeff_mulc                 C   s$   t |tstdt| t| |S )a4  Returns the term-wise addition of 'self' and 'other'.

        ``other`` should be a sequence.

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n
        >>> SeqFormula(n**2) + SeqFormula(n**3)
        SeqFormula(n**3 + n**2, (n, 0, oo))
        zcannot add sequence and %s
isinstancer   	TypeErrortypeSeqAddrJ   r&   r&   r'   __add__   s    
zSeqBase.__add__rS   c                 C   s   | | S Nr&   rJ   r&   r&   r'   __radd__   s    zSeqBase.__radd__c                 C   s&   t |tstdt| t| | S )a7  Returns the term-wise subtraction of ``self`` and ``other``.

        ``other`` should be a sequence.

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n
        >>> SeqFormula(n**2) - (SeqFormula(n))
        SeqFormula(n**2 - n, (n, 0, oo))
        zcannot subtract sequence and %srN   rJ   r&   r&   r'   __sub__   s    
zSeqBase.__sub__rV   c                 C   s
   |  | S rT   r&   rJ   r&   r&   r'   __rsub__   s    zSeqBase.__rsub__c                 C   s
   |  dS )zNegates the sequence.

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n
        >>> -SeqFormula(n**2)
        SeqFormula(-n**2, (n, 0, oo))
        rD   )rM   r0   r&   r&   r'   __neg__   s    zSeqBase.__neg__c                 C   s$   t |tstdt| t| |S )a{  Returns the term-wise multiplication of 'self' and 'other'.

        ``other`` should be a sequence. For ``other`` not being a
        sequence see :func:`coeff_mul` method.

        Examples
        ========

        >>> from sympy import SeqFormula
        >>> from sympy.abc import n
        >>> SeqFormula(n**2) * (SeqFormula(n))
        SeqFormula(n**3, (n, 0, oo))
        zcannot multiply sequence and %s)rO   r   rP   rQ   SeqMulrJ   r&   r&   r'   __mul__	  s    
zSeqBase.__mul__rZ   c                 C   s   | | S rT   r&   rJ   r&   r&   r'   __rmul__  s    zSeqBase.__rmul__c                 c   s*   t | jD ]}| |}| |V  q
d S rT   )ranger4   rI   rB   )r,   r9   rA   r&   r&   r'   __iter__  s    
zSeqBase.__iter__c                    st   t |tr |} |S t |trp|j|j }}|d u rBd}|d u rP j} fddt|||j	phdD S d S )Nr   c                    s   g | ]}   |qS r&   )rB   rI   )r8   r9   r0   r&   r'   
<listcomp>.      z'SeqBase.__getitem__.<locals>.<listcomp>rE   )
rO   intrI   rB   slicer    r3   r4   r\   rH   )r,   indexr    r3   r&   r0   r'   __getitem__$  s    




zSeqBase.__getitem__Nc                 C   s8  ddl m} dd | d| D }t|}|du r<|d }nt||d }g }td|d D ]}	d|	 }
g }t|	D ]}|||||	   qt||}| dkr\t||||	|
 }||
krt	|ddd } qFg }t|	||	 D ]}|||||	   q||}|| |||
d kr\t	|ddd } qFq\|du rT|S t|}	|	dkrng dfS ||	d  ||	d   d||	d  ||	    }}t|	d D ]n}||| ||  7 }t|	| d D ]*}||| ||  ||| d   8 }q||| ||d   8 }q|tt
|t
| fS dS )	a  
        Finds the shortest linear recurrence that satisfies the first n
        terms of sequence of order `\leq` ``n/2`` if possible.
        If ``d`` is specified, find shortest linear recurrence of order
        `\leq` min(d, n/2) if possible.
        Returns list of coefficients ``[b(1), b(2), ...]`` corresponding to the
        recurrence relation ``x(n) = b(1)*x(n-1) + b(2)*x(n-2) + ...``
        Returns ``[]`` if no recurrence is found.
        If gfvar is specified, also returns ordinary generating function as a
        function of gfvar.

        Examples
        ========

        >>> from sympy import sequence, sqrt, oo, lucas
        >>> from sympy.abc import n, x, y
        >>> sequence(n**2).find_linear_recurrence(10, 2)
        []
        >>> sequence(n**2).find_linear_recurrence(10)
        [3, -3, 1]
        >>> sequence(2**n).find_linear_recurrence(10)
        [2]
        >>> sequence(23*n**4+91*n**2).find_linear_recurrence(10)
        [5, -10, 10, -5, 1]
        >>> sequence(sqrt(5)*(((1 + sqrt(5))/2)**n - (-(1 + sqrt(5))/2)**(-n))/5).find_linear_recurrence(10)
        [1, 1]
        >>> sequence(x+y*(-2)**(-n), (n, 0, oo)).find_linear_recurrence(30)
        [1/2, 1/2]
        >>> sequence(3*5**n + 12).find_linear_recurrence(20,gfvar=x)
        ([6, -5], 3*(5 - 21*x)/((x - 1)*(5*x - 1)))
        >>> sequence(lucas(n)).find_linear_recurrence(15,gfvar=x)
        ([1, 1], (x - 2)/(x**2 + x - 1))
        r   )Matrixc                 S   s   g | ]}t t|qS r&   )r   r   )r8   tr&   r&   r'   r^   T  r_   z2SeqBase.find_linear_recurrence.<locals>.<listcomp>N   rE   rD   )Zsympy.matricesrd   lenminr\   appendZdetr   ZLUsolver   r   )r,   ndZgfvarrd   xZlxrZcoeffsll2Zmlistkmyr9   r:   r&   r&   r'   find_linear_recurrence1  sJ    "


2(zSeqBase.find_linear_recurrence)NN)#__name__
__module____qualname____doc__Zis_commutativeZ_op_prioritystaticmethodr(   r.   propertyr1   r)   r    r3   r4   r5   r6   r   rB   r?   rI   rK   rL   rM   rS   r   rU   rV   rW   rX   rZ   r[   r]   rc   rs   r&   r&   r&   r'   r      sP   








,


r   c                   @   s8   e Zd ZdZedd Zedd Zdd Zdd	 Zd
S )EmptySequencea  Represents an empty sequence.

    The empty sequence is also available as a singleton as
    ``S.EmptySequence``.

    Examples
    ========

    >>> from sympy import EmptySequence, SeqPer
    >>> from sympy.abc import x
    >>> EmptySequence
    EmptySequence
    >>> SeqPer((1, 2), (x, 0, 10)) + EmptySequence
    SeqPer((1, 2), (x, 0, 10))
    >>> SeqPer((1, 2)) * EmptySequence
    EmptySequence
    >>> EmptySequence.coeff_mul(-1)
    EmptySequence
    c                 C   s   t jS rT   )r   EmptySetr0   r&   r&   r'   r)     s    zEmptySequence.intervalc                 C   s   t jS rT   )r   ZZeror0   r&   r&   r'   r4     s    zEmptySequence.lengthc                 C   s   | S )"See docstring of SeqBase.coeff_mulr&   )r,   rB   r&   r&   r'   rM     s    zEmptySequence.coeff_mulc                 C   s   t g S rT   )iterr0   r&   r&   r'   r]     s    zEmptySequence.__iter__N)	rt   ru   rv   rw   ry   r)   r4   rM   r]   r&   r&   r&   r'   rz   |  s   

rz   )	metaclassc                   @   sX   e Zd ZdZedd Zedd Zedd Zedd	 Zed
d Z	edd Z
dS )SeqExpra  Sequence expression class.

    Various sequences should inherit from this class.

    Examples
    ========

    >>> from sympy.series.sequences import SeqExpr
    >>> from sympy.abc import x
    >>> from sympy import Tuple
    >>> s = SeqExpr(Tuple(1, 2, 3), Tuple(x, 0, 10))
    >>> s.gen
    (1, 2, 3)
    >>> s.interval
    Interval(0, 10)
    >>> s.length
    11

    See Also
    ========

    sympy.series.sequences.SeqPer
    sympy.series.sequences.SeqFormula
    c                 C   s
   | j d S Nr   r<   r0   r&   r&   r'   r1     s    zSeqExpr.genc                 C   s   t | jd d | jd d S )NrE   rf   )r   r=   r0   r&   r&   r'   r)     s    zSeqExpr.intervalc                 C   s   | j jS rT   r)   r*   r0   r&   r&   r'   r      s    zSeqExpr.startc                 C   s   | j jS rT   r)   r+   r0   r&   r&   r'   r3     s    zSeqExpr.stopc                 C   s   | j | j d S NrE   r3   r    r0   r&   r&   r'   r4     s    zSeqExpr.lengthc                 C   s   | j d d fS )NrE   r   r<   r0   r&   r&   r'   r5     s    zSeqExpr.variablesN)rt   ru   rv   rw   ry   r1   r)   r    r3   r4   r5   r&   r&   r&   r'   r     s   




r   c                   @   sR   e Zd ZdZdddZedd Zedd Zd	d
 Zdd Z	dd Z
dd ZdS )SeqPera  
    Represents a periodic sequence.

    The elements are repeated after a given period.

    Examples
    ========

    >>> from sympy import SeqPer, oo
    >>> from sympy.abc import k

    >>> s = SeqPer((1, 2, 3), (0, 5))
    >>> s.periodical
    (1, 2, 3)
    >>> s.period
    3

    For value at a particular point

    >>> s.coeff(3)
    1

    supports slicing

    >>> s[:]
    [1, 2, 3, 1, 2, 3]

    iterable

    >>> list(s)
    [1, 2, 3, 1, 2, 3]

    sequence starts from negative infinity

    >>> SeqPer((1, 2, 3), (-oo, 0))[0:6]
    [1, 2, 3, 1, 2, 3]

    Periodic formulas

    >>> SeqPer((k, k**2, k**3), (k, 0, oo))[0:6]
    [0, 1, 8, 3, 16, 125]

    See Also
    ========

    sympy.series.sequences.SeqFormula
    Nc                 C   s$  t |}dd }d\}}}|d u r8||dtj  }}}t|trvt|dkrZ|\}}}nt|dkrv||}|\}}t|ttfr|d u s|d u rt	dt
| |tju r|tju rt	dt |||f}t|trt tt|}nt	d	| t|d
 |d tju rtjS t| ||S )Nc                 S   s(   | j }t| j dkr| S tdS d S )NrE   rp   )r6   rg   popr   )
periodicalfreer&   r&   r'   _find_x  s    zSeqPer.__new__.<locals>._find_xNNNr      rf   Invalid limits given: %sz/Both the start and end valuecannot be unboundedz6invalid period %s should be something like e.g (1, 2) rE   )r   r   r$   r   r   rg   rO   r   r   r#   strrF   tupler   r   r{   rz   r   __new__)clsr   limitsr   rl   r    r3   r&   r&   r'   r     s0    


zSeqPer.__new__c                 C   s
   t | jS rT   )rg   r1   r0   r&   r&   r'   period-  s    zSeqPer.periodc                 C   s   | j S rT   r1   r0   r&   r&   r'   r   1  s    zSeqPer.periodicalc                 C   sF   | j tju r| j| | j }n|| j  | j }| j| | jd |S r   )r    r   rF   r3   r   r   subsr5   )r,   rA   idxr&   r&   r'   r?   5  s    zSeqPer._eval_coeffc                 C   s   t |tr| j| j }}|j|j }}t||}g }t|D ]*}|||  }	|||  }
||	|
  q<| |\}}t|| jd ||fS dS zSee docstring of SeqBase._addr   N	rO   r   r   r   r   r\   ri   r.   r5   r,   r-   Zper1Zlper1Zper2Zlper2Z
per_lengthZnew_perrl   Zele1Zele2r    r3   r&   r&   r'   rK   <  s    

zSeqPer._addc                 C   s   t |tr| j| j }}|j|j }}t||}g }t|D ]*}|||  }	|||  }
||	|
  q<| |\}}t|| jd ||fS dS zSee docstring of SeqBase._mulr   Nr   r   r&   r&   r'   rL   M  s    

zSeqPer._mulc                    s,   t    fdd| jD }t|| jd S )r|   c                    s   g | ]}|  qS r&   r&   r8   rl   rB   r&   r'   r^   a  r_   z$SeqPer.coeff_mul.<locals>.<listcomp>rE   )r   r   r   r=   )r,   rB   Zperr&   r   r'   rM   ^  s    zSeqPer.coeff_mul)N)rt   ru   rv   rw   r   ry   r   r   r?   rK   rL   rM   r&   r&   r&   r'   r     s   0
(

r   c                   @   sN   e Zd ZdZdddZedd Zdd Zd	d
 Zdd Z	dd Z
dd ZdS )
SeqFormulaaf  
    Represents sequence based on a formula.

    Elements are generated using a formula.

    Examples
    ========

    >>> from sympy import SeqFormula, oo, Symbol
    >>> n = Symbol('n')
    >>> s = SeqFormula(n**2, (n, 0, 5))
    >>> s.formula
    n**2

    For value at a particular point

    >>> s.coeff(3)
    9

    supports slicing

    >>> s[:]
    [0, 1, 4, 9, 16, 25]

    iterable

    >>> list(s)
    [0, 1, 4, 9, 16, 25]

    sequence starts from negative infinity

    >>> SeqFormula(n**2, (-oo, 0))[0:6]
    [0, 1, 4, 9, 16, 25]

    See Also
    ========

    sympy.series.sequences.SeqPer
    Nc                 C   s   t |}dd }d\}}}|d u r8||dtj  }}}t|trvt|dkrZ|\}}}nt|dkrv||}|\}}t|ttfr|d u s|d u rt	dt
| |tju r|tju rt	dt |||f}t|d	 |d tju rtjS t| ||S )
Nc                 S   s6   | j }t|dkr| S |s&tdS td|  d S )NrE   rp   z specify dummy variables for %s. If the formula contains more than one free symbol, a dummy variable should be supplied explicitly e.g., SeqFormula(m*n**2, (n, 0, 5)))r6   rg   r   r   r#   )formular   r&   r&   r'   r     s    z#SeqFormula.__new__.<locals>._find_xr   r   r   rf   r   z0Both the start and end value cannot be unboundedrE   )r   r   r$   r   r   rg   rO   r   r   r#   r   rF   r   r{   rz   r   r   )r   r   r   r   rl   r    r3   r&   r&   r'   r     s&    

zSeqFormula.__new__c                 C   s   | j S rT   r   r0   r&   r&   r'   r     s    zSeqFormula.formulac                 C   s   | j d }| j||S r   )r5   r   r   )r,   rA   rk   r&   r&   r'   r?     s    
zSeqFormula._eval_coeffc           	      C   s`   t |tr\| j| jd  }}|j|jd  }}|||| }| |\}}t||||fS dS r   rO   r   r   r5   r   r.   	r,   r-   Zform1v1Zform2v2r   r    r3   r&   r&   r'   rK     s    
zSeqFormula._addc           	      C   s`   t |tr\| j| jd  }}|j|jd  }}|||| }| |\}}t||||fS dS r   r   r   r&   r&   r'   rL     s    
zSeqFormula._mulc                 C   s"   t |}| j| }t|| jd S )r|   rE   )r   r   r   r=   )r,   rB   r   r&   r&   r'   rM     s    
zSeqFormula.coeff_mulc                 O   s$   t t| jg|R i || jd S r   )r   r   r   r=   )r,   r=   kwargsr&   r&   r'   r     s    zSeqFormula.expand)N)rt   ru   rv   rw   r   ry   r   r?   rK   rL   rM   r   r&   r&   r&   r'   r   e  s   (
'
		r   c                   @   s   e Zd ZdZdddZedd Zedd	 Zed
d Zedd Z	edd Z
edd Zedd Zedd Zedd Zdd Zdd ZdS )RecursiveSeqa  
    A finite degree recursive sequence.

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

    That is, a sequence a(n) that depends on a fixed, finite number of its
    previous values. The general form is

        a(n) = f(a(n - 1), a(n - 2), ..., a(n - d))

    for some fixed, positive integer d, where f is some function defined by a
    SymPy expression.

    Parameters
    ==========

    recurrence : SymPy expression defining recurrence
        This is *not* an equality, only the expression that the nth term is
        equal to. For example, if :code:`a(n) = f(a(n - 1), ..., a(n - d))`,
        then the expression should be :code:`f(a(n - 1), ..., a(n - d))`.

    yn : applied undefined function
        Represents the nth term of the sequence as e.g. :code:`y(n)` where
        :code:`y` is an undefined function and `n` is the sequence index.

    n : symbolic argument
        The name of the variable that the recurrence is in, e.g., :code:`n` if
        the recurrence function is :code:`y(n)`.

    initial : iterable with length equal to the degree of the recurrence
        The initial values of the recurrence.

    start : start value of sequence (inclusive)

    Examples
    ========

    >>> from sympy import Function, symbols
    >>> from sympy.series.sequences import RecursiveSeq
    >>> y = Function("y")
    >>> n = symbols("n")
    >>> fib = RecursiveSeq(y(n - 1) + y(n - 2), y(n), n, [0, 1])

    >>> fib.coeff(3) # Value at a particular point
    2

    >>> fib[:6] # supports slicing
    [0, 1, 1, 2, 3, 5]

    >>> fib.recurrence # inspect recurrence
    Eq(y(n), y(n - 2) + y(n - 1))

    >>> fib.degree # automatically determine degree
    2

    >>> for x in zip(range(10), fib): # supports iteration
    ...     print(x)
    (0, 0)
    (1, 1)
    (2, 1)
    (3, 2)
    (4, 3)
    (5, 5)
    (6, 8)
    (7, 13)
    (8, 21)
    (9, 34)

    See Also
    ========

    sympy.series.sequences.SeqFormula

    Nr   c                    s^  t |tstd|t |tr(|js6td||j|fkrJtd|jtd|fd}d}|	}|D ]f}	t
|	jdkrtd|	jd || | }
|
 r|
jr|
dk std	|	|
 |krp|
 }qp|sd
d t|D }t
||krtdt|}t  tdd |D  }t| |||| } fddt|D |_||_|S )NzErecurrence sequence must be an applied undefined function, found `{}`z0recurrence variable must be a symbol, found `{}`z)recurrence sequence does not match symbolrp   )excluder   rE   z)Recurrence should be in a single variablezDRecurrence should have constant, negative, integer shifts (found {})c                 S   s   g | ]}t d |qS )zc_{})r   format)r8   rp   r&   r&   r'   r^   I  r_   z(RecursiveSeq.__new__.<locals>.<listcomp>z)Number of initial terms must equal degreec                 s   s   | ]}t |V  qd S rT   r   r   r&   r&   r'   	<genexpr>Q  r_   z'RecursiveSeq.__new__.<locals>.<genexpr>c                    s   i | ]\}} | |qS r&   r&   )r8   rp   initr    rr   r&   r'   
<dictcomp>U  r_   z(RecursiveSeq.__new__.<locals>.<dictcomp>)rO   r   rP   r   r   Z	is_symbolr=   rC   r   findrg   matchis_constant
is_integerr\   r#   r
   r   r   r   	enumeratecachedegree)r   
recurrenceynrj   rG   r    rp   r   Zprev_ysZprev_yshiftseqr&   r   r'   r   %  sF    


zRecursiveSeq.__new__c                 C   s
   | j d S zEquation defining recurrence.r   r<   r0   r&   r&   r'   _recurrenceZ  s    zRecursiveSeq._recurrencec                 C   s   t | j| jd S r   )r   r   r=   r0   r&   r&   r'   r   _  s    zRecursiveSeq.recurrencec                 C   s
   | j d S )z*Applied function representing the nth termrE   r<   r0   r&   r&   r'   r   d  s    zRecursiveSeq.ync                 C   s   | j jS )z3Undefined function for the nth term of the sequence)r   rC   r0   r&   r&   r'   rr   i  s    zRecursiveSeq.yc                 C   s
   | j d S )zSequence index symbolrf   r<   r0   r&   r&   r'   rj   n  s    zRecursiveSeq.nc                 C   s
   | j d S )z"The initial values of the sequencer   r<   r0   r&   r&   r'   rG   s  s    zRecursiveSeq.initialc                 C   s
   | j d S )r2      r<   r0   r&   r&   r'   r    x  s    zRecursiveSeq.startc                 C   s   t jS )z&The ending point of the sequence. (oo))r   r$   r0   r&   r&   r'   r3   }  s    zRecursiveSeq.stopc                 C   s   | j tjfS )z&Interval on which sequence is defined.)r    r   r$   r0   r&   r&   r'   r)     s    zRecursiveSeq.intervalc                 C   s   || j  t| jk r$| j| | S tt| j|d D ]<}| j | }| j| j|i}|| j}|| j| |< q8| j| | j |  S r   )r    rg   r   rr   r\   r   Zxreplacerj   )r,   rb   currentZ	seq_indexZcurrent_recurrenceZnew_termr&   r&   r'   r?     s    
zRecursiveSeq._eval_coeffc                 c   s    | j }| |V  |d7 }qd S r   )r    r?   )r,   rb   r&   r&   r'   r]     s    zRecursiveSeq.__iter__)Nr   )rt   ru   rv   rw   r   ry   r   r   r   rr   rj   rG   r    r3   r)   r?   r]   r&   r&   r&   r'   r     s,   L
5








r   Nc                 C   s*   t | } t| trt| |S t| |S dS )a  
    Returns appropriate sequence object.

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

    If ``seq`` is a SymPy sequence, returns :class:`SeqPer` object
    otherwise returns :class:`SeqFormula` object.

    Examples
    ========

    >>> from sympy import sequence
    >>> from sympy.abc import n
    >>> sequence(n**2, (n, 0, 5))
    SeqFormula(n**2, (n, 0, 5))
    >>> sequence((1, 2, 3), (n, 0, 5))
    SeqPer((1, 2, 3), (n, 0, 5))

    See Also
    ========

    sympy.series.sequences.SeqPer
    sympy.series.sequences.SeqFormula
    N)r   r   r   r   r   )r   r   r&   r&   r'   sequence  s    

r   c                   @   sX   e Zd ZdZedd Zedd Zedd Zedd	 Zed
d Z	edd Z
dS )	SeqExprOpa  
    Base class for operations on sequences.

    Examples
    ========

    >>> from sympy.series.sequences import SeqExprOp, sequence
    >>> from sympy.abc import n
    >>> s1 = sequence(n**2, (n, 0, 10))
    >>> s2 = sequence((1, 2, 3), (n, 5, 10))
    >>> s = SeqExprOp(s1, s2)
    >>> s.gen
    (n**2, (1, 2, 3))
    >>> s.interval
    Interval(5, 10)
    >>> s.length
    6

    See Also
    ========

    sympy.series.sequences.SeqAdd
    sympy.series.sequences.SeqMul
    c                 C   s   t dd | jD S )zjGenerator for the sequence.

        returns a tuple of generators of all the argument sequences.
        c                 s   s   | ]}|j V  qd S rT   r   r8   ar&   r&   r'   r     r_   z SeqExprOp.gen.<locals>.<genexpr>)r   r=   r0   r&   r&   r'   r1     s    zSeqExprOp.genc                 C   s   t dd | jD  S )zeSequence is defined on the intersection
        of all the intervals of respective sequences
        c                 s   s   | ]}|j V  qd S rT   r)   r   r&   r&   r'   r     r_   z%SeqExprOp.interval.<locals>.<genexpr>)r   r=   r0   r&   r&   r'   r)     s    zSeqExprOp.intervalc                 C   s   | j jS rT   r   r0   r&   r&   r'   r      s    zSeqExprOp.startc                 C   s   | j jS rT   r   r0   r&   r&   r'   r3     s    zSeqExprOp.stopc                 C   s   t tdd | jD S )z%Cumulative of all the bound variablesc                 S   s   g | ]
}|j qS r&   )r5   r   r&   r&   r'   r^     r_   z'SeqExprOp.variables.<locals>.<listcomp>)r   r   r=   r0   r&   r&   r'   r5     s    zSeqExprOp.variablesc                 C   s   | j | j d S r   r   r0   r&   r&   r'   r4     s    zSeqExprOp.lengthN)rt   ru   rv   rw   ry   r1   r)   r    r3   r5   r4   r&   r&   r&   r'   r     s   




r   c                   @   s,   e Zd ZdZdd Zedd Zdd ZdS )	rR   a  Represents term-wise addition of sequences.

    Rules:
        * The interval on which sequence is defined is the intersection
          of respective intervals of sequences.
        * Anything + :class:`EmptySequence` remains unchanged.
        * Other rules are defined in ``_add`` methods of sequence classes.

    Examples
    ========

    >>> from sympy import EmptySequence, oo, SeqAdd, SeqPer, SeqFormula
    >>> from sympy.abc import n
    >>> SeqAdd(SeqPer((1, 2), (n, 0, oo)), EmptySequence)
    SeqPer((1, 2), (n, 0, oo))
    >>> SeqAdd(SeqPer((1, 2), (n, 0, 5)), SeqPer((1, 2), (n, 6, 10)))
    EmptySequence
    >>> SeqAdd(SeqPer((1, 2), (n, 0, oo)), SeqFormula(n**2, (n, 0, oo)))
    SeqAdd(SeqFormula(n**2, (n, 0, oo)), SeqPer((1, 2), (n, 0, oo)))
    >>> SeqAdd(SeqFormula(n**3), SeqFormula(n**2))
    SeqFormula(n**3 + n**2, (n, 0, oo))

    See Also
    ========

    sympy.series.sequences.SeqMul
    c                    s   | dtj}t|} fdd  |}dd |D }|sBtjS tdd |D  tju r`tjS |rnt	|S tt
|tj}tj| g|R  S )Nevaluatec                    sP   t | tr,t | tr&tt | jg S | gS t| rDtt | g S tdd S Nz2Input must be Sequences or  iterables of Sequences)rO   r   rR   summapr=   r   rP   arg_flattenr&   r'   r   "  s    

z SeqAdd.__new__.<locals>._flattenc                 S   s   g | ]}|t jur|qS r&   )r   rz   r   r&   r&   r'   r^   .  r_   z"SeqAdd.__new__.<locals>.<listcomp>c                 s   s   | ]}|j V  qd S rT   r   r   r&   r&   r'   r   4  r_   z!SeqAdd.__new__.<locals>.<genexpr>)getr   r   listr   rz   r   r{   rR   reducer   r   r(   r   r   r   r=   r   r   r&   r   r'   r     s    

zSeqAdd.__new__c                    s   d}|r|t | D ]h\} d}t | D ]F\}||kr6q$ }|dur$ fdd| D }||  qlq$|r|}  qqqt| dkr|  S t| ddS dS )a  Simplify :class:`SeqAdd` using known rules.

        Iterates through all pairs and ask the constituent
        sequences if they can simplify themselves with any other constituent.

        Notes
        =====

        adapted from ``Union.reduce``

        TFNc                    s   g | ]}| fvr|qS r&   r&   r   sre   r&   r'   r^   W  r_   z!SeqAdd.reduce.<locals>.<listcomp>rE   r   )r   rK   ri   rg   r   rR   r=   new_argsZid1Zid2Znew_seqr&   r   r'   r   ?  s$    

zSeqAdd.reducec                    s   t  fdd| jD S )z9adds up the coefficients of all the sequences at point ptc                 3   s   | ]}|  V  qd S rT   r   r   rA   r&   r'   r   e  r_   z%SeqAdd._eval_coeff.<locals>.<genexpr>)r   r=   r@   r&   r   r'   r?   c  s    zSeqAdd._eval_coeffNrt   ru   rv   rw   r   rx   r   r?   r&   r&   r&   r'   rR     s
   $
#rR   c                   @   s,   e Zd ZdZdd Zedd Zdd ZdS )	rY   a'  Represents term-wise multiplication of sequences.

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

    Handles multiplication of sequences only. For multiplication
    with other objects see :func:`SeqBase.coeff_mul`.

    Rules:
        * The interval on which sequence is defined is the intersection
          of respective intervals of sequences.
        * Anything \* :class:`EmptySequence` returns :class:`EmptySequence`.
        * Other rules are defined in ``_mul`` methods of sequence classes.

    Examples
    ========

    >>> from sympy import EmptySequence, oo, SeqMul, SeqPer, SeqFormula
    >>> from sympy.abc import n
    >>> SeqMul(SeqPer((1, 2), (n, 0, oo)), EmptySequence)
    EmptySequence
    >>> SeqMul(SeqPer((1, 2), (n, 0, 5)), SeqPer((1, 2), (n, 6, 10)))
    EmptySequence
    >>> SeqMul(SeqPer((1, 2), (n, 0, oo)), SeqFormula(n**2))
    SeqMul(SeqFormula(n**2, (n, 0, oo)), SeqPer((1, 2), (n, 0, oo)))
    >>> SeqMul(SeqFormula(n**3), SeqFormula(n**2))
    SeqFormula(n**5, (n, 0, oo))

    See Also
    ========

    sympy.series.sequences.SeqAdd
    c                    s   | dtj}t|} fdd  |}|s4tjS tdd |D  tju rRtjS |r`t	|S tt
|tj}tj| g|R  S )Nr   c                    sR   t | tr.t | tr&tt | jg S | gS nt| rFtt | g S tdd S r   )rO   r   rY   r   r   r=   r   rP   r   r   r&   r'   r     s    

z SeqMul.__new__.<locals>._flattenc                 s   s   | ]}|j V  qd S rT   r   r   r&   r&   r'   r     r_   z!SeqMul.__new__.<locals>.<genexpr>)r   r   r   r   r   rz   r   r{   rY   r   r   r   r(   r   r   r   r&   r   r'   r     s    

zSeqMul.__new__c                    s   d}|r|t | D ]h\} d}t | D ]F\}||kr6q$ }|dur$ fdd| D }||  qlq$|r|}  qqqt| dkr|  S t| ddS dS )a.  Simplify a :class:`SeqMul` using known rules.

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

        Iterates through all pairs and ask the constituent
        sequences if they can simplify themselves with any other constituent.

        Notes
        =====

        adapted from ``Union.reduce``

        TFNc                    s   g | ]}| fvr|qS r&   r&   r   r   r&   r'   r^     r_   z!SeqMul.reduce.<locals>.<listcomp>rE   r   )r   rL   ri   rg   r   rY   r   r&   r   r'   r     s$    

zSeqMul.reducec                 C   s"   d}| j D ]}|||9 }q
|S )z<multiplies the coefficients of all the sequences at point ptrE   )r=   rB   )r,   rA   valr   r&   r&   r'   r?     s    
zSeqMul._eval_coeffNr   r&   r&   r&   r'   rY   h  s
   ""
&rY   )N)6Zsympy.core.basicr   Zsympy.core.cacher   Zsympy.core.containersr   Zsympy.core.decoratorsr   Zsympy.core.parametersr   Zsympy.core.functionr   Zsympy.core.mulr	   Zsympy.core.numbersr
   Zsympy.core.relationalr   Zsympy.core.singletonr   r   Zsympy.core.sortingr   Zsympy.core.symbolr   r   r   Zsympy.core.sympifyr   Zsympy.polysr   r   Zsympy.sets.setsr   r   Zsympy.simplifyr   Zsympy.tensor.indexedr   Zsympy.utilities.iterablesr   r   r   r   r   rz   r   r   r   r   r   r   rR   rY   r&   r&   r&   r'   <module>   s@     c%3 s F
':j