initial commit

scripts/fortran_toolchain.py

@@ -0,0 +1,81 @@
+#!/usr/bin/env python3
+"""Intel oneAPI Fortran discovery helpers for no-Abaqus validation."""
+
+from __future__ import annotations
+
+import os
+import shutil
+import subprocess
+from pathlib import Path
+from typing import Iterable
+
+
+COMPILER_CANDIDATES = ("ifx", "ifort")
+ONEAPI_VARS_CANDIDATES = [
+    Path(r"C:\Program Files (x86)\Intel\oneAPI\compiler\latest\env\vars.bat"),
+    Path(r"C:\Program Files\Intel\oneAPI\compiler\latest\env\vars.bat"),
+    Path(r"C:\Program Files (x86)\Intel\oneAPI\setvars.bat"),
+    Path(r"C:\Program Files\Intel\oneAPI\setvars.bat"),
+]
+
+
+class FortranToolchain:
+    def __init__(self, *, name: str, executable: str, env_script: Path | None):
+        self.name = name
+        self.executable = executable
+        self.env_script = env_script
+
+
+def compiler_candidates(preference: str | None = None) -> list[str]:
+    preference = (preference or os.environ.get("HARNESS_FORTRAN_COMPILER") or "auto").lower()
+    if preference == "auto":
+        return list(COMPILER_CANDIDATES)
+    if preference not in COMPILER_CANDIDATES:
+        raise ValueError(f"Unsupported HARNESS_FORTRAN_COMPILER: {preference}")
+    return [preference]
+
+
+def discover_oneapi_env_script(env: dict[str, str] | None = None) -> Path | None:
+    env = env or os.environ
+    configured = env.get("HARNESS_ONEAPI_VARS_BAT")
+    if configured:
+        path = Path(configured)
+        return path if path.exists() else None
+
+    for path in ONEAPI_VARS_CANDIDATES:
+        if path.exists():
+            return path
+    return None
+
+
+def resolve_toolchain(env: dict[str, str] | None = None) -> FortranToolchain | None:
+    env = env or os.environ
+    candidates = compiler_candidates(env.get("HARNESS_FORTRAN_COMPILER"))
+    for compiler in candidates:
+        resolved = shutil.which(compiler)
+        if resolved:
+            return FortranToolchain(name=compiler, executable=resolved, env_script=None)
+
+    env_script = discover_oneapi_env_script(env)
+    if env_script is None:
+        return None
+
+    compiler = candidates[0]
+    return FortranToolchain(name=compiler, executable=compiler, env_script=env_script)
+
+
+def shell_join(args: Iterable[str | Path]) -> str:
+    return subprocess.list2cmdline([str(arg) for arg in args])
+
+
+def quote_path(path: str | Path) -> str:
+    return shell_join([path])
+
+
+def wrap_command(toolchain: FortranToolchain, args: list[str | Path]) -> str:
+    command = shell_join(args)
+    if toolchain.env_script is None:
+        return command
+
+    env_script = quote_path(toolchain.env_script)
+    return f'cmd /d /s /c "call {env_script} intel64 >nul && {command}"'