mcdc-vvp/launch.py at dev · mcdc-project/mcdc-vvp · GitHub

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
"""Launch the configured MC/DC VVP suites."""

import argparse
import subprocess
import sys
from pathlib import Path

# ======================================================================================
# Bootstrap VVP imports
# ======================================================================================

REPO_DIR = Path(__file__).resolve().parent

# Support launching this script from any working directory without installing VVP.
if str(REPO_DIR) not in sys.path:
    sys.path.insert(0, str(REPO_DIR))


# ======================================================================================
# Load launch configuration
# ======================================================================================

from configs.launch_config import LAUNCH_CONFIG

# ======================================================================================
# Command-line arguments
# ======================================================================================

parser = argparse.ArgumentParser(description="Launch enabled MC/DC VVP suites.")
parser.add_argument(
    "--platform",
    default=None,
    help="Select suites configured for PLATFORM; omit for local run.",
)
args = parser.parse_args()

active_platform = args.platform
platform_name = active_platform or "local"

# Select only suites assigned to this launch invocation.
selected_suites = {
    suite: options
    for suite, options in LAUNCH_CONFIG.items()
    if options.get("enabled", False) and options.get("platform") == active_platform
}
if not selected_suites:
    parser.error(f"No enabled suites are configured for platform '{platform_name}'.")


# ======================================================================================
# Launch enabled, compatible suites
# ======================================================================================

for suite, options in selected_suites.items():
    suite_platform = options.get("platform")
    suite_dir = REPO_DIR / suite
    launcher = suite_dir / "launch.py"

    if not suite_dir.is_dir():
        raise FileNotFoundError(f"Suite directory not found: {suite_dir}")

    if not launcher.is_file():
        raise FileNotFoundError(f"Suite launcher not found: {launcher}")

    # Reuse the active interpreter so suite launchers inherit this environment.
    command = [sys.executable, str(launcher)]

    # Forward only options that are meaningful for the configured suite.
    if suite_platform is not None:
        command.extend(["--platform", suite_platform])

    if options.get("N_node") is not None:
        command.extend(["--N_node", str(options["N_node"])])

    if options.get("walltime") is not None:
        command.extend(["--walltime", str(options["walltime"])])

    print("=" * 80)
    print(f"Launching suite: {suite}")
    print("Command:", " ".join(command))
    print("=" * 80)

    subprocess.run(command, cwd=suite_dir, check=True)


# ======================================================================================
# Summary
# ======================================================================================

print()
print("Launch complete.")