setup.py

from pathlib import Path
import subprocess
import sys
import os

from setuptools import setup, find_packages, Extension

import numpy
from Cython.Build import cythonize

cwd = Path(__file__).parent

str_true = "True"
MINIMAL_INSTALL = os.environ.get('INFINIGEN_MINIMAL_INSTALL') == str_true
BUILD_TERRAIN = os.environ.get('INFINIGEN_INSTALL_TERRAIN', str_true) == str_true
BUILD_OPENGL = os.environ.get('INFINIGEN_INSTALL_CUSTOMGT', "False") == str_true

dont_build_steps = ["clean", "egg_info", "dist_info", "sdist", "--help"]
is_build_step = not any(x in sys.argv[1] for x in dont_build_steps) 

def ensure_submodules():
    # Inspired by https://github.com/pytorch/pytorch/blob/main/setup.py

    with (cwd/'.gitmodules').open() as f:
        submodule_folders = [
            cwd/line.split("=", 1)[1].strip()
            for line in f.readlines()
            if line.strip().startswith("path")
        ]

    if any(not p.exists() or not any(p.iterdir()) for p in submodule_folders):
        subprocess.run(
            ["git", "submodule", "update", "--init", "--recursive"], 
            cwd=cwd,
            check=True
        )    

ensure_submodules()

# inspired by https://github.com/pytorch/pytorch/blob/161ea463e690dcb91a30faacbf7d100b98524b6b/setup.py#L290
# theirs seems to not exclude dist_info but this causes duplicate compiling in my tests
if is_build_step and not MINIMAL_INSTALL:
    if BUILD_TERRAIN:
        subprocess.run(['make', 'terrain'], cwd=cwd, check=True)
    if BUILD_OPENGL:
        subprocess.run(['make', 'customgt'], cwd=cwd, check=True)

cython_extensions = []

if not MINIMAL_INSTALL:
    cython_extensions.append(Extension(
        name="bnurbs",
        sources=["infinigen/assets/creatures/util/geometry/cpp_utils/bnurbs.pyx"],
        include_dirs=[numpy.get_include()]
    ))
    if BUILD_TERRAIN:
        cython_extensions.append(
            Extension(
                name="infinigen.terrain.marching_cubes",
                sources=["infinigen/terrain/marching_cubes/_marching_cubes_lewiner_cy.pyx"],
                include_dirs=[numpy.get_include()]
            )
        )

setup(
    ext_modules=[
        *cythonize(cython_extensions)
    ]
    # other opts come from pyproject.toml
)