o h] @sdZddlmZddlmZmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZzdd l Z Wne!ytZ"zdd l#m$Z$e$e"Z WYd Z"["nd Z"["wwGd ddZ%Gddde%Z&Gddde%Z'Gddde%Z(GdddZ)d2ddZ*d3d4ddZ+d3d4dd Z,d5d&d'Z-d6d*d+Z.d7d.d/Z/d8d0d1Z0d S)9z8Functions 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@sheZdZUdZded< ded< ddZedd Zedd d ZeddZ dddZ dddZ 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'P/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/pillow_heif/heif.py__init__4s zBaseImage.__init__cCs||jS)zaDecodes image and returns image data. :returns: ``bytes`` of the decoded image. )loadr$r%r'r'r(data9szBaseImage.datareturnintcCs||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(r/BszBaseImage.stridecCs|t|jt|jd}t|jddkrd}nt|d}d}|jd|f}t|jddkr>|t|jdf7}||d|jdS) zNumpy array interface support.rrz|u1zrLrArr r+r'r'r(__repr__0zHeifDepthImage.__repr__r-r8cst}|j|_|S)r9)rIr<rJrr%imagerKr'r(r<s  zHeifDepthImage.to_pillowr?)rArBrCrDr)rTr< __classcell__r'r'rKr(rGvs  rGc@seZdZdZddZdS) HeifAuxImagezdClass representing the auxiliary image associated with the :py:class:`~pillow_heif.HeifImage` class.c CrMrNrSr+r'r'r(rTrUzHeifAuxImage.__repr__N)rArBrCrDrTr'r'r'r(rYs rYcsleZdZdZfddZddZeddd Zedd d Zej dd d Zdfdd Z dddZ Z S) HeifImagez;One image in a :py:class:`~pillow_heif.HeifFile` container.c s:t||j}t|}t|}tjrdd|jDng}tjr)dd|j Dng}t |j t |j ||||d|_tjr_i}|jD]}||} | |vrRg|| <|| |qC||jd<t|} |rj||jd<| rq| |jd<t||j|j} | r| dd vr| d |jd <| d|jd <dS| d |jd <dSdS)NcSsg|]}|dur|qSr!r'.0ir'r'r( sz&HeifImage.__init__..cSg|] }|durt|qSr!)rGr[r'r'r(r^)primary bit_depthexifrH thumbnails depth_imagesauxxmpheiftype)rICCprofr, icc_profileicc_profile_type nclx_profile)rIr)rHrrr THUMBNAILSrd DEPTH_IMAGESdepth_image_listboolrar.rbrJ AUX_IMAGES aux_image_ids get_aux_typeappendrr color_profile) r%r&rHrcrgrdre ctx_aux_infoaux_idaux_type heif_metarwrKr'r(r)sF        zHeifImage.__init__cCsp|js t|jtrt|jdnd}d|jjd|jdd|jdd|j d|d t|j d gd S) Nz bytesnorOrPrrQr with z image data and rdz thumbnails>) r$ isinstancer#r lenr,rLrArr rJget)r%s_bytesr'r'r(rTs&,zHeifImage.__repr__r-rrcCs|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.rrrrr)rrr rr+r'r'r(premultiplied_alphaszHeifImage.premultiplied_alphavaluecCs.|jr|j|r dnd|rdnd|_dSdS)Nrr)rr replacer%rr'r'r(rs$r8cs*t}|j|_t|j|jd<|S)r9original_orientation)rIr<rJrrrVrKr'r(r<s  zHeifImage.to_pillowryr.rYcCs|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_imagerY)r%ry aux_imager'r'r(rs zHeifImage.get_aux_imager-rrrrrr?)ryr.r-rY) rArBrCrDr)rTrFrrsetterr<rrXr'r'rKr(rZs '    rZc@seZdZdZd?ddZeddZed d Zed d Zed dZ e j d@ddZ eddZ eddZ eddZ dAddZdBddZdd Zd!d"Zd#d$Zd%d&Zd'd(ZdCd-d.ZdDd1d2ZdEd3d4Zed5d6Zd7d8Zd9d:Zd;d<Zd=d>ZeZdS)FHeifFileaRepresentation 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|durg}d}nKt|}t|}|ddkr+tjdd}n|ddks9|ddkrAtjd d}nd}t |tj |||d d |d d |d tj |tj }||_dd|D|_d|_t|jD]\} } | jddr| |_qrdS)NseekravifrAVIFheicrhHEIF remove_strideT hdr_to_16bit reload_sizecSr_r!)rZr[r'r'r(r^r`z%HeifFile.__init__..raF)hasattrrrrrfindr PREFERRED_DECODERr _pillow_heif load_fileDECODE_THREADSALLOW_INCORRECT_HEADERSDISABLE_SECURITY_LIMITSmimetype_images primary_index enumeraterJ) r%fpconvert_hdr_to_8bitbgr_modekwargsimagesrfp_bytespreferred_decoderindexr>r'r'r(r)s>      zHeifFile.__init__cC|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(rz HeifFile.sizecCr)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 rz HeifFile.modecCr)z:attr:`~pillow_heif.HeifImage.has_alpha` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrrr+r'r'r(r(rzHeifFile.has_alphacCr)z:attr:`~pillow_heif.HeifImage.premultiplied_alpha` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. rrrr+r'r'r(r0rzHeifFile.premultiplied_alpharrrcCs||j|j_dSr!rrr'r'r(r8scCr)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,<rz HeifFile.datacCr)z:attr:`~pillow_heif.HeifImage.stride` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. )rrr/r+r'r'r(r/DrzHeifFile.stridecCr)z`info`` dict of the primary :class:`~pillow_heif.HeifImage` in the container. :exception IndexError: If there are no images. )rrrJr+r'r'r(rJLrz HeifFile.infor-r8cCs|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)NrOr}z images: cSsg|]}t|qSr')rr[r'r'r(r^sz%HeifFile.__repr__..rR)rLrArr+r'r'r(rTs*zHeifFile.__repr__cCs t|jSr!)rrr+r'r'r(__len__s zHeifFile.__len__ccs|jEdHdSr!)rr+r'r'r(__iter__szHeifFile.__iter__cCs.|dks |t|jkrtd||j|SNrzinvalid image index: rr IndexError)r%rr'r'r( __getitem__s zHeifFile.__getitem__cCs0|dks |t|jkrtd||j|=dSrr)r%keyr'r'r( __delitem__s zHeifFile.__delitem__r rrrcKs(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. )rZr rrv)r%r rr,r added_imager'r'r( add_frombytess zHeifFile.add_frombytesrWrZcCs@||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. )r/raN) r*rr rr,r/rrJpop)r%rWrr'r'r( add_from_heifs zHeifFile.add_from_heifc Cs |jddks|jddkrtd|j}t||d<t|}|r'||d<t|}t|}|dur<|dkr\}} | | j } d | d <||krm| jdi|d| d <| dd|j| j| j| jfdt| i| d| jiqL||dS)NformatrrzNo z encoder found. append_imagessave_allTrz)Cannot write file with no images as HEIF.rFrar/rimage_orientationr')rr AV1HEVCr get_lib_info RuntimeErrorrrr rr*rJrupdater add_imagerr r,rr/r) rrr compressioncompression_formatimages_to_saver ctx_writer]rrJr'r'r(rQs>       r pil_imager8cCst}|||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 rr)TF)r-r)r rrrr-r=)rrr-r=)rr8r-r)r rrrr-r)1rD __future__rrriortypingrPILrrr constantsr miscr r r rrrrrrrrrrrrrr ImportErrorex_deffered_errorrrrGrYrZrrrrrrrrr'r'r'r(s<      H  NS