a C gA) @sdZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddl m Zdd lmZdd lmZdd lmZdd lmZmZmZmZmZmZmZmZz dd lZWn8eyZ z ddl!m"Z"e"e ZWYd Z [ n d Z [ 00Gddde j Z#Gddde#Z$ddddZ%ddddddZ&ddddddZ'ddd d!Z(Gd"d#d#e#Z)ddd$d%Z*ddddd&d'Z+ddddd(d)Z,ddd*d+Z-d,d-Z.ddd.d/d0d1Z/ddd.d/d2d3Z0d4dddd5d6d7Z1d S)8zPlugins for the Pillow library.) annotations)chain)IO)warn)Image ImageFile ImageSequence) __version__)options)HeifCompressionFormat)HeifFile) CtxEncode_exif_from_pillow _get_bytes_get_orientation_for_encoder_get_primary_index_pil_to_supported_mode_xmp_from_pillowset_orientationN) DeferredErrorcseZdZUdZdZded<dZded<fdd Zd d Zfd d Z e dddvrdddddZ ddddZ ddddZ ddddZeddddZeddd d!Zddd"d#Zddd$d%d&ZZS)'_LibHeifImageFilezVBase class with all functionality for ``HeifImageFile`` and ``AvifImageFile`` classes.NzHeifFile | None _heif_fileTstr_modecsd|_tj|i|dS)Nr)_LibHeifImageFile__framesuper__init__)selfargskwargs __class__T/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/pillow_heif/as_plugin.pyr)sz_LibHeifImageFile.__init__c Cszt|jdddd}Wn<tttttfyR}ztt|dWYd}~n d}~00|j|_ ||_ |j |_ | |j g|_dS)NTF)Zconvert_hdr_to_8bitZ hdr_to_16bitZ remove_stride)r fpOSError ValueError SyntaxError RuntimeErrorEOFErrorrmimetypeZcustom_mimetyper primary_indexr_init_from_heif_fileZtile)rZ heif_file exceptionr#r#r$_open-s$ z_LibHeifImageFile._opencs|jr|j|}z0|j}|j|_|||d|j|jfWn"t yft j sZ|Yn0|j rt |ddrt|j dr|j d|_ |jsd|_tS)NrawZ _exclusive_fpFclose)rtelldatasize_sizeZ load_prepare frombytesmodeZstrider*rZLOAD_TRUNCATED_IMAGESr%getattrhasattrr1 is_animatedrload)rZ frame_heifr3r!r#r$r;9s"  z_LibHeifImageFile.load)z10.1z10.2z10.3dictreturncCs:|jddr6|jddd}|dr6||dSiS)zReturns a dictionary containing the XMP tags. Requires ``defusedxml`` to be installed. :returns: XMP tags in a dictionary. xmpNr r)infogetrsplitZ_getxmp)rZxmp_datar#r#r$getxmpOs z_LibHeifImageFile.getxmpint)framecCsl||sdS||_||tdddkrBtj|j|j|_ t |dd}|durht |ddrhd|_ dS)Nz10.Z_exif_loadedF) _seek_checkrr- pil_versionrcorenewrr5imr8rI)rrGexifr#r#r$seekZs   z_LibHeifImageFile.seekcCs|jSN)rrr#r#r$r2ksz_LibHeifImageFile.tellNonecCsdSrQr#rRr#r#r$verifynsz_LibHeifImageFile.verifycCs|jrt|jSdS)zbReturns the number of available frames. :returns: Frame number, starting with 0. r )rlenrRr#r#r$n_framesqsz_LibHeifImageFile.n_framesboolcCs |jdkS)zTReturns ``True`` if this image contains more than one frame, or ``False`` otherwise.r )rVrRr#r#r$r:ysz_LibHeifImageFile.is_animatedcCs&|dks||jkrtd||kS)Nrz attempt to seek outside sequence)rVr*r2)rrGr#r#r$rJ~sz_LibHeifImageFile._seek_check) img_indexr?cCsD|jr@|j|j|_|j|j|_|j|j|_t|j|jd<dS)NZoriginal_orientation)rr4r5r7rrBr)rrXr#r#r$r-s z&_LibHeifImageFile._init_from_heif_file)__name__ __module__ __qualname____doc__r__annotations__Z!_close_exclusive_fp_after_loadingrr/r;rKrErPr2rTpropertyrVr:rJr- __classcell__r#r#r!r$r"s$     rc@seZdZdZdZdZdS) HeifImageFilez1Pillow plugin class type for a HEIF image format.HEIFzHEIF containerNrYrZr[r\formatformat_descriptionr#r#r#r$r`sr`rWr>cCs.t|d}|dddkrdS|dddvS)N r<ftypF) sheicsheixsheimsheisshevcshevxshevmshevssmif1smsf1rr%magicr#r#r$_is_supported_heifs rkz Image.Imagez IO[bytes]z str | bytesrNr% _filenamecCst||tjdSrQ) __save_oner HEVCrlr#r#r$ _save_heifsrpcCst||tjdSrQ) __save_allr rorlr#r#r$_save_all_heifsrrrScKsntfi|ttjtttdrFttjt t tjt gd}t tjdt tj|dS)zRegisters a Pillow plugin for HEIF format. :param kwargs: dictionary with values to set in options. See: :ref:`options`. ra)z.heicz.heicsz.heifz.heifsz.hifz image/heifN)__options_updater register_openr`rcrk _pillow_heif get_lib_info register_saverpregister_save_allrr register_mimeregister_extensionsr extensionsr#r#r$register_heif_openers r}c@seZdZdZdZdZdS) AvifImageFilez2Pillow plugin class type for an AVIF image format.AVIFzAVIF containerNrbr#r#r#r$r~sr~cCs.t|d}|dddkrdS|dddkS)Nrer<rfrgFsavifrhrir#r#r$_is_supported_avifs rcCst||tjdSrQ)rnr AV1rlr#r#r$ _save_avifsrcCst||tjdSrQ)rqr rrlr#r#r$_save_all_avifsrcKstdtddtds*tddddStfi|ttjtt t tjt t tjt dg}ttjd ttj|dS) zRegisters a Pillow plugin for AVIF format. :param kwargs: dictionary with values to set in options. See: :ref:`options`. zThe AVIF support in this library is marked as deprecated and will be removed in the next version. If you still need AVIF support until it natively appears in Pillow, use the https://github.com/fdintino/pillow-avif-plugin project instead. stacklevelrz=This version of `pillow-heif` was built without AVIF support.r Nz.avifz image/avif)rDeprecationWarningrurvrsrrtr~rcrrwrrxrryrzr{r#r#r$register_avif_openers  rcKs|D]\}}|dkr |t_q|dkr0|t_q|dkr@|t_q|dkrP|t_q|dkr`|t_q|dkrp|t_q|dkr|t_q|dkr|t_ q|d kr|t_ q|d kr|t_ qt d |d d qdS)z`Internal function to set options from `register_avif_opener` and `register_heif_opener` methods.Z thumbnailsZ depth_imagesZ aux_imagesZqualityZ save_to_12bitZdecode_threadsZallow_incorrect_headersZsave_nclx_profileZpreferred_encoderZpreferred_decoderzUnknown option: r rN) itemsr Z THUMBNAILSZ DEPTH_IMAGESZ AUX_IMAGESZQUALITYZSAVE_HDR_TO_12_BITZDECODE_THREADSZALLOW_INCORRECT_HEADERSZSAVE_NCLX_PROFILEZPREFERRED_ENCODERZPREFERRED_DECODERr)r kvr#r#r$rss,rsr )rNr%compression_formatcCs6t|fi|j}t||dfi|j||dS)NT)r encoderinfo_pil_encode_imagesave)rNr%r ctx_writer#r#r$rnsrnc Cst|fi|j}t|dr$|nd}|jdg}ttt|||jdd}t tt||D]"\}}t ||||kfi|jqj|durt|dr| || |dS)Nr2 append_imagesr,rP) rrr9r2rCrrrIterator enumeraterrPr) rNr%rr current_framerr,irGr#r#r$rqs rqr)ctximgprimaryr?cKs|jddks|jddkr$td|j}t||d<t||d<|jfi|||d<|jdkr|j|fdt |i|n.t |}|j |j|j| fdt |i|dS) Nrr zEmpty images are not supported.rOr@rZYCbCrZimage_orientation) r4r'rBcopyrrupdater7Zadd_image_ycbcrrrZ add_imagetobytes)rrrr rBr#r#r$rs    r)2r\ __future__r itertoolsrtypingrwarningsrZPILrrrr rKr constantsr Zheifr miscrrrrrrrrru ImportErrorexZ_deffered_errorrrr`rkrprrr}r~rrrrrsrnrqrr#r#r#r$s<        (   i