source: python/lib/generator.py @ c2e0aef

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5pitchshiftsamplertimestretchyinfft+
Last change on this file since c2e0aef was 7609f6d, checked in by Paul Brossier <piem@piem.org>, 11 years ago

python/lib/generator.py: remove audio_unit

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