o 8Va} @shdZddZddZddZddZd d Zdd d ZddZddZddZ ddZ edddZ d S)z& Generic SymPy-Independent Strategies cCs|SN)xrr7/usr/lib/python3/dist-packages/sympy/strategies/core.pysrcfdd}|S)z0 Apply a rule repeatedly until it has no effect cs0||}}||kr||}}||ks |Srr)exprnewoldrulerr exhaustive_rls zexhaust..exhaustive_rlr)r r rr rexhausts r csifdd}|S)z Memoized version of a rule cs$|vr|S|}||<|Srrrresultcacher rr memoized_rls zmemoize..memoized_rlr)r rrrrmemoizesrcfdd}|S)z& Only apply rule if condition is true cs|r|S|Srrrcondr rrconditioned_rlsz!condition..conditioned_rlr)rr rrrr conditionrcr)zQ Compose a sequence of rules so that they apply to the expr sequentially csD]}||}q|Srr)rr rulesrrchain_rl's zchain..chain_rlr)rrrrrchain#s rNcs*dur ddlm}|fdd}|S)z? Print out before and after expressions each time rule is used N)stdoutcsD|d}|i|}||kr djd||f|S)Nrz Rule: %s zIn: %s Out: %s )write__name__)argskwargsrrfiler rrdebug_rl2s zdebug..debug_rl)sysr )r r&r r'rr%rdebug-s  r)cr)z+ Return original expr if rule returns None cs|}|dur |S|Srrrr rr null_safe_rl=sznull_safe..null_safe_rlr)r r*rr r null_safe; r+cr)z/ Return original expr if rule raises exception cs"z|WSy|YSwrrr exceptionr rrtry_rlGs   ztryit..try_rlr)r r.r/rr-rtryitErr0cr)z3 Try each of the rules until one works. Then stop. cs&D]}||}||kr|Sq|Srr)rrlrrrr do_one_rlPs zdo_one..do_one_rlr)rr2rrrdo_oneNr,r3cr)zA Select a rule based on the result of key called on the function cs|t}||Sr)getidentity)rr1keyruledictrr switch_rlZszswitch..switch_rlr)r7r8r9rr6rswitchXsr:) objectivecr)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 cstfddDdS)Ncsg|]}|qSrr).0r rrr osz-minimize..minrule..)r7)minrr;rrrminrulenszminimize..minruler)r;rr@rr?rminimize_srAr) __doc__r5r rrrr)r+r0r3r:rArrrrs