1 | #! /usr/bin/python |
---|
2 | |
---|
3 | """ This file generates a c file from a list of cpp prototypes. """ |
---|
4 | |
---|
5 | import os, sys, shutil |
---|
6 | from gen_pyobject import write_msg, gen_new_init, gen_do, gen_members, gen_methods, gen_finish |
---|
7 | |
---|
8 | def get_cpp_objects(): |
---|
9 | |
---|
10 | cpp_output = [l.strip() for l in os.popen('cpp -DAUBIO_UNSTABLE=1 -I../build/src ../src/aubio.h').readlines()] |
---|
11 | |
---|
12 | cpp_output = filter(lambda y: len(y) > 1, cpp_output) |
---|
13 | cpp_output = filter(lambda y: not y.startswith('#'), cpp_output) |
---|
14 | |
---|
15 | i = 1 |
---|
16 | while 1: |
---|
17 | if i >= len(cpp_output): break |
---|
18 | if cpp_output[i-1].endswith(',') or cpp_output[i-1].endswith('{') or cpp_output[i].startswith('}'): |
---|
19 | cpp_output[i] = cpp_output[i-1] + ' ' + cpp_output[i] |
---|
20 | cpp_output.pop(i-1) |
---|
21 | else: |
---|
22 | i += 1 |
---|
23 | |
---|
24 | typedefs = filter(lambda y: y.startswith ('typedef struct _aubio'), cpp_output) |
---|
25 | |
---|
26 | cpp_objects = [a.split()[3][:-1] for a in typedefs] |
---|
27 | |
---|
28 | return cpp_output, cpp_objects |
---|
29 | |
---|
30 | def generate_object_files(output_path): |
---|
31 | if os.path.isdir(output_path): shutil.rmtree(output_path) |
---|
32 | os.mkdir(output_path) |
---|
33 | |
---|
34 | generated_objects = [] |
---|
35 | cpp_output, cpp_objects = get_cpp_objects() |
---|
36 | skip_objects = [ |
---|
37 | # already in ext/ |
---|
38 | 'fft', |
---|
39 | 'pvoc', |
---|
40 | 'filter', |
---|
41 | 'filterbank', |
---|
42 | #'resampler', |
---|
43 | # AUBIO_UNSTABLE |
---|
44 | 'hist', |
---|
45 | 'parameter', |
---|
46 | 'scale', |
---|
47 | 'beattracking', |
---|
48 | 'resampler', |
---|
49 | 'sndfile', |
---|
50 | 'peakpicker', |
---|
51 | 'pitchfcomb', |
---|
52 | 'pitchmcomb', |
---|
53 | 'pitchschmitt', |
---|
54 | 'pitchyin', |
---|
55 | 'pitchyinfft', |
---|
56 | 'sink_apple_audio', |
---|
57 | 'sink_sndfile', |
---|
58 | 'source_apple_audio', |
---|
59 | 'source_sndfile', |
---|
60 | 'source_avcodec', |
---|
61 | #'sampler', |
---|
62 | 'audio_unit', |
---|
63 | ] |
---|
64 | |
---|
65 | write_msg("-- INFO: %d objects in total" % len(cpp_objects)) |
---|
66 | |
---|
67 | for this_object in cpp_objects: |
---|
68 | lint = 0 |
---|
69 | |
---|
70 | if this_object[-2:] == '_t': |
---|
71 | object_name = this_object[:-2] |
---|
72 | else: |
---|
73 | object_name = this_object |
---|
74 | write_msg("-- WARNING: %s does not end in _t" % this_object) |
---|
75 | |
---|
76 | if object_name[:len('aubio_')] != 'aubio_': |
---|
77 | write_msg("-- WARNING: %s does not start n aubio_" % this_object) |
---|
78 | |
---|
79 | write_msg("-- INFO: looking at", object_name) |
---|
80 | object_methods = filter(lambda x: this_object in x, cpp_output) |
---|
81 | object_methods = [a.strip() for a in object_methods] |
---|
82 | object_methods = filter(lambda x: not x.startswith('typedef'), object_methods) |
---|
83 | #for method in object_methods: |
---|
84 | # write_msg(method) |
---|
85 | new_methods = filter(lambda x: 'new_'+object_name in x, object_methods) |
---|
86 | if len(new_methods) > 1: |
---|
87 | write_msg("-- WARNING: more than one new method for", object_name) |
---|
88 | for method in new_methods: |
---|
89 | write_msg(method) |
---|
90 | elif len(new_methods) < 1: |
---|
91 | write_msg("-- WARNING: no new method for", object_name) |
---|
92 | elif 0: |
---|
93 | for method in new_methods: |
---|
94 | write_msg(method) |
---|
95 | |
---|
96 | del_methods = filter(lambda x: 'del_'+object_name in x, object_methods) |
---|
97 | if len(del_methods) > 1: |
---|
98 | write_msg("-- WARNING: more than one del method for", object_name) |
---|
99 | for method in del_methods: |
---|
100 | write_msg(method) |
---|
101 | elif len(del_methods) < 1: |
---|
102 | write_msg("-- WARNING: no del method for", object_name) |
---|
103 | |
---|
104 | do_methods = filter(lambda x: object_name+'_do' in x, object_methods) |
---|
105 | if len(do_methods) > 1: |
---|
106 | pass |
---|
107 | #write_msg("-- WARNING: more than one do method for", object_name) |
---|
108 | #for method in do_methods: |
---|
109 | # write_msg(method) |
---|
110 | elif len(do_methods) < 1: |
---|
111 | write_msg("-- WARNING: no do method for", object_name) |
---|
112 | elif 0: |
---|
113 | for method in do_methods: |
---|
114 | write_msg(method) |
---|
115 | |
---|
116 | # check do methods return void |
---|
117 | for method in do_methods: |
---|
118 | if (method.split()[0] != 'void'): |
---|
119 | write_msg("-- ERROR: _do method does not return void:", method ) |
---|
120 | |
---|
121 | get_methods = filter(lambda x: object_name+'_get_' in x, object_methods) |
---|
122 | |
---|
123 | set_methods = filter(lambda x: object_name+'_set_' in x, object_methods) |
---|
124 | for method in set_methods: |
---|
125 | if (method.split()[0] != 'uint_t'): |
---|
126 | write_msg("-- ERROR: _set method does not return uint_t:", method ) |
---|
127 | |
---|
128 | other_methods = filter(lambda x: x not in new_methods, object_methods) |
---|
129 | other_methods = filter(lambda x: x not in del_methods, other_methods) |
---|
130 | other_methods = filter(lambda x: x not in do_methods, other_methods) |
---|
131 | other_methods = filter(lambda x: x not in get_methods, other_methods) |
---|
132 | other_methods = filter(lambda x: x not in set_methods, other_methods) |
---|
133 | |
---|
134 | if len(other_methods) > 0: |
---|
135 | write_msg("-- WARNING: some methods for", object_name, "were unidentified") |
---|
136 | for method in other_methods: |
---|
137 | write_msg(method) |
---|
138 | |
---|
139 | |
---|
140 | # generate this_object |
---|
141 | short_name = object_name[len('aubio_'):] |
---|
142 | if short_name in skip_objects: |
---|
143 | write_msg("-- INFO: skipping object", short_name ) |
---|
144 | continue |
---|
145 | if 1: #try: |
---|
146 | s = gen_new_init(new_methods[0], short_name) |
---|
147 | s += gen_do(do_methods[0], short_name) |
---|
148 | s += gen_members(new_methods[0], short_name) |
---|
149 | s += gen_methods(get_methods, set_methods, short_name) |
---|
150 | s += gen_finish(short_name) |
---|
151 | generated_filepath = os.path.join(output_path,'gen-'+short_name+'.c') |
---|
152 | fd = open(generated_filepath, 'w') |
---|
153 | fd.write(s) |
---|
154 | #except Exception, e: |
---|
155 | # write_msg("-- ERROR:", type(e), str(e), "in", short_name) |
---|
156 | # continue |
---|
157 | generated_objects += [this_object] |
---|
158 | |
---|
159 | s = """// generated list of objects created with generator.py |
---|
160 | |
---|
161 | """ |
---|
162 | |
---|
163 | types_ready = [] |
---|
164 | for each in generated_objects: |
---|
165 | types_ready.append(" PyType_Ready (&Py_%sType) < 0" % \ |
---|
166 | each.replace('aubio_','').replace('_t','') ) |
---|
167 | |
---|
168 | s = """// generated list of objects created with generator.py |
---|
169 | |
---|
170 | #include "aubio-generated.h" |
---|
171 | """ |
---|
172 | |
---|
173 | s += """ |
---|
174 | int generated_types_ready (void) |
---|
175 | { |
---|
176 | return ( |
---|
177 | """ |
---|
178 | s += ('\n ||').join(types_ready) |
---|
179 | s += """); |
---|
180 | } |
---|
181 | """ |
---|
182 | |
---|
183 | s += """ |
---|
184 | void add_generated_objects ( PyObject *m ) |
---|
185 | {""" |
---|
186 | for each in generated_objects: |
---|
187 | s += """ |
---|
188 | Py_INCREF (&Py_%(name)sType); |
---|
189 | PyModule_AddObject (m, "%(name)s", (PyObject *) & Py_%(name)sType);""" % \ |
---|
190 | { 'name': ( each.replace('aubio_','').replace('_t','') ) } |
---|
191 | |
---|
192 | s += """ |
---|
193 | }""" |
---|
194 | |
---|
195 | fd = open(os.path.join(output_path,'aubio-generated.c'), 'w') |
---|
196 | fd.write(s) |
---|
197 | |
---|
198 | s = """// generated list of objects created with generator.py |
---|
199 | |
---|
200 | #include <Python.h> |
---|
201 | |
---|
202 | """ |
---|
203 | |
---|
204 | for each in generated_objects: |
---|
205 | s += "extern PyTypeObject Py_%sType;\n" % \ |
---|
206 | each.replace('aubio_','').replace('_t','') |
---|
207 | |
---|
208 | s+= "int generated_objects ( void );\n" |
---|
209 | s+= "void add_generated_objects( PyObject *m );\n" |
---|
210 | |
---|
211 | fd = open(os.path.join(output_path,'aubio-generated.h'), 'w') |
---|
212 | fd.write(s) |
---|
213 | |
---|
214 | from os import listdir |
---|
215 | generated_files = listdir(output_path) |
---|
216 | generated_files = filter(lambda x: x.endswith('.c'), generated_files) |
---|
217 | generated_files = [output_path+'/'+f for f in generated_files] |
---|
218 | return generated_files |
---|
219 | |
---|
220 | if __name__ == '__main__': |
---|
221 | generate_object_files('gen') |
---|