
    iP                         d dl Z d dlm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)Optional)
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	e   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    s    ~~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                 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 / r.   r/   (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_CleanDiv/   s    ##D))r   c                 >    | j                  |j                  d         S Nr   r   r   s     r   _print_IdentityzExprPrinter._print_Identity2   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_Float5   s2    :: u{{4R011 t9r   c                     |j                   \  }}|t        |      k(  sJ |       t        |      }|dk\  sJ |dkD  r| j                  |g|z  dt        d         S y)Nr   r   Mul1)r   intr   r   r   r   baseexps       r   
_print_PowzExprPrinter._print_PowE   s`    II	cc#h###haxx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_ToFloatS   s    !$GT
|"TUUr   c                 0    t        dt        |              )Nz$_print_Infinity not implemented for rP   r   s     r   _print_InfinityzExprPrinter._print_InfinityV       !$Hd"UVVr   c                 0    t        dt        |              )Nz,_print_NegativeInfinity not implemented for rP   r   s     r   _print_NegativeInfinityz#ExprPrinter._print_NegativeInfinityY   s    !:4:,G
 	
r   c                 0    t        dt        |              )Nz$_print_FloorDiv not implemented for rP   r   s     r   r9   zExprPrinter._print_FloorDiv^   rV   r   c                 0    t        dt        |              )Nz%_print_PythonMod not implemented for rP   r   s     r   _print_PythonModzExprPrinter._print_PythonModa   s    !$I$t*"VWWr   c                 0    t        dt        |              )Nz&_print_IntTrueDiv not implemented for rP   r   s     r   _print_IntTrueDivzExprPrinter._print_IntTrueDivd       !$J4PT:,"WXXr   c                 0    t        dt        |              )N(_print_PowByNatural not implemented for rP   r   s     r   _print_PowByNaturalzExprPrinter._print_PowByNaturalg       !6tDzlC
 	
r   c                 0    t        dt        |              )Nz$_print_FloatPow not implemented for rP   r   s     r   _print_FloatPowzExprPrinter._print_FloatPowl   rV   r   c                 0    t        dt        |              )Nz&_print_TruncToInt not implemented for rP   r   s     r   _print_TruncToIntzExprPrinter._print_TruncToInto   r^   r   c                 0    t        dt        |              )Nz&_print_RoundToInt not implemented for rP   r   s     r   _print_RoundToIntzExprPrinter._print_RoundToIntr   r^   r   c                 0    t        dt        |              )Nz(_print_RoundDecimal not implemented for rP   r   s     r   _print_RoundDecimalzExprPrinter._print_RoundDecimalu   rb   r   c                 0    t        dt        |              )Nz(_print_TruncToFloat not implemented for rP   r   s     r   _print_TruncToFloatzExprPrinter._print_TruncToFloat   rb   r   r8   ) __name__
__module____qualname__printmethodrD   ExprrC   r   r   r   r   r#   r'   r*   r0   r6   r:   r=   rF   rN   rS   rU   rX   r9   r[   r]   ra   rd   rf   rh   rj   rl    r   r   r
   r
      se   #K@uzz @c @4uzz 4c 4Buzz B(3- B3 BOejj OS OJ J JI

 Is I
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&y$)%PythonPrinterr   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )N   ztorch.sym_float(r   r   lenr   r   r   s     r   rS   zPythonPrinter._print_ToFloat   s;    499~""" "$++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r.   r/   Nparenthesizer   .0argr   s     r   	<genexpr>z7PythonPrinter._print_ModularIndexing.<locals>.<genexpr>   s-      
ADDc:f#5#;<
   &)rI   r4    // r   r-   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.infrr   r   s     r   rU   zPythonPrinter._print_Infinity   s    r   c                      y)Nz	-math.infrr   r   s     r   rX   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   r1   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   r9   zPythonPrinter._print_FloorDiv   s%    XdiiX3Dr   c                 P    | j                  |j                  dt        d   dz
        S )Nr3   r.   r/   r&   r   s     r   r]   zPythonPrinter._print_IntTrueDiv   r1   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   rd   zPythonPrinter._print_FloatPow       ~~diiE1BCCr   c                 J    | j                  |j                  dt        d         S r   r&   r   s     r   ra   z!PythonPrinter._print_PowByNatural   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS Nrv   zmath.floor(r   r   rw   r   s     r   _print_floorzPythonPrinter._print_floor   9    499~"""T[[167q99r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS r   rw   r   s     r   _print_FloorToIntzPythonPrinter._print_FloorToInt   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   zmath.trunc(r   r   rw   r   s     r   rf   zPythonPrinter._print_TruncToInt   9    499~"""T[[167q99r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS Nrv   z
math.ceil(r   r   rw   r   s     r   _print_ceilingzPythonPrinter._print_ceiling   9    499~"""DKK		!56a88r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS r   rw   r   s     r   _print_CeilToIntzPythonPrinter._print_CeilToInt   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   zabs(r   r   rw   r   s     r   
_print_AbszPythonPrinter._print_Abs   s9    499~"""dkk$))A,/022r   c                     t        |j                        dk\  sJ ddj                  t        | j                  |j                               dS )N   zmax(, r   rx   r   joinmapr   r   s     r   
_print_MaxzPythonPrinter._print_Max   @    499~"""diiDKK ;<=Q??r   c                     t        |j                        dk\  sJ ddj                  t        | j                  |j                               dS )Nr   zmin(r   r   r   r   s     r   
_print_MinzPythonPrinter._print_Min   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	math.cos(r   r   rw   r   s     r   _print_OpaqueUnaryFn_cosz&PythonPrinter._print_OpaqueUnaryFn_cos   9    499~"""4;;tyy|45Q77r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.cosh(r   r   rw   r   s     r   _print_OpaqueUnaryFn_coshz'PythonPrinter._print_OpaqueUnaryFn_cosh   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.acos(r   r   rw   r   s     r   _print_OpaqueUnaryFn_acosz'PythonPrinter._print_OpaqueUnaryFn_acos   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	math.sin(r   r   rw   r   s     r   _print_OpaqueUnaryFn_sinz&PythonPrinter._print_OpaqueUnaryFn_sin   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.sinh(r   r   rw   r   s     r   _print_OpaqueUnaryFn_sinhz'PythonPrinter._print_OpaqueUnaryFn_sinh   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.asin(r   r   rw   r   s     r   _print_OpaqueUnaryFn_asinz'PythonPrinter._print_OpaqueUnaryFn_asin   r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	math.tan(r   r   rw   r   s     r   _print_OpaqueUnaryFn_tanz&PythonPrinter._print_OpaqueUnaryFn_tan  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.tanh(r   r   rw   r   s     r   _print_OpaqueUnaryFn_tanhz'PythonPrinter._print_OpaqueUnaryFn_tanh  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.atan(r   r   rw   r   s     r   _print_OpaqueUnaryFn_atanz'PythonPrinter._print_OpaqueUnaryFn_atan
  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
math.log2(r   r   rw   r   s     r   _print_OpaqueUnaryFn_log2z'PythonPrinter._print_OpaqueUnaryFn_log2  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   round(r   r   rw   r   s     r   rh   zPythonPrinter._print_RoundToInt  s9    499~"""DIIaL12!44r   c                     t        |j                        dk(  sJ |j                  \  }}t        |t        j                        sJ d| j                  |       d| dS )Nr   r   r   r   )rx   r   
isinstancerD   Integerr   )r   r   numberndigitss       r   rj   z!PythonPrinter._print_RoundDecimal  sX    499~"""))'5==111F+,Bwiq99r   N)'rm   rn   ro   rD   rq   rC   rS   rz   r|   r   rU   rX   r[   r9   r]   r   r   rd   ra   r   r   rf   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rh   rj   rr   r   r   rt   rt      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 :r   rt   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'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k(  sJ d| dS | | S )N)darwinwin32LLLz too big to convert to r   z(-1z << 63))sysplatformrJ   INDEX_TYPE_MAXINDEX_TYPE_MINOverflowError
INDEX_TYPE)r   r   suffixis       r   _print_IntegerzCppPrinter._print_Integer  sw    )<<#I~^!31#%<ZL IJJ. 
?"? ((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>+  s-      
ADDc:f#5#;<
r   z ? z : r   )r   r   cpqs   `    r   _print_WherezCppPrinter._print_Where*  s5    
HL		
1a Cs#aS!!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 )
Nrv   ,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_ModularIndexing0  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   r9   zCppPrinter._print_FloorDiv<  sc    3LLOll3??A!D\]`\aacdd=aS@WX[W\\^__r   c                     t        |j                        dk(  sJ d| j                  |j                  d          d}|j                  rdt         d| dS |S Nrv   zstd::floor(r   r   static_cast<r   rx   r   r   r   r   r   r   rs      r   r   zCppPrinter._print_floorD  X    499~"""$++diil34A648OOj\A3a0JJr   c                     t        |j                        dk(  sJ d| j                  |j                  d          d}|j                  rdt         d| dS |S r   r   r   s      r   r   zCppPrinter._print_FloorToIntI  r   r   c                     t        |j                        dk(  sJ d| j                  |j                  d          d}dt         d| dS )Nrv   std::trunc(r   r   r   r   )rx   r   r   r   r   s      r   rf   zCppPrinter._print_TruncToIntN  sM    499~"""$++diil34A6j\A3a00r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   r   r   r   rw   r   s     r   rl   zCppPrinter._print_TruncToFloatS  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   static_cast<double>(r   r   rw   r   s     r   rS   zCppPrinter._print_ToFloatW  s9    499~"""%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   r]   zCppPrinter._print_IntTrueDiva  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   r`   )r   r   rQ   rR   rK   s       r   ra   zCppPrinter._print_PowByNaturalh  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  rK   s       r   rd   zCppPrinter._print_FloatPowq  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 )Nr/   g      
std::sqrt(r   z1.0/std::sqrt(r   r   rH   z1.0/(z1.0/z1.0r   r   r  r   )	r   r   r   rJ   r   r   absr   float)r   r   rL   rM   r   s        r   rN   zCppPrinter._print_Powu  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 )Nrv   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k(  sJ d| j                  |j                  d          d}|j                  rdt         d| dS |S Nrv   z
std::ceil(r   r   r   r   r   r   s      r   r   zCppPrinter._print_ceiling  X    499~"""TYYq\231548OOj\A3a0JJr   c                     t        |j                        dk(  sJ d| j                  |j                  d          d}|j                  rdt         d| dS |S r  r   r   s      r   r   zCppPrinter._print_CeilToInt  r  r   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<rv   r   {r   }z	std::min<r   r   r   rx   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   r  rv   r   r  r   r  z	std::max<r   r  r  s        r   r   zCppPrinter._print_Max  r  r  c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	std::abs(r   r   rw   r   s     r   r   zCppPrinter._print_Abs  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	std::cos(r   r   rw   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_cos  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::cosh(r   r   rw   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_cosh  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::acos(r   r   rw   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_acos  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	std::sin(r   r   rw   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_sin  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::sinh(r   r   rw   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_sinh  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::asin(r   r   rw   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_asin  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z	std::tan(r   r   rw   r   s     r   r   z#CppPrinter._print_OpaqueUnaryFn_tan  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::tanh(r   r   rw   r   s     r   r   z$CppPrinter._print_OpaqueUnaryFn_tanh  r   r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   z
std::atan(r   r   rw   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_log2  r*  r   c                 z    t        |j                        dk(  sJ d| j                  |j                  d          dS )Nrv   zstd::lrint(r   r   rw   r   s     r   rh   zCppPrinter._print_RoundToInt  r   r   c                     t        |j                        dk(  sJ |j                  \  }}|j                  r|dk  sJ t        d| d      | j	                  |t
        d         }d| d| d|  d	S )
Nr   r   zOFor integer inputs, only non-negative ndigits are currently supported, but got .rH   z%static_cast<double>(std::nearbyint(1ez * z) * 1er   )rx   r   r   
ValueErrorr   r   )r   r   r   r   
number_strs        r   rj   zCppPrinter._print_RoundDecimal  s    499~"""))Q;;abiajjkl  &&vz%/@A
6wis:,fV]U]T^^_``r   c                      y)Ntruerr   r   s     r   _print_BooleanTruezCppPrinter._print_BooleanTrue  s    r   c                      y)Nfalserr   r   s     r   _print_BooleanFalsezCppPrinter._print_BooleanFalse  s    r   c                      y)Nz'std::numeric_limits<double>::infinity()rr   r   s     r   rU   zCppPrinter._print_Infinity  s    8r   c                 *    d| j                  |       S )N-)rU   r   s     r   rX   z"CppPrinter._print_NegativeInfinity  s    4''-.//r   N)+rm   rn   ro   rD   rq   rC   r   r   r   r9   r   r   rf   rl   rS   r[   r]   ra   rd   rN   r  r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rh   rj   r3  r6  rU   rX   rr   r   r   r   r     sv   
5:: 
# 
" " "
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   typingr   rD   sympy.printing.precedencer   r   sympy.printing.strr   r   r   r   r
   rt   r   rr   r   r   <module>r=     sQ    
   < ) 

r
* r
jU:K U:p`0 `0r   