docling-parse/local_build.py

#!/usr/bin/env python

import os

import sys

import subprocess
from typing import List
import pybind11

ROOT_DIR = os.path.abspath("./")
BUILD_DIR = os.path.join(ROOT_DIR, "build")

def get_pybind11_cmake_args():
        pybind11_sys_path = os.getenv("PYBIND11_SYSPATH")
        if pybind11_sys_path:
            # pybind11_include_dir = os.path.join(pybind11_sys_path, "include")
            pybind11_cmake_dir = os.path.join(pybind11_sys_path, "share", "cmake", "pybind11")
        else:
            # pybind11_include_dir = pybind11.get_include()
            pybind11_cmake_dir = pybind11.get_cmake_dir()
        # print(f"{pybind11_include_dir=}")
        print(f"{pybind11_cmake_dir=}")
        return [f"-Dpybind11_DIR={pybind11_cmake_dir}"]

def run(cmd: List[str], cwd: str="./"):

    print_cmd = " ".join(cmd)
    print(f"\nlaunch: {print_cmd}")

    message = subprocess.run(cmd, cwd=cwd)

    if "returncode=0" in str(message):
        print(f" -> SUCCESS")
        return True

    print(f" -> ERROR with message: '{message}'\n")
    return False


def build_local(num_threads: int):

    USE_SYSTEM_DEPS = os.getenv("USE_SYSTEM_DEPS", "OFF")


    print("python prefix: ", sys.exec_prefix)
    print("python executable: ", sys.executable)
    config_cmd = [
        "cmake",
        "-B", f"{BUILD_DIR}",
        f"-DUSE_SYSTEM_DEPS={USE_SYSTEM_DEPS}",
        f"-DPYTHON_EXECUTABLE={sys.executable}",
    ]
    config_cmd.extend(get_pybind11_cmake_args())
    success = run(config_cmd, cwd=ROOT_DIR)
    if not success:
        raise RuntimeError("Error building.")

    build_cmd = [
        "cmake",
        "--build", f"{BUILD_DIR}",
        "--target=install",
    ]
    if num_threads > 1:
        build_cmd.extend(["-j", f"{num_threads}"])
    success = run(build_cmd, cwd=ROOT_DIR)
    if not success:
        raise RuntimeError("Error building.")


if "__main__" == __name__:
    num_threads = int(os.getenv("BUILD_THREADS", "4"))
    print(f"Building with {num_threads} threads (BUILD_THREADS={os.getenv('BUILD_THREADS', 'not set, defaulting to 4')})")
    build_local(num_threads=num_threads)