Merge pull request #46 from thewtex/window-module-wheel

Add windows_build_module_wheels.py

Author: thewtex

scripts/internal/windows_build_common.py

@@ -0,0 +1,41 @@
+__all__ = ['DEFAULT_PY_ENVS', 'venv_paths']
+
+import os
+
+DEFAULT_PY_ENVS = ["27-x64", "35-x64", "36-x64"]
+
+SCRIPT_DIR = os.path.dirname(__file__)
+ROOT_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, "..", ".."))
+
+def venv_paths(python_version):
+    venv_dir = os.path.join(ROOT_DIR, "venv-%s" % python_version)
+
+    python_executable = os.path.join(venv_dir, "Scripts", "python.exe")
+    python_include_dir = os.path.join(venv_dir, "Include")
+
+    # XXX It should be possible to query skbuild for the library dir associated
+    #     with a given interpreter.
+    xy_ver = python_version.split("-")[0]
+
+    python_library = "C:/Python%s/libs/python%s.lib" % (python_version, xy_ver)
+
+    print("")
+    print("PYTHON_EXECUTABLE: %s" % python_executable)
+    print("PYTHON_INCLUDE_DIR: %s" % python_include_dir)
+    print("PYTHON_LIBRARY: %s" % python_library)
+
+    pip = os.path.join(venv_dir, "Scripts", "pip.exe")
+
+    ninja_executable = os.path.join(
+        ROOT_DIR, "venv-27-x64", "Scripts", "ninja.exe")
+    print("NINJA_EXECUTABLE:%s" % ninja_executable)
+
+    # Update PATH
+    path = os.path.join(venv_dir, "Scripts")
+
+    return python_executable, \
+        python_include_dir, \
+        python_library, \
+        pip, \
+        ninja_executable, \
+        path

scripts/windows_build_module_wheels.py

@@ -0,0 +1,58 @@
+from subprocess import check_call
+import os
+import sys
+
+SCRIPT_DIR = os.path.dirname(__file__)
+ROOT_DIR = os.path.abspath(os.getcwd())
+
+
+print("SCRIPT_DIR: %s" % SCRIPT_DIR)
+print("ROOT_DIR: %s" % ROOT_DIR)
+
+sys.path.insert(0, os.path.join(SCRIPT_DIR, "internal"))
+
+from wheel_builder_utils import push_dir, push_env
+from windows_build_common import DEFAULT_PY_ENVS, venv_paths
+
+def build_wheels(py_envs=DEFAULT_PY_ENVS):
+    for py_env in py_envs:
+        python_executable, \
+                python_include_dir, \
+                python_library, \
+                pip, \
+                ninja_executable, \
+                path = venv_paths(py_env)
+
+        with push_env(PATH="%s%s%s" % (path, os.pathsep, os.environ["PATH"])):
+
+            # Install dependencies
+            requirements_file = os.path.join(ROOT_DIR, "requirements-dev.txt")
+            if os.path.exists(requirements_file):
+                check_call([pip, "install", "--upgrade", "-r", requirements_file])
+            check_call([pip, "install", "cmake"])
+
+            build_type = "Release"
+            source_path = ROOT_DIR
+            itk_build_path = os.path.abspath("%s/ITK-win_%s" % (os.path.join(SCRIPT_DIR, '..'), py_env))
+            print('ITKDIR: %s' % itk_build_path)
+
+            # Generate wheel
+            check_call([
+                python_executable,
+                "setup.py", "bdist_wheel",
+                "--build-type", build_type, "-G", "Ninja",
+                "--",
+                "-DCMAKE_MAKE_PROGRAM:FILEPATH=%s" % ninja_executable,
+                "-DITK_DIR:PATH=%s" % itk_build_path,
+                "-DWRAP_ITK_INSTALL_COMPONENT_IDENTIFIER:STRING=PythonWheel",
+                "-DSWIG_EXECUTABLE:FILEPATH=%s/Wrapping/Generators/SwigInterface/swig/bin/swig.exe" % itk_build_path,
+                "-DBUILD_TESTING:BOOL=OFF",
+                "-DPYTHON_EXECUTABLE:FILEPATH=%s" % python_executable,
+                "-DPYTHON_INCLUDE_DIR:PATH=%s" % python_include_dir,
+                "-DPYTHON_LIBRARY:FILEPATH=%s" % python_library
+            ])
+            # Cleanup
+            check_call([python_executable, "setup.py", "clean"])
+
+if __name__ == '__main__':
+    build_wheels()

scripts/windows_build_wheels.py

@@ -20,6 +20,7 @@
 
 sys.path.insert(0, os.path.join(SCRIPT_DIR, "internal"))
 from wheel_builder_utils import push_dir, push_env
+from windows_build_common import DEFAULT_PY_ENVS, venv_paths
 
 
 def pip_install(python_dir, package, upgrade=True):
@@ -104,30 +105,13 @@ def build_wrapped_itk(
 def build_wheel(python_version, single_wheel=False,
                 cleanup=False, wheel_names=None):
 
-    venv_dir = os.path.join(ROOT_DIR, "venv-%s" % python_version)
-
-    python_executable = os.path.join(venv_dir, "Scripts", "python.exe")
-    python_include_dir = os.path.join(venv_dir, "Include")
-
-    # XXX It should be possible to query skbuild for the library dir associated
-    #     with a given interpreter.
-    xy_ver = python_version.split("-")[0]
-
-    python_library = "C:/Python%s/libs/python%s.lib" % (python_version, xy_ver)
+    python_executable, \
+            python_include_dir, \
+            python_library, \
+            pip, \
+            ninja_executable, \
+            path = venv_paths(python_version)
 
-    print("")
-    print("PYTHON_EXECUTABLE: %s" % python_executable)
-    print("PYTHON_INCLUDE_DIR: %s" % python_include_dir)
-    print("PYTHON_LIBRARY: %s" % python_library)
-
-    pip = os.path.join(venv_dir, "Scripts", "pip.exe")
-
-    ninja_executable = os.path.join(
-        ROOT_DIR, "venv-27-x64", "Scripts", "ninja.exe")
-    print("NINJA_EXECUTABLE:%s" % ninja_executable)
-
-    # Update PATH
-    path = os.path.join(venv_dir, "Scripts")
     with push_env(PATH="%s%s%s" % (path, os.pathsep, os.environ["PATH"])):
 
         # Install dependencies
@@ -234,12 +218,9 @@ def test_wheels(single_wheel=False):
     pass
 
 
-def build_wheels(py_envs=None, single_wheel=False,
+def build_wheels(py_envs=DEFAULT_PY_ENVS, single_wheel=False,
                  cleanup=False, wheel_names=None):
 
-    if py_envs is None:
-        py_envs = ["27-x64", "35-x64", "36-x64"]
-
     prepare_build_env("27-x64")
     prepare_build_env("35-x64")
     prepare_build_env("36-x64")
@@ -269,7 +250,7 @@ def build_wheels(py_envs=None, single_wheel=False,
             cleanup=cleanup, wheel_names=wheel_names)
 
 
-def main(py_envs=None, wheel_names=None, cleanup=True):
+def main(py_envs=DEFAULT_PY_ENVS, wheel_names=None, cleanup=True):
     single_wheel = False
 
     build_wheels(