1import sys, os.path as p 2wdir = p.abspath(p.dirname(__file__)) 3topdir = p.normpath(p.join(wdir, p.pardir, p.pardir)) 4srcdir = p.join(topdir, 'src') 5sys.path.insert(0, p.join(topdir, 'conf')) 6 7from mpiscanner import Scanner 8scanner = Scanner() 9libmpi_pxd = p.join(srcdir, 'mpi4py', 'libmpi.pxd') 10scanner.parse_file(libmpi_pxd) 11libmpi_h = p.join(wdir, 'libmpi.h') 12scanner.dump_header_h(libmpi_h) 13 14#try: 15# from cStringIO import StringIO 16#except ImportError: 17# from io import StringIO 18#libmpi_h = StringIO() 19#scanner.dump_header_h(libmpi_h) 20#print libmpi_h.read() 21 22libmpi_c = p.join(wdir, 'libmpi.c.in') 23with open(libmpi_c, 'w') as f: 24 f.write("""\ 25#include <mpi.h> 26#include "%(srcdir)s/lib-mpi/config.h" 27#include "%(srcdir)s/lib-mpi/missing.h" 28#include "%(srcdir)s/lib-mpi/fallback.h" 29#include "%(srcdir)s/lib-mpi/compat.h" 30""" % vars()) 31