U
a+d @ s@ d Z ddlmZmZ dadd ZdZed ZG dd d eZdS )
an Read/Write images using SimpleITK.
Backend: `Insight Toolkit `_
.. note::
To use this plugin you have to install its backend::
pip install imageio[itk]
The ItkFormat uses the ITK or SimpleITK library to support a range of
ITK-related formats. It also supports a few common formats (e.g. PNG and JPEG).
Parameters
----------
None
)Format
has_moduleNc C sh zdd l atjatjaW nJ tk
rb zdd latjatj aW n tk
r\ tdY nX Y nX tS )N zitk could not be found. Please try python -m pip install itk or python -m pip install simpleitk or refer to https://itkpythonpackage.readthedocs.io/ for further instructions.)
Zitk_itkZimread_read_functionZimwrite_write_function ImportError SimpleITKZ ReadImageZ
WriteImage r
r
=/tmp/pip-unpacked-wheel-xbmu82vq/imageio/plugins/simpleitk.pyload_lib s
r )z.giplz.iplz.mhaz.mhdz.nhdrZniahdrz.nrrdz.niiz.nii.gzz.imgz.img.gzz.vtkZhdf5ZlsmmncZmnc2Zmghr Zpic) z.bmpz.jpegz.jpgz.pngz.tiffz.tifz.dicomz.dcmz.gdcmc @ sD e Zd ZdZdd Zdd ZG dd dejZG dd d ejZd
S ) ItkFormatz$See :mod:`imageio.plugins.simpleitk`c C s, |j tkrdS tds tdr(|j tkS d S NTzitk.ImageIOBaser extension ITK_FORMATSr ALL_FORMATSselfrequestr
r
r _can_read] s
zItkFormat._can_readc C s, |j tkrdS tds tdr(|j tkS d S r r r r
r
r
_can_writeg s
zItkFormat._can_writec @ s6 e Zd Zd ddZdd Zdd Zdd Zd
d ZdS )
zItkFormat.ReaderNc K sL t s
t d}|d k r2||f7 }|d k r2||f7 }t| j f| | _d S )Nr
)r r r r get_local_filename_img)r Z
pixel_typeZ
fallback_onlykwargsargsr
r
r _openp s
zItkFormat.Reader._openc C s dS )N r
r r
r
r _get_lengthz s z ItkFormat.Reader._get_lengthc C s d S Nr
r r
r
r _close} s zItkFormat.Reader._closec C s$ |dkrd}t |t| ji fS )Nr z.Index out of range while reading from itk file)
IndexErrorr ZGetArrayFromImager r index error_msgr
r
r _get_data s zItkFormat.Reader._get_datac C s d}t |d S Nz5The itk plugin does not support meta data, currently. RuntimeErrorr% r
r
r _get_meta_data s z ItkFormat.Reader._get_meta_data)NN)__name__
__module__ __qualname__r r! r# r( r, r
r
r
r Readero s
r0 c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
zItkFormat.Writerc C s t s
t d S r" )r r r r
r
r r s zItkFormat.Writer._openc C s d S r" r
r r
r
r r# s zItkFormat.Writer._closec C s t |}t|| j d S r" )r ZGetImageFromArrayr r r )r ZimmetaZ_itk_imgr
r
r _append_data s
z ItkFormat.Writer._append_datac C s d}t |d S r) r* )r r1 r' r
r
r
set_meta_data s z ItkFormat.Writer.set_meta_dataN)r- r. r/ r r# r2 r3 r
r
r
r Writer s r4 N) r- r. r/ __doc__r r r r0 r4 r
r
r
r r Z s
r ) r5 corer r r r r r r r
r
r
r s