Windows configure script

configure.py

@@ -0,0 +1,140 @@
+import sys, platform
+if sys.platform != 'win32':
+    print("This script is for Windows only! Use configure.sh instead.")
+    exit(1)
+if sys.version_info.major != 3 or sys.version_info.minor != 8:
+    print("You neeed to use Python 3.8.x (due to the requirements.txt). You are using %s instead." % platform.python_version())
+    exit(1)
+
+import argparse
+import subprocess
+import os
+import stat
+import urllib.request
+import shutil 
+import zipfile
+
+from venv import EnvBuilder
+
+parser = argparse.ArgumentParser(description='ODM Windows Configure Script')
+parser.add_argument('action',
+                type=str,
+                choices=["build", "clean", "dist", "vcpkg_export"],
+                help='Action: %(choices)s')
+parser.add_argument('--build-vcpkg',
+                    type=bool,
+                    help='Build VCPKG environment from scratch instead of downloading prebuilt one.')
+parser.add_argument('--vcpkg-archive-url',
+                    type=str,
+                    default='https://github.com/OpenDroneMap/windows-deps/releases/download/2.5.0/vcpkg-export-250.zip',
+                    required=False,
+                    help='Path to VCPKG export archive')
+args = parser.parse_args()
+
+def run(cmd, cwd=os.getcwd()):
+    env = os.environ.copy()
+    print(cmd)
+    p = subprocess.Popen(cmd, shell=True, env=env, cwd=cwd)
+    retcode = p.wait()
+    if retcode != 0:
+        raise Exception("Command returned %s" % retcode)
+
+# https://izziswift.com/shutil-rmtree-fails-on-windows-with-access-is-denied/
+def rmtree(top):
+    for root, dirs, files in os.walk(top, topdown=False):
+        for name in files:
+            filename = os.path.join(root, name)
+            os.chmod(filename, stat.S_IWUSR)
+            os.remove(filename)
+        for name in dirs:
+            os.rmdir(os.path.join(root, name))
+    os.rmdir(top)      
+
+def vcpkg_requirements():
+    with open("vcpkg-requirements.txt") as f:
+        pckgs = list(filter(lambda l: len(l) > 0, map(str.strip, f.read().split("\n"))))
+    return pckgs
+        
+
+def build():
+    # Create python virtual env
+    if not os.path.isdir("venv"):
+        print("Creating virtual env --> venv/")
+        ebuilder = EnvBuilder(with_pip=True)
+        ebuilder.create("venv")
+
+    run("venv\\Scripts\\pip install --ignore-installed -r requirements.txt")
+    
+    # Download / build VCPKG environment
+    if not os.path.isdir("vcpkg"):
+        if args.build_vcpkg:
+            print("TODO")
+            # git clone vcpkg repo
+            # bootstrap
+            # install requirements
+
+        else:
+            if not os.path.exists("vcpkg-env.zip"):
+                print("Downloading %s" % args.vcpkg_archive_url)
+                with urllib.request.urlopen(args.vcpkg_archive_url) as response, open( "vcpkg-env.zip", 'wb') as out_file:
+                    shutil.copyfileobj(response, out_file)
+            if not os.path.exists("vcpkg"):
+                print("Extracting vcpkg-env.zip --> vcpkg/")
+                with zipfile.ZipFile("vcpkg-env.zip") as z:
+                    top_dir = z.namelist()[0]
+                    z.extractall(".")
+
+                    if os.path.exists(top_dir):
+                        os.rename(top_dir, "vcpkg")
+                    else:
+                        print("Warning! Something looks wrong in the VCPKG archive... check the vcpkg/ directory.")
+
+    if not os.path.exists(os.path.join("SuperBuild", "build")) or not os.path.exists(os.path.join("SuperBuild", "install")):
+        print("Compiling SuperBuild")
+        
+        build_dir = os.path.join("SuperBuild", "build")
+        if not os.path.isdir(build_dir):
+            os.mkdir(build_dir)
+
+        toolchain_file = os.path.join(os.getcwd(), "vcpkg", "scripts", "buildsystems", "vcpkg.cmake")
+        run("cmake .. -DCMAKE_TOOLCHAIN_FILE=\"%s\"" % toolchain_file,  cwd=build_dir)
+        run("cmake --build . --config Release", cwd=build_dir)
+
+def vcpkg_export():
+    if not os.path.exists("vcpkg"):
+        print("vcpkg directory does not exist. Did you build the environment?")
+        exit(1)
+
+    pkgs = vcpkg_requirements()
+    out = "vcpkg-export-%s" % odm_version().replace(".", "")
+    run("vcpkg\\vcpkg export %s --output=%s --zip" % (" ".join(pkgs), out))
+
+def odm_version():
+    with open("VERSION") as f:
+        return f.read().split("\n")[0].strip()
+
+def safe_remove(path):
+    if os.path.isdir(path):
+        rmtree(path)
+    elif os.path.isfile(path):
+        os.remove(path)
+
+def clean():
+    safe_remove("vcpkg-download.zip")
+    safe_remove("vcpkg")
+    safe_remove("venv")
+    safe_remove(os.path.join("SuperBuild", "build"))
+    safe_remove(os.path.join("SuperBuild", "download"))
+    safe_remove(os.path.join("SuperBuild", "src"))
+    safe_remove(os.path.join("SuperBuild", "install"))
+
+
+if args.action == 'build':
+    build()
+elif args.action == 'vcpkg_export':
+    vcpkg_export()
+elif args.action == 'clean':
+    clean()
+else:
+    args.print_help()
+    exit(1)