a d=@sdZddlmZddlZddlZddlmZddlm Z ddddd d Z ddddd d Z ddddd dZ ddddddZ ddddddZddddddZddddddZddddddZddddddZddddddZddddddZdddddd Zddddd!d"Zddddd#d$Zddddd%d&Zddddd'd(Zddddd)d*Zddddd+d,Zddddd-d.Zddddd/d0Zddddd1d2Zddddd3d4Zddd5d6d7Z dddd8d9d:Z!ddd5d;d<Z"dddd=d>d?Z#ddddd@dAZ$dddddBdCZ%dddddDdEZ&dddddFdGZ'dddddHdIdJZ(dddddHdKdLZ)dddddHdMdNZ*dddddHdOdPZ+dddQdRdSZ,dddQdTdUZ-dddVddWdXdYZ.dad[d[d\d]d[d^d_d`Z/dS)ba8Provide blending functions and types. Adapted from https://github.com/addisonElliott/pypdn/blob/master/pypdn/reader.py and https://gitlab.com/inklabapp/pyora/-/blob/master/pyora/BlendNonSep.py MIT License Copyright (c) 2020 FredHappyface Credits to: MIT License Copyright (c) 2019 Paul Jewell For implementing blending from the Open Raster Image Spec MIT License Copyright (c) 2018 Addison Elliott For implementing blending from Paint.NET MIT License Copyright (c) 2017 pashango For implementing a number of blending functions used by other popular image editors ) annotationsN)Image) BlendTypez np.ndarray) background foregroundreturncCs~|S)zBlendType.NORMAL.rrr r Y/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/blendmodes/blend.pynormalsr cCst||ddS)zBlendType.MULTIPLY.?npclipr r r r multiply$srcCst||dS)zBlendType.ADDITIVE.rrminimumr r r r additive)src CsVtjdd6t|dktdd||ddWdS1sH0YdS)zBlendType.COLOURBURN.ignoredivider rN)rerrstatewheremaximumr r r r colourburn.src CsRtjdd2t|dkt|d|ddWdS1sD0YdS)zBlendType.COLOURDODGE.rrrNrrrrr r r r colourdodge6src CsVtjdd6t|dkt|dd|ddWdS1sH0YdS)zBlendType.REFLECT.rrrNrr r r r reflect<sr c CsVtjdd6t|dkt|dd|ddWdS1sH0YdS)zBlendType.GLOW.rrrrNrr r r r glowDsr!cCs.t|dkd||ddd|d|S)zBlendType.OVERLAY.?rr)rrr r r r overlayLs  r#cCst||S)zBlendType.DIFFERENCE.)rabsr r r r differenceUsr%cCst||dS)zBlendType.NEGATION.r rrr r r r negationZsr'cCs t||S)zBlendType.LIGHTEN.r&r r r r lighten_sr(cCs t||S)zBlendType.DARKEN.rr r r r darkendsr)cCs||||S)zBlendType.SCREEN.r r r r r screenisr*cCsJt.tdtt|t|AWdS1s<0YdS)zBlendType.XOR.rN)warningscatch_warnings simplefilterimageIntToFloatimageFloatToIntr r r r xorns  r0cCs(d||||dd|d|S)zBlendType.SOFTLIGHT.rr r r r r softlightvsr1c CsBt|dkt|d|dtdd|d|dddS)zBlendType.HARDLIGHT.r"rr@)rrrr r r r hardlight}s "r3cCst||dddS)zBlendType.GRAINEXTRACT.r"r rrr r r r grainextractsr4cCst||dddS)zBlendType.GRAINMERGE.r"r rrr r r r grainmergesr5cCstd|d|dS)zBlendType.DIVIDE.g?gp?rrr r r r rsrcCs4t|d||dkt|d|d|dkS)zBlendType.PINLIGHT.rr")rrrr r r r pinlights  r6cCs0t||d|dkt|d|d|dkS)zBlendType.VIVIDLIGHT.rr")rrr r r r vividlights  r7cCs||d||S)zBlendType.EXCLUSION.r2r r r r r exclusionsr8)coloursrcCsN|dddddfd|dddddfd|dddddfdS)zLuminosity. :param colours: x by x by 3 matrix of rgb color components of pixels :return: x by x by 3 matrix of luminosity of pixels NrgA`"?rgbX9?rgv/?r r9r r r _lumsr;)originalColours newLuminosityrc CsH|}t|}||}|dddddf|7<|dddddf|7<|dddddf|7<t|}tj|dd}tj|dd}t|jdD]}t|jdD]}|||} |||f}|||f} |||f} | dkr|| |||| |||<| dkr|| |d|| ||||<qq|S)z3Set a new luminosity value for the matrix of color.NrrrZaxis)copyr;rminmaxrangeshape) r<r=_coloursZ _luminosityZdeltaLumZ _minColoursZ _MaxColoursij_colourZ minColourZ maxColourr r r _setLums0      rHcCstj|ddtj|ddS)zzSaturation. :param colours: x by x by 3 matrix of rgb color components of pixels :return: int of saturation of pixels rr>)rrAr@r:r r r _satsrI)r< newSaturationrc Cs6|}t|jdD]}t|jdD]}|||}d}d}d}||||krb||}}||||kr|||}}||||kr||}}||||dkr|||||||f||||||||<|||f||||<n d||||<d||||<d||||<q*q|S)aSet a new saturation value for the matrix of color. The current implementation cannot be vectorized in an efficient manner, so it is very slow, O(m*n) at least. This might be able to be improved with openCL if that is the direction that the lib takes. :param c: x by x by 3 matrix of rgb color components of pixels :param s: int of the new saturation value for the matrix :return: x by x by 3 matrix of luminosity of pixels rrrr )r?rBrC) r<rJrDrErFrGZminIZmidIZmaxIr r r _setSats,     rKcCstt|t|t|S)zBlendType.HUE.rHrKrIr;r r r r huesrMcCstt|t|t|S)zBlendType.SATURATION.rLr r r r saturationsrNcCst|t|S)zBlendType.COLOUR.rHr;r r r r coloursrPcCst|t|S)zBlendType.LUMINOSITY.rOr r r r luminosity srQ)backgroundAlphaforegroundAlphabackgroundColourforegroundColourc Cs|~||}tjdddLtt||dddddf||dddddf}Wdn1sj0Y||fS)u 'clip' composite mode. All parts of 'layer above' which are alpha in 'layer below' will be made also alpha in 'layer above' (to whatever degree of alpha they were) Destination which overlaps the source, replaces the source. Fa = 0; Fb = αs co = αb x Cb x αs αo = αb x αs rrinvalidNrrrrrRrSrTrUZoutAlphaZoutRGBr r r destins "rZc Cs~|d|}tjdddPtt|d|dddddf||dddddf}Wdn1sr0Y||fS)zReverse 'Clip' composite mode. All parts of 'layer below' which are alpha in 'layer above' will be made also alpha in 'layer below' (to whatever degree of alpha they were) rrrVNrXrYr r r destout*s  $"r[c Cs|d|||}tjdddrtt|d|dddddf|t||dddddf||dddddf}Wdn1s0Y||fS)zPlace the layer below above the 'layer above' in places where the 'layer above' exists... where 'layer below' does not exist, but 'layer above' does, place 'layer-above' rrrVNrXrYr r r destatopAs $ "r\c Cs|||d|}tjdddrtt||dddddf|t|d|dddddf||dddddf}Wdn1s0Y||fS)zWPlace the layer below above the 'layer above' in places where the 'layer above' exists.rrrVNrXrYr r r srcatopVs $"r])imagercCs|dS)zConvert a numpy array representing an image to an array of floats. Args: image (np.ndarray): numpy array of ints Returns: np.ndarray: numpy array of floats r r^r r r r.hs r.cCs|dtjS)zConvert a numpy array representing an image to an array of ints. Args: image (np.ndarray): numpy array of floats Returns: np.ndarray: numpy array of ints r_)ZastyperZuint8r`r r r r/ts r/r)rr blendTyperc0Cstjttjttjttjttj t tj t tj ttjttjttjttjttjttjttjttjttjt tj!t"tj#t$tj%t&tj't(tj)t*tj+t,tj-t.tj/t0i}||vrt||S||||S)aBlend pixels. Args: background (np.ndarray): background foreground (np.ndarray): foreground blendType (BlendType): the blend type Returns: np.ndarray: new array representing the image background: np.ndarray, foreground: np.ndarray and the return are in the form [[[0. 0. 0.] [0. 0. 0.] [0. 0. 0.] ... [0. 0. 0.] [0. 0. 0.] [0. 0. 0.]] ... [[0. 0. 0.] [0. 0. 0.] [0. 0. 0.] ... [0. 0. 0.] [0. 0. 0.] [0. 0. 0.]]] )1rNORMALr ZMULTIPLYrZ COLOURBURNrZ COLOURDODGErZREFLECTr ZOVERLAYr#Z DIFFERENCEr%ZLIGHTENr(ZDARKENr)ZSCREENr*Z SOFTLIGHTr1Z HARDLIGHTr3Z GRAINEXTRACTr4Z GRAINMERGEr5ZDIVIDErZHUErMZ SATURATIONrNZCOLOURrPZ LUMINOSITYrQZXORr0ZNEGATIONr'ZPINLIGHTr6Z VIVIDLIGHTr7Z EXCLUSIONr8)rrraZ blendLookupr r r blends8! rcrz Image.ImagezBlendType | tuple[str, ...]float)rrraopacityrc Csttt|d}tt|d}|dddddf}|dddddf|}||}|ddddddf} |ddddddf} tjttjttj t tj t i} || vrt ttjt| |||| | dddS||dddddf| ||dddddf| |dddddft| | |} |||} t ttjt| | fdddS)a,Blend layers using numpy array. Args: background (Image.Image): background layer foreground (Image.Image): foreground layer (must be same size as background) blendType (BlendType): The blendtype opacity (float): The opacity of the foreground image Returns: Image.Image: combined image ZRGBANrr)Za_minZa_max)r.rarrayconvertrZDESTINrZZDESTOUTr[ZSRCATOPr]ZDESTATOPr\rZ fromarrayr/rZdstackrc)rrrareZ npForegroundZ npBackgroundrRrSZ combinedAlphaZbackgroundColorZforegroundColorZ alphaFuncZcolorComponentsZalphaComponentr r r blendLayerssH  ri)r)0__doc__ __future__rr+numpyrZPILrZ blendtyperr rrrrr r!r#r%r'r(r)r*r0r1r3r4r5rr6r7r8r;rHrIrKrMrNrPrQrZr[r\r]r.r/rcrir r r r sX       $  D