Wj jdZddlmZmZddlmZddlZddlm Z ddfd Z d Z d Z d Z d ZdZdZy)zDThis submodule contains tools for working with numpy.poly1d objects.)divisionabsolute_import) combinationsN)iscloseFcy)NT)rs g/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/svgpathtools/polytools.pyr sdctj|}|r0|Dcgc]%}t|jds|j'}}|Dcgc] }||s |}}g}t t |dD]&\}\}}t||s|j|(t |Dcgc] \}}||vs |c}}Scc}wcc}wcc}}w)a{ Returns the roots of a polynomial with coefficients given in p. p[0] * x**n + p[1] * x**(n-1) + ... + p[n-1]*x + p[n] INPUT: p - Rank-1 array-like object of polynomial coefficients. realroots - a boolean. If true, only real roots will be returned and the condition function can be written assuming all roots are real. condition - a boolean-valued function. Only roots satisfying this will be returned. If realroots==True, these conditions should assume the roots are real. OUTPUT: A list containing the roots of the polynomial. NOTE: This uses np.isclose and np.rootsr)nprootsrimagreal enumeraterappend) p realroots conditionrr duplicatesidxr1r2s r polyrootsr s HHQKE!&=A'!&&!*<= .11QE .J"<q#9: Xb" 2r?   c ";&e, F,&#q:0EA, FF> . Gs"CC CC. C ;C c t|ddS)zReturns the real roots between 0 and 1 of the polynomial with coefficients given in p, p[0] * x**n + p[1] * x**(n-1) + ... + p[n-1]*x + p[n] p can also be a np.poly1d object. See polyroots for more information.Tc"d|cxkxrdkScS)Nrrr )tvals r r zpolyroots01..+s qD~A~r )rr)r)rs r polyroots01r!&s Q$2M NNr cZt|tjrt|tjsJ|tjdgk7sJ||dk7r||||z S||dk(r)t|j |j |St d)zLComputes the limit of the rational function (f/g)(t) as t approaches t0.rzLimit does not exist.) isinstancerpoly1drational_limitderiv ValueError)fgt0s r r%r%.s a # 1bii(@@ @  1#  uzuQrU{ 2!aggiB77011r c tj|jjS#t$r|jcYSwxYwN)rr$coeffsrAttributeErrorzs r rr;6yy'' vv (+AAc tj|jjS#t$r|jcYSwxYwr,)rr$r-rr.r/s r rrBr1r2cTtj|jjSz Deprecated.)rr$r-rpolys r poly_real_partr8I 99T[[%% &&r cTtj|jjSr5)rr$r-rr6s r poly_imag_partr;Nr9r )__doc__ __future__rr itertoolsrnumpyr misctoolsrrr!r%rrr8r;r r r rAsDJ1"!NG4O 2' 'r