abacus-develop/tools/windows/install-abacus.bat at 19925b1da9b2e64b17180df9a525fd841959af12 · deepmodeling/abacus-develop · 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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
@echo off
setlocal EnableExtensions EnableDelayedExpansion
title ABACUS for Windows (WSL2) Installer

set "DISTRO=Ubuntu-22.04"
set "INSTALL_DIR=%LOCALAPPDATA%\ABACUS"
set "BIN_DIR=%INSTALL_DIR%\bin"

echo =========================================
echo   ABACUS Windows Installer (WSL2)
echo =========================================
echo.

REM --- 1. Admin check ---
net session >nul 2>&1
if errorlevel 1 (
    echo [!] Please right-click this script and choose "Run as administrator".
    pause
    exit /b 1
)

REM --- 2. Windows build check (WSL2 requires 19041+) ---
for /f "tokens=4-6 delims=. " %%a in ('ver') do set "BUILD=%%c"
if %BUILD% LSS 19041 (
    echo [!] Windows build %BUILD% detected.
    echo [!] WSL2 requires Windows 10 build 19041 or newer, or Windows 11.
    pause
    exit /b 1
)

REM --- 2b. Offer China mirror (TUNA) for faster downloads ---
if not defined ABACUS_CHINA_MIRROR (
    set "ABACUS_CHINA_MIRROR=0"
    set /p CHINA_INPUT="Use TUNA mirror for faster downloads (recommended inside Mainland China)? (y/N): "
    if /i "!CHINA_INPUT!"=="y" set "ABACUS_CHINA_MIRROR=1"
)

REM --- 3. Ensure WSL itself is present ---
where wsl >nul 2>&1
if errorlevel 1 (
    echo [*] WSL not found. Installing WSL2 runtime...
    wsl --install --no-launch
    if errorlevel 1 (
        echo [!] WSL installation failed.
        echo [!] Make sure virtualization is enabled in BIOS/UEFI.
        pause & exit /b 1
    )
    echo.
    echo [!] WSL was just installed. Please REBOOT Windows,
    echo [!] then run this script again to continue.
    pause
    exit /b 0
)

REM --- 4. Ensure target distro registered ---
echo [*] Probing WSL state. First call may take 30-60s while WSL2 cold-starts...
wsl --set-default-version 2 >nul 2>&1

REM Probe: if `wsl -d DISTRO -- true` returns 0, the distro already exists.
REM This avoids parsing `wsl -l -q` (UTF-16 output trips findstr).
set "DISTRO_PREEXISTED=1"
wsl -d %DISTRO% -u root -- /bin/true >nul 2>&1
if errorlevel 1 (
    set "DISTRO_PREEXISTED=0"
    echo [*] Installing %DISTRO% into WSL ^(this may take a few minutes^)...
    wsl --install -d %DISTRO% --no-launch
    if errorlevel 1 (
        echo [!] Failed to install %DISTRO%. Reboot and retry, or run
        echo [!]     wsl --install -d %DISTRO%
        echo [!] manually and finish any setup prompt, then re-run this script.
        pause & exit /b 1
    )
    REM Poke the distro once as root so any first-run steps complete without OOBE.
    wsl -d %DISTRO% -u root -- /bin/true >nul 2>&1
)

REM --- 5. Run provisioning script inside the distro ---
echo [*] Provisioning ABACUS via conda-forge inside %DISTRO%.
echo     This downloads ~400 MB and takes 5-15 minutes on first run.
echo.

echo [*] Translating script path into WSL...
set "WSL_SCRIPT="
for /f "usebackq delims=" %%i in (`wsl -d %DISTRO% wslpath "%~dp0provision.sh" 2^>nul`) do set "WSL_SCRIPT=%%i"
if not defined WSL_SCRIPT (
    echo [!] Could not translate script path into WSL. Aborting.
    pause & exit /b 1
)

wsl -d %DISTRO% -u root -- env ABACUS_CHINA_MIRROR=!ABACUS_CHINA_MIRROR! bash "!WSL_SCRIPT!"
if errorlevel 1 (
    echo [!] Provisioning failed. See output above.
    pause & exit /b 1
)

REM --- 6. Install Windows-side launchers ---
if not exist "%BIN_DIR%" mkdir "%BIN_DIR%"

> "%BIN_DIR%\abacus.cmd" echo @echo off
>> "%BIN_DIR%\abacus.cmd" echo wsl -d %DISTRO% --cd "%%CD%%" -- abacus %%*

> "%BIN_DIR%\abacus-mpi.cmd" echo @echo off
>> "%BIN_DIR%\abacus-mpi.cmd" echo wsl -d %DISTRO% --cd "%%CD%%" -- abacus-mpi %%*

REM Record install state so the uninstaller knows whether we added the distro.
> "%INSTALL_DIR%\install-state.txt" echo distro=%DISTRO%
>> "%INSTALL_DIR%\install-state.txt" echo distro_preexisted=!DISTRO_PREEXISTED!

REM --- 7. Add BIN_DIR to user PATH (idempotent, no 1024-char truncation) ---
powershell -NoProfile -ExecutionPolicy Bypass -Command ^
 "$b='%BIN_DIR%'; $p=[Environment]::GetEnvironmentVariable('PATH','User'); if([string]::IsNullOrEmpty($p)){[Environment]::SetEnvironmentVariable('PATH',$b,'User')} elseif(($p -split ';') -notcontains $b){[Environment]::SetEnvironmentVariable('PATH',$p.TrimEnd(';')+';'+$b,'User')}"

echo.
echo =========================================
echo   Installation complete!
echo =========================================
echo.
echo Open a NEW terminal window, cd into a case directory, then run:
echo     abacus
echo.
echo For parallel execution with N processes:
echo     abacus-mpi -n 4
echo.
echo To enter the Linux shell for advanced use:
echo     wsl -d %DISTRO%
echo.
pause
exit /b 0