Current File : //lib/python3/dist-packages/pythran/analyses/node_count.py
"""
NodeCount counts the number of nodes in a node
"""
from pythran.passmanager import NodeAnalysis
class NodeCount(NodeAnalysis):
"""
Count the number of nodes included in a node
This has nothing to do with execution time or whatever,
its mainly use is to prevent the AST from growing too much when unrolling
>>> import gast as ast
>>> from pythran import passmanager, backend
>>> node = ast.parse("if 1: return 3")
>>> pm = passmanager.PassManager("test")
>>> print(pm.gather(NodeCount, node))
5
"""
def __init__(self):
self.result = 0
super(NodeCount, self).__init__()
def generic_visit(self, node):
self.result += 1
super(NodeCount, self).generic_visit(node)
Mini Shell