monolithium 1.2.0

Finding the Largest Minecraft Alpha Monoliths
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

import os
import shutil
import subprocess
import sys
from pathlib import Path
from subprocess import PIPE, CompletedProcess, Popen
from typing import Union

import tomli


class Paths:
    PACKAGE: Path = Path(__file__).parent
    REPO:    Path = (PACKAGE.parent)
    BUILD:   Path = (Path.cwd()/"build")

class Tools:
    PYTHON: tuple[str] = (sys.executable,)
    MESON:  tuple[str] = (*PYTHON, "-m", "mesonbuild.mesonmain")
    NINJA:  tuple[str] = (*PYTHON, "-m", "ninja")
    RUSTUP: tuple[str] = (shutil.which("rustup"),)
    CARGO:  tuple[str] = (shutil.which("cargo"),)

# ---------------------------------------------------------------------------- #

def rustlith(
    *args: list[str],
    Popen: bool=False,
    toolchain: str="stable",
    **kwargs
) -> Union[CompletedProcess, Popen]:
    """Run the Rust version of Monolithium"""
    args = list(map(str, (args or sys.argv[1:])))

    # Have a rust toolchain
    if subprocess.run(
        (*Tools.RUSTUP, "run", toolchain, "rustc", "--version"),
        stdout=PIPE, stderr=PIPE
    ).returncode != 0:
        subprocess.check_call((
            *Tools.RUSTUP,
            "default", toolchain
        ))

    # Simple features handling via anywhere in args flags
    cargo = tomli.loads((Paths.PACKAGE/"Cargo.toml").read_text(encoding="utf-8"))
    features = list()

    for feature in cargo["features"]:
        if (flag := f"--{feature}") in args:
            features.append("--features")
            features.append(feature)
            args.remove(flag)

    os.environ.update(RUSTFLAGS="-C target-cpu=native")

    return (subprocess.Popen if Popen else subprocess.run)((
        *Tools.CARGO, "run",
        "--manifest-path", (Paths.PACKAGE/"Cargo.toml"),
        "--target-dir", str(Path.cwd()/"target"),
        "--release", *features,
        "--", *args,
    ), **kwargs)

# ---------------------------------------------------------------------------- #

def cudalith(
    *args: list[str],
    Popen: bool=False,
    **kwargs
) -> Union[CompletedProcess, Popen]:
    """Run the CUDA version of Monolithium"""
    args = list(map(str, (args or sys.argv[1:])))

    if not shutil.which("nvcc"):
        raise RuntimeError("nvcc wasn't found in path, do you have cuda toolkit?")

    # Generate meson build files
    subprocess.check_call((
        *Tools.MESON,
        "setup", Paths.BUILD,
        "--buildtype", "release",
        "--reconfigure"
    ), cwd=Paths.PACKAGE)

    # Compile the cuda part
    subprocess.check_call((
        *Tools.NINJA, "-C",
        Paths.BUILD
    ))

    return (subprocess.Popen if Popen else subprocess.run)((
        Paths.BUILD/"cudalith",
        *args
    ), **kwargs)