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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# $Id: EmptyModuleRemover.py,v 1.7 2003/12/04 01:08:36 stefan Exp $
#
# Copyright (C) 2000 Stefan Seefeld
# Copyright (C) 2000 Stephen Davies
# All rights reserved.
# Licensed to the public under the terms of the GNU LGPL (>= 2),
# see the file COPYING for details.
#
from Synopsis.Processor import Processor, Parameter
from Synopsis import AST, Type
class EmptyModuleRemover (Processor, AST.Visitor):
"""A processor that removes empty modules"""
def process(self, ast, **kwds):
self.set_parameters(kwds)
self.ast = self.merge_input(ast)
self.__scopestack = []
self.__currscope = []
for decl in self.ast.declarations():
decl.accept(self)
self.ast.declarations()[:] = self.__currscope
return self.output_and_return_ast()
def push(self):
"""Pushes the current scope onto the stack and starts a new one"""
self.__scopestack.append(self.__currscope)
self.__currscope = []
def pop(self, decl):
"""Pops the current scope from the stack, and appends the given
declaration to it"""
self.__currscope = self.__scopestack.pop()
self.__currscope.append(decl)
def pop_only(self):
"""Only pops, doesn't append to scope"""
self.__currscope = self.__scopestack.pop()
def add(self, decl):
"""Adds the given decl to the current scope"""
self.__currscope.append(decl)
def visitDeclaration(self, decl):
"""Adds declaration to scope"""
self.add(decl)
visitBuiltin = visitDeclaration
visitGroup = visitDeclaration
visitEnum = visitDeclaration
def visitModule(self, module):
"""Visits all children of the module, and if there are no declarations
after that removes the module"""
self.push()
for decl in module.declarations():
decl.accept(self)
module.declarations()[:] = self.__currscope
# count the number of non-forward declarations in the current scope
count = reduce(lambda x, y: x + y,
map(lambda x: not isinstance(x, AST.Forward),
self.__currscope), 0)
if count: self.pop(module)
else: self.pop_only()
|
