o @az"@sdZddlmZmZddlmZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZmZdd lZdd lmZdd lZed ZGd d d eZGdddeZd S)z< ConstantFolding performs some kind of partial evaluation. )ConstantExpressions ASTMatcher)Transformation)MODULES)to_astConversionError ToNotEvalmangle)DamnTooLongPattern)PythranSyntaxError) isintegralisnumN)deepcopyZpythrancsTeZdZdZddZfddZddZeZZe j Z Z e j Z Zdd Z ZS) ConstantFoldingae Replace constant expression by their evaluation. >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(): return 1+3") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(ConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(): return 4 cCt|tdSNr__init__rselfrH/usr/lib/python3/dist-packages/pythran/optimizations/constant_folding.pyr!zConstantFolding.__init__c st|tjsJdtdi|_tD]}|dkr.t|}z t||j|<Wqty-Yqwqtdd|jDg}t t t |dd|jt t ||dS)Nbuiltins __dispatch__cSsg|] }t|tjs|qSr) isinstanceastZImport).0srrr 6s  z+ConstantFolding.prepare..zexec)rrZModule __import__envrr ImportErrorbodyevalcompile gast_to_astsuperrprepare)rnodeZ module_nameZalias_module_nameZ dummy_module __class__rrr)$s*   zConstantFolding.preparecCs|Srrrr*rrrskip?szConstantFolding.skipc Cst|tjr||jvrt|}tt|dd}z:t||j}t |}zt | |s3d|_ |WWSWnt yL}z td|dWYd}~nd}~wwt||WStye}ztd|d}~wtyst||YSty}ztd|jdt||WYd}~Sd}~wty}z)d |jdvrt||WYd}~Std|jdt||WYd}~Sd}~wty}ztt||d}~wwt||S) Nzr%TzW: z Assume no update happened.zerror in constant folding: z-During constant folding, bailing out due to: rr)rrexprconstant_expressionsZ Expressionr&r'r%r"rrsearchupdater printr generic_visitrrAttributeErrorloggerinfoargs NameError Exceptionr str)rr*Z fake_nodecodevalueZnew_nodeerrrr4GsV      zConstantFolding.generic_visit)__name__ __module__ __qualname____doc__rr)r.Zvisit_ConstantZ visit_Namerr4Z visit_ListZ visit_SetZ visit_DictZ visit_Tuple __classcell__rrr+rrs   rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)PartialConstantFoldinga Replace partially constant expression by their evaluation. >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(n): return [n] * 2") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(PartialConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(n): return [n, n] >>> node = ast.parse("def foo(n): return 2 * (n,)") >>> _, node = pm.apply(PartialConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(n): return (n, n) >>> node = ast.parse("def foo(n): return [n] + [n]") >>> _, node = pm.apply(PartialConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(n): return [n, n] >>> node = ast.parse("def foo(n, m): return (n,) + (m, n)") >>> _, node = pm.apply(PartialConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(n, m): return (n, m, n) cCrrrrrrrrrzPartialConstantFolding.__init__cCHt|jtjtjfs dSt|jsdSt|jstd|t|j tj SNFz!Multiplying a sequence by a float) rleftrListTupler rightr r opMultr-rrrfold_mult_left   z%PartialConstantFolding.fold_mult_leftcCrErF) rrJrrHrIr rGr r rKrLr-rrrfold_mult_rightrNz&PartialConstantFolding.fold_mult_rightcCs.t|j|sdSt|j|sdSt|jtjS)NF)rrGrJrKrAddrr*Ztyrrrfold_adds  zPartialConstantFolding.fold_addcs|jvrS||r&d|_fddtjjDj_jS| r.)r0r4rMr2rangerJr=rGrSrOZvisitrrHrIrRrQrrVr visit_BinOps(          z"PartialConstantFolding.visit_BinOpc Cs||t|jtjs|St|jjtjs|St|js|S|jj}|j}|j}|jp1t dd}|j p:t dd}t |t t |t ||_d|_|S)a >>> import gast as ast >>> from pythran import passmanager, backend >>> pm = passmanager.PassManager("test") >>> node = ast.parse("def foo(a): a[1:][3]") >>> _, node = pm.apply(PartialConstantFolding, node) >>> _, node = pm.apply(ConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(a): a[4] >>> node = ast.parse("def foo(a): a[::2][3]") >>> _, node = pm.apply(PartialConstantFolding, node) >>> _, node = pm.apply(ConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(a): a[6] >>> node = ast.parse("def foo(a): a[-4:][5]") >>> _, node = pm.apply(PartialConstantFolding, node) >>> _, node = pm.apply(ConstantFolding, node) >>> print(pm.dump(backend.Python, node)) def foo(a): a[1] rNT)r4rr=rZ SubscriptsliceZSlicer lowerZConstantstepZBinOprPrLr2)rr*Zslice_indexr[r\rrrvisit_Subscripts*  z&PartialConstantFolding.visit_SubscriptN) r?r@rArBrrMrOrRrXr^rrrrrDos    rD)rBZpythran.analysesrrZpythran.passmanagerrZpythran.tablesrZpythran.conversionrrrr Zpythran.analyses.ast_matcherr Zpythran.syntaxr Z pythran.utilsr r ZgastrcopyrZloggingZ getLoggerr6rrDrrrrs      ]