o í@Ëa˜ã@s<dZddlmZddlZGdd„deƒZGdd„deƒZdS)z‰ LocalNameDeclarations gathers name of declarations local to a node. LocalNodeDeclarations gathers node of declarations local to a node. é)ÚNodeAnalysisNcs(eZdZdZ‡fdd„Zdd„Z‡ZS)ÚLocalNodeDeclarationsa² Gathers all local symbols from a function. It should not be use from outside a function, but can be used on a function (but in that case, parameters are not taken into account) >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse(''' ... def foo(a): ... b = a + 1''') >>> pm = passmanager.PassManager("test") >>> [name.id for name in pm.gather(LocalNodeDeclarations, node)] ['b'] >>> node = ast.parse(''' ... for c in range(n): ... b = a + 1''') >>> pm = passmanager.PassManager("test") >>> sorted([name.id for name in pm.gather(LocalNodeDeclarations, node)]) ['b', 'c'] cótƒ|_tt|ƒ ¡dS©z% Initialize empty set as the result. N)ÚsetÚresultÚsuperrÚ__init__©Úself©Ú __class__©úE/usr/lib/python3/dist-packages/pythran/analyses/local_declarations.pyr #ózLocalNodeDeclarations.__init__cCs"t|jtjƒr|j |¡dSdS)z3 Any node with Store context is a new declaration. N)Ú isinstanceÚctxÚastÚStorerÚadd©rZnoderrrÚ visit_Name(sÿz LocalNodeDeclarations.visit_Name)Ú__name__Ú __module__Ú__qualname__Ú__doc__r rÚ __classcell__rrrrrsrcs0eZdZdZ‡fdd„Zdd„Zdd„Z‡ZS)ÚLocalNameDeclarationsaF Gathers all local identifiers from a node. >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse(''' ... def foo(a): ... b = a + 1''') >>> pm = passmanager.PassManager("test") >>> sorted(pm.gather(LocalNameDeclarations, node)) ['a', 'b', 'foo'] crr)rrrrr r rrrr =rzLocalNameDeclarations.__init__cCs*t|jtjtjfƒr|j |j¡dSdS)z; Any node with Store or Param context is a new identifier. N)rrrrZParamrrÚidrrrrrBsÿz LocalNameDeclarations.visit_NamecCs|j |j¡| |¡dS)z) Function name is a possible identifier. N)rrÚnameZ generic_visitrrrrÚvisit_FunctionDefGsz'LocalNameDeclarations.visit_FunctionDef)rrrrr rr rrrrrr.s r)rZpythran.passmanagerrZgastrrrrrrrÚs #