import distutils.ccompiler import sys, os, subprocess, glob header = os.path.join('src', 'aubio.h') output_path = os.path.join('python', 'gen') source_header = """// this file is generated! do not modify #include "aubio-types.h" """ skip_objects = [ # already in ext/ 'fft', 'pvoc', 'filter', 'filterbank', #'resampler', # AUBIO_UNSTABLE 'hist', 'parameter', 'scale', 'beattracking', 'resampler', 'peakpicker', 'pitchfcomb', 'pitchmcomb', 'pitchschmitt', 'pitchspecacf', 'pitchyin', 'pitchyinfft', 'sink', 'sink_apple_audio', 'sink_sndfile', 'sink_wavwrite', #'mfcc', 'source', 'source_apple_audio', 'source_sndfile', 'source_avcodec', 'source_wavread', #'sampler', 'audio_unit', ] def get_preprocessor(): # findout which compiler to use from distutils.sysconfig import customize_compiler compiler_name = distutils.ccompiler.get_default_compiler() compiler = distutils.ccompiler.new_compiler(compiler=compiler_name) try: customize_compiler(compiler) except AttributeError as e: print("Warning: failed customizing compiler ({:s})".format(repr(e))) if hasattr(compiler, 'initialize'): try: compiler.initialize() except ValueError as e: print("Warning: failed initializing compiler ({:s})".format(repr(e))) cpp_cmd = None if hasattr(compiler, 'preprocessor'): # for unixccompiler cpp_cmd = compiler.preprocessor elif hasattr(compiler, 'compiler'): # for ccompiler cpp_cmd = compiler.compiler.split() cpp_cmd += ['-E'] elif hasattr(compiler, 'cc'): # for msvccompiler cpp_cmd = compiler.cc.split() cpp_cmd += ['-E'] if not cpp_cmd: print("Warning: could not guess preprocessor, using env's CC") cpp_cmd = os.environ.get('CC', 'cc').split() cpp_cmd += ['-E'] return cpp_cmd def get_cpp_objects(header=header): cpp_cmd = get_preprocessor() macros = [('AUBIO_UNSTABLE', 1)] if not os.path.isfile(header): raise Exception("could not find include file " + header) includes = [os.path.dirname(header)] cpp_cmd += distutils.ccompiler.gen_preprocess_options(macros, includes) cpp_cmd += [header] print("Running command: {:s}".format(" ".join(cpp_cmd))) proc = subprocess.Popen(cpp_cmd, stderr=subprocess.PIPE, stdout=subprocess.PIPE) assert proc, 'Proc was none' cpp_output = proc.stdout.read() err_output = proc.stderr.read() if not cpp_output: raise Exception("preprocessor output is empty:\n%s" % err_output) elif err_output: print ("Warning: preprocessor produced warnings:\n%s" % err_output) if not isinstance(cpp_output, list): cpp_output = [l.strip() for l in cpp_output.decode('utf8').split('\n')] cpp_output = filter(lambda y: len(y) > 1, cpp_output) cpp_output = list(filter(lambda y: not y.startswith('#'), cpp_output)) i = 1 while 1: if i >= len(cpp_output): break if cpp_output[i-1].endswith(',') or cpp_output[i-1].endswith('{') or cpp_output[i].startswith('}'): cpp_output[i] = cpp_output[i-1] + ' ' + cpp_output[i] cpp_output.pop(i-1) else: i += 1 typedefs = filter(lambda y: y.startswith ('typedef struct _aubio'), cpp_output) cpp_objects = [a.split()[3][:-1] for a in typedefs] return cpp_output, cpp_objects def analyze_cpp_output(cpp_objects, cpp_output): lib = {} for o in cpp_objects: if o[:6] != 'aubio_': continue shortname = o[6:-2] if shortname in skip_objects: continue lib[shortname] = {'struct': [], 'new': [], 'del': [], 'do': [], 'get': [], 'set': [], 'other': []} lib[shortname]['longname'] = o lib[shortname]['shortname'] = shortname for fn in cpp_output: if o[:-1] in fn: #print "found", o[:-1], "in", fn if 'typedef struct ' in fn: lib[shortname]['struct'].append(fn) elif '_do' in fn: lib[shortname]['do'].append(fn) elif 'new_' in fn: lib[shortname]['new'].append(fn) elif 'del_' in fn: lib[shortname]['del'].append(fn) elif '_get_' in fn: lib[shortname]['get'].append(fn) elif '_set_' in fn: lib[shortname]['set'].append(fn) else: #print "no idea what to do about", fn lib[shortname]['other'].append(fn) return lib def print_cpp_output_results(lib, cpp_output): for fn in cpp_output: found = 0 for o in lib: for family in lib[o]: if fn in lib[o][family]: found = 1 if found == 0: print ("missing", fn) for o in lib: for family in lib[o]: if type(lib[o][family]) == str: print ( "{:15s} {:10s} {:s}".format(o, family, lib[o][family] ) ) elif len(lib[o][family]) == 1: print ( "{:15s} {:10s} {:s}".format(o, family, lib[o][family][0] ) ) else: print ( "{:15s} {:10s} {:s}".format(o, family, lib[o][family] ) ) def generate_external(header=header, output_path=output_path, usedouble=False, overwrite=True): if not os.path.isdir(output_path): os.mkdir(output_path) elif not overwrite: return glob.glob(os.path.join(output_path, '*.c')) cpp_output, cpp_objects = get_cpp_objects(header) lib = analyze_cpp_output(cpp_objects, cpp_output) # print_cpp_output_results(lib, cpp_output) sources_list = [] try: from .gen_code import MappedObject except (SystemError, ValueError): from gen_code import MappedObject for o in lib: out = source_header mapped = MappedObject(lib[o], usedouble = usedouble) out += mapped.gen_code() output_file = os.path.join(output_path, 'gen-%s.c' % o) with open(output_file, 'w') as f: f.write(out) print ("wrote %s" % output_file ) sources_list.append(output_file) out = source_header out += "#include \"aubio-generated.h\"" check_types = "\n || ".join(["PyType_Ready(&Py_%sType) < 0" % o for o in lib]) out += """ int generated_types_ready (void) {{ return ({pycheck_types}); }} """.format(pycheck_types = check_types) add_types = "".join([""" Py_INCREF (&Py_{name}Type); PyModule_AddObject(m, "{name}", (PyObject *) & Py_{name}Type);""".format(name = o) for o in lib]) out += """ void add_generated_objects ( PyObject *m ) {{ {add_types} }} """.format(add_types = add_types) output_file = os.path.join(output_path, 'aubio-generated.c') with open(output_file, 'w') as f: f.write(out) print ("wrote %s" % output_file ) sources_list.append(output_file) objlist = "".join(["extern PyTypeObject Py_%sType;\n" % p for p in lib]) out = """// generated list of objects created with gen_external.py #include """ if usedouble: out += """ #ifndef HAVE_AUBIO_DOUBLE #define HAVE_AUBIO_DOUBLE 1 #endif """ out += """ {objlist} int generated_objects ( void ); void add_generated_objects( PyObject *m ); """.format(objlist = objlist) output_file = os.path.join(output_path, 'aubio-generated.h') with open(output_file, 'w') as f: f.write(out) print ("wrote %s" % output_file ) # no need to add header to list of sources return sources_list if __name__ == '__main__': if len(sys.argv) > 1: header = sys.argv[1] if len(sys.argv) > 2: output_path = sys.argv[2] generate_external(header, output_path)