U 9%e<@sxddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZGd d d eZd S) )Add gcd_terms)Function) NumberKind) fuzzy_and fuzzy_not)Mul) equal_valued)Sc@s@eZdZdZeZeddZddZddZ dd Z d d Z d S) ModaRepresents a modulo operation on symbolic expressions. Parameters ========== p : Expr Dividend. q : Expr Divisor. Notes ===== The convention used is the same as Python's: the remainder always has the same sign as the divisor. Examples ======== >>> from sympy.abc import x, y >>> x**2 % y Mod(x**2, y) >>> _.subs({x: 5, y: 6}) 1 csdd}||}|dk r|St|rd|jd}|dkrN|jdS||jr`|Snt| r| jd}|dkr| jd S||jr|Snt|tr4ggf}\}}|jD]} |t| | q|rhtfdd|Drht|tdd|D} | Sn4t|trhggf}\}}|jD]} |t| | qV|r6tfd d|Dr6td d|jDr6jr6fd d|D}g} g} |D].} t| r| | jdn | | qt| }t| }td d|D}||} || Sj r\t j k r\fd d|D}t||}ddl m }ddlm}z4||tdsfdd|fD\}Wn|k rt j YnX|}}|jrFg}|jD]<}|}||kr||n ||q|t|jkrt|}nr|\}}\}d}|jrr|js||}t|dr|9|t||9}d}|s||}||rrdd|fD\}||}|dk r|Sjr,tdr,|9}|ddSjrvjdjrvtjddrvjd|}tjdd||f||fkdS)NcSs|jrtd|tjks6|tjks6|jdks6|jdkr.number_evalrrc3s|]}|jdkVqdSrNargs.0innerr r%r& szMod.eval..cSsg|]}|jdqSrr)r,ir%r%r& szMod.eval..c3s|]}|jdkVqdSr(r)r+r.r%r&r/scss|] }|jVqdS)N)r)r,tr%r%r&r/scsg|]}|qSr%r%)r,x)clsr r%r&r3scSsg|]}|jdqSr0r)r1r%r%r&r3scs,g|]$}|jr$|tjk r$|n|qSr%) is_Integerr rr1r.r%r&r3s)PolynomialError)gcdcsg|]}t|dddqS)F)clearfractionrr1)Gr%r&r3sFTcSsg|] }| qSr%r%r1r%r%r&r3s)evaluate)rr*Zis_nonnegativeZis_nonpositiverappendallr rr7r rZsympy.polys.polyerrorsr8Zsympy.polys.polytoolsr9r Zis_AddcountlistZ as_coeff_MulZ is_RationalrZcould_extract_minus_signZis_FloatZis_MulZ _from_args)r6rr r'r!ZqinnerZboth_lZ non_mod_lZmod_largnetmodZnon_modjZprod_modZ prod_non_modZ prod_mod1r8r9ZpwasZqwasr*r2acpZcqokr"r%)r<r6r r&eval*s;           <                   (zMod.evalcCs*|j\}}t|j|jt|jgr&dSdS)NT)r*rrrr)selfrr r%r%r&_eval_is_integers zMod._eval_is_integercCs|jdjrdSdSNrT)r*rrJr%r%r&_eval_is_nonnegatives zMod._eval_is_nonnegativecCs|jdjrdSdSrL)r*rrMr%r%r&_eval_is_nonpositives zMod._eval_is_nonpositivecKs ddlm}|||||S)Nr)floor)Z#sympy.functions.elementary.integersrP)rJrFbkwargsrPr%r%r&_eval_rewrite_as_floors zMod._eval_rewrite_as_floorN) __name__ __module__ __qualname____doc__rkind classmethodrIrKrNrOrSr%r%r%r&r s 5r N)addrZ exprtoolsrfunctionrrXrZlogicrrmulr numbersr Z singletonr r r%r%r%r&s