Aluode/PerceptionLabPortable
0
1o
2 ��Yi� � @ s� d dl mZ d dlZd dlZddlmZmZ dd d3�ZG d d � d ej�ZG d
d� dej �Z4e� ej ee� e�
de5� e�ej ddg� dS )� )� annotationsN� )�Image� ImageFile�prefix�bytes�return�boolc C s6 | � d�S )N� SIMPLE)�7startswith)r � r �}E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\PIL/FitsImagePlugin.py�_accept s 8r c @ s2 e Zd ZdZdZddd�Zdd d �Zddd�ZdS )�
FitsImageFileZFITSr �Nonec 9 C sF | j d usJ �i }d}d} | j �d�}|s d}t|��|d d� �� }|dv r+d}n&|r0|s0nS|dkrQ| j �t�| j �� d �d � |sN| �|�\}}}d}q
|rTq
|dd � � d10�d �� } | �11d �rn| d
d � �� } |s~t |�rx| dkr~d}t |��| ||< q|s�d}t
|��|| j �� d 7 }t�|d| j ||�g| _d S )NF� T�P zTruncated FITS file� )r12 � XTENSIONs ENDi@ � /r � =r � TzNot a FITS filez
No image data�r r )�fp�read�OSError�strip�seek�math�ceil�tell�_parse_headers�splitr r � SyntaxError�13ValueErrorr Z_Tile�sizeZtile)14�self�headersZheader_in_progress� decoder_name�header�msg�keyword�offset�args�valuer r r
�_open sF 15� zFitsImageFile._openr' �dict[bytes, bytes]r r �tuple[int, int] | Nonec C sX t ||d �}|dkrd S |dkr dt ||d �fS t ||d �t ||d �fS )Ns NAXISr r s NAXIS1s NAXIS2)�int)r& r' r Znaxisr r r
� _get_sizeH s zFitsImageFile._get_size�&tuple[str, int, tuple[str | int, ...]]c C s� d}d}d}|� d�dkr8|� d�dkr8|d d kr8| �||�p!d16}t|d �}|d |d |d
}d}d}| �||�}|sBdS || _t||d �}|d
krUd| _n|dkr]d| _n|dkred| _n|dv rld| _|dkrw| jddf}n|f}|||fS )N� �rawr r s17 'BINTABLE's ZIMAGEr s ZCMPTYPEs18 'GZIP_1 'r s BITPIXr r � Z� fits_gzip)r r r �L� zI;16� �I)i����i�����F�����)�getr3 r2 �_size�_mode�mode) r& r' r r( r, Zno_prefix_size�number_of_bitsr% r- r r r
r! T s8 19z FitsImageFile._parse_headersN)r r )r' r0 r r r r1 )r' r0 r r4 )�__name__�20__module__� __qualname__�format�format_descriptionr/ r3 r! r r r r
r s 2122- r c @ s e Zd ZdZd dd�ZdS )23�FitsGzipDecoderT�buffer�$bytes | Image.SupportsArrayInterfacer �tuple[int, int]c C s� | j d usJ �t�| j �� �}g }d}t| jd d d�}t| jj�D ]%}t � }t| jj24�D ]}|||d| |d � 7 }|d7 }q.|� |� q#| � t
dd� |d d d� D ��� dS )Nr r � c S s g | ] }|D ]}|�qqS r r )�.0�rowZpixelr r r
�25<listcomp>� s z*FitsGzipDecoder.decode.<locals>.<listcomp>r> )r> r )�fd�gzip�26decompressr �minr- �range�stateZysize� bytearrayZxsize�appendZ27set_as_rawr ) r& rJ r. �rowsr, rC �yrO �xr r r
�decode� s 28 "zFitsGzipDecoder.decodeN)rJ rK r rL )rD rE rF Z _pulls_fdr\ r r r r
rI ~ s rI r8 z.fitz.fits)r r r r )�29__future__r rR r r r r r r Z PyDecoderrI Z
register_openrG Zregister_decoderZregister_extensionsr r r r
�<module> s 3031g