U : j0Y @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(4/tmp/pip-unpacked-wheel-8l127pjy/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-rLr>r r!r&Z_bytesr(r(r)__repr__s&zHeifDepthImage.__repr__r.cst}|j|_|S)r:)rHr<rIrr&imagerKr(r)r<s  zHeifDepthImage.to_pillow) r>r?r@rAr*rVr r< __classcell__r(r(rKr)rFts rFcsteZdZdZfddZddZeedddZeedd 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")rFr[r(r(r)r^s)primary bit_depthexifrG thumbnailsZ depth_imagesxmpheiftype)ZrICCZprofr- icc_profileicc_profile_type nclx_profile)rHr*rGrrr Z THUMBNAILSrbZ DEPTH_IMAGESZdepth_image_listrboolr_r7r`rIrZ color_profile) r&r'rJZ_exif_xmpZ _thumbnailsZ _depth_imagesZ _heif_metaZ_color_profilerKr(r)r*s6     zHeifImage.__init__cCsp|jst|jtr"t|jdnd}d|jjd|jdd|jdd|j d|d t|j d gd S) NrMrNrOrPrrQr  with z image data and rbz thumbnails>) r%rSr$rrTr-rLr>r r!rIgetrUr(r(r)rVs&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.rmrnrrprr)rir!rsr,r(r(r)premultiplied_alphaszHeifImage.premultiplied_alphavaluecCs*|jr&|j|rdnd|rdnd|_dS)Nrqrr)rtr!replacer&rwr(r(r)ruscs*t}|j|_t|j|jd<|S)r:original_orientation)rHr<rIrrrWrKr(r)r<s  zHeifImage.to_pillow)r>r?r@rAr*rVrErirtrusetterr r<rYr(r(rKr)rZs !rZc@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)NseekrZavifrpAVIFZheicrdHEIFZ remove_strideTZ hdr_to_16bit reload_sizecSsg|]}|dk rt|qSr")rZr[r(r(r)r^sz%HeifFile.__init__..r_F)hasattrr}rrrfindr ZPREFERRED_DECODERrl _pillow_heifZ load_fileZDECODE_THREADSZALLOW_INCORRECT_HEADERSmimetype_images primary_index enumeraterI) 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. )rrrtr,r(r(r)rtszHeifFile.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. rrrur,r(r(r)ruszHeifFile.premultiplied_alpharvcCs||j|j_dSr"rryr(r(r)ru!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. )rrrIr,r(r(r)rI5sz 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)NrOrkz images: cSsg|] }t|qSr()rDr[r(r(r)r^rsz%HeifFile.__repr__..rR)rLr>rTr,r(r(r)rVqszHeifFile.__repr__cCs t|jSr")rTrr,r(r(r)__len__tszHeifFile.__len__ccs|jEdHdSr")rr,r(r(r)__iter__wszHeifFile.__iter__cCs.|dks|t|jkr$td||j|SNrzinvalid image index: rTr 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. )rZrrappend)r&r!r r-r added_imager(r(r) add_frombytess  zHeifFile.add_frombytes)rXr/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-r0rrIpop)r&rXrr(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.rarcN)r`rbrfrg)rhrG)r ValueErrorrIrrrrrrr+rr!tobytesr)r&rX_inforjrzZ_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 rIrr)r&im_descZimim_datar(r(r) __getstate__s   zHeifFile.__getstate__c CsD||\|_|_}|D]$}|\}}}}||||}||_qdSr")r*rrrrI) 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?r@rAr*rEr r!rtrur{rir-r0rIr r<rrVrrrrrDrBrrZrrr9rrZ_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)rrZr)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)5rArriortypingrrrrr ZPILr r~r constantsr miscrrrrrrrrrrrrrrrrr ImportErrorexZ_deffered_errorrrrFrZr|rirrrrDrBrrrrr(r(r(r)s2    H  NE