a C gcD@sddlmZzddlmZmZWney2Yn0ddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZGdddeZddZdd ZGd d d e ZegZd S) )BytesIO)AvifImagePluginHeifImagePlugin) AvifImageFileBadImageOperationError BMPImageFile GIFImageFile HeicImageFile IcoImageFileImage JPEGImageFile PNGImageFileRGBAImageBufferRGBImageBuffer TIFFImageFile WebPImageFilec@s eZdZdS)UnsupportedRotationN)__name__ __module__ __qualname__rrT/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/willow/plugins/pillow.pyrsrcCsddl}|jSNr)Z PIL.Imager PILrrr _PIL_ImagesrcCsddl}|jSr)Z PIL.ImageCmsImageCmsrrrr _PIL_ImageCms#src @seZdZddZeddZeddZejddZ ejd d Z ejd d Z ejd dZ ejddZ ejddZejddZejddZddZddZejdAddZejdBeeeed!d"d#ZejdCeed$d%d&ZejdDed'd(d)ZejdEeeed+d,d-ZejdFeeed+d.d/ZejdGd0d1ZejdHd2d3Zejd4d5Zejd6d7Zeee ee!eje"d8d9ee#ee$ee%ee&ee'ee(d:d;Z)e*e+dd?Z.d@S)I PillowImagecCs ||_dSNimageselfr!rrr__init__*szPillowImage.__init__cCs tdSr)r)clsrrrcheck-szPillowImage.checkcCst}||vSr)rZregistered_extensionsvalues)r%Z image_formatformatsrrris_format_supported1s zPillowImage.is_format_supportedcCs|jjSr)r!sizer#rrrget_size6szPillowImage.get_sizecCsdS)Nrr+rrrget_frame_count:szPillowImage.get_frame_countcCs$|j}|jdvp"|jdko"d|jvS)N)RGBAZLAP transparency)r!modeinfo)r#imgrrr has_alpha?s zPillowImage.has_alphacCsdS)NFrr+rrr has_animationFszPillowImage.has_animationcCsL|jjdvr0|r"|jd}q6|jd}n|j}t||tjjS)N1r0r/RGB) r!r2r5convertrresizerZ ResamplingZLANCZOS)r#r*r!rrrr;Ks  zPillowImage.resizec Cs|\}}}}|jj\}}||ksH||ksH|dksH||ksH||ksH|dkrVtd|td|td|t||t||f}t|j|S)NrzInvalid crop dimensions: )r!r*rmaxminrcrop) r#ZrectlefttoprightZbottomwidthheightZ clamped_rectrrrr>Zs*  zPillowImage.cropcCsXt}|jj|jj|jjd}|d}|s.|S||}|sDtd|j|}t |S)zp Accept a multiple of 90 to pass to the underlying Pillow function to rotate the image. )ZiihzLSorry - we only support right angle rotations - i.e. multiples of 90 degrees) rZ TransposeZ ROTATE_90Z ROTATE_180Z ROTATE_270getrr!Z transposer)r#Zangler ZORIENTATION_TO_TRANSPOSEZ modulo_angleZtranspose_codeZrotatedrrrrotaters  zPillowImage.rotatecCs|s |St|ttfr&t|dks.td|jd}t d|jj |d|d|ddf}t |drx| |nt ||}t |d S) Nz6the 'color' argument must be a 3-element tuple or listr/rr-alpha_compositer9)r5 isinstancetuplelistlen TypeErrorr!r:rnewr*hasattrrKr)r#colorr!Z new_imagerrrset_background_color_rgbs   z$PillowImage.set_background_color_rgbcCs|jjdS)N icc_profiler!r3rFr+rrrget_icc_profileszPillowImage.get_icc_profilecCs|jjdS)NexifrVr+rrr get_exif_dataszPillowImage.get_exif_datarcCs\|}|dur|St}|t|}|r4dnd}|j|j||d||d}t|S)aY Transforms the color of the image to fit inside sRGB color gamut using the embedded ICC profile. The resulting image will always be in RGB(A) mode and will have a small generic sRGB ICC profile embedded. If the image does not have an ICC profile this operation is a no-op. Images without a profile are commonly assumed to be in sRGB color space already. :param rendering_intent: Controls how out-of-gamut colors and handled. Defaults to 0 (perceptual) because this is what Pillow defaults to. :return: PillowImage in RGB mode :raises: PIL.ImageCms.PyCMSError Further reading: * https://pillow.readthedocs.io/en/stable/reference/ImageCms.html#PIL.ImageCms.profileToProfile * https://www.permajet.com/blog/rendering-intents-explained/ Nr/r9ZsRGB)ZrenderingIntentZ outputMode) rWrZImageCmsProfilerr5ZprofileToProfiler!Z createProfiler)r#Zrendering_intentrUrZ output_moder!rrrtransform_colorspace_to_srgbsz(PillowImage.transform_colorspace_to_srgbUFT)qualityoptimize progressiveapply_optimizersc Cs|jjdvr|jd}n|j}d|i}|r4d|d<|r@d|d<|}|durX||d<|} | durp| |d <|j|d fi||r||d t|S) a Save the image as a JPEG file. :param f: the file or file-like object to save to :param quality: the image quality :param optimize: Whether Pillow should optimize the file. When True, Pillow will attempt to compress the palette by eliminating unused colors. :param progressive: whether to save as progressive JPEG file. :param apply_optimizers: controls whether to run any configured optimizer libraries :return: JPEGImageFile r7r9r\Tr]r^NrUrXZJPEGZjpeg)r!r2r:rWrYsaver]r ) r#fr\r]r^r_r!kwargsrU exif_datarrr save_as_jpegs$  zPillowImage.save_as_jpeg)r]r_c Csi}|j}|}|durL|jjdkrB|}|j}||d<q`||d<n|jdkr`|d}|rld|d<|}|dur||d<|j|dfi||r||d t|S) a Save the image as a PNG file. :param f: the file or file-like object to save to :param optimize: Whether Pillow should optimize the file. When True, Pillow will attempt to compress the palette by eliminating unused colors. :param apply_optimizers: controls whether to run any configured optimizer libraries :return: PNGImageFile NCMYKrUr9Tr]rXZPNGZpng) r!rWr2rZr:rYr`r]r ) r#rar]r_rbr!rU pillow_imagercrrr save_as_pngs(      zPillowImage.save_as_png)r_cCsl|j}|jdvr$|jdtjjd}i}d|jvr@|jd|d<|j|dfi||rd||dt |S)N)Lr0r0)Zpaletter1ZGIFZgif) r!r2r:rZPaletteZADAPTIVEr3r`r]r)r#rar_r!rbrrr save_as_gif<s   zPillowImage.save_as_gifP)r\losslessr_c Cs~||d}|j}|}|durN|jdkrF|}|j}||d<n||d<|j|dfi||rv|sv||dt|S)a Save the image as a WEBP file. :param f: the file or file-like object to save to :param quality: the image quality :param lossless: whether to save as lossless WEBP file. :param apply_optimizers: controls whether to run any configured optimizer libraries. Note that when lossless=True, this will be ignored. :return: WebPImageFile )r\rkNrerUZWEBPZwebp)r!rWr2rZr`r]r r#rar\rkr_rbr!rUrfrrr save_as_webpOs   zPillowImage.save_as_webpc Csd|i}|rddd}|j}|}|durZ|jdkrR|}|j}||d<n||d<|j|dfi||s|r||d t|S) a Save the image as a HEIC file. :param f: the file or file-like object to save to :param quality: the image quality :param lossless: whether to save as lossless HEIC/HEIF file. :param apply_optimizers: controls whether to run any configured optimizer libraries. Note that when lossless=True, this will be ignored. :return: HeicImageFile r\r\ZchromaNrerUZHEIFheic)r!rWr2rZr`r]r rlrrr save_as_heiczs   zPillowImage.save_as_heicc Csd|i}|rddd}|j}|}|durZ|jdkrR|}|j}||d<n||d<|j|dfi||s|r||dt|S) Nr\rnrorprerUZAVIFrq)r!rWr2rZr`r]rrlrrr save_as_avifs   zPillowImage.save_as_avifcCs&|j|d|r||dt|S)NZICOZico)r!r`r]r )r#rar_rrr save_as_icos zPillowImage.save_as_icocCs ddlm}||j}t|S)Nr)ImageOps)rruZexif_transposer!r)r#rur!rrr auto_orients  zPillowImage.auto_orientcCs|jSrr r+rrrget_pillow_imageszPillowImage.get_pillow_image)ZcostcCs*|jdt|j}|||Sr)raseekropenload)r%Z image_filer!rrrrzs zPillowImage.opencCs*|j}|jdkr|d}t|j|S)Nr9)r!r2r:rr*tobytesr"rrr to_buffer_rgbs  zPillowImage.to_buffer_rgbcCs*|j}|jdkr|d}t|j|S)Nr/)r!r2r:rr*r|r"rrrto_buffer_rgbas  zPillowImage.to_buffer_rgbaN)r)r[FFT)FT)T)rjFT)rjFT)rjFT)T)/rrrr$ classmethodr&r)r Z operationr,r.r5r6r;r>rGrTrWrYrZintboolrdrgrirmrrrsrtrvrwZconverter_fromr r rrrrr rr rzZ converter_torr}rr~rrrrr)s         !  *+,*-      rN)iorZ pillow_heifrr ImportErrorZ willow.imagerrrrr r r r r rrrr ExceptionrrrrZwillow_image_classesrrrrs  <[