U ڲg,@s4ddlmZddlZddlmZddlmZmZddlm Z m Z ddl m Z Gdd d Z d d d d d Zd d d d dZd dd dddZd dd dddZd dd dddZd dd dddZd dd dddZeeeeeeedZifddd d d!d"d#Zifddd d d!d$d%Zifddd d d&d'd(ZdS))) annotationsN)CodeType)AnyCallable)Image _imagingmath) deprecatec@seZdZdZddddZddddd ZdRd dd d ddddZddddZddddZddddZ ddddZ dddddZ dddddZ dddd d!Z dddd"d#Zdddd$d%Zdddd&d'Zdddd(d)Zdddd*d+Zdddd,d-Zdddd.d/Zdddd0d1Zdddd2d3Zddd4d5Zdddd6d7Zdddd8d9Zdddd:d;Zddddd?Zdddd@dAZddddBdCZddddDdEZdFdGZ dHdIZ!ddddJdKZ"ddddLdMZ#ddddNdOZ$ddddPdQZ%d S)S_Operandz4Wraps an image operand, providing standard operatorsz Image.ImageimcCs ||_dSNr )selfr r1/tmp/pip-unpacked-wheel-xela2va7/PIL/ImageMath.py__init__sz_Operand.__init__z_Operand | float)im1returncCst|trL|jjdkr"|jdS|jjdkr4|jSd|jj}t|n>t|ttfrx|jjdkrxt d|jj |St d|jj |SdS)N)1LI)rFzunsupported mode: )rrrr) isinstancer r modeconvert ValueErrorintfloatrnewsize)rrmsgrrrZ__fixup!s     z_Operand.__fixupNstr_Operand | float | Nonez str | None)oprim2rrc Cs||}|dkrt|p|j|jd}|ztt|d|j}Wn8tk r}zd|d}t ||W5d}~XYnXt ||j j |j j nF||} |j| jkr|jdkr| d}| jdkr| d} |j| jkrJt|jd| jdt|jd| jdf} |j| kr0|d| }| j| krJ| d| } t|pX|j|jd}|| ztt|d|j}Wn:tk r}zd|d}t ||W5d}~XYnXt||j j |j j | j j t|S)N_zbad operand type for ''rrr)rr)_Operand__fixuprrrrloadgetattrrAttributeError TypeErrorZunopr idrminZcropZbinopr ) rr#rr$rZim_1outer Zim_2rrrrapply3sD           z_Operand.applybool)rcCs|jdk Sr )r Zgetbboxrrrr__bool__dsz_Operand.__bool__cCs |d|S)Nabsr0r2rrr__abs__hsz_Operand.__abs__cCs|Sr rr2rrr__pos__ksz_Operand.__pos__cCs |d|S)Nnegr5r2rrr__neg__nsz_Operand.__neg__)otherrcCs|d||SNaddr5rr:rrr__add__rsz_Operand.__add__cCs|d||Sr;r5r=rrr__radd__usz_Operand.__radd__cCs|d||SNsubr5r=rrr__sub__xsz_Operand.__sub__cCs|d||Sr@r5r=rrr__rsub__{sz_Operand.__rsub__cCs|d||SNmulr5r=rrr__mul__~sz_Operand.__mul__cCs|d||SrDr5r=rrr__rmul__sz_Operand.__rmul__cCs|d||SNdivr5r=rrr __truediv__sz_Operand.__truediv__cCs|d||SrHr5r=rrr __rtruediv__sz_Operand.__rtruediv__cCs|d||SNmodr5r=rrr__mod__sz_Operand.__mod__cCs|d||SrLr5r=rrr__rmod__sz_Operand.__rmod__cCs|d||SNpowr5r=rrr__pow__sz_Operand.__pow__cCs|d||SrPr5r=rrr__rpow__sz_Operand.__rpow__cCs |d|S)Ninvertr5r2rrr __invert__sz_Operand.__invert__cCs|d||SNandr5r=rrr__and__sz_Operand.__and__cCs|d||SrVr5r=rrr__rand__sz_Operand.__rand__cCs|d||SNorr5r=rrr__or__sz_Operand.__or__cCs|d||SrZr5r=rrr__ror__sz_Operand.__ror__cCs|d||SNxorr5r=rrr__xor__sz_Operand.__xor__cCs|d||Sr^r5r=rrr__rxor__sz_Operand.__rxor__cCs|d||S)Nlshiftr5r=rrr __lshift__sz_Operand.__lshift__cCs|d||S)Nrshiftr5r=rrr __rshift__sz_Operand.__rshift__cCs|d||S)Neqr5r=rrr__eq__sz_Operand.__eq__cCs|d||S)Nner5r=rrr__ne__sz_Operand.__ne__cCs|d||S)Nltr5r=rrr__lt__sz_Operand.__lt__cCs|d||S)Nler5r=rrr__le__sz_Operand.__le__cCs|d||S)Ngtr5r=rrr__gt__sz_Operand.__gt__cCs|d||S)Nger5r=rrr__ge__sz_Operand.__ge__)NN)&__name__ __module__ __qualname____doc__rr'r0r3r6r7r9r>r?rBrCrFrGrJrKrNrOrRrSrUrXrYr\r]r`rarcrergrirkrmrorqrrrrr sJ1r )rrcCst|jdS)Nrr r rr2rrr imagemath_intsrwcCst|jdS)Nrrvr2rrrimagemath_floatsrxr")rr:rcCs|jd||ddS)Nrfrrr5r=rrrimagemath_equalsrzcCs|jd||ddS)Nrhrryr5r=rrrimagemath_notequalsr{cCs|d||S)Nr-r5r=rrr imagemath_minsr|cCs|d||S)Nmaxr5r=rrr imagemath_maxsr~r!)rrrcCst|j|Sr rv)rrrrrimagemath_convertsr)rrequalZnotequalr-r}rzCallable[[dict[str, Any]], Any]zdict[str, Any]r) expressionoptionskwrcKsrt}|||||D]\}}t|dr$t|||<q$||}z|jWStk rl|YSXdS)a 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 options: Values to add to the function's dictionary. You can either use a dictionary, or one or more keyword arguments. :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. r N)opscopyupdateitemshasattrr r r*)rrrargskvr.rrr lambda_evals   rc stt|t|D]*}d|ks6tt|r d|d}t|q ||D]\}}t|drht ||<qht |ddddd fd d t |d d t ii}z|j WStk r|YSXdS)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 options: Values to add to the evaluation context. You can either use a dictionary, or one or more keyword arguments. :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 allowedr zevalrNone)codercsX|jD]}t|tkr|q|jD](}|kr*|dkr*d|d}t|q*dS)Nr4r&r) co_conststypeco_namesr)rconstnamer rZ compiled_codescanrrr9s    zunsafe_eval..scanZ __builtinsr4N)rrlistkeysrbuiltinsrrrr compilerr4r r*)rrrrr rr.rrr unsafe_evals$       r)r_dictrrcKstdddt||f|S)a1 Evaluates an image expression. Deprecated. Use lambda_eval() or unsafe_eval() instead. :param expression: A string containing a Python-style expression. :param _dict: Values to add to the evaluation context. You can either use a dictionary, or one or more keyword arguments. :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. .. deprecated:: 10.3.0 zImageMath.eval z.ImageMath.lambda_eval or ImageMath.unsafe_eval)r r)rrrrrrrKs r) __future__rrtypesrtypingrrrrZ _deprecater r rwrxrzr{r|r~rrrrrrrrrs8   - %;