source: interfaces/python/generator.py @ 0b5b7b2

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

generator.py: skip sinks for now

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