o
    *ήc{-                     @   sp   d Z ddlmZmZ ddlmZ ddlmZ ddl	m
Z
mZ ddlmZmZ G dd	 d	eZeed
d ZdS )z
A Printer for generating executable code.

The most important function here is srepr that returns a string so that the
relation eval(srepr(expr))=expr holds in an appropriate environment.
    )AnyDict)AppliedUndef)Mul)repr_dpsto_str   )Printerprint_functionc                   @   s  e Zd ZdZdddZdd Zdd Zdod	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-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Z d=d> Z!dod?d@Z"dAdB Z#dCdD Z$dEdF Z%dGdH Z&dIdJ Z'dKdL Z(dMdN Z)dOdP Z*dQdR Z+dSdT Z,dUdV Z-dWdX Z.dYdZ Z/d[d\ Z0d]d^ Z1d_d` Z2dadb Z3dcdd Z4dedf Z5dgdh Z6didj Z7dkdl Z8dmdn Z9dS )pReprPrinter
_sympyreprNT)orderperm_cyclicc                    s   |  fdd|D S )zG
        Prints each item in `args` and joins them with `sep`.
        c                       g | ]}  |qS  doprint).0itemselfr   :/tmp/pip-target-vg8gfxp4/lib/python/sympy/printing/repr.py
<listcomp>       z'ReprPrinter.reprify.<locals>.<listcomp>)join)r   argssepr   r   r   reprify   s   zReprPrinter.reprifyc                 C   s   t |tr|S t|dr| S t|dr6t|jdr6g }|jD ]
}|| | q |jjdd	|  S t|drHt|drHd|j
|jf S t|S )	z'
        The fallback printer.
        	__srepr__r   __iter__(%s), 
__module____name__z	<'%s.%s'>)
isinstancestrhasattrr   r   append_print	__class__r#   r   r"   )r   exprlor   r   r   emptyPrinter   s   


zReprPrinter.emptyPrinterc                 C   s6   | j ||d}t| j|}t|j}|dd|  S )N)r   r    r!   )_as_ordered_termsmapr(   typer#   r   r   r*   r   r   clsnamer   r   r   
_print_Add1   s   
zReprPrinter._print_Addc                 C   s   |  S N)__repr__r   r*   r   r   r   _print_Cycle7      zReprPrinter._print_Cyclec                 C   s@  ddl m}m} ddlm} |j}|d ur"|d| ddddd	 n| jd
d}|rb|js0dS |||jd 	 t
dd  }|d}|dks^d||d  vr^||d  |d |  }d| S | }|sy|jdk rtdt|j S d|j S t|jd |d d  d|j  }t|j }	}
t
|t
|
k r|}	d|	 S )Nr   )PermutationCycle)sympy_deprecation_warningzw
                Setting Permutation.print_cyclic is deprecated. Instead use
                init_printing(perm_cyclic=z).
                z1.6z#deprecated-permutation-print_cyclic   )deprecated_since_versionactive_deprecations_target
stacklevelr   TzPermutation()r   r:   (,zPermutation%s   zPermutation(%s)zPermutation([], size=%s)z	, size=%s) sympy.combinatorics.permutationsr9   r:   sympy.utilities.exceptionsr;   print_cyclic	_settingsgetsizer5   lenrfindsupportr%   
array_form)r   r*   r9   r:   r;   r   slasttrimusefullr   r   r   _print_Permutation:   s>   
"


$zReprPrinter._print_Permutationc                    2     |j}|dd fdd|jD  7 }|S )Nr    r!   c                    r   r   r(   r   ar   r   r   r   d   r   z/ReprPrinter._print_Function.<locals>.<listcomp>)r(   funcr   r   r   r*   rr   r   r   _print_Functionb   s   "zReprPrinter._print_Functionc                    rT   )Nr    r!   c                    r   r   rU   rV   r   r   r   r   k   r   z0ReprPrinter._print_Heaviside.<locals>.<listcomp>)r(   rX   r   pargsrY   r   r   r   _print_Heavisideg   s   "zReprPrinter._print_Heavisidec                 C   s   t |tr
d|j S |jS )NzFunction(%r))
issubclassr   r#   r6   r   r   r   _print_FunctionClassn   s   

z ReprPrinter._print_FunctionClassc                 C      dS )NzRational(1, 2)r   r6   r   r   r   _print_Halft      zReprPrinter._print_Halfc                 C      t |S r4   r%   r6   r   r   r   _print_RationalConstantw   r8   z#ReprPrinter._print_RationalConstantc                 C   rc   r4   rd   r6   r   r   r   _print_AtomicExprz   r8   zReprPrinter._print_AtomicExprc                 C   rc   r4   rd   r6   r   r   r   _print_NumberSymbol}   r8   zReprPrinter._print_NumberSymbolc                 C   
   d|j  S )NzInteger(%i))pr6   r   r   r   _print_Integer      
zReprPrinter._print_Integerc                 C   r`   )N	Complexesr   r6   r   r   r   _print_Complexes   rb   zReprPrinter._print_Complexesc                 C   r`   )NIntegersr   r6   r   r   r   _print_Integers   rb   zReprPrinter._print_Integersc                 C   r`   )NNaturalsr   r6   r   r   r   _print_Naturals   rb   zReprPrinter._print_Naturalsc                 C   r`   )N	Naturals0r   r6   r   r   r   _print_Naturals0   rb   zReprPrinter._print_Naturals0c                 C   r`   )N	Rationalsr   r6   r   r   r   _print_Rationals   rb   zReprPrinter._print_Rationalsc                 C   r`   )NRealsr   r6   r   r   r   _print_Reals   rb   zReprPrinter._print_Realsc                 C   r`   )NEmptySetr   r6   r   r   r   _print_EmptySet   rb   zReprPrinter._print_EmptySetc                 C   r`   )NUniversalSetr   r6   r   r   r   _print_UniversalSet   rb   zReprPrinter._print_UniversalSetc                 C   r`   )NEmptySequencer   r6   r   r   r   _print_EmptySequence   rb   z ReprPrinter._print_EmptySequencec                 C   s   d|  |d S )Nz[%s]r!   r   r6   r   r   r   _print_list   s   zReprPrinter._print_listc                    s(   d} fdd|  D }d|| S )Nr!   c                    s(   g | ]\}}d   |  |f qS )z%s: %sr   )r   keyvaluer   r   r   r      s   ( z+ReprPrinter._print_dict.<locals>.<listcomp>{%s})itemsr   )r   r*   r   dict_kvsr   r   r   _print_dict   s   zReprPrinter._print_dictc                 C   s   |sdS d|  |d S )Nzset()r   r!   r~   r6   r   r   r   
_print_set   s   zReprPrinter._print_setc                 C   s   |j dk|jdkA rd|jj| |j | |j| g f S g }t|j D ]}|g  t|jD ]}|d |||f  q2q&d|jj| |f S )Nr   %s(%s, %s, %s)rC   %s(%s))rowscolsr)   r#   r(   ranger'   )r   r*   r+   ijr   r   r   _print_MatrixBase   s   


zReprPrinter._print_MatrixBasec                 C   r`   )Ntruer   r6   r   r   r   _print_BooleanTrue   rb   zReprPrinter._print_BooleanTruec                 C   r`   )Nfalser   r6   r   r   r   _print_BooleanFalse   rb   zReprPrinter._print_BooleanFalsec                 C   r`   )Nnanr   r6   r   r   r   
_print_NaN   rb   zReprPrinter._print_NaNc                 C   sF   | j dvr
| }nt|}t| j|}t|j}|dd|  S )N)oldnoner    r!   )	r   as_ordered_factorsr   	make_argsr/   r(   r0   r#   r   r1   r   r   r   
_print_Mul   s   



zReprPrinter._print_Mulc                 C      d|  |j|  |jf S )NzRational(%s, %s))r(   ri   qr6   r   r   r   _print_Rational      zReprPrinter._print_Rationalc                 C   s   d|j j|j|jf S )Nz
%s(%d, %d))r)   r#   ri   r   r6   r   r   r   _print_PythonRational   s   z!ReprPrinter._print_PythonRationalc                 C   r   )NzFraction(%s, %s))r(   	numeratordenominatorr6   r   r   r   _print_Fraction   r   zReprPrinter._print_Fractionc                 C   s&   t |jt|j}d|jj||jf S )Nz%s('%s', precision=%i))mlib_to_str_mpf_r   _precr)   r#   rY   r   r   r   _print_Float   s   zReprPrinter._print_Floatc                 C   s0   d|  |j|  |j|  |j|  |jf S )NzSum2(%s, (%s, %s, %s)))r(   fr   rW   br6   r   r   r   _print_Sum2   s   zReprPrinter._print_Sum2c                 C   s   d|j j| |jf S Nr   )r)   r#   r(   name)r   rN   r   r   r   
_print_Str   s   zReprPrinter._print_Strc                 C   sj   |j j}|jr|j|d< |i krd|jj| |jf S dd | D }d|jj| |jd	|f S )Ndummy_indexr   c                 S      g | ]
\}}d ||f qS z%s=%sr   r   kvr   r   r   r          z-ReprPrinter._print_Symbol.<locals>.<listcomp>
%s(%s, %s)r!   )
_assumptions	generatoris_Dummyr   r)   r#   r(   r   r   r   r   r*   dattrr   r   r   _print_Symbol   s   
zReprPrinter._print_Symbolc                 C   sn   |j j}|i krd|jj| |j| |jf S dd | D }d|jj| |j| |jd|f S )Nr   c                 S   r   r   r   r   r   r   r   r      r   z7ReprPrinter._print_CoordinateSymbol.<locals>.<listcomp>r   r!   )	r   r   r)   r#   r(   	coord_sysindexr   r   r   r   r   r   _print_CoordinateSymbol   s   



z#ReprPrinter._print_CoordinateSymbolc                 C   rh   )NzQ.%s)r   r6   r   r   r   _print_Predicate   rk   zReprPrinter._print_Predicatec                 C   s   |j }d|jj| |df S )Nr   r!   )_argsr)   r#   r   )r   r*   r   r   r   r   _print_AppliedPredicate  s   z#ReprPrinter._print_AppliedPredicatec                 C   rc   r4   )reprr6   r   r   r   
_print_str  r8   zReprPrinter._print_strc                 C   s.   t |dkrd| |d  S d| |d S )Nr   z(%s,)r   r    r!   )rJ   r(   r   r6   r   r   r   _print_tuple  s   zReprPrinter._print_tuplec                 C   s   d|j j|jf S )Nz%s('%s'))r)   r#   r   r6   r   r   r   _print_WildFunction  s   zReprPrinter._print_WildFunctionc                 C   s$   d|j j| |j| | f S )Nr   )r)   r#   r(   rootcoeffsr6   r   r   r   _print_AlgebraicNumber  s   z"ReprPrinter._print_AlgebraicNumberc                 C   ,   d|j j| |j| |j| |jf S Nr   r)   r#   r(   symbolsdomainr   )r   ringr   r   r   _print_PolyRing     zReprPrinter._print_PolyRingc                 C   r   r   r   )r   fieldr   r   r   _print_FracField  r   zReprPrinter._print_FracFieldc                 C   s>   t | }|j|jjdd d|jj| |j| |f S )NTr   reverser   )listtermssortr   r   r)   r#   r(   )r   polyr   r   r   r   _print_PolyElement   s    zReprPrinter._print_PolyElementc                 C   sp   t |j }|j|jjdd t |j }|j|jjdd | |}| |}d|jj	| |j||f S )NTr   r   )
r   numerr   r   r   r   denomr(   r)   r#   )r   fracnumer_termsdenom_termsr   r   r   r   r   _print_FracElement%  s   

zReprPrinter._print_FracElementc                 C   s    |j j}| |j}d||f S r   )r)   r#   r(   r   )r   r   clsr   r   r   r   _print_FractionField.  s   z ReprPrinter._print_FractionFieldc                 C   sZ   |j j}| |j}dt| j|j}t|j}||j	kr#d| }nd}d||||f S )Nr!   z, order= %s(%s, %s%s))
r)   r#   r(   r   r   r/   gensr%   r   default_order)r   r   r   domr   r   orderstrr   r   r   _print_PolynomialRingBase3  s   


z%ReprPrinter._print_PolynomialRingBasec                 C   sP   |j j}| |j}| |j}|jd urd| |j }nd}d||||f S )Nz, ring=r   r   )r)   r#   r(   repr   r   )r   ri   r   r   r   ringstrr   r   r   
_print_DMP>  s   
zReprPrinter._print_DMPc                 C   s   dt |j S )NzFiniteExtension(%s))r%   modulus)r   extr   r   r   _print_MonogenicFiniteExtensionH     z+ReprPrinter._print_MonogenicFiniteExtensionc                 C   s$   |  |j}|  |j}d||f S )NzExtElem(%s, %s))r(   r   r   )r   r   r   r   r   r   r   _print_ExtensionElementM  s   z#ReprPrinter._print_ExtensionElementr4   ):r#   r"   __qualname__printmethod_default_settingsr   r-   r3   r7   rS   r[   r]   r_   ra   re   rf   rg   rj   rm   ro   rq   rs   ru   rw   ry   r{   r}   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r      st    
(
	
r   c                 K   s   t || S )zreturn expr in repr form)r   r   )r*   settingsr   r   r   sreprR  r   r   N)__doc__typingr   r   tDictsympy.core.functionr   sympy.core.mulr   mpmath.libmpr   r   r   printerr	   r
   r   r   r   r   r   r   <module>   s      C