blob: de5252b5ba83aad90e9955a4a151693f0ce55ea0 [file] [log] [blame]
eli.bendersky38165b72011-02-04 08:13:39 +02001import sys
Eli Bendersky3921e8e2010-05-21 09:05:39 +03002from pycparser.c_ast import *
3from pycparser.c_parser import CParser, Coord, ParseError
eli.bendersky38165b72011-02-04 08:13:39 +02004from pycparser.c_lexer import CLexer
Eli Bendersky3921e8e2010-05-21 09:05:39 +03005
6
7def expand_decl(decl):
8 """ Converts the declaration into a nested list.
9 """
10 typ = type(decl)
11
12 if typ == TypeDecl:
13 return ['TypeDecl', expand_decl(decl.type)]
14 elif typ == IdentifierType:
15 return ['IdentifierType', decl.names]
16 elif typ == ID:
17 return ['ID', decl.name]
18 elif typ in [Struct, Union]:
19 decls = [expand_decl(d) for d in decl.decls or []]
20 return [typ.__name__, decl.name, decls]
21 else:
22 nested = expand_decl(decl.type)
23
24 if typ == Decl:
25 if decl.quals:
26 return ['Decl', decl.quals, decl.name, nested]
27 else:
28 return ['Decl', decl.name, nested]
29 elif typ == Typename: # for function parameters
30 if decl.quals:
31 return ['Typename', decl.quals, nested]
32 else:
33 return ['Typename', nested]
34 elif typ == ArrayDecl:
35 dimval = decl.dim.value if decl.dim else ''
36 return ['ArrayDecl', dimval, nested]
37 elif typ == PtrDecl:
38 return ['PtrDecl', nested]
39 elif typ == Typedef:
40 return ['Typedef', decl.name, nested]
41 elif typ == FuncDecl:
42 if decl.args:
43 params = [expand_decl(param) for param in decl.args.params]
44 else:
45 params = []
46 return ['FuncDecl', params, nested]
47
48#-----------------------------------------------------------------
eli.bendersky724b1cc2011-03-05 10:45:08 +020049class NodeVisitor(object):
50 def __init__(self):
eli.bendersky3b52ac02011-03-18 06:32:23 +020051 self.current_parent = None
eli.bendersky724b1cc2011-03-05 10:45:08 +020052
53 def visit(self, node):
54 """ Visit a node.
55 """
56 method = 'visit_' + node.__class__.__name__
57 visitor = getattr(self, method, self.generic_visit)
58 return visitor(node)
eli.bendersky3b52ac02011-03-18 06:32:23 +020059
60 def visit_FuncCall(self, node):
61 print("Visiting FuncCall")
62 print(node.show())
63 print('---- parent ----')
64 print(self.current_parent.show())
65
eli.bendersky724b1cc2011-03-05 10:45:08 +020066 def generic_visit(self, node):
67 """ Called if no explicit visitor function exists for a
68 node. Implements preorder visiting of the node.
69 """
eli.bendersky3b52ac02011-03-18 06:32:23 +020070 oldparent = self.current_parent
71 self.current_parent = node
eli.bendersky724b1cc2011-03-05 10:45:08 +020072 for c in node.children():
73 self.visit(c)
eli.bendersky3b52ac02011-03-18 06:32:23 +020074 self.current_parent = oldparent
eli.bendersky724b1cc2011-03-05 10:45:08 +020075
Eli Bendersky3921e8e2010-05-21 09:05:39 +030076
eli.bendersky0e0a71f2010-10-09 08:32:00 +020077if __name__ == "__main__":
78 source_code = """
eli.bendersky84a6a632011-04-29 09:00:43 +030079typedef char FlagType;
Eli Bendersky3921e8e2010-05-21 09:05:39 +030080
eli.bendersky84a6a632011-04-29 09:00:43 +030081int main()
82{
83}
eli.bendersky724b1cc2011-03-05 10:45:08 +020084
eli.bendersky84a6a632011-04-29 09:00:43 +030085int myproc( int FlagType)
86{
87
88}
eli.bendersky724b1cc2011-03-05 10:45:08 +020089
eli.bendersky724b1cc2011-03-05 10:45:08 +020090
eli.bendersky724b1cc2011-03-05 10:45:08 +020091"""
eli.bendersky38165b72011-02-04 08:13:39 +020092
93 #--------------- Lexing
eli.bendersky38165b72011-02-04 08:13:39 +020094 #~ def errfoo(msg, a, b):
95 #~ printme(msg)
96 #~ sys.exit()
97 #~ clex = CLexer(errfoo, lambda t: False)
98 #~ clex.build()
99 #~ clex.input(source_code)
100
101 #~ while 1:
102 #~ tok = clex.token()
103 #~ if not tok: break
104
105 #~ printme([tok.value, tok.type, tok.lineno, clex.filename, tok.lexpos])
106
107 #--------------- Parsing
eli.bendersky0e0a71f2010-10-09 08:32:00 +0200108 parser = CParser()
eli.bendersky724b1cc2011-03-05 10:45:08 +0200109 ast = parser.parse(source_code, filename='zz')
110 ast.show(showcoord=True)
111 nv=NodeVisitor()
112 nv.visit(ast)
113
Eli Bendersky3921e8e2010-05-21 09:05:39 +0300114