
    ic                     ~    d dl Z d dlZd dlmZmZ d dlmZ dZdZdZ	 G d de      Z
 G d	 d
e
      Z G d de
      Zy)    N)
PRECEDENCE
precedence)
StrPrinterint64_tl             c                      e Zd ZdZdej
                  defdZdej
                  defdZddej
                  dedz  defdZ	dej
                  defd	Z
dej
                  defd
Zdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZdej
                  defdZy) ExprPrinter_torch_sympystrexprreturnc                 N    | j                  |j                  dt        |            S )N*	stringifyargsr   selfr   s     U/var/www/html/engine/venv/lib/python3.12/site-packages/torch/utils/_sympy/printers.py
_print_MulzExprPrinter._print_Mul   s    ~~diij.>??    c                 F    d| j                  |j                  d          dS )Nznot (r   )_printr   r   s     r   
_print_NotzExprPrinter._print_Not   s#    t{{499Q<0133r   Norderc                 N    | j                  |j                  dt        |            S )Nz + r   )r   r   r   s      r   
_print_AddzExprPrinter._print_Add   s    ~~dii
40@AAr   c                 j    | j                  |j                  d|j                   dt        |            S )N )r   r   rel_opr   r   s     r   _print_RelationalzExprPrinter._print_Relational   s+    ~~dii1T[[M);Z=MNNr   c                 J    | j                  |j                  dt        d         S )Nz & 
BitwiseAndr   r   r   r   s     r   _print_BitwiseFn_bitwise_andz(ExprPrinter._print_BitwiseFn_bitwise_and       ~~dii
<0HIIr   c                 J    | j                  |j                  dt        d         S )Nz | 	BitwiseOrr%   r   s     r   _print_BitwiseFn_bitwise_orz'ExprPrinter._print_BitwiseFn_bitwise_or"   s    ~~dii
;0GHHr   c                 J    | j                  |j                  dt        d         S )Nz ^ 
BitwiseXorr%   r   s     r   _print_BitwiseFn_bitwise_xorz(ExprPrinter._print_BitwiseFn_bitwise_xor%   r'   r   c                 P    | j                  |j                  dt        d   dz
        S N % Atom      ?r%   r   s     r   
_print_ModzExprPrinter._print_Mod*   #    ~~dii
60BS0HIIr   c                 \    | j                  |j                  dt        d   dz
        }d| dS )N / r1   r2   (r   r%   )r   r   ss      r   _print_FloatTrueDivzExprPrinter._print_FloatTrueDiv-   s/    NN499eZ-?#-EF1#Qxr   c                 $    | j                  |      S N)_print_FloorDivr   s     r   _print_CleanDivzExprPrinter._print_CleanDiv1   s    ##D))r   c                 >    | j                  |j                  d         S Nr   r   r   s     r   _print_IdentityzExprPrinter._print_Identity4   s    {{499Q<((r   c                 v    |j                   dk(  r t        t        j                  |d            S t        |      S )N5      )dps)_precstrsympyFloatr   s     r   _print_FloatzExprPrinter._print_Float7   s2    :: u{{4R011 t9r   c                     |j                   \  }}|t        |      k7  rt        |      t        |      }|dk  rt        d|       |dkD  r| j                  |g|z  dt        d         S y)Nr   z#exponent must be non-negative, got r   Mul1)r   intAssertionErrorr   r   r   r   baseexps       r   
_print_PowzExprPrinter._print_PowH   sp    II	c#c(? %%#h7 #Fse!LMM7>>4&3,Z5FGGr   c                 0    t        dt        |              )Nz#_print_ToFloat not implemented for NotImplementedErrortyper   s     r   _print_ToFloatzExprPrinter._print_ToFloatX   s    !$GT
|"TUUr   c                 0    t        dt        |              )Nz$_print_Infinity not implemented for rT   r   s     r   _print_InfinityzExprPrinter._print_Infinity[       !$Hd"UVVr   c                 0    t        dt        |              )Nz,_print_NegativeInfinity not implemented for rT   r   s     r   _print_NegativeInfinityz#ExprPrinter._print_NegativeInfinity^   s    !:4:,G
 	
r   c                 0    t        dt        |              )Nz$_print_FloorDiv not implemented for rT   r   s     r   r<   zExprPrinter._print_FloorDivc   rZ   r   c                 0    t        dt        |              )Nz%_print_PythonMod not implemented for rT   r   s     r   _print_PythonModzExprPrinter._print_PythonModf   s    !$I$t*"VWWr   c                 0    t        dt        |              )Nz&_print_IntTrueDiv not implemented for rT   r   s     r   _print_IntTrueDivzExprPrinter._print_IntTrueDivi       !$J4PT:,"WXXr   c                 0    t        dt        |              )N(_print_PowByNatural not implemented for rT   r   s     r   _print_PowByNaturalzExprPrinter._print_PowByNaturall       !6tDzlC
 	
r   c                 0    t        dt        |              )Nz$_print_FloatPow not implemented for rT   r   s     r   _print_FloatPowzExprPrinter._print_FloatPowq   rZ   r   c                 0    t        dt        |              )Nz&_print_TruncToInt not implemented for rT   r   s     r   _print_TruncToIntzExprPrinter._print_TruncToIntt   rb   r   c                 0    t        dt        |              )Nz&_print_RoundToInt not implemented for rT   r   s     r   _print_RoundToIntzExprPrinter._print_RoundToIntw   rb   r   c                 0    t        dt        |              )Nz(_print_RoundDecimal not implemented for rT   r   s     r   _print_RoundDecimalzExprPrinter._print_RoundDecimalz   rf   r   c                 0    t        dt        |              )Nz(_print_TruncToFloat not implemented for rT   r   s     r   _print_TruncToFloatzExprPrinter._print_TruncToFloat   rf   r   r;   ) __name__
__module____qualname__printmethodrG   ExprrF   r   r   r   r"   r&   r*   r-   r3   r9   r=   r@   rI   rR   rW   rY   r\   r<   r_   ra   re   rh   rj   rl   rn   rp    r   r   r	   r	      s   #K@uzz @c @4uzz 4c 4Buzz B#* B BOejj OS OJ J JI

 Is IJ J J
Juzz Jc J

 s *EJJ *3 *)EJJ )3 )  "	uzz 	c 	 V5:: V# VWEJJ W3 W
EJJ 
3 

WEJJ W3 WXUZZ XC XYejj YS Y


 
s 

WEJJ W3 WYejj YS YYejj YS Y


 
s 



 
s 
r   r	   c                      e Zd Zdej                  defdZdej                  defdZdej                  defdZdej                  defdZ	dej                  defdZ
dej                  defdZdej                  defd	Zdej                  defd
Zdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZ dej                  defdZ!dej                  defdZ"dej                  defd Z#dej                  defd!Z$dej                  defd"Z%dej                  defd#Z&dej                  defd$Z'y%)&PythonPrinterr   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )N   $ToFloat expects exactly one argumentztorch.sym_float(r   r   lenr   rN   r   r   s     r   rW   zPythonPrinter._print_ToFloat   sA    tyy>Q !GHH "$++diil";!<A>>r   c                 N    | j                  |j                  dt        |            S )Nz and r   r   s     r   
_print_AndzPythonPrinter._print_And   s    ~~dii*T2BCCr   c                 N    | j                  |j                  dt        |            S )Nz or r   r   s     r   	_print_OrzPythonPrinter._print_Or   s    ~~diiD1ABBr   c                 d      fd|j                   D        \  }}}|dk7  r	d| d| d}d| d| dS )Nc              3   V   K   | ]   }j                  |t        d    dz
         " ywr1   r2   Nparenthesizer   .0argr   s     r   	<genexpr>z7PythonPrinter._print_ModularIndexing.<locals>.<genexpr>   s-      
ADDc:f#5#;<
   &)rL   r7    // r   r0   r   r   r   xdivmods   `    r   _print_ModularIndexingz$PythonPrinter._print_ModularIndexing   sN    
HL		
3 #:A3d3%q!A1#SQr   c                      y)Nzmath.infrv   r   s     r   rY   zPythonPrinter._print_Infinity   s    r   c                      y)Nz	-math.infrv   r   s     r   r\   z%PythonPrinter._print_NegativeInfinity   s    r   c                 P    | j                  |j                  dt        d   dz
        S r/   r%   r   s     r   r_   zPythonPrinter._print_PythonMod   r4   r   c                 B      fd|j                   D        \  }}| d| S )Nc              3   V   K   | ]   }j                  |t        d    dz
         " ywr   r   r   s     r   r   z0PythonPrinter._print_FloorDiv.<locals>.<genexpr>   s(     Xs$##CF);c)ABXr   r   r   r   r   r   r   s   `   r   r<   zPythonPrinter._print_FloorDiv   s%    XdiiX3Dr   c                 P    | j                  |j                  dt        d   dz
        S )Nr6   r1   r2   r%   r   s     r   ra   zPythonPrinter._print_IntTrueDiv   r4   r   c                 ,    d| j                  |       dS )Nz
math.sqrt(r   )r   r   s     r   _helper_sqrtzPythonPrinter._helper_sqrt   s    DKK-.a00r   c                 >    | j                  |j                  d         S r?   )r   r   r   s     r   _print_OpaqueUnaryFn_sqrtz'PythonPrinter._print_OpaqueUnaryFn_sqrt   s      1..r   c                 J    | j                  |j                  dt        d         S Nz ** Powr%   r   s     r   rh   zPythonPrinter._print_FloatPow       ~~diiE1BCCr   c                 J    | j                  |j                  dt        d         S r   r%   r   s     r   re   z!PythonPrinter._print_PowByNatural   r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   "floor expects exactly one argumentmath.floor(r   r   r|   r   s     r   _print_floorzPythonPrinter._print_floor   s?    tyy>Q !EFFT[[167q99r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   'FloorToInt expects exactly one argumentr   r   r   r|   r   s     r   _print_FloorToIntzPythonPrinter._print_FloorToInt   s?    tyy>Q !JKKT[[167q99r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   'TruncToInt expects exactly one argumentzmath.trunc(r   r   r|   r   s     r   rj   zPythonPrinter._print_TruncToInt   ?    tyy>Q !JKKT[[167q99r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   $ceiling expects exactly one argument
math.ceil(r   r   r|   r   s     r   _print_ceilingzPythonPrinter._print_ceiling   s?    tyy>Q !GHHDKK		!56a88r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   &CeilToInt expects exactly one argumentr   r   r   r|   r   s     r   _print_CeilToIntzPythonPrinter._print_CeilToInt   s?    tyy>Q !IJJDKK		!56a88r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz    Abs expects exactly one argumentzabs(r   r   r|   r   s     r   
_print_AbszPythonPrinter._print_Abs   s?    tyy>Q !CDDdkk$))A,/022r   c                     t        |j                        dk  rt        d      ddj                  t	        | j
                  |j                               dS )N   z"Max expects at least two argumentszmax(, r   r}   r   rN   joinmapr   r   s     r   
_print_MaxzPythonPrinter._print_Max   F    tyy>A !EFFdiiDKK ;<=Q??r   c                     t        |j                        dk  rt        d      ddj                  t	        | j
                  |j                               dS )Nr   z"Min expects at least two argumentszmin(r   r   r   r   s     r   
_print_MinzPythonPrinter._print_Min   r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz    cos expects exactly one argumentz	math.cos(r   r   r|   r   s     r   _print_OpaqueUnaryFn_cosz&PythonPrinter._print_OpaqueUnaryFn_cos   ?    tyy>Q !CDD4;;tyy|45Q77r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !cosh expects exactly one argumentz
math.cosh(r   r   r|   r   s     r   _print_OpaqueUnaryFn_coshz'PythonPrinter._print_OpaqueUnaryFn_cosh   ?    tyy>Q !DEEDKK		!56a88r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !acos expects exactly one argumentz
math.acos(r   r   r|   r   s     r   _print_OpaqueUnaryFn_acosz'PythonPrinter._print_OpaqueUnaryFn_acos  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS Nrz   z sin expects exactly one argumentz	math.sin(r   r   r|   r   s     r   _print_OpaqueUnaryFn_sinz&PythonPrinter._print_OpaqueUnaryFn_sin  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !sinh expects exactly one argumentz
math.sinh(r   r   r|   r   s     r   _print_OpaqueUnaryFn_sinhz'PythonPrinter._print_OpaqueUnaryFn_sinh  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !asin expects exactly one argumentz
math.asin(r   r   r|   r   s     r   _print_OpaqueUnaryFn_asinz'PythonPrinter._print_OpaqueUnaryFn_asin  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz    tan expects exactly one argumentz	math.tan(r   r   r|   r   s     r   _print_OpaqueUnaryFn_tanz&PythonPrinter._print_OpaqueUnaryFn_tan  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !tanh expects exactly one argumentz
math.tanh(r   r   r|   r   s     r   _print_OpaqueUnaryFn_tanhz'PythonPrinter._print_OpaqueUnaryFn_tanh  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   !atan expects exactly one argumentz
math.atan(r   r   r|   r   s     r   _print_OpaqueUnaryFn_atanz'PythonPrinter._print_OpaqueUnaryFn_atan   r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   z!log2 expects exactly one argumentz
math.log2(r   r   r|   r   s     r   _print_OpaqueUnaryFn_log2z'PythonPrinter._print_OpaqueUnaryFn_log2%  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   'RoundToInt expects exactly one argumentround(r   r   r|   r   s     r   rl   zPythonPrinter._print_RoundToInt*  s?    tyy>Q !JKKDIIaL12!44r   c                     t        |j                        dk7  rt        d      |j                  \  }}t        |t        j
                        st        d      d| j                  |       d| dS )Nr   *RoundDecimal expects exactly two argumentsz,ndigits must be an instance of sympy.Integerr   r   r   )r}   r   rN   
isinstancerG   Integer	TypeErrorr   )r   r   numberndigitss       r   rn   z!PythonPrinter._print_RoundDecimal/  sd    tyy>Q !MNN))'5==1JKKF+,Bwiq99r   c                     d }t        |j                        D ]@  \  }}| j                  |      }|dk(  r|}| j                  |      }||}5d| d| d| d}B |r|S dS )NTr7   z if z else r   0)reversedr   r   r   r   resultexpr_icond_iexpr_strcond_strs          r   _print_PiecewisezPythonPrinter._print_Piecewise7  s     "&tyy1 
	INFF{{6*H~!;;v.>%F 
$xjvhaHF
	I  v(S(r   N)(rq   rr   rs   rG   ru   rF   rW   r   r   r   rY   r\   r_   r<   ra   r   r   rh   re   r   r   rj   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rl   rn   r   rv   r   r   rx   rx      s(   ?5:: ?# ?$Duzz Dc DCejj CS C 5::  #  EJJ 3 EJJ 3 JUZZ JC JEJJ 3 Jejj JS J1 1 1/ejj /S /DEJJ D3 DD

 Ds D: : :
:ejj :S :
:ejj :S :95:: 9# 9
9UZZ 9C 9
3uzz 3c 3@uzz @c @
@uzz @c @
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
9ejj 9S 9
5ejj 5S 5
:

 :s :)UZZ )C )r   rx   c                   d   e Zd Zdej                  defdZdej                  defdZdej                  defdZdej                  defdZ	dej                  defdZ
dej                  defdZdej                  defd	Zdej                  defd
Zdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZdej                  defdZ dej                  defdZ!dej                  defdZ"dej                  defd Z#dej                  defd!Z$dej                  defd"Z%dej                  defd#Z&dej                  defd$Z'dej                  defd%Z(dej                  defd&Z)dej                  defd'Z*dej                  defd(Z+y))*
CppPrinterr   r   c                     t         j                  dv rdnd}t        |      }|t        kD  s	|t        k  rt        | dt               |t        k(  r|dk7  rt        d      d| dS | | S )	N)darwinwin32LLLz too big to convert to r   z#unexpected minimum index type valuez(-1z << 63))sysplatformrM   INDEX_TYPE_MAXINDEX_TYPE_MINOverflowError
INDEX_TYPErN   )r   r   suffixis       r   _print_IntegerzCppPrinter._print_IntegerK  s~    )<<#I~^!31#%<ZL IJJ. J$%JKK ((VH~r   c                 J      fd|j                   D        \  }}}| d| d| S )Nc              3   V   K   | ]   }j                  |t        d    dz
         " ywr   r   r   s     r   r   z*CppPrinter._print_Where.<locals>.<genexpr>Y  s-      
ADDc:f#5#;<
r    ?  : r   )r   r   cpqs   `    r   _print_WherezCppPrinter._print_WhereX  s5    
HL		
1a Cs#aS!!r   c                     d }t        |j                        D ]T  \  }}| j                  |t        d   dz
        }|dk(  r|}*| j                  |t        d   dz
        }||}K| d| d| }V |rd| dS dS )	Nr1   r2   Tr  r  r7   r   r   )r   r   r   r   r   s          r   r   zCppPrinter._print_Piecewise^  s     "&tyy1 
	CNFF((F1Cc1IJH~!,,VZ5G#5MN>%F (zXJc&BF
	C !'6(!}/C/r   c           	          |j                   \  }}}| j                  |      }|dk7  r0| j                  |      }|j                  r
d| d| d}n	d| d| d}| j                  |      }dt         d| d	t         d| d	S )
Nrz   ,c10::div_floor_integer(static_cast<int64_t>(), static_cast<int64_t>()),c10::div_floor_floating(static_cast<double>(), static_cast<double>(z(static_cast<>(z) % static_cast<)r   doprint
is_integerr   r   s        r   r   z!CppPrinter._print_ModularIndexingp  s    ii3LLO!8,,s#CB1#E]^a]bbdeB1#E\]`\aacdll3zl"QC/?
|2cURTUUr   c                     |j                   \  }}| j                  |      }| j                  |      }|j                  r	d| d| dS d| d| dS )Nr  r  r  r  r  )r   r  r  r   s       r   r<   zCppPrinter._print_FloorDiv|  sc    3LLOll3??A!D\]`\aacdd=aS@WX[W\\^__r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          d}|j                  rdt
         d| dS |S )Nrz   r   std::floor(r   r   static_cast<r  r}   r   rN   r   r  r   r   r   rs      r   r   zCppPrinter._print_floor  s^    tyy>Q !EFF$++diil34A648OOj\A3a0JJr   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          d}|j                  rdt
         d| dS |S )Nrz   r   r  r   r   r  r  r  r  s      r   r   zCppPrinter._print_FloorToInt  s^    tyy>Q !JKK$++diil34A648OOj\A3a0JJr   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          d}dt         d| dS )Nrz   r   std::trunc(r   r   r  r  )r}   r   rN   r   r   r  s      r   rj   zCppPrinter._print_TruncToInt  sS    tyy>Q !JKK$++diil34A6j\A3a00r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   z)TruncToFloat expects exactly one argumentr  r   r   r|   r   s     r   rp   zCppPrinter._print_TruncToFloat  s?    tyy>Q !LMMT[[167q99r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r{   static_cast<double>(r   r   r|   r   s     r   rW   zCppPrinter._print_ToFloat  s?    tyy>Q !GHH%dkk$))A,&?%@BBr   c                 v    |j                   \  }}| j                  |      }| j                  |      }d| d| dS )Nzc10::div_mod(r   r   )r   r  r   s       r   r_   zCppPrinter._print_PythonMod  s?    3LLOll3qcC5**r   c                 n    |j                   \  }}d| j                  |       d| j                  |       dS )Nr"  z) / static_cast<double>(r   r   r   )r   r   lhsrhss       r   ra   zCppPrinter._print_IntTrueDiv  s>    99S%dkk#&6%77OPTP[P[\_P`Oaabccr   c                     |j                   \  }}|dk(  rd| j                  |       dS t        dt        |              )Nr   z(1 << (r  rd   )r   r   rU   rV   rO   s       r   re   zCppPrinter._print_PowByNatural  sJ    II	c19T[[-.b11!6tDzlC
 	
r   c                 n    |j                   \  }}d| j                  |       d| j                  |       dS )N	std::pow(r   r   r%  rO   s       r   rh   zCppPrinter._print_FloatPow  s8    II	c4;;t,-RC0@/ACCr   c                    |j                   \  }}|dk(  s|dk(  r"| j                  |      }|dk(  rd| dS d| dS |j                  rt        |      }|dkD  r| j	                  |g|z  dt
        d         }nO|d	k  r.d
| j	                  |gt        |      z  dt
        d         z   dz   }n|d	k(  rd| j                  |      z   }nd}|j                  rdt         d| dS |S d| dt        |       dS )Nr2   g      
std::sqrt(r   z1.0/std::sqrt(r   r   rK   z1.0/(z1.0/z1.0r  r  r*  r   )	r   r   r  rM   r   r   absr   float)r   r   rP   rQ   r  s        r   rR   zCppPrinter._print_Pow  s   II	c#:;;t$D+.#:ZvQ'S^D6QR;SS>>c(CQwNND6C<j6GHrnndVc#h%6Z=NOP 
 T[[..8<\*Rs!4NQN tfBuSzl!44r   c                     |j                   dk(  r|j                   }n|j                   d|j                    d}|j                  rdt         d| dS |S )Nrz   z.0/z.0r  r  r   )r
  r	  r  r   r  s      r   _print_RationalzCppPrinter._print_Rational  sR    66Q;66(A66(#dffXR(A48OOj\A3a0JJr   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          d}|j                  rdt
         d| dS |S )Nrz   r   
std::ceil(r   r   r  r  r  r  s      r   r   zCppPrinter._print_ceiling  s^    tyy>Q !GHHTYYq\231548OOj\A3a0JJr   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          d}|j                  rdt
         d| dS |S )Nrz   r   r3  r   r   r  r  r  r  s      r   r   zCppPrinter._print_CeilToInt  s^    tyy>Q !IJJTYYq\231548OOj\A3a0JJr   c           	          |j                   D cg c]  }| j                  |       }}t        |      dk(  rdt         d|d    dt         d|d    d	S dd	j	                  |      z   d
z   }dt         d| dS c c}w )Nr   zstd::min(static_cast<r  r   ), static_cast<rz   r  {r   }z	std::min<r   r   r   r}   r   r   r   r   ar   ils        r   r   zCppPrinter._print_Min      (,		21A22t9>*:,ba	Q[P\\^_cde_f^ggijj tyy&,Bzl"RD22 3   A8c           	          |j                   D cg c]  }| j                  |       }}t        |      dk(  rdt         d|d    dt         d|d    d	S dd	j	                  |      z   d
z   }dt         d| dS c c}w )Nr   zstd::max(static_cast<r  r   r6  rz   r  r7  r   r8  z	std::max<r   r9  r:  s        r   r   zCppPrinter._print_Max  r=  r>  c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z	std::abs(r   r   r|   r   s     r   r   zCppPrinter._print_Abs  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z	std::cos(r   r   r|   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_cos   r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::cosh(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_cosh  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::acos(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_acos
  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS r   r|   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_sin  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::sinh(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_sinh  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::asin(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_asin  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z	std::tan(r   r   r|   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_tan  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::tanh(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_tanh#  r   r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   z
std::atan(r   r   r|   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_atan(  r   r   c                 F    d| j                  |j                  d          dS )Nr,  r   r   r   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_sqrt-  #    DKK		!56a88r   c                 F    d| j                  |j                  d          dS )Nz
std::log2(r   r   r   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_log20  rK  r   c                     t        |j                        dk7  rt        d      d| j                  |j                  d          dS )Nrz   r   zstd::lrint(r   r   r|   r   s     r   rl   zCppPrinter._print_RoundToInt3  r   r   c                    t        |j                        dk7  rt        d      |j                  \  }}|j                  r|dk\  rt        d      t	        d| d      | j                  |t        d         }d| d	| d
|  dS )Nr   r   r   z+ndigits must be negative for integer inputszOFor integer inputs, only non-negative ndigits are currently supported, but got .rK   z%static_cast<double>(std::nearbyint(1ez * z) * 1er   )r}   r   rN   r  
ValueErrorr   r   )r   r   r   r   
number_strs        r   rn   zCppPrinter._print_RoundDecimal9  s    tyy>Q !MNN))!|$%RSSabiajjkl  &&vz%/@A
6wis:,fV]U]T^^_``r   c                      y)Ntruerv   r   s     r   _print_BooleanTruezCppPrinter._print_BooleanTrueG  s    r   c                      y)Nfalserv   r   s     r   _print_BooleanFalsezCppPrinter._print_BooleanFalseJ  s    r   c                      y)Nz'std::numeric_limits<double>::infinity()rv   r   s     r   rY   zCppPrinter._print_InfinityM  s    8r   c                 *    d| j                  |       S )N-)rY   r   s     r   r\   z"CppPrinter._print_NegativeInfinityP  s    4''-.//r   N),rq   rr   rs   rG   ru   rF   r  r  r   r   r<   r   r   rj   rp   rW   r_   ra   re   rh   rR   r1  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rl   rn   rT  rW  rY   r\   rv   r   r   r   r   J  s   5:: # " " "0UZZ 0C 0$
V5:: 
V# 
V`EJJ `3 `K K KKejj KS K1ejj 1S 1:

 :s :
C5:: C# C
+UZZ +C +dejj dS d


 
s 
DEJJ D3 D5uzz 5c 56KEJJ K3 KK5:: K# KKUZZ KC K3uzz 3c 33uzz 3c 38uzz 8c 8
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
8UZZ 8C 8
9ejj 9S 9
9ejj 9S 9
9ejj 9S 99ejj 9S 9:ejj :S :a

 as auzz c 

 s 9EJJ 93 90EJJ 03 0r   r   )r   rG   sympy.printing.precedencer   r   sympy.printing.strr   r   r   r   r	   rx   r   rv   r   r   <module>r]     sN    
  < ) 

x
* x
v})K })@G0 G0r   