o
    FZhF                     @   sf  d dl mZ G dd deZdd Z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edd Zedd Zedd Zed d! Zed"d# Zedhd%d&Zed'd( Zed)d* Zed+d, Zed-d. Zed/d0 Zed1d2 Zedid4d5Zedjd7d8Zed9d: Zed;d< Zed=d> Zed?d@ Z edAdB Z!edCdD Z"edEdF Z#edkdHdIZ$edJdK Z%edLdM Z&edNdO Z'edPdQ Z(dRdS Z)d3dGl*Z*d3dGl+Z+dTdU Z,dVdW Z-edidXdYZ.edhdZd[Z/djd\d]Z0ed^d_ Z1ed`da Z2edbdc Z3edjdddeZ4edjdfdgZ5dGS )l   )xrangec                   @   s   e Zd ZdZi ZdZ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dd Zdd Zdd Zdd Zdd Zdd ZdS ) SpecialFunctionsa  
    This class implements special functions using high-level code.

    Elementary and some other functions (e.g. gamma function, basecase
    hypergeometric series) are assumed to be predefined by the context as
    "builtins" or "low-level" functions.
    gP?c              
   C   s.  | j }|jD ]}|j| \}}|||| q| d| _| d| _| d| _| d| _| d| _| d| _	| d| _
| d| _| d	| _| d
| _| d| _| d| _| d| _| d| _| d| _| d| _| d| _i | _| jdddddddd | | j| _d S )N)   r       r   )r   r   )   r   )r      )r      )r   r	   )   r   )r   r   )   r   )r
   r   )r   r   )r   r   )r   r   )r      )r
   r   )r
   r   argconjrootpsizetaZfibZfac)Zphase	conjugateZnthrootZ	polygammaZhurwitzZ	fibonacci	factorial)	__class__defined_functions_wrap_specfunZ_mpqZmpq_1Zmpq_0Zmpq_1_2Zmpq_3_2Zmpq_1_4Zmpq_1_16Zmpq_3_16Zmpq_5_2Zmpq_3_4Zmpq_7_4Zmpq_5_4Zmpq_1_3Zmpq_2_3Zmpq_4_3Zmpq_1_6Zmpq_5_6Zmpq_5_3Z_misc_const_cache_aliasesupdatememoizeZzetazeroZzetazero_memoized)selfclsnamefwrap r   I/var/www/auris/lib/python3.10/site-packages/mpmath/functions/functions.py__init__   s@   
zSpecialFunctions.__init__c                 C   s   t | || d S N)setattr)r   r   r   r   r   r   r    r   =   s   zSpecialFunctions._wrap_specfunc                 C      t r"   NotImplementedErrorctxnzr   r   r    _besseljD       zSpecialFunctions._besseljc                 C   r$   r"   r%   r(   r*   r   r   r    _erfE   r,   zSpecialFunctions._erfc                 C   r$   r"   r%   r-   r   r   r    _erfcF   r,   zSpecialFunctions._erfcc                 C   r$   r"   r%   )r(   r*   ar   r   r    _gamma_upper_intG   r,   z!SpecialFunctions._gamma_upper_intc                 C   r$   r"   r%   r'   r   r   r    _expint_intH   r,   zSpecialFunctions._expint_intc                 C   r$   r"   r%   r(   sr   r   r    _zetaI   r,   zSpecialFunctions._zetac                 C   r$   r"   r%   )r(   r4   r0   r)   ZderivativesZreflectr   r   r    _zetasum_fastJ   r,   zSpecialFunctions._zetasum_fastc                 C   r$   r"   r%   r-   r   r   r    _eiK   r,   zSpecialFunctions._eic                 C   r$   r"   r%   r-   r   r   r    _e1L   r,   zSpecialFunctions._e1c                 C   r$   r"   r%   r-   r   r   r    _ciM   r,   zSpecialFunctions._cic                 C   r$   r"   r%   r-   r   r   r    _siN   r,   zSpecialFunctions._sic                 C   r$   r"   r%   r3   r   r   r    _altzetaO   r,   zSpecialFunctions._altzetaN)__name__
__module____qualname____doc__r   ZTHETA_Q_LIMr!   classmethodr   r+   r.   r/   r1   r2   r5   r6   r7   r8   r9   r:   r;   r   r   r   r    r      s&    +
r   c                 C      | dft j| j< | S )NTr   r   r<   r   r   r   r    defun_wrappedQ      rD   c                 C   rA   )NFrB   rC   r   r   r    defunU   rE   rF   c                 C   s   t t| j|  | S r"   )r#   r   r<   rC   r   r   r    defun_staticY   s   rG   c                 C      | j | | S r"   )onetanr-   r   r   r    cot]      rK   c                 C   rH   r"   )rI   cosr-   r   r   r    sec`   rL   rN   c                 C   rH   r"   )rI   sinr-   r   r   r    cscc   rL   rP   c                 C   rH   r"   )rI   tanhr-   r   r   r    cothf   rL   rR   c                 C   rH   r"   )rI   coshr-   r   r   r    sechi   rL   rT   c                 C   rH   r"   )rI   sinhr-   r   r   r    cschl   rL   rV   c                 C      |s| j d S | | j| S )N      ?)piatanrI   r-   r   r   r    acoto      
r[   c                 C      |  | j| S r"   )acosrI   r-   r   r   r    asecv   rL   r_   c                 C   r]   r"   )asinrI   r-   r   r   r    acscy   rL   ra   c                 C   rW   )Ny              ?)rY   atanhrI   r-   r   r   r    acoth|   r\   rc   c                 C   r]   r"   )acoshrI   r-   r   r   r    asech   rL   re   c                 C   r]   r"   )asinhrI   r-   r   r   r    acsch   rL   rg   c                 C   sH   |  |}|r| |r|S | |r|dkr| jS | j S |t| S )Nr   )convertisnanZ_is_real_typerI   absr(   xr   r   r    sign   s   

rm   r   c                 C   s2   |dkr	|  |S | |}| |}| ||S Nr   )Zagm1rh   Z_agm)r(   r0   br   r   r    agm   s
   


rp   c                 C   s,   |  |r	d| S |s|d S | || S rn   )isinfrO   rk   r   r   r    sinc   s
   
rr   c                 C   s2   |  |r	d| S |s|d S | || j|  S rn   )rq   ZsinpirY   rk   r   r   r    sincpi   s
   
rs   c                    sB   s j S   j k rdd   S   fdddS )NrX   r   c                      s   t  dgS N)iterexpr   rk   r   r    <lambda>       zexpm1.<locals>.<lambda>r   )zeromagprecsum_accuratelyrk   r   rk   r    expm1   s
   r~   c                 C   sH   |s| j S | || j k r|d|d   S | | jd|d| j dS )NrX   r   r   r|   )rz   r{   r|   logZfaddrk   r   r   r    log1p   s
   r   c           
         s   | j }| j}  | }||}|dkr|S |s%r# dv r%| r%|S  | }|}|  }	|||	 | j k rH|	 |	 d d  S |  fdddS )Ni)r   ru   y              ?y             r   c                      s   t   dgS rt   )rv   r   rl   yr   r    rx      s    zpowm1.<locals>.<lambda>r   )r{   rI   Zisintlnr|   r}   )
r(   rl   r   r{   rI   wMx1ZmagyZlnxr   r   r    powm1   s   
r   c                 C   sx   t |}t |}||; }|s| jS d| |kr| j S d| |kr$| jS d| d| kr0| j S | d| | | S )Nr   r   r   )intrI   jZexpjpimpf)r(   kr)   r   r   r    _rootof1   s   r   r   c                 C   s   t |}| |}|rN|d@ r+d| |d kr+| |s+| |dk r+| | | S | j}z|  jd7  _| ||d| || }W || _|
 S || _w | ||S )Nr   r   r   
   )r   rh   imrer   r|   r   Z_nthroot)r(   rl   r)   r   r|   vr   r   r    r      s   
0r   Fc                    st    j  j}z(  jd7  _|r fddtD }n fddtD }W | _n| _w dd |D S )Nr   c                    s&   g | ]}|d kr  |qS r   r   .0r   r(   gcdr)   r   r    
<listcomp>   s   & zunitroots.<locals>.<listcomp>c                    s   g | ]}  |qS r   r   r   )r(   r)   r   r    r     s    c                 S   s   g | ]}|
 qS r   r   )r   rl   r   r   r    r     ry   )Z_gcdr|   range)r(   r)   Z	primitiver|   r   r   r   r    	unitroots   s   r   c                 C   s*   |  |}| |}| |}| ||S r"   )rh   _re_imatan2)r(   rl   r   r   r   r   r    r     s   


r   c                 C   s   t | |S r"   )rj   rh   rk   r   r   r    fabs  s   r   c                 C   s   |  |}t|dr|jS |S )Nreal)rh   hasattrr   rk   r   r   r    r     s   

r   c                 C   s    |  |}t|dr|jS | jS )Nimag)rh   r   r   rz   rk   r   r   r    r     s   

r   c                 C   s,   |  |}z| W S  ty   | Y S w r"   )rh   r   AttributeErrorrk   r   r   r    r      s   

r   c                 C   s   |  || |fS r"   )r   r   r-   r   r   r    polar(     r   c                 C   s   || j | |  S r"   )ZmpcZcos_sin)r(   rphir   r   r    rect,  r   r   Nc                 C   s8   |d u r	|  |S | jd }| j ||d| j ||d S )N   r   )r   r|   )r(   rl   ro   wpr   r   r    r   0  s   

r   c                 C   s   |  |dS )Nr   )r   rk   r   r   r    log107  s   r   c                 C   s   |  ||  | S r"   )rh   )r(   rl   r   r   r   r    fmod;  r   r   c                 C   s
   || j  S r"   Zdegreerk   r   r   r    degrees?     
r   c                 C   s
   || j  S r"   r   rk   r   r   r    radiansC  r   r   c                 C   sv   |s|s|S | j | S || jkr"|dkr|S |d| | j | j  S || j kr6| d| d | j | j  S | |S )Nr   r   r   )ZninfinfrY   r   r   )r(   r*   r   r   r   r    _lambertw_specialG  s   



r   c                 C   s  d}t | drt| j}| j}|rd|dk  }t|}nt| }d}d}|s(d}t||} |dkrd|  k r;dk rn nd|  k rGdk rn nt|r{|d	krWd
d| d   S |dkrcdd| d   S |dk rodd| d   S |dk r{dd| d   S |dk r|dkrdd| d   S dd| d   S d}|s||kr|} |dk rdd | | d!   d"| |   S |d!k r| S d#d$|   S |s|dkrt|}t|}nt| }t|}n|dkrXd}|s||  k rdk rn n|} |dkr|d%k rd&|  k rdk rn ndd | | d!   d"| |   S |s7d|  kr'dk r7n nt| }|t|  S |dkrL|sL|dk rLt| d' }nt| d( }t|}|| ||  ||d)  d)|d)    S )*Nr   r   ru   g        g      g      @g      g      @      ?yx&1?p=
ף?yh|?5?ʡEƿy      ?      @g      ?y)\(?&1?y      ?L7A`y      ?      ?yx&1?p=
ףyh|?5?ʡE?y      ?      g      пy)\(?&1ʿy      ?L7A`?y      ?      g      y'1ZԿq=
ףp?yM`"r   y'1ZԿq=
ףpyM`"?g2,6V׿gɿg4@rX   g}tp?g?g333333?g?g333333y        -DT!	@y        -DT!@r   )r   floatr   r   complexmathr   cmath)r*   r   Z	imag_signrl   r   r   L1L2r   r   r    _lambertw_approx_hybridZ  s`   



0 
0 "
(r   c                    s    }d|  k rdk rn nd|  k rdk rn n|dk r	td dk r	|dksJ|d	kr= dksJ|dkr	 dk r	  fd
d}  | }  j|7  _ d j d  }  j|8  _ d	 dd d d	d}|dkr| } j}	t	t
d|D ]`vr݈ fddt	dD |< d d  d |d  d   d  | d  d  d   <  |  }
|	|
7 }	  |
| k r|	df  S d7 q  j|d 7  _|	dfS |dks|d	krt|dfS |dkr7|d	k r,d  dfS  } |}n=|d	krc scd   k rQdk rcn n  }| |  dfS  d j |  } |}|| ||  ||d  d|d    dfS )z
    Return rough approximation for W_k(z) from an asymptotic series,
    sufficiently accurate for the Halley iteration to converge to
    the correct value.
    ii  ii  r   g,6V?g?r   ru   c                      s     dgS rt   )rw   r   r-   r   r    rx     s    z"_lambertw_series.<locals>.<lambda>r   r   c                 3   s(    | ]}|  d  |   V  qdS )r   Nr   )r   r   )lur   r    	<genexpr>  s   & z#_lambertw_series.<locals>.<genexpr>r   TFg,6V׿y               @)r{   rj   r   r}   r|   sqrter   rz   r   maxfsumr   r   r   rY   )r(   r*   r   tolZmagzdeltaZcancellationpr0   r4   termr   r   r   )r(   r   r   r*   r    _lambertw_series  sN   
6$T



8
,r   c                 C   s  |  |}t|}| |st| ||S | j}|  jd| |p d 7  _| j}|d }t| |||\}}|s| d}tdD ]7}	| 	|}
||
 }|| }||||
 || | || |     }| || | || kru|} n|}q@|	dkr| 
d|  || _|
 S )Nr   r   r
   r   d   z1Lambert W iteration failed to converge for z = %s)rh   r   Zisnormalr   r|   r{   r   r   r   rw   warn)r(   r*   r   r|   r   r   r   donetwoiewZwewZwewzZwnr   r   r    lambertw  s0   



(r   c                 C   s   |  |}|s| |r|S t|dS | |s(| |s(| |s(| |r,|| S |dkr2|S |dkr<||d  S |dkrE| |S t| ||d| | S )Nr   r   r   T)rh   ri   typerq   rs   _polyexprw   )r(   r)   rl   r   r   r    bell  s   

(r   c                    s     fdd} j |ddS )Nc                  3   s@    r	  V  } d}	 | |  V  |d7 }|  | } qrn   )rs   )tr   r(   extrar)   rl   r   r    _terms  s   z_polyexp.<locals>._termsr   )Z
check_step)r}   )r(   r)   rl   r   r   r   r   r    r     s   	r   c                 C   s   |  |s|  |s| |s| |r|| S |dkr || S |dkr)| |S |dkr4| || S |dkrC| || |d  S t| ||S )Nr   r   r   )rq   ri   r~   rw   r   )r(   r4   r*   r   r   r    polyexp  s   (r   c                 C   s  t |}|dk rtd| j}|dkr|S |dkr|| S |dkr%|| S d}d}d}d}td|d D ]6}|| sj| || }	| || }
|
rQ||
|	 9 }q4|	dkr^||9 }|d7 }q4|	dkrj||9 }|d7 }q4|r||krw|d9 }|S ||9 }|| }|S )Nr   zn cannot be negativer   r   ru   )r   
ValueErrorrI   r   Zmoebiusr   )r(   r)   r*   r   Za_prodZb_prodZ	num_zerosZ	num_polesdr   ro   r   r   r    
cyclotomic  sD   
r   c                 C   s  t |}|dk r| jS |d dkr ||d @ dkr| j
 S | jS dD ]*}|| sL|| d}}|dkrEt||\}}|rA| j  S |dks3| |  S q"| |rW| |S |dkr]td}	 t |d|  d }|dk rq| jS || |kr| |r| |S |d7 }q`)a  
    Evaluates the von Mangoldt function `\Lambda(n) = \log p`
    if `n = p^k` a power of a prime, and `\Lambda(n) = 0` otherwise.

    **Examples**

        >>> from mpmath import *
        >>> mp.dps = 25; mp.pretty = True
        >>> [mangoldt(n) for n in range(-2,3)]
        [0.0, 0.0, 0.0, 0.0, 0.6931471805599453094172321]
        >>> mangoldt(6)
        0.0
        >>> mangoldt(7)
        1.945910149055313305105353
        >>> mangoldt(8)
        0.6931471805599453094172321
        >>> fsum(mangoldt(n) for n in range(101))
        94.04531122935739224600493
        >>> fsum(mangoldt(n) for n in range(10001))
        10013.39669326311478372032

    r   r   r   )
r   r
   r                        l       73Me'r   rX   )r   rz   Zln2divmodr   Zisprimer&   )r(   r)   r   qr   r   r   r   r    mangoldt;  s>   




r   c                 C   *   |  t|t|}|rt|S | |S r"   )Z
_stirling1r   r   r(   r)   r   exactr   r   r   r    	stirling1w     
r   c                 C   r   r"   )Z
_stirling2r   r   r   r   r   r    	stirling2  r   r   r   )r   )Fr"   )6Zlibmp.backendr   objectr   rD   rF   rG   rK   rN   rP   rR   rT   rV   r[   r_   ra   rc   re   rg   rm   rp   rr   rs   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    <module>   s    N















	













?6

	
*
;