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 = ['fft',
  37       'pvoc',
  38       'filter',
  39       'filterbank',
  40       'resampler',
  41       'sndfile',
  42       'sink_apple_audio',
  43       'sink_sndfile',
  44       'source_apple_audio',
  45       'source_sndfile']
  46
  47   write_msg("-- INFO: %d objects in total" % len(cpp_objects))
  48
  49   for this_object in cpp_objects:
  50       lint = 0
  51
  52       if this_object[-2:] == '_t':
  53           object_name = this_object[:-2]
  54       else:
  55           object_name = this_object
  56           write_msg("-- WARNING: %s does not end in _t" % this_object)
  57
  58       if object_name[:len('aubio_')] != 'aubio_':
  59           write_msg("-- WARNING: %s does not start n aubio_" % this_object)
  60
  61       write_msg("-- INFO: looking at", object_name)
  62       object_methods = filter(lambda x: this_object in x, cpp_output)
  63       object_methods = [a.strip() for a in object_methods]
  64       object_methods = filter(lambda x: not x.startswith('typedef'), object_methods)
  65       #for method in object_methods:
  66       #    write_msg(method)
  67       new_methods = filter(lambda x: 'new_'+object_name in x, object_methods)
  68       if len(new_methods) > 1:
  69           write_msg("-- WARNING: more than one new method for", object_name)
  70           for method in new_methods:
  71               write_msg(method)
  72       elif len(new_methods) < 1:
  73           write_msg("-- WARNING: no new method for", object_name)
  74       elif 0:
  75           for method in new_methods:
  76               write_msg(method)
  77
  78       del_methods = filter(lambda x: 'del_'+object_name in x, object_methods)
  79       if len(del_methods) > 1:
  80           write_msg("-- WARNING: more than one del method for", object_name)
  81           for method in del_methods:
  82               write_msg(method)
  83       elif len(del_methods) < 1:
  84           write_msg("-- WARNING: no del method for", object_name)
  85
  86       do_methods = filter(lambda x: object_name+'_do' in x, object_methods)
  87       if len(do_methods) > 1:
  88           pass
  89           #write_msg("-- WARNING: more than one do method for", object_name)
  90           #for method in do_methods:
  91           #    write_msg(method)
  92       elif len(do_methods) < 1:
  93           write_msg("-- WARNING: no do method for", object_name)
  94       elif 0:
  95           for method in do_methods:
  96               write_msg(method)
  97
  98       # check do methods return void
  99       for method in do_methods:
 100           if (method.split()[0] != 'void'):
 101               write_msg("-- ERROR: _do method does not return void:", method )
 102
 103       get_methods = filter(lambda x: object_name+'_get_' in x, object_methods)
 104
 105       set_methods = filter(lambda x: object_name+'_set_' in x, object_methods)
 106       for method in set_methods:
 107           if (method.split()[0] != 'uint_t'):
 108               write_msg("-- ERROR: _set method does not return uint_t:", method )
 109
 110       other_methods = filter(lambda x: x not in new_methods, object_methods)
 111       other_methods = filter(lambda x: x not in del_methods, other_methods)
 112       other_methods = filter(lambda x: x not in    do_methods, other_methods)
 113       other_methods = filter(lambda x: x not in get_methods, other_methods)
 114       other_methods = filter(lambda x: x not in set_methods, other_methods)
 115
 116       if len(other_methods) > 0:
 117           write_msg("-- WARNING: some methods for", object_name, "were unidentified")
 118           for method in other_methods:
 119               write_msg(method)
 120
 121
 122       # generate this_object
 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 = os.path.join(output_path,'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(os.path.join(output_path,'aubio-generated.h'), 'w')
 178   fd.write(s)
 179
 180   from os import listdir
 181   generated_files = listdir(output_path)
 182   generated_files = filter(lambda x: x.endswith('.c'), generated_files)
 183   generated_files = [output_path+'/'+f for f in generated_files]
 184   return generated_files
 185
 186 if __name__ == '__main__':
 187   generate_object_files('gen')