U
    ?h                     @   s0   d Z ddlmZ dadd ZG dd deZdS )a  Read FITS files.

Backend Library: `Astropy <https://www.astropy.org/>`_

.. note::
    To use this plugin you have to install its backend::

        pip install imageio[fits]

Flexible Image Transport System (FITS) is an open standard defining a
digital file format useful for storage, transmission and processing of
scientific and other images. FITS is the most commonly used digital
file format in astronomy.


Parameters
----------
cache : bool
    If the file name is a URL, `~astropy.utils.data.download_file` is used
    to open the file.  This specifies whether or not to save the file
    locally in Astropy's download cache (default: `True`).
uint : bool
    Interpret signed integer data where ``BZERO`` is the
    central value and ``BSCALE == 1`` as unsigned integer
    data.  For example, ``int16`` data with ``BZERO = 32768``
    and ``BSCALE = 1`` would be treated as ``uint16`` data.

    Note, for backward compatibility, the kwarg **uint16** may
    be used instead.  The kwarg was renamed when support was
    added for integers of any size.
ignore_missing_end : bool
    Do not issue an exception when opening a file that is
    missing an ``END`` card in the last header.
checksum : bool or str
    If `True`, verifies that both ``DATASUM`` and
    ``CHECKSUM`` card values (when present in the HDU header)
    match the header and data of all HDU's in the file.  Updates to a
    file that already has a checksum will preserve and update the
    existing checksums unless this argument is given a value of
    'remove', in which case the CHECKSUM and DATASUM values are not
    checked, and are removed when saving changes to the file.
disable_image_compression : bool, optional
    If `True`, treats compressed image HDU's like normal
    binary table HDU's.
do_not_scale_image_data : bool
    If `True`, image data is not scaled using BSCALE/BZERO values
    when read.
ignore_blank : bool
    If `True`, the BLANK keyword is ignored if present.
scale_back : bool
    If `True`, when saving changes to a file that contained scaled
    image data, restore the data to the original type and reapply the
    original BSCALE/BZERO values.  This could lead to loss of accuracy
    if scaling back to integer values after performing floating point
    operations on the data.

   )FormatNc                   C   s2   zddl ma W n tk
r,   tdY nX tS )N    )fitszoThe FITS format relies on the astropy package.Please refer to http://www.astropy.org/ for further instructions.)Z
astropy.ior   _fitsImportError r   r   F/var/www/html/venv/lib/python3.8/site-packages/imageio/plugins/fits.pyload_libC   s    
r	   c                   @   s2   e Zd ZdZdd Zdd ZG dd dejZdS )	
FitsFormatzSee :mod:`imageio.plugins.fits`c                 C   s   |j | jkS N)	extension
extensionsselfrequestr   r   r   	_can_readS   s    zFitsFormat._can_readc                 C   s   dS )NFr   r   r   r   r   
_can_writeX   s    zFitsFormat._can_writec                   @   s6   e Zd ZdddZdd Zdd Zdd	 Zd
d ZdS )zFitsFormat.ReaderFc                 K   s   t s
t  t j| j fd|i|}g | _t jt jt jf}t	t
t||D ](\}}t||rN|jdkrN| j| qN|| _d S )Ncacher   )r   r	   openr   Zget_file_indexZImageHDUZ
PrimaryHDUZCompImageHDUziprangelen
isinstancesizeappend_hdulist)r   r   kwargsZhdulistZallowed_hdu_typesnZhdur   r   r   _open_   s    

zFitsFormat.Reader._openc                 C   s   | j   d S r   )r   closer   r   r   r   _closen   s    zFitsFormat.Reader._closec                 C   s
   t | jS r   )r   r   r!   r   r   r   _get_lengthq   s    zFitsFormat.Reader._get_lengthc                 C   s8   |dk s|t | jkrtd| j| j|  j}|i fS )Nr   z*Index out of range while reading from fits)r   r   
IndexErrorr   data)r   indexZimr   r   r   	_get_datat   s    zFitsFormat.Reader._get_datac                 C   s   t dd S )Nz+The fits format does not support meta data.)RuntimeError)r   r&   r   r   r   _get_meta_data|   s    z FitsFormat.Reader._get_meta_dataN)F)__name__
__module____qualname__r   r"   r#   r'   r)   r   r   r   r   Reader^   s
   
r-   N)r*   r+   r,   __doc__r   r   r   r-   r   r   r   r   r
   P   s   r
   )r.   corer   r   r	   r
   r   r   r   r   <module>   s   :