unroller.py 文件源码

def flatten_array_lookups(root):
    class Transformer(ast.NodeTransformer):
        def visit_FunctionDef(self, node):
            return node

        def visit_Subscript(self, node):
            self.generic_visit(node)

            # Get the indices being accessed.
            indices = slice_node_to_tuple_of_numbers(node.slice)
            variable_name = node.value.id
            index_name = flattened_array_name(variable_name, indices)
            return ast.copy_location(ast.Name(index_name, node.ctx), node)

    return Transformer().visit(root)