B \ @s`ddlmZddlmZmZddlmZmZedGdddeZedGdd d eZ d S) )print_function)irerrors)register_rewriteRewritezbefore-inferencec@s eZdZdZddZddZdS)RewritePrintCallszU Rewrite calls to the print() global function to dedicated IR print() nodes. c Csi|_}||_x|tjD]v}t|jtjr|jjdkr|j}y| |j }Wnt j k rjwYnX|t kr|jrd} t | |j|||<qWt|dkS)NZcallzKNumba's print() function implementation does not support keyword arguments.r)printsblock find_instsrAssign isinstancevalueZExpropinfer_constantfuncrConstantInferenceErrorprintZkwsZUnsupportedErrorloclen) selffunc_irr typemap calltypesr instexprZcalleemsgr6lib/python3.7/site-packages/numba/rewrites/ir_print.pymatch s  zRewritePrintCalls.matchcCs|j}|xz|jjD]n}||jkr|j|}tj|j|j|j d}| |tj tj d|j d|j |j d}| |q| |qW|S)z{ Rewrite `var = call (...)` as a sequence of `print(...)` and `var = const(None)`. )argsvarargrN)r)rtargetr)r copyclearbodyr rPrintr r!rappendr ZConstr")rZ new_blockrrZ print_nodeZ assign_noderrrapply!s        zRewritePrintCalls.applyN)__name__ __module__ __qualname____doc__rr(rrrrrsrc@s eZdZdZddZddZdS)DetectConstPrintArgumentsz? Detect and store constant arguments to print() nodes. c Csi|_}||_xl|tjD]\}|jr*qxNt|jD]@\}}y||} Wntj k rdw6YnX| | |i|<q6WqWt |dkS)Nr) constsr r rr& enumerater rrr setdefaultr) rrr rrr.ridxvarZconstrrrr=s zDetectConstPrintArguments.matchcCs.x&|jjD]}||jkr |j||_q W|jS)zC Store detected constant arguments on their nodes. )r r%r.)rrrrrr(Ms zDetectConstPrintArguments.applyN)r)r*r+r,rr(rrrrr-7sr-N) Z __future__rZnumbarrrrrr-rrrrs /