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       #'sampler',
  61       'audio_unit',
  62       ]
  63
  64   write_msg("-- INFO: %d objects in total" % len(cpp_objects))
  65
  66   for this_object in cpp_objects:
  67       lint = 0
  68
  69       if this_object[-2:] == '_t':
  70           object_name = this_object[:-2]
  71       else:
  72           object_name = this_object
  73           write_msg("-- WARNING: %s does not end in _t" % this_object)
  74
  75       if object_name[:len('aubio_')] != 'aubio_':
  76           write_msg("-- WARNING: %s does not start n aubio_" % this_object)
  77
  78       write_msg("-- INFO: looking at", object_name)
  79       object_methods = filter(lambda x: this_object in x, cpp_output)
  80       object_methods = [a.strip() for a in object_methods]
  81       object_methods = filter(lambda x: not x.startswith('typedef'), object_methods)
  82       #for method in object_methods:
  83       #    write_msg(method)
  84       new_methods = filter(lambda x: 'new_'+object_name in x, object_methods)
  85       if len(new_methods) > 1:
  86           write_msg("-- WARNING: more than one new method for", object_name)
  87           for method in new_methods:
  88               write_msg(method)
  89       elif len(new_methods) < 1:
  90           write_msg("-- WARNING: no new method for", object_name)
  91       elif 0:
  92           for method in new_methods:
  93               write_msg(method)
  94
  95       del_methods = filter(lambda x: 'del_'+object_name in x, object_methods)
  96       if len(del_methods) > 1:
  97           write_msg("-- WARNING: more than one del method for", object_name)
  98           for method in del_methods:
  99               write_msg(method)
 100       elif len(del_methods) < 1:
 101           write_msg("-- WARNING: no del method for", object_name)
 102
 103       do_methods = filter(lambda x: object_name+'_do' in x, object_methods)
 104       if len(do_methods) > 1:
 105           pass
 106           #write_msg("-- WARNING: more than one do method for", object_name)
 107           #for method in do_methods:
 108           #    write_msg(method)
 109       elif len(do_methods) < 1:
 110           write_msg("-- WARNING: no do method for", object_name)
 111       elif 0:
 112           for method in do_methods:
 113               write_msg(method)
 114
 115       # check do methods return void
 116       for method in do_methods:
 117           if (method.split()[0] != 'void'):
 118               write_msg("-- ERROR: _do method does not return void:", method )
 119
 120       get_methods = filter(lambda x: object_name+'_get_' in x, object_methods)
 121
 122       set_methods = filter(lambda x: object_name+'_set_' in x, object_methods)
 123       for method in set_methods:
 124           if (method.split()[0] != 'uint_t'):
 125               write_msg("-- ERROR: _set method does not return uint_t:", method )
 126
 127       other_methods = filter(lambda x: x not in new_methods, object_methods)
 128       other_methods = filter(lambda x: x not in del_methods, other_methods)
 129       other_methods = filter(lambda x: x not in    do_methods, other_methods)
 130       other_methods = filter(lambda x: x not in get_methods, other_methods)
 131       other_methods = filter(lambda x: x not in set_methods, other_methods)
 132
 133       if len(other_methods) > 0:
 134           write_msg("-- WARNING: some methods for", object_name, "were unidentified")
 135           for method in other_methods:
 136               write_msg(method)
 137
 138
 139       # generate this_object
 140       short_name = object_name[len('aubio_'):]
 141       if short_name in skip_objects:
 142               write_msg("-- INFO: skipping object", short_name )
 143               continue
 144       if 1: #try:
 145           s = gen_new_init(new_methods[0], short_name)
 146           s += gen_do(do_methods[0], short_name)
 147           s += gen_members(new_methods[0], short_name)
 148           s += gen_methods(get_methods, set_methods, short_name)
 149           s += gen_finish(short_name)
 150           generated_filepath = os.path.join(output_path,'gen-'+short_name+'.c')
 151           fd = open(generated_filepath, 'w')
 152           fd.write(s)
 153       #except Exception, e:
 154       #        write_msg("-- ERROR:", type(e), str(e), "in", short_name)
 155       #        continue
 156       generated_objects += [this_object]
 157
 158   s = """// generated list of objects created with generator.py
 159
 160 """
 161
 162   types_ready = []
 163   for each in generated_objects:
 164       types_ready.append("  PyType_Ready (&Py_%sType) < 0" % \
 165               each.replace('aubio_','').replace('_t','') )
 166
 167   s = """// generated list of objects created with generator.py
 168
 169 #include "aubio-generated.h"
 170 """
 171
 172   s += """
 173 int generated_types_ready (void)
 174 {
 175   return (
 176 """
 177   s += ('\n     ||').join(types_ready)
 178   s += """);
 179 }
 180 """
 181
 182   s += """
 183 void add_generated_objects ( PyObject *m )
 184 {"""
 185   for each in generated_objects:
 186     s += """
 187   Py_INCREF (&Py_%(name)sType);
 188   PyModule_AddObject (m, "%(name)s", (PyObject *) & Py_%(name)sType);""" % \
 189           { 'name': ( each.replace('aubio_','').replace('_t','') ) }
 190
 191   s += """
 192 }"""
 193
 194   fd = open(os.path.join(output_path,'aubio-generated.c'), 'w')
 195   fd.write(s)
 196
 197   s = """// generated list of objects created with generator.py
 198
 199 #include <Python.h>
 200
 201 """
 202
 203   for each in generated_objects:
 204       s += "extern PyTypeObject Py_%sType;\n" % \
 205               each.replace('aubio_','').replace('_t','')
 206
 207   s+= "int generated_objects ( void );\n"
 208   s+= "void add_generated_objects( PyObject *m );\n"
 209
 210   fd = open(os.path.join(output_path,'aubio-generated.h'), 'w')
 211   fd.write(s)
 212
 213   from os import listdir
 214   generated_files = listdir(output_path)
 215   generated_files = filter(lambda x: x.endswith('.c'), generated_files)
 216   generated_files = [output_path+'/'+f for f in generated_files]
 217   return generated_files
 218
 219 if __name__ == '__main__':
 220   generate_object_files('gen')