o Ԧ/f,@sddlmZddlZddlmZddlmZmZddlm Z m Z ddl m Z Gdd d Z d,d d Zd,ddZd-ddZd-ddZd-ddZd-ddZd.ddZeeeeeeedZifd/d%d&Zifd0d'd(Zifd1d*d+ZdS)2) annotationsN)CodeType)AnyCallable)Image _imagingmath) deprecatec@sfeZdZdZdTddZdUd 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@dAZdZdBdCZdZdDdEZdZdFdGZdHdIZ dJdKZ!dZdLdMZ"dZdNdOZ#dZdPdQZ$dZdRdSZ%d S)[_Operandz4Wraps an image operand, providing standard operatorsim Image.ImagecCs ||_dSN)r )selfr r-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)rrmsgrrrZ__fixup!s    z_Operand.__fixupNopstrim2_Operand | float | Noner str | Nonec Cs||}|durKt|p|j|jd}|z tt|d|j}Wnty;}z d|d}t ||d}~wwt ||j j |j j t|S||} |j| jkrj|jdkr`| d}| jdkrj| 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||j j |j j | j j t|S)N_zbad operand type for ''rrr)rr)_Operand__fixuprrrr loadgetattrrAttributeError TypeErrorZunopr idrminZcropZbinopr ) rr"rr$rZim_1outer!Zim_2r rrrapply3sN             z_Operand.applyboolcCs|jduSr )r Zgetbboxrrrr__bool__dsz_Operand.__bool__cC |d|S)Nabsr2r4rrr__abs__h z_Operand.__abs__cCs|Sr rr4rrr__pos__ksz_Operand.__pos__cCr6)Nnegr8r4rrr__neg__nr:z_Operand.__neg__othercC|d||SNaddr8rr>rrr__add__rz_Operand.__add__cC|d||Sr@r8rBrrr__radd__urDz_Operand.__radd__cCr?Nsubr8rBrrr__sub__xrDz_Operand.__sub__cCrErGr8rBrrr__rsub__{rDz_Operand.__rsub__cCr?Nmulr8rBrrr__mul__~rDz_Operand.__mul__cCrErKr8rBrrr__rmul__rDz_Operand.__rmul__cCr?NZdivr8rBrrr __truediv__rDz_Operand.__truediv__cCrErOr8rBrrr __rtruediv__rDz_Operand.__rtruediv__cCr?Nmodr8rBrrr__mod__rDz_Operand.__mod__cCrErRr8rBrrr__rmod__rDz_Operand.__rmod__cCr?Npowr8rBrrr__pow__rDz_Operand.__pow__cCrErVr8rBrrr__rpow__rDz_Operand.__rpow__cCr6)Ninvertr8r4rrr __invert__r:z_Operand.__invert__cCr?Nandr8rBrrr__and__rDz_Operand.__and__cCrEr\r8rBrrr__rand__rDz_Operand.__rand__cCr?Norr8rBrrr__or__rDz_Operand.__or__cCrEr`r8rBrrr__ror__rDz_Operand.__ror__cCr?Nxorr8rBrrr__xor__rDz_Operand.__xor__cCrErdr8rBrrr__rxor__rDz_Operand.__rxor__cCr?)Nlshiftr8rBrrr __lshift__rDz_Operand.__lshift__cCr?)Nrshiftr8rBrrr __rshift__rDz_Operand.__rshift__cCr?)Neqr8rBrrr__eq__rDz_Operand.__eq__cCr?)Nner8rBrrr__ne__rDz_Operand.__ne__cCr?)Nltr8rBrrr__lt__rDz_Operand.__lt__cCr?)Nler8rBrrr__le__rDz_Operand.__le__cCr?)Ngtr8rBrrr__gt__rDz_Operand.__gt__cCr?)Nger8rBrrr__ge__rDz_Operand.__ge__)r r )rrrr )NN) r"r#rrr$r%rr&rr )rr3)rr )r>rrr )&__name__ __module__ __qualname____doc__rr)r2r5r9r;r=rCrFrIrJrMrNrPrQrTrUrXrYr[r^r_rbrcrfrgrirkrmrorqrsrurwrrrrr sL   1                           r rrcCt|jdS)Nrr r rr4rrr imagemath_intr~cCr|)Nrr}r4rrrimagemath_floatrrr>r%cC|jd||ddS)Nrlrrr8rBrrrimagemath_equalrcCr)Nrnrrr8rBrrrimagemath_notequalrrcCr?)Nr/r8rBrrr imagemath_minrDrcCr?)Nmaxr8rBrrr imagemath_maxrDrrr#cCst|j|Sr r})rrrrrimagemath_convertrr)rrZequalZnotequalr/rr expressionCallable[[dict[str, Any]], Any]optionsdict[str, Any]kwrcKslt}|||||D]\}}t|dr!t|||<q||}z|jWSty5|YSw)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 )opscopyupdateitemshasattrr r r,)rrrargskvr0rrr lambda_evals     rc stt|t|D]}d|vstt|r%d|d}t|q||D]\}}t|drCt ||<q4t |dddfd d t |d dt ii}z|j WStyq|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 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 zevalcoderrNonecsX|jD]}t|tur|q|jD]}|vr)|dkr)d|d}t|qdS)Nr7r(r) co_conststypeco_namesr)rconstnamer!rZ compiled_codescanrrr9s   zunsafe_eval..scanZ __builtinsr7N)rrrr)rrlistkeysrbuiltinsrrrr compilerr7r r,)rrrrr!rr0rrr unsafe_evals*        r_dictcKstdddt||fi|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)rr rr )rr r>r%rr )rr rr#rr )rrrrrrrr)rr#rrrrrr)rr#rrrrrr)Z __future__rrtypesrtypingrrrrZ _deprecater r r~rrrrrrrrrrrrrrs:    -       % ;