fractionfield.py 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181
  1. """Implementation of :class:`FractionField` class. """
  2. from sympy.polys.domains.compositedomain import CompositeDomain
  3. from sympy.polys.domains.field import Field
  4. from sympy.polys.polyerrors import CoercionFailed, GeneratorsError
  5. from sympy.utilities import public
  6. @public
  7. class FractionField(Field, CompositeDomain):
  8. """A class for representing multivariate rational function fields. """
  9. is_FractionField = is_Frac = True
  10. has_assoc_Ring = True
  11. has_assoc_Field = True
  12. def __init__(self, domain_or_field, symbols=None, order=None):
  13. from sympy.polys.fields import FracField
  14. if isinstance(domain_or_field, FracField) and symbols is None and order is None:
  15. field = domain_or_field
  16. else:
  17. field = FracField(symbols, domain_or_field, order)
  18. self.field = field
  19. self.dtype = field.dtype
  20. self.gens = field.gens
  21. self.ngens = field.ngens
  22. self.symbols = field.symbols
  23. self.domain = field.domain
  24. # TODO: remove this
  25. self.dom = self.domain
  26. def new(self, element):
  27. return self.field.field_new(element)
  28. def of_type(self, element):
  29. """Check if ``a`` is of type ``dtype``. """
  30. return self.field.is_element(element)
  31. @property
  32. def zero(self):
  33. return self.field.zero
  34. @property
  35. def one(self):
  36. return self.field.one
  37. @property
  38. def order(self):
  39. return self.field.order
  40. def __str__(self):
  41. return str(self.domain) + '(' + ','.join(map(str, self.symbols)) + ')'
  42. def __hash__(self):
  43. return hash((self.__class__.__name__, self.field, self.domain, self.symbols))
  44. def __eq__(self, other):
  45. """Returns ``True`` if two domains are equivalent. """
  46. if not isinstance(other, FractionField):
  47. return NotImplemented
  48. return self.field == other.field
  49. def to_sympy(self, a):
  50. """Convert ``a`` to a SymPy object. """
  51. return a.as_expr()
  52. def from_sympy(self, a):
  53. """Convert SymPy's expression to ``dtype``. """
  54. return self.field.from_expr(a)
  55. def from_ZZ(K1, a, K0):
  56. """Convert a Python ``int`` object to ``dtype``. """
  57. return K1(K1.domain.convert(a, K0))
  58. def from_ZZ_python(K1, a, K0):
  59. """Convert a Python ``int`` object to ``dtype``. """
  60. return K1(K1.domain.convert(a, K0))
  61. def from_QQ(K1, a, K0):
  62. """Convert a Python ``Fraction`` object to ``dtype``. """
  63. dom = K1.domain
  64. conv = dom.convert_from
  65. if dom.is_ZZ:
  66. return K1(conv(K0.numer(a), K0)) / K1(conv(K0.denom(a), K0))
  67. else:
  68. return K1(conv(a, K0))
  69. def from_QQ_python(K1, a, K0):
  70. """Convert a Python ``Fraction`` object to ``dtype``. """
  71. return K1(K1.domain.convert(a, K0))
  72. def from_ZZ_gmpy(K1, a, K0):
  73. """Convert a GMPY ``mpz`` object to ``dtype``. """
  74. return K1(K1.domain.convert(a, K0))
  75. def from_QQ_gmpy(K1, a, K0):
  76. """Convert a GMPY ``mpq`` object to ``dtype``. """
  77. return K1(K1.domain.convert(a, K0))
  78. def from_GaussianRationalField(K1, a, K0):
  79. """Convert a ``GaussianRational`` object to ``dtype``. """
  80. return K1(K1.domain.convert(a, K0))
  81. def from_GaussianIntegerRing(K1, a, K0):
  82. """Convert a ``GaussianInteger`` object to ``dtype``. """
  83. return K1(K1.domain.convert(a, K0))
  84. def from_RealField(K1, a, K0):
  85. """Convert a mpmath ``mpf`` object to ``dtype``. """
  86. return K1(K1.domain.convert(a, K0))
  87. def from_ComplexField(K1, a, K0):
  88. """Convert a mpmath ``mpf`` object to ``dtype``. """
  89. return K1(K1.domain.convert(a, K0))
  90. def from_AlgebraicField(K1, a, K0):
  91. """Convert an algebraic number to ``dtype``. """
  92. if K1.domain != K0:
  93. a = K1.domain.convert_from(a, K0)
  94. if a is not None:
  95. return K1.new(a)
  96. def from_PolynomialRing(K1, a, K0):
  97. """Convert a polynomial to ``dtype``. """
  98. if a.is_ground:
  99. return K1.convert_from(a.coeff(1), K0.domain)
  100. try:
  101. return K1.new(a.set_ring(K1.field.ring))
  102. except (CoercionFailed, GeneratorsError):
  103. # XXX: We get here if K1=ZZ(x,y) and K0=QQ[x,y]
  104. # and the poly a in K0 has non-integer coefficients.
  105. # It seems that K1.new can handle this but K1.new doesn't work
  106. # when K0.domain is an algebraic field...
  107. try:
  108. return K1.new(a)
  109. except (CoercionFailed, GeneratorsError):
  110. return None
  111. def from_FractionField(K1, a, K0):
  112. """Convert a rational function to ``dtype``. """
  113. try:
  114. return a.set_field(K1.field)
  115. except (CoercionFailed, GeneratorsError):
  116. return None
  117. def get_ring(self):
  118. """Returns a field associated with ``self``. """
  119. return self.field.to_ring().to_domain()
  120. def is_positive(self, a):
  121. """Returns True if ``LC(a)`` is positive. """
  122. return self.domain.is_positive(a.numer.LC)
  123. def is_negative(self, a):
  124. """Returns True if ``LC(a)`` is negative. """
  125. return self.domain.is_negative(a.numer.LC)
  126. def is_nonpositive(self, a):
  127. """Returns True if ``LC(a)`` is non-positive. """
  128. return self.domain.is_nonpositive(a.numer.LC)
  129. def is_nonnegative(self, a):
  130. """Returns True if ``LC(a)`` is non-negative. """
  131. return self.domain.is_nonnegative(a.numer.LC)
  132. def numer(self, a):
  133. """Returns numerator of ``a``. """
  134. return a.numer
  135. def denom(self, a):
  136. """Returns denominator of ``a``. """
  137. return a.denom
  138. def factorial(self, a):
  139. """Returns factorial of ``a``. """
  140. return self.dtype(self.domain.factorial(a))