o
h @ 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.)
itk_itkimread_read_functionimwrite_write_function ImportError SimpleITK ReadImage
WriteImage r r T/var/www/html/scripts/venv/lib/python3.10/site-packages/imageio/plugins/simpleitk.pyload_lib s"
r )z.giplz.iplz.mhaz.mhdz.nhdrniahdrz.nrrdz.niiz.nii.gzz.imgz.img.gzz.vtkhdf5lsmmncmnc2mghr pic) 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"
pixel_type
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} r0 zItkFormat.Reader._closec C s$ |dkr
d}t |t| ji fS )Nr z.Index out of range while reading from itk file)
IndexErrorr GetArrayFromImager' 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. RuntimeErrorr6 r r r _get_meta_data z ItkFormat.Reader._get_meta_data)NN)__name__
__module__ __qualname__r, r/ r3 r9 r> r r r r Readero s
rC c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
zItkFormat.Writerc C s t st d S d S r2 )r r r. r r r r, s
zItkFormat.Writer._openc C r1 r2 r r. r r r r3 r0 zItkFormat.Writer._closec C s t |}t|| j d S r2 )r GetImageFromArrayr
r# r& )r" immeta_itk_imgr r r _append_data s
z ItkFormat.Writer._append_datac C r: r; r< )r" rF r8 r r r
set_meta_data r? z ItkFormat.Writer.set_meta_dataN)r@ rA rB r, r3 rH rI r r r r Writer s
rJ N) r@ rA rB __doc__r$ r% r rC rJ r r r r r Z s
r ) rK corer r r r r r r r r r r s