U
    9%e>                     @   s   d dl mZm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dZed#ddZedd Zedd Zedd  Zd!S )$   )defundefun_wrappedc                 C   s<  |  |\}}| |}| j }|sld| jg|dgg ||d  gg g df}|rf|d d  || 7  < |fS | | p| |dkp| |dko| |dk}| jd d }|r| j| j|||ddd	d
}	| j|| j	d|d|d}
n|}
| j|
|
|d}| j
d||d}| j|d	d
}| j|
d	d
}|rdd|
g||gg g || ||d  gg |f}|g}nld|g||gg g || ||d  gg |f}d| j|g|d ddgg || g||d  gd| g|f}||g}|r4| |	}tt|D ]F}|| d d  || 7  < || d | || d d qt|S )z
    Combined calculation of the Hermite polynomial H_n(z) (and its
    generalization to complex n) and the parabolic cylinder
    function D.
             ?r             )prec      пTexact)_convert_paramconvertmpq_1_2piisnpintreZimr	   fmulsqrtZfdivZfnegexprangelenappendtuple)ctxnzZparabolic_cylinderZntypqT1Zcan_use_2f0ZexpprecuwZw2Zrw2Znrw2nwtermsT2Zexpui r%   Z/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/mpmath/functions/orthogonal.py_hermite_param   sB    
&**:
r'   c                    s    j  fddg f|S )Nc                      s   t  dS )Nr   r'   r%   r   r   r   r%   r&   <lambda>>       zhermite.<locals>.<lambda>	hypercombr   r   r   kwargsr%   r)   r&   hermite<   s    r0   c                    s    j  fddg f|S )a8  
    Gives the parabolic cylinder function in Whittaker's notation
    `D_n(z) = U(-n-1/2, z)` (see :func:`~mpmath.pcfu`).
    It solves the differential equation

    .. math ::

        y'' + \left(n + \frac{1}{2} - \frac{1}{4} z^2\right) y = 0.

    and can be represented in terms of Hermite polynomials
    (see :func:`~mpmath.hermite`) as

    .. math ::

        D_n(z) = 2^{-n/2} e^{-z^2/4} H_n\left(\frac{z}{\sqrt{2}}\right).

    **Plots**

    .. literalinclude :: /plots/pcfd.py
    .. image :: /plots/pcfd.png

    **Examples**

        >>> from mpmath import *
        >>> mp.dps = 25; mp.pretty = True
        >>> pcfd(0,0); pcfd(1,0); pcfd(2,0); pcfd(3,0)
        1.0
        0.0
        -1.0
        0.0
        >>> pcfd(4,0); pcfd(-3,0)
        3.0
        0.6266570686577501256039413
        >>> pcfd('1/2', 2+3j)
        (-5.363331161232920734849056 - 3.858877821790010714163487j)
        >>> pcfd(2, -10)
        1.374906442631438038871515e-9

    Verifying the differential equation::

        >>> n = mpf(2.5)
        >>> y = lambda z: pcfd(n,z)
        >>> z = 1.75
        >>> chop(diff(y,z,2) + (n+0.5-0.25*z**2)*y(z))
        0.0

    Rational Taylor series expansion when `n` is an integer::

        >>> taylor(lambda z: pcfd(5,z), 0, 7)
        [0.0, 15.0, 0.0, -13.75, 0.0, 3.96875, 0.0, -0.6015625]

    c                      s   t  dS Nr   r(   r%   r)   r%   r&   r*   v   r+   zpcfd.<locals>.<lambda>r,   r.   r%   r)   r&   pcfd@   s    6r2   c                 K   s"   |  |\}}| | | j |S )a  
    Gives the parabolic cylinder function `U(a,z)`, which may be
    defined for `\Re(z) > 0` in terms of the confluent
    U-function (see :func:`~mpmath.hyperu`) by

    .. math ::

        U(a,z) = 2^{-\frac{1}{4}-\frac{a}{2}} e^{-\frac{1}{4} z^2}
            U\left(\frac{a}{2}+\frac{1}{4},
            \frac{1}{2}, \frac{1}{2}z^2\right)

    or, for arbitrary `z`,

    .. math ::

        e^{-\frac{1}{4}z^2} U(a,z) =
            U(a,0) \,_1F_1\left(-\tfrac{a}{2}+\tfrac{1}{4};
            \tfrac{1}{2}; -\tfrac{1}{2}z^2\right) +
            U'(a,0) z \,_1F_1\left(-\tfrac{a}{2}+\tfrac{3}{4};
            \tfrac{3}{2}; -\tfrac{1}{2}z^2\right).

    **Examples**

    Connection to other functions::

        >>> from mpmath import *
        >>> mp.dps = 25; mp.pretty = True
        >>> z = mpf(3)
        >>> pcfu(0.5,z)
        0.03210358129311151450551963
        >>> sqrt(pi/2)*exp(z**2/4)*erfc(z/sqrt(2))
        0.03210358129311151450551963
        >>> pcfu(0.5,-z)
        23.75012332835297233711255
        >>> sqrt(pi/2)*exp(z**2/4)*erfc(-z/sqrt(2))
        23.75012332835297233711255
        >>> pcfu(0.5,-z)
        23.75012332835297233711255
        >>> sqrt(pi/2)*exp(z**2/4)*erfc(-z/sqrt(2))
        23.75012332835297233711255

    )r   r2   r   )r   ar   r/   r   _r%   r%   r&   pcfux   s    ,r5   c                    s     |\}  j j|dkr d r fdd} j|g f|} r| r| |}|S  fdd} j|gf|S dS )a  
    Gives the parabolic cylinder function `V(a,z)`, which can be
    represented in terms of :func:`~mpmath.pcfu` as

    .. math ::

        V(a,z) = \frac{\Gamma(a+\tfrac{1}{2}) (U(a,-z)-\sin(\pi a) U(a,z)}{\pi}.

    **Examples**

    Wronskian relation between `U` and `V`::

        >>> from mpmath import *
        >>> mp.dps = 25; mp.pretty = True
        >>> a, z = 2, 3
        >>> pcfu(a,z)*diff(pcfv,(a,z),(0,1))-diff(pcfu,(a,z),(0,1))*pcfv(a,z)
        0.7978845608028653558798921
        >>> sqrt(2/pi)
        0.7978845608028653558798921
        >>> a, z = 2.5, 3
        >>> pcfu(a,z)*diff(pcfv,(a,z),(0,1))-diff(pcfu,(a,z),(0,1))*pcfv(a,z)
        0.7978845608028653558798921
        >>> a, z = 0.25, -1
        >>> pcfu(a,z)*diff(pcfv,(a,z),(0,1))-diff(pcfu,(a,z),(0,1))*pcfv(a,z)
        0.7978845608028653558798921
        >>> a, z = 2+1j, 2+3j
        >>> chop(pcfu(a,z)*diff(pcfv,(a,z),(0,1))-diff(pcfu,(a,z),(0,1))*pcfv(a,z))
        0.7978845608028653558798921

    Qr   c                     s    j ddd} t   d}t  | d}|D ]2}|d d |d d |d   q:    d j  }|D ] }|d | |d d q|| S )	Ny             Tr   r   r                 ?   r   )r   r'   r   expjpir   r   )ZjzZT1termsZT2termsTr   r   r   r   rr   r%   r&   h   s    "zpcfv.<locals>.hc           
         s     d}  d} |} j |g}| |   dg|   gg |   gg|f}|g  |   ddgd |   gg |  d  gd g|f} |   \}}|d | |d | ||fD ]$}	|	d d |	d |   q||fS )Nr
   r   r   r   r8   )Zsquare_exp_argr   r   cospi_sinpir   )
r   r    r   elZY1ZY2csY)r   r   r<   r   r%   r&   r=      s    
8LN)r   r   r   Zmpq_1_4isintr-   _is_real_type_re)r   r3   r   r/   Zntyper=   vr%   r;   r&   pcfv   s     

rH   c                    sT     |\}  fdd} |} rP rP |}|S )aI  
    Gives the parabolic cylinder function `W(a,z)` defined in (DLMF 12.14).

    **Examples**

    Value at the origin::

        >>> from mpmath import *
        >>> mp.dps = 25; mp.pretty = True
        >>> a = mpf(0.25)
        >>> pcfw(a,0)
        0.9722833245718180765617104
        >>> power(2,-0.75)*sqrt(abs(gamma(0.25+0.5j*a)/gamma(0.75+0.5j*a)))
        0.9722833245718180765617104
        >>> diff(pcfw,(a,0),(0,1))
        -0.5142533944210078966003624
        >>> -power(2,-0.25)*sqrt(abs(gamma(0.75+0.5j*a)/gamma(0.25+0.5j*a)))
        -0.5142533944210078966003624

    c                  3   s     d j  }  d j   d j   d }  jd d|   } d d j     j  } |d  d j   }| |   j  	d  V  | |    j   	d  V  d S )Nr   y               @   r   r   g      ?r
   )
arggammajZloggammar   r   r   expjr5   r9   )Zphi2rhokCr)   r%   r&   r"     s    ,.",zpcfw.<locals>.terms)r   r   Zsum_accuratelyrE   rF   )r   r3   r   r/   r4   r"   rG   r%   r)   r&   pcfw   s    
	

rQ   c                    sz   |   rd  S |   d rZ|  d r:tdfdd}| j| gf|S  fdd}| j|gf|S )Nr   r   r   z#Gegenbauer function with two limitsc              	      sF   d|  }g g  | g d |g   | g| d gdd  f}|gS Nr   r   r   r%   )r3   a2r:   r   r   r%   r&   r=   =  s    8zgegenbauer.<locals>.hc              	      sF   d  }g g | | g| d |g|  | | g d gdd  f}|gS rR   r%   )r   rS   r:   )r3   r   r%   r&   r=   B  s    8)r   NotImplementedErrorr-   r   r   r3   r   r/   r=   r%   )r3   r   r   r&   
gegenbauer3  s    
rW   c                    s   |   s, fdd}| j||gf|S | sXfdd}| j|| gf|S | |  || j| d|     d d d f| S )Nc                    sJ   g g  |  d g| d  d g|    |  d g d gd d ffS Nr   r   r%   )r   r3   bxr%   r&   r=   K  s    zjacobi.<locals>.hc                    sF   g g   g| d   |  g|  |  |  d g d gd d ffS rX   r%   )r   r3   )rZ   r[   r%   r&   r=   O  s    r   r   )r   r-   rD   Zbinomialhyp2f1)r   r   r3   rZ   r[   r/   r=   r%   rY   r&   jacobiH  s    

r]   c                    s     fdd}| j ||gf|S )Nc                    s4   g g |   d g| d  d g  g| d gffS r1   r%   )r3   rT   r%   r&   r=   Z  s    zlaguerre.<locals>.hr,   rV   r%   rT   r&   laguerreU  s    r^   c                 K   s   |  |rbt|}||dk  d@ rb|s*|S | |}|d| j d k rJ|S |dk rb|  j| 7  _| j| |d dd| d f|S )Nr   r   
   r   )rD   intmagr	   r\   )r   r   r[   r/   rc   r%   r%   r&   legendre^  s    

rd   r   c                    s   |  |}|  |}|s(| j| f|S |dkrP fdd}| j|||gf|S |dkrx fdd}| j|||gf|S tdd S )Nr   c              	      sP   |d }d  d  g|| gg d| g|  | d gd| gdd   f}|fS Nr   r   r%   r   mgr:   r   r%   r&   r=   w  s    Bzlegenp.<locals>.hr8   c              	      sP   |d } d  d g|| gg d| g|  | d gd| gdd   f}|fS re   r%   rf   ri   r%   r&   r=   }  s    Brequires type=2 or type=3)r   rd   r-   
ValueErrorr   r   rg   r   typer/   r=   r%   ri   r&   legenpm  s    

rn   c                    s     |}  |}  dkr, jS |dkrV fdd} j|||gf|S |dkrtdkr fdd} j|||gf|S  fdd} j|||gf|S td	d S )
N)r   r   c                    s     |\}}d|  j }|}d }d }|d }d d }	||||gdd|| gg d| g|  | d gd| g|	f}
| ||gd| |g| | d g| | d |d g|  | d g|d g|	f}|
|fS Nr   r   ro   )r>   r   )r   rg   cossinrB   rA   r3   rZ   r   r    r   r#   r   r   r%   r&   r=     s$       2  zlegenq.<locals>.hr8   r   c                    s     |d jd d gd|  d d|  | d d| d| g| | d g| d gdd|  |  dd|  |  g| d gd f}|gS )Nr   r   r   g      ?r_   )r9   r   )r   rg   r   rs   r%   r&   r=     s    &   c           
         s   d  |  j } |}d }d }|d }d d }||||gdd|| gg d| g|  | d gd| g|f}| |||gdd| |g| | d g| | d |d g|  | d g|d g|f}	||	fS rp   )Zsinpir   r9   )
r   rg   rB   rA   r3   rZ   r   r    r   r#   rs   r%   r&   r=     s"    
   6  rj   )r   nanr-   absrk   rl   r%   rs   r&   legenq  s    


rv   c                 K   sJ   |s,|  |r,t| |d dkr,|d S | j| |dd| d f|S )Nr   r   r   )r   r   rD   rb   rF   r\   r   r   r[   r/   r%   r%   r&   chebyt  s    $ry   c                 K   sV   |s,|  |r,t| |d dkr,|d S |d | j| |d dd| d f| S )Nr   r   r   )r8   r   rw   rx   r%   r%   r&   chebyu  s    $rz   c           
         s     |}  |}     |}|o<|dk} |}|rr|dk rr|rr j|d  |f|S dkr|r|dk r jd S |r|rt||kr jd S  fdd}	n fdd}	 j|	||gf|S )Nr   r   r7   c              
      s   t |}d | d|  d  | |   j  | |   d  |dg}d|  |d  ddd| d| d g}||g g ||  | | d g|d g d d ffS )Nro   r   r   r   )ru   rM   Zfacr   rr   sign)r@   rg   ZabsmrP   Pr   phithetar%   r&   r=     s    , ,"zspherharm.<locals>.hc                    s     | | d s2  | | d s2  d| rJdgdgg g g g dffS  d \}}d |  d|  d  j  | | d  | | d |d |d g}ddddd| d| g}||g d| g|  | d gd| g|d ffS )Nr   r   ro   r   r   g      )r   Zcos_sinrM   r   rK   )r@   rg   rq   rr   rP   r|   r}   r%   r&   r=     s    2   )r   rD   	spherharmzeroru   r-   )
r   r@   rg   r   r~   r/   Zl_isintZ	l_naturalZm_isintr=   r%   r}   r&   r     s"    







	r   N)r   )r   )Z	functionsr   r   r'   r0   r2   r5   rH   rQ   rW   r]   r^   rd   rn   rv   ry   rz   r   r%   r%   r%   r&   <module>   s:   9

7
.
H
B



5

