o W_>@sdZddlmZmZmZddlmZddlmZm Z ddl m Z m Z ddl mZdd lmZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZddZGdddeZGdddeZGdddeZGdddeZd S)!z[ Objects related to parsing headers of JPEG image streams, both JFIF and Exif sub-formats. )absolute_importdivisionprint_function)BytesIO)JPEG_MARKER_CODE MIME_TYPE) BIG_ENDIAN StreamReader)BaseImageHeader)Tiffc@s(eZdZdZeddZeddZdS)Jpegz2 Base class for JFIF and EXIF subclasses. cCstjS)zi MIME content type for this image, unconditionally `image/jpeg` for JPEG images. )r ZJPEGselfr?D:\Projects\ConvertPro\env\Lib\site-packages\docx\image\jpeg.py content_typezJpeg.content_typecCsdS)zJ Default filename extension, always 'jpg' for JPG images. Zjpgrrrrr default_extszJpeg.default_extN)__name__ __module__ __qualname____doc__propertyrrrrrrrs  rc@eZdZdZeddZdS)Exifz3 Image header parser for Exif image format cC8t|}|jj}|jj}|jj}|jj}|||||S)zm Return |Exif| instance having header properties parsed from Exif image in *stream*. ) _JfifMarkers from_streamsofpx_width px_heightapp1horz_dpivert_dpiclsstreammarkersr!r"r$r%rrrr)s zExif.from_streamNrrrr classmethodrrrrrr%rc@r)Jfifz3 Image header parser for JFIF image format cCr)zj Return a |Jfif| instance having header properties parsed from image in *stream*. )rrr r!r"app0r$r%r&rrrr>s zJfif.from_streamNr*rrrrr-:r,r-csXeZdZdZfddZddZeddZedd Z ed d Z ed d Z Z S)rzp Sequence of markers in a JPEG file, perhaps truncated at first SOS marker for performance reasons. cstt|t||_dSN)superr__init__list_markers)rr) __class__rrr1Ssz_JfifMarkers.__init__cCsNd}d}g}|jD]}|||j|jt|j|jfq |g|}d|S)z Returns a tabular listing of the markers in this instance, which can be handy for debugging and perhaps other uses. z4 offset seglen mc name ======= ====== == =====z%7d %6d %02X %s )r3appendoffsetsegment_lengthord marker_codenamejoin)rheadertmplZrowsmarkerlinesrrr__str__Ws    z_JfifMarkers.__str__cCsDt|}g}|D]}|||jtjkr||Sq ||S)z Return a |_JfifMarkers| instance containing a |_JfifMarker| subclass instance for each marker in *stream*. ) _MarkerParserr iter_markersr7r;rZSOS)r'r(Z marker_parserr)r@rrrrgs    z_JfifMarkers.from_streamcC(|jD] }|jtjkr|Sqtd)z5 First APP0 marker in image markers. zno APP0 marker in image)r3r;rAPP0KeyErrorrmrrrr.u  z_JfifMarkers.app0cCrE)z5 First APP1 marker in image markers. zno APP1 marker in image)r3r;rAPP1rGrHrrrr#rJz_JfifMarkers.app1cCs(|jD] }|jtjvr|Sqtd)zF First start of frame (SOFn) marker in this sequence. z(no start of frame (SOFn) marker in image)r3r;rSOF_MARKER_CODESrGrHrrrr rJz_JfifMarkers.sof) rrrrr1rBr+rrr.r#r __classcell__rrr4rrNs     rcs4eZdZdZfddZeddZddZZS)rCz_ Service class that knows how to parse a JFIF stream and iterate over its markers. ctt|||_dSr/)r0rCr1_stream)r stream_readerr4rrr1 z_MarkerParser.__init__cCst|t}||S)z` Return a |_MarkerParser| instance to parse JFIF markers from *stream*. )r r )r'r(rPrrrrs z_MarkerParser.from_streamccs^t|j}d}d}|tjkr-||\}}t||j|}|V||j}|tjksdSdS)z Generate a (marker_code, segment_offset) 2-tuple for each marker in the JPEG *stream*, in the order they occur in the stream. rN) _MarkerFinderrrOrZEOInext_MarkerFactoryr9)rZ marker_finderstartr;segment_offsetr@rrrrDs   z_MarkerParser.iter_markers) rrrrr1r+rrDrMrrr4rrCs   rCcsLeZdZdZfddZeddZddZdd Zd d Z d d Z Z S)rRzP Service class that knows how to find the next JFIF marker in a stream. crNr/)r0rRr1rO)rr(r4rrr1rQz_MarkerFinder.__init__cCs||S)zU Return a |_MarkerFinder| instance to find JFIF markers in *stream*. r)r'r(rrrrsz_MarkerFinder.from_streamcCsH|} |j|d}|j|dd\}}|dkrq||d}} ||fS)a` Return a (marker_code, segment_offset) 2-tuple identifying and locating the first marker in *stream* occuring after offset *start*. The returned *segment_offset* points to the position immediately following the 2-byte marker code, the start of the marker segment, for those markers that have a segment. T)rUr)_offset_of_next_ff_byte_next_non_ff_byte)rrUpositionbyte_r;rVrrrrSs z_MarkerFinder.nextcCsB|j||}|dkr|}|dks|jd}||fS)u Return an offset, byte 2-tuple for the next byte in *stream* that is not 'ÿ', starting with the byte at offset *start*. If the byte at offset *start* is not 'ÿ', *start* and the returned *offset* will be the same. rrOseek _read_bytetell)rrUr[Zoffset_of_non_ff_byterrrrYs z_MarkerFinder._next_non_ff_bytecCs>|j||}|dkr|}|dks|jd}|S)u Return the offset of the next 'ÿ' byte in *stream* starting with the byte at offset *start*. Returns *start* if the byte at that offset is a hex 255; it does not necessarily advance in the stream. r\rr])rrUr[Zoffset_of_ff_byterrrrXs z%_MarkerFinder._offset_of_next_ff_bytecCs|jd}|s td|S)zm Return the next byte read from stream. Raise Exception if stream is at end of file. rzunexpected end of file)rOread Exception)rr[rrrr_s z_MarkerFinder._read_byte) rrrrr1r+rrSrYrXr_rMrrr4rrRs   rRcCsB|tjkrt}n|tjkrt}n |tjvrt}nt}||||S)zx Return |_Marker| or subclass instance appropriate for marker at *offset* in *stream* having *marker_code*. ) rrF _App0MarkerrK _App1MarkerrL _SofMarker_Markerr)r;r(r8Z marker_clsrrrrTs   rTcs\eZdZdZfddZeddZeddZedd Z ed d Z ed d Z Z S)rfzu Base class for JFIF marker classes. Represents a marker and its segment occuring in a JPEG byte stream. cs$tt|||_||_||_dSr/)r0rfr1 _marker_code_offset_segment_length)rr;r8r9r4rrr1s z_Marker.__init__cCs&t|rd}n||}||||S)zz Return a generic |_Marker| instance for the marker at *offset* in *stream* having *marker_code*. r)rZ is_standalone read_short)r'r(r;r8r9rrrrs   z_Marker.from_streamcC|jS)u The single-byte code that identifies the type of this marker, e.g. ``'à'`` for start of image (SOI). )rgrrrrr;)rz_Marker.marker_codecCs tj|jSr/)rZ marker_namesrgrrrrr<1s z _Marker.namecCrkr/)rhrrrrr85sz_Marker.offsetcCrk)z> The length in bytes of this marker's segment )rirrrrr99z_Marker.segment_length) rrrrr1r+rrr;r<r8r9rMrrr4rrfs     rfcsLeZdZdZfddZeddZeddZdd Ze d d Z Z S) rcz0 Represents a JFIF APP0 marker segment. cs*tt||||||_||_||_dSr/)r0rcr1_density_units _x_density _y_density)rr;r8length density_units x_density y_densityr4rrr1Es z_App0Marker.__init__cC ||jSzm Horizontal dots per inch specified in this marker, defaults to 72 if not specified. )_dpirnrrrrr$M z_App0Marker.horz_dpicCrtzk Vertical dots per inch specified in this marker, defaults to 72 if not specified. )rvrorrrrr%Urwz_App0Marker.vert_dpicCs8|jdkr |}|S|jdkrtt|d}|Sd}|S)zH Return dots per inch corresponding to *density* value. rrgRQ@H)rmintround)rZdensityZdpirrrrv]s  z_App0Marker._dpicCs@||}||d}||d}||d}|||||||S)zg Return an |_App0Marker| instance for the APP0 marker at *offset* in *stream*. )rjZ read_byte)r'r(r;r8r9rqrrrsrrrris     z_App0Marker.from_stream) rrrrr1rr$r%rvr+rrMrrr4rrcAs    rccs\eZdZdZfddZeddZeddZedd Z ed d Z ed d Z Z S)rdz7 Represents a JFIF APP1 (Exif) marker segment. c$tt||||||_||_dSr/)r0rdr1 _horz_dpi _vert_dpi)rr;r8rpr$r%r4rrr1s z_App1Marker.__init__cCsH||}|||r||||ddS||||}|||||j|jS)z Extract the horizontal and vertical dots-per-inch value from the APP1 header at *offset* in *stream*. ry)rj_is_non_Exif_APP1_segment_tiff_from_exif_segmentr$r%)r'r(r;r8r9tiffrrrrs  z_App1Marker.from_streamcCrkru)rrrrrr$rz_App1Marker.horz_dpicCrkrx)rrrrrr%rz_App1Marker.vert_dpicCs ||d|d}|dkS)z Return True if the APP1 segment at *offset* in *stream* is NOT an Exif segment, as determined by the ``'Exif'`` signature at offset 2 in the segment. rsExif)r^ra)r'r(r8Zexif_signaturerrrrs z%_App1Marker._is_non_Exif_APP1_segmentcCs.||d||d}t|}t|S)z Return a |Tiff| instance parsed from the Exif APP1 segment of *segment_length* at *offset* in *stream*. )r^rarr r)r'r(r8r9Z segment_bytesZ substreamrrrrs z#_App1Marker._tiff_from_exif_segment) rrrrr1r+rrr$r%rrrMrrr4rrds      rdcsDeZdZdZfddZeddZeddZedd Z Z S) rezA Represents a JFIF start of frame (SOFx) marker segment. crr/)r0rer1 _px_width _px_height)rr;r8r9r!r"r4rrr1s z_SofMarker.__init__cCs2||}||d}||d}||||||S)zd Return an |_SofMarker| instance for the SOFn marker at *offset* in stream. )rj)r'r(r;r8r9r"r!rrrrs  z_SofMarker.from_streamcCrk)z( Image height in pixels )rrrrrr"rlz_SofMarker.px_heightcCrk)z' Image width in pixels )rrrrrr!rlz_SofMarker.px_width) rrrrr1r+rrr"r!rMrrr4rres   reN)rZ __future__rrrcompatr constantsrr helpersr r imager rr rrr-objectrrCrRrTrfrcrdrerrrrs$   F#K/BG