U Zʹj3@sldZddlmZddlmZefddZddZefd d Zd d Zd dZ ddZ efddZ ddZ dS)zP Generic Rules for SymPy This file assumes knowledge of Basic and little else. )sift)newcsfdd}|S)a Create a rule to remove identities. isid - fn :: x -> Bool --- whether or not this element is an identity. Examples ======== >>> from sympy.strategies import rm_id >>> from sympy import Basic, S >>> remove_zeros = rm_id(lambda x: x==0) >>> remove_zeros(Basic(S(1), S(0), S(2))) Basic(1, 2) >>> remove_zeros(Basic(S(0), S(0))) # If only identites then we keep one Basic(0) See Also: unpack cshtt|j}t|dkr |St|t|krR|jfddt|j|DS|j|jdSdS)z Remove identities rcSsg|]\}}|s|qSr).0argxrrR/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/strategies/rl.py $sz/rm_id..ident_remove..N)listmapargssumlen __class__zip)exprZidsisidrrr ident_removes zrm_id..ident_remover)rrrrrr rm_id s rcsfdd}|S)a6 Create a rule to conglomerate identical args. Examples ======== >>> from sympy.strategies import glom >>> from sympy import Add >>> from sympy.abc import x >>> key = lambda x: x.as_coeff_Mul()[1] >>> count = lambda x: x.as_coeff_Mul()[0] >>> combine = lambda cnt, arg: cnt * arg >>> rl = glom(key, count, combine) >>> rl(Add(x, -x, 3*x, 2, 3, evaluate=False)) 3*x + 5 Wait, how are key, count and combine supposed to work? >>> key(2*x) x >>> count(2*x) 2 >>> combine(2, x) 2*x csdt|j}fdd|D}fdd|D}t|t|jkr\tt|f|S|SdS)z2 Conglomerate together identical args x + x -> 2x cs i|]\}}|tt|qSr)rr )rkr )countrr Isz.glom..conglomerate..csg|]\}}||qSrr)rZmatZcnt)combinerr r Jsz.glom..conglomerate..N)rr itemssetrtype)rgroupscountsZnewargsrrkeyrr conglomerateFs  zglom..conglomerater)r!rrr"rr r glom+s r#csfdd}|S)z Create a rule to sort by a key function. Examples ======== >>> from sympy.strategies import sort >>> from sympy import Basic, S >>> sort_rl = sort(str) >>> sort_rl(Basic(S(3), S(1), S(2))) Basic(1, 2, 3) cs|jft|jdS)N)r!)rsortedr rr!rrr sort_rl`szsort..sort_rlr)r!rr'rr&r sortSs r(csfdd}|S)aW Turns an A containing Bs into a B of As where A, B are container types >>> from sympy.strategies import distribute >>> from sympy import Add, Mul, symbols >>> x, y = symbols('x,y') >>> dist = distribute(Mul, Add) >>> expr = Mul(2, x+y, evaluate=False) >>> expr 2*(x + y) >>> dist(expr) 2*x + 2*y cspt|jD]`\}}t|r |jd||j||j|dd}fdd|jDSq |S)Nrcsg|]}|fqSrr)rr)Afirsttailrr r ysz5distribute..distribute_rl..) enumerater isinstance)rirbr)B)r*r+r distribute_rlus  ."z!distribute..distribute_rlr)r)r1r2rr0r distributeesr3csfdd}|S)z Replace expressions exactly cs|kr S|SdS)Nrr%ar/rr subs_rlszsubs..subs_rlr)r5r/r6rr4r subs~sr7cCs t|jdkr|jdS|SdS)z Rule to unpack singleton args >>> from sympy.strategies import unpack >>> from sympy import Basic, S >>> unpack(Basic(S(2))) 2 rrN)rr r%rrr unpacks r8cCsH|j}g}|jD]&}|j|kr,||jq||q||jf|S)z9 Flatten T(a, b, T(c, d), T2(e)) to T(a, b, c, d, T2(e)) )rr extendappend)rrclsr rrrr flattens   r<cCs$|jr |S|jttt|jSdS)z Rebuild a SymPy tree. Explanation =========== This function recursively calls constructors in the expression tree. This forces canonicalization and removes ugliness introduced by the use of Basic.__new__ N)Zis_Atomfuncr r rebuildr r%rrr r>s r>N) __doc__Zsympy.utilities.iterablesrutilrrr#r(r3r7r8r<r>rrrr s   !(