o Ð'f ã@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.)rÚPNG©Úself©r úF/home/ubuntu/flask/venv/lib/python3.10/site-packages/docx/image/png.pyÚ content_type szPng.content_typecCsdS)z8Default filename extension, always 'png' for PNG images.Úpngr 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__ÚpropertyrrÚ 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 r2r r r r=óz_PngParser.horz_dpicCr0)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%r2r3r4Úvert_px_per_unitr6r r r rHr7z_PngParser.vert_dpicCs |dkr|rtt|dƒƒSdS)zWReturn dots per inch value calculated from `units_specifier` and `px_per_unit`.rg¦ F%uš?r1)ÚintÚround)r4Ú px_per_unitr r r r3Ss z_PngParser._dpi)rrrrr$r rrrrrrÚ staticmethodr3Ú __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 Úchunk_iterabler'r r r$_sz_Chunks.__init__cCst |¡}t| ¡ƒ}||ƒS)zBReturn a |_Chunks| instance containing the PNG chunks in `stream`.)Ú _ChunkParserrr>Ú iter_chunks)rrÚ 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.cSrCr")rDrr2rEr r r rGvrHz_Chunks.pHYs..)rI)r rJr r r r2ss z _Chunks.pHYscCs |jD] }||ƒr |SqdS)zGReturn first chunk in stream order returning True for function `match`.N)r%)r rJrFr r r rIys ÿz_Chunks._find_first) rrrrr$r rrr-r2rIr=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.cr!r")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)rrrLr 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Ú _ChunkFactoryrK)r Ú chunk_typeÚoffsetrFr r r rAs €þ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éÚIENDN)rKÚ read_longÚread_str)r Ú chunk_offsetÚchunk_data_lenrOÚ data_offsetr r r rM–s€  øz _ChunkParser._iter_chunk_offsets) rrrrr$r rrArMr=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-Ú _IHDRChunkr2Ú _pHYsChunkÚgetÚ_ChunkÚ from_offset)rOrLrPÚ chunk_cls_mapÚ chunk_clsr r r rN¨s þ rNcs8eZdZdZ‡fdd„Zedd„ƒZedd„ƒZ‡Z S)r\zUBase class for specific chunk types. Also serves as the default chunk type. cr!r")r#r\r$Ú _chunk_type)r rOr'r r r$¹r)z_Chunk.__init__cCs||ƒS)z@Return a default _Chunk instance that only knows its chunk type.r )rrOrLrPr r r r]½sz_Chunk.from_offsetcCó|jS)z.The chunk type name, e.g. 'IHDR', 'pHYs', etc.)r`r r r r rDÂsz_Chunk.type_name) rrrrr$r r]rrDr=r r r'r r\³s  r\csDeZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ ‡Z S) rYz*IHDR chunk, contains the image dimensions.cs tt|ƒ |¡||_||_dSr")r#rYr$Ú _px_widthÚ _px_height)r rOrrr'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`.rR)rT)rrOrLrPrrr r r r]Ðs   z_IHDRChunk.from_offsetcCrar")rbr r r r rØóz_IHDRChunk.px_widthcCrar")rcr r r r rÜrdz_IHDRChunk.px_height) rrrrr$r r]rrrr=r r r'r rYÈs   rYcsPeZdZdZ‡fdd„Zedd„ƒZedd„ƒZedd „ƒZ ed d „ƒZ ‡Z S) rZz/PYHs chunk, contains the image dpi information.cs&tt|ƒ |¡||_||_||_dSr")r#rZr$Ú_horz_px_per_unitÚ_vert_px_per_unitÚ_units_specifier)r rOr5r8r4r'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`.rRrQ)rTÚ read_byte)rrOrLrPr5r8r4r r r r]ês   z_pHYsChunk.from_offsetcCrar")rer r r r r5órdz_pHYsChunk.horz_px_per_unitcCrar")rfr r r r r8÷rdz_pHYsChunk.vert_px_per_unitcCrar")rgr r r r r4ûrdz_pHYsChunk.units_specifier) rrrrr$r r]rr5r8r4r=r r r'r rZás    rZN)Ú constantsrrÚ exceptionsrÚhelpersrrÚimagerrrr*r@rNr\rYrZr r r r Ús  9%'