a {d<@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}||}|dur|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 ur\fd d|D}t||}ddl m }ddlm}z4||tdsfdd|fD\}Wn|yt j Yn0|}}|jrDg}|jD]<}|}||kr||n ||q|t|jkrt|}nr|\}}\}d}|jrp|js||}t|dr|9|t||9}d}|s||}||rrdd|fD\}||}|dur|Sjr*tdr*|9}|ddSjrtjdjrttjddrtjd|}tjdd||f||fkdS)NcSs|jrtd|tjus6|tjus6|jdus6|jdur.number_evalrrc3s|]}|jdkVqdSrNargs.0innerr r%r& zMod.eval..cSsg|]}|jdqSrr)r,ir%r%r& r0zMod.eval..c3s|]}|jdkVqdSr(r)r+r.r%r&r/r0css|] }|jVqdS)N)r)r,tr%r%r&r/r0csg|]}|qSr%r%)r,x)clsr r%r&r4r0cSsg|]}|jdqSr1r)r2r%r%r&r4r0cs,g|]$}|jr$|tjur$|n|qSr%) is_Integerr rr2r.r%r&r4s)PolynomialError)gcdcsg|]}t|dddqS)F)clearfractionrr2)Gr%r&r4sFTcSsg|] }| qSr%r%r2r%r%r&r4r0)evaluate)rr*Zis_nonnegativeZis_nonpositiverappendallr rr8r rZsympy.polys.polyerrorsr9Zsympy.polys.polytoolsr:r Zis_AddcountlistZ as_coeff_MulZ is_RationalrZcould_extract_minus_signZis_FloatZis_MulZ _from_args)r7rr r'r!ZqinnerZboth_lZ non_mod_lZmod_largnetmodZnon_modjZprod_modZ prod_non_modZ prod_mod1r9r:ZpwasZqwasr*r3acpZcqokr"r%)r=r7r 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*rrKr%r%r&_eval_is_nonnegatives zMod._eval_is_nonnegativecCs|jdjrdSdSrM)r*rrNr%r%r&_eval_is_nonpositives zMod._eval_is_nonpositivecKs ddlm}|||||S)Nr)floor)Z#sympy.functions.elementary.integersrQ)rKrGbkwargsrQr%r%r&_eval_rewrite_as_floors zMod._eval_rewrite_as_floorN) __name__ __module__ __qualname____doc__rkind classmethodrJrLrOrPrTr%r%r%r&r s 5r N)addrZ exprtoolsrfunctionrrYrZlogicrrmulr Znumbersr Z singletonr r r%r%r%r&s