Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1 | from pycparser.c_ast import *
|
| 2 | from pycparser.c_parser import CParser, Coord, ParseError
|
| 3 |
|
| 4 |
|
| 5 | def expand_decl(decl):
|
| 6 | """ Converts the declaration into a nested list.
|
| 7 | """
|
| 8 | typ = type(decl)
|
| 9 |
|
| 10 | if typ == TypeDecl:
|
| 11 | return ['TypeDecl', expand_decl(decl.type)]
|
| 12 | elif typ == IdentifierType:
|
| 13 | return ['IdentifierType', decl.names]
|
| 14 | elif typ == ID:
|
| 15 | return ['ID', decl.name]
|
| 16 | elif typ in [Struct, Union]:
|
| 17 | decls = [expand_decl(d) for d in decl.decls or []]
|
| 18 | return [typ.__name__, decl.name, decls]
|
| 19 | else:
|
| 20 | nested = expand_decl(decl.type)
|
| 21 |
|
| 22 | if typ == Decl:
|
| 23 | if decl.quals:
|
| 24 | return ['Decl', decl.quals, decl.name, nested]
|
| 25 | else:
|
| 26 | return ['Decl', decl.name, nested]
|
| 27 | elif typ == Typename: # for function parameters
|
| 28 | if decl.quals:
|
| 29 | return ['Typename', decl.quals, nested]
|
| 30 | else:
|
| 31 | return ['Typename', nested]
|
| 32 | elif typ == ArrayDecl:
|
| 33 | dimval = decl.dim.value if decl.dim else ''
|
| 34 | return ['ArrayDecl', dimval, nested]
|
| 35 | elif typ == PtrDecl:
|
| 36 | return ['PtrDecl', nested]
|
| 37 | elif typ == Typedef:
|
| 38 | return ['Typedef', decl.name, nested]
|
| 39 | elif typ == FuncDecl:
|
| 40 | if decl.args:
|
| 41 | params = [expand_decl(param) for param in decl.args.params]
|
| 42 | else:
|
| 43 | params = []
|
| 44 | return ['FuncDecl', params, nested]
|
| 45 |
|
| 46 | #-----------------------------------------------------------------
|
| 47 |
|
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 48 | if __name__ == "__main__":
|
| 49 | source_code = """
|
| 50 | int main()
|
| 51 | {
|
eli.bendersky | 4a31472 | 2010-12-10 07:19:31 +0200 | [diff] [blame^] | 52 | const union blahunion tt = {
|
| 53 | .joe = {0, 1},
|
| 54 | .boo.bar[2] = 4};
|
eli.bendersky | 38ed9a9 | 2010-10-09 09:29:59 +0200 | [diff] [blame] | 55 | p++;
|
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 56 | int a;
|
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 57 | }
|
| 58 | """
|
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 59 |
|
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 60 | parser = CParser()
|
| 61 | ast = parser.parse(source_code)
|
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 62 |
|
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 63 | ast.show()
|
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 64 |
|