o jq i ã@s¬ddlmZmZddlmZddlmZmZddlm Z Gdd„de ƒZ Gdd„dƒZ Gd d „d ƒZ Gd d „d ƒZ d d„ZGdd„dƒZGdd„deƒZGdd„deƒZdS)é)Ú MIME_TYPEÚPNG_CHUNK_TYPE)ÚInvalidImageStreamError)Ú BIG_ENDIANÚ StreamReader)ÚBaseImageHeaderc@s4eZdZdZedd„ƒZedd„ƒZedd„ƒZdS) ÚPngz#Image header parser for PNG images.cCstjS)zUMIME content type for this image, unconditionally `image/png` for PNG images.)rZPNG©Úself©r úU/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/docx/image/png.pyÚ content_type szPng.content_typecCsdS)z8Default filename extension, always 'png' for PNG images.Zpngr r r r r Ú default_extszPng.default_extcCs0t |¡}|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Úparserrrrrr r r Ú from_streams zPng.from_streamN) Ú__name__Ú __module__Ú __qualname__Ú__doc__Úpropertyr rÚ classmethodrr r r r rs  rcsheZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ ed d „ƒZ ed d „ƒZ e dd„ƒZ ‡ZS)rzNParses a PNG image stream to extract the image properties found in its chunks.cótt|ƒ ¡||_dS©N)ÚsuperrÚ__init__Ú_chunks)r Úchunks©Ú __class__r r r"&ó z_PngParser.__init__cCst |¡}||ƒS)znReturn a |_PngParser| instance containing the header properties parsed from the PNG image in `stream`.)Ú_Chunksr)rrr$r r r r*ó z_PngParser.parsecCó|jj}|jS)z.The number of pixels in each row of the image.)r#ÚIHDRr©r r+r r r r1óz_PngParser.px_widthcCr*)z2The number of stacked rows of pixels in the image.)r#r+rr,r r r r7r-z_PngParser.px_heightcCó$|jj}|dur 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 r0r r r r=óz_PngParser.horz_dpicCr.)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. Nr/)r#r0r1r2Úvert_px_per_unitr4r r r rHr5z_PngParser.vert_dpicCs |dkr|rtt|dƒƒSdS)zWReturn dots per inch value calculated from `units_specifier` and `px_per_unit`.rg¦ F%uš?r/)ÚintÚround)r2Z px_per_unitr r r r1Ss z_PngParser._dpi)rrrrr"rrrrrrrÚ staticmethodr1Ú __classcell__r r r%r r#s       rcsLeZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ d d „Z ‡Z S) r(z8Collection of the chunks parsed from a PNG image stream.cstt|ƒ ¡t|ƒ|_dSr )r!r(r"Úlistr#)r Zchunk_iterabler%r r r"_sz_Chunks.__init__cCst |¡}t| ¡ƒ}||ƒS)zBReturn a |_Chunks| instance containing the PNG chunks in `stream`.)Ú _ChunkParserrr;Ú iter_chunks)rrZ chunk_parserr$r r r rcs  z_Chunks.from_streamcCs&dd„}| |¡}|durtdƒ‚|S)zIHDR chunk in PNG image.cSó |jtjkSr )Ú type_namerr+©Úchunkr r r Úmó z_Chunks.IHDR..Nzno IHDR chunk in PNG image)Ú _find_firstr)r Úmatchr+r r r r+js  z _Chunks.IHDRcCsdd„}| |¡S)z2PHYs chunk in PNG image, or |None| if not present.cSr>r )r?rr0r@r r r rBvrCz_Chunks.pHYs..)rD)r rEr r r r0ss z _Chunks.pHYscCs |jD] }||ƒr |SqdS)zGReturn first chunk in stream order returning True for function `match`.N)r#)r rErAr r r rDys ÿz_Chunks._find_first) rrrrr"rrrr+r0rDr:r r r%r r(\s    r(cs<eZdZdZ‡fdd„Zedd„ƒZdd„Zdd „Z‡Z S) r<z(Extracts chunks from a PNG image stream.crr )r!r<r"Ú _stream_rdr)r Ú stream_rdrr%r r r"„r'z_ChunkParser.__init__cCst|tƒ}||ƒS)zdReturn a |_ChunkParser| instance that can extract the chunks from the PNG image in `stream`.)rr)rrrGr r r rˆr)z_ChunkParser.from_streamccs,| ¡D]\}}t||j|ƒ}|VqdS)zGenerate a |_Chunk| subclass instance for each chunk in this parser's PNG stream, in the order encountered in the stream.N)Ú_iter_chunk_offsetsÚ _ChunkFactoryrF)r Ú chunk_typeÚoffsetrAr r r r=s €þz_ChunkParser.iter_chunksccsTd} |j |¡}|j d|d¡}|d}||fV|dkr!dS|d|d7}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éZIENDN)rFÚ read_longZread_str)r Z chunk_offsetZchunk_data_lenrJZ data_offsetr r r rH–s€  øz _ChunkParser._iter_chunk_offsets) rrrrr"rrr=rHr:r r r%r r<s  r<cCs*tjttjti}| |t¡}| |||¡S)ziReturn a |_Chunk| subclass instance appropriate to `chunk_type` parsed from `stream_rdr` at `offset`.)rr+Ú _IHDRChunkr0Ú _pHYsChunkÚgetÚ_ChunkÚ from_offset)rJrGrKZ chunk_cls_mapZ chunk_clsr r r rI¨s þ rIcs8eZdZdZ‡fdd„Zedd„ƒZedd„ƒZ‡Z S)rRzUBase class for specific chunk types. Also serves as the default chunk type. crr )r!rRr"Ú _chunk_type)r rJr%r r r"¹r'z_Chunk.__init__cCs||ƒS)z@Return a default _Chunk instance that only knows its chunk type.r )rrJrGrKr r r rS½sz_Chunk.from_offsetcCó|jS)z.The chunk type name, e.g. 'IHDR', 'pHYs', etc.)rTr r r r r?Âsz_Chunk.type_name) rrrrr"rrSrr?r:r r r%r rR³s  rRcsDeZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ ‡Z S) rOz*IHDR chunk, contains the image dimensions.cs tt|ƒ |¡||_||_dSr )r!rOr"Ú _px_widthÚ _px_height)r rJrrr%r r r"Ës z_IHDRChunk.__init__cCs"| |¡}| |d¡}||||ƒS)z|Return an _IHDRChunk instance containing the image dimensions extracted from the IHDR chunk in `stream` at `offset`.rM)rN)rrJrGrKrrr r r rSÐs   z_IHDRChunk.from_offsetcCrUr )rVr r r r rØóz_IHDRChunk.px_widthcCrUr )rWr r r r rÜrXz_IHDRChunk.px_height) rrrrr"rrSrrrr:r r r%r rOÈs   rOcsPeZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ ed d „ƒZ ‡Z S) rPz/PYHs chunk, contains the image dpi information.cs&tt|ƒ |¡||_||_||_dSr )r!rPr"Ú_horz_px_per_unitÚ_vert_px_per_unitÚ_units_specifier)r rJr3r6r2r%r r r"äs z_pHYsChunk.__init__cCs0| |¡}| |d¡}| |d¡}|||||ƒS)z{Return a _pHYsChunk instance containing the image resolution extracted from the pHYs chunk in `stream` at `offset`.rMrL)rNÚ read_byte)rrJrGrKr3r6r2r r r rSês   z_pHYsChunk.from_offsetcCrUr )rYr r r r r3órXz_pHYsChunk.horz_px_per_unitcCrUr )rZr r r r r6÷rXz_pHYsChunk.vert_px_per_unitcCrUr )r[r r r r r2ûrXz_pHYsChunk.units_specifier) rrrrr"rrSrr3r6r2r:r r r%r rPás    rPN)Ú constantsrrÚ exceptionsrÚhelpersrrÚimagerrrr(r<rIrRrOrPr r r r Ús  9%'