a C g] @sdZddlmZddlmZmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZz dd l Z Wn8e!yZ"z dd l#m$Z$e$e"Z WYd Z"["n d Z"["00Gd ddZ%Gddde%Z&Gddde%Z'Gddde%Z(GdddZ)ddddZ*d1ddddZ+d2dddd Z,d!d"d#d$d%d&Z-d'd#d(d)d*Z.d+dd,d-d.Z/d!d"dd$d/d0Z0d S)3z8Functions and classes for heif images to read and write.) annotations)copydeepcopy)SEEK_SET)Any)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@speZdZUdZded<ded<ddZedd Zed d d d ZeddZ dd ddZ dd ddZ dS) BaseImagez^Base class for :py:class:`HeifImage`, :py:class:`HeifDepthImage` and :py:class:`HeifAuxImage`.tuple[int, int]sizestrmodecCs|j\|_|_||_d|_dSN) size_moderr _c_image_data)selfc_imager'O/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/pillow_heif/heif.py__init__4szBaseImage.__init__cCs||jS)zaDecodes image and returns image data. :returns: ``bytes`` of the decoded image. )loadr$r%r'r'r(data9szBaseImage.dataintreturncCs||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(r0BszBaseImage.stridecCs|t|jt|jd}t|jddkr8d}nt|d}d}|jd|f}t|jddkr||t|jdf7}||d|jdS) zNumpy array interface support.rrz|u1zrKr@rr r+r'r'r(__repr__szHeifDepthImage.__repr__r9r.cst}|j|_|S)r:)rHr=rIrr%imagerJr'r(r=s  zHeifDepthImage.to_pillow)r@rArBrCr)rRr= __classcell__r'r'rJr(rFvs rFc@seZdZdZddZdS) HeifAuxImagezdClass representing the auxiliary image associated with the :py:class:`~pillow_heif.HeifImage` class.c Cs0d|jjd|jdd|jdd|jd SrLrQr+r'r'r(rRszHeifAuxImage.__repr__N)r@rArBrCrRr'r'r'r(rVsrVcseZdZdZfddZddZedddd Zeddd d Zej dd d d Zddfdd Z dddddZ Z S) HeifImagez;One image in a :py:class:`~pillow_heif.HeifFile` container.c s8t||j}t|}t|}tjr8dd|jDng}tjrRdd|j Dng}t |j t |j ||||d|_tjri}|jD],}||} | |vrg|| <|| |q||jd<t|} |r||jd<| r| |jd<t||j|j} | r4| dd vr&| d |jd <| d|jd <n| d |jd <dS)NcSsg|]}|dur|qSr!r'.0ir'r'r( z&HeifImage.__init__..cSsg|]}|durt|qSr!)rFrXr'r'r(r[r\)primary bit_depthexifrG thumbnails depth_imagesZauxxmpheiftype)ZrICCZprofr, icc_profileicc_profile_type nclx_profile)rHr)rGrrr Z THUMBNAILSr`Z DEPTH_IMAGESZdepth_image_listboolr]r-r^rIZ AUX_IMAGESZ aux_image_idsZ get_aux_typeappendrr color_profile) r%r&rGr_rbr`raZ ctx_aux_infoaux_idZaux_typeZ heif_metarjrJr'r(r)sD       zHeifImage.__init__cCsp|jst|jtr"t|jdnd}d|jjd|jdd|jdd|j d|d t|j d gd S) Nz bytesnorMrNrrOr with z image data and r`z thumbnails>) r$ isinstancer#r lenr,rKr@rr rIget)r%Zs_bytesr'r'r(rRs&,zHeifImage.__repr__rhr.cCs|jjdddddvS)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.rqrrrrtrv)rhr rwr+r'r'r(premultiplied_alphaszHeifImage.premultiplied_alphavaluecCs*|jr&|j|rdnd|rdnd|_dS)Nrurv)rxr replacer%r{r'r'r(rysr9cs*t}|j|_t|j|jd<|S)r:original_orientation)rHr=rIrrrSrJr'r(r=s  zHeifImage.to_pillowr-rV)rkr/cCs|j|}t|S)zMethod to retrieve the auxiliary image at the given ID. :returns: a :py:class:`~pillow_heif.HeifAuxImage` class instance. )r# get_aux_imagerV)r%rkZ aux_imager'r'r(rs zHeifImage.get_aux_image) r@rArBrCr)rRrErxrysetterr=rrUr'r'rJr(rWs ' rWc@s&eZdZdZd>ddZeddZed d Zed d Zed dZ e j ddddZ eddZ eddZ eddZ ddddZddddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*d+d,d-Zd.d.d/d0d1Zdd.d/d2d3Zed4d5Zd6d7Zd8d9Zd:d;Zd.r]F)hasattrrrrrfindr ZPREFERRED_DECODERrp _pillow_heifZ load_fileZDECODE_THREADSZALLOW_INCORRECT_HEADERSZDISABLE_SECURITY_LIMITSmimetype_images primary_index enumeraterI) r%fpconvert_hdr_to_8bitbgr_modekwargsimagesrZfp_bytesZpreferred_decoderindexr?r'r'r(r)s:      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. )rrrr+r'r'r(rsz 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. )rrrxr+r'r'r(rx(szHeifFile.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. rrryr+r'r'r(ry0szHeifFile.premultiplied_alpharhrzcCs||j|j_dSr!rr}r'r'r(ry8scCs|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(r0DszHeifFile.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(rILsz HeifFile.infor9r.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=TszHeifFile.to_pillowr>cKst|j|fi|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(save[s+z HeifFile.savecCs*d|jjdt|ddd|DdS)NrMrmz images: cSsg|] }t|qSr')rrXr'r'r(r[r\z%HeifFile.__repr__..rP)rKr@ror+r'r'r(rRszHeifFile.__repr__cCs t|jSr!)rorr+r'r'r(__len__szHeifFile.__len__ccs|jEdHdSr!)rr+r'r'r(__iter__szHeifFile.__iter__cCs.|dks|t|jkr$td||j|SNrzinvalid image index: ror IndexError)r%rr'r'r( __getitem__szHeifFile.__getitem__cCs0|dks|t|jkr$td||j|=dSrr)r%keyr'r'r( __delitem__szHeifFile.__delitem__rr)r rcKs(tt|||fi|}|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. )rWr rri)r%r rr,r added_imager'r'r( add_frombytess  zHeifFile.add_frombytesrW)rTr/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 rr,r0rrIpop)r%rTrr'r'r( add_from_heifs zHeifFile.add_from_heifc Cs |jddks|jddkr$td|j}t||d<t|}|rN||d<t|}t|}|durx|dkrxt||}| | |j |j| }dD]}||jvr|j||j|<qdD]"}||jvrt |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. rrzEmpty images are not supported.r_rbN)r^r`rerf)rgrG)r ValueErrorrIrrrrrrr*rr tobytesr) r%rTrIrbr~imgrrZim_xmpr'r'r(add_from_pillows8      zHeifFile.add_from_pillowcCs|j|jjS)z+Returns the primary image as a numpy array.)rrr8r+r'r'r(r8szHeifFile.__array_interface__cCs@g}|jD]&}t|j}||j|j||jgq |j|j|gSr!) rbytesr,rir rrIrr)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!)rrrrr)r%Zim_copyr'r'r(Z__copys  zHeifFile.__copycCs|j|j|S)z`get_aux_image`` method of the primary :class:`~pillow_heif.HeifImage` in the container. :exception IndexError: If there are no images. )rrr)r%rkr'r'r(rszHeifFile.get_aux_image)NTF)r@rArBrCr)rErr rxryrr,r0rIr=rrRrrrrrrrr8rrZ_HeifFile__copyr__copy__r'r'r'r(rsF "       -$ rrhr.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)rZf_datar'r'r( is_supporteds rTFcKst|||fi|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)rr*)rrrrretrr'r'r( read_heif*s rrrr>)r rr/cKs,ttt|||fi|g|fi|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)rrWr )r rr,rrr'r'r(encodeFsrzlist[HeifImage])rr/c Ks$|dd}|dkrtjntj}t|s       rr9) 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. )rr)rr?r'r'r( from_pillowps rcKs t}|j|||fi||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. )rr)r rr,rr?r'r'r( from_bytes|s r)TF)TF)1rC __future__rrriortypingrZPILrrr constantsr miscr r r rrrrrrrrrrrrrr ImportErrorexZ_deffered_errorrrrFrVrWrrrrrrrrr'r'r'r(s6      H  NS