o GZht@sdZddlmZddlmZmZddlmZddlm Z edZ edZ d/d d Z d0ddZ d1ddZd2ddZd3ddZd4ddZd5ddZd0d d!Zd3d"d#Zd6d(d)Zd*d+Zed,d7d-d.ZdS)8z& Generic SymPy-Independent Strategies ) annotations)CallableMapping)TypeVarstdout_S_TxreturncC|SNr rrD/var/www/auris/lib/python3.10/site-packages/sympy/strategies/core.pyidentity rruleCallable[[_T], _T]cdfdd }|S)z0 Apply a rule repeatedly until it has no effect exprr r cs0||}}||kr||}}||ks |Sr r)rnewoldrrr exhaustive_rls zexhaust..exhaustive_rlNrr r r r)rrrrrexhaustrCallable[[_S], _T]csidfdd }|S) zMemoized version of a rule Notes ===== This cache can grow infinitely, so it is not recommended to use this than ``functools.lru_cache`` unless you need very heavy computation. rrr r cs$|vr|S|}||<|Sr rrresultcacherrr memoized_rl%s zmemoize..memoized_rlNrrr r r)rr#rr!rmemoizes r%condCallable[[_T], bool]cdfdd }|S)z& Only apply rule if condition is true rr r cs|r|S|Sr rrr&rrrconditioned_rl3sz!condition..conditioned_rlNrr)r&rr+rr*r condition/sr,rulescr)zQ Compose a sequence of rules so that they apply to the expr sequentially rr r csD]}||}q|Sr r)rrr-rrchain_rl>s zchain..chain_rlNrr)r-r/rr.rchain:sr0Ncsdurtfdd}|S)z? Print out before and after expressions each time rule is used NcsD|d}|i|}||kr djd||f|S)Nrz Rule: %s zIn: %s Out: %s )write__name__)argskwargsrr filerrrdebug_rlJs zdebug..debug_rlr)rr6r7rr5rdebugEsr8Callable[[_T], _T | None]cr)z+ Return original expr if rule returns None rr r cs|}|dur |S|Sr rrrrr null_safe_rlVsznull_safe..null_safe_rlNrr)rr:rrr null_safeTrr;cr()z/ Return original expr if rule raises exception rr r cs"z|WSy|YSwr rr) exceptionrrrtry_rl`s   ztryit..try_rlNrr)rr=r>rr<rtryit^sr?cr)z3 Try each of the rules until one works. Then stop. rr r cs&D]}||}||kr|Sq|Sr r)rrlr r.rr do_one_rljs zdo_one..do_one_rlNrr)r-rArr.rdo_onehsrBkeyruledictMapping[_T, Callable[[_S], _S]]Callable[[_S], _S]cr()zA Select a rule based on the result of key called on the function rrr cs|t}||Sr )getr)rr@rCrDrr switch_rlxszswitch..switch_rlN)rrr rr)rCrDrIrrHrswitchssrJcCr r rrrrr _identityrrK) objectivecsdfdd }|S) a- Select result of rules that minimizes objective >>> from sympy.strategies import minimize >>> inc = lambda x: x + 1 >>> dec = lambda x: x - 1 >>> rl = minimize(inc, dec) >>> rl(4) 3 >>> rl = minimize(inc, dec, objective=lambda x: -x) # maximize >>> rl(4) 5 rrr r cstfddDdS)Ncsg|]}|qSrr).0rr)rr sz-minimize..minrule..)rC)minr)rLr-r)rminruleszminimize..minruleNr$r)rLr-rQrrPrminimizesrR)r r r r )rrr r)rrr r)r&r'rrr r)r-rr rr )rr9r r)rCrrDrEr rF)r-rr r)__doc__ __future__rcollections.abcrrtypingrsysrrr rrr%r,r0r8r;r?rBrJrKrRrrrrs(