
    ݷek                        d Z ddlZddlZddlmZ ddlmZ ddlmZm	Z	 ddl
mZmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZ ddlmZ g d	Z e       Z	  e d
        ee d
      Z! e#edd       Z$d Z%dFdZ&d Z'dGdZ(dGdZ)d Z*e+fdZ,d Z-e-Z.d Z/d Z0d Z1dGdZ2d Z3	 ddl
m4Z5 d Z4e3j                   e4_          G d de7      Z8d Z9d  Z:dHd!Z;d" Z<d# Z=d$ Z>dGd%Z?dGd&Z@dGd'ZAdId(ZBd)d*d+ZCdGd,ZDd- ZEd. ZFd/ ZGd0 ZHd1 ZId2 ZJd3 ZKd4 ZLd5 ZMd6 ZNdJd7ZOd8 ZPd9dd:ZQed;k\  rdd<l
mRZS d9dd=ZRneQZReQj                   eR_         d> ZTd? ZUd@ ZVdA ZWdB ZXdC ZYdD ZZdE Z[y# e"$ r e Z!Y w xY w# e6$ r e3Z4Y w xY w)Ka  Imported from the recipes section of the itertools documentation.

All functions taken from the recipes section of the itertools library docs
[1]_.
Some backward-compatible usability improvements have been made.

.. [1] http://docs.python.org/library/itertools.html#recipes

    N)deque)Sized)partialreduce)chaincombinationscompresscountcyclegroupbyisliceproductrepeatstarmapteezip_longest)	randrangesamplechoice)
hexversion)-	all_equalbatchedbefore_and_afterconsumeconvolve
dotproduct
first_truefactorflattengrouperiter_except
iter_indexmatmulncyclesnthnth_combinationpadnonepad_nonepairwise	partitionpolynomial_evalpolynomial_from_rootspolynomial_derivativepowersetprependquantifyreshape#random_combination_with_replacementrandom_combinationrandom_permutationrandom_product
repeatfunc
roundrobinsievesliding_window	subslicessum_of_squarestabulatetailtaketotient	transpose
triplewiseunique_everseenunique_justseenTstrictsumprodc                     t        | |      S N)r   )xys     8/usr/lib/python3/dist-packages/more_itertools/recipes.py<lambda>rL   ]   s    Aq1A     c                 ,    t        t        ||             S )zReturn first *n* items of the iterable as a list.

        >>> take(3, range(10))
        [0, 1, 2]

    If there are fewer than *n* items in the iterable, all of them are
    returned.

        >>> take(10, range(3))
        [0, 1, 2]

    )listr   niterables     rK   r>   r>   `   s     x#$$rM   c                 ,    t        | t        |            S )a  Return an iterator over the results of ``func(start)``,
    ``func(start + 1)``, ``func(start + 2)``...

    *func* should be a function that accepts one integer argument.

    If *start* is not specified it defaults to 0. It will be incremented each
    time the iterator is advanced.

        >>> square = lambda x: x ** 2
        >>> iterator = tabulate(square, -3)
        >>> take(4, iterator)
        [9, 4, 1, 0]

    )mapr
   )functionstarts     rK   r<   r<   p   s     xu&&rM   c           	   #      K   t        |t              r,t        |t        dt	        |      | z
        d      E d{    yt        t        ||             E d{    y7 $7 w)zReturn an iterator over the last *n* items of *iterable*.

    >>> t = tail(3, 'ABCDEFG')
    >>> list(t)
    ['E', 'F', 'G']

    r   Nmaxlen)
isinstancer   r   maxleniterr   rP   s     rK   r=   r=      sS      (E"(C3x=1+<$=tDDDhq1222 	E2s!   7A"AA"A A" A"c                 R    |t        | d       yt        t        | ||      d       y)aX  Advance *iterable* by *n* steps. If *n* is ``None``, consume it
    entirely.

    Efficiently exhausts an iterator without returning values. Defaults to
    consuming the whole iterator, but an optional second argument may be
    provided to limit consumption.

        >>> i = (x for x in range(10))
        >>> next(i)
        0
        >>> consume(i, 3)
        >>> next(i)
        4
        >>> consume(i)
        >>> next(i)
        Traceback (most recent call last):
          File "<stdin>", line 1, in <module>
        StopIteration

    If the iterator has fewer items remaining than the provided limit, the
    whole iterator will be consumed.

        >>> i = (x for x in range(3))
        >>> consume(i, 5)
        >>> next(i)
        Traceback (most recent call last):
          File "<stdin>", line 1, in <module>
        StopIteration

    Nr   rX   )r   nextr   )iteratorrQ   s     rK   r   r      s)    @ 	yhq! 	VHa#T*rM   c                 0    t        t        | |d      |      S )zReturns the nth item or a default value.

    >>> l = range(10)
    >>> nth(l, 3)
    3
    >>> nth(l, 20, "zebra")
    'zebra'

    N)r_   r   )rR   rQ   defaults      rK   r%   r%      s     xD)733rM   c                 N    t        |       }t        |d      xr t        |d       S )z
    Returns ``True`` if all the elements are equal to each other.

        >>> all_equal('aaaa')
        True
        >>> all_equal('aaab')
        False

    TF)r   r_   )rR   gs     rK   r   r      s(     	A4=/a//rM   c                 ,    t        t        ||             S )zcReturn the how many times the predicate is true.

    >>> quantify([True, False, True])
    2

    )sumrT   )rR   preds     rK   r0   r0      s     s4"##rM   c                 ,    t        | t        d            S )a   Returns the sequence of elements and then returns ``None`` indefinitely.

        >>> take(5, pad_none(range(3)))
        [0, 1, 2, None, None]

    Useful for emulating the behavior of the built-in :func:`map` function.

    See also :func:`padded`.

    N)r   r   rR   s    rK   r(   r(      s     6$<((rM   c                 R    t        j                  t        t        |       |            S )zvReturns the sequence elements *n* times

    >>> list(ncycles(["a", "b"], 3))
    ['a', 'b', 'a', 'b', 'a', 'b']

    )r   from_iterabler   tuple)rR   rQ   s     rK   r$   r$      s      veHoq9::rM   c                 J    t        t        t        j                  | |            S )zcReturns the dot product of the two iterables.

    >>> dotproduct([10, 10], [20, 20])
    400

    )rf   rT   operatormul)vec1vec2s     rK   r   r      s     s8<<t,--rM   c                 ,    t        j                  |       S )zReturn an iterator flattening one level of nesting in a list of lists.

        >>> list(flatten([[0, 1], [2, 3]]))
        [0, 1, 2, 3]

    See also :func:`collapse`, which can flatten multiple levels of nesting.

    )r   rk   )listOfListss    rK   r   r     s     {++rM   c                 \    |t        | t        |            S t        | t        ||            S )aG  Call *func* with *args* repeatedly, returning an iterable over the
    results.

    If *times* is specified, the iterable will terminate after that many
    repetitions:

        >>> from operator import add
        >>> times = 4
        >>> args = 3, 5
        >>> list(repeatfunc(add, times, *args))
        [8, 8, 8, 8]

    If *times* is ``None`` the iterable will not terminate:

        >>> from random import randrange
        >>> times = None
        >>> args = 1, 11
        >>> take(6, repeatfunc(randrange, times, *args))  # doctest:+SKIP
        [2, 4, 8, 1, 8, 4]

    )r   r   )functimesargss      rK   r6   r6     s.    , }tVD\**4e,--rM   c                 N    t        |       \  }}t        |d       t        ||      S )zReturns an iterator of paired items, overlapping, from the original

    >>> take(4, pairwise(count()))
    [(0, 1), (1, 2), (2, 3), (3, 4)]

    On Python 3.10 and above, this is an alias for :func:`itertools.pairwise`.

    N)r   r_   zip)rR   abs      rK   	_pairwiser|   -  s&     x=DAqDMq!9rM   r)   c                     t        |       S rH   )itertools_pairwiseri   s    rK   r)   r)   A  s    !(++rM   c                         e Zd Zd fd	Z xZS )UnequalIterablesErrorc                 P    d}|| dj                   | z  }t        | 	  |       y )Nz Iterables have different lengthsz/: index 0 has length {}; index {} has length {})formatsuper__init__)selfdetailsmsg	__class__s      rK   r   zUnequalIterablesError.__init__H  s;    0MEMM C 	rM   rH   )__name__
__module____qualname__r   __classcell__)r   s   @rK   r   r   G  s     rM   r   c              #   n   K   t        | dt        iD ]  }|D ]  }|t        u st                | ! y w)N	fillvalue)r   _markerr   )	iterablescombovals      rK   _zip_equal_generatorr   R  sE     i;7;  	.Cg~+--	. 	s    55c                      	 t        | d         }t        | dd  d      D ]$  \  }}t        |      }||k7  st        |||f       t        |  S # t        $ r t        |       cY S w xY w)Nr      )r   )r\   	enumerater   ry   	TypeErrorr   )r   
first_sizeiitsizes        rK   
_zip_equalr   Z  s    /1&
y}a0 	KEArr7Dz!+ZD4IJJ	K
 I  /#I../s   3A A A%$A%c                     t        |       g|z  }|dk(  rt        |d|iS |dk(  rt        | S |dk(  rt        | S t	        d      )a  Group elements from *iterable* into fixed-length groups of length *n*.

    >>> list(grouper('ABCDEF', 3))
    [('A', 'B', 'C'), ('D', 'E', 'F')]

    The keyword arguments *incomplete* and *fillvalue* control what happens for
    iterables whose length is not a multiple of *n*.

    When *incomplete* is `'fill'`, the last group will contain instances of
    *fillvalue*.

    >>> list(grouper('ABCDEFG', 3, incomplete='fill', fillvalue='x'))
    [('A', 'B', 'C'), ('D', 'E', 'F'), ('G', 'x', 'x')]

    When *incomplete* is `'ignore'`, the last group will not be emitted.

    >>> list(grouper('ABCDEFG', 3, incomplete='ignore', fillvalue='x'))
    [('A', 'B', 'C'), ('D', 'E', 'F')]

    When *incomplete* is `'strict'`, a subclass of `ValueError` will be raised.

    >>> it = grouper('ABCDEFG', 3, incomplete='strict')
    >>> list(it)  # doctest: +IGNORE_EXCEPTION_DETAIL
    Traceback (most recent call last):
    ...
    UnequalIterablesError

    fillr   rE   ignorez Expected fill, strict, or ignore)r]   r   r   ry   
ValueError)rR   rQ   
incompleter   rw   s        rK   r    r    j  s^    : NaDVD6I66X4  XDz;<<rM   c               '      K   t        |       }t        d | D              }|r	 |D ]  } |         	 |ryy# t        $ r |dz  }t        t        ||            }Y *w xY ww)aJ  Yields an item from each iterable, alternating between them.

        >>> list(roundrobin('ABC', 'D', 'EF'))
        ['A', 'D', 'E', 'B', 'F', 'C']

    This function produces the same output as :func:`interleave_longest`, but
    may perform better for some inputs (in particular when the number of
    iterables is small).

    c              3   F   K   | ]  }t        |      j                    y wrH   )r]   __next__).0r   s     rK   	<genexpr>zroundrobin.<locals>.<genexpr>  s     8$r(##8s   !r   N)r\   r   StopIterationr   )r   pendingnextsr_   s       rK   r7   r7     sn      )nG8i88E
	2 f   	2qLG&01E	2s*    A"9 A"A"#AA"AA"c                     | t         } t        |d      \  }}}t        t        | |            \  }}t        |t        t        j
                  |            t        ||      fS )a  
    Returns a 2-tuple of iterables derived from the input iterable.
    The first yields the items that have ``pred(item) == False``.
    The second yields the items that have ``pred(item) == True``.

        >>> is_odd = lambda x: x % 2 != 0
        >>> iterable = range(10)
        >>> even_items, odd_items = partition(is_odd, iterable)
        >>> list(even_items), list(odd_items)
        ([0, 2, 4, 6, 8], [1, 3, 5, 7, 9])

    If *pred* is None, :func:`bool` is used.

        >>> iterable = [0, 1, False, True, '', ' ']
        >>> false_items, true_items = partition(None, iterable)
        >>> list(false_items), list(true_items)
        ([0, False, ''], [1, True, ' '])

       )boolr   rT   r	   rn   not_)rg   rR   t1t2pp1p2s          rK   r*   r*     sX    ( |Ha IBAT1FBRX]]B/0(2r2BCCrM   c                     t        |       t        j                  fdt        t	              dz         D              S )a  Yields all possible subsets of the iterable.

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

    :func:`powerset` will operate on iterables that aren't :class:`set`
    instances, so repeated elements in the input will produce repeated elements
    in the output. Use :func:`unique_everseen` on the input to avoid generating
    duplicates:

        >>> seq = [1, 1, 0]
        >>> list(powerset(seq))
        [(), (1,), (1,), (0,), (1, 1), (1, 0), (1, 0), (1, 1, 0)]
        >>> from more_itertools import unique_everseen
        >>> list(powerset(unique_everseen(seq)))
        [(), (1,), (0,), (1, 0)]

    c              3   6   K   | ]  }t        |        y wrH   )r   )r   rss     rK   r   zpowerset.<locals>.<genexpr>  s     Ma|Aq1Ms   r   )rO   r   rk   ranger\   )rR   r   s    @rK   r.   r.     s2    & 	XAM5Q!;LMMMrM   c              #      K   t               }|j                  }g }|j                  }|du}| D ]  }|r ||      n|}	 ||vr ||       | ! y# t        $ r ||vr ||       | Y >w xY ww)a  
    Yield unique elements, preserving order.

        >>> list(unique_everseen('AAAABBBCCDAABBB'))
        ['A', 'B', 'C', 'D']
        >>> list(unique_everseen('ABBCcAD', str.lower))
        ['A', 'B', 'C', 'D']

    Sequences with a mix of hashable and unhashable items can be used.
    The function will be slower (i.e., `O(n^2)`) for unhashable items.

    Remember that ``list`` objects are unhashable - you can use the *key*
    parameter to transform the list to a tuple (which is hashable) to
    avoid a slowdown.

        >>> iterable = ([1, 2], [2, 3], [1, 2])
        >>> list(unique_everseen(iterable))  # Slow
        [[1, 2], [2, 3]]
        >>> list(unique_everseen(iterable, key=tuple))  # Faster
        [[1, 2], [2, 3]]

    Similarly, you may want to convert unhashable ``set`` objects with
    ``key=frozenset``. For ``dict`` objects,
    ``key=lambda x: frozenset(x.items())`` can be used.

    N)setaddappendr   )	rR   keyseensetseenset_addseenlistseenlist_adduse_keyelementks	            rK   rB   rB     s     6 eG++KH??LoG 	#CL	A	  	 Q	s(   :A/AA/A,)A/+A,,A/c           
          |(t        t        j                  d      t        |             S t        t        t        t        j                  d      t        | |                  S )zYields elements in order, ignoring serial duplicates

    >>> list(unique_justseen('AAAABBBCCDAABBB'))
    ['A', 'B', 'C', 'D', 'A', 'B']
    >>> list(unique_justseen('ABBCcAD', str.lower))
    ['A', 'B', 'C', 'A', 'D']

    r   r   )rT   rn   
itemgetterr   r_   )rR   r   s     rK   rC   rC   	  sL     {8&&q)78+<==tS,,Q/31GHIIrM   c              #   N   K   	 |	 |        	  |         
# |$ r Y yw xY ww)a  Yields results from a function repeatedly until an exception is raised.

    Converts a call-until-exception interface to an iterator interface.
    Like ``iter(func, sentinel)``, but uses an exception instead of a sentinel
    to end the loop.

        >>> l = [0, 1, 2]
        >>> list(iter_except(l.pop, IndexError))
        [2, 1, 0]

    Multiple exceptions can be specified as a stopping condition:

        >>> l = [1, 2, 3, '...', 4, 5, 6]
        >>> list(iter_except(lambda: 1 + l.pop(), (IndexError, TypeError)))
        [7, 6, 5]
        >>> list(iter_except(lambda: 1 + l.pop(), (IndexError, TypeError)))
        [4, 3, 2]
        >>> list(iter_except(lambda: 1 + l.pop(), (IndexError, TypeError)))
        []

    N )ru   	exceptionfirsts      rK   r!   r!     s8     ,'M&L  s   % "%"%c                 .    t        t        ||       |      S )a  
    Returns the first true value in the iterable.

    If no true value is found, returns *default*

    If *pred* is not None, returns the first item for which
    ``pred(item) == True`` .

        >>> first_true(range(10))
        1
        >>> first_true(range(10), pred=lambda x: x > 5)
        6
        >>> first_true(range(10), default='missing', pred=lambda x: x > 9)
        'missing'

    )r_   filter)rR   rb   rg   s      rK   r   r   7  s    " tX&00rM   r   )r   c                 h    |D cg c]  }t        |       c}| z  }t        d |D              S c c}w )a  Draw an item at random from each of the input iterables.

        >>> random_product('abc', range(4), 'XYZ')  # doctest:+SKIP
        ('c', 3, 'Z')

    If *repeat* is provided as a keyword argument, that many items will be
    drawn from each iterable.

        >>> random_product('abcd', range(4), repeat=2)  # doctest:+SKIP
        ('a', 2, 'd', 3)

    This equivalent to taking a random selection from
    ``itertools.product(*args, **kwarg)``.

    c              3   2   K   | ]  }t        |        y wrH   )r   )r   pools     rK   r   z!random_product.<locals>.<genexpr>\  s     0$0s   )rl   )r   rw   r   poolss       rK   r5   r5   K  s3      &**TU4[*V3E0%000 +s   /c                 `    t        |       }|t        |      n|}t        t        ||            S )ab  Return a random *r* length permutation of the elements in *iterable*.

    If *r* is not specified or is ``None``, then *r* defaults to the length of
    *iterable*.

        >>> random_permutation(range(5))  # doctest:+SKIP
        (3, 4, 0, 1, 2)

    This equivalent to taking a random selection from
    ``itertools.permutations(iterable, r)``.

    )rl   r\   r   )rR   r   r   s      rK   r4   r4   _  s-     ?DYD	AAa!!rM   c                     t        |       t              }t        t        t	        |      |            }t        fd|D              S )zReturn a random *r* length subsequence of the elements in *iterable*.

        >>> random_combination(range(5), 3)  # doctest:+SKIP
        (2, 3, 4)

    This equivalent to taking a random selection from
    ``itertools.combinations(iterable, r)``.

    c              3   (   K   | ]	  }|     y wrH   r   r   r   r   s     rK   r   z%random_combination.<locals>.<genexpr>~       *Qa*   )rl   r\   sortedr   r   )rR   r   rQ   indicesr   s       @rK   r3   r3   q  s=     ?DD	AVE!Ha()G*'***rM   c                     t        |       t              t        fdt        |      D              }t        fd|D              S )aS  Return a random *r* length subsequence of elements in *iterable*,
    allowing individual elements to be repeated.

        >>> random_combination_with_replacement(range(3), 5) # doctest:+SKIP
        (0, 0, 1, 2, 2)

    This equivalent to taking a random selection from
    ``itertools.combinations_with_replacement(iterable, r)``.

    c              3   4   K   | ]  }t                y wrH   )r   )r   r   rQ   s     rK   r   z6random_combination_with_replacement.<locals>.<genexpr>  s     4aYq\4s   c              3   (   K   | ]	  }|     y wrH   r   r   s     rK   r   z6random_combination_with_replacement.<locals>.<genexpr>  r   r   )rl   r\   r   r   )rR   r   r   rQ   r   s      @@rK   r2   r2     s<     ?DD	A45844G*'***rM   c                    t        |       }t        |      }|dk  s||kD  rt        d}t        |||z
        }t	        d|dz         D ]  }|||z
  |z   z  |z  } |dk  r||z  }|dk  s||k\  rt
        g }|rL||z  |z  |dz
  |dz
  }}}||k\  r||z  }|||z
  z  |z  |dz
  }}||k\  r|j                  |d|z
            |rLt        |      S )a  Equivalent to ``list(combinations(iterable, r))[index]``.

    The subsequences of *iterable* that are of length *r* can be ordered
    lexicographically. :func:`nth_combination` computes the subsequence at
    sort position *index* directly, without computing the previous
    subsequences.

        >>> nth_combination(range(5), 3, 5)
        (0, 3, 4)

    ``ValueError`` will be raised If *r* is negative or greater than the length
    of *iterable*.
    ``IndexError`` will be raised if the given *index* is invalid.
    r   r   )rl   r\   r   minr   
IndexErrorr   )	rR   r   indexr   rQ   cr   r   results	            rK   r&   r&     s    ?DD	A	A1q5	AAq1uA1a!e_ !QOq ! qy
	uzF
a%1*a!eQUa1qjQJEA;!#QUqA qj 	d26l#  =rM   c                     t        | g|      S )a  Yield *value*, followed by the elements in *iterator*.

        >>> value = '0'
        >>> iterator = ['1', '2', '3']
        >>> list(prepend(value, iterator))
        ['0', '1', '2', '3']

    To prepend multiple values, see :func:`itertools.chain`
    or :func:`value_chain`.

    )r   )valuer`   s     rK   r/   r/     s     %(##rM   c              #      K   t        |      ddd   }t        |      }t        dg|      |z  }t        | t	        d|dz
              D ]!  }|j                  |       t        ||       # yw)aB  Convolve the iterable *signal* with the iterable *kernel*.

        >>> signal = (1, 2, 3, 4, 5)
        >>> kernel = [3, 2, 1]
        >>> list(convolve(signal, kernel))
        [3, 8, 14, 20, 26, 14, 5]

    Note: the input arguments are not interchangeable, as the *kernel*
    is immediately consumed and stored.

    Nr   r   rX   r   )rl   r\   r   r   r   r   _sumprod)signalkernelrQ   windowrI   s        rK   r   r     sp      6]4R4 FFAA3q!A%F66!QU+, 'avv&&'s   A,A.c                 Z     t              g  fd}t              } |       |fS )a  A variant of :func:`takewhile` that allows complete access to the
    remainder of the iterator.

         >>> it = iter('ABCdEfGhI')
         >>> all_upper, remainder = before_and_after(str.isupper, it)
         >>> ''.join(all_upper)
         'ABC'
         >>> ''.join(remainder) # takewhile() would lose the 'd'
         'dEfGhI'

    Note that the first iterator must be fully consumed before the second
    iterator can generate valid results.
    c               3   Z   K   D ]!  }  |       r|  j                  |         y  y wrH   )r   )elemr   	predicate
transitions    rK   true_iteratorz'before_and_after.<locals>.true_iterator  s3      	D
!!$'	s   (+)r]   r   )r   r   r   remainder_iteratorr   s   ``  @rK   r   r     s5     
bBJ z2.?...rM   c              #   ^   K   t        t        |             D ]  \  \  }}\  }}|||f  yw)zReturn overlapping triplets from *iterable*.

    >>> list(triplewise('ABCDE'))
    [('A', 'B', 'C'), ('B', 'C', 'D'), ('C', 'D', 'E')]

    Nr}   )rR   rz   _r{   r   s        rK   rA   rA     s8      #8H#56 AAAgs   +-c              #      K   t        |       }t        t        ||dz
        |      }|D ]   }|j                  |       t	        |       " yw)aY  Return a sliding window of width *n* over *iterable*.

        >>> list(sliding_window(range(6), 4))
        [(0, 1, 2, 3), (1, 2, 3, 4), (2, 3, 4, 5)]

    If *iterable* has fewer than *n* items, then nothing is yielded:

        >>> list(sliding_window(range(3), 4))
        []

    For a variant with more features, see :func:`windowed`.
    r   rX   N)r]   r   r   r   rl   )rR   rQ   r   r   rI   s        rK   r9   r9     sM      
hB6"a!e$Q/F aFms   AAc           
          t        |       }t        t        t        t	        t        |      dz         d            }t        t        j                  t        |      |      S )zReturn all contiguous non-empty subslices of *iterable*.

        >>> list(subslices('ABC'))
        [['A'], ['A', 'B'], ['A', 'B', 'C'], ['B'], ['B', 'C'], ['C']]

    This is similar to :func:`substrings`, but emits items in a different
    order.
    r      )
rO   r   slicer   r   r\   rT   rn   getitemr   )rR   seqslicess      rK   r:   r:   !  sF     x.CULs3x!|)<a@AFxf55rM   c                     t        t        d      t        t        j                  |             }t        t        t        |dg            S )zCompute a polynomial's coefficients from its roots.

    >>> roots = [5, -4, 3]  # (x - 5) * (x + 4) * (x - 3)
    >>> polynomial_from_roots(roots)  # x^3 - 4 * x^2 - 17 * x + 60
    [1, -4, -17, 60]
    r   )ry   r   rT   rn   negrO   r   r   )rootsfactorss     rK   r,   r,   /  s5     &)Su56Gx1#.//rM   c              #      K   t        | dd      }|0t        | ||      }t        ||      D ]  \  }}||u s||k(  s|  y|t        |       n|}|dz
  }	 	  |||dz   |      x} # t        $ r Y yw xY ww)a  Yield the index of each place in *iterable* that *value* occurs,
    beginning with index *start* and ending before index *stop*.

    See :func:`locate` for a more general means of finding the indexes
    associated with particular values.

    >>> list(iter_index('AABCADEAF', 'A'))
    [0, 1, 4, 7]
    >>> list(iter_index('AABCADEAF', 'A', 1))  # start index is inclusive
    [1, 4, 7]
    >>> list(iter_index('AABCADEAF', 'A', 1, 7))  # stop index is not inclusive
    [1, 4]
    r   Nr   )getattrr   r   r\   r   )rR   r   rV   stop	seq_indexr   r   r   s           rK   r"   r"   :  s      '40IHeT*#B. 	JAw%7e#3	
 !%s8}$AI	%eQUD99q:  		s(   8A9A9A* *	A63A95A66A9c              #   j  K   | dkD  rd d}t        d      | dz  z  }t        j                  |       dz   }t        |d||      D ]Q  }t        |d|||z        E d{    t	        t        t        ||z  | ||z                     |||z  | ||z   <   ||z  }S t        |d|      E d{    y7 R7 w)zdYield the primes less than n.

    >>> list(sieve(30))
    [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
    r   r   )r   r   r   N)	bytearraymathisqrtr"   bytesr\   r   )rQ   rV   datalimitr   s        rK   r8   r8   Z  s      	1uEVQ'DJJqMAEa. dAua!e444"'E!a%AE,B(C"DQUQQA $5))) 	5 *s%   AB3B/AB3)B1*B31B3Fc             #      K   |dk  rt        d      t        |       }t        t        ||            x}r8|rt	        |      |k7  rt        d      | t        t        ||            x}r7yyw)a  Batch data into tuples of length *n*. If the number of items in
    *iterable* is not divisible by *n*:
    * The last batch will be shorter if *strict* is ``False``.
    * :exc:`ValueError` will be raised if *strict* is ``True``.

    >>> list(batched('ABCDEFG', 3))
    [('A', 'B', 'C'), ('D', 'E', 'F'), ('G',)]

    On Python 3.13 and above, this is an alias for :func:`itertools.batched`.
    r   zn must be at least onezbatched(): incomplete batchN)r   r]   rl   r   r\   )rR   rQ   rE   r   batchs        rK   _batchedr  l  sp      	1u122	hBA'
'%
'c%jAo:;; A'
'%
's   A)A.,A.i )r   c                    t        | ||      S )NrD   )itertools_batched)rR   rQ   rE   s      rK   r   r     s     1V<<rM   c                     t        |  S )a  Swap the rows and columns of the input matrix.

    >>> list(transpose([(1, 2, 3), (11, 22, 33)]))
    [(1, 11), (2, 22), (3, 33)]

    The caller should ensure that the dimensions of the input are compatible.
    If the input is empty, no output will be produced.
    )_zip_strictr   s    rK   r@   r@     s     rM   c                 @    t        t        j                  |       |      S )zReshape the 2-D input *matrix* to have a column count given by *cols*.

    >>> matrix = [(0, 1), (2, 3), (4, 5)]
    >>> cols = 3
    >>> list(reshape(matrix, cols))
    [(0, 1, 2), (3, 4, 5)]
    )r   r   rk   )matrixcolss     rK   r1   r1     s     5&&v.55rM   c                 x    t        |d         }t        t        t        t	        | t        |                  |      S )zMultiply two matrices.

    >>> list(matmul([(7, 5), (3, 5)], [(2, 5), (7, 9)]))
    [(49, 80), (41, 60)]

    The caller should ensure that the dimensions of the input matrices are
    compatible with each other.
    r   )r\   r   r   r   r   r@   )m1m2rQ   s      rK   r#   r#     s0     	BqE
A78WR2%?@!DDrM   c              #      K   t        t        j                  |       dz         D ]  }| |z  r	| | |z  } | dk(  r y| |z  s  | dkD  r|  yyw)zTYield the prime factors of n.

    >>> list(factor(360))
    [2, 2, 2, 3, 3, 5]
    r   N)r8   r  r  )rQ   primes     rK   r   r     sd      tzz!}q() e)K%KAAv	 e) 	1u s   *AAAc           	          t        |       }|dk(  r|dz  S t        t        t        |      t	        t        |                  }t        | |      S )zEvaluate a polynomial at a specific value.

    Example: evaluating x^3 - 4 * x^2 - 17 * x + 60 at x = 2.5:

    >>> coefficients = [1, -4, -17, 60]
    >>> x = 2.5
    >>> polynomial_eval(coefficients, x)
    8.125
    r   )r\   rT   powr   reversedr   r   )coefficientsrI   rQ   powerss       rK   r+   r+     sF     	LAAv1ufQi%(!34FL&))rM   c                 $    t        t        |        S )zfReturn the sum of the squares of the input values.

    >>> sum_of_squares([10, 20, 30])
    1400
    )r   r   r  s    rK   r;   r;     s     SWrM   c                     t        |       }t        t        d|            }t        t	        t
        j                  | |            S )a  Compute the first derivative of a polynomial.

    Example: evaluating the derivative of x^3 - 4 * x^2 - 17 * x + 60

    >>> coefficients = [1, -4, -17, 60]
    >>> derivative_coefficients = polynomial_derivative(coefficients)
    >>> derivative_coefficients
    [3, -8, -17]
    r   )r\   r  r   rO   rT   rn   ro   )r  rQ   r  s      rK   r-   r-     s6     	LAeAqk"FHLL,788rM   c                 N    t        t        |             D ]  }| |z  |dz
  z  }  | S )zReturn the count of natural numbers up to *n* that are coprime with *n*.

    >>> totient(9)
    6
    >>> totient(12)
    4
    r   )rC   r   )rQ   r   s     rK   r?   r?     s5     VAY' Fa!e HrM   )r   rH   )r   N)NN)r   N)\__doc__r  rn   collectionsr   collections.abcr   	functoolsr   r   	itertoolsr   r   r	   r
   r   r   r   r   r   r   r   r   randomr   r   r   sysr   __all__objectr   ry   r  r   r   r   r>   r<   r=   r   r%   r   r   r0   r(   r'   r$   r   r   r6   r|   r)   r   ImportErrorr   r   r   r   r    r7   r*   r.   rB   rC   r!   r   r5   r4   r3   r2   r&   r/   r   r   rA   r9   r:   r,   r"   r8   r  r   r  r@   r1   r#   r   r+   r;   r-   r?   r   rM   rK   <module>r+     s      ! %    - , .` (,t #d+K 4$AB% '$3$%+P
40 ! $) ;.	,.6	)8
, !((HJ / %=P2.D8N.*ZJ>1( "# 1("$+ +"'T$',/B(60@*$ %* ( 6', = G&&GO	6
E *"9e  KL  Hs$   	E  %E-  E*)E*-E76E7