g5je]dZddlmZmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZ dd l Z Gd dZ%Gdde%Z&Gdde%Z'Gdde%Z(GddZ)de*fdZ+d#de)fdZ,d#de)fdZ-de.de/e0e0fdd fdZ1de2e(dd fdZ3d e jde)fd!Z4de.de/e0e0fde)fd"Z5y #e!$rZ"dd l#m$Z$e$e"Z Yd Z"["d Z"["wwxYw)$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) DeferredErrorceZdZUdZeeefed< eed< dZe dZ e defdZ e dZ de jfd Zd d Zy ) BaseImagez^Base class for :py:class:`HeifImage`, :py:class:`HeifDepthImage` and :py:class:`HeifAuxImage`.sizemodecp|j\|_|_||_d|_t |_yN) size_moderr _c_image_datar _load_lock)selfc_images B/var/www/html/qr/venv/lib/python3.12/site-packages/pi_heif/heif.py__init__zBaseImage.__init__3s-&00 49  &c:|j|jS)zaDecodes image and returns image data. :returns: ``bytes`` of the decoded image. )loadr%r's r)datazBaseImage.data9s zzr+returncN|j|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.s r)r2zBaseImage.strideBs }}###r+c||jt|jt|jdz }t|jddkrd}nt|dz }d}|j d|f}t|jddkDr|t|jdfz }||d|j dS) zNumpy array interface support.rr z|u1zrTrErr r.s r)__repr__zHeifDepthImage.__repr__C4>>**+1TYYq\N!DIIaL>499+UVWWr+r0cbt|}|jj|_|S)r>)rRrArSrr'imagerTs r)rAzHeifDepthImage.to_pillows) !#YY^^%  r+) rErFrGrHr*r\rrA __classcell__rTs@r)rOrOys&f3X5;;r+rOceZdZdZdZy) HeifAuxImagez`Class representing the auxiliary image associated with the :py:class:`~pi_heif.HeifImage` class.c d|jjd|jdd|jdd|jd SrVr[r.s r)r\zHeifAuxImage.__repr__r]r+N)rErFrGrHr\rMr+r)rdrds jXr+rdceZdZdZfdZdZedefdZedefdZ e jdefdZ de jffd Z d e defd ZxZS) HeifImagez7One image in a :py:class:`~pi_heif.HeifFile` container.ct|||j}t|}t |}t j r|jDcgc]}|| c}ng}t jr&|jDcgc]}|t|c}ng}t|jt|j||||d|_t j rPi}|j"D]0} |j%| } | |vrg|| <|| j'| 2||jd<t)|} |r||jd<| r| |jd<|j*} | r| |jd<t-||j|j.} | r?| ddvr%| d|jd <| d|jd <y| d|jd <yycc}wcc}w) N)primary bit_depthexifrQ thumbnails depth_imagesauxxmpheifpixel_aspect_ratiotype)rICCprofr/ icc_profileicc_profile_type nclx_profile)rRr*rQrrr THUMBNAILSrl DEPTH_IMAGESdepth_image_listrOboolrir:rjrS AUX_IMAGES aux_image_ids get_aux_typeappendrrqr color_profile)r'r(rQrkroirlrm ctx_aux_infoaux_idaux_type heif_metarqrrTs r)r*zHeifImage.__init__s !&//h'H%[b[m[m73E3E'Wa'Wsu V]VjVj(@(@ R1AM^A  Rpr GOO,W../ $(      L!// 6"//7</-/L*X&--f5  6 ,DIIe "7+ "DIIe   )DIIf $77 .@DII* +w 2(/(=(= V$(88+8+@ -(0=f0E ,-,9&,A .) =(X RsGGG! G!cb|jst|jtrt |j dnd}d|j jd|jdd|jdd|jd|d t |jjd gd S) Nz bytesnorWrXrrYr  with z image data and rlz thumbnails>) r% isinstancer$rlenr/rTrErr rSget)r's_bytess r)r\zHeifImage.__repr__s/3zzZ W`=aS^$F+gk''($))A,q1a {S9,S|R1P-Q,RR^ ` r+r0cJ|jjddddvS)zD``True`` for images with the ``alpha`` channel, ``False`` otherwise.;sepr)Aa)r splitr.s r) has_alphazHeifImage.has_alphas(yy3'*2.*<> ))#CSVWDI r+ct|}|jj|_t |j|jd<|S)r>original_orientation)rRrArSrrr_s r)rAzHeifImage.to_pillowsA !#YY^^% -40!// F! ?E(W!1(W !$,,/ +HE1vvzz)U+%*" +)Xs #F + F cH|j|jjS)z:attr:`~pi_heif.HeifImage.size` property of the primary :class:`~pi_heif.HeifImage`. :exception IndexError: If there are no images. )rrrr.s r)rz HeifFile.size ||D../444r+cH|j|jjS)z:attr:`~pi_heif.HeifImage.mode` property of the primary :class:`~pi_heif.HeifImage`. :exception IndexError: If there are no images. )rrr r.s r)r z HeifFile.mode%rr+cH|j|jjS)z:attr:`~pi_heif.HeifImage.has_alpha` property of the primary :class:`~pi_heif.HeifImage`. :exception IndexError: If there are no images. )rrrr.s r)rzHeifFile.has_alpha-s ||D../999r+cH|j|jjS)z:attr:`~pi_heif.HeifImage.premultiplied_alpha` property of the primary :class:`~pi_heif.HeifImage`. :exception IndexError: If there are no images. rrrr.s r)rzHeifFile.premultiplied_alpha5s ||D../CCCr+rc@||j|j_yr"rrs r)rzHeifFile.premultiplied_alpha=s?D T''(` 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:`~pi_heif.options.SAVE_NCLX_PROFILE` ``matrix_coefficients`` - int, nclx profile: color conversion matrix coef., default=6 (BT.601, 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'rrs r)savez HeifFile.save`sR t||R262r+c d|jjdt|d|Dcgc] }t|c}dScc}w)NrWrz images: rZ)rTrErrK)r'rs r)r\zHeifFile.__repr__sB4>>**+6#d)IW[F\RSs1vF\E]]^__F\sAc,t|jSr")rrr.s r)__len__zHeifFile.__len__s4<<  r+c#8K|jEd{y7wr")rr.s r)__iter__zHeifFile.__iter__s<<s cv|dks|t|jk\rtd||j|SNrzinvalid image index: rr IndexError)r'rs r) __getitem__zHeifFile.__getitem__s; 19T\\!224UG<= =||E""r+ct|dks|t|jk\rtd||j|=yrr)r'keys r) __delitem__zHeifFile.__delitem__s8 7cS..4SE:; ; LL r+r rc jtt|||fi|}|jj||S)aLAdds 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:`~pi_heif.HeifImage` added object. )rgrrr)r'r rr/r added_images r) add_frombyteszHeifFile.add_frombytess4 $d Ef EF  K(r+r`c |j|j|j|j|j|j }t |j|_|jjdd|S)zAdd image to the container. :param image: :py:class:`~pi_heif.HeifImage` class to add from. :returns: :py:class:`~pi_heif.HeifImage` added object. )r2riN) r-rr rr/r2rrSpop)r'r`rs r) add_from_heifzHeifFile.add_from_heifsm  (( JJ JJ JJ<< ) $EJJ/ Y-r+c|jddks|jddkr td|jj}t ||d<t |}|r||d<t |}t|}||dk7r t||}|j|j|j|j|j}dD]-}||jvs|j||j|</dD]6}||jvst|j||j|<8t ||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:`~pi_heif.HeifImage` added object. rr zEmpty images are not supported.rkro)rjrlrurvrq)rwrQ)r ValueErrorrSrrrrrrr-rr tobytesr) r'r`rSrorimgrrim_xmps r)add_from_pillowzHeifFile.add_from_pillowso ::a=A A!!3>? ?zz (/V u% DK.t4$U+  +0D0Ic#78C  (( HH HH KKM h 8Cejj (- 3   % 80 BCejj (0C(A   % B$5U#;  !%( &,K  U #r+cH|j|jjS)z+Returns the primary image as a numpy array.)rrr<r.s r)r<zHeifFile.__array_interface__s||D../CCCr+cg}|jD]J}t|j}|j|j|j ||j gL|j|j|gSr") rbytesr/rr rrSrr)r'im_descimim_datas r) __getstate__zHeifFile.__getstate__sc,, ABBGGnG NNBGGRWWgrww? @ A""DMM7;;r+c|j|\|_|_}|D]#}|\}}}}|j|||}||_%yr")r*rrrrS) r'staterrim_modeim_sizerim_infors r) __setstate__zHeifFile.__setstate__sU 491DM6 'G18 .GWgw,,WgwGK&K  'r+ct}t|j|_|j|_|j|_|Sr")rrrrr)r'im_copys r)__copyzHeifFile.__copys8*t||,== $ 2 2r+cR|j|jj|S)z`get_aux_image`` method of the primary :class:`~pi_heif.HeifImage` in the container. :exception IndexError: If there are no images. )rrr)r'rs r)rzHeifFile.get_aux_images$ ||D../==fEEr+)NTFrD)$rErFrGrHr*rLrr rrrr{r/r2rSrrArr\rrrrrKrIr:rrgrrr<rr_HeifFile__copyr__copy__rMr+r)rrsg +B5555::DDEE E557755<5;;<)3V`! #  # U38_ 9$"U[["Y"HDD<'FHr+rr0cHt|d}|dddk7ryt|dk7S)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. r4sftypFr)rr)rf_datas r) is_supportedrs1B F a{g V $ **r+c t|||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:`~pi_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)rrrrs r) open_heifrs& B+X @ @@r+c Pt|||fi|}|D]}|j|S)aNOpens the given HEIF image file and decodes all images. .. note:: In most cases it is better to call :py:meth:`~pi_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:`~pi_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. )rr-)rrrrretrs r) read_heifr-s5, 2*H ? ?C   Jr+r rc Jttt|||fi|g|fi|y)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)rrgr)r rr/rrs r)encoderIs)IidDCFCDErTVTr+rc |jdd}|dk(rtjntj}t j |st d|d||jdgz}|jdds|dd }|s td t||jd }t|fi|}t|D]\}} | j| jj} d | d <||k(r| jdi|d| d <| jdd|j | j"| j$| j&fdt)| i| d| j*i|j-|y)NformatrrzNo z encoder found. append_imagessave_allTr z)Cannot write file with no images as HEIF.rFrir2rimage_orientationrM)rr AV1HEVCr get_lib_info RuntimeErrorrrr rr-rSrupdater add_imagerr r/rr2r) rrr compressioncompression_formatimages_to_saver ctx_writerrrSs r)rrTsz**Xv.K6AV6K.22QfQkQk  "; /S _=>>&,vzz/2/N&NN ::j$ ''+ DEE&~vzz/7RSM,77IN+ 3  xx}}Y  DKK !& !"DO 1  HH HH HH ;4@    ::   NN2r+ pil_imagec<t}|j||S)zCreates :py:class:`~pi_heif.HeifFile` from a Pillow Image. :param pil_image: Pillow :external:py:class:`~PIL.Image.Image` class. :returns: New :py:class:`~pi_heif.HeifFile` object. )rr)rrCs r) from_pillowrss  Ai Hr+c Dt}|j|||fi||S)aBCreates :py:class:`~pi_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:`~pi_heif.HeifFile` object. )rr)r rr/rrCs r) from_bytesrs'  AAOOD$// Hr+)TF)6rHrrior threadingrtypingrPILrrr constantsr miscr r rrrrrrrrrrrrrrr ImportErrorex_deffered_errorrrrOrdrgrr{rrrrKrIr:rlistrrrrMr+r)r$s=>,&!O&O&dY0X9XS' S'lWWt + +AA,8UUE#s(OUDU4 ?T>  5;;  8    S  c3h  H  }!.R H!sCC:"C55C: