a ]j @sddlmZmZddlmZddlmZmZddlm Z Gddde Z GdddZ Gd d d Z Gd d d Z d dZGdddZGdddeZGdddeZdS)) MIME_TYPEPNG_CHUNK_TYPE)InvalidImageStreamError) BIG_ENDIAN StreamReader)BaseImageHeaderc@s4eZdZdZeddZeddZeddZdS) Pngz#Image header parser for PNG images.cCstjS)zUMIME content type for this image, unconditionally `image/png` for PNG images.)rZPNGselfr J/var/www/html/assistant/venv/lib/python3.9/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`.) _PngParserparsepx_width px_heighthorz_dpivert_dpi)clsstreamparserrrrrr r r from_streams  zPng.from_streamN) __name__ __module__ __qualname____doc__propertyr r classmethodrr r r r rs  rcsheZdZdZfddZeddZeddZedd Z ed d Z ed d Z e ddZ ZS)rzNParses a PNG image stream to extract the image properties found in its chunks.cstt|||_dSN)superr__init___chunks)r chunks __class__r r r!&sz_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*s z_PngParser.parsecCs|jj}|jS)z.The number of pixels in each row of the image.)r"IHDRrr r'r r r r1sz_PngParser.px_widthcCs|jj}|jS)z2The number of stacked rows of pixels in the image.)r"r'rr(r r r r7sz_PngParser.px_heightcCs$|jj}|durdS||j|jS)zInteger dots per inch for the width of this image. Defaults to 72 when not present in the file, as is often the case. NH)r"pHYs_dpiunits_specifierhorz_px_per_unitr r*r r r r=sz_PngParser.horz_dpicCs$|jj}|durdS||j|jS)zInteger 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"r*r+r,vert_px_per_unitr.r r r rHsz_PngParser.vert_dpicCs |dkr|rtt|dSdS)zWReturn dots per inch value calculated from `units_specifier` and `px_per_unit`.rg F%u?r))intround)r,Z px_per_unitr r r r+Ss z_PngParser._dpi)rrrrr!rrrrrrr staticmethodr+ __classcell__r r r$r r#s      rcsLeZdZdZfddZeddZeddZedd 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`.) _ChunkParserrr4 iter_chunks)rrZ chunk_parserr#r r r rcs  z_Chunks.from_streamcCs&dd}||}|dur"td|S)zIHDR chunk in PNG image.cSs |jtjkSr) type_namerr'chunkr r r mz_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.cSs |jtjkSr)r7rr*r8r r r r:vr;z_Chunks.pHYs..)r<)r r=r r r r*ssz _Chunks.pHYscCs |jD]}||r|SqdS)zGReturn first chunk in stream order returning True for function `match`.N)r")r r=r9r r r r<ys  z_Chunks._find_first) rrrrr!rrrr'r*r<r3r r r$r r&\s    r&cs<eZdZdZfddZeddZddZdd ZZ S) r5z(Extracts chunks from a PNG image stream.cstt|||_dSr)r r5r! _stream_rdr)r stream_rdrr$r r r!sz_ChunkParser.__init__cCst|t}||S)zdReturn a |_ChunkParser| instance that can extract the chunks from the PNG image in `stream`.)rr)rrr?r r r rs 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 _ChunkFactoryr>)r chunk_typeoffsetr9r r r r6sz_ChunkParser.iter_chunksccsRd}|j|}|jd|d}|d}||fV|dkr read_longZread_str)r Z chunk_offsetZchunk_data_lenrBZ data_offsetr r r r@s  z _ChunkParser._iter_chunk_offsets) rrrrr!rrr6r@r3r r r$r r5s   r5cCs*tjttjti}||t}||||S)ziReturn a |_Chunk| subclass instance appropriate to `chunk_type` parsed from `stream_rdr` at `offset`.)rr' _IHDRChunkr* _pHYsChunkget_Chunk from_offset)rBr?rCZ chunk_cls_mapZ chunk_clsr r r rAs  rAcs8eZdZdZfddZeddZeddZZ S)rJzUBase class for specific chunk types. Also serves as the default chunk type. cstt|||_dSr)r rJr! _chunk_type)r rBr$r r r!sz_Chunk.__init__cCs||S)z@Return a default _Chunk instance that only knows its chunk type.r )rrBr?rCr r r rKsz_Chunk.from_offsetcCs|jS)z.The chunk type name, e.g. 'IHDR', 'pHYs', etc.)rLr r r r r7sz_Chunk.type_name) rrrrr!rrKrr7r3r r r$r rJs   rJcsDeZdZdZfddZeddZeddZedd Z Z S) rGz*IHDR chunk, contains the image dimensions.cs tt||||_||_dSr)r rGr! _px_width _px_height)r rBrrr$r r r!sz_IHDRChunk.__init__cCs"||}||d}||||S)z|Return an _IHDRChunk instance containing the image dimensions extracted from the IHDR chunk in `stream` at `offset`.rE)rF)rrBr?rCrrr r r rKs  z_IHDRChunk.from_offsetcCs|jSr)rMr r r r rsz_IHDRChunk.px_widthcCs|jSr)rNr r r r rsz_IHDRChunk.px_height) rrrrr!rrKrrrr3r r r$r rGs   rGcsPeZdZdZfddZeddZeddZedd Z ed d Z Z S) rHz/PYHs chunk, contains the image dpi information.cs&tt||||_||_||_dSr)r rHr!_horz_px_per_unit_vert_px_per_unit_units_specifier)r rBr-r/r,r$r r r!sz_pHYsChunk.__init__cCs0||}||d}||d}|||||S)z{Return a _pHYsChunk instance containing the image resolution extracted from the pHYs chunk in `stream` at `offset`.rErD)rF read_byte)rrBr?rCr-r/r,r r r rKs   z_pHYsChunk.from_offsetcCs|jSr)rOr r r r r-sz_pHYsChunk.horz_px_per_unitcCs|jSr)rPr r r r r/sz_pHYsChunk.vert_px_per_unitcCs|jSr)rQr r r r r,sz_pHYsChunk.units_specifier) rrrrr!rrKrr-r/r,r3r r r$r rHs    rHN) constantsrr exceptionsrZhelpersrrimagerrrr&r5rArJrGrHr r r r s  9%'