a SG5d 2@sdZddlZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZdd lmZGd d d eZd dddZddZdddZddZdS)z,Solvers of systems of polynomial equations. N)S)default_sort_key)Polygroebnerroots)parallel_poly_from_expr)ComputationFailedPolificationFailedCoercionFailedrcollect) postfixes) filldedentc@seZdZdZdS) SolveFailedz.Raised when solver's conditions were not met. N)__name__ __module__ __qualname____doc__rrQ/var/www/html/django/DPS/env/lib/python3.9/site-packages/sympy/solvers/polysys.pyrsrFstrictc Oszt|g|Ri|\}}Wn4tyR}ztdt||WYd}~n d}~00t|t|jkrrdkrnnH|\}}tdd||Drzt|||WStyYn0t |||dS)a9 Solve a system of polynomial equations. Parameters ========== seq: a list/tuple/set Listing all the equations that are needed to be solved gens: generators generators of the equations in seq for which we want the solutions strict: a boolean (default is False) if strict is True, NotImplementedError will be raised if the solution is known to be incomplete (which can occur if not all solutions are expressible in radicals) args: Keyword arguments Special options for solving the equations. Returns ======= List[Tuple] A List of tuples. Solutions for symbols that satisfy the equations listed in seq Examples ======== >>> from sympy import solve_poly_system >>> from sympy.abc import x, y >>> solve_poly_system([x*y - 2*y, 2*y**2 - x**2], x, y) [(0, 0), (2, -sqrt(2)), (2, sqrt(2))] >>> solve_poly_system([x**5 - x + y**3, y**2 - 1], x, y, strict=True) Traceback (most recent call last): ... UnsolvableFactorError solve_poly_systemNcss|]}|dkVqdS)rNr).0irrr Fz$solve_poly_system..r) rr rlengensall degree_listsolve_biquadraticr solve_generic) seqrrargspolysoptexcfgrrrrs+&" rc st||g}t|dkr&|djr&dSt|dkr6t|j\}|\}}||jsXtt||dd}fddt|D}| d }t t|}fd dt ||D} t | td S) aSolve a system of two bivariate quadratic polynomial equations. Parameters ========== f: a single Expr or Poly First equation g: a single Expr or Poly Second Equation opt: an Options object For specifying keyword arguments and generators Returns ======= List[Tuple] A List of tuples. Solutions for symbols that satisfy the equations listed in seq. Examples ======== >>> from sympy import Options, Poly >>> from sympy.abc import x, y >>> from sympy.solvers.polysys import solve_biquadratic >>> NewOption = Options((x, y), {'domain': 'ZZ'}) >>> a = Poly(y**2 - 4 + x, y, x, domain='ZZ') >>> b = Poly(y*2 + 3*x - 7, y, x, domain='ZZ') >>> solve_biquadratic(a, b, NewOption) [(1/3, 3), (41/27, 11/9)] >>> a = Poly(y + x**2 - 3, y, x, domain='ZZ') >>> b = Poly(-y + x - 4, y, x, domain='ZZ') >>> solve_biquadratic(a, b, NewOption) [(7/2 - sqrt(29)/2, -sqrt(29)/2 - 1/2), (sqrt(29)/2 + 7/2, -1/2 + sqrt(29)/2)] rNrF)expandcsg|]}t|qSrr )rexpryrr rz%solve_biquadratic..cs g|]\}}|||fqSr)subs)rZq_rootZp_rootr.rrr0rkey)rr is_groundrrgcdrrkeysltrimlist itertoolsproductsortedr) r)r*r'GxpqZp_rootsZq_roots solutionsrr.rr"Os"'       r"cslddddd fdd z||jdd }WntyNtYn0|d urdt|td Sd Sd S) a Solve a generic system of polynomial equations. Returns all possible solutions over C[x_1, x_2, ..., x_m] of a set F = { f_1, f_2, ..., f_n } of polynomial equations, using Groebner basis approach. For now only zero-dimensional systems are supported, which means F can have at most a finite number of solutions. The algorithm works by the fact that, supposing G is the basis of F with respect to an elimination order (here lexicographic order is used), G and F generate the same ideal, they have the same set of solutions. By the elimination property, if G is a reduced, zero-dimensional Groebner basis, then there exists an univariate polynomial in G (in its last variable). This can be solved by computing its roots. Substituting all computed roots for the last (eliminated) variable in other elements of G, new polynomial system is generated. Applying the above procedure recursively, a finite number of solutions can be found. The ability of finding all solutions by this procedure depends on the root finding algorithms. If no solutions were found, it means only that roots() failed, but the system is solvable. To overcome this difficulty use numerical algorithms instead. Parameters ========== polys: a list/tuple/set Listing all the polynomial equations that are needed to be solved opt: an Options object For specifying keyword arguments and generators strict: a boolean If strict is True, NotImplementedError will be raised if the solution is known to be incomplete Returns ======= List[Tuple] A List of tuples. Solutions for symbols that satisfy the equations listed in seq References ========== .. [Buchberger01] B. Buchberger, Groebner Bases: A Short Introduction for Systems Theorists, In: R. Moreno-Diaz, B. Buchberger, J.L. Freire, Proceedings of EUROCAST'01, February, 2001 .. [Cox97] D. Cox, J. Little, D. O'Shea, Ideals, Varieties and Algorithms, Springer, Second Edition, 1997, pp. 112 Raises ======== NotImplementedError If the system is not zero-dimensional. (does not have a finite number of solutions) UnsolvableFactorError If ``strict`` is True and not all solution components are expressible in radicals Examples ======== >>> from sympy import Poly, Options >>> from sympy.solvers.polysys import solve_generic >>> from sympy.abc import x, y >>> NewOption = Options((x, y), {'domain': 'ZZ'}) >>> a = Poly(x - y + 5, x, y, domain='ZZ') >>> b = Poly(x + y - 3, x, y, domain='ZZ') >>> solve_generic([a, b], NewOption) [(-1, 4)] >>> a = Poly(x - 2*y + 5, x, y, domain='ZZ') >>> b = Poly(2*x - y - 3, x, y, domain='ZZ') >>> solve_generic([a, b], NewOption) [(11/3, 13/3)] >>> a = Poly(x**2 + y, x, y, domain='ZZ') >>> b = Poly(x + y*4, x, y, domain='ZZ') >>> solve_generic([a, b], NewOption) [(0, 0), (1/4, -1/16)] >>> a = Poly(x**5 - x + y**3, x, y, domain='ZZ') >>> b = Poly(y**2 - 1, x, y, domain='ZZ') >>> solve_generic([a, b], NewOption, strict=True) Traceback (most recent call last): ... UnsolvableFactorError cSs(|D]}t|ddrdSqdS)z8Returns True if 'f' is univariate in its last variable. Nr1FT)monomsany)r)monomrrr_is_univariates z%solve_generic.._is_univariatecSs,|||i}||dkr(|jdd}|S)z:Replace generator with a root so that the result is nice. rF)deep)as_exprdegreer,)r)genzeror?rrr _subs_roots z!solve_generic.._subs_rootFcst|t|krdkrLnn,tt|d|dd}dd|DSt||dd}t|dkr||djr||sxgSd Stt|}t|t|krttd t|dkr| }n ttd |j }|d}tt| |d}|sgSt|dkrd d|DSg}|D]n} g} |d d} |d dD](} | || } | t j ur<| | q<| | D]}||| fqpq|rt|dt|krttd |S) z/Recursively solves reduced polynomial systems. r+rr1rcSsg|] }|fqSrrrrJrrrr0rz@solve_generic.._solve_reduced_system..Tr&Nzv only zero-dimensional systems supported (finite number of solutions) cSsg|] }|fqSrrrLrrrr0-r)rr9rr7rr5filterNotImplementedErrorrpoprr8rZeroappend)systemrentryzerosbasisZ univariater)rIrArJZ new_systemnew_gensbeqsolutionrE_solve_reduced_systemrKrrrr\sD         z,solve_generic.._solve_reduced_systemT)rTNr3)F)rr rOr<r)r&r'rresultrr[rr#sa C   r#cOst||dd}tt|}|d}|durLt|D]\}}||||<q4|dd|dd}}|}|} t } | D]} | | f|fqt|dd} t |dd} t | | D]\}}t }| D]\}}gtt ||}}|D]V}|f|}|j |r||dkr||t|}|||kr||qt|dd d }|} | D]2} | js|| }n|}| | f||fqhq|} qt| } t| D]\}\}}|| |<qt| td S) a Solve a polynomial system using Gianni-Kalkbrenner algorithm. The algorithm proceeds by computing one Groebner basis in the ground domain and then by iteratively computing polynomial factorizations in appropriately constructed algebraic extensions of the ground domain. Parameters ========== polys: a list/tuple/set Listing all the equations that are needed to be solved gens: generators generators of the equations in polys for which we want the solutions args: Keyword arguments Special options for solving the equations Returns ======= List[Tuple] A List of tuples. Solutions for symbols that satisfy the equations listed in polys Examples ======== >>> from sympy import solve_triangulated >>> from sympy.abc import x, y, z >>> F = [x**2 + y + z - 1, x + y**2 + z - 1, x + y + z**2 - 1] >>> solve_triangulated(F, x, y, z) [(0, 0, 1), (0, 1, 0), (1, 0, 0)] References ========== 1. Patrizia Gianni, Teo Mora, Algebraic Solution of System of Polynomial Equations using Groebner Bases, AAECC-5 on Applied Algebra, Algebraic Algorithms and Error-Correcting Codes, LNCS 356 247--257, 1989 TrMdomainNrr1r+cSs|S)N)rH)hrrrrz$solve_triangulated..r3)rr9reversedget enumerate set_domainr8 get_domain ground_rootssetaddr zip has_only_gensrHevaldictrRmin is_Rationalalgebraic_fieldr<r)r&rr%r=r^rr*r)domrUrArJZvar_seqZvars_seqvarvarsZ _solutionsvaluesHmapping_varsr_r?Zdom_zerorZ_rrrsolve_triangulatedPsH-       rx)F)rr: sympy.corersympy.core.sortingr sympy.polysrrrsympy.polys.polytoolsrsympy.polys.polyerrorsrr r sympy.simplifyr Zsympy.utilitiesr sympy.utilities.miscr Exceptionrrr"r#rxrrrrs      <A A