o i(@sddlmZddlZddlmZmZdZer(ddlmZddl m Z ddl m Z Gd d d Z d(d dZd(ddZd)ddZd)ddZd)ddZd)ddZd*ddZeeeeeeedZd+d$d%Zd,d&d'ZdS)-) annotationsN)Image _imagingmathF)Callable)CodeType)Anyc@sjeZdZdZdTddZdUd d Z dVdWddZdXddZdYddZdYddZ dYddZ dZddZ dZd d!Z dZd"d#Z dZd$d%ZdZd&d'ZdZd(d)ZdZd*d+ZdZd,d-ZdZd.d/ZdZd0d1ZdZd2d3ZdZd4d5ZdYd6d7ZdZd8d9ZdZd:d;ZdZdd?ZdZd@dAZdZdBdCZdZdDdEZdZdFdGZdZdHdIZ dZdJdKZ!dZdLdMZ"dZdNdOZ#dZdPdQZ$dZdRdSZ%d S)[_Operandz4Wraps an image operand, providing standard operatorsim Image.ImagecCs ||_dSN)r )selfr rE/home/air/sos/back/venv/lib/python3.10/site-packages/PIL/ImageMath.py__init__!s z_Operand.__init__im1_Operand | floatreturncCst|tr%|jjdvr|jdS|jjdvr|jSd|jj}t|t|ttfr;|jjdvr;t d|jj |St d|jj |S)N)1LI)rFzunsupported mode: )rrrr) isinstancer r modeconvert ValueErrorintfloatrnewsize)r rmsgrrr__fixup$s    z_Operand.__fixupNopstrim2_Operand | float | Noner str | Nonec Cs||}|durHt|p|j|jd}z tt|d|j}Wnty8}z d|d}t||d}~wwt || | t|S||} |j| jkrg|jdkr]| d}| jdkrg| d} |j| jkrt |jd| jdt |jd| jdf} |j| kr| d| }| j| kr| d| } t|p|j|jd}z tt|d|j}Wnty}z d|d}t||d}~wwt|| | | t|S)N_zbad operand type for ''rrr)rr)_Operand__fixuprrrrgetattrrAttributeError TypeErrorunopgetimrmincropbinopr ) r r"rr$rim_1outer im_2rrrrapply6sH             z_Operand.applyboolcCs|jduSr )r getbboxr rrr__bool__dsz_Operand.__bool__cC |d|S)Nabsr6r9rrr__abs__h z_Operand.__abs__cCs|Sr rr9rrr__pos__ksz_Operand.__pos__cCr;)Nnegr=r9rrr__neg__nr?z_Operand.__neg__othercC|d||SNaddr=r rCrrr__add__rz_Operand.__add__cC|d||SrEr=rGrrr__radd__urIz_Operand.__radd__cCrDNsubr=rGrrr__sub__xrIz_Operand.__sub__cCrJrLr=rGrrr__rsub__{rIz_Operand.__rsub__cCrDNmulr=rGrrr__mul__~rIz_Operand.__mul__cCrJrPr=rGrrr__rmul__rIz_Operand.__rmul__cCrDNdivr=rGrrr __truediv__rIz_Operand.__truediv__cCrJrTr=rGrrr __rtruediv__rIz_Operand.__rtruediv__cCrDNmodr=rGrrr__mod__rIz_Operand.__mod__cCrJrXr=rGrrr__rmod__rIz_Operand.__rmod__cCrDNpowr=rGrrr__pow__rIz_Operand.__pow__cCrJr\r=rGrrr__rpow__rIz_Operand.__rpow__cCr;)Ninvertr=r9rrr __invert__r?z_Operand.__invert__cCrDNandr=rGrrr__and__rIz_Operand.__and__cCrJrbr=rGrrr__rand__rIz_Operand.__rand__cCrDNorr=rGrrr__or__rIz_Operand.__or__cCrJrfr=rGrrr__ror__rIz_Operand.__ror__cCrDNxorr=rGrrr__xor__rIz_Operand.__xor__cCrJrjr=rGrrr__rxor__rIz_Operand.__rxor__cCrD)Nlshiftr=rGrrr __lshift__rIz_Operand.__lshift__cCrD)Nrshiftr=rGrrr __rshift__rIz_Operand.__rshift__cCrD)Neqr=rGrrr__eq__rIz_Operand.__eq__cCrD)Nner=rGrrr__ne__rIz_Operand.__ne__cCrD)Nltr=rGrrr__lt__rIz_Operand.__lt__cCrD)Nler=rGrrr__le__rIz_Operand.__le__cCrD)Ngtr=rGrrr__gt__rIz_Operand.__gt__cCrD)Nger=rGrrr__ge__rIz_Operand.__ge__)r r )rrrr )NN) r"r#rrr$r%rr&rr )rr7)rr )rCrrr )&__name__ __module__ __qualname____doc__rr)r6r:r>r@rBrHrKrNrOrRrSrVrWrZr[r^r_rardrerhrirlrmrorqrsrurwryr{r}rrrrr sL   .                             r r rcCt|jdS)Nrr r rr9rrr imagemath_intrcCr)Nrrr9rrrimagemath_floatrrrCr%cC|jd||ddS)Nrrrrr=rGrrrimagemath_equalrcCr)Nrtrrr=rGrrrimagemath_notequalrrcCrD)Nr/r=rGrrr imagemath_minrIrcCrD)Nmaxr=rGrrr imagemath_maxrIrrr#cCst|j|Sr r)r rrrrimagemath_convertrr)rrequalnotequalr/rr expressionCallable[[dict[str, Any]], Any]kwrcKsdt}|||D]\}}t|tjrt|||<q ||}z|jWSty1|YSw)a1 Returns the result of an image function. :py:mod:`~PIL.ImageMath` only supports single-layer images. To process multi-band images, use the :py:meth:`~PIL.Image.Image.split` method or :py:func:`~PIL.Image.merge` function. :param expression: A function that receives a dictionary. :param **kw: Values to add to the function's dictionary. :return: The expression result. This is usually an image object, but can also be an integer, a floating point value, or a pixel tuple, depending on the expression. ) opscopyupdateitemsrrr r r+)rrargskvr3rrr lambda_evals    rc st|D]}d|vstt|rd|d}t|q|D]\}}t|tjr5t ||<q%t |dddfd d t |d d t ii}z|j WStyc|YSw)a- Evaluates an image expression. This uses Python's ``eval()`` function to process the expression string, and carries the security risks of doing so. It is not recommended to process expressions without considering this. :py:meth:`~lambda_eval` is a more secure alternative. :py:mod:`~PIL.ImageMath` only supports single-layer images. To process multi-band images, use the :py:meth:`~PIL.Image.Image.split` method or :py:func:`~PIL.Image.merge` function. :param expression: A string containing a Python-style expression. :param **kw: Values to add to the evaluation context. :return: The evaluated expression. This is usually an image object, but can also be an integer, a floating point value, or a pixel tuple, depending on the expression. __r( ' not allowedzevalcoderrNonecsX|jD]}t|tur|q|jD]}|vr)|dkr)d|d}t|qdS)Nr<r(r) co_conststypeco_namesr)rconstnamer r compiled_codescanrrr+s   zunsafe_eval..scan __builtinsr<N)rrrr)rrhasattrbuiltinsrrrrrr compilerr<r r+)rrrr rr3rrr unsafe_eval s(       r)r r rr )r r rCr%rr )r r rr#rr )rrrrrr)rr#rrrr) __future__rrrr TYPE_CHECKINGcollections.abcrtypesrtypingrr rrrrrrrrrrrrrrs6     *