o
e @ 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 s^ z
dd l atjatjaW tS ty. zdd latjatj aW Y tS ty- tdw w )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
ID:\Projects\ConvertPro\env\Lib\site-packages\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 , |j tv rdS tdstdr|j tv S d S NTzitk.ImageIOBaser extension ITK_FORMATSr ALL_FORMATSselfrequestr
r
r _can_read] s
zItkFormat._can_readc C r 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 sN t st d}|d ur||f7 }|d ur||f7 }t| j g|R | _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 z ItkFormat.Reader._get_lengthc C d S Nr
r! r
r
r _close} r# zItkFormat.Reader._closec C s$ |dkr
d}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 d}t |Nz5The itk plugin does not support meta data, currently. RuntimeErrorr( r
r
r _get_meta_data z ItkFormat.Reader._get_meta_data)NN)__name__
__module__ __qualname__r r" r& r+ r0 r
r
r
r Readero s
r5 c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
zItkFormat.Writerc C s t st d S d S r% )r r r! r
r
r r s
zItkFormat.Writer._openc C r$ r% r
r! r
r
r r& r# 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 r, r- r. )r r6 r* r
r
r
set_meta_data r1 z ItkFormat.Writer.set_meta_dataN)r2 r3 r4 r r& r7 r8 r
r
r
r Writer s
r9 N) r2 r3 r4 __doc__r r r r5 r9 r
r
r
r r Z s
r ) r: corer r r r r r r r
r
r
r s