diff options
Diffstat (limited to 'contrib/find_typedefs.py')
-rw-r--r-- | contrib/find_typedefs.py | 95 |
1 files changed, 95 insertions, 0 deletions
diff --git a/contrib/find_typedefs.py b/contrib/find_typedefs.py new file mode 100644 index 000000000..d6db597da --- /dev/null +++ b/contrib/find_typedefs.py | |||
@@ -0,0 +1,95 @@ | |||
1 | from __future__ import print_function | ||
2 | import os | ||
3 | import re | ||
4 | import sys | ||
5 | |||
6 | debug = False | ||
7 | |||
8 | def get_td_from_function_signature (line, file, num): | ||
9 | left_paren = line.find ('(') | ||
10 | if left_paren > 0: | ||
11 | left_paren += 1 | ||
12 | line = line[left_paren:] | ||
13 | right_paren = line.find (')') | ||
14 | if right_paren > 0 and right_paren > left_paren and line[right_paren:].find ('(') >= 0: | ||
15 | fname = line[:right_paren] | ||
16 | fname = fname.lstrip (' ').lstrip ('*').lstrip (' ').rstrip (' ') | ||
17 | if len (fname) > 0: | ||
18 | if debug: | ||
19 | print ("from {0}:{1}".format (file, num)) | ||
20 | print ("-T {0}".format (fname)) | ||
21 | |||
22 | def get_td_from_simple_type (line, file, num): | ||
23 | line = line.rstrip (' ').rstrip ('\t').rstrip (' ').rstrip ('\t') | ||
24 | right_space = line.rfind (' ') | ||
25 | right_tab = line.rfind ('\t') | ||
26 | sep = right_tab if right_tab > right_space else right_space | ||
27 | sep += 1 | ||
28 | tname = line[sep:] | ||
29 | tname = tname.lstrip ('*') | ||
30 | if len (tname) > 0: | ||
31 | if debug: | ||
32 | print ("from {0}:{1}".format (file, num)) | ||
33 | print ("-T {0}".format (tname)) | ||
34 | |||
35 | def find_typedefs (file): | ||
36 | with open (file, 'rb') as f: | ||
37 | td = False | ||
38 | td_struct = False | ||
39 | td_level = 0 | ||
40 | td_line = [] | ||
41 | data = f.read () | ||
42 | for i, l in enumerate (data.splitlines (False)): | ||
43 | # Don't try to be too smart: only count lines that begin with 'typedef ' | ||
44 | l = l.rstrip (' ').rstrip ('\t') | ||
45 | if len (l) == 0: | ||
46 | continue | ||
47 | if not td: | ||
48 | if l[:8] != 'typedef ': | ||
49 | continue | ||
50 | else: | ||
51 | td = True | ||
52 | if l[8:].lstrip (' ').lstrip ('\t')[:6] == 'struct': | ||
53 | td_struct = True | ||
54 | if td_struct: | ||
55 | leftcbrace = l.find ('{') | ||
56 | if leftcbrace >= 0: | ||
57 | if td_level == 0: | ||
58 | td_line.append (l[:leftcbrace]) | ||
59 | l = l[leftcbrace + 1:] | ||
60 | td_level += 1 | ||
61 | rightcbrace = l.rfind ('}') | ||
62 | if rightcbrace >= 0: | ||
63 | td_level -= 1 | ||
64 | if td_level == 0: | ||
65 | td_line.append (l[rightcbrace + 1:]) | ||
66 | else: | ||
67 | td_line.append (l) | ||
68 | if len (l) > 0 and l[-1] == ';' and (not td_struct or td_level == 0): | ||
69 | td_line = ' '.join (td_line) | ||
70 | td_line = td_line[:-1] | ||
71 | if len (td_line) > 0: | ||
72 | if td_line[-1] == ')': | ||
73 | get_td_from_function_signature (td_line, file, i) | ||
74 | else: | ||
75 | get_td_from_simple_type (td_line, file, i) | ||
76 | td_line = [] | ||
77 | td = False | ||
78 | td_struct = False | ||
79 | td_level = 0 | ||
80 | |||
81 | |||
82 | def scan_dir (d): | ||
83 | for dirpath, dirs, files in os.walk (d): | ||
84 | for f in files: | ||
85 | if re.match (r'(?!lt_).+\.(c|cc|h)$', f): | ||
86 | file = os.path.join (dirpath, f) | ||
87 | find_typedefs (file) | ||
88 | |||
89 | |||
90 | if __name__ == '__main__': | ||
91 | if len (sys.argv[1:]) == 0: | ||
92 | arg = os.getcwd () | ||
93 | else: | ||
94 | arg = sys.argv[1] | ||
95 | scan_dir (arg) \ No newline at end of file | ||