U ;qLeR@sPdZddlmZmZmZddlmZddlmZddl m Z ddZ dd Z d S) zS Optimizations of the expression tree representation for better CSE opportunities. )AddBasicMul)S)default_sort_key)preorder_traversalcCsdd|tD}i}t}|D]0}| }|jr>||q"ttj|g||<q"| |}t |t ri}t |tt dD]F}||krq~||kr||||<q~|r~ttjtj| g||<q~| |}|S)zD Replace y - x with -(x - y) if -1 can be extracted from y - x. cSsg|]}|r|qS)could_extract_minus_sign).0arrn/home/p21-0144/sympy/latex2sympy2solve-back-end/sympyEq/lib/python3.8/site-packages/sympy/simplify/cse_opts.py szsub_pre..)key)atomsrsetis_Muladdr _from_argsr NegativeOnexreplace isinstancersortedrr One)eZaddsrepsignorer nanegsrrr sub_pre s*    rc Cs|g}t|D]N}t|tr |jdtjkr |jdtjkr ||t|jdd fq |D]\}}| ||i}q`|S)z Replace 1*-1*x with -x. rN) rrrargsrrrappendrr)r replacementsnode replacementrrr sub_post)s  " r&N) __doc__ sympy.corerrrZsympy.core.singletonrsympy.core.sortingrsympy.core.traversalrrr&rrrr s