B
    3Rc                 @   sF   d dl mZ d dlZd dlZd dlZd dlmZ d	ddZdd ZdS )
    )RDConfigN)Image   r   c          	   C   s  t | tr| g} |g}|d kr0tjdddd}|}t| t|ksHt|d }|d }d}	d}
xJ|
t| k r| |
 }||
 }|s|	dt  7 }	n|	d	t  7 }	|
d7 }
qbW |	rtj	d
ddd}|
|	d  W d Q R X yd||jf }t| W n: tk
r2   dd l}|  tj
d|  d}Y n*X d}x"|D ]}tj|s>d}P q>W yt|j W n tk
r   Y nX |S )Nz/usrZlocalbinZcstsr       zens get [ens create {%(smi)s}] E_GIF {} {width %(width)d height %(height)d bgcolor white filename %(name)s format gif frame %(frame)d};zens get [ens create {%(smi)s}] E_GIF {} {width %(width)d height %(height)d bgcolor white filename %(name)s format gif symbolfontsize 24 frame %(frame)d linewidth 2.8 linespacing 4.0};zw+z.cmdF)suffixdelete
z%s < %szCMD: %s
)
isinstancestrospathjoinlenAssertionErrorlocalstempfileNamedTemporaryFilewritenamesystem	Exception	traceback	print_excsysstderrexistsunlink)smilesZ	fileNamessizecmdZdblSizeframeZbaseCmdwidthZheightargsZnDoneZsmir   tmpr   Zres r'   1lib/python3.7/site-packages/rdkit/utils/cactvs.pySmilesToGif
   sN    


r)   c          	   K   sB   t jdd,}t| |jf|}|r0t|j}nd }W d Q R X |S )Nz.gif)r	   )r   r   r)   r   r   open)r    kwargsr&   okZimgr'   r'   r(   SmilesToImage8   s    r-   )r   Nr   r   )	Zrdkitr   r   r   r   ZPILr   r)   r-   r'   r'   r'   r(   <module>   s   
.