blob: 23c6b3ad30f8fc78ba01c118ecccd9bb4007911a (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
"""
PotentialIterator finds if it is possible to use an iterator.
"""
from pythran.analyses.aliases import Aliases
from pythran.analyses.argument_read_once import ArgumentReadOnce
from pythran.passmanager import NodeAnalysis
import gast as ast
class PotentialIterator(NodeAnalysis):
"""Find whether an expression can be replaced with an iterator."""
def __init__(self):
self.result = set()
NodeAnalysis.__init__(self, Aliases, ArgumentReadOnce)
def visit_For(self, node):
self.result.add(node.iter)
self.generic_visit(node)
def visit_Compare(self, node):
if isinstance(node.ops[0], (ast.In, ast.NotIn)):
self.result.update(node.comparators)
self.generic_visit(node)
def visit_Call(self, node):
for i, arg in enumerate(node.args):
def isReadOnce(f, i):
return (f in self.argument_read_once and
self.argument_read_once[f][i] <= 1)
if all(isReadOnce(alias, i) for alias in self.aliases[node.func]):
self.result.add(arg)
self.generic_visit(node)
|
