B
    3Rc+                 @   sl   d dl T d dlmZ d dlZdddZedkrhd d	lmZ ed
ddZeeddddddddd	 e	  dS )    )*)pidNF      c          
   C   s  |	d kr$| j d d | j d d f}	ttd| | }tj| d }|| }t|| ||  }t|tt|t | }td||||| tj| d }|}t|t}xt|D ]}|	d |	d fg}x^t|D ]R}|	d |t	|  }|	d |t
|  }|||f ||
| 7 }||7 }qW |	d |t	|  }|	d |t
|  }|||f | j|tjt|d |d |d dd ||
| 8 }||7 }qW d S )Nr      r   ih  g     f@zINFO:)Z	edgeColorZ	fillColorclosed)sizeintmathZceilZpifloatZarrayprintrangeZcosZsinappendZdrawPolygonr   ZtransparentZColor)ZcanvasZ
startColorZendColorZstartRadiusZ	endRadiusZnLoopsdegsPerSliceZdegsPerStep
startAngleZ	centerPosdirZnSlicesZ
radPerStepZstepsPerSliceZ
radiusStepZ	colorStepZangleZradiusZcoloriZptsjZxPosZyPos r   1lib/python3.7/site-packages/rdkit/utils/spiral.py
DrawSpiral   s4    

 r   __main__)	SVGCanvas)iX  iX  ztest.svg)r   name)g?g?r   )g?g?g      ?   2      iy   )r   r   r   )r   r   r   Nr   )
ZnumpyZrdkit.spingr   r
   r   __name__Zrdkit.sping.SVG.pidSVGr   ZcanvZsaver   r   r   r   <module>   s    
 