o
    à8VaÉ  ã                   @   sd   d Z ddlmZmZmZmZmZmZm	Z
 ddlmZ ddlmZ ddlmZ eG dd„ deƒƒZdS )	z2Implementation of :class:`GMPYIntegerRing` class. é    )ÚGMPYIntegerÚSymPyIntegerÚ	factorialÚ
gmpy_gcdexÚgmpy_gcdÚgmpy_lcmÚsqrt)ÚIntegerRing)ÚCoercionFailed)Úpublicc                   @   s¸   e Zd ZdZeZedƒZedƒZeeƒZ	dZ
dd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'S )(ÚGMPYIntegerRingz³Integer ring based on GMPY's ``mpz`` type.

    This will be the implementation of :ref:`ZZ` if ``gmpy`` or ``gmpy2`` is
    installed. Elements will be of type ``gmpy.mpz``.
    r   é   ZZZ_gmpyc                 C   s   dS )z$Allow instantiation of this domain. N© )Úselfr   r   úE/usr/lib/python3/dist-packages/sympy/polys/domains/gmpyintegerring.pyÚ__init__   s    zGMPYIntegerRing.__init__c                 C   s   t t|ƒƒS )z!Convert ``a`` to a SymPy object. )r   Úint©r   Úar   r   r   Úto_sympy   ó   zGMPYIntegerRing.to_sympyc                 C   s:   |j rt|jƒS |jrt|ƒ|krtt|ƒƒS td| ƒ‚)z&Convert SymPy's Integer to ``dtype``. zexpected an integer, got %s)Z
is_Integerr   ÚpZis_Floatr   r
   r   r   r   r   Ú
from_sympy"   s
   
zGMPYIntegerRing.from_sympyc                 C   s   t | ¡ ƒS )z3Convert ``ModularInteger(int)`` to GMPY's ``mpz``. )r   Úto_int©ÚK1r   ÚK0r   r   r   Úfrom_FF_python+   r   zGMPYIntegerRing.from_FF_pythonc                 C   ó   t |ƒS )z,Convert Python's ``int`` to GMPY's ``mpz``. )r   r   r   r   r   Úfrom_ZZ_python/   ó   zGMPYIntegerRing.from_ZZ_pythonc                 C   ó   |j dkr
t|jƒS dS ©z1Convert Python's ``Fraction`` to GMPY's ``mpz``. r   N©Údenominatorr   Ú	numeratorr   r   r   r   Úfrom_QQ3   ó   

ÿzGMPYIntegerRing.from_QQc                 C   r!   r"   r#   r   r   r   r   Úfrom_QQ_python8   r'   zGMPYIntegerRing.from_QQ_pythonc                 C   s   |  ¡ S )z3Convert ``ModularInteger(mpz)`` to GMPY's ``mpz``. )r   r   r   r   r   Úfrom_FF_gmpy=   r    zGMPYIntegerRing.from_FF_gmpyc                 C   s   |S )z*Convert GMPY's ``mpz`` to GMPY's ``mpz``. r   r   r   r   r   Úfrom_ZZ_gmpyA   s   zGMPYIntegerRing.from_ZZ_gmpyc                 C   s   |j dkr|jS dS )z(Convert GMPY ``mpq`` to GMPY's ``mpz``. r   N)r$   r%   r   r   r   r   Úfrom_QQ_gmpyE   s   
ÿzGMPYIntegerRing.from_QQ_gmpyc                 C   s"   |  |¡\}}|dkrt|ƒS dS )z,Convert mpmath's ``mpf`` to GMPY's ``mpz``. r   N)Zto_rationalr   )r   r   r   r   Úqr   r   r   Úfrom_RealFieldJ   s   ÿzGMPYIntegerRing.from_RealFieldc                 C   s   |j dkr|jS d S )Nr   )ÚyÚxr   r   r   r   Úfrom_GaussianIntegerRingQ   s   
ÿz(GMPYIntegerRing.from_GaussianIntegerRingc                 C   s   t ||ƒ\}}}|||fS )z)Compute extended GCD of ``a`` and ``b``. )r   )r   r   ÚbÚhÚsÚtr   r   r   ÚgcdexU   s   
zGMPYIntegerRing.gcdexc                 C   ó
   t ||ƒS )z Compute GCD of ``a`` and ``b``. )r   ©r   r   r1   r   r   r   ÚgcdZ   ó   
zGMPYIntegerRing.gcdc                 C   r6   )z Compute LCM of ``a`` and ``b``. )r   r7   r   r   r   Úlcm^   r9   zGMPYIntegerRing.lcmc                 C   r   )zCompute square root of ``a``. )Ú	gmpy_sqrtr   r   r   r   r   b   r    zGMPYIntegerRing.sqrtc                 C   r   )zCompute factorial of ``a``. )Úgmpy_factorialr   r   r   r   r   f   r    zGMPYIntegerRing.factorialN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ZdtypeZzeroZoneÚtypeÚtpÚaliasr   r   r   r   r   r&   r(   r)   r*   r+   r-   r0   r5   r8   r:   r   r   r   r   r   r   r      s0    	r   N)r@   Zsympy.polys.domains.groundtypesr   r   r   r<   r   r   r   r   r;   Zsympy.polys.domains.integerringr	   Zsympy.polys.polyerrorsr
   Zsympy.utilitiesr   r   r   r   r   r   Ú<module>   s    $