o
    	'f(                     @   s   d dl mZmZmZ d dlmZmZmZ d dlm	Z	 G dd de	Z
G dd dZG dd	 d	ZG d
d dZdd ZG dd dZG dd deZG dd deZG dd deZG dd deZdS )   )	MIME_TYPETIFF_FLDTIFF_TAG)
BIG_ENDIANLITTLE_ENDIANStreamReader)BaseImageHeaderc                   @   s4   e Zd ZdZedd Zedd Zedd ZdS )	Tiffz`Image header parser for TIFF images.

    Handles both big and little endian byte ordering.
    c                 C   s   t jS )z[Return the MIME type of this TIFF image, unconditionally the string
        ``image/tiff``.)r   TIFFself r   G/home/ubuntu/flask/venv/lib/python3.10/site-packages/docx/image/tiff.pycontent_type   s   zTiff.content_typec                 C      dS )z:Default filename extension, always 'tiff' for TIFF images.tiffr   r   r   r   r   default_ext   s   zTiff.default_extc                 C   s0   t |}|j}|j}|j}|j}| ||||S )zYReturn a |Tiff| instance containing the properties of the TIFF image in
        `stream`.)_TiffParserparsepx_width	px_heighthorz_dpivert_dpi)clsstreamparserr   r   r   r   r   r   r   from_stream   s   
zTiff.from_streamN)	__name__
__module____qualname____doc__propertyr   r   classmethodr   r   r   r   r   r	      s    

r	   c                       s|   e Zd ZdZ fddZedd Zedd Zedd	 Z	ed
d Z
edd Zedd Zdd Zedd Z  ZS )r   zkParses a TIFF image stream to extract the image properties found in its main
    image file directory (IFD)c                       t t|   || _d S N)superr   __init___ifd_entries)r   ifd_entries	__class__r   r   r&   )      
z_TiffParser.__init__c                 C   s(   |  |}|d}t||}| |S )zmReturn an instance of |_TiffParser| containing the properties parsed from the
        TIFF image in `stream`.   )_make_stream_reader	read_long_IfdEntriesr   )r   r   
stream_rdrifd0_offsetr(   r   r   r   r   -   s   

z_TiffParser.parsec                 C      |  tjS )zThe horizontal dots per inch value calculated from the XResolution and
        ResolutionUnit tags of the IFD; defaults to 72 if those tags are not present.)_dpir   X_RESOLUTIONr   r   r   r   r   6      z_TiffParser.horz_dpic                 C   r2   )zThe vertical dots per inch value calculated from the XResolution and
        ResolutionUnit tags of the IFD; defaults to 72 if those tags are not present.)r3   r   Y_RESOLUTIONr   r   r   r   r   <   r5   z_TiffParser.vert_dpic                 C      | j tjS )zThe number of stacked rows of pixels in the image, |None| if the IFD contains
        no ``ImageLength`` tag, the expected case when the TIFF is embeded in an Exif
        image.)r'   getr   IMAGE_LENGTHr   r   r   r   r   B      z_TiffParser.px_heightc                 C   r7   )zThe number of pixels in each row in the image, |None| if the IFD contains no
        ``ImageWidth`` tag, the expected case when the TIFF is embeded in an Exif
        image.)r'   r8   r   IMAGE_WIDTHr   r   r   r   r   I   r:   z_TiffParser.px_widthc                 C   s$   | d |d}|dkrtS tS )zReturn either BIG_ENDIAN or LITTLE_ENDIAN depending on the endian indicator
        found in the TIFF `stream` header, either 'MM' or 'II'.       s   MM)seekreadr   r   )r   r   
endian_strr   r   r   _detect_endianP   s   

z_TiffParser._detect_endianc                 C   s^   | j }||vr	dS tj|v r|tj nd}|dkrdS |dkr!dnd}|| }tt|| S )a!  Return the dpi value calculated for `resolution_tag`, which can be either
        TIFF_TAG.X_RESOLUTION or TIFF_TAG.Y_RESOLUTION.

        The calculation is based on the values of both that tag and the
        TIFF_TAG.RESOLUTION_UNIT tag in this parser's |_IfdEntries| instance.
        H   r=   r   gRQ@)r'   r   RESOLUTION_UNITintround)r   resolution_tagr(   resolution_unitunits_per_inchdots_per_unitr   r   r   r3   X   s   

z_TiffParser._dpic                 C   s   |  |}t||S )zReturn a |StreamReader| instance with wrapping `stream` and having "endian-
        ness" determined by the 'MM' or 'II' indicator in the TIFF stream header.)rA   r   )r   r   endianr   r   r   r-   r   s   

z_TiffParser._make_stream_reader)r   r   r   r    r&   r"   r   r!   r   r   r   r   rA   r3   r-   __classcell__r   r   r)   r   r   %   s$    





r   c                       sF   e Zd ZdZ fddZdd Zdd Zedd	 ZdddZ	  Z
S )r/   z}Image File Directory for a TIFF image, having mapping (dict) semantics allowing
    "tag" values to be retrieved by tag code.c                    r#   r$   )r%   r/   r&   _entries)r   entriesr)   r   r   r&   ~   r+   z_IfdEntries.__init__c                 C      | j |S )z5Provides ``in`` operator, e.g. ``tag in ifd_entries``)rL   __contains__r   keyr   r   r   rO         z_IfdEntries.__contains__c                 C   rN   )zCProvides indexed access, e.g. ``tag_value = ifd_entries[tag_code]``)rL   __getitem__rP   r   r   r   rS      rR   z_IfdEntries.__getitem__c                 C   s$   t ||}dd | D }| |S )zVReturn a new |_IfdEntries| instance parsed from `stream` starting at
        `offset`.c                 S   s   i | ]}|j |jqS r   )tagvalue).0er   r   r   
<dictcomp>   s    z+_IfdEntries.from_stream.<locals>.<dictcomp>)
_IfdParseriter_entries)r   r   offset
ifd_parserrM   r   r   r   r      s   
z_IfdEntries.from_streamNc                 C   s   | j ||S )zhReturn value of IFD entry having tag matching `tag_code`, or `default` if no
        matching tag found.)rL   r8   )r   tag_codedefaultr   r   r   r8         z_IfdEntries.getr$   )r   r   r   r    r&   rO   rS   r"   r   r8   rK   r   r   r)   r   r/   z   s    
r/   c                       s4   e Zd ZdZ fddZdd Zedd Z  ZS )rY   zaService object that knows how to extract directory entries from an Image File
    Directory (IFD)c                       t t|   || _|| _d S r$   )r%   rY   r&   _stream_rdr_offset)r   r0   r[   r)   r   r   r&         
z_IfdParser.__init__c                 c   s:    t | jD ]}| jd |d  }t| j|}|V  qdS )zVGenerate an |_IfdEntry| instance corresponding to each entry in the
        directory.r=      N)range_entry_countrb   _IfdEntryFactoryra   )r   idxdir_entry_offset	ifd_entryr   r   r   rZ      s   z_IfdParser.iter_entriesc                 C   s   | j | jS )zDThe count of directory entries, read from the top of the IFD header.)ra   
read_shortrb   r   r   r   r   rf      r_   z_IfdParser._entry_count)	r   r   r   r    r&   rZ   r!   rf   rK   r   r   r)   r   rY      s    rY   c                 C   s@   t jtt jtt jtt jti}| 	|d}|
|t}|| |S )ztReturn an |_IfdEntry| subclass instance containing the value of the directory
    entry at `offset` in `stream_rdr`.r=   )r   ASCII_AsciiIfdEntrySHORT_ShortIfdEntryLONG_LongIfdEntryRATIONAL_RationalIfdEntryrk   r8   	_IfdEntryr   )r0   r[   ifd_entry_classes
field_typeEntryClsr   r   r   rg      s   rg   c                       sP   e Zd ZdZ fddZedd Zedd Zedd	 Z	ed
d Z
  ZS )rt   zsBase class for IFD entry classes.

    Subclasses are differentiated by value type, e.g. ASCII, long int, etc.
    c                    r`   r$   )r%   rt   r&   	_tag_code_value)r   r]   rU   r)   r   r   r&      rc   z_IfdEntry.__init__c                 C   s>   | |d}||d}||d}| ||||}| ||S )a!  Return an |_IfdEntry| subclass instance containing the tag and value of the
        tag parsed from `stream_rdr` at `offset`.

        Note this method is common to all subclasses. Override the ``_parse_value()``
        method to provide distinctive behavior based on field type.
        r<   r,      )rk   r.   _parse_value)r   r0   r[   r]   value_countvalue_offsetrU   r   r   r   r      s
   
z_IfdEntry.from_streamc                 C   r   )zReturn the value of this field parsed from `stream_rdr` at `offset`.

        Intended to be overridden by subclasses.
        zUNIMPLEMENTED FIELD TYPEr   r   r0   r[   r|   r}   r   r   r   r{      s   z_IfdEntry._parse_valuec                 C      | j S )z.Short int code that identifies this IFD entry.)rx   r   r   r   r   rT         z_IfdEntry.tagc                 C   r   )z7Value of this tag, its type being dependent on the tag.)ry   r   r   r   r   rU      r   z_IfdEntry.value)r   r   r   r    r&   r"   r   r{   r!   rT   rU   rK   r   r   r)   r   rt      s    


rt   c                   @      e Zd ZdZedd ZdS )rm   z<IFD entry having the form of a NULL-terminated ASCII string.c                 C   s   | |d |S )zReturn the ASCII string parsed from `stream_rdr` at `value_offset`.

        The length of the string, including a terminating ' ' (NUL) character, is in
        `value_count`.
        r   )read_strr~   r   r   r   r{      s   z_AsciiIfdEntry._parse_valueNr   r   r   r    r"   r{   r   r   r   r   rm          rm   c                   @   r   )ro   z0IFD entry expressed as a short (2-byte) integer.c                 C      |dkr
| |dS dS )zReturn the short int value contained in the `value_offset` field of this
        entry.

        Only supports single values at present.
        r   rz   z)Multi-value short integer NOT IMPLEMENTED)rk   r~   r   r   r   r{         z_ShortIfdEntry._parse_valueNr   r   r   r   r   ro      r   ro   c                   @   r   )rq   z/IFD entry expressed as a long (4-byte) integer.c                 C   r   )zReturn the long int value contained in the `value_offset` field of this
        entry.

        Only supports single values at present.
        r   rz   z(Multi-value long integer NOT IMPLEMENTEDr.   r~   r   r   r   r{   	  r   z_LongIfdEntry._parse_valueNr   r   r   r   r   rq     r   rq   c                   @   r   )rs   z5IFD entry expressed as a numerator, denominator pair.c                 C   s*   |dkr| |}| |d}|| S dS )zReturn the rational (numerator / denominator) value at `value_offset` in
        `stream_rdr` as a floating-point number.

        Only supports single values at present.
        r   r,   z$Multi-value Rational NOT IMPLEMENTEDr   )r   r0   r[   r|   r}   	numeratordenominatorr   r   r   r{     s
   
z_RationalIfdEntry._parse_valueNr   r   r   r   r   rs     r   rs   N)	constantsr   r   r   helpersr   r   r   imager   r	   r   r/   rY   rg   rt   rm   ro   rq   rs   r   r   r   r   <module>   s    U,