o jf^ @sdZddlmZmZddlmZddlmZddlmZddl 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)de*fddZ+d,de)fddZ,d,de)fddZ-d e.d!e/e0e0fdd fd"d#Z1d$e2e(dd fd%d&Z3d'e j de)fd(d)Z4d e.d!e/e0e0fde)fd*d+Z5d S)-z8Functions and classes for heif images to read and write.)copydeepcopy)SEEK_SET)Lock)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@szeZdZUdZeeefed< eed< ddZe ddZ e defd d Z e d d Z de j fd dZdddZdS) BaseImagez^Base class for :py:class:`HeifImage`, :py:class:`HeifDepthImage` and :py:class:`HeifAuxImage`.sizemodecCs&|j\|_|_||_d|_t|_dSN) size_moderr_c_image_datar _load_lock)selfc_imager&ZC:\Users\psant\Downloads\Anu Raj Backend\school\venv\Lib\site-packages\pillow_heif/heif.py__init__3s zBaseImage.__init__cCs||jS)zaDecodes image and returns image data. :returns: ``bytes`` of the decoded image. N)loadr"r$r&r&r'data9szBaseImage.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. N)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|u1zrIr<rrr*r&r&r'__repr__0zHeifDepthImage.__repr__r,cst}|j|_|S)r7N)rFr9rGrr$imagerHr&r'r9s  zHeifDepthImage.to_pillow) r<r=r>r?r(rQrr9 __classcell__r&r&rHr'rDys  rDc@seZdZdZddZdS) HeifAuxImagezdClass representing the auxiliary image associated with the :py:class:`~pillow_heif.HeifImage` class.c CrJrKrPr*r&r&r'rQrRzHeifAuxImage.__repr__N)r<r=r>r?rQr&r&r&r'rVs rVcseZdZdZfddZddZedefddZedefd d Z e j d efd d Z de j ffd d Z de defddZZS) HeifImagez;One image in a :py:class:`~pillow_heif.HeifFile` container.c sNt||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<|j} | r{| |jd<t||j|j} | r| d d vr| d |jd <| d |jd <dS| d |jd<dSdS)NcSsg|]}|dur|qSrr&.0ir&r&r' sz&HeifImage.__init__..cSg|] }|durt|qSr)rDrXr&r&r'r[)primary bit_depthexifrE thumbnails depth_imagesZauxxmpheifpixel_aspect_ratiotype)ZrICCZprofr+ icc_profileicc_profile_type nclx_profile)rFr(rErrr Z THUMBNAILSraZ DEPTH_IMAGESZdepth_image_listboolr^r4r_rGZ AUX_IMAGESZ aux_image_idsZ get_aux_typeappendrrer color_profile) r$r%rEr`rcrarbZ ctx_aux_infoaux_idZaux_typeZ heif_metarerlrHr&r'r(sL         zHeifImage.__init__cCsp|js t|jtrt|jdnd}d|jjd|jdd|jdd|j d|d t|j d gd S) Nz bytesnorLrMrrNr with z image data and raz thumbnails>) r" isinstancer!r lenr+rIr<rrrGget)r$Zs_bytesr&r&r'rQs&,zHeifImage.__repr__r,cCs|jjdddddvS)zD``True`` for images with the ``alpha`` channel, ``False`` otherwise.;sepr)AaN)rsplitr*r&r&r' has_alphaszHeifImage.has_alphacCst|jjdddddkS)zN``True`` for images with ``premultiplied alpha`` channel, ``False`` otherwise.rsrtrrvrxN)rjrryr*r&r&r'premultiplied_alphaszHeifImage.premultiplied_alphavaluecCs.|jr|j|r dnd|rdnd|_dSdS)Nrwrx)rzrreplacer$r|r&r&r'r{s$cs*t}|j|_t|j|jd<|S)r7original_orientationN)rFr9rGrrrSrHr&r'r9s  zHeifImage.to_pillowrmcCs|j|}t|S)zMethod to retrieve the auxiliary image at the given ID. :returns: a :py:class:`~pillow_heif.HeifAuxImage` class instance. N)r! get_aux_imagerV)r$rmZ aux_imager&r&r'rs zHeifImage.get_aux_image)r<r=r>r?r(rQrCrjrzr{setterrr9r4rVrrUr&r&rHr'rWs * rWc@s4eZdZdZd9ddZeddZed d Zed d Zed dZ e j de fddZ eddZ eddZ eddZdejfddZd:ddZddZddZd d!Zd"d#Zd$d%Zd&ed'eeeffd(d)Zd*edefd+d,Zd*ejdefd-d.Zed/d0Zd1d2Zd3d4Z d5d6Z!d7d8Z"e!Z#dS);HeifFileaRepresentation 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)NseekrZavifrvAVIFZheicrdHEIFZ remove_strideTZ hdr_to_16bit reload_sizecSr\r)rWrXr&r&r'r[r]z%HeifFile.__init__..r^F)hasattrrrrrfindr ZPREFERRED_DECODERrr _pillow_heifZ load_fileZDECODE_THREADSZALLOW_INCORRECT_HEADERSZDISABLE_SECURITY_LIMITSmimetype_images primary_index enumeraterG) r$fpconvert_hdr_to_8bitbgr_modekwargsimagesrZfp_bytesZpreferred_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. N)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. N)rrrr*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. N)rrrzr*r&r&r'rz.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. Nrrr{r*r&r&r'r{6rzHeifFile.premultiplied_alphar|cCs||j|j_dSrrr~r&r&r'r{>scCr)z:attr:`~pillow_heif.HeifImage.data` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. N)rrr+r*r&r&r'r+Brz HeifFile.datacCr)z:attr:`~pillow_heif.HeifImage.stride` property of the primary :class:`~pillow_heif.HeifImage`. :exception IndexError: If there are no images. N)rrr-r*r&r&r'r-JrzHeifFile.stridecCr)z`info`` dict of the primary :class:`~pillow_heif.HeifImage` in the container. :exception IndexError: If there are no images. N)rrrGr*r&r&r'rGRrz 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. N)rrr9r*r&r&r'r9ZszHeifFile.to_pillowcKst|j|fi|dS)a*Saves 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'. ``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'saveas)z HeifFile.savecCs*d|jjdt|ddd|DdS)NrLroz images: cSsg|]}t|qSr&)rBrXr&r&r'r[sz%HeifFile.__repr__..rO)rIr<rqr*r&r&r'rQs*zHeifFile.__repr__cCs t|jSr)rqrr*r&r&r'__len__s zHeifFile.__len__ccs|jEdHdSr)rr*r&r&r'__iter__szHeifFile.__iter__cCs.|dks |t|jkrtd||j|SNrzinvalid image index: rqr IndexError)r$rr&r&r' __getitem__s zHeifFile.__getitem__cCs0|dks |t|jkrtd||j|=dSrr)r$keyr&r&r' __delitem__s zHeifFile.__delitem__rrcKs(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. N)rWr rrk)r$rrr+r added_imager&r&r' add_frombytess zHeifFile.add_frombytesrTcCs@||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-r^N) r)rrrr+r-rrGpop)r$rTrr&r&r' add_from_heifs zHeifFile.add_from_heifc Cs |jddks|jddkrtd|j}t||d<t|}|r'||d<t|}t|}|dur<|dkrr?r(rCrrrzr{rrjr+r-rGrr9rrQrrrrrBr@r4rrWrrr6rrZ_HeifFile__copyr__copy__r&r&r&r'rsH "        +$ rr,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. r.sftypFrN)rr)rZf_datar&r&r' is_supported s  rTFcKst|||fi|S)aOpens the given HEIF 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. N)r)rrrrr&r&r' open_heifsrcKs.t|||fddi|}|D]}|q|S)aVOpens the given HEIF 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. rTN)rr))rrrrretrr&r&r' read_heif.s rrrcKs,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 )rrr+rrr&r&r'encodeJs,rrc Ks$|dd}|dkr tjntj}t|std|d||dg}|dds2|dd }|s8td t||d }t |fi|}t |D]>\}} | | j } d | d <||krm| jdi|d| d <| dd|j| j| j| jfdt| i| d| jiqL||dS)NformatrrzNo z encoder found.Z append_imagesZsave_allTrz)Cannot write file with no images as HEIF.rFr^r-rZimage_orientationr&)rrr ZAV1ZHEVCrZ get_lib_info RuntimeErrorrrr rr)rGrupdaterZ add_imagerrr+rr-r) rrr compressionZcompression_formatZimages_to_saverZ ctx_writerZrrGr&r&r'rUs>       r pil_imagecCst}|||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. N)rr)rr:r&r&r' from_pillowts 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. N)rr)rrr+rr:r&r&r' from_bytess r)TF)6r?rrior threadingrtypingrZPILrrr constantsr miscr r r rrrrrrrrrrrrrr ImportErrorexZ_deffered_errorrrrDrVrWrrjrrrrBr@r4rlistrrrr&r&r&r's<      H  RV "