gtdZddlmZddlmZmZddlmZddlm Z edZ edZ d#d Z d$d Z d%dZd&dZd'dZd(dZd)dZd$dZd'dZd*dZd Zed!d+d"ZdS),z& Generic SymPy-Independent Strategies ) annotations)CallableMapping)TypeVarstdout_S_Txreturnc|SNr s Q/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/strategies/core.pyidentityr  HruleCallable[[_T], _T]cdfd }|S)z0 Apply a rule repeatedly until it has no effect exprr r cT||}}||kr||}}||k|Srr)rnewoldrs r exhaustive_rlzexhaust..exhaustive_rls>4::tSSjjtCyy#CSjj rrr r r r)rrs` rexhaustrs) rCallable[[_S], _T]cidfd }|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. rr r r c@|vr|S|}||<|Srr)rresultcachers r memoized_rlzmemoize..memoized_rl%s1 5==; T$ZZF E$KMrrr r r r)rr$r#s` @rmemoizer&s6E rcondCallable[[_T], bool]cdfd }|S)z& Only apply rule if condition is true rr r c4|r |S|Srr)rr'rs rconditioned_rlz!condition..conditioned_rl3s& 4:: 4::  rrr)r'rr+s`` r conditionr,/s/ rrulescdfd }|S)zQ Compose a sequence of rules so that they apply to the expr sequentially rr r c(D] }||}|Srr)rrr-s rchain_rlzchain..chain_rl>s&  D4::DD rrr)r-r0s` rchainr1:s( OrNc(tfd}|S)z? Print out before and after expressions each time rule is used Nc|d}|i|}||kr9djzd|d|d|S)Nrz Rule: %s zIn: z Out: z )write__name__)argskwargsrr"filers rdebug_rlzdebug..debug_rlJshAwt&v&& T>> JJ|dm3 4 4 4 JJJ444@ A A A rr)rr8r9s`` rdebugr:Es3 | OrCallable[[_T], _T | None]cdfd }|S)z+ Return original expr if rule returns None rr r c&|}||S|Srr)rr"rs r null_safe_rlznull_safe..null_safe_rlVsd >K rrr)rr>s` r null_safer?Ts) rcdfd }|S)z/ Return original expr if rule raises exception rr r c6 |S#$r|cYSwxYwrr)r exceptionrs rtry_rlztryit..try_rl`s: 4::    KKK s rr)rrBrCs`` rtryitrD^s. Mrcdfd }|S)z3 Try each of the rules until one works. Then stop. rr r c<D]}||}||kr|cS|Srr)rrlr"r-s r do_one_rlzdo_one..do_one_rljs;  BRXXF~~  rrr)r-rHs` rdo_onerIhs) rkeyruledictMapping[_T, Callable[[_S], _S]]Callable[[_S], _S]cdfd }|S)zA Select a rule based on the result of key called on the function rr r cb|t}||Sr)getr)rrGrJrKs r switch_rlzswitch..switch_rlxs+ \\##d))X . .r$xxr)rr r r r)rJrKrQs`` rswitchrRss/  rc|Srrrs r _identityrTrr) objectivecdfd }|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 rr r r c@tfdDS)Nc&g|] }|Srr).0rrs r z-minimize..minrule..s!1114DDJJ111r)rJ)min)rrUr-s`rminrulezminimize..minrules,11115111yAAAArr%r)rUr-r\s`` rminimizer]s3"BBBBBBB Nr)r r r r )rrr r)rrr r)r'r(rrr r)r-rr rr)rr;r r)rJrrKrLr rM)r-rr r)__doc__ __future__rcollections.abcrrtypingrsysrr r rrr&r,r1r:r?rDrIrRrTr]rrrrcsy,,""""""-------- WT]] WT]]    *       r