U
    Hvf5  ã                   @   sj   d Z ddlZz"ddlZddlmZ e d¡ZW n ek
rB   Y nX dd„ ZdZdd	„ Z	e
d
krfe	ƒ  dS )zÊPrecompute the polynomials for the asymptotic expansion of the
generalized exponential integral.

Sources
-------
[1] NIST, Digital Library of Mathematical Functions,
    https://dlmf.nist.gov/8.20#ii

é    N)ÚPolyÚxc                 C   s^   t dtƒg}t| ƒD ]D}| t dd| t  tƒ||  t ttd  ƒ||  ¡   ¡ q|S )Né   é   )r   r   ÚrangeÚappendZdiff)ÚKÚAÚk© r   úF/tmp/pip-unpacked-wheel-96ln3f52/scipy/special/_precompute/expn_asy.pyÚ
generate_A   s    Br   ze/* This file was automatically generated by _precompute/expn_asy.py.
 * Do not edit it manually!
 */
c               	   C   sü   t tƒ tj ddd¡} d}t|ƒ}t| d dƒ°}| t¡ | d 	t
|ƒ¡¡ t|ƒD ]2\}}d d	d
„ | ¡ D ƒ¡}| d 	||¡¡ qZd dd
„ t|d ƒD ƒ¡}| d 	|¡¡ d dd
„ |D ƒ¡}| d 	|¡¡ W 5 Q R X t | d | ¡ d S )Nz..Zcepheszexpn.hé   z.newÚwz#define nA {}
z, c                 S   s   g | ]}t | d ¡ƒ‘qS )é   )ÚstrZevalf)Ú.0r   r   r   r   Ú
<listcomp>,   s     zmain.<locals>.<listcomp>z$static const double A{}[] = {{{}}};
c                 S   s   g | ]}d   |¡‘qS )zA{})Úformat)r   r
   r   r   r   r   .   s     r   z#static const double *A[] = {{{}}};
c                 S   s   g | ]}t | ¡ ƒ‘qS r   )r   Zdegree)r   ÚAkr   r   r   r   0   s     z#static const int Adegs[] = {{{}}};
)ÚprintÚ__doc__ÚosÚpathÚjoinr   ÚopenÚwriteÚWARNINGr   ÚlenÚ	enumerateZcoeffsr   Úrename)Úfnr   r	   Úfr
   r   Útmpr   r   r   Úmain"   s    
r$   Ú__main__)r   r   Zsympyr   Úsymbolsr   ÚImportErrorr   r   r$   Ú__name__r   r   r   r   Ú<module>   s   	