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