File: //home/arjun/projects/env/lib/python3.10/site-packages/docx/image/__pycache__/png.cpython-310.pyc
o
$we � @ s� d dl mZmZ d dlmZ d dlmZmZ d dlm Z G dd� de �Z
G dd� d�Z G d d
� d
�Z G d d � d �Z
d
d� ZG dd� d�ZG dd� de�ZG dd� de�ZdS )� )� MIME_TYPE�PNG_CHUNK_TYPE)�InvalidImageStreamError)�
BIG_ENDIAN� StreamReader)�BaseImageHeaderc @ s4 e Zd ZdZedd� �Zedd� �Zedd� �ZdS ) �Pngz#Image header parser for PNG images.c C s t jS )zUMIME content type for this image, unconditionally `image/png` for PNG
images.)r �PNG��self� r �G/home/arjun/projects/env/lib/python3.10/site-packages/docx/image/png.py� content_type
s zPng.content_typec C s dS )z8Default filename extension, always 'png' for PNG images.�pngr r
r r r
� default_ext s zPng.default_extc C s0 t �|�}|j}|j}|j}|j}| ||||�S )zWReturn a |Png| instance having header properties parsed from image in
`stream`.)�
_PngParser�parse�px_width� px_height�horz_dpi�vert_dpi)�cls�stream�parserr r r r r r r
� from_stream s
zPng.from_streamN) �__name__�
__module__� __qualname__�__doc__�propertyr r � classmethodr r r r r
r s
r c sh e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Z ed
d � �Z
ed d
� �Z e dd� �Z
� ZS )r zNParses a PNG image stream to extract the image properties found in its chunks.c � t t| ��� || _d S �N)�superr �__init__�_chunks)r �chunks�� __class__r r
r$ & �
z_PngParser.__init__c C s t �|�}| |�S )znReturn a |_PngParser| instance containing the header properties parsed from
the PNG image in `stream`.)�_Chunksr )r r r& r r r
r * �
z_PngParser.parsec C � | j j}|jS )z.The number of pixels in each row of the image.)r% �IHDRr �r r- r r r
r 1 � z_PngParser.px_widthc C r, )z2The number of stacked rows of pixels in the image.)r% r- r r. r r r
r 7 r/ z_PngParser.px_heightc C �$ | j j}|du r
dS | �|j|j�S )z�Integer dots per inch for the width of this image.
Defaults to 72 when not present in the file, as is often the case.
N�H )r% �pHYs�_dpi�units_specifier�horz_px_per_unit�r r2 r r r
r = � z_PngParser.horz_dpic C r0 )z�Integer dots per inch for the height of this image.
Defaults to 72 when not present in the file, as is often the case.
Nr1 )r% r2 r3 r4 �vert_px_per_unitr6 r r r
r H r7 z_PngParser.vert_dpic C s | dkr|rt t|d ��S dS )zWReturn dots per inch value calculated from `units_specifier` and
`px_per_unit`.r g�
F%u�?r1 )�int�round)r4 � px_per_unitr r r
r3 S s z_PngParser._dpi)r r r r r$ r r r r r r r � staticmethodr3 �
__classcell__r r r' r
r # s
r c sL e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Z d
d � Z
� Z S ) r* z8Collection of the chunks parsed from a PNG image stream.c s t t| ��� t|�| _d S r" )r# r* r$ �listr% )r �chunk_iterabler' r r
r$ _ s z_Chunks.__init__c C s t �|�}t|�� �}| |�S )zBReturn a |_Chunks| instance containing the PNG chunks in `stream`.)� _ChunkParserr r> � iter_chunks)r r � chunk_parserr& r r r
r c s
z_Chunks.from_streamc C s& dd� }| � |�}|du rtd��|S )zIHDR chunk in PNG image.c S � | j tjkS r" )� type_namer r- ��chunkr r r
�<lambda>m � z _Chunks.IHDR.<locals>.<lambda>Nzno IHDR chunk in PNG image)� _find_firstr )r �matchr- r r r
r- j s
z _Chunks.IHDRc C s dd� }| � |�S )z2PHYs chunk in PNG image, or |None| if not present.c S rC r" )rD r r2 rE r r r
rG v rH z _Chunks.pHYs.<locals>.<lambda>)rI )r rJ r r r
r2 s s
z _Chunks.pHYsc C s | j D ]
}||�r
| S qdS )zGReturn first chunk in stream order returning True for function `match`.N)r% )r rJ rF r r r
rI y s
�z_Chunks._find_first) r r r r r$ r r r r- r2 rI r= r r r' r
r* \ s
r* c s< e Zd ZdZ� fdd�Zedd� �Zdd� Zdd � Z� Z S )
r@ z(Extracts chunks from a PNG image stream.c r! r" )r# r@ r$ � _stream_rdr)r �
stream_rdrr' r r
r$ � r) z_ChunkParser.__init__c C s t |t�}| |�S )zdReturn a |_ChunkParser| instance that can extract the chunks from the PNG
image in `stream`.)r r )r r rL r r r
r � r+ z_ChunkParser.from_streamc c s, � | � � D ]\}}t|| j|�}|V qdS )z�Generate a |_Chunk| subclass instance for each chunk in this parser's PNG
stream, in the order encountered in the stream.N)�_iter_chunk_offsets�
_ChunkFactoryrK )r �
chunk_type�offsetrF r r r
rA � s
��z_ChunkParser.iter_chunksc c sT � d} | j �|�}| j �d|d�}|d }||fV |dkr!dS |d| d 7 }q)z�Generate a (chunk_type, chunk_offset) 2-tuple for each of the chunks in the
PNG image stream.
Iteration stops after the IEND chunk is returned.
� T� �IENDN)rK � read_long�read_str)r � chunk_offset�chunk_data_lenrO � data_offsetr r r
rM � s �
�z _ChunkParser._iter_chunk_offsets)
r r r r r$ r r rA rM r= r r r' r
r@ � s
r@ c C s* t jtt jti}|�| t�}|�| ||�S )ziReturn a |_Chunk| subclass instance appropriate to `chunk_type` parsed from
`stream_rdr` at `offset`.)r r- �
_IHDRChunkr2 �
_pHYsChunk�get�_Chunk� from_offset)rO rL rP �
chunk_cls_map� chunk_clsr r r
rN � s
� rN c s8 e Zd ZdZ� fdd�Zedd� �Zedd� �Z� Z S )r\ zUBase class for specific chunk types.
Also serves as the default chunk type.
c r! r" )r# r\ r$ � _chunk_type)r rO r' r r
r$ � r) z_Chunk.__init__c C s | |�S )z@Return a default _Chunk instance that only knows its chunk type.r )r rO rL rP r r r
r] � s z_Chunk.from_offsetc C � | j S )z.The chunk type name, e.g. 'IHDR', 'pHYs', etc.)r` r
r r r
rD � s z_Chunk.type_name)
r r r r r$ r r] r rD r= r r r' r
r\ � s
r\ c sD e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Z � Z
S )
rY z*IHDR chunk, contains the image dimensions.c s t t| ��|� || _|| _d S r" )r# rY r$ � _px_width�
_px_height)r rO r r r' r r
r$ � s
z_IHDRChunk.__init__c C s" |� |�}|� |d�}| |||�S )z|Return an _IHDRChunk instance containing the image dimensions extracted from
the IHDR chunk in `stream` at `offset`.rR )rT )r rO rL rP r r r r r
r] � s
z_IHDRChunk.from_offsetc C ra r" )rb r
r r r
r � � z_IHDRChunk.px_widthc C ra r" )rc r
r r r
r � rd z_IHDRChunk.px_height) r r r r r$ r r] r r r r= r r r' r
rY � s
rY c sP e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Z ed
d � �Z
� Z S ) rZ z/PYHs chunk, contains the image dpi information.c s& t t| ��|� || _|| _|| _d S r" )r# rZ r$ �_horz_px_per_unit�_vert_px_per_unit�_units_specifier)r rO r5 r8 r4 r' r r
r$ � s
z_pHYsChunk.__init__c C s0 |� |�}|� |d�}|�|d�}| ||||�S )z{Return a _pHYsChunk instance containing the image resolution extracted from
the pHYs chunk in `stream` at `offset`.rR rQ )rT � read_byte)r rO rL rP r5 r8 r4 r r r
r] � s
z_pHYsChunk.from_offsetc C ra r" )re r
r r r
r5 � rd z_pHYsChunk.horz_px_per_unitc C ra r" )rf r
r r r
r8 � rd z_pHYsChunk.vert_px_per_unitc C ra r" )rg r
r r r
r4 � rd z_pHYsChunk.units_specifier) r r r r r$ r r] r r5 r8 r4 r= r r r' r
rZ � s
rZ N)� constantsr r �
exceptionsr �helpersr r �imager r r r* r@ rN r\ rY rZ r r r r
�<module> s 9%'