FEAT: numpy.core and numpy.lib both build

1 files changed, 59 insertions, 1 deletions

diff --git a/bscript b/bscript
index 8d7e55664..3796b839b 100644
--- a/bscript
+++ b/bscript
@@ -1,4 +1,5 @@
 import sys
+import shutil
 
 # Ugly but necessary hack: import numpy here so that wscript in sub directories
 # will see this numpy and not an already installed one
@@ -7,11 +8,13 @@ __builtin__.__NUMPY_SETUP__ = True
 
 from bento.commands.hooks \
     import \
-        pre_configure
+        pre_configure, post_build
 from bento.commands.extras.waf \
     import \
         ConfigureWafContext, BuildWafContext
 
+import waflib
+
 def check_blas_lapack(conf):
     conf.env.HAS_CBLAS = False
     if sys.platform == "win32":
@@ -56,8 +59,63 @@ def configure(context):
         conf.env.CC = ["/usr/bin/gcc-4.0"]
         conf.env.LINK_CC = ["/usr/bin/gcc-4.0"]
 
+    conf.env.CFLAGS_PYEXT.append("-Wfatal-errors")
     check_blas_lapack(conf)
 
+# FIXME: abstract those module gen tasks...
+class write_module(waflib.Task.Task):
+    color = "CYAN"
+    vars = ["CONTENT"]
+    def run(self):
+        # FIXME: put actual data here
+        self.outputs[0].write(self.env.CONTENT)
+
+@waflib.TaskGen.feature("gen_pymodule")
+def process_write_config(self):
+    if not hasattr(self, "content"):
+        raise ValueError("task gen %r expects a 'content' argument" % self.name)
+    else:
+        self.env.CONTENT = self.content
+    tsk = self.create_task("write_module")
+    tsk.set_outputs(self.path.find_or_declare(self.target))
+    return tsk
+
+@post_build()
+def pbuild(context):
+    bld = context.waf_context
+
+    bld(features="gen_pymodule",
+        target="numpy/__config__.py",
+        content="""\
+def show():
+    pass
+""",
+        always=True)
+    bld(features="gen_pymodule",
+        target="numpy/version.py",
+        content="""\
+git_revision = ""
+version = ""
+""",
+        always=True)
+    bld.compile()
+
+    # Poor man's and temporary replacement for in-place build
+    if "-i" in sys.argv:
+        for g in bld.groups:
+            for task_gen in g:
+                if hasattr(task_gen, "link_task"):
+                    ltask = task_gen.link_task
+                    for output in ltask.outputs:
+                        if output.is_child_of(bld.bldnode):
+                            shutil.copy(output.abspath(), output.path_from(bld.bldnode))
+                elif "gen_pymodule" in task_gen.features:
+                    if len(task_gen.tasks) > 0:
+                        task = task_gen.tasks[0]
+                        for output in task.outputs:
+                            if output.is_child_of(bld.bldnode):
+                                shutil.copy(output.abspath(), output.path_from(bld.bldnode))
+
 def startup(context):
     context.register_context("configure", ConfigureWafContext)
     context.register_context("build", BuildWafContext)