| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186 |
- """OO layer for several polynomial representations. """
- from __future__ import annotations
- from sympy.external.gmpy import GROUND_TYPES
- from sympy.utilities.exceptions import sympy_deprecation_warning
- from sympy.core.numbers import oo
- from sympy.core.sympify import CantSympify
- from sympy.polys.polyutils import PicklableWithSlots, _sort_factors
- from sympy.polys.domains import Domain, ZZ, QQ
- from sympy.polys.polyerrors import (
- CoercionFailed,
- ExactQuotientFailed,
- DomainError,
- NotInvertible,
- )
- from sympy.polys.densebasic import (
- ninf,
- dmp_validate,
- dup_normal, dmp_normal,
- dup_convert, dmp_convert,
- dmp_from_sympy,
- dup_strip,
- dmp_degree_in,
- dmp_degree_list,
- dmp_negative_p,
- dmp_ground_LC,
- dmp_ground_TC,
- dmp_ground_nth,
- dmp_one, dmp_ground,
- dmp_zero, dmp_zero_p, dmp_one_p, dmp_ground_p,
- dup_from_dict, dmp_from_dict,
- dmp_to_dict,
- dmp_deflate,
- dmp_inject, dmp_eject,
- dmp_terms_gcd,
- dmp_list_terms, dmp_exclude,
- dup_slice, dmp_slice_in, dmp_permute,
- dmp_to_tuple,)
- from sympy.polys.densearith import (
- dmp_add_ground,
- dmp_sub_ground,
- dmp_mul_ground,
- dmp_quo_ground,
- dmp_exquo_ground,
- dmp_abs,
- dmp_neg,
- dmp_add,
- dmp_sub,
- dmp_mul,
- dmp_sqr,
- dmp_pow,
- dmp_pdiv,
- dmp_prem,
- dmp_pquo,
- dmp_pexquo,
- dmp_div,
- dmp_rem,
- dmp_quo,
- dmp_exquo,
- dmp_add_mul, dmp_sub_mul,
- dmp_max_norm,
- dmp_l1_norm,
- dmp_l2_norm_squared)
- from sympy.polys.densetools import (
- dmp_clear_denoms,
- dmp_integrate_in,
- dmp_diff_in,
- dmp_eval_in,
- dup_revert,
- dmp_ground_trunc,
- dmp_ground_content,
- dmp_ground_primitive,
- dmp_ground_monic,
- dmp_compose,
- dup_decompose,
- dup_shift,
- dmp_shift,
- dup_transform,
- dmp_lift)
- from sympy.polys.euclidtools import (
- dup_half_gcdex, dup_gcdex, dup_invert,
- dmp_subresultants,
- dmp_resultant,
- dmp_discriminant,
- dmp_inner_gcd,
- dmp_gcd,
- dmp_lcm,
- dmp_cancel)
- from sympy.polys.sqfreetools import (
- dup_gff_list,
- dmp_norm,
- dmp_sqf_p,
- dmp_sqf_norm,
- dmp_sqf_part,
- dmp_sqf_list, dmp_sqf_list_include)
- from sympy.polys.factortools import (
- dup_cyclotomic_p, dmp_irreducible_p,
- dmp_factor_list, dmp_factor_list_include)
- from sympy.polys.rootisolation import (
- dup_isolate_real_roots_sqf,
- dup_isolate_real_roots,
- dup_isolate_all_roots_sqf,
- dup_isolate_all_roots,
- dup_refine_real_root,
- dup_count_real_roots,
- dup_count_complex_roots,
- dup_sturm,
- dup_cauchy_upper_bound,
- dup_cauchy_lower_bound,
- dup_mignotte_sep_bound_squared)
- from sympy.polys.polyerrors import (
- UnificationFailed,
- PolynomialError)
- if GROUND_TYPES == 'flint':
- import flint
- def _supported_flint_domain(D):
- return D.is_ZZ or D.is_QQ or D.is_FF and D._is_flint
- else:
- flint = None
- def _supported_flint_domain(D):
- return False
- class DMP(CantSympify):
- """Dense Multivariate Polynomials over `K`. """
- __slots__ = ()
- lev: int
- dom: Domain
- def __new__(cls, rep, dom, lev=None):
- if lev is None:
- rep, lev = dmp_validate(rep)
- elif not isinstance(rep, list):
- raise CoercionFailed("expected list, got %s" % type(rep))
- return cls.new(rep, dom, lev)
- @classmethod
- def new(cls, rep, dom, lev):
- # It would be too slow to call _validate_args always at runtime.
- # Ideally this checking would be handled by a static type checker.
- #
- #cls._validate_args(rep, dom, lev)
- if flint is not None:
- if lev == 0 and _supported_flint_domain(dom):
- return DUP_Flint._new(rep, dom, lev)
- return DMP_Python._new(rep, dom, lev)
- @property
- def rep(f):
- """Get the representation of ``f``. """
- sympy_deprecation_warning("""
- Accessing the ``DMP.rep`` attribute is deprecated. The internal
- representation of ``DMP`` instances can now be ``DUP_Flint`` when the
- ground types are ``flint``. In this case the ``DMP`` instance does not
- have a ``rep`` attribute. Use ``DMP.to_list()`` instead. Using
- ``DMP.to_list()`` also works in previous versions of SymPy.
- """,
- deprecated_since_version="1.13",
- active_deprecations_target="dmp-rep",
- )
- return f.to_list()
- def to_best(f):
- """Convert to DUP_Flint if possible.
- This method should be used when the domain or level is changed and it
- potentially becomes possible to convert from DMP_Python to DUP_Flint.
- """
- if flint is not None:
- if isinstance(f, DMP_Python) and f.lev == 0 and _supported_flint_domain(f.dom):
- return DUP_Flint.new(f._rep, f.dom, f.lev)
- return f
- @classmethod
- def _validate_args(cls, rep, dom, lev):
- assert isinstance(dom, Domain)
- assert isinstance(lev, int) and lev >= 0
- def validate_rep(rep, lev):
- assert isinstance(rep, list)
- if lev == 0:
- assert all(dom.of_type(c) for c in rep)
- else:
- for r in rep:
- validate_rep(r, lev - 1)
- validate_rep(rep, lev)
- @classmethod
- def from_dict(cls, rep, lev, dom):
- rep = dmp_from_dict(rep, lev, dom)
- return cls.new(rep, dom, lev)
- @classmethod
- def from_list(cls, rep, lev, dom):
- """Create an instance of ``cls`` given a list of native coefficients. """
- return cls.new(dmp_convert(rep, lev, None, dom), dom, lev)
- @classmethod
- def from_sympy_list(cls, rep, lev, dom):
- """Create an instance of ``cls`` given a list of SymPy coefficients. """
- return cls.new(dmp_from_sympy(rep, lev, dom), dom, lev)
- @classmethod
- def from_monoms_coeffs(cls, monoms, coeffs, lev, dom):
- return cls(dict(list(zip(monoms, coeffs))), dom, lev)
- def convert(f, dom):
- """Convert ``f`` to a ``DMP`` over the new domain. """
- if f.dom == dom:
- return f
- elif f.lev or flint is None:
- return f._convert(dom)
- elif isinstance(f, DUP_Flint):
- if _supported_flint_domain(dom):
- return f._convert(dom)
- else:
- return f.to_DMP_Python()._convert(dom)
- elif isinstance(f, DMP_Python):
- if _supported_flint_domain(dom):
- return f._convert(dom).to_DUP_Flint()
- else:
- return f._convert(dom)
- else:
- raise RuntimeError("unreachable code")
- def _convert(f, dom):
- raise NotImplementedError
- @classmethod
- def zero(cls, lev, dom):
- return DMP(dmp_zero(lev), dom, lev)
- @classmethod
- def one(cls, lev, dom):
- return DMP(dmp_one(lev, dom), dom, lev)
- def _one(f):
- raise NotImplementedError
- def __repr__(f):
- return "%s(%s, %s)" % (f.__class__.__name__, f.to_list(), f.dom)
- def __hash__(f):
- return hash((f.__class__.__name__, f.to_tuple(), f.lev, f.dom))
- def __getnewargs__(self):
- return self.to_list(), self.dom, self.lev
- def ground_new(f, coeff):
- """Construct a new ground instance of ``f``. """
- raise NotImplementedError
- def unify_DMP(f, g):
- """Unify and return ``DMP`` instances of ``f`` and ``g``. """
- if not isinstance(g, DMP) or f.lev != g.lev:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- if f.dom != g.dom:
- dom = f.dom.unify(g.dom)
- f = f.convert(dom)
- g = g.convert(dom)
- return f, g
- def to_dict(f, zero=False):
- """Convert ``f`` to a dict representation with native coefficients. """
- return dmp_to_dict(f.to_list(), f.lev, f.dom, zero=zero)
- def to_sympy_dict(f, zero=False):
- """Convert ``f`` to a dict representation with SymPy coefficients. """
- rep = f.to_dict(zero=zero)
- for k, v in rep.items():
- rep[k] = f.dom.to_sympy(v)
- return rep
- def to_sympy_list(f):
- """Convert ``f`` to a list representation with SymPy coefficients. """
- def sympify_nested_list(rep):
- out = []
- for val in rep:
- if isinstance(val, list):
- out.append(sympify_nested_list(val))
- else:
- out.append(f.dom.to_sympy(val))
- return out
- return sympify_nested_list(f.to_list())
- def to_list(f):
- """Convert ``f`` to a list representation with native coefficients. """
- raise NotImplementedError
- def to_tuple(f):
- """
- Convert ``f`` to a tuple representation with native coefficients.
- This is needed for hashing.
- """
- raise NotImplementedError
- def to_ring(f):
- """Make the ground domain a ring. """
- return f.convert(f.dom.get_ring())
- def to_field(f):
- """Make the ground domain a field. """
- return f.convert(f.dom.get_field())
- def to_exact(f):
- """Make the ground domain exact. """
- return f.convert(f.dom.get_exact())
- def slice(f, m, n, j=0):
- """Take a continuous subsequence of terms of ``f``. """
- if not f.lev and not j:
- return f._slice(m, n)
- else:
- return f._slice_lev(m, n, j)
- def _slice(f, m, n):
- raise NotImplementedError
- def _slice_lev(f, m, n, j):
- raise NotImplementedError
- def coeffs(f, order=None):
- """Returns all non-zero coefficients from ``f`` in lex order. """
- return [ c for _, c in f.terms(order=order) ]
- def monoms(f, order=None):
- """Returns all non-zero monomials from ``f`` in lex order. """
- return [ m for m, _ in f.terms(order=order) ]
- def terms(f, order=None):
- """Returns all non-zero terms from ``f`` in lex order. """
- if f.is_zero:
- zero_monom = (0,)*(f.lev + 1)
- return [(zero_monom, f.dom.zero)]
- else:
- return f._terms(order=order)
- def _terms(f, order=None):
- raise NotImplementedError
- def all_coeffs(f):
- """Returns all coefficients from ``f``. """
- if f.lev:
- raise PolynomialError('multivariate polynomials not supported')
- if not f:
- return [f.dom.zero]
- else:
- return list(f.to_list())
- def all_monoms(f):
- """Returns all monomials from ``f``. """
- if f.lev:
- raise PolynomialError('multivariate polynomials not supported')
- n = f.degree()
- if n < 0:
- return [(0,)]
- else:
- return [ (n - i,) for i, c in enumerate(f.to_list()) ]
- def all_terms(f):
- """Returns all terms from a ``f``. """
- if f.lev:
- raise PolynomialError('multivariate polynomials not supported')
- n = f.degree()
- if n < 0:
- return [((0,), f.dom.zero)]
- else:
- return [ ((n - i,), c) for i, c in enumerate(f.to_list()) ]
- def lift(f):
- """Convert algebraic coefficients to rationals. """
- return f._lift().to_best()
- def _lift(f):
- raise NotImplementedError
- def deflate(f):
- """Reduce degree of `f` by mapping `x_i^m` to `y_i`. """
- raise NotImplementedError
- def inject(f, front=False):
- """Inject ground domain generators into ``f``. """
- raise NotImplementedError
- def eject(f, dom, front=False):
- """Eject selected generators into the ground domain. """
- raise NotImplementedError
- def exclude(f):
- r"""
- Remove useless generators from ``f``.
- Returns the removed generators and the new excluded ``f``.
- Examples
- ========
- >>> from sympy.polys.polyclasses import DMP
- >>> from sympy.polys.domains import ZZ
- >>> DMP([[[ZZ(1)]], [[ZZ(1)], [ZZ(2)]]], ZZ).exclude()
- ([2], DMP_Python([[1], [1, 2]], ZZ))
- """
- J, F = f._exclude()
- return J, F.to_best()
- def _exclude(f):
- raise NotImplementedError
- def permute(f, P):
- r"""
- Returns a polynomial in `K[x_{P(1)}, ..., x_{P(n)}]`.
- Examples
- ========
- >>> from sympy.polys.polyclasses import DMP
- >>> from sympy.polys.domains import ZZ
- >>> DMP([[[ZZ(2)], [ZZ(1), ZZ(0)]], [[]]], ZZ).permute([1, 0, 2])
- DMP_Python([[[2], []], [[1, 0], []]], ZZ)
- >>> DMP([[[ZZ(2)], [ZZ(1), ZZ(0)]], [[]]], ZZ).permute([1, 2, 0])
- DMP_Python([[[1], []], [[2, 0], []]], ZZ)
- """
- return f._permute(P)
- def _permute(f, P):
- raise NotImplementedError
- def terms_gcd(f):
- """Remove GCD of terms from the polynomial ``f``. """
- raise NotImplementedError
- def abs(f):
- """Make all coefficients in ``f`` positive. """
- raise NotImplementedError
- def neg(f):
- """Negate all coefficients in ``f``. """
- raise NotImplementedError
- def add_ground(f, c):
- """Add an element of the ground domain to ``f``. """
- return f._add_ground(f.dom.convert(c))
- def sub_ground(f, c):
- """Subtract an element of the ground domain from ``f``. """
- return f._sub_ground(f.dom.convert(c))
- def mul_ground(f, c):
- """Multiply ``f`` by a an element of the ground domain. """
- return f._mul_ground(f.dom.convert(c))
- def quo_ground(f, c):
- """Quotient of ``f`` by a an element of the ground domain. """
- return f._quo_ground(f.dom.convert(c))
- def exquo_ground(f, c):
- """Exact quotient of ``f`` by a an element of the ground domain. """
- return f._exquo_ground(f.dom.convert(c))
- def add(f, g):
- """Add two multivariate polynomials ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._add(G)
- def sub(f, g):
- """Subtract two multivariate polynomials ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._sub(G)
- def mul(f, g):
- """Multiply two multivariate polynomials ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._mul(G)
- def sqr(f):
- """Square a multivariate polynomial ``f``. """
- return f._sqr()
- def pow(f, n):
- """Raise ``f`` to a non-negative power ``n``. """
- if not isinstance(n, int):
- raise TypeError("``int`` expected, got %s" % type(n))
- return f._pow(n)
- def pdiv(f, g):
- """Polynomial pseudo-division of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._pdiv(G)
- def prem(f, g):
- """Polynomial pseudo-remainder of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._prem(G)
- def pquo(f, g):
- """Polynomial pseudo-quotient of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._pquo(G)
- def pexquo(f, g):
- """Polynomial exact pseudo-quotient of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._pexquo(G)
- def div(f, g):
- """Polynomial division with remainder of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._div(G)
- def rem(f, g):
- """Computes polynomial remainder of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._rem(G)
- def quo(f, g):
- """Computes polynomial quotient of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._quo(G)
- def exquo(f, g):
- """Computes polynomial exact quotient of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._exquo(G)
- def _add_ground(f, c):
- raise NotImplementedError
- def _sub_ground(f, c):
- raise NotImplementedError
- def _mul_ground(f, c):
- raise NotImplementedError
- def _quo_ground(f, c):
- raise NotImplementedError
- def _exquo_ground(f, c):
- raise NotImplementedError
- def _add(f, g):
- raise NotImplementedError
- def _sub(f, g):
- raise NotImplementedError
- def _mul(f, g):
- raise NotImplementedError
- def _sqr(f):
- raise NotImplementedError
- def _pow(f, n):
- raise NotImplementedError
- def _pdiv(f, g):
- raise NotImplementedError
- def _prem(f, g):
- raise NotImplementedError
- def _pquo(f, g):
- raise NotImplementedError
- def _pexquo(f, g):
- raise NotImplementedError
- def _div(f, g):
- raise NotImplementedError
- def _rem(f, g):
- raise NotImplementedError
- def _quo(f, g):
- raise NotImplementedError
- def _exquo(f, g):
- raise NotImplementedError
- def degree(f, j=0):
- """Returns the leading degree of ``f`` in ``x_j``. """
- if not isinstance(j, int):
- raise TypeError("``int`` expected, got %s" % type(j))
- return f._degree(j)
- def _degree(f, j):
- raise NotImplementedError
- def degree_list(f):
- """Returns a list of degrees of ``f``. """
- raise NotImplementedError
- def total_degree(f):
- """Returns the total degree of ``f``. """
- raise NotImplementedError
- def homogenize(f, s):
- """Return homogeneous polynomial of ``f``"""
- td = f.total_degree()
- result = {}
- new_symbol = (s == len(f.terms()[0][0]))
- for term in f.terms():
- d = sum(term[0])
- if d < td:
- i = td - d
- else:
- i = 0
- if new_symbol:
- result[term[0] + (i,)] = term[1]
- else:
- l = list(term[0])
- l[s] += i
- result[tuple(l)] = term[1]
- return DMP.from_dict(result, f.lev + int(new_symbol), f.dom)
- def homogeneous_order(f):
- """Returns the homogeneous order of ``f``. """
- if f.is_zero:
- return -oo
- monoms = f.monoms()
- tdeg = sum(monoms[0])
- for monom in monoms:
- _tdeg = sum(monom)
- if _tdeg != tdeg:
- return None
- return tdeg
- def LC(f):
- """Returns the leading coefficient of ``f``. """
- raise NotImplementedError
- def TC(f):
- """Returns the trailing coefficient of ``f``. """
- raise NotImplementedError
- def nth(f, *N):
- """Returns the ``n``-th coefficient of ``f``. """
- if all(isinstance(n, int) for n in N):
- return f._nth(N)
- else:
- raise TypeError("a sequence of integers expected")
- def _nth(f, N):
- raise NotImplementedError
- def max_norm(f):
- """Returns maximum norm of ``f``. """
- raise NotImplementedError
- def l1_norm(f):
- """Returns l1 norm of ``f``. """
- raise NotImplementedError
- def l2_norm_squared(f):
- """Return squared l2 norm of ``f``. """
- raise NotImplementedError
- def clear_denoms(f):
- """Clear denominators, but keep the ground domain. """
- raise NotImplementedError
- def integrate(f, m=1, j=0):
- """Computes the ``m``-th order indefinite integral of ``f`` in ``x_j``. """
- if not isinstance(m, int):
- raise TypeError("``int`` expected, got %s" % type(m))
- if not isinstance(j, int):
- raise TypeError("``int`` expected, got %s" % type(j))
- return f._integrate(m, j)
- def _integrate(f, m, j):
- raise NotImplementedError
- def diff(f, m=1, j=0):
- """Computes the ``m``-th order derivative of ``f`` in ``x_j``. """
- if not isinstance(m, int):
- raise TypeError("``int`` expected, got %s" % type(m))
- if not isinstance(j, int):
- raise TypeError("``int`` expected, got %s" % type(j))
- return f._diff(m, j)
- def _diff(f, m, j):
- raise NotImplementedError
- def eval(f, a, j=0):
- """Evaluates ``f`` at the given point ``a`` in ``x_j``. """
- if not isinstance(j, int):
- raise TypeError("``int`` expected, got %s" % type(j))
- elif not (0 <= j <= f.lev):
- raise ValueError("invalid variable index %s" % j)
- if f.lev:
- return f._eval_lev(a, j)
- else:
- return f._eval(a)
- def _eval(f, a):
- raise NotImplementedError
- def _eval_lev(f, a, j):
- raise NotImplementedError
- def half_gcdex(f, g):
- """Half extended Euclidean algorithm, if univariate. """
- F, G = f.unify_DMP(g)
- if F.lev:
- raise ValueError('univariate polynomial expected')
- return F._half_gcdex(G)
- def _half_gcdex(f, g):
- raise NotImplementedError
- def gcdex(f, g):
- """Extended Euclidean algorithm, if univariate. """
- F, G = f.unify_DMP(g)
- if F.lev:
- raise ValueError('univariate polynomial expected')
- if not F.dom.is_Field:
- raise DomainError('ground domain must be a field')
- return F._gcdex(G)
- def _gcdex(f, g):
- raise NotImplementedError
- def invert(f, g):
- """Invert ``f`` modulo ``g``, if possible. """
- F, G = f.unify_DMP(g)
- if F.lev:
- raise ValueError('univariate polynomial expected')
- return F._invert(G)
- def _invert(f, g):
- raise NotImplementedError
- def revert(f, n):
- """Compute ``f**(-1)`` mod ``x**n``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._revert(n)
- def _revert(f, n):
- raise NotImplementedError
- def subresultants(f, g):
- """Computes subresultant PRS sequence of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._subresultants(G)
- def _subresultants(f, g):
- raise NotImplementedError
- def resultant(f, g, includePRS=False):
- """Computes resultant of ``f`` and ``g`` via PRS. """
- F, G = f.unify_DMP(g)
- if includePRS:
- return F._resultant_includePRS(G)
- else:
- return F._resultant(G)
- def _resultant(f, g, includePRS=False):
- raise NotImplementedError
- def discriminant(f):
- """Computes discriminant of ``f``. """
- raise NotImplementedError
- def cofactors(f, g):
- """Returns GCD of ``f`` and ``g`` and their cofactors. """
- F, G = f.unify_DMP(g)
- return F._cofactors(G)
- def _cofactors(f, g):
- raise NotImplementedError
- def gcd(f, g):
- """Returns polynomial GCD of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._gcd(G)
- def _gcd(f, g):
- raise NotImplementedError
- def lcm(f, g):
- """Returns polynomial LCM of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._lcm(G)
- def _lcm(f, g):
- raise NotImplementedError
- def cancel(f, g, include=True):
- """Cancel common factors in a rational function ``f/g``. """
- F, G = f.unify_DMP(g)
- if include:
- return F._cancel_include(G)
- else:
- return F._cancel(G)
- def _cancel(f, g):
- raise NotImplementedError
- def _cancel_include(f, g):
- raise NotImplementedError
- def trunc(f, p):
- """Reduce ``f`` modulo a constant ``p``. """
- return f._trunc(f.dom.convert(p))
- def _trunc(f, p):
- raise NotImplementedError
- def monic(f):
- """Divides all coefficients by ``LC(f)``. """
- raise NotImplementedError
- def content(f):
- """Returns GCD of polynomial coefficients. """
- raise NotImplementedError
- def primitive(f):
- """Returns content and a primitive form of ``f``. """
- raise NotImplementedError
- def compose(f, g):
- """Computes functional composition of ``f`` and ``g``. """
- F, G = f.unify_DMP(g)
- return F._compose(G)
- def _compose(f, g):
- raise NotImplementedError
- def decompose(f):
- """Computes functional decomposition of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._decompose()
- def _decompose(f):
- raise NotImplementedError
- def shift(f, a):
- """Efficiently compute Taylor shift ``f(x + a)``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._shift(f.dom.convert(a))
- def shift_list(f, a):
- """Efficiently compute Taylor shift ``f(X + A)``. """
- a = [f.dom.convert(ai) for ai in a]
- return f._shift_list(a)
- def _shift(f, a):
- raise NotImplementedError
- def transform(f, p, q):
- """Evaluate functional transformation ``q**n * f(p/q)``."""
- if f.lev:
- raise ValueError('univariate polynomial expected')
- P, Q = p.unify_DMP(q)
- F, P = f.unify_DMP(P)
- F, Q = F.unify_DMP(Q)
- return F._transform(P, Q)
- def _transform(f, p, q):
- raise NotImplementedError
- def sturm(f):
- """Computes the Sturm sequence of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._sturm()
- def _sturm(f):
- raise NotImplementedError
- def cauchy_upper_bound(f):
- """Computes the Cauchy upper bound on the roots of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._cauchy_upper_bound()
- def _cauchy_upper_bound(f):
- raise NotImplementedError
- def cauchy_lower_bound(f):
- """Computes the Cauchy lower bound on the nonzero roots of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._cauchy_lower_bound()
- def _cauchy_lower_bound(f):
- raise NotImplementedError
- def mignotte_sep_bound_squared(f):
- """Computes the squared Mignotte bound on root separations of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._mignotte_sep_bound_squared()
- def _mignotte_sep_bound_squared(f):
- raise NotImplementedError
- def gff_list(f):
- """Computes greatest factorial factorization of ``f``. """
- if f.lev:
- raise ValueError('univariate polynomial expected')
- return f._gff_list()
- def _gff_list(f):
- raise NotImplementedError
- def norm(f):
- """Computes ``Norm(f)``."""
- raise NotImplementedError
- def sqf_norm(f):
- """Computes square-free norm of ``f``. """
- raise NotImplementedError
- def sqf_part(f):
- """Computes square-free part of ``f``. """
- raise NotImplementedError
- def sqf_list(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- raise NotImplementedError
- def sqf_list_include(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- raise NotImplementedError
- def factor_list(f):
- """Returns a list of irreducible factors of ``f``. """
- raise NotImplementedError
- def factor_list_include(f):
- """Returns a list of irreducible factors of ``f``. """
- raise NotImplementedError
- def intervals(f, all=False, eps=None, inf=None, sup=None, fast=False, sqf=False):
- """Compute isolating intervals for roots of ``f``. """
- if f.lev:
- raise PolynomialError("Cannot isolate roots of a multivariate polynomial")
- if all and sqf:
- return f._isolate_all_roots_sqf(eps=eps, inf=inf, sup=sup, fast=fast)
- elif all and not sqf:
- return f._isolate_all_roots(eps=eps, inf=inf, sup=sup, fast=fast)
- elif not all and sqf:
- return f._isolate_real_roots_sqf(eps=eps, inf=inf, sup=sup, fast=fast)
- else:
- return f._isolate_real_roots(eps=eps, inf=inf, sup=sup, fast=fast)
- def _isolate_all_roots(f, eps, inf, sup, fast):
- raise NotImplementedError
- def _isolate_all_roots_sqf(f, eps, inf, sup, fast):
- raise NotImplementedError
- def _isolate_real_roots(f, eps, inf, sup, fast):
- raise NotImplementedError
- def _isolate_real_roots_sqf(f, eps, inf, sup, fast):
- raise NotImplementedError
- def refine_root(f, s, t, eps=None, steps=None, fast=False):
- """
- Refine an isolating interval to the given precision.
- ``eps`` should be a rational number.
- """
- if f.lev:
- raise PolynomialError(
- "Cannot refine a root of a multivariate polynomial")
- return f._refine_real_root(s, t, eps=eps, steps=steps, fast=fast)
- def _refine_real_root(f, s, t, eps, steps, fast):
- raise NotImplementedError
- def count_real_roots(f, inf=None, sup=None):
- """Return the number of real roots of ``f`` in ``[inf, sup]``. """
- raise NotImplementedError
- def count_complex_roots(f, inf=None, sup=None):
- """Return the number of complex roots of ``f`` in ``[inf, sup]``. """
- raise NotImplementedError
- @property
- def is_zero(f):
- """Returns ``True`` if ``f`` is a zero polynomial. """
- raise NotImplementedError
- @property
- def is_one(f):
- """Returns ``True`` if ``f`` is a unit polynomial. """
- raise NotImplementedError
- @property
- def is_ground(f):
- """Returns ``True`` if ``f`` is an element of the ground domain. """
- raise NotImplementedError
- @property
- def is_sqf(f):
- """Returns ``True`` if ``f`` is a square-free polynomial. """
- raise NotImplementedError
- @property
- def is_monic(f):
- """Returns ``True`` if the leading coefficient of ``f`` is one. """
- raise NotImplementedError
- @property
- def is_primitive(f):
- """Returns ``True`` if the GCD of the coefficients of ``f`` is one. """
- raise NotImplementedError
- @property
- def is_linear(f):
- """Returns ``True`` if ``f`` is linear in all its variables. """
- raise NotImplementedError
- @property
- def is_quadratic(f):
- """Returns ``True`` if ``f`` is quadratic in all its variables. """
- raise NotImplementedError
- @property
- def is_monomial(f):
- """Returns ``True`` if ``f`` is zero or has only one term. """
- raise NotImplementedError
- @property
- def is_homogeneous(f):
- """Returns ``True`` if ``f`` is a homogeneous polynomial. """
- raise NotImplementedError
- @property
- def is_irreducible(f):
- """Returns ``True`` if ``f`` has no factors over its domain. """
- raise NotImplementedError
- @property
- def is_cyclotomic(f):
- """Returns ``True`` if ``f`` is a cyclotomic polynomial. """
- raise NotImplementedError
- def __abs__(f):
- return f.abs()
- def __neg__(f):
- return f.neg()
- def __add__(f, g):
- if isinstance(g, DMP):
- return f.add(g)
- else:
- try:
- return f.add_ground(g)
- except CoercionFailed:
- return NotImplemented
- def __radd__(f, g):
- return f.__add__(g)
- def __sub__(f, g):
- if isinstance(g, DMP):
- return f.sub(g)
- else:
- try:
- return f.sub_ground(g)
- except CoercionFailed:
- return NotImplemented
- def __rsub__(f, g):
- return (-f).__add__(g)
- def __mul__(f, g):
- if isinstance(g, DMP):
- return f.mul(g)
- else:
- try:
- return f.mul_ground(g)
- except CoercionFailed:
- return NotImplemented
- def __rmul__(f, g):
- return f.__mul__(g)
- def __truediv__(f, g):
- if isinstance(g, DMP):
- return f.exquo(g)
- else:
- try:
- return f.mul_ground(g)
- except CoercionFailed:
- return NotImplemented
- def __rtruediv__(f, g):
- if isinstance(g, DMP):
- return g.exquo(f)
- else:
- try:
- return f._one().mul_ground(g).exquo(f)
- except CoercionFailed:
- return NotImplemented
- def __pow__(f, n):
- return f.pow(n)
- def __divmod__(f, g):
- return f.div(g)
- def __mod__(f, g):
- return f.rem(g)
- def __floordiv__(f, g):
- if isinstance(g, DMP):
- return f.quo(g)
- else:
- try:
- return f.quo_ground(g)
- except TypeError:
- return NotImplemented
- def __eq__(f, g):
- if f is g:
- return True
- if not isinstance(g, DMP):
- return NotImplemented
- try:
- F, G = f.unify_DMP(g)
- except UnificationFailed:
- return False
- else:
- return F._strict_eq(G)
- def _strict_eq(f, g):
- raise NotImplementedError
- def eq(f, g, strict=False):
- if not strict:
- return f == g
- else:
- return f._strict_eq(g)
- def ne(f, g, strict=False):
- return not f.eq(g, strict=strict)
- def __lt__(f, g):
- F, G = f.unify_DMP(g)
- return F.to_list() < G.to_list()
- def __le__(f, g):
- F, G = f.unify_DMP(g)
- return F.to_list() <= G.to_list()
- def __gt__(f, g):
- F, G = f.unify_DMP(g)
- return F.to_list() > G.to_list()
- def __ge__(f, g):
- F, G = f.unify_DMP(g)
- return F.to_list() >= G.to_list()
- def __bool__(f):
- return not f.is_zero
- class DMP_Python(DMP):
- """Dense Multivariate Polynomials over `K`. """
- __slots__ = ('_rep', 'dom', 'lev')
- @classmethod
- def _new(cls, rep, dom, lev):
- obj = object.__new__(cls)
- obj._rep = rep
- obj.lev = lev
- obj.dom = dom
- return obj
- def _strict_eq(f, g):
- if type(f) != type(g):
- return False
- return f.lev == g.lev and f.dom == g.dom and f._rep == g._rep
- def per(f, rep):
- """Create a DMP out of the given representation. """
- return f._new(rep, f.dom, f.lev)
- def ground_new(f, coeff):
- """Construct a new ground instance of ``f``. """
- return f._new(dmp_ground(coeff, f.lev), f.dom, f.lev)
- def _one(f):
- return f.one(f.lev, f.dom)
- def unify(f, g):
- """Unify representations of two multivariate polynomials. """
- # XXX: This function is not really used any more since there is
- # unify_DMP now.
- if not isinstance(g, DMP) or f.lev != g.lev:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- if f.dom == g.dom:
- return f.lev, f.dom, f.per, f._rep, g._rep
- else:
- lev, dom = f.lev, f.dom.unify(g.dom)
- F = dmp_convert(f._rep, lev, f.dom, dom)
- G = dmp_convert(g._rep, lev, g.dom, dom)
- def per(rep):
- return f._new(rep, dom, lev)
- return lev, dom, per, F, G
- def to_DUP_Flint(f):
- """Convert ``f`` to a Flint representation. """
- return DUP_Flint._new(f._rep, f.dom, f.lev)
- def to_list(f):
- """Convert ``f`` to a list representation with native coefficients. """
- return list(f._rep)
- def to_tuple(f):
- """Convert ``f`` to a tuple representation with native coefficients. """
- return dmp_to_tuple(f._rep, f.lev)
- def _convert(f, dom):
- """Convert the ground domain of ``f``. """
- return f._new(dmp_convert(f._rep, f.lev, f.dom, dom), dom, f.lev)
- def _slice(f, m, n):
- """Take a continuous subsequence of terms of ``f``. """
- rep = dup_slice(f._rep, m, n, f.dom)
- return f._new(rep, f.dom, f.lev)
- def _slice_lev(f, m, n, j):
- """Take a continuous subsequence of terms of ``f``. """
- rep = dmp_slice_in(f._rep, m, n, j, f.lev, f.dom)
- return f._new(rep, f.dom, f.lev)
- def _terms(f, order=None):
- """Returns all non-zero terms from ``f`` in lex order. """
- return dmp_list_terms(f._rep, f.lev, f.dom, order=order)
- def _lift(f):
- """Convert algebraic coefficients to rationals. """
- r = dmp_lift(f._rep, f.lev, f.dom)
- return f._new(r, f.dom.dom, f.lev)
- def deflate(f):
- """Reduce degree of `f` by mapping `x_i^m` to `y_i`. """
- J, F = dmp_deflate(f._rep, f.lev, f.dom)
- return J, f.per(F)
- def inject(f, front=False):
- """Inject ground domain generators into ``f``. """
- F, lev = dmp_inject(f._rep, f.lev, f.dom, front=front)
- # XXX: domain and level changed here
- return f._new(F, f.dom.dom, lev)
- def eject(f, dom, front=False):
- """Eject selected generators into the ground domain. """
- F = dmp_eject(f._rep, f.lev, dom, front=front)
- # XXX: domain and level changed here
- return f._new(F, dom, f.lev - len(dom.symbols))
- def _exclude(f):
- """Remove useless generators from ``f``. """
- J, F, u = dmp_exclude(f._rep, f.lev, f.dom)
- # XXX: level changed here
- return J, f._new(F, f.dom, u)
- def _permute(f, P):
- """Returns a polynomial in `K[x_{P(1)}, ..., x_{P(n)}]`. """
- return f.per(dmp_permute(f._rep, P, f.lev, f.dom))
- def terms_gcd(f):
- """Remove GCD of terms from the polynomial ``f``. """
- J, F = dmp_terms_gcd(f._rep, f.lev, f.dom)
- return J, f.per(F)
- def _add_ground(f, c):
- """Add an element of the ground domain to ``f``. """
- return f.per(dmp_add_ground(f._rep, c, f.lev, f.dom))
- def _sub_ground(f, c):
- """Subtract an element of the ground domain from ``f``. """
- return f.per(dmp_sub_ground(f._rep, c, f.lev, f.dom))
- def _mul_ground(f, c):
- """Multiply ``f`` by a an element of the ground domain. """
- return f.per(dmp_mul_ground(f._rep, c, f.lev, f.dom))
- def _quo_ground(f, c):
- """Quotient of ``f`` by a an element of the ground domain. """
- return f.per(dmp_quo_ground(f._rep, c, f.lev, f.dom))
- def _exquo_ground(f, c):
- """Exact quotient of ``f`` by a an element of the ground domain. """
- return f.per(dmp_exquo_ground(f._rep, c, f.lev, f.dom))
- def abs(f):
- """Make all coefficients in ``f`` positive. """
- return f.per(dmp_abs(f._rep, f.lev, f.dom))
- def neg(f):
- """Negate all coefficients in ``f``. """
- return f.per(dmp_neg(f._rep, f.lev, f.dom))
- def _add(f, g):
- """Add two multivariate polynomials ``f`` and ``g``. """
- return f.per(dmp_add(f._rep, g._rep, f.lev, f.dom))
- def _sub(f, g):
- """Subtract two multivariate polynomials ``f`` and ``g``. """
- return f.per(dmp_sub(f._rep, g._rep, f.lev, f.dom))
- def _mul(f, g):
- """Multiply two multivariate polynomials ``f`` and ``g``. """
- return f.per(dmp_mul(f._rep, g._rep, f.lev, f.dom))
- def sqr(f):
- """Square a multivariate polynomial ``f``. """
- return f.per(dmp_sqr(f._rep, f.lev, f.dom))
- def _pow(f, n):
- """Raise ``f`` to a non-negative power ``n``. """
- return f.per(dmp_pow(f._rep, n, f.lev, f.dom))
- def _pdiv(f, g):
- """Polynomial pseudo-division of ``f`` and ``g``. """
- q, r = dmp_pdiv(f._rep, g._rep, f.lev, f.dom)
- return f.per(q), f.per(r)
- def _prem(f, g):
- """Polynomial pseudo-remainder of ``f`` and ``g``. """
- return f.per(dmp_prem(f._rep, g._rep, f.lev, f.dom))
- def _pquo(f, g):
- """Polynomial pseudo-quotient of ``f`` and ``g``. """
- return f.per(dmp_pquo(f._rep, g._rep, f.lev, f.dom))
- def _pexquo(f, g):
- """Polynomial exact pseudo-quotient of ``f`` and ``g``. """
- return f.per(dmp_pexquo(f._rep, g._rep, f.lev, f.dom))
- def _div(f, g):
- """Polynomial division with remainder of ``f`` and ``g``. """
- q, r = dmp_div(f._rep, g._rep, f.lev, f.dom)
- return f.per(q), f.per(r)
- def _rem(f, g):
- """Computes polynomial remainder of ``f`` and ``g``. """
- return f.per(dmp_rem(f._rep, g._rep, f.lev, f.dom))
- def _quo(f, g):
- """Computes polynomial quotient of ``f`` and ``g``. """
- return f.per(dmp_quo(f._rep, g._rep, f.lev, f.dom))
- def _exquo(f, g):
- """Computes polynomial exact quotient of ``f`` and ``g``. """
- return f.per(dmp_exquo(f._rep, g._rep, f.lev, f.dom))
- def _degree(f, j=0):
- """Returns the leading degree of ``f`` in ``x_j``. """
- return dmp_degree_in(f._rep, j, f.lev)
- def degree_list(f):
- """Returns a list of degrees of ``f``. """
- return dmp_degree_list(f._rep, f.lev)
- def total_degree(f):
- """Returns the total degree of ``f``. """
- return max(sum(m) for m in f.monoms())
- def LC(f):
- """Returns the leading coefficient of ``f``. """
- return dmp_ground_LC(f._rep, f.lev, f.dom)
- def TC(f):
- """Returns the trailing coefficient of ``f``. """
- return dmp_ground_TC(f._rep, f.lev, f.dom)
- def _nth(f, N):
- """Returns the ``n``-th coefficient of ``f``. """
- return dmp_ground_nth(f._rep, N, f.lev, f.dom)
- def max_norm(f):
- """Returns maximum norm of ``f``. """
- return dmp_max_norm(f._rep, f.lev, f.dom)
- def l1_norm(f):
- """Returns l1 norm of ``f``. """
- return dmp_l1_norm(f._rep, f.lev, f.dom)
- def l2_norm_squared(f):
- """Return squared l2 norm of ``f``. """
- return dmp_l2_norm_squared(f._rep, f.lev, f.dom)
- def clear_denoms(f):
- """Clear denominators, but keep the ground domain. """
- coeff, F = dmp_clear_denoms(f._rep, f.lev, f.dom)
- return coeff, f.per(F)
- def _integrate(f, m=1, j=0):
- """Computes the ``m``-th order indefinite integral of ``f`` in ``x_j``. """
- return f.per(dmp_integrate_in(f._rep, m, j, f.lev, f.dom))
- def _diff(f, m=1, j=0):
- """Computes the ``m``-th order derivative of ``f`` in ``x_j``. """
- return f.per(dmp_diff_in(f._rep, m, j, f.lev, f.dom))
- def _eval(f, a):
- return dmp_eval_in(f._rep, f.dom.convert(a), 0, f.lev, f.dom)
- def _eval_lev(f, a, j):
- rep = dmp_eval_in(f._rep, f.dom.convert(a), j, f.lev, f.dom)
- return f.new(rep, f.dom, f.lev - 1)
- def _half_gcdex(f, g):
- """Half extended Euclidean algorithm, if univariate. """
- s, h = dup_half_gcdex(f._rep, g._rep, f.dom)
- return f.per(s), f.per(h)
- def _gcdex(f, g):
- """Extended Euclidean algorithm, if univariate. """
- s, t, h = dup_gcdex(f._rep, g._rep, f.dom)
- return f.per(s), f.per(t), f.per(h)
- def _invert(f, g):
- """Invert ``f`` modulo ``g``, if possible. """
- s = dup_invert(f._rep, g._rep, f.dom)
- return f.per(s)
- def _revert(f, n):
- """Compute ``f**(-1)`` mod ``x**n``. """
- return f.per(dup_revert(f._rep, n, f.dom))
- def _subresultants(f, g):
- """Computes subresultant PRS sequence of ``f`` and ``g``. """
- R = dmp_subresultants(f._rep, g._rep, f.lev, f.dom)
- return list(map(f.per, R))
- def _resultant_includePRS(f, g):
- """Computes resultant of ``f`` and ``g`` via PRS. """
- res, R = dmp_resultant(f._rep, g._rep, f.lev, f.dom, includePRS=True)
- if f.lev:
- res = f.new(res, f.dom, f.lev - 1)
- return res, list(map(f.per, R))
- def _resultant(f, g):
- res = dmp_resultant(f._rep, g._rep, f.lev, f.dom)
- if f.lev:
- res = f.new(res, f.dom, f.lev - 1)
- return res
- def discriminant(f):
- """Computes discriminant of ``f``. """
- res = dmp_discriminant(f._rep, f.lev, f.dom)
- if f.lev:
- res = f.new(res, f.dom, f.lev - 1)
- return res
- def _cofactors(f, g):
- """Returns GCD of ``f`` and ``g`` and their cofactors. """
- h, cff, cfg = dmp_inner_gcd(f._rep, g._rep, f.lev, f.dom)
- return f.per(h), f.per(cff), f.per(cfg)
- def _gcd(f, g):
- """Returns polynomial GCD of ``f`` and ``g``. """
- return f.per(dmp_gcd(f._rep, g._rep, f.lev, f.dom))
- def _lcm(f, g):
- """Returns polynomial LCM of ``f`` and ``g``. """
- return f.per(dmp_lcm(f._rep, g._rep, f.lev, f.dom))
- def _cancel(f, g):
- """Cancel common factors in a rational function ``f/g``. """
- cF, cG, F, G = dmp_cancel(f._rep, g._rep, f.lev, f.dom, include=False)
- return cF, cG, f.per(F), f.per(G)
- def _cancel_include(f, g):
- """Cancel common factors in a rational function ``f/g``. """
- F, G = dmp_cancel(f._rep, g._rep, f.lev, f.dom, include=True)
- return f.per(F), f.per(G)
- def _trunc(f, p):
- """Reduce ``f`` modulo a constant ``p``. """
- return f.per(dmp_ground_trunc(f._rep, p, f.lev, f.dom))
- def monic(f):
- """Divides all coefficients by ``LC(f)``. """
- return f.per(dmp_ground_monic(f._rep, f.lev, f.dom))
- def content(f):
- """Returns GCD of polynomial coefficients. """
- return dmp_ground_content(f._rep, f.lev, f.dom)
- def primitive(f):
- """Returns content and a primitive form of ``f``. """
- cont, F = dmp_ground_primitive(f._rep, f.lev, f.dom)
- return cont, f.per(F)
- def _compose(f, g):
- """Computes functional composition of ``f`` and ``g``. """
- return f.per(dmp_compose(f._rep, g._rep, f.lev, f.dom))
- def _decompose(f):
- """Computes functional decomposition of ``f``. """
- return list(map(f.per, dup_decompose(f._rep, f.dom)))
- def _shift(f, a):
- """Efficiently compute Taylor shift ``f(x + a)``. """
- return f.per(dup_shift(f._rep, a, f.dom))
- def _shift_list(f, a):
- """Efficiently compute Taylor shift ``f(X + A)``. """
- return f.per(dmp_shift(f._rep, a, f.lev, f.dom))
- def _transform(f, p, q):
- """Evaluate functional transformation ``q**n * f(p/q)``."""
- return f.per(dup_transform(f._rep, p._rep, q._rep, f.dom))
- def _sturm(f):
- """Computes the Sturm sequence of ``f``. """
- return list(map(f.per, dup_sturm(f._rep, f.dom)))
- def _cauchy_upper_bound(f):
- """Computes the Cauchy upper bound on the roots of ``f``. """
- return dup_cauchy_upper_bound(f._rep, f.dom)
- def _cauchy_lower_bound(f):
- """Computes the Cauchy lower bound on the nonzero roots of ``f``. """
- return dup_cauchy_lower_bound(f._rep, f.dom)
- def _mignotte_sep_bound_squared(f):
- """Computes the squared Mignotte bound on root separations of ``f``. """
- return dup_mignotte_sep_bound_squared(f._rep, f.dom)
- def _gff_list(f):
- """Computes greatest factorial factorization of ``f``. """
- return [ (f.per(g), k) for g, k in dup_gff_list(f._rep, f.dom) ]
- def norm(f):
- """Computes ``Norm(f)``."""
- r = dmp_norm(f._rep, f.lev, f.dom)
- return f.new(r, f.dom.dom, f.lev)
- def sqf_norm(f):
- """Computes square-free norm of ``f``. """
- s, g, r = dmp_sqf_norm(f._rep, f.lev, f.dom)
- return s, f.per(g), f.new(r, f.dom.dom, f.lev)
- def sqf_part(f):
- """Computes square-free part of ``f``. """
- return f.per(dmp_sqf_part(f._rep, f.lev, f.dom))
- def sqf_list(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- coeff, factors = dmp_sqf_list(f._rep, f.lev, f.dom, all)
- return coeff, [ (f.per(g), k) for g, k in factors ]
- def sqf_list_include(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- factors = dmp_sqf_list_include(f._rep, f.lev, f.dom, all)
- return [ (f.per(g), k) for g, k in factors ]
- def factor_list(f):
- """Returns a list of irreducible factors of ``f``. """
- coeff, factors = dmp_factor_list(f._rep, f.lev, f.dom)
- return coeff, [ (f.per(g), k) for g, k in factors ]
- def factor_list_include(f):
- """Returns a list of irreducible factors of ``f``. """
- factors = dmp_factor_list_include(f._rep, f.lev, f.dom)
- return [ (f.per(g), k) for g, k in factors ]
- def _isolate_real_roots(f, eps, inf, sup, fast):
- return dup_isolate_real_roots(f._rep, f.dom, eps=eps, inf=inf, sup=sup, fast=fast)
- def _isolate_real_roots_sqf(f, eps, inf, sup, fast):
- return dup_isolate_real_roots_sqf(f._rep, f.dom, eps=eps, inf=inf, sup=sup, fast=fast)
- def _isolate_all_roots(f, eps, inf, sup, fast):
- return dup_isolate_all_roots(f._rep, f.dom, eps=eps, inf=inf, sup=sup, fast=fast)
- def _isolate_all_roots_sqf(f, eps, inf, sup, fast):
- return dup_isolate_all_roots_sqf(f._rep, f.dom, eps=eps, inf=inf, sup=sup, fast=fast)
- def _refine_real_root(f, s, t, eps, steps, fast):
- return dup_refine_real_root(f._rep, s, t, f.dom, eps=eps, steps=steps, fast=fast)
- def count_real_roots(f, inf=None, sup=None):
- """Return the number of real roots of ``f`` in ``[inf, sup]``. """
- return dup_count_real_roots(f._rep, f.dom, inf=inf, sup=sup)
- def count_complex_roots(f, inf=None, sup=None):
- """Return the number of complex roots of ``f`` in ``[inf, sup]``. """
- return dup_count_complex_roots(f._rep, f.dom, inf=inf, sup=sup)
- @property
- def is_zero(f):
- """Returns ``True`` if ``f`` is a zero polynomial. """
- return dmp_zero_p(f._rep, f.lev)
- @property
- def is_one(f):
- """Returns ``True`` if ``f`` is a unit polynomial. """
- return dmp_one_p(f._rep, f.lev, f.dom)
- @property
- def is_ground(f):
- """Returns ``True`` if ``f`` is an element of the ground domain. """
- return dmp_ground_p(f._rep, None, f.lev)
- @property
- def is_sqf(f):
- """Returns ``True`` if ``f`` is a square-free polynomial. """
- return dmp_sqf_p(f._rep, f.lev, f.dom)
- @property
- def is_monic(f):
- """Returns ``True`` if the leading coefficient of ``f`` is one. """
- return f.dom.is_one(dmp_ground_LC(f._rep, f.lev, f.dom))
- @property
- def is_primitive(f):
- """Returns ``True`` if the GCD of the coefficients of ``f`` is one. """
- return f.dom.is_one(dmp_ground_content(f._rep, f.lev, f.dom))
- @property
- def is_linear(f):
- """Returns ``True`` if ``f`` is linear in all its variables. """
- return all(sum(monom) <= 1 for monom in dmp_to_dict(f._rep, f.lev, f.dom).keys())
- @property
- def is_quadratic(f):
- """Returns ``True`` if ``f`` is quadratic in all its variables. """
- return all(sum(monom) <= 2 for monom in dmp_to_dict(f._rep, f.lev, f.dom).keys())
- @property
- def is_monomial(f):
- """Returns ``True`` if ``f`` is zero or has only one term. """
- return len(f.to_dict()) <= 1
- @property
- def is_homogeneous(f):
- """Returns ``True`` if ``f`` is a homogeneous polynomial. """
- return f.homogeneous_order() is not None
- @property
- def is_irreducible(f):
- """Returns ``True`` if ``f`` has no factors over its domain. """
- return dmp_irreducible_p(f._rep, f.lev, f.dom)
- @property
- def is_cyclotomic(f):
- """Returns ``True`` if ``f`` is a cyclotomic polynomial. """
- if not f.lev:
- return dup_cyclotomic_p(f._rep, f.dom)
- else:
- return False
- class DUP_Flint(DMP):
- """Dense Multivariate Polynomials over `K`. """
- lev = 0
- __slots__ = ('_rep', 'dom', '_cls')
- def __reduce__(self):
- return self.__class__, (self.to_list(), self.dom, self.lev)
- @classmethod
- def _new(cls, rep, dom, lev):
- rep = cls._flint_poly(rep[::-1], dom, lev)
- return cls.from_rep(rep, dom)
- def to_list(f):
- """Convert ``f`` to a list representation with native coefficients. """
- return f._rep.coeffs()[::-1]
- @classmethod
- def _flint_poly(cls, rep, dom, lev):
- assert _supported_flint_domain(dom)
- assert lev == 0
- flint_cls = cls._get_flint_poly_cls(dom)
- return flint_cls(rep)
- @classmethod
- def _get_flint_poly_cls(cls, dom):
- if dom.is_ZZ:
- return flint.fmpz_poly
- elif dom.is_QQ:
- return flint.fmpq_poly
- elif dom.is_FF:
- return dom._poly_ctx
- else:
- raise RuntimeError("Domain %s is not supported with flint" % dom)
- @classmethod
- def from_rep(cls, rep, dom):
- """Create a DMP from the given representation. """
- if dom.is_ZZ:
- assert isinstance(rep, flint.fmpz_poly)
- _cls = flint.fmpz_poly
- elif dom.is_QQ:
- assert isinstance(rep, flint.fmpq_poly)
- _cls = flint.fmpq_poly
- elif dom.is_FF:
- assert isinstance(rep, (flint.nmod_poly, flint.fmpz_mod_poly))
- c = dom.characteristic()
- __cls = type(rep)
- _cls = lambda e: __cls(e, c)
- else:
- raise RuntimeError("Domain %s is not supported with flint" % dom)
- obj = object.__new__(cls)
- obj.dom = dom
- obj._rep = rep
- obj._cls = _cls
- return obj
- def _strict_eq(f, g):
- if type(f) != type(g):
- return False
- return f.dom == g.dom and f._rep == g._rep
- def ground_new(f, coeff):
- """Construct a new ground instance of ``f``. """
- return f.from_rep(f._cls([coeff]), f.dom)
- def _one(f):
- return f.ground_new(f.dom.one)
- def unify(f, g):
- """Unify representations of two polynomials. """
- raise RuntimeError
- def to_DMP_Python(f):
- """Convert ``f`` to a Python native representation. """
- return DMP_Python._new(f.to_list(), f.dom, f.lev)
- def to_tuple(f):
- """Convert ``f`` to a tuple representation with native coefficients. """
- return tuple(f.to_list())
- def _convert(f, dom):
- """Convert the ground domain of ``f``. """
- if dom == QQ and f.dom == ZZ:
- return f.from_rep(flint.fmpq_poly(f._rep), dom)
- elif _supported_flint_domain(dom) and _supported_flint_domain(f.dom):
- # XXX: python-flint should provide a faster way to do this.
- return f.to_DMP_Python()._convert(dom).to_DUP_Flint()
- else:
- raise RuntimeError(f"DUP_Flint: Cannot convert {f.dom} to {dom}")
- def _slice(f, m, n):
- """Take a continuous subsequence of terms of ``f``. """
- coeffs = f._rep.coeffs()[m:n]
- return f.from_rep(f._cls(coeffs), f.dom)
- def _slice_lev(f, m, n, j):
- """Take a continuous subsequence of terms of ``f``. """
- # Only makes sense for multivariate polynomials
- raise NotImplementedError
- def _terms(f, order=None):
- """Returns all non-zero terms from ``f`` in lex order. """
- if order is None or order.alias == 'lex':
- terms = [ ((n,), c) for n, c in enumerate(f._rep.coeffs()) if c ]
- return terms[::-1]
- else:
- # XXX: InverseOrder (ilex) comes here. We could handle that case
- # efficiently by reversing the coefficients but it is not clear
- # how to test if the order is InverseOrder.
- #
- # Otherwise why would the order ever be different for univariate
- # polynomials?
- return f.to_DMP_Python()._terms(order=order)
- def _lift(f):
- """Convert algebraic coefficients to rationals. """
- # This is for algebraic number fields which DUP_Flint does not support
- raise NotImplementedError
- def deflate(f):
- """Reduce degree of `f` by mapping `x_i^m` to `y_i`. """
- # XXX: Check because otherwise this segfaults with python-flint:
- #
- # >>> flint.fmpz_poly([]).deflation()
- # Exception (fmpz_poly_deflate). Division by zero.
- # Aborted (core dumped
- #
- if f.is_zero:
- return (1,), f
- g, n = f._rep.deflation()
- return (n,), f.from_rep(g, f.dom)
- def inject(f, front=False):
- """Inject ground domain generators into ``f``. """
- # Ground domain would need to be a poly ring
- raise NotImplementedError
- def eject(f, dom, front=False):
- """Eject selected generators into the ground domain. """
- # Only makes sense for multivariate polynomials
- raise NotImplementedError
- def _exclude(f):
- """Remove useless generators from ``f``. """
- # Only makes sense for multivariate polynomials
- raise NotImplementedError
- def _permute(f, P):
- """Returns a polynomial in `K[x_{P(1)}, ..., x_{P(n)}]`. """
- # Only makes sense for multivariate polynomials
- raise NotImplementedError
- def terms_gcd(f):
- """Remove GCD of terms from the polynomial ``f``. """
- # XXX: python-flint should have primitive, content, etc methods.
- J, F = f.to_DMP_Python().terms_gcd()
- return J, F.to_DUP_Flint()
- def _add_ground(f, c):
- """Add an element of the ground domain to ``f``. """
- return f.from_rep(f._rep + c, f.dom)
- def _sub_ground(f, c):
- """Subtract an element of the ground domain from ``f``. """
- return f.from_rep(f._rep - c, f.dom)
- def _mul_ground(f, c):
- """Multiply ``f`` by a an element of the ground domain. """
- return f.from_rep(f._rep * c, f.dom)
- def _quo_ground(f, c):
- """Quotient of ``f`` by a an element of the ground domain. """
- return f.from_rep(f._rep // c, f.dom)
- def _exquo_ground(f, c):
- """Exact quotient of ``f`` by an element of the ground domain. """
- q, r = divmod(f._rep, c)
- if r:
- raise ExactQuotientFailed(f, c)
- return f.from_rep(q, f.dom)
- def abs(f):
- """Make all coefficients in ``f`` positive. """
- return f.to_DMP_Python().abs().to_DUP_Flint()
- def neg(f):
- """Negate all coefficients in ``f``. """
- return f.from_rep(-f._rep, f.dom)
- def _add(f, g):
- """Add two multivariate polynomials ``f`` and ``g``. """
- return f.from_rep(f._rep + g._rep, f.dom)
- def _sub(f, g):
- """Subtract two multivariate polynomials ``f`` and ``g``. """
- return f.from_rep(f._rep - g._rep, f.dom)
- def _mul(f, g):
- """Multiply two multivariate polynomials ``f`` and ``g``. """
- return f.from_rep(f._rep * g._rep, f.dom)
- def sqr(f):
- """Square a multivariate polynomial ``f``. """
- return f.from_rep(f._rep ** 2, f.dom)
- def _pow(f, n):
- """Raise ``f`` to a non-negative power ``n``. """
- return f.from_rep(f._rep ** n, f.dom)
- def _pdiv(f, g):
- """Polynomial pseudo-division of ``f`` and ``g``. """
- d = f.degree() - g.degree() + 1
- q, r = divmod(g.LC()**d * f._rep, g._rep)
- return f.from_rep(q, f.dom), f.from_rep(r, f.dom)
- def _prem(f, g):
- """Polynomial pseudo-remainder of ``f`` and ``g``. """
- d = f.degree() - g.degree() + 1
- q = (g.LC()**d * f._rep) % g._rep
- return f.from_rep(q, f.dom)
- def _pquo(f, g):
- """Polynomial pseudo-quotient of ``f`` and ``g``. """
- d = f.degree() - g.degree() + 1
- r = (g.LC()**d * f._rep) // g._rep
- return f.from_rep(r, f.dom)
- def _pexquo(f, g):
- """Polynomial exact pseudo-quotient of ``f`` and ``g``. """
- d = f.degree() - g.degree() + 1
- q, r = divmod(g.LC()**d * f._rep, g._rep)
- if r:
- raise ExactQuotientFailed(f, g)
- return f.from_rep(q, f.dom)
- def _div(f, g):
- """Polynomial division with remainder of ``f`` and ``g``. """
- if f.dom.is_Field:
- q, r = divmod(f._rep, g._rep)
- return f.from_rep(q, f.dom), f.from_rep(r, f.dom)
- else:
- # XXX: python-flint defines division in ZZ[x] differently
- q, r = f.to_DMP_Python()._div(g.to_DMP_Python())
- return q.to_DUP_Flint(), r.to_DUP_Flint()
- def _rem(f, g):
- """Computes polynomial remainder of ``f`` and ``g``. """
- return f.from_rep(f._rep % g._rep, f.dom)
- def _quo(f, g):
- """Computes polynomial quotient of ``f`` and ``g``. """
- return f.from_rep(f._rep // g._rep, f.dom)
- def _exquo(f, g):
- """Computes polynomial exact quotient of ``f`` and ``g``. """
- q, r = f._div(g)
- if r:
- raise ExactQuotientFailed(f, g)
- return q
- def _degree(f, j=0):
- """Returns the leading degree of ``f`` in ``x_j``. """
- d = f._rep.degree()
- if d == -1:
- d = ninf
- return d
- def degree_list(f):
- """Returns a list of degrees of ``f``. """
- return ( f._degree() ,)
- def total_degree(f):
- """Returns the total degree of ``f``. """
- return f._degree()
- def LC(f):
- """Returns the leading coefficient of ``f``. """
- return f._rep[f._rep.degree()]
- def TC(f):
- """Returns the trailing coefficient of ``f``. """
- return f._rep[0]
- def _nth(f, N):
- """Returns the ``n``-th coefficient of ``f``. """
- [n] = N
- return f._rep[n]
- def max_norm(f):
- """Returns maximum norm of ``f``. """
- return f.to_DMP_Python().max_norm()
- def l1_norm(f):
- """Returns l1 norm of ``f``. """
- return f.to_DMP_Python().l1_norm()
- def l2_norm_squared(f):
- """Return squared l2 norm of ``f``. """
- return f.to_DMP_Python().l2_norm_squared()
- def clear_denoms(f):
- """Clear denominators, but keep the ground domain. """
- R = f.dom
- if R.is_QQ:
- denom = f._rep.denom()
- numer = f.from_rep(f._cls(f._rep.numer()), f.dom)
- return denom, numer
- elif R.is_ZZ or R.is_FiniteField:
- return R.one, f
- else:
- raise NotImplementedError
- def _integrate(f, m=1, j=0):
- """Computes the ``m``-th order indefinite integral of ``f`` in ``x_j``. """
- assert j == 0
- if f.dom.is_Field:
- rep = f._rep
- for i in range(m):
- rep = rep.integral()
- return f.from_rep(rep, f.dom)
- else:
- return f.to_DMP_Python()._integrate(m=m, j=j).to_DUP_Flint()
- def _diff(f, m=1, j=0):
- """Computes the ``m``-th order derivative of ``f``. """
- assert j == 0
- rep = f._rep
- for i in range(m):
- rep = rep.derivative()
- return f.from_rep(rep, f.dom)
- def _eval(f, a):
- # XXX: This method is called with many different input types. Ideally
- # we could use e.g. fmpz_poly.__call__ here but more thought needs to
- # go into which types this is supposed to be called with and what types
- # it should return.
- return f.to_DMP_Python()._eval(a)
- def _eval_lev(f, a, j):
- # Only makes sense for multivariate polynomials
- raise NotImplementedError
- def _half_gcdex(f, g):
- """Half extended Euclidean algorithm. """
- s, h = f.to_DMP_Python()._half_gcdex(g.to_DMP_Python())
- return s.to_DUP_Flint(), h.to_DUP_Flint()
- def _gcdex(f, g):
- """Extended Euclidean algorithm. """
- h, s, t = f._rep.xgcd(g._rep)
- return f.from_rep(s, f.dom), f.from_rep(t, f.dom), f.from_rep(h, f.dom)
- def _invert(f, g):
- """Invert ``f`` modulo ``g``, if possible. """
- R = f.dom
- if R.is_Field:
- gcd, F_inv, _ = f._rep.xgcd(g._rep)
- # XXX: Should be gcd != 1 but nmod_poly does not compare equal to
- # other types.
- if gcd != 0*gcd + 1:
- raise NotInvertible("zero divisor")
- return f.from_rep(F_inv, R)
- else:
- # fmpz_poly does not have xgcd or invert and this is not well
- # defined in general.
- return f.to_DMP_Python()._invert(g.to_DMP_Python()).to_DUP_Flint()
- def _revert(f, n):
- """Compute ``f**(-1)`` mod ``x**n``. """
- # XXX: Use fmpz_series etc for reversion?
- # Maybe python-flint should provide revert for fmpz_poly...
- return f.to_DMP_Python()._revert(n).to_DUP_Flint()
- def _subresultants(f, g):
- """Computes subresultant PRS sequence of ``f`` and ``g``. """
- # XXX: Maybe _fmpz_poly_pseudo_rem_cohen could be used...
- R = f.to_DMP_Python()._subresultants(g.to_DMP_Python())
- return [ g.to_DUP_Flint() for g in R ]
- def _resultant_includePRS(f, g):
- """Computes resultant of ``f`` and ``g`` via PRS. """
- # XXX: Maybe _fmpz_poly_pseudo_rem_cohen could be used...
- res, R = f.to_DMP_Python()._resultant_includePRS(g.to_DMP_Python())
- return res, [ g.to_DUP_Flint() for g in R ]
- def _resultant(f, g):
- """Computes resultant of ``f`` and ``g``. """
- # XXX: Use fmpz_mpoly etc when possible...
- return f.to_DMP_Python()._resultant(g.to_DMP_Python())
- def discriminant(f):
- """Computes discriminant of ``f``. """
- # XXX: Use fmpz_mpoly etc when possible...
- return f.to_DMP_Python().discriminant()
- def _cofactors(f, g):
- """Returns GCD of ``f`` and ``g`` and their cofactors. """
- h = f.gcd(g)
- return h, f.exquo(h), g.exquo(h)
- def _gcd(f, g):
- """Returns polynomial GCD of ``f`` and ``g``. """
- return f.from_rep(f._rep.gcd(g._rep), f.dom)
- def _lcm(f, g):
- """Returns polynomial LCM of ``f`` and ``g``. """
- # XXX: python-flint should have a lcm method
- if not (f and g):
- return f.ground_new(f.dom.zero)
- l = f._mul(g)._exquo(f._gcd(g))
- if l.dom.is_Field:
- l = l.monic()
- elif l.LC() < 0:
- l = l.neg()
- return l
- def _cancel(f, g):
- """Cancel common factors in a rational function ``f/g``. """
- assert f.dom == g.dom
- R = f.dom
- # Think carefully about how to handle denominators and coefficient
- # canonicalisation if more domains are permitted...
- assert R.is_ZZ or R.is_QQ or R.is_FiniteField
- if R.is_FiniteField:
- h = f._gcd(g)
- F, G = f.exquo(h), g.exquo(h)
- return R.one, R.one, F, G
- if R.is_QQ:
- cG, F = f.clear_denoms()
- cF, G = g.clear_denoms()
- else:
- cG, F = R.one, f
- cF, G = R.one, g
- cH = cF.gcd(cG)
- cF, cG = cF // cH, cG // cH
- H = F._gcd(G)
- F, G = F.exquo(H), G.exquo(H)
- f_neg = F.LC() < 0
- g_neg = G.LC() < 0
- if f_neg and g_neg:
- F, G = F.neg(), G.neg()
- elif f_neg:
- cF, F = -cF, F.neg()
- elif g_neg:
- cF, G = -cF, G.neg()
- return cF, cG, F, G
- def _cancel_include(f, g):
- """Cancel common factors in a rational function ``f/g``. """
- cF, cG, F, G = f._cancel(g)
- return F._mul_ground(cF), G._mul_ground(cG)
- def _trunc(f, p):
- """Reduce ``f`` modulo a constant ``p``. """
- return f.to_DMP_Python()._trunc(p).to_DUP_Flint()
- def monic(f):
- """Divides all coefficients by ``LC(f)``. """
- # XXX: python-flint should add monic
- return f._exquo_ground(f.LC())
- def content(f):
- """Returns GCD of polynomial coefficients. """
- # XXX: python-flint should have a content method
- return f.to_DMP_Python().content()
- def primitive(f):
- """Returns content and a primitive form of ``f``. """
- cont = f.content()
- if f.is_zero:
- return f.dom.zero, f
- prim = f._exquo_ground(cont)
- return cont, prim
- def _compose(f, g):
- """Computes functional composition of ``f`` and ``g``. """
- return f.from_rep(f._rep(g._rep), f.dom)
- def _decompose(f):
- """Computes functional decomposition of ``f``. """
- return [ g.to_DUP_Flint() for g in f.to_DMP_Python()._decompose() ]
- def _shift(f, a):
- """Efficiently compute Taylor shift ``f(x + a)``. """
- x_plus_a = f._cls([a, f.dom.one])
- return f.from_rep(f._rep(x_plus_a), f.dom)
- def _transform(f, p, q):
- """Evaluate functional transformation ``q**n * f(p/q)``."""
- F, P, Q = f.to_DMP_Python(), p.to_DMP_Python(), q.to_DMP_Python()
- return F.transform(P, Q).to_DUP_Flint()
- def _sturm(f):
- """Computes the Sturm sequence of ``f``. """
- return [ g.to_DUP_Flint() for g in f.to_DMP_Python()._sturm() ]
- def _cauchy_upper_bound(f):
- """Computes the Cauchy upper bound on the roots of ``f``. """
- return f.to_DMP_Python()._cauchy_upper_bound()
- def _cauchy_lower_bound(f):
- """Computes the Cauchy lower bound on the nonzero roots of ``f``. """
- return f.to_DMP_Python()._cauchy_lower_bound()
- def _mignotte_sep_bound_squared(f):
- """Computes the squared Mignotte bound on root separations of ``f``. """
- return f.to_DMP_Python()._mignotte_sep_bound_squared()
- def _gff_list(f):
- """Computes greatest factorial factorization of ``f``. """
- F = f.to_DMP_Python()
- return [ (g.to_DUP_Flint(), k) for g, k in F.gff_list() ]
- def norm(f):
- """Computes ``Norm(f)``."""
- # This is for algebraic number fields which DUP_Flint does not support
- raise NotImplementedError
- def sqf_norm(f):
- """Computes square-free norm of ``f``. """
- # This is for algebraic number fields which DUP_Flint does not support
- raise NotImplementedError
- def sqf_part(f):
- """Computes square-free part of ``f``. """
- return f._exquo(f._gcd(f._diff()))
- def sqf_list(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- # XXX: python-flint should provide square free factorisation.
- coeff, factors = f.to_DMP_Python().sqf_list(all=all)
- return coeff, [ (g.to_DUP_Flint(), k) for g, k in factors ]
- def sqf_list_include(f, all=False):
- """Returns a list of square-free factors of ``f``. """
- factors = f.to_DMP_Python().sqf_list_include(all=all)
- return [ (g.to_DUP_Flint(), k) for g, k in factors ]
- def factor_list(f):
- """Returns a list of irreducible factors of ``f``. """
- if f.dom.is_ZZ or f.dom.is_FF:
- # python-flint matches polys here
- coeff, factors = f._rep.factor()
- factors = [ (f.from_rep(g, f.dom), k) for g, k in factors ]
- elif f.dom.is_QQ:
- # python-flint returns monic factors over QQ whereas polys returns
- # denominator free factors.
- coeff, factors = f._rep.factor()
- factors_monic = [ (f.from_rep(g, f.dom), k) for g, k in factors ]
- # Absorb the denominators into coeff
- factors = []
- for g, k in factors_monic:
- d, g = g.clear_denoms()
- coeff /= d**k
- factors.append((g, k))
- else:
- # Check carefully when adding more domains here...
- raise RuntimeError("Domain %s is not supported with flint" % f.dom)
- # We need to match the way that polys orders the factors
- factors = f._sort_factors(factors)
- return coeff, factors
- def factor_list_include(f):
- """Returns a list of irreducible factors of ``f``. """
- # XXX: factor_list_include seems to be broken in general:
- #
- # >>> Poly(2*(x - 1)**3, x).factor_list_include()
- # [(Poly(2*x - 2, x, domain='ZZ'), 3)]
- #
- # Let's not try to implement it here.
- factors = f.to_DMP_Python().factor_list_include()
- return [ (g.to_DUP_Flint(), k) for g, k in factors ]
- def _sort_factors(f, factors):
- """Sort a list of factors to canonical order. """
- # Convert the factors to lists and use _sort_factors from polys
- factors = [ (g.to_list(), k) for g, k in factors ]
- factors = _sort_factors(factors, multiple=True)
- to_dup_flint = lambda g: f.from_rep(f._cls(g[::-1]), f.dom)
- return [ (to_dup_flint(g), k) for g, k in factors ]
- def _isolate_real_roots(f, eps, inf, sup, fast):
- return f.to_DMP_Python()._isolate_real_roots(eps, inf, sup, fast)
- def _isolate_real_roots_sqf(f, eps, inf, sup, fast):
- return f.to_DMP_Python()._isolate_real_roots_sqf(eps, inf, sup, fast)
- def _isolate_all_roots(f, eps, inf, sup, fast):
- # fmpz_poly and fmpq_poly have a complex_roots method that could be
- # used here. It probably makes more sense to add analogous methods in
- # python-flint though.
- return f.to_DMP_Python()._isolate_all_roots(eps, inf, sup, fast)
- def _isolate_all_roots_sqf(f, eps, inf, sup, fast):
- return f.to_DMP_Python()._isolate_all_roots_sqf(eps, inf, sup, fast)
- def _refine_real_root(f, s, t, eps, steps, fast):
- return f.to_DMP_Python()._refine_real_root(s, t, eps, steps, fast)
- def count_real_roots(f, inf=None, sup=None):
- """Return the number of real roots of ``f`` in ``[inf, sup]``. """
- return f.to_DMP_Python().count_real_roots(inf=inf, sup=sup)
- def count_complex_roots(f, inf=None, sup=None):
- """Return the number of complex roots of ``f`` in ``[inf, sup]``. """
- return f.to_DMP_Python().count_complex_roots(inf=inf, sup=sup)
- @property
- def is_zero(f):
- """Returns ``True`` if ``f`` is a zero polynomial. """
- return not f._rep
- @property
- def is_one(f):
- """Returns ``True`` if ``f`` is a unit polynomial. """
- return f._rep == f.dom.one
- @property
- def is_ground(f):
- """Returns ``True`` if ``f`` is an element of the ground domain. """
- return f._rep.degree() <= 0
- @property
- def is_linear(f):
- """Returns ``True`` if ``f`` is linear in all its variables. """
- return f._rep.degree() <= 1
- @property
- def is_quadratic(f):
- """Returns ``True`` if ``f`` is quadratic in all its variables. """
- return f._rep.degree() <= 2
- @property
- def is_monomial(f):
- """Returns ``True`` if ``f`` is zero or has only one term. """
- fr = f._rep
- return fr.degree() < 0 or not any(fr[n] for n in range(fr.degree()))
- @property
- def is_monic(f):
- """Returns ``True`` if the leading coefficient of ``f`` is one. """
- return f.LC() == f.dom.one
- @property
- def is_primitive(f):
- """Returns ``True`` if the GCD of the coefficients of ``f`` is one. """
- return f.to_DMP_Python().is_primitive
- @property
- def is_homogeneous(f):
- """Returns ``True`` if ``f`` is a homogeneous polynomial. """
- return f.to_DMP_Python().is_homogeneous
- @property
- def is_sqf(f):
- """Returns ``True`` if ``f`` is a square-free polynomial. """
- g = f._rep.gcd(f._rep.derivative())
- return g.degree() <= 0
- @property
- def is_irreducible(f):
- """Returns ``True`` if ``f`` has no factors over its domain. """
- _, factors = f._rep.factor()
- if len(factors) == 0:
- return True
- elif len(factors) == 1:
- return factors[0][1] == 1
- else:
- return False
- @property
- def is_cyclotomic(f):
- """Returns ``True`` if ``f`` is a cyclotomic polynomial. """
- if f.dom.is_QQ:
- try:
- f = f.convert(ZZ)
- except CoercionFailed:
- return False
- if f.dom.is_ZZ:
- return bool(f._rep.is_cyclotomic())
- else:
- # This is what dup_cyclotomic_p does...
- return False
- def init_normal_DMF(num, den, lev, dom):
- return DMF(dmp_normal(num, lev, dom),
- dmp_normal(den, lev, dom), dom, lev)
- class DMF(PicklableWithSlots, CantSympify):
- """Dense Multivariate Fractions over `K`. """
- __slots__ = ('num', 'den', 'lev', 'dom')
- def __init__(self, rep, dom, lev=None):
- num, den, lev = self._parse(rep, dom, lev)
- num, den = dmp_cancel(num, den, lev, dom)
- self.num = num
- self.den = den
- self.lev = lev
- self.dom = dom
- @classmethod
- def new(cls, rep, dom, lev=None):
- num, den, lev = cls._parse(rep, dom, lev)
- obj = object.__new__(cls)
- obj.num = num
- obj.den = den
- obj.lev = lev
- obj.dom = dom
- return obj
- def ground_new(self, rep):
- return self.new(rep, self.dom, self.lev)
- @classmethod
- def _parse(cls, rep, dom, lev=None):
- if isinstance(rep, tuple):
- num, den = rep
- if lev is not None:
- if isinstance(num, dict):
- num = dmp_from_dict(num, lev, dom)
- if isinstance(den, dict):
- den = dmp_from_dict(den, lev, dom)
- else:
- num, num_lev = dmp_validate(num)
- den, den_lev = dmp_validate(den)
- if num_lev == den_lev:
- lev = num_lev
- else:
- raise ValueError('inconsistent number of levels')
- if dmp_zero_p(den, lev):
- raise ZeroDivisionError('fraction denominator')
- if dmp_zero_p(num, lev):
- den = dmp_one(lev, dom)
- else:
- if dmp_negative_p(den, lev, dom):
- num = dmp_neg(num, lev, dom)
- den = dmp_neg(den, lev, dom)
- else:
- num = rep
- if lev is not None:
- if isinstance(num, dict):
- num = dmp_from_dict(num, lev, dom)
- elif not isinstance(num, list):
- num = dmp_ground(dom.convert(num), lev)
- else:
- num, lev = dmp_validate(num)
- den = dmp_one(lev, dom)
- return num, den, lev
- def __repr__(f):
- return "%s((%s, %s), %s)" % (f.__class__.__name__, f.num, f.den, f.dom)
- def __hash__(f):
- return hash((f.__class__.__name__, dmp_to_tuple(f.num, f.lev),
- dmp_to_tuple(f.den, f.lev), f.lev, f.dom))
- def poly_unify(f, g):
- """Unify a multivariate fraction and a polynomial. """
- if not isinstance(g, DMP) or f.lev != g.lev:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- if f.dom == g.dom:
- return (f.lev, f.dom, f.per, (f.num, f.den), g._rep)
- else:
- lev, dom = f.lev, f.dom.unify(g.dom)
- F = (dmp_convert(f.num, lev, f.dom, dom),
- dmp_convert(f.den, lev, f.dom, dom))
- G = dmp_convert(g._rep, lev, g.dom, dom)
- def per(num, den, cancel=True, kill=False, lev=lev):
- if kill:
- if not lev:
- return num/den
- else:
- lev = lev - 1
- if cancel:
- num, den = dmp_cancel(num, den, lev, dom)
- return f.__class__.new((num, den), dom, lev)
- return lev, dom, per, F, G
- def frac_unify(f, g):
- """Unify representations of two multivariate fractions. """
- if not isinstance(g, DMF) or f.lev != g.lev:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- if f.dom == g.dom:
- return (f.lev, f.dom, f.per, (f.num, f.den),
- (g.num, g.den))
- else:
- lev, dom = f.lev, f.dom.unify(g.dom)
- F = (dmp_convert(f.num, lev, f.dom, dom),
- dmp_convert(f.den, lev, f.dom, dom))
- G = (dmp_convert(g.num, lev, g.dom, dom),
- dmp_convert(g.den, lev, g.dom, dom))
- def per(num, den, cancel=True, kill=False, lev=lev):
- if kill:
- if not lev:
- return num/den
- else:
- lev = lev - 1
- if cancel:
- num, den = dmp_cancel(num, den, lev, dom)
- return f.__class__.new((num, den), dom, lev)
- return lev, dom, per, F, G
- def per(f, num, den, cancel=True, kill=False):
- """Create a DMF out of the given representation. """
- lev, dom = f.lev, f.dom
- if kill:
- if not lev:
- return num/den
- else:
- lev -= 1
- if cancel:
- num, den = dmp_cancel(num, den, lev, dom)
- return f.__class__.new((num, den), dom, lev)
- def half_per(f, rep, kill=False):
- """Create a DMP out of the given representation. """
- lev = f.lev
- if kill:
- if not lev:
- return rep
- else:
- lev -= 1
- return DMP(rep, f.dom, lev)
- @classmethod
- def zero(cls, lev, dom):
- return cls.new(0, dom, lev)
- @classmethod
- def one(cls, lev, dom):
- return cls.new(1, dom, lev)
- def numer(f):
- """Returns the numerator of ``f``. """
- return f.half_per(f.num)
- def denom(f):
- """Returns the denominator of ``f``. """
- return f.half_per(f.den)
- def cancel(f):
- """Remove common factors from ``f.num`` and ``f.den``. """
- return f.per(f.num, f.den)
- def neg(f):
- """Negate all coefficients in ``f``. """
- return f.per(dmp_neg(f.num, f.lev, f.dom), f.den, cancel=False)
- def add_ground(f, c):
- """Add an element of the ground domain to ``f``. """
- return f + f.ground_new(c)
- def add(f, g):
- """Add two multivariate fractions ``f`` and ``g``. """
- if isinstance(g, DMP):
- lev, dom, per, (F_num, F_den), G = f.poly_unify(g)
- num, den = dmp_add_mul(F_num, F_den, G, lev, dom), F_den
- else:
- lev, dom, per, F, G = f.frac_unify(g)
- (F_num, F_den), (G_num, G_den) = F, G
- num = dmp_add(dmp_mul(F_num, G_den, lev, dom),
- dmp_mul(F_den, G_num, lev, dom), lev, dom)
- den = dmp_mul(F_den, G_den, lev, dom)
- return per(num, den)
- def sub(f, g):
- """Subtract two multivariate fractions ``f`` and ``g``. """
- if isinstance(g, DMP):
- lev, dom, per, (F_num, F_den), G = f.poly_unify(g)
- num, den = dmp_sub_mul(F_num, F_den, G, lev, dom), F_den
- else:
- lev, dom, per, F, G = f.frac_unify(g)
- (F_num, F_den), (G_num, G_den) = F, G
- num = dmp_sub(dmp_mul(F_num, G_den, lev, dom),
- dmp_mul(F_den, G_num, lev, dom), lev, dom)
- den = dmp_mul(F_den, G_den, lev, dom)
- return per(num, den)
- def mul(f, g):
- """Multiply two multivariate fractions ``f`` and ``g``. """
- if isinstance(g, DMP):
- lev, dom, per, (F_num, F_den), G = f.poly_unify(g)
- num, den = dmp_mul(F_num, G, lev, dom), F_den
- else:
- lev, dom, per, F, G = f.frac_unify(g)
- (F_num, F_den), (G_num, G_den) = F, G
- num = dmp_mul(F_num, G_num, lev, dom)
- den = dmp_mul(F_den, G_den, lev, dom)
- return per(num, den)
- def pow(f, n):
- """Raise ``f`` to a non-negative power ``n``. """
- if isinstance(n, int):
- num, den = f.num, f.den
- if n < 0:
- num, den, n = den, num, -n
- return f.per(dmp_pow(num, n, f.lev, f.dom),
- dmp_pow(den, n, f.lev, f.dom), cancel=False)
- else:
- raise TypeError("``int`` expected, got %s" % type(n))
- def quo(f, g):
- """Computes quotient of fractions ``f`` and ``g``. """
- if isinstance(g, DMP):
- lev, dom, per, (F_num, F_den), G = f.poly_unify(g)
- num, den = F_num, dmp_mul(F_den, G, lev, dom)
- else:
- lev, dom, per, F, G = f.frac_unify(g)
- (F_num, F_den), (G_num, G_den) = F, G
- num = dmp_mul(F_num, G_den, lev, dom)
- den = dmp_mul(F_den, G_num, lev, dom)
- return per(num, den)
- exquo = quo
- def invert(f, check=True):
- """Computes inverse of a fraction ``f``. """
- return f.per(f.den, f.num, cancel=False)
- @property
- def is_zero(f):
- """Returns ``True`` if ``f`` is a zero fraction. """
- return dmp_zero_p(f.num, f.lev)
- @property
- def is_one(f):
- """Returns ``True`` if ``f`` is a unit fraction. """
- return dmp_one_p(f.num, f.lev, f.dom) and \
- dmp_one_p(f.den, f.lev, f.dom)
- def __neg__(f):
- return f.neg()
- def __add__(f, g):
- if isinstance(g, (DMP, DMF)):
- return f.add(g)
- elif g in f.dom:
- return f.add_ground(f.dom.convert(g))
- try:
- return f.add(f.half_per(g))
- except (TypeError, CoercionFailed, NotImplementedError):
- return NotImplemented
- def __radd__(f, g):
- return f.__add__(g)
- def __sub__(f, g):
- if isinstance(g, (DMP, DMF)):
- return f.sub(g)
- try:
- return f.sub(f.half_per(g))
- except (TypeError, CoercionFailed, NotImplementedError):
- return NotImplemented
- def __rsub__(f, g):
- return (-f).__add__(g)
- def __mul__(f, g):
- if isinstance(g, (DMP, DMF)):
- return f.mul(g)
- try:
- return f.mul(f.half_per(g))
- except (TypeError, CoercionFailed, NotImplementedError):
- return NotImplemented
- def __rmul__(f, g):
- return f.__mul__(g)
- def __pow__(f, n):
- return f.pow(n)
- def __truediv__(f, g):
- if isinstance(g, (DMP, DMF)):
- return f.quo(g)
- try:
- return f.quo(f.half_per(g))
- except (TypeError, CoercionFailed, NotImplementedError):
- return NotImplemented
- def __rtruediv__(self, g):
- return self.invert(check=False)*g
- def __eq__(f, g):
- try:
- if isinstance(g, DMP):
- _, _, _, (F_num, F_den), G = f.poly_unify(g)
- if f.lev == g.lev:
- return dmp_one_p(F_den, f.lev, f.dom) and F_num == G
- else:
- _, _, _, F, G = f.frac_unify(g)
- if f.lev == g.lev:
- return F == G
- except UnificationFailed:
- pass
- return False
- def __ne__(f, g):
- try:
- if isinstance(g, DMP):
- _, _, _, (F_num, F_den), G = f.poly_unify(g)
- if f.lev == g.lev:
- return not (dmp_one_p(F_den, f.lev, f.dom) and F_num == G)
- else:
- _, _, _, F, G = f.frac_unify(g)
- if f.lev == g.lev:
- return F != G
- except UnificationFailed:
- pass
- return True
- def __lt__(f, g):
- _, _, _, F, G = f.frac_unify(g)
- return F < G
- def __le__(f, g):
- _, _, _, F, G = f.frac_unify(g)
- return F <= G
- def __gt__(f, g):
- _, _, _, F, G = f.frac_unify(g)
- return F > G
- def __ge__(f, g):
- _, _, _, F, G = f.frac_unify(g)
- return F >= G
- def __bool__(f):
- return not dmp_zero_p(f.num, f.lev)
- def init_normal_ANP(rep, mod, dom):
- return ANP(dup_normal(rep, dom),
- dup_normal(mod, dom), dom)
- class ANP(CantSympify):
- """Dense Algebraic Number Polynomials over a field. """
- __slots__ = ('_rep', '_mod', 'dom')
- def __new__(cls, rep, mod, dom):
- if isinstance(rep, DMP):
- pass
- elif type(rep) is dict: # don't use isinstance
- rep = DMP(dup_from_dict(rep, dom), dom, 0)
- else:
- if isinstance(rep, list):
- rep = [dom.convert(a) for a in rep]
- else:
- rep = [dom.convert(rep)]
- rep = DMP(dup_strip(rep), dom, 0)
- if isinstance(mod, DMP):
- pass
- elif isinstance(mod, dict):
- mod = DMP(dup_from_dict(mod, dom), dom, 0)
- else:
- mod = DMP(dup_strip(mod), dom, 0)
- return cls.new(rep, mod, dom)
- @classmethod
- def new(cls, rep, mod, dom):
- if not (rep.dom == mod.dom == dom):
- raise RuntimeError("Inconsistent domain")
- obj = super().__new__(cls)
- obj._rep = rep
- obj._mod = mod
- obj.dom = dom
- return obj
- # XXX: It should be possible to use __getnewargs__ rather than __reduce__
- # but it doesn't work for some reason. Probably this would be easier if
- # python-flint supported pickling for polynomial types.
- def __reduce__(self):
- return ANP, (self.rep, self.mod, self.dom)
- @property
- def rep(self):
- return self._rep.to_list()
- @property
- def mod(self):
- return self.mod_to_list()
- def to_DMP(self):
- return self._rep
- def mod_to_DMP(self):
- return self._mod
- def per(f, rep):
- return f.new(rep, f._mod, f.dom)
- def __repr__(f):
- return "%s(%s, %s, %s)" % (f.__class__.__name__, f._rep.to_list(), f._mod.to_list(), f.dom)
- def __hash__(f):
- return hash((f.__class__.__name__, f.to_tuple(), f._mod.to_tuple(), f.dom))
- def convert(f, dom):
- """Convert ``f`` to a ``ANP`` over a new domain. """
- if f.dom == dom:
- return f
- else:
- return f.new(f._rep.convert(dom), f._mod.convert(dom), dom)
- def unify(f, g):
- """Unify representations of two algebraic numbers. """
- # XXX: This unify method is not used any more because unify_ANP is used
- # instead.
- if not isinstance(g, ANP) or f.mod != g.mod:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- if f.dom == g.dom:
- return f.dom, f.per, f.rep, g.rep, f.mod
- else:
- dom = f.dom.unify(g.dom)
- F = dup_convert(f.rep, f.dom, dom)
- G = dup_convert(g.rep, g.dom, dom)
- if dom != f.dom and dom != g.dom:
- mod = dup_convert(f.mod, f.dom, dom)
- else:
- if dom == f.dom:
- mod = f.mod
- else:
- mod = g.mod
- per = lambda rep: ANP(rep, mod, dom)
- return dom, per, F, G, mod
- def unify_ANP(f, g):
- """Unify and return ``DMP`` instances of ``f`` and ``g``. """
- if not isinstance(g, ANP) or f._mod != g._mod:
- raise UnificationFailed("Cannot unify %s with %s" % (f, g))
- # The domain is almost always QQ but there are some tests involving ZZ
- if f.dom != g.dom:
- dom = f.dom.unify(g.dom)
- f = f.convert(dom)
- g = g.convert(dom)
- return f._rep, g._rep, f._mod, f.dom
- @classmethod
- def zero(cls, mod, dom):
- return ANP(0, mod, dom)
- @classmethod
- def one(cls, mod, dom):
- return ANP(1, mod, dom)
- def to_dict(f):
- """Convert ``f`` to a dict representation with native coefficients. """
- return f._rep.to_dict()
- def to_sympy_dict(f):
- """Convert ``f`` to a dict representation with SymPy coefficients. """
- rep = dmp_to_dict(f.rep, 0, f.dom)
- for k, v in rep.items():
- rep[k] = f.dom.to_sympy(v)
- return rep
- def to_list(f):
- """Convert ``f`` to a list representation with native coefficients. """
- return f._rep.to_list()
- def mod_to_list(f):
- """Return ``f.mod`` as a list with native coefficients. """
- return f._mod.to_list()
- def to_sympy_list(f):
- """Convert ``f`` to a list representation with SymPy coefficients. """
- return [ f.dom.to_sympy(c) for c in f.to_list() ]
- def to_tuple(f):
- """
- Convert ``f`` to a tuple representation with native coefficients.
- This is needed for hashing.
- """
- return f._rep.to_tuple()
- @classmethod
- def from_list(cls, rep, mod, dom):
- return ANP(dup_strip(list(map(dom.convert, rep))), mod, dom)
- def add_ground(f, c):
- """Add an element of the ground domain to ``f``. """
- return f.per(f._rep.add_ground(c))
- def sub_ground(f, c):
- """Subtract an element of the ground domain from ``f``. """
- return f.per(f._rep.sub_ground(c))
- def mul_ground(f, c):
- """Multiply ``f`` by an element of the ground domain. """
- return f.per(f._rep.mul_ground(c))
- def quo_ground(f, c):
- """Quotient of ``f`` by an element of the ground domain. """
- return f.per(f._rep.quo_ground(c))
- def neg(f):
- return f.per(f._rep.neg())
- def add(f, g):
- F, G, mod, dom = f.unify_ANP(g)
- return f.new(F.add(G), mod, dom)
- def sub(f, g):
- F, G, mod, dom = f.unify_ANP(g)
- return f.new(F.sub(G), mod, dom)
- def mul(f, g):
- F, G, mod, dom = f.unify_ANP(g)
- return f.new(F.mul(G).rem(mod), mod, dom)
- def pow(f, n):
- """Raise ``f`` to a non-negative power ``n``. """
- if not isinstance(n, int):
- raise TypeError("``int`` expected, got %s" % type(n))
- mod = f._mod
- F = f._rep
- if n < 0:
- F, n = F.invert(mod), -n
- # XXX: Need a pow_mod method for DMP
- return f.new(F.pow(n).rem(f._mod), mod, f.dom)
- def exquo(f, g):
- F, G, mod, dom = f.unify_ANP(g)
- return f.new(F.mul(G.invert(mod)).rem(mod), mod, dom)
- def div(f, g):
- return f.exquo(g), f.zero(f._mod, f.dom)
- def quo(f, g):
- return f.exquo(g)
- def rem(f, g):
- F, G, mod, dom = f.unify_ANP(g)
- s, h = F.half_gcdex(G)
- if h.is_one:
- return f.zero(mod, dom)
- else:
- raise NotInvertible("zero divisor")
- def LC(f):
- """Returns the leading coefficient of ``f``. """
- return f._rep.LC()
- def TC(f):
- """Returns the trailing coefficient of ``f``. """
- return f._rep.TC()
- @property
- def is_zero(f):
- """Returns ``True`` if ``f`` is a zero algebraic number. """
- return f._rep.is_zero
- @property
- def is_one(f):
- """Returns ``True`` if ``f`` is a unit algebraic number. """
- return f._rep.is_one
- @property
- def is_ground(f):
- """Returns ``True`` if ``f`` is an element of the ground domain. """
- return f._rep.is_ground
- def __pos__(f):
- return f
- def __neg__(f):
- return f.neg()
- def __add__(f, g):
- if isinstance(g, ANP):
- return f.add(g)
- try:
- g = f.dom.convert(g)
- except CoercionFailed:
- return NotImplemented
- else:
- return f.add_ground(g)
- def __radd__(f, g):
- return f.__add__(g)
- def __sub__(f, g):
- if isinstance(g, ANP):
- return f.sub(g)
- try:
- g = f.dom.convert(g)
- except CoercionFailed:
- return NotImplemented
- else:
- return f.sub_ground(g)
- def __rsub__(f, g):
- return (-f).__add__(g)
- def __mul__(f, g):
- if isinstance(g, ANP):
- return f.mul(g)
- try:
- g = f.dom.convert(g)
- except CoercionFailed:
- return NotImplemented
- else:
- return f.mul_ground(g)
- def __rmul__(f, g):
- return f.__mul__(g)
- def __pow__(f, n):
- return f.pow(n)
- def __divmod__(f, g):
- return f.div(g)
- def __mod__(f, g):
- return f.rem(g)
- def __truediv__(f, g):
- if isinstance(g, ANP):
- return f.quo(g)
- try:
- g = f.dom.convert(g)
- except CoercionFailed:
- return NotImplemented
- else:
- return f.quo_ground(g)
- def __eq__(f, g):
- try:
- F, G, _, _ = f.unify_ANP(g)
- except UnificationFailed:
- return NotImplemented
- return F == G
- def __ne__(f, g):
- try:
- F, G, _, _ = f.unify_ANP(g)
- except UnificationFailed:
- return NotImplemented
- return F != G
- def __lt__(f, g):
- F, G, _, _ = f.unify_ANP(g)
- return F < G
- def __le__(f, g):
- F, G, _, _ = f.unify_ANP(g)
- return F <= G
- def __gt__(f, g):
- F, G, _, _ = f.unify_ANP(g)
- return F > G
- def __ge__(f, g):
- F, G, _, _ = f.unify_ANP(g)
- return F >= G
- def __bool__(f):
- return bool(f._rep)
|