U wh0Y @sdZddlmZmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!z dd l"Z"Wn6e#k rZ$zdd l%m&Z&e&e$Z"W5d Z$[$XYnXGd d d Z'Gddde'Z(Gddde'Z)GdddZ*e+dddZ,d(e*dddZ-d)e*dddZ.e/e0d dddZ1ee)d d d!d"Z2e j e*d#d$d%Z3e/e0e*dd&d'Z4d S)*z8Functions and classes for heif images to read and write.)copydeepcopy)SEEK_SET)AnyDictListOptionalTuple)Image)options)HeifCompressionFormat) MODE_INFO CtxEncode MimCImage_exif_from_pillow _get_bytes_get_heif_meta_get_orientation_for_encoder_get_primary_index_pil_to_supported_mode_retrieve_exif _retrieve_xmp _rotate_pil_xmp_from_pillowget_file_mimetypesave_colorspace_chromaset_orientationN) DeferredErrorc@sreZdZUdZeed<eed<ddZeddZ ee dd d Z ed d Z e j dd dZddddZdS) BaseImagezDBase class for :py:class:`HeifImage` and :py:class:`HeifDepthImage`.sizemodecCs|j\|_|_||_d|_dSN) size_moder r!_c_image_data)selfc_imager(X/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/pillow_heif/heif.py__init__2szBaseImage.__init__cCs||jS)zaDecodes image and returns image data. :returns: ``bytes`` of the decoded image. )loadr%r&r(r(r)data7szBaseImage.datareturncCs||jjS)zStride of the image. .. note:: from `0.10.0` version this value always will have width * sizeof pixel in default usage mode. :returns: An Int value indicating the image stride after decoding. )r+r$strider,r(r(r)r0@szBaseImage.stridecCs|t|jt|jd}t|jddkr8d}nt|d}d}|jd|f}t|jddkr||t|jdf7}||d|jdS) zNumpy array interface support.rr z|u1z) r% isinstancer$rlenr-rMr?r r!r&Z_bytesr(r(r)__repr__s&zHeifDepthImage.__repr__r.cst}|j|_|S)r:)rIr=rJrr&imagerLr(r)r=s  zHeifDepthImage.to_pillow) r?r@rArBr*rWr r= __classcell__r(r(rLr)rGts rGcsteZdZdZfddZddZeedddZeedd d Z e j ed d d Z e j dfd d Z Z S) HeifImagez;One image in a :py:class:`~pillow_heif.HeifFile` container.c st||j}t|}t|}tjr8dd|jDng}tjrRdd|j Dng}t |}t |j t |j||||d|_|r||jd<|r||jd<t||j|j}|r|ddkr|d |jd <|d|jd <n|d |jd <dS) NcSsg|]}|dk r|qSr"r(.0ir(r(r) sz&HeifImage.__init__..cSsg|]}|dk rt|qSr")rGr\r(r(r)r_s)primary bit_depthexifrH thumbnailsZ depth_imagesxmpheiftype)ZrICCZprofr- icc_profileicc_profile_type nclx_profile)rIr*rHrrr Z THUMBNAILSrcZ DEPTH_IMAGESZdepth_image_listrboolr`r7rarJrZ color_profile) r&r'rKZ_exif_xmpZ _thumbnailsZ _depth_imagesZ _heif_metaZ_color_profilerLr(r)r*s6     zHeifImage.__init__cCsp|jst|jtr"t|jdnd}d|jjd|jdd|jdd|j d|d t|j d gd S) NrNrOrPrQrrRr  with z image data and rcz thumbnails>) r%rTr$rrUr-rMr?r r!rJgetrVr(r(r)rWs&HzHeifImage.__repr__r.cCs|jjdddddkS)zD``True`` for images with the ``alpha`` channel, ``False`` otherwise.;sepr)Aa)r!splitr,r(r(r) has_alphaszHeifImage.has_alphacCst|jjdddddkS)zN``True`` for images with ``premultiplied alpha`` channel, ``False`` otherwise.rnrorrqrs)rjr!rtr,r(r(r)premultiplied_alphaszHeifImage.premultiplied_alphavaluecCs*|jr&|j|rdnd|rdnd|_dS)Nrrrs)rur!replacer&rxr(r(r)rvscs*t}|j|_t|j|jd<|S)r:original_orientation)rIr=rJrrrXrLr(r)r=s  zHeifImage.to_pillow)r?r@rArBr*rWrFrjrurvsetterr r=rZr(r(rLr)r[s !r[c@s"eZdZdZd6ddZeddZed d Zed d Zed dZ e j e dddZ eddZ eddZ eddZejdddZddddZddZddZd d!Zd"d#Zd$d%Zeed&d'd(Zeed)d*d+Zejed)d,d-Zed.d/Zd0d1Zd2d3Zd4d5Z e Z!dS)7HeifFileaRepresentation of the :py:class:`~pillow_heif.HeifImage` classes container. To create :py:class:`~pillow_heif.HeifFile` object, use the appropriate factory functions. * :py:func:`~pillow_heif.open_heif` * :py:func:`~pillow_heif.read_heif` * :py:func:`~pillow_heif.from_pillow` * :py:func:`~pillow_heif.from_bytes` Exceptions that can be raised when working with methods: `ValueError`, `EOFError`, `SyntaxError`, `RuntimeError`, `OSError` NTFc Kst|dr|dt|dkr(g}d}nt|}t|}|ddkrVtjdd}n0|ddksr|ddkrtjd d}nd}t |tj |||d d |d d |d tj |}||_ dd|D|_d|_t|jD]\} } | jddr| |_qdS)NseekrZavifrqAVIFZheicreHEIFZ remove_strideTZ hdr_to_16bit reload_sizecSsg|]}|dk rt|qSr")r[r\r(r(r)r_sz%HeifFile.__init__..r`F)hasattrr~rrrfindr ZPREFERRED_DECODERrm _pillow_heifZ load_fileZDECODE_THREADSZALLOW_INCORRECT_HEADERSmimetype_images primary_index enumeraterJ) r&fpconvert_hdr_to_8bitbgr_modekwargsimagesrZfp_bytesZpreferred_decoderindexr>r(r(r)r*s8      zHeifFile.__init__cCs|j|jjS)z:attr:`~pillow_heif.HeifImage.size` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrr r,r(r(r)r sz HeifFile.sizecCs|j|jjS)z:attr:`~pillow_heif.HeifImage.mode` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrr!r,r(r(r)r! sz HeifFile.modecCs|j|jjS)z:attr:`~pillow_heif.HeifImage.has_alpha` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrrur,r(r(r)ruszHeifFile.has_alphacCs|j|jjS)z:attr:`~pillow_heif.HeifImage.premultiplied_alpha` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. rrrvr,r(r(r)rvszHeifFile.premultiplied_alpharwcCs||j|j_dSr"rrzr(r(r)rv!scCs|j|jjS)z:attr:`~pillow_heif.HeifImage.data` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrr-r,r(r(r)r-%sz HeifFile.datacCs|j|jjS)z:attr:`~pillow_heif.HeifImage.stride` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrr0r,r(r(r)r0-szHeifFile.stridecCs|j|jjS)z`info`` dict of the primary :class:`~pillow_heif.HeifImage` in the container. :exception IndexError: If there are no images. )rrrJr,r(r(r)rJ5sz HeifFile.infor.cCs|j|jS)zHelper method to create Pillow :external:py:class:`~PIL.Image.Image`. :returns: :external:py:class:`~PIL.Image.Image` class created from the primary image. )rrr=r,r(r(r)r==szHeifFile.to_pillowcKst|j|f|dS)aSaves image(s) under the given fp. Keyword options can be used to provide additional instructions to the writer. If a writer does not recognize an option, it is silently ignored. Supported options: ``save_all`` - boolean. Should all images from ``HeiFile`` be saved? (default = ``True``) ``append_images`` - do the same as in Pillow. Accepts the list of ``HeifImage`` .. note:: Appended images always will have ``info["primary"]=False`` ``quality`` - see :py:attr:`~pillow_heif.options.QUALITY` ``enc_params`` - dictionary with key:value to pass to :ref:`x265 ` encoder. ``exif`` - override primary image's EXIF with specified. Accepts ``None``, ``bytes`` or ``PIL.Image.Exif`` class. ``xmp`` - override primary image's XMP with specified. Accepts ``None`` or ``bytes``. ``primary_index`` - ignore ``info["primary"]`` and set `PrimaryImage` by index. ``chroma`` - custom subsampling value. Possible values: ``444``, ``422`` or ``420`` (``x265`` default). ``subsampling`` - synonym for *chroma*. Format is string, compatible with Pillow: ``x:x:x``, e.g. '4:4:4'. ``format`` - string with encoder format name. Possible values: ``HEIF`` (default) or ``AVIF``. ``save_nclx_profile`` - boolean, see :py:attr:`~pillow_heif.options.SAVE_NCLX_PROFILE` ``matrix_coefficients`` - int, nclx profile: color conversion matrix coefficients, default=6 (see h.273) ``color_primaries`` - int, nclx profile: color primaries (see h.273) ``transfer_characteristic`` - int, nclx profile: transfer characteristics (see h.273) ``full_range_flag`` - nclx profile: full range flag, default: 1 :param fp: A filename (string), pathlib.Path object or an object with `write` method. N)_encode_imagesr)r&rrr(r(r)saveDs+z HeifFile.savecCs*d|jjdt|ddd|DdS)NrPrlz images: cSsg|] }t|qSr()rEr\r(r(r)r_rsz%HeifFile.__repr__..rS)rMr?rUr,r(r(r)rWqszHeifFile.__repr__cCs t|jSr")rUrr,r(r(r)__len__tszHeifFile.__len__ccs|jEdHdSr")rr,r(r(r)__iter__wszHeifFile.__iter__cCs.|dks|t|jkr$td||j|SNrzinvalid image index: rUr IndexError)r&rr(r(r) __getitem__zszHeifFile.__getitem__cCs0|dks|t|jkr$td||j|=dSrr)r&keyr(r(r) __delitem__szHeifFile.__delitem__)r!r cKs$tt|||f|}|j||S)aPAdds image from bytes to container. .. note:: Supports ``stride`` value if needed. :param mode: see :ref:`image-modes`. :param size: tuple with ``width`` and ``height`` of image. :param data: bytes object with raw image data. :returns: :py:class:`~pillow_heif.HeifImage` added object. )r[rrappend)r&r!r r-r added_imager(r(r) add_frombytess  zHeifFile.add_frombytes)rYr/cCs@||j|j|j|j|jd}t|j|_|jdd|S)zAdd image to the container. :param image: :py:class:`~pillow_heif.HeifImage` class to add from. :returns: :py:class:`~pillow_heif.HeifImage` added object. )r0r`N) r+rr!r r-r0rrJpop)r&rYrr(r(r) add_from_heifs zHeifFile.add_from_heifcCs |jddks|jddkr$td|j}t||d<t|}|rN||d<t|}t|}|dk rx|dkrxt||}| | |j |j| }dD]}||jkr|j||j|<qdD]"}||jkrt |j||j|<qt||jd<t|}|r||jd<|S) zAdd image to the container. :param image: Pillow :external:py:class:`~PIL.Image.Image` class to add from. :returns: :py:class:`~pillow_heif.HeifImage` added object. rr zEmpty images are not supported.rbrdN)rarcrgrh)rirH)r ValueErrorrJrrrrrrr+rr!tobytesr)r&rY_inforkr{Z_imgrrr(r(r)add_from_pillows8      zHeifFile.add_from_pillowcCs|j|jjS)z+Returns the primary image as a numpy array.)rrr9r,r(r(r)r9szHeifFile.__array_interface__cCs@g}|jD]&}t|j}||j|j||jgq |j|j|gSr") rbytesr-rr!r rJrr)r&im_descZimim_datar(r(r) __getstate__s   zHeifFile.__getstate__c CsD||\|_|_}|D]$}|\}}}}||||}||_qdSr")r*rrrrJ) r&staterrZim_modeZim_sizerZim_inforr(r(r) __setstate__s  zHeifFile.__setstate__cCs&t}t|j|_|j|_|j|_|Sr")r}rrrr)r&Z_im_copyr(r(r)Z__copys  zHeifFile.__copy)NTF)"r?r@rArBr*rFr r!rurvr|rjr-r0rJr r=rrWrrrrrErCrr[rrr9rrZ_HeifFile__copy__copy__r(r(r(r)r}sD !       -$ r}r.cCs*t|d}|dddkrdSt|dkS)aVChecks if the given `fp` object contains a supported file type. :param fp: A filename (string), pathlib.Path object or a file object. The file object must implement ``file.read``, ``file.seek``, and ``file.tell`` methods, and be opened in binary mode. :returns: A boolean indicating if the object can be opened. r1sftypFr)rr)rZ__datar(r(r) is_supporteds rTFcKst|||f|S)aOpens the given HEIF(AVIF) image file. :param fp: See parameter ``fp`` in :func:`is_supported` :param convert_hdr_to_8bit: Boolean indicating should 10 bit or 12 bit images be converted to 8-bit images during decoding. Otherwise, they will open in 16-bit mode. ``Does not affect "monochrome" or "depth images".`` :param bgr_mode: Boolean indicating should be `RGB(A)` images be opened in `BGR(A)` mode. :param kwargs: **hdr_to_16bit** a boolean value indicating that 10/12-bit image data should be converted to 16-bit mode during decoding. `Has lower priority than convert_hdr_to_8bit`! Default = **True** :returns: :py:class:`~pillow_heif.HeifFile` object. :exception ValueError: invalid input data. :exception EOFError: corrupted image data. :exception SyntaxError: unsupported feature. :exception RuntimeError: some other error. :exception OSError: out of memory. )r})rrrrr(r(r) open_heifsrcKs.t|||fddi|}|D] }|q|S)a\Opens the given HEIF(AVIF) image file and decodes all images. .. note:: In most cases it is better to call :py:meth:`~pillow_heif.open_heif`, and let images decoded automatically only when needed. :param fp: See parameter ``fp`` in :func:`is_supported` :param convert_hdr_to_8bit: Boolean indicating should 10 bit or 12 bit images be converted to 8-bit images during decoding. Otherwise, they will open in 16-bit mode. ``Does not affect "monochrome" or "depth images".`` :param bgr_mode: Boolean indicating should be `RGB(A)` images be opened in `BGR(A)` mode. :param kwargs: **hdr_to_16bit** a boolean value indicating that 10/12-bit image data should be converted to 16-bit mode during decoding. `Has lower priority than convert_hdr_to_8bit`! Default = **True** :returns: :py:class:`~pillow_heif.HeifFile` object. :exception ValueError: invalid input data. :exception EOFError: corrupted image data. :exception SyntaxError: unsupported feature. :exception RuntimeError: some other error. :exception OSError: out of memory. rT)r}r+)rrrrretimgr(r(r) read_heif s r)r!r r/cKs$ttt|||f|g|f|dS)aREncodes data in a ``fp``. :param mode: `BGR(A);16`, `RGB(A);16`, LA;16`, `L;16`, `I;16L`, `BGR(A)`, `RGB(A)`, `LA`, `L` :param size: tuple with ``width`` and ``height`` of an image. :param data: bytes object with raw image data. :param fp: A filename (string), pathlib.Path object or an object with ``write`` method. N)rr[r)r!r r-rrr(r(r)encode(sr)rr/c Ks|dd}|dkrtjntj}t|s         r) pil_imager/cCst}|||S)zCreates :py:class:`~pillow_heif.HeifFile` from a Pillow Image. :param pil_image: Pillow :external:py:class:`~PIL.Image.Image` class. :returns: New :py:class:`~pillow_heif.HeifFile` object. )r}r)rr>r(r(r) from_pillowRs rcKst}|j|||f||S)aJCreates :py:class:`~pillow_heif.HeifFile` from bytes. .. note:: Supports ``stride`` value if needed. :param mode: see :ref:`image-modes`. :param size: tuple with ``width`` and ``height`` of an image. :param data: bytes object with raw image data. :returns: New :py:class:`~pillow_heif.HeifFile` object. )r}r)r!r r-rr>r(r(r) from_bytes^s r)TF)TF)5rBrriortypingrrrrr ZPILr rr constantsr miscrrrrrrrrrrrrrrrrr ImportErrorexZ_deffered_errorrrrGr[r}rjrrrrErCrrrrr(r(r(r)s2    H  NE