python/lib/generator.py

   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')