3
€dç  ã               @   s  d Z ddlZddlmZ ejdƒZday ddlZejj	ƒ ejj
ƒ faW n ek
rd   ejdƒ Y nX ejdƒZdejd< d	az<ydd
lmZ daW n" ek
r¶   ejdƒ dZY nX W dedk	rÐeejd< nejd= X dd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )z‚
vtkbase provides some helpers to use VTK through the tvtk package (mayavi)

Code using tvtk should import it through this module
é    Né   )Úloggingznipype.interfacezVTK was not foundZETS_TOOLKITÚnullF)ÚtvtkTztvtk wasn't foundc               C   s   t S )zGet VTK version)Ú_vtk_version© r   r   ú;/tmp/pip-build-7vycvbft/nipype/nipype/interfaces/vtkbase.pyÚvtk_version-   s    r	   c               C   s   t dkS )z?Checks if VTK is installed and the python wrapper is functionalN)r   r   r   r   r   Úno_vtk3   s    r
   c               C   s   t  S )zChecks if tvtk was found)Ú
_have_tvtkr   r   r   r   Úno_tvtk9   s    r   c               C   s   t dkrtdƒ‚t d dk S )z3Checks if VTK uses the old-style pipeline (VTK<6.0)NzVTK is not correctly installed.r   é   )r   ZRuntimeExceptionr   r   r   r   Úvtk_old?   s    r   c             C   s   t ƒ r|| _n
| j|ƒ dS )zh
    Configure the input data for vtk pipeline object obj.
    Copied from latest version of mayavi
    N)r   ÚinputZset_input_data)ÚobjÚdatar   r   r   Úconfigure_input_dataG   s    r   c             C   s   t ƒ r| jS | jƒ S )z5Configure the input data for vtk pipeline object obj.)r   ÚoutputZ
get_output)r   r   r   r   Ú
vtk_outputR   s    r   )Ú__doc__ÚosÚ r   Ú	getLoggerZifloggerr   ZvtkZ
vtkVersionZGetVTKMajorVersionZGetVTKMinorVersionÚImportErrorÚwarningÚgetenvZold_etsÚenvironr   Ztvtk.apir   r	   r
   r   r   r   r   r   r   r   r   Ú<module>   s8   




