build.ps1

#Requires -Version 5.1
[CmdletBinding()]
param (
    [switch]$ForceRebuild,
    [switch]$EnableTQ,
    [switch]$EnableAvm,
    [switch]$EnableVvenc,
    [switch]$EnableX265,
    [switch]$EnableX264,
    [string]$Backend = "cuda",
    [int]$SvtVariantId = 1,
    [int]$ArchChoiceId = 1,
    [switch]$NoPrompt
)

Set-StrictMode -Version Latest
$ErrorActionPreference = 'Stop'

$COMMON_FLAGS = "-flto -O3 -DNDEBUG -march=native"

$CudaReleases = @{
    '12.9' = @{
        Version    = '12.9.1'
        Url        = 'https://developer.download.nvidia.com/compute/cuda/12.9.1/network_installers/cuda_12.9.1_windows_network.exe'
        Sha256     = 'DE45C336AB4E9825018B546357DC9DE5DED9F89C4F5200EF52B8EEAC3B1B4072'
        Components = @('nvcc_12.9', 'cudart_12.9')
    }
    '13.3' = @{
        Version    = '13.3.1'
        Url        = 'https://developer.download.nvidia.com/compute/cuda/13.3.1/network_installers/cuda_13.3.1_windows_network.exe'
        Sha256     = '0A4964DB02D01A80F897605B0CB2C18FDDAF277C44940A0F3436301A577B3C57'
        Components = @('nvcc_13.3', 'cudart_13.3', 'crt_13.3', 'nvvm_13.3')
    }
}

function Install-CudaMinimal {
    param([string]$Version, [string]$Url, [string]$Sha256, [string[]]$Components)
    $mm = ($Version -split '\.')[0..1] -join '.'
    $installer = "$env:TEMP\cuda_${Version}_network.exe"
    $logPath = "$env:TEMP\cuda_${Version}_install.log"

    Write-Host "[INFO] Downloading CUDA Toolkit $Version network installer (minimal stub, ~16 MB)..." -ForegroundColor Cyan
    Invoke-WebRequest -Uri $Url -OutFile $installer -UseBasicParsing

    Write-Host "[INFO] Verifying installer checksum..." -ForegroundColor Cyan
    $actual = (Get-FileHash $installer -Algorithm SHA256).Hash
    if ($actual -ne $Sha256) {
        Write-Host "[ERROR] CUDA installer checksum mismatch. Expected $Sha256, got $actual." -ForegroundColor Red
        exit 1
    }

    Write-Host "[INFO] Installing CUDA $Version ($($Components -join ', '), no driver)..." -ForegroundColor Cyan
    $proc = Start-Process -FilePath $installer -ArgumentList "-s $($Components -join ' ') -n -log:`"$logPath`"" -Wait -PassThru
    if ($proc.ExitCode -ne 0) {
        Write-Host "[ERROR] CUDA $Version installer failed (exit code $($proc.ExitCode)). Log: $logPath" -ForegroundColor Red
        exit 1
    }

    Update-SessionEnvironment
    if (-not $env:CUDA_PATH) {
        $default = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v$mm"
        if (Test-Path "$default\bin\nvcc.exe") { $env:CUDA_PATH = $default }
    }
    Write-Host "[INFO] CUDA Toolkit $Version (minimal) installed successfully." -ForegroundColor Green
    Remove-Item -Path $installer -Force -ErrorAction SilentlyContinue
}

function Invoke-Step {
    param([string]$Label, [scriptblock]$Action)
    Write-Host "[INFO] $Label..." -ForegroundColor Cyan
    & $Action
    if ($LASTEXITCODE -and $LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] $Label failed (exit code $LASTEXITCODE)." -ForegroundColor Red
        exit 1
    }
}

# Refresh PATH and SDK env vars from registry so newly installed tools are usable immediately.
function Update-SessionEnvironment {
    $machinePath = [System.Environment]::GetEnvironmentVariable('PATH', 'Machine')
    $userPath = [System.Environment]::GetEnvironmentVariable('PATH', 'User')
    $newPath = "$machinePath;$userPath"

    $commonPaths = @(
        "$env:ProgramFiles\LLVM\bin",
		"$env:ProgramFiles\NASM",
        "$env:LOCALAPPDATA\bin\NASM",
        "$env:ProgramFiles\Meson",
        "$env:USERPROFILE\.cargo\bin"
    )
    foreach ($p in $commonPaths) {
        if ((Test-Path $p) -and ($newPath -notlike "*$p*")) {
            $newPath = "$p;$newPath"
        }
    }
    $env:PATH = $newPath

    foreach ($var in @('CUDA_PATH', 'HIP_PATH')) {
        $val = [System.Environment]::GetEnvironmentVariable($var, 'Machine')
        if (-not $val) { $val = [System.Environment]::GetEnvironmentVariable($var, 'User') }
        if ($val) { Set-Item "env:$var" $val }
    }
}

function Assert-Command {
    param([string]$Cmd)
    return [bool](Get-Command $Cmd -ErrorAction SilentlyContinue)
}

function Test-CpuFlag {
    param([string]$MsysExe, [string]$Flag)
    try {
        & $MsysExe -lc "grep '^flags' /proc/cpuinfo | grep -q -w '$Flag'" 2>$null
        return ($LASTEXITCODE -eq 0)
    }
    catch {
        return $false
    }
}

function Confirm-Install {
    param([string]$AppName, [string]$WingetId)
    Write-Host ""
    Write-Host "[PROMPT] $AppName is missing." -ForegroundColor Yellow
    $choice = Read-Host "Do you want to install it using winget? (Y/N) [Default: Y]"
    if ($choice -ieq 'N') {
        Write-Host "[ERROR] $AppName is required. Exiting." -ForegroundColor Red
        exit 1
    }
    Write-Host "[INFO] Installing $AppName..." -ForegroundColor Cyan
    winget install -e --id $WingetId --accept-source-agreements --accept-package-agreements | Out-Host
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to install $AppName." -ForegroundColor Red
        exit 1
    }
    Update-SessionEnvironment
    Write-Host "[INFO] $AppName installed successfully." -ForegroundColor Green
}

function Find-Msys2Root {
    # Well-known default path
    $candidates = @('C:\msys64')

    # Scoop (per-user and global)
    $candidates += "$env:USERPROFILE\scoop\apps\msys2\current"
    $candidates += "$env:ProgramData\scoop\apps\msys2\current"

    $found = $candidates | Where-Object { Test-Path "$_\usr\bin\bash.exe" } | Select-Object -First 1
    if ($found) { return $found }

    # Registry uninstall entries (covers custom install paths)
    $regRoots = @(
        'HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall',
        'HKLM:\SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall',
        'HKCU:\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall'
    )
    foreach ($root in $regRoots) {
        if (-not (Test-Path $root)) { continue }
        $entry = Get-ChildItem $root -ErrorAction SilentlyContinue |
        Get-ItemProperty -ErrorAction SilentlyContinue |
        Where-Object { $_.PSObject.Properties['DisplayName'] -and $_.DisplayName -match 'MSYS2' } |
        Select-Object -First 1
        if ($entry -and $entry.InstallLocation -and (Test-Path "$($entry.InstallLocation)\usr\bin\bash.exe")) {
            return $entry.InstallLocation
        }
    }
    return $null
}

function Install-Msys2 {
    $msysRoot = Find-Msys2Root
    if (-not $msysRoot) {
        Confirm-Install "MSYS2" "MSYS2.MSYS2"
        $msysRoot = Find-Msys2Root
        if (-not $msysRoot) {
            Write-Host "[ERROR] MSYS2 not found after install. Please re-run the script or set the path manually." -ForegroundColor Red
            exit 1
        }
    }
    Write-Host "[INFO] Found MSYS2 at $msysRoot" -ForegroundColor Cyan
    $msysExe = "$msysRoot\usr\bin\bash.exe"
    Invoke-Step "Installing MSYS2 base dependencies" {
        & $msysExe -lc "pacman --noconfirm -S --needed autoconf automake libtool base-devel pkg-config" | Out-Host
    }
    return $msysExe
}

function Install-VsBuildTools {
    param([bool]$VsIncludeV143)
    $vswhere = "${env:ProgramFiles(x86)}\Microsoft Visual Studio\Installer\vswhere.exe"
    $hasCppTools = (Test-Path $vswhere) -and
    (& $vswhere -latest -products * -requires 'Microsoft.VisualStudio.Component.VC.Tools.x86.x64' -property installationPath 2>$null)
    
    if (-not $hasCppTools) {
        Write-Host ""
        Write-Host "[PROMPT] Visual Studio Build Tools with C++ workload is missing." -ForegroundColor Yellow
        $choice = Read-Host "Do you want to install it? (Y/N) [Default: Y]"
        if ($choice -ieq 'N') {
            Write-Host "[ERROR] Visual Studio Build Tools are required. Exiting." -ForegroundColor Red
            exit 1
        }
        Write-Host "[INFO] Installing Visual Studio Build Tools with Desktop C++ workload (this may take a while)..." -ForegroundColor Cyan

        $vsWorkloadComponents = "--add Microsoft.VisualStudio.Workload.VCTools --add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --add Microsoft.VisualStudio.Component.Windows11SDK.26100"
        if ($VsIncludeV143) {
            $vsWorkloadComponents += " --add Microsoft.VisualStudio.ComponentGroup.VC.Tools.143.x86.x64"
        }
        $vsWingetArgs = "--quiet --wait --norestart $vsWorkloadComponents"
        $vsModifyArgs = "--quiet --norestart $vsWorkloadComponents"

        winget install -e --id 'Microsoft.VisualStudio.BuildTools' --source winget `
            --accept-source-agreements --accept-package-agreements `
            --override $vsWingetArgs | Out-Host

        if ($LASTEXITCODE -ne 0) {
            $existingInstallPath = & $vswhere -latest -products * -property installationPath 2>$null
            if (-not $existingInstallPath) {
                Write-Host "[ERROR] winget failed to install Visual Studio Build Tools (exit code $LASTEXITCODE) and no existing installation was found." -ForegroundColor Red
                exit 1
            }
            Write-Host "[INFO] VS Build Tools already installed, but there is no C++ workload." -ForegroundColor Cyan
            Write-Host "[INFO] Installing C++ workload..." -ForegroundColor Cyan
            $vsInstallerPath = "${env:ProgramFiles(x86)}\Microsoft Visual Studio\Installer\vs_installer.exe"
            if (-not (Test-Path $vsInstallerPath)) {
                Write-Host "[ERROR] Could not find vs_installer.exe. Please open Visual Studio Installer manually and add the 'Desktop development with C++' workload." -ForegroundColor Red
                exit 1
            }
            Start-Process -FilePath $vsInstallerPath `
                -ArgumentList "modify --installPath `"$existingInstallPath`" $vsModifyArgs" `
                -Verb RunAs -Wait
            if ($LASTEXITCODE -ne 0) {
                Write-Host "[ERROR] Failed to modify Visual Studio Build Tools. Please open Visual Studio Installer manually and add the 'Desktop development with C++' workload." -ForegroundColor Red
                exit 1
            }
        }
        Update-SessionEnvironment
        Write-Host "[INFO] Visual Studio Build Tools with C++ workload installed successfully." -ForegroundColor Green
    }

    $vsPathResult = & $vswhere -latest -products * -requires Microsoft.VisualStudio.Component.VC.Tools.x86.x64 -property installationPath
    if (-not $vsPathResult) {
        Write-Host "[ERROR] Visual Studio with C++ tools not found." -ForegroundColor Red
        exit 1
    }
    return $vsPathResult
}

function Import-Vcvars {
    param([string]$VsPath, [bool]$VsIncludeV143)
    $devShell = Join-Path $VsPath 'Common7\Tools\Microsoft.VisualStudio.DevShell.dll'
    if (-not (Test-Path $devShell)) {
        Write-Host "[ERROR] VsDevShell not found at $devShell" -ForegroundColor Red
        exit 1
    }

    $devArgs = '-arch=x64'
    if ($VsIncludeV143) { $devArgs += ' -vcvars_ver=14.4' }

    Import-Module $devShell
    Enter-VsDevShell -VsInstallPath $VsPath -SkipAutomaticLocation -DevCmdArguments $devArgs
}

function Install-GpuSdk {
    param([string]$VshipBackend, [bool]$VsIncludeV143)
    if ($VshipBackend -eq 'cuda' -and -not $env:CUDA_PATH) {
        $mm = if ($VsIncludeV143) { '12.9' } else { '13.3' }
        $rel = $CudaReleases[$mm]
        Write-Host ""
        Write-Host "[PROMPT] NVIDIA CUDA Toolkit $mm is missing." -ForegroundColor Yellow
        $choice = Read-Host "Do you want to install it (minimal: $($rel.Components -join ', '))? (Y/N) [Default: Y]"
        if ($choice -ieq 'N') { Write-Host "[ERROR] CUDA Toolkit is required. Exiting." -ForegroundColor Red; exit 1 }
        Install-CudaMinimal -Version $rel.Version -Url $rel.Url -Sha256 $rel.Sha256 -Components $rel.Components
        if (-not $env:CUDA_PATH) {
            Write-Host "[ERROR] CUDA_PATH still not set after install. Try restarting your terminal, or set CUDA_PATH manually." -ForegroundColor Red
            exit 1
        }
    }

    if ($VshipBackend -eq 'hip' -and -not $env:HIP_PATH) {
        Write-Host ""
        Write-Host "[PROMPT] AMD HIP SDK is missing." -ForegroundColor Yellow
        $choice = Read-Host "Do you want to install it? (Y/N) [Default: Y]"
        if ($choice -ieq 'N') { Write-Host "[ERROR] AMD HIP SDK is required. Exiting." -ForegroundColor Red; exit 1 }
        
        Write-Host "[INFO] Downloading AMD HIP SDK..." -ForegroundColor Cyan
        $hipInstaller = "$env:TEMP\AMD-HIP-Setup.exe"
        Invoke-WebRequest -Uri "https://download.amd.com/developer/eula/rocm-hub/AMD-Software-PRO-Edition-26.Q1-Win11-For-HIP.exe" -OutFile $hipInstaller -UseBasicParsing
        Write-Host "[INFO] Installing AMD HIP SDK (this may take a while)..." -ForegroundColor Cyan
        Start-Process -FilePath $hipInstaller -ArgumentList '-install' -Wait
        Update-SessionEnvironment
        if (-not $env:HIP_PATH) {
            Write-Host "[ERROR] HIP_PATH still not set after install. Try restarting your terminal, or set HIP_PATH manually." -ForegroundColor Red
            exit 1
        }
        Remove-Item -Path $hipInstaller -Force -ErrorAction SilentlyContinue
    }
}

function Install-Dependencies {
    param([string]$VshipBackend, [bool]$VsIncludeV143)

    $basicTools = [ordered]@{
        'git'   = 'Git.Git'
        'cmake' = 'Kitware.CMake'
        'ninja' = 'Ninja-build.Ninja'
        'cargo' = 'Rustlang.Rustup'
    }

    foreach ($cmd in $basicTools.Keys) {
        if (-not (Assert-Command $cmd)) {
            Confirm-Install $cmd $basicTools[$cmd]
        }
    }

    $llvmBin = "$env:ProgramFiles\LLVM\bin"
    if (-not (Assert-Command 'clang++')) {
        if (-not (Test-Path "$llvmBin\clang++.exe")) { Confirm-Install "LLVM" "LLVM.LLVM" }
    }

    $nasmPaths = @("$env:ProgramFiles\NASM", "$env:LOCALAPPDATA\bin\NASM")
    if (-not (Assert-Command 'nasm')) {
        $nasmFound = $nasmPaths | Where-Object { Test-Path "$_\nasm.exe" } | Select-Object -First 1
        if (-not $nasmFound) { Confirm-Install "NASM" "NASM.NASM" }
    }

    Update-SessionEnvironment

    if (-not (Assert-Command 'clang')) { Write-Host "[ERROR] clang not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'clang++')) { Write-Host "[ERROR] clang++ not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'llvm-ar')) { Write-Host "[ERROR] llvm-ar not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'llvm-lib')) { Write-Host "[ERROR] llvm-lib not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'llvm-nm')) { Write-Host "[ERROR] llvm-nm not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'llvm-ranlib')) { Write-Host "[ERROR] llvm-ranlib not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'llvm-strip')) { Write-Host "[ERROR] llvm-strip not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'lld-link')) { Write-Host "[ERROR] lld-link not found after LLVM setup." -ForegroundColor Red; exit 1 }
    if (-not (Assert-Command 'nasm')) { Write-Host "[ERROR] nasm not found after NASM setup." -ForegroundColor Red; exit 1 }

    Write-Host "[INFO] Setting Rust toolchain to nightly..." -ForegroundColor Cyan
    rustup default nightly | Out-Host
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to set Rust toolchain to nightly." -ForegroundColor Red; exit 1
    }
    Write-Host "[INFO] Ensuring rust-src component (required for build-std)..." -ForegroundColor Cyan
    rustup component add rust-src --toolchain nightly | Out-Host
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to add rust-src component." -ForegroundColor Red; exit 1
    }

    Install-GpuSdk -VshipBackend $VshipBackend -VsIncludeV143 $VsIncludeV143
    $msysExe = Install-Msys2
    $vsPathResult = Install-VsBuildTools -VsIncludeV143 $VsIncludeV143

    return @{
        MsysExe = $msysExe
        VsPath  = $vsPathResult
    }
}

function Build-Vship {
    param([string]$Backend, [bool]$VsIncludeV143, [string]$VsPath)
    if (Test-Path 'lib\libvship.lib') {
        Write-Host "[INFO] Vship already compiled. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (Test-Path 'Vship') {
            Push-Location Vship
            git pull
            Pop-Location
        }
        else {
            git clone --depth 1 https://codeberg.org/Line-fr/Vship.git
        }
        Push-Location Vship

        switch ($Backend) {
            'cuda' {
                $msvcBin = Split-Path (Get-Command cl.exe -ErrorAction SilentlyContinue).Definition
                if (-not $msvcBin) {
                    Write-Host "[WARNING] cl.exe not found in PATH. nvcc may fail." -ForegroundColor Yellow
                    $ccbinArg = @()
                }
                else {
                    Write-Host "[INFO] Using MSVC host compiler for nvcc: $msvcBin" -ForegroundColor Cyan
                    $ccbinArg = @('-ccbin', $msvcBin)
                }
                Invoke-Step "Compiling Vship (CUDA)" {
                    & "$env:CUDA_PATH\bin\nvcc.exe" @ccbinArg -x cu src/VshipLib.cpp -std=c++17 -I include -arch=native -Xcompiler /MT --lib -o libvship.lib
                }
            }
            'hip' {
                Invoke-Step "Compiling Vship (HIP)" {
                    & "$env:HIP_PATH\bin\hipcc" -c src/VshipLib.cpp -std=c++17 -I include `
                        --offload-arch=native `
                        -Wno-unused-result -Wno-ignored-attributes -o libvship.o
                }
                if (Test-Path 'libvship.lib') { Remove-Item 'libvship.lib' }
                Invoke-Step "Archiving Vship (HIP)" { llvm-ar rcs libvship.lib libvship.o }
            }
            'vulkan' {
                Invoke-Step "Compiling shaderEmbedder" {
                    clang++ src/Vulkan/spvFileToCppHeader.cpp -std=c++17 -O2 -o shaderEmbedder.exe
                }
                Invoke-Step "Embedding shaders" {
                    .\shaderEmbedder.exe libvshipSpvShaders include/libvshipSpvShaders.hpp
                }
                Invoke-Step "Compiling Vship (Vulkan)" {
                    clang++ -c src/VshipLib.cpp -DVULKANBUILD -DNDEBUG -std=c++17 -O2 -Wall `
                        -Wno-ignored-attributes -Wno-unused-variable -Wno-nullability-completeness `
                        -Wno-unused-private-field -I include -I ..\install\include -o libvship.o
                }
                if (Test-Path 'libvship.lib') { Remove-Item 'libvship.lib' }
                Invoke-Step "Archiving Vship (Vulkan)" { llvm-ar rcs libvship.lib libvship.o }
            }
        }

        Copy-Item libvship.lib ..\lib\ -Force

        Pop-Location
    }
}

function Patch-SvtAv1 {
    param([string]$Variant, [string]$MsysExe)

    # should be fixed in mainline/tritium
    $apply8MbStackPatch = $Variant -notin @('svt-av1-tritium', 'svt-av1-skibidi', 'mainline')

    $applySharedRtcdPatch = $Variant -ne '5fish'

    $patchScript = @'
#!/bin/sh
set -e
sed -i 's/\r//g' CMakeLists.txt Source/Lib/Globals/enc_handle.c Source/Lib/Codec/aom_dsp_rtcd.c Source/Lib/Codec/common_dsp_rtcd.c Source/Lib/Codec/enc_inter_prediction.c
sed -i 's/set(CMAKE_POSITION_INDEPENDENT_CODE ON)/set(CMAKE_POSITION_INDEPENDENT_CODE OFF)/' CMakeLists.txt
sed -i 's/set(CMAKE_C_STANDARD 99)/set(CMAKE_C_STANDARD 23)/' CMakeLists.txt
sed -i 's/set(CMAKE_CXX_STANDARD 11)/set(CMAKE_CXX_STANDARD 23)/' CMakeLists.txt
sed -i '/relro/s/^/#/' CMakeLists.txt
sed -i '/mno-avx/s/^/#/' CMakeLists.txt
sed -i '/fstack-protector-strong/s/^/#/' CMakeLists.txt
sed -i '/FORTIFY_SOURCE/s/^/#/' CMakeLists.txt
sed -i '/gdwarf/s/^/#/' CMakeLists.txt
sed -i '/gnull/s/^/#/' CMakeLists.txt
sed -i 's|^        return_error = svt_av1_set_default_params(config_ptr);$|        return_error = config_ptr ? svt_av1_set_default_params(config_ptr) : EB_ErrorNone;|' Source/Lib/Globals/enc_handle.c
sed -i 's|^            if (svt_aom_copy_metadata_buffer(dst, src->metadata) != EB_ErrorNone)$|            if (1)|' Source/Lib/Globals/enc_handle.c
sed -i 's/void NOINLINE svt_aom_enc_make_inter_predictor(/void svt_aom_enc_make_inter_predictor(/' Source/Lib/Codec/enc_inter_prediction.c
'@

    if ($applySharedRtcdPatch) {
        $patchScript += @'

sed -i '/^    svt_aom_setup_common_rtcd_internal(scs->static_config.use_cpu_flags);$/,/^    svt_aom_build_blk_geom(scs->svt_aom_geom_idx, scs->blk_geom_mds);$/c\
    return_error = svt_shared_setup(scs);\
    if (return_error != EB_ErrorNone)\
        return return_error;' Source/Lib/Globals/enc_handle.c
grep -q init_shared_rtcd Source/Lib/Globals/enc_handle.c || sed -i '/^DEFINE_ONCE(global_tables_once);$/a\
\
static uint64_t          shared_cpu_flags;\
static uint32_t          shared_geom_idx;\
static uint16_t          shared_geom_cnt;\
static struct BlockGeom *shared_blk_geom;\
static uint32_t          shared_blk_geom_idx;\
\
static ONCE_ROUTINE(init_shared_rtcd) {\
    svt_aom_setup_common_rtcd_internal(shared_cpu_flags);\
    svt_aom_setup_rtcd_internal(shared_cpu_flags);\
    ONCE_ROUTINE_EPILOG;\
}\
DEFINE_ONCE(shared_rtcd_once);\
\
static ONCE_ROUTINE(init_shared_blk_geom) {\
    shared_blk_geom_idx = shared_geom_idx;\
    EB_MALLOC_ARRAY_NO_CHECK(shared_blk_geom, shared_geom_cnt);\
    if (shared_blk_geom)\
        svt_aom_build_blk_geom(shared_blk_geom_idx, shared_blk_geom);\
    ONCE_ROUTINE_EPILOG;\
}\
DEFINE_ONCE(shared_blk_geom_once);\
\
static EbErrorType svt_shared_setup(SequenceControlSet *scs) {\
    shared_cpu_flags = scs->static_config.use_cpu_flags;\
    svt_run_once(\&shared_rtcd_once, init_shared_rtcd);\
    svt_run_once(\&global_tables_once, init_global_tables);\
    shared_geom_idx = scs->svt_aom_geom_idx;\
    shared_geom_cnt = scs->max_block_cnt;\
    svt_run_once(\&shared_blk_geom_once, init_shared_blk_geom);\
    if (shared_blk_geom \&\& shared_blk_geom_idx == scs->svt_aom_geom_idx) {\
        scs->blk_geom_mds = shared_blk_geom;\
        return EB_ErrorNone;\
    }\
    EB_MALLOC_ARRAY(scs->blk_geom_mds, scs->max_block_cnt);\
    svt_aom_build_blk_geom(scs->svt_aom_geom_idx, scs->blk_geom_mds);\
    return EB_ErrorNone;\
}' Source/Lib/Globals/enc_handle.c
sed -i 's|handle->scs_instance->scs->blk_geom_mds != NULL) {|handle->scs_instance->scs->blk_geom_mds != NULL \&\& handle->scs_instance->scs->blk_geom_mds != shared_blk_geom) {|' Source/Lib/Globals/enc_handle.c
sed -i 's|^        SET_FUNCTIONS_X86(ptr, neon, neon_dotprod, neon_i8mm, sve, sve2) *\\$|        SET_FUNCTIONS_X86(ptr, mmx, sse, sse2, sse3, ssse3, sse4_1, sse4_2, avx, avx2, avx512) \\|' Source/Lib/Codec/aom_dsp_rtcd.c Source/Lib/Codec/common_dsp_rtcd.c
sed -i 's|^    if (scs->static_config.encoder_bit_depth == EB_EIGHT_BIT) {$|    if (0) {|' Source/Lib/Globals/enc_handle.c
sed -i 's|^    if (validate_on_the_fly_settings(p_buffer,scs, enc_handle_ptr->scs_instance->config_mutex)) {$|    if (0) {|' Source/Lib/Globals/enc_handle.c
sed -i 's|^    EbErrorType return_error = svt_av1_verify_settings(scs);$|    EbErrorType return_error = EB_ErrorNone;|' Source/Lib/Globals/enc_handle.c
'@
    }
    else {
        $patchScript += @'

grep -q svt5fish_cpuinfo Source/Lib/Globals/enc_handle.c || {
    find third_party/cpuinfo -type f \( -name '*.c' -o -name '*.h' \) -exec sed -i 's/cpuinfo_/svt5fish_cpuinfo_/g' {} +
    sed -i 's/cpuinfo_/svt5fish_cpuinfo_/g' Source/Lib/Codec/common_dsp_rtcd.c Source/Lib/Globals/enc_handle.c
}
sed -i 's|^    if(validate_on_the_fly_settings(|    if(0 \&\& validate_on_the_fly_settings(|' Source/Lib/Globals/enc_handle.c
sed -i '/^    EbErrorType return_error = svt_av1_verify_settings($/,+1c\
    EbErrorType return_error = EB_ErrorNone;' Source/Lib/Globals/enc_handle.c
'@
    }

    if ($apply8MbStackPatch) {
        $patchScript += @'

sed -i 's/\r//g' Source/Lib/Codec/svt_threads.c
sed -i 's/1 MiB/8 MiB/g' Source/Lib/Codec/svt_threads.c
sed -i 's|0, // default stack size|8 * 1024 * 1024, // default stack size|' Source/Lib/Codec/svt_threads.c
sed -i 's|0, // thread active when created|STACK_SIZE_PARAM_IS_A_RESERVATION, // thread active when created|' Source/Lib/Codec/svt_threads.c
sed -i 's|const size_t min_stack_size = 1024 \* 1024;|const size_t min_stack_size = 8 * 1024 * 1024;|' Source/Lib/Codec/svt_threads.c
'@
    }

    if (Test-CpuFlag -MsysExe $MsysExe -Flag 'avx2') {
        $patchScript += @'

if [ -f Source/API/EbConfigMacros.h ]; then
    sed -i 's/\r//g' Source/API/EbConfigMacros.h
    sed -i '/^#ifndef CONFIG_X86_AVX2_IS_GUARANTEED$/,/^#endif$/s/       0$/       1/' Source/API/EbConfigMacros.h
fi
'@
    }

    Set-Content -Path 'patch_svt.sh' -Value ($patchScript.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixPath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching SVT-AV1 (sed)" {
        & $MsysExe -lc "cd `"$unixPath`" && sh ./patch_svt.sh"
    }
    Remove-Item 'patch_svt.sh' -Force -ErrorAction SilentlyContinue
}

function Build-SvtAv1 {
    param([string]$Variant, [string]$Dir, [string]$Branch, [string]$Repo, [string]$ExtraCFlags, [string]$ArchFlags, [bool]$NoPrompt, [string]$MsysExe)

    if (Test-Path 'lib\SvtAv1Enc.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] $Variant is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile ${Variant}? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping $Variant compilation..." -ForegroundColor Cyan
            return
        }
    }

    $avx512Supported = Test-CpuFlag -MsysExe $MsysExe -Flag 'avx512f'
    $svtAvx512Flag = if ($avx512Supported) { 'ON' } else { 'OFF' }
    Write-Host "[INFO] Detected AVX512 support: $avx512Supported. SVT-AV1 will be built with -DENABLE_AVX512=$svtAvx512Flag." -ForegroundColor Cyan

    if (Test-Path $Dir) { Remove-Item -Recurse -Force $Dir }
    if ($Branch) {
        # depth 300 to get the git tag
        git clone --depth 300 --branch $Branch $Repo $Dir
    }
    else {
        git clone --depth 300 $Repo $Dir
    }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone $Repo into $Dir." -ForegroundColor Red
        exit 1
    }
    Push-Location $Dir

    Patch-SvtAv1 -Variant $Variant -MsysExe $MsysExe

    $pgoDir = "$PWD/svt_pgo_data"
    if (Test-Path $pgoDir) { Remove-Item -Recurse -Force $pgoDir }
    New-Item -ItemType Directory $pgoDir | Out-Null

    if (Test-Path svt_build_gen) { Remove-Item -Recurse -Force svt_build_gen }

    Invoke-Step "Configuring $Variant (PGO Gen)" {
        cmake -B svt_build_gen -G Ninja -DCMAKE_BUILD_TYPE=Release -DBUILD_SHARED_LIBS=OFF `
            -DSVT_AV1_LTO=OFF -DLIBDOVI_FOUND=0 -DLIBHDR10PLUS_RS_FOUND=0 -DENABLE_AVX512=$svtAvx512Flag `
            -DCMAKE_CXX_FLAGS_RELEASE="-flto -DNDEBUG -O2 $ArchFlags $ExtraCFlags -fprofile-generate=$pgoDir" `
            -DCMAKE_C_FLAGS_RELEASE="-flto -DNDEBUG -O2 $ArchFlags $ExtraCFlags -fprofile-generate=$pgoDir" `
            -DLOG_QUIET=ON -DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded -DCMAKE_OUTPUT_DIRECTORY=svt_build_gen
    }

    Invoke-Step "Building $Variant (PGO Gen)" { ninja -C svt_build_gen }

    $clipFile = @("y4m", "mkv", "mp4", "webm") |
    ForEach-Object { "..\pgo_clip.$_" } |
    Where-Object { Test-Path $_ } |
    Select-Object -First 1

    if ($clipFile) {
        Write-Host "[INFO] $clipFile found. Using it for PGO." -ForegroundColor Cyan
    }
    else {
        $clipUrl = "https://media.xiph.org/video/derf/webm/Netflix_FoodMarket2_4096x2160_60fps_10bit_420.webm"
        $clipHash = "F625E9460AA7964855D00C4CAD535D910EC4EEC7594B4CCEB5611CB00CC5F75B"
        $clipFile = "Netflix_FoodMarket2_4096x2160_60fps_10bit_420.webm"

        if (-not (Test-Path "..\$clipFile")) {
            Write-Host "[INFO] Downloading PGO clip..." -ForegroundColor Cyan
            Invoke-WebRequest -Uri $clipUrl -OutFile "..\$clipFile" -UseBasicParsing
        }

        $actualHash = (Get-FileHash "..\$clipFile" -Algorithm SHA256).Hash
        if ($actualHash -ne $clipHash) {
            Write-Host "[ERROR] PGO clip hash mismatch. Expected $clipHash, got $actualHash." -ForegroundColor Red
            exit 1
        }

        if (-not (Test-Path "..\Netflix_FoodMarket2_1920x1080_60fps_10bit_420_96f.y4m")) {
            & "..\FFmpeg\ffmpeg.exe" -hide_banner -v error -stats -y -nostdin -i "..\$clipFile" -frames:v 96 -vf "scale=1920:1080:flags=lanczos+accurate_rnd+full_chroma_int:param0=4" -pix_fmt yuv420p10le -strict -1 -f yuv4mpegpipe "..\Netflix_FoodMarket2_1920x1080_60fps_10bit_420_96f.y4m"
        }
        $clipFile = "..\Netflix_FoodMarket2_1920x1080_60fps_10bit_420_96f.y4m"
    }

    Invoke-Step "PGO Run for $Variant" {
        if ($clipFile -notlike '*.y4m') {
            & "..\FFmpeg\ffmpeg.exe" -i "$clipFile" -f yuv4mpegpipe -pix_fmt yuv420p10le -strict -1 - | & ".\svt_build_gen\SvtAv1EncApp.exe" -i - -b NUL --preset 3 | Out-Host
        }
        else {
            & ".\svt_build_gen\SvtAv1EncApp.exe" -i "$clipFile" -b NUL --preset 3 | Out-Host
        }
    }

    $profdata = "$pgoDir/default.profdata"
    Invoke-Step "Merging PGO data" {
        & "llvm-profdata" merge --sparse=true -o $profdata $pgoDir | Out-Host
    }

    if (Test-Path svt_build_use) { Remove-Item -Recurse -Force svt_build_use }

    Invoke-Step "Configuring $Variant (PGO Use)" {
        cmake -B svt_build_use -G Ninja -DCMAKE_BUILD_TYPE=Release -DBUILD_SHARED_LIBS=OFF `
            -DSVT_AV1_LTO=OFF -DLIBDOVI_FOUND=0 -DLIBHDR10PLUS_RS_FOUND=0 -DENABLE_AVX512=$svtAvx512Flag `
            -DCMAKE_CXX_FLAGS_RELEASE="-flto -DNDEBUG -O2 $ArchFlags $ExtraCFlags -fprofile-use=$profdata" `
            -DCMAKE_C_FLAGS_RELEASE="-flto -DNDEBUG -O2 $ArchFlags $ExtraCFlags -fprofile-use=$profdata" `
            -DLOG_QUIET=ON -DBUILD_APPS=OFF -DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded -DCMAKE_OUTPUT_DIRECTORY=svt_build_use
    }

    Invoke-Step "Building $Variant (PGO Use)" { ninja -C svt_build_use }
    Pop-Location
    Copy-Item "$Dir\svt_build_use\SvtAv1Enc.lib" lib\ -Force
}

function Build-Opus {
    if (Test-Path 'lib\opus.lib') {
        Write-Host "[INFO] Opus already compiled. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (Test-Path 'opus') { Push-Location opus; git pull; Pop-Location }
        else { git clone --depth 1 https://gitlab.xiph.org/xiph/opus.git }
        Push-Location opus
        Invoke-Step "Configuring Opus" {
            cmake -B build -G Ninja `
                -DCMAKE_BUILD_TYPE=Release `
                -DCMAKE_C_FLAGS_RELEASE="$COMMON_FLAGS" `
                -DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded
        }
        Invoke-Step "Building Opus" { ninja -C build }
        Pop-Location
        Copy-Item opus\build\opus.lib lib\ -Force
    }
}

function Build-Vulkan {
    param([string]$VsPath)
    if ((Test-Path 'lib\vulkan-1.lib') -and (Test-Path 'install\include\spirv\unified1\spirv.h')) {
        Write-Host "[INFO] Vulkan and SPIRV-Headers already compiled/installed. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (-not (Test-Path 'vulkan')) { New-Item -ItemType Directory 'vulkan' | Out-Null }
        Push-Location vulkan

        if (Test-Path 'Vulkan-Headers') { Push-Location 'Vulkan-Headers'; git pull; Pop-Location }
        else { git clone --depth 1 https://github.com/KhronosGroup/Vulkan-Headers.git }

        Invoke-Step "Configuring Vulkan Headers" {
            cmake -S Vulkan-Headers -B Vulkan-Headers/build -G Ninja `
                -DCMAKE_BUILD_TYPE=Release `
                -DCMAKE_INSTALL_PREFIX="$PWD/../install" `
                -DCMAKE_C_FLAGS_RELEASE="$COMMON_FLAGS" `
                -DCMAKE_CXX_FLAGS_RELEASE="$COMMON_FLAGS"
        }
        Invoke-Step "Installing Vulkan Headers" {
            ninja -C Vulkan-Headers/build install
        }

        if (Test-Path 'SPIRV-Headers') { Push-Location 'SPIRV-Headers'; git pull; Pop-Location }
        else { git clone --depth 1 https://github.com/KhronosGroup/SPIRV-Headers.git }

        Invoke-Step "Configuring SPIRV-Headers" {
            cmake -S SPIRV-Headers -B SPIRV-Headers/build -G Ninja `
                -DCMAKE_BUILD_TYPE=Release `
                -DCMAKE_INSTALL_PREFIX="$PWD/../install"
        }
        Invoke-Step "Installing SPIRV-Headers" {
            ninja -C SPIRV-Headers/build install
        }

        if (Test-Path 'Vulkan-Loader') { Push-Location 'Vulkan-Loader'; git pull; Pop-Location }
        else { git clone --depth 1 https://github.com/KhronosGroup/Vulkan-Loader.git }

        Invoke-Step "Building Vulkan Loader" {
            cmake -S Vulkan-Loader -B Vulkan-Loader/build -G Ninja `
                -DCMAKE_BUILD_TYPE=Release `
                -DCMAKE_INSTALL_PREFIX="$PWD/../install" `
                -DBUILD_SHARED_LIBS=ON `
                -DCMAKE_ASM_MASM_COMPILER="ml64.exe" `
                -DVULKAN_HEADERS_INSTALL_DIR="$PWD/../install" `
                -DCMAKE_C_FLAGS_RELEASE="$COMMON_FLAGS"
            ninja -C Vulkan-Loader/build
            ninja -C Vulkan-Loader/build install
        }

        Pop-Location

        Copy-Item 'install\lib\vulkan-1.lib' 'lib\vulkan-1.lib' -Force
    }
}

function Build-NvHeaders {
    param([string]$MsysExe)
    if (Test-Path 'nv-codec-headers\install\lib\pkgconfig\ffnvcodec.pc') {
        Write-Host "[INFO] nv-codec-headers already installed. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (Test-Path 'nv-codec-headers') {
            Push-Location nv-codec-headers
            git pull
            Pop-Location
        }
        else {
            git clone --depth 1 https://github.com/FFmpeg/nv-codec-headers.git
        }
        Push-Location nv-codec-headers
        $unixPwd = $PWD.Path -replace '\\', '/'
        Invoke-Step "Installing nv-codec-headers" {
            & $MsysExe -lc "cd `"$unixPwd`" && make install PREFIX=`"$unixPwd/install`""
        }
        Pop-Location
    }
}

# dav1d
function Build-Dav1d {
    if (Test-Path 'lib\dav1d.lib') {
        Write-Host "[INFO] dav1d already compiled. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (Test-Path 'dav1d') { Push-Location dav1d; git pull; Pop-Location }
        else { git clone --depth 1 https://code.videolan.org/videolan/dav1d.git }
        Push-Location dav1d
        if (-not (Assert-Command 'meson')) {
            $mesonVersion = '1.12.0'
            $mesonUrl = "https://github.com/mesonbuild/meson/releases/download/$mesonVersion/meson-$mesonVersion-64.msi"
            $mesonHash = 'c35383106327e20795a3c4010d40db0bfe952baf5490584c2535f70899833dce'
            $mesonMsi = "$env:TEMP\meson-$mesonVersion-64.msi"
            $mesonExe = "$env:ProgramFiles\Meson\meson.exe"

            Write-Host ""
            Write-Host "[PROMPT] Meson is missing." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to install Meson $($mesonVersion)? (Y/N) [Default: Y]"
            if ($choice -ieq 'N') {
                Write-Host "[ERROR] Meson is required to build dav1d. Exiting." -ForegroundColor Red
                exit 1
            }

            Write-Host "[INFO] Downloading Meson $mesonVersion..." -ForegroundColor Cyan
            Invoke-WebRequest -Uri $mesonUrl -OutFile $mesonMsi -UseBasicParsing

            $actual = (Get-FileHash $mesonMsi -Algorithm SHA256).Hash
            if ($actual -ne $mesonHash.ToUpper()) {
                Write-Host "[ERROR] Meson installer checksum mismatch. Expected $mesonHash, got $actual." -ForegroundColor Red
                exit 1
            }

            Write-Host "[INFO] Installing Meson $mesonVersion silently..." -ForegroundColor Cyan
            $proc = Start-Process -FilePath 'msiexec.exe' -ArgumentList "/i `"$mesonMsi`" /quiet /qn ALLUSERS=1" -Wait -PassThru -Verb RunAs
            if ($proc.ExitCode -ne 0) {
                Write-Host "[ERROR] Meson installer failed (exit code $($proc.ExitCode))." -ForegroundColor Red
                exit 1
            }

            if (Test-Path $mesonExe) {
                $mesonBin = Split-Path $mesonExe
                if ($env:PATH -notlike "*$mesonBin*") {
                    $env:PATH = "$mesonBin;$env:PATH"
                }
            }
            else {
                Write-Host "[ERROR] meson.exe not found at $mesonExe after install." -ForegroundColor Red
                exit 1
            }
            Write-Host "[INFO] Meson $mesonVersion installed successfully." -ForegroundColor Green
        }
        Invoke-Step "Building dav1d" {
            meson setup --reconfigure build --prefix="$PWD/../install" --default-library=static --buildtype=release -Db_vscrt=mt -Db_lto=true -Doptimization=3 -Denable_tools=false -Denable_examples=false -Dc_args="-DNDEBUG -march=native" -Dc_link_args="-fuse-ld=lld"
            ninja -C build
            meson install -C build
        }
        Pop-Location
        Copy-Item dav1d\build\src\libdav1d.a lib\dav1d.lib -Force
    }
}

function Patch-FFmpeg {
    param([string]$MsysExe)

    $patchScript = @'
#!/bin/sh
set -e
grep -q '^disable inline_asm_nonlocal_labels$' configure || sed -i '/^check_optflags -fno-signed-zeros$/a\
disable inline_asm_nonlocal_labels\
disable inline_asm_direct_symbol_refs' configure
'@
    Set-Content -Path 'patch_ffmpeg.sh' -Value ($patchScript.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixPath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching FFmpeg (sed)" {
        & $MsysExe -lc "cd `"$unixPath`" && sh ./patch_ffmpeg.sh"
    }
    Remove-Item 'patch_ffmpeg.sh' -Force -ErrorAction SilentlyContinue
}

function Build-FFmpeg {
    param([string]$VsPath, [string]$MsysExe, [string]$Backend)
    
    $env:INCLUDE = "$PWD\dav1d\include;$PWD\dav1d\build\include;$PWD\install\include;$env:INCLUDE"
    if ($Backend -eq 'cuda') {
        $env:INCLUDE = "$env:CUDA_PATH\include;$env:INCLUDE"
        $env:LIB = "$env:CUDA_PATH\lib\x64;$PWD\lib;$PWD\install\lib;$env:LIB"
    }
    else {
        $env:LIB = "$PWD\lib;$PWD\install\lib;$env:LIB"
    }

    if (Test-Path 'lib\avcodec.lib') {
        Write-Host "[INFO] FFmpeg already compiled. Skipping..." -ForegroundColor Cyan
    }
    else {
        if (Test-Path 'FFmpeg') {
            Push-Location FFmpeg
            git reset --hard
            git pull
            Pop-Location
        }
        else { git clone --depth 1 https://github.com/FFmpeg/FFmpeg.git }
        Push-Location FFmpeg
        Patch-FFmpeg -MsysExe $MsysExe
        Invoke-Step "Building FFmpeg" {
            $llvmBinWin = "C:\Program Files\LLVM\bin"

            if ($Backend -eq 'cuda') {
                $hwArgs = @'
    --enable-ffnvcodec \
    --enable-nvdec \
    --enable-cuvid \
    --enable-decoder=h264_cuvid \
    --enable-decoder=hevc_cuvid \
    --enable-decoder=av1_cuvid \
    --enable-decoder=vp9_cuvid \
    --enable-decoder=vc1_cuvid \
'@
                $hwPkgConfig = 'export PKG_CONFIG_PATH="$(pwd)/../nv-codec-headers/install/lib/pkgconfig:$PKG_CONFIG_PATH"'
            }
            else {
                $hwArgs = @'
    --enable-vulkan \
    --enable-hwaccel=h264_vulkan \
    --enable-hwaccel=hevc_vulkan \
    --enable-hwaccel=av1_vulkan \
    --enable-hwaccel=vp9_vulkan \
'@
                $hwPkgConfig = ""
            }

            $bashScript = @"
#!/bin/sh
set -e
export PATH="`$(cygpath -u '$llvmBinWin'):`$PATH"
"@ + "`n" + @'
export PKG_CONFIG_PATH="$(pwd)/../install/lib/pkgconfig"

./configure \
    --disable-all \
    --disable-everything \
    --cc="clang-cl" \
    --cxx="clang-cl" \
    --ld="lld-link" \
    --ar="llvm-ar" \
    --nm="llvm-nm" \
    --ranlib="llvm-ranlib" \
    --strip="llvm-strip" \
    --toolchain="msvc" \
    --extra-cflags="-flto /clang:-O3 -DNDEBUG -march=native" \
	--extra-cxxflags="-flto /clang:-O3 -DNDEBUG -march=native" \
    --disable-shared \
    --enable-static \
    --pkg-config-flags="--static" \
    --disable-programs \
    --disable-doc \
    --disable-network \
    --disable-autodetect \
    --disable-debug \
    --enable-ffmpeg \
    --disable-ffprobe \
    --disable-ffplay \
    --enable-avcodec \
    --enable-avformat \
    --enable-avutil \
    --enable-avfilter \
    --enable-swscale \
    --enable-swresample \
    --enable-protocol=file \
    --enable-protocol=pipe \
    --enable-demuxer=matroska \
    --enable-demuxer=mov \
    --enable-demuxer=mpegts \
    --enable-demuxer=mpegps \
    --enable-demuxer=flv \
    --enable-demuxer=avi \
    --enable-demuxer=ivf \
    --enable-demuxer=yuv4mpegpipe \
    --enable-muxer=yuv4mpegpipe \
    --enable-demuxer=h264 \
    --enable-demuxer=hevc \
    --enable-demuxer=vvc \
    --enable-decoder=ffv1 \
    --enable-decoder=rawvideo \
    --enable-encoder=wrapped_avframe \
    --enable-decoder=h264 \
    --enable-decoder=hevc \
    --enable-decoder=mpeg2video \
    --enable-decoder=mpeg1video \
    --enable-decoder=mpeg4 \
    --enable-decoder=av1 \
    --enable-decoder=libdav1d \
    --enable-decoder=vp9 \
    --enable-decoder=vc1 \
    --enable-decoder=vvc \
    --enable-decoder=utvideo \
    --enable-decoder=aac \
    --enable-decoder=aac_latm \
    --enable-decoder=ac3 \
    --enable-decoder=eac3 \
    --enable-decoder=dca \
    --enable-decoder=truehd \
    --enable-decoder=mlp \
    --enable-decoder=mp1 \
    --enable-decoder=mp1float \
    --enable-decoder=mp2 \
    --enable-decoder=mp2float \
    --enable-decoder=mp3 \
    --enable-decoder=mp3float \
    --enable-decoder=opus \
    --enable-decoder=vorbis \
    --enable-decoder=flac \
    --enable-decoder=alac \
    --enable-decoder=ape \
    --enable-decoder=tak \
    --enable-decoder=tta \
    --enable-decoder=wavpack \
    --enable-decoder=wmalossless \
    --enable-decoder=wmapro \
    --enable-decoder=wmav1 \
    --enable-decoder=wmav2 \
    --enable-decoder=mpc7 \
    --enable-decoder=mpc8 \
    --enable-decoder=dsd_lsbf \
    --enable-decoder=dsd_lsbf_planar \
    --enable-decoder=dsd_msbf \
    --enable-decoder=dsd_msbf_planar \
    --enable-decoder=pcm_s16le \
    --enable-decoder=pcm_s16be \
    --enable-decoder=pcm_s24le \
    --enable-decoder=pcm_s24be \
    --enable-decoder=pcm_s32le \
    --enable-decoder=pcm_s32be \
    --enable-decoder=pcm_f32le \
    --enable-decoder=pcm_f32be \
    --enable-decoder=pcm_f64le \
    --enable-decoder=pcm_f64be \
    --enable-decoder=pcm_bluray \
    --enable-decoder=pcm_dvd \
    --enable-libdav1d \
    --enable-parser=h264 \
    --enable-parser=hevc \
    --enable-parser=mpeg4video \
    --enable-parser=mpegvideo \
    --enable-parser=av1 \
    --enable-parser=vp9 \
    --enable-parser=vvc \
    --enable-parser=vc1 \
    --enable-parser=aac \
    --enable-parser=ac3 \
    --enable-parser=dca \
    --enable-parser=mpegaudio \
    --enable-parser=opus \
    --enable-parser=vorbis \
    --enable-parser=flac \
    __HW_ARGS__
    --enable-bsf=extract_extradata \
    --enable-demuxer=ogg \
    --enable-filter=scale \
    --enable-filter=format
make -j$(nproc)
'@
            $bashScript = $bashScript.Replace('export PKG_CONFIG_PATH="$(pwd)/../install/lib/pkgconfig"', 'export PKG_CONFIG_PATH="$(pwd)/../install/lib/pkgconfig"' + "`n" + $hwPkgConfig).Replace('__HW_ARGS__', $hwArgs.TrimEnd([char[]]("`n", "`r")))
            Set-Content -Path 'build_ffmpeg.sh' -Value $bashScript -Encoding Ascii
            $env:MSYS2_PATH_TYPE = 'inherit'
            $unixPath = $PWD.Path -replace '\\', '/'
            & $MsysExe -lc "cd `"$unixPath`" && sh ./build_ffmpeg.sh"
        }
        Pop-Location
        Copy-Item 'FFmpeg\libavcodec\avcodec.lib' 'lib\avcodec.lib' -Force
        Copy-Item 'FFmpeg\libavformat\avformat.lib' 'lib\avformat.lib' -Force
        Copy-Item 'FFmpeg\libavutil\avutil.lib' 'lib\avutil.lib' -Force
        Copy-Item 'FFmpeg\libswresample\swresample.lib' 'lib\swresample.lib' -Force
    }
}

function Patch-Avm {
    param([string]$MsysExe)

    $patchScript = @'
#!/bin/sh
set -e
sed -i 's/\r//g' av2/encoder/part_split_prune_tflite.cc av2/encoder/intra_dip_mode_prune_tflite.cc avm_scale/avm_scale.cmake avm/src/avm_encoder.c av2/av2_cx_iface.c
grep -q tls_part_split av2/encoder/part_split_prune_tflite.cc || sed -i '/^static void ensure_tflite_init(void \*\*context, MODEL_TYPE model_type) {$/i\
static thread_local PartSplitContext *tls_part_split;' av2/encoder/part_split_prune_tflite.cc
sed -i 's|^  if (\*context == nullptr) \*context = new PartSplitContext();$|  if (*context == nullptr) {\n    if (tls_part_split == nullptr) tls_part_split = new PartSplitContext();\n    *context = tls_part_split;\n  }|' av2/encoder/part_split_prune_tflite.cc
sed -i '/^extern "C" void av2_part_prune_tflite_close(void \*\*context) {$/,/^}$/c\
extern "C" void av2_part_prune_tflite_close(void **context) { *context = nullptr; }' av2/encoder/part_split_prune_tflite.cc
grep -q tls_dip av2/encoder/intra_dip_mode_prune_tflite.cc || sed -i '/^static void ensure_tflite_init(void \*\*context, int model_index) {$/i\
static thread_local DipContext *tls_dip;' av2/encoder/intra_dip_mode_prune_tflite.cc
sed -i 's|^    \*context = new DipContext();$|    if (tls_dip == nullptr) tls_dip = new DipContext();\n    *context = tls_dip;|' av2/encoder/intra_dip_mode_prune_tflite.cc
sed -i '/^extern "C" void intra_dip_mode_prune_close(void \*\*context) {$/,/^}$/c\
extern "C" void intra_dip_mode_prune_close(void **context) { *context = nullptr; }' av2/encoder/intra_dip_mode_prune_tflite.cc
sed -i '\|generic/avm_scale.c"$|d;\|generic/gen_scalers.c"$|d' avm_scale/avm_scale.cmake
sed -i 's/^#if ARCH_X86 || ARCH_X86_64$/#if 0/' avm/src/avm_encoder.c
sed -i 's/if (!ctx || (img \&\& !duration))/if (0)/' avm/src/avm_encoder.c
sed -i 's/if (!ctx->iface || !ctx->priv)/if (0)/g' avm/src/avm_encoder.c
sed -i 's/if (!(ctx->iface->caps \& AVM_CODEC_CAP_ENCODER))/if (0)/g' avm/src/avm_encoder.c
sed -i 's/if (!iter)/if (0)/' avm/src/avm_encoder.c
sed -i 's|^                                    const avm_image_t \*img) {$|                                    const avm_image_t *img) { return AVM_CODEC_OK;|' av2/av2_cx_iface.c
sed -i 's|^                                       const struct av2_extracfg \*extra_cfg) {$|                                       const struct av2_extracfg *extra_cfg) { return AVM_CODEC_OK;|' av2/av2_cx_iface.c
grep -q '^    if (ctx->cx_data == NULL) {$' av2/av2_cx_iface.c || sed -i '/^static avm_codec_err_t encoder_encode(/,/^    if (res == AVM_CODEC_OK) {$/s|^    if (res == AVM_CODEC_OK) {$|    if (ctx->cx_data == NULL) {|' av2/av2_cx_iface.c
'@
    Set-Content -Path 'patch_avm.sh' -Value ($patchScript.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixPath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching AVM (sed)" {
        & $MsysExe -lc "cd `"$unixPath`" && sh ./patch_avm.sh"
    }
    Remove-Item 'patch_avm.sh' -Force -ErrorAction SilentlyContinue
}

function Build-Avm {
    param([bool]$NoPrompt, [string]$MsysExe)
    if (Test-Path 'lib\avm_full.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] avm is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile avm? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping avm compilation..." -ForegroundColor Cyan
            return
        }
    }
    if (Test-Path 'avm') {
        Push-Location avm
        git reset --hard
        git pull
        Pop-Location
    }
    else { git clone --depth 1 https://github.com/AOMediaCodec/avm }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone https://github.com/AOMediaCodec/avm into avm." -ForegroundColor Red
        exit 1
    }
    Push-Location avm
    Invoke-Step "Patching AVM TFLite CMakeLists" { git apply "..\patch\tflite-windows-cmakelists.patch" }
    Patch-Avm -MsysExe $MsysExe
    if (Test-Path build) { Remove-Item -Recurse -Force build }
    $cmakeArgs = @('-B', 'build', '-G', 'Ninja',
        '-DCMAKE_BUILD_TYPE=Release',
        '-DCMAKE_C_COMPILER=clang-cl',
        '-DCMAKE_CXX_COMPILER=clang-cl',
        '-DCMAKE_C_FLAGS=-flto /clang:-O3 -DNDEBUG -march=native',
        '-DCMAKE_CXX_FLAGS=-flto /clang:-O3 -DNDEBUG -march=native /EHsc',
        '-DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded',
        '-DBUILD_SHARED_LIBS=OFF',
        '-DENABLE_APPS=0',
        '-DENABLE_EXAMPLES=0',
        '-DENABLE_TESTS=0',
        '-DENABLE_TOOLS=0',
        '-DENABLE_DOCS=0',
        '-DENABLE_NASM=1',
        '-DCONFIG_AV2_ENCODER=1',
        '-DCONFIG_AV2_DECODER=0',
        '-DCONFIG_WEBM_IO=0',
        '-DCONFIG_RUNTIME_CPU_DETECT=0',
        '-DCONFIG_MULTITHREAD=0',
        '-DCONFIG_LIBYUV=0',
        '-DCONFIG_LANCZOS_RESAMPLE=0',
        '-DCONFIG_SPATIAL_RESAMPLING=0',
        '-DCONFIG_12BIT_PROFILE=0',
        '-DCONFIG_DENOISE=0',
        '-DCONFIG_TENSORFLOW_LITE=1')
    Invoke-Step "Configuring AVM" { cmake @cmakeArgs }
    $litePatch = @'
#!/bin/sh
set -e
sed -i 's/\r//g' third_party/tensorflow/tensorflow/lite/CMakeLists.txt
sed -i 's#FILTER "(.*_test_util_internal|test_.*|.*_ops_wrapper)\\\\.(cc|h)"#FILTER "(/[^/]*_test_util_internal[^/]*|/test_[^/]*|/[^/]*_ops_wrapper)\\\\.(cc|h)$"#' third_party/tensorflow/tensorflow/lite/CMakeLists.txt
'@
    Set-Content -Path 'patch_avm_lite.sh' -Value ($litePatch.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixLitePath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching AVM TFLite file list (sed)" {
        & $MsysExe -lc "cd `"$unixLitePath`" && sh ./patch_avm_lite.sh"
    }
    Remove-Item 'patch_avm_lite.sh' -Force -ErrorAction SilentlyContinue
    Invoke-Step "Reconfiguring AVM" { cmake @cmakeArgs }
    Invoke-Step "Building AVM" { ninja -C build avm }
    if (-not (Test-Path 'build\avm.lib')) {
        Write-Host "[ERROR] build\avm.lib not found after AVM build." -ForegroundColor Red
        exit 1
    }
    $depLibs = @(Get-ChildItem -Path 'build' -Filter '*.lib' -Recurse -File | Where-Object { $_.Name -ne 'avm.lib' -and $_.Name -ne 'avm_full.lib' })
    $libArgs = New-Object System.Collections.Generic.List[string]
    $libArgs.Add('/OUT:build\avm_full.lib')
    $libArgs.Add('build\avm.lib')
    foreach ($d in $depLibs) { $libArgs.Add($d.FullName) }
    Invoke-Step "Archiving AVM" { llvm-lib @libArgs }
    Pop-Location
    Copy-Item 'avm\build\avm_full.lib' 'lib\avm_full.lib' -Force
}

function Patch-Vvenc {
    param([string]$MsysExe)

    $patchScript = @'
#!/bin/sh
set -e
sed -i 's/\r//g' CMakeLists.txt source/Lib/vvenc/vvencimpl.cpp source/Lib/vvenc/vvencimpl.h source/Lib/vvenc/vvenc.cpp source/Lib/vvenc/vvencCfg.cpp source/Lib/EncoderLib/EncGOP.cpp source/Lib/EncoderLib/EncPicture.cpp source/Lib/CommonLib/TypeDef.h
sed -i 's/set( CMAKE_POSITION_INDEPENDENT_CODE TRUE )/set( CMAKE_POSITION_INDEPENDENT_CODE FALSE )/' CMakeLists.txt
sed -i 's/set( CMAKE_CXX_STANDARD 14 )/set( CMAKE_CXX_STANDARD 20 )/' CMakeLists.txt
sed -i '/setSIMDExtension( nullptr );/d' source/Lib/vvenc/vvencimpl.cpp
sed -i '/m_cEncoderInfo = createEncoderInfoStr();/d' source/Lib/vvenc/vvencimpl.cpp
sed -i '/m_cVVEncCfgExt = \*config;/d' source/Lib/vvenc/vvencimpl.cpp
sed -i '/vvenc_config           m_cVVEncCfgExt;/d' source/Lib/vvenc/vvencimpl.h
sed -i 's/if ( vvenc_init_config_parameter(&m_cVVEncCfg) )/if ( !m_cVVEncCfg.m_configDone \&\& vvenc_init_config_parameter(\&m_cVVEncCfg) )/' source/Lib/vvenc/vvencimpl.cpp
sed -i '/malloc_trim(0);/d' source/Lib/vvenc/vvencimpl.cpp
sed -i 's/! xConvertVerifyYUVBuffer( pcYUVBuffer )/false/' source/Lib/vvenc/vvencimpl.cpp
grep -q 'false && !m_bInitialized' source/Lib/vvenc/vvencimpl.cpp || {
    sed -i '/^int VVEncImpl::encode(/,/^  int iRet= VVENC_OK;$/s|^  if(|  if( false \&\& |' source/Lib/vvenc/vvencimpl.cpp
    sed -i '/^    if( m_eState == INTERNAL_STATE_FLUSHING ) { m_cErrorString = "encoder already received flush indication/,/^    if ( false )$/s|^    if(|    if( false \&\& |' source/Lib/vvenc/vvencimpl.cpp
}
sed -i '/memset( accessUnit->infoString, 0, sizeof( accessUnit->infoString ) );/d' source/Lib/vvenc/vvenc.cpp
sed -i -e '/^  accessUnit->cts             = 0;$/d' -e '/^  accessUnit->dts             = 0;$/d' \
    -e '/^  accessUnit->ctsValid        = false;$/d' -e '/^  accessUnit->dtsValid        = false;$/d' \
    -e '/^  accessUnit->sliceType       = VVENC_NUMBER_OF_SLICE_TYPES;$/d' \
    -e '/^  accessUnit->refPic          = false;$/d' -e '/^  accessUnit->temporalLayer   = 0;$/d' \
    -e '/^  accessUnit->poc             = 0;$/d' -e '/^  accessUnit->status          = 0;$/d' \
    -e '/accessUnit->infoString\[0\]/d' source/Lib/vvenc/vvenc.cpp
sed -i '/^VVENC_DECL void vvenc_accessUnit_reset/,/^}/s|^  if( nullptr == accessUnit )$|  if( false )|' source/Lib/vvenc/vvenc.cpp
sed -i 's|^  if ( ! bflag )$|  if ( true )|' source/Lib/vvenc/vvencCfg.cpp
sed -i 's/m_nalUnitData\.str()\.c_str()/m_nalUnitData.view().data()/g' source/Lib/vvenc/vvencimpl.cpp source/Lib/EncoderLib/EncGOP.cpp
sed -i 's/m_nalUnitData\.str()\.size()/m_nalUnitData.view().size()/g' source/Lib/vvenc/vvencimpl.cpp source/Lib/EncoderLib/EncGOP.cpp
sed -i '/xPrintPictureInfo ( pic, au, digestStr, m_pcEncCfg->m_printFrameMSE, isEncodeLtRef );/d' source/Lib/EncoderLib/EncGOP.cpp
sed -i '/xCalcDistortion( pic, \*slice->sps );/d' source/Lib/EncoderLib/EncPicture.cpp
sed -i 's|^#define CHECK(c,x)          if(c){ THROW(x); }$|#define CHECK(c,x)|' source/Lib/CommonLib/TypeDef.h
'@
    Set-Content -Path 'patch_vvenc.sh' -Value ($patchScript.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixPath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching VVenC (sed)" {
        & $MsysExe -lc "cd `"$unixPath`" && sh ./patch_vvenc.sh"
    }
    Remove-Item 'patch_vvenc.sh' -Force -ErrorAction SilentlyContinue
}

function Build-Vvenc {
    param([bool]$NoPrompt, [string]$MsysExe)
    if (Test-Path 'lib\vvenc.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] vvenc is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile vvenc? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping vvenc compilation..." -ForegroundColor Cyan
            return
        }
    }
    if (Test-Path 'vvenc') {
        Push-Location vvenc
        git reset --hard
        git pull
        Pop-Location
    }
    else { git clone --depth 1 https://github.com/fraunhoferhhi/vvenc }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone https://github.com/fraunhoferhhi/vvenc into vvenc." -ForegroundColor Red
        exit 1
    }
    Push-Location vvenc
    Patch-Vvenc -MsysExe $MsysExe
    if (Test-Path build) { Remove-Item -Recurse -Force build }
    if (Test-Path lib) { Remove-Item -Recurse -Force lib }
    $cmakeArgs = @('-B', 'build', '-G', 'Ninja',
        '-DCMAKE_BUILD_TYPE=Release',
        '-DCMAKE_C_COMPILER=clang',
        '-DCMAKE_CXX_COMPILER=clang++',
        '-DCMAKE_C_FLAGS=-flto -O3 -DNDEBUG -march=native',
        '-DCMAKE_CXX_FLAGS=-flto -O3 -DNDEBUG -march=native',
        '-DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded',
        '-DBUILD_SHARED_LIBS=OFF',
        '-DVVENC_ENABLE_WERROR=OFF',
        '-DVVENC_ENABLE_INSTALL=OFF',
        '-DVVENC_ENABLE_LINK_TIME_OPT=OFF',
        '-DVVENC_ENABLE_UNSTABLE_API=OFF',
        '-DVVENC_ENABLE_TRACING=OFF')
    Invoke-Step "Configuring VVenC" { cmake @cmakeArgs }
    Invoke-Step "Building VVenC" { ninja -C build vvenc }
    if (-not (Test-Path 'lib\release-static\vvenc.lib')) {
        Write-Host "[ERROR] lib\release-static\vvenc.lib not found after VVenC build." -ForegroundColor Red
        exit 1
    }
    Pop-Location
    Copy-Item 'vvenc\lib\release-static\vvenc.lib' 'lib\vvenc.lib' -Force
}

function Patch-Vvdec {
    param([string]$MsysExe)

    $patchScript = @'
#!/bin/sh
set -e
sed -i 's/\r//g' CMakeLists.txt source/Lib/vvdec/vvdecimpl.cpp source/Lib/DecoderLib/DecLib.cpp
sed -i 's/set( CMAKE_POSITION_INDEPENDENT_CODE TRUE )/set( CMAKE_POSITION_INDEPENDENT_CODE FALSE )/' CMakeLists.txt
sed -i 's/set( CMAKE_CXX_STANDARD 14 )/set( CMAKE_CXX_STANDARD 20 )/' CMakeLists.txt
grep -q 'false && !m_bInitialized' source/Lib/vvdec/vvdecimpl.cpp || {
    sed -i '/^int VVDecImpl::decode(/,/not supported feature detected/s|^  if(|  if( false \&\& |' source/Lib/vvdec/vvdecimpl.cpp
    sed -i '/^  if( !rcAccessUnit.payload )$/,/^  int iRet = VVDEC_OK;$/s|^  if(|  if( false \&\& |' source/Lib/vvdec/vvdecimpl.cpp
}
sed -i '/^      bool bStartCodeFound = false;$/,/^      pushNalEndPos( rcAccessUnit.payloadUsedSize );$/c\
      const size_t iStartCodeSizeVec[1] = { rcAccessUnit.payload[2] == 1 ? (size_t)3 : (size_t)4 };\
      const size_t iStartCodePosVec[1] = { iStartCodeSizeVec[0] };\
      int iLastPos = rcAccessUnit.payloadUsedSize;\
      while( iLastPos > 0 \&\& rcAccessUnit.payload[iLastPos-1] == 0 )\
      {\
        iLastPos--;\
      }\
      const size_t iAUEndPosVec[1] = { (size_t)iLastPos };' source/Lib/vvdec/vvdecimpl.cpp
sed -i 's/!iStartCodePosVec.empty() && iStartCodePosVec\[0\] != iStartCodeSizeVec\[0\]/false/' source/Lib/vvdec/vvdecimpl.cpp
sed -i 's/iAU < iStartCodePosVec.size()/iAU < 1/' source/Lib/vvdec/vvdecimpl.cpp
sed -i 's|parserFrameDelay = std::min<int>( ( numDecThreads \* DEFAULT_PARSE_DELAY_FACTOR ) >> 4, DEFAULT_PARSE_DELAY_MAX );|parserFrameDelay = (int) m_decLibRecon.size();|' source/Lib/DecoderLib/DecLib.cpp
'@
    Set-Content -Path 'patch_vvdec.sh' -Value ($patchScript.Replace("`r`n", "`n") + "`n") -Encoding Ascii -NoNewline
    $unixPath = $PWD.Path -replace '\\', '/'
    Invoke-Step "Patching VVdeC (sed)" {
        & $MsysExe -lc "cd `"$unixPath`" && sh ./patch_vvdec.sh"
    }
    Remove-Item 'patch_vvdec.sh' -Force -ErrorAction SilentlyContinue
}

function Build-Vvdec {
    param([bool]$NoPrompt, [string]$MsysExe)
    if (Test-Path 'lib\vvdec.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] vvdec is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile vvdec? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping vvdec compilation..." -ForegroundColor Cyan
            return
        }
    }
    if (Test-Path 'vvdec') {
        Push-Location vvdec
        git reset --hard
        git pull
        Pop-Location
    }
    else { git clone --depth 1 https://github.com/fraunhoferhhi/vvdec }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone https://github.com/fraunhoferhhi/vvdec into vvdec." -ForegroundColor Red
        exit 1
    }
    Push-Location vvdec
    Patch-Vvdec -MsysExe $MsysExe
    if (Test-Path build) { Remove-Item -Recurse -Force build }
    if (Test-Path lib) { Remove-Item -Recurse -Force lib }
    $cmakeArgs = @('-B', 'build', '-G', 'Ninja',
        '-DCMAKE_BUILD_TYPE=Release',
        '-DCMAKE_C_COMPILER=clang',
        '-DCMAKE_CXX_COMPILER=clang++',
        '-DCMAKE_C_FLAGS=-flto -O3 -DNDEBUG -march=native',
        '-DCMAKE_CXX_FLAGS=-flto -O3 -DNDEBUG -march=native',
        '-DCMAKE_MSVC_RUNTIME_LIBRARY=MultiThreaded',
        '-DBUILD_SHARED_LIBS=OFF',
        '-DVVDEC_LIBRARY_ONLY=ON',
        '-DVVDEC_ENABLE_WERROR=OFF',
        '-DVVDEC_ENABLE_LINK_TIME_OPT=OFF',
        '-DVVDEC_ENABLE_UNSTABLE_API=OFF',
        '-DVVDEC_ENABLE_TRACING=OFF')
    Invoke-Step "Configuring VVdeC" { cmake @cmakeArgs }
    Invoke-Step "Building VVdeC" { ninja -C build vvdec }
    if (-not (Test-Path 'lib\release-static\vvdec.lib')) {
        Write-Host "[ERROR] lib\release-static\vvdec.lib not found after VVdeC build." -ForegroundColor Red
        exit 1
    }
    Pop-Location
    Copy-Item 'vvdec\lib\release-static\vvdec.lib' 'lib\vvdec.lib' -Force
}

function Build-X265 {
    param([bool]$NoPrompt, [string]$MsysExe)
    if (Test-Path 'lib\x265.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] x265 is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile x265? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping x265 compilation..." -ForegroundColor Cyan
            return
        }
    }
    if (Test-Path 'x265_git') {
        Push-Location x265_git
        git reset --hard
        git pull
        Pop-Location
    }
    else { git clone --depth 1 https://bitbucket.org/multicoreware/x265_git }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone https://bitbucket.org/multicoreware/x265_git into x265_git." -ForegroundColor Red
        exit 1
    }
    Push-Location x265_git
    Copy-Item '..\patch\x265.sh' 'source\x265.sh' -Force
    $unixSource = (Join-Path $PWD 'source') -replace '\\', '/'
    Invoke-Step "Patching x265 (sed)" {
        & $MsysExe -lc "cd `"$unixSource`" && sh ./x265.sh"
    }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] x265 patch failed." -ForegroundColor Red
        Pop-Location
        exit 1
    }
    if (Test-Path 'source\build-xav') { Remove-Item -Recurse -Force 'source\build-xav' }
    $cmakeArgs = @('-S', 'source', '-B', 'source\build-xav', '-G', 'Ninja',
        '-DCMAKE_BUILD_TYPE=Release',
        '-DCMAKE_C_COMPILER=clang-cl',
        '-DCMAKE_CXX_COMPILER=clang-cl',
        '-DCMAKE_C_FLAGS=/clang:-O3 -DNDEBUG -march=native',
        '-DCMAKE_CXX_FLAGS=/clang:-O3 -DNDEBUG -march=native /EHsc',
        '-DSTATIC_LINK_CRT=ON',
        '-DHIGH_BIT_DEPTH=ON',
        '-DMAIN12=OFF',
        '-DEXPORT_C_API=ON',
        '-DENABLE_SHARED=OFF',
        '-DENABLE_CLI=OFF',
        '-DENABLE_PIC=OFF',
        '-DENABLE_ASSEMBLY=ON',
        '-DENABLE_LIBNUMA=OFF',
        '-DENABLE_HDR10_PLUS=OFF',
        '-DENABLE_SVT_HEVC=OFF',
        '-DENABLE_LIBVMAF=OFF',
        '-DENABLE_ALPHA=OFF',
        '-DENABLE_MULTIVIEW=OFF',
        '-DENABLE_SCC_EXT=OFF',
        '-DENABLE_TESTS=OFF',
        '-DDETAILED_CU_STATS=OFF',
        '-DCHECKED_BUILD=OFF',
        '-DWARNINGS_AS_ERRORS=OFF')
    Invoke-Step "Configuring x265" { cmake @cmakeArgs }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] x265 cmake configure failed." -ForegroundColor Red
        Pop-Location
        exit 1
    }
    Invoke-Step "Building x265" { ninja -C source\build-xav x265-static }
    Pop-Location
    if (-not (Test-Path 'x265_git\source\build-xav\x265-static.lib')) {
        Write-Host "[ERROR] x265-static.lib not found after x265 build." -ForegroundColor Red
        exit 1
    }
    Copy-Item 'x265_git\source\build-xav\x265-static.lib' 'lib\x265.lib' -Force
}

function Build-X264 {
    param([bool]$NoPrompt, [string]$MsysExe)
    if (Test-Path 'lib\x264.lib') {
        if ($NoPrompt) {
            $choice = if ($ForceRebuild) { 'Y' } else { 'N' }
        }
        else {
            Write-Host ""
            Write-Host "[PROMPT] x264 is already compiled." -ForegroundColor Yellow
            $choice = Read-Host "Do you want to update and recompile x264? (Y/N) [Default: N]"
        }
        if ($choice -notmatch '^[Yy]') {
            Write-Host "[INFO] Skipping x264 compilation..." -ForegroundColor Cyan
            return
        }
    }
    if (Test-Path 'x264') {
        Push-Location x264
        git reset --hard
        git pull
        Pop-Location
    }
    else { git clone --depth 1 https://code.videolan.org/videolan/x264.git }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] Failed to clone https://code.videolan.org/videolan/x264.git into x264." -ForegroundColor Red
        exit 1
    }
    Push-Location x264
    Copy-Item @('..\patch\x264.sh', '..\patch\x264-cc-filter.sh') -Destination '.' -Force
    $unixX264 = $PWD.Path -replace '\\', '/'
    Invoke-Step "Building x264" {
        & $MsysExe -lc "cd `"$unixX264`" && sh ./x264.sh"
    }
    if ($LASTEXITCODE -ne 0) {
        Write-Host "[ERROR] x264 build failed." -ForegroundColor Red
        Pop-Location
        exit 1
    }
    Pop-Location
    if (-not (Test-Path 'x264\libx264.a')) {
        Write-Host "[ERROR] libx264.a not found after x264 build." -ForegroundColor Red
        exit 1
    }
    Copy-Item 'x264\libx264.a' 'lib\x264.lib' -Force
}

function Build-Xav {
    param([string]$Backend, [string]$SvtChoice, [bool]$enableTQ, [bool]$enableAvm, [bool]$enableVvenc, [bool]$enableX265, [bool]$enableX264)
    $features = @()
    if ($enableAvm) { $features += "avm" }
    if ($enableVvenc) { $features += "vvenc" }
    if ($enableX265) { $features += "x265" }
    if ($enableX264) { $features += "x264" }
    if ($enableTQ) {
        $features += "vship"
        if ($Backend -eq 'cuda') { $features += "cuda" }
        elseif ($Backend -eq 'hip') { $features += "amd" }
    }

    if ($SvtChoice -eq '3') {
        $features += "5fish"
    }
    elseif ($SvtChoice -eq '2' -or $SvtChoice -eq '6') {
        $features += "svt-essential"
    }

    $featureArg = $features -join ","
    $releaseRustflags = "build.rustflags=['-C', 'debuginfo=0', '-C', 'target-cpu=native', '-C', 'opt-level=3', '-C', 'codegen-units=1', '-C', 'strip=symbols', '-C', 'panic=immediate-abort', '-Z', 'unstable-options', '-C', 'linker=lld-link', '-C', 'lto=thin', '-C', 'embed-bitcode=yes', '-C', 'default-linker-libraries=yes', '-C', 'relocation-model=static', '-Z', 'panic_abort_tests', '-Z', 'location-detail=none', '-Z', 'fmt-debug=none', '-C', 'target-feature=+crt-static', '-C', 'link-arg=/OPT:REF', '-C', 'link-arg=/OPT:ICF']"
    Invoke-Step "Cargo build ($Backend)" {
        cargo update
        cargo build --release --no-default-features --features "$featureArg" --config $releaseRustflags
    }

    Write-Host ""
    if (-not (Test-Path 'target\x86_64-pc-windows-msvc\release')) { New-Item -ItemType Directory 'target\x86_64-pc-windows-msvc\release' | Out-Null }
}

# main

if (Test-Path 'lib') {
    $existingLibs = @(Get-ChildItem -Path 'lib\*.lib' -ErrorAction SilentlyContinue)
    if ($existingLibs.Count -gt 0) {
        if ($NoPrompt) {
            $rebuildChoice = if ($ForceRebuild) { 'Y' } else { 'N' }
        } else {
            Write-Host "[PROMPT] Do you want to update and rebuild all dependencies?" -ForegroundColor Yellow
            $rebuildChoice = Read-Host "Enter choice (Y/N) [Default: N]"
        }
        if ($rebuildChoice -ieq 'Y') {
            Remove-Item -Path 'lib\*.lib' -Force -ErrorAction SilentlyContinue
        }
        if (-not $NoPrompt) { Write-Host "" }
    }
}

if ($NoPrompt) {
    $tqChoice = if ($EnableTQ) { 'Y' } else { 'N' }
} else {
    Write-Host "[PROMPT] Compile with target quality feature?" -ForegroundColor Yellow
    $tqChoice = Read-Host "Enter choice (Y/N) [Default: Y]"
    if (-not $tqChoice) { $tqChoice = 'Y' }
}

$enableTQ = $tqChoice -ieq 'Y'

if ($enableTQ) {
    if ($NoPrompt) {
        $vshipBackend = $Backend.ToLower()
    } else {
        Write-Host ""
        Write-Host "[PROMPT] Select Vship backend to compile:" -ForegroundColor Yellow
        Write-Host "  1. CUDA"
        Write-Host "  2. HIP"
        Write-Host "  3. Vulkan"
        $vshipChoice = Read-Host "Enter choice (1-3) [Default: 1]"
        if (-not $vshipChoice) { $vshipChoice = '1' }

        switch ($vshipChoice) {
            '1' { $vshipBackend = 'cuda' }
            '2' { $vshipBackend = 'hip' }
            '3' { $vshipBackend = 'vulkan' }
            default {
                Write-Host "[ERROR] Invalid choice." -ForegroundColor Red
                exit 1
            }
        }
    }
}
else {
    $vshipBackend = 'none'
}

if ($NoPrompt) {
    $avmChoice = if ($EnableAvm) { 'Y' } else { 'N' }
} else {
    Write-Host ""
    Write-Host "[PROMPT] Compile with avm feature?" -ForegroundColor Yellow
    $avmChoice = Read-Host "Enter choice (Y/N) [Default: Y]"
    if (-not $avmChoice) { $avmChoice = 'Y' }
}

$enableAvm = $avmChoice -ieq 'Y'

if ($NoPrompt) {
    $vvencChoice = if ($EnableVvenc) { 'Y' } else { 'N' }
} else {
    Write-Host ""
    Write-Host "[PROMPT] Compile with vvenc feature?" -ForegroundColor Yellow
    $vvencChoice = Read-Host "Enter choice (Y/N) [Default: Y]"
    if (-not $vvencChoice) { $vvencChoice = 'Y' }
}

$enableVvenc = $vvencChoice -ieq 'Y'

if ($NoPrompt) {
    $x265Choice = if ($EnableX265) { 'Y' } else { 'N' }
} else {
    Write-Host ""
    Write-Host "[PROMPT] Compile with x265 feature?" -ForegroundColor Yellow
    $x265Choice = Read-Host "Enter choice (Y/N) [Default: Y]"
    if (-not $x265Choice) { $x265Choice = 'Y' }
}

$enableX265 = $x265Choice -ieq 'Y'

if ($NoPrompt) {
    $x264Choice = if ($EnableX264) { 'Y' } else { 'N' }
} else {
    Write-Host ""
    Write-Host "[PROMPT] Compile with x264 feature?" -ForegroundColor Yellow
    $x264Choice = Read-Host "Enter choice (Y/N) [Default: Y]"
    if (-not $x264Choice) { $x264Choice = 'Y' }
}

$enableX264 = $x264Choice -ieq 'Y'

if ($NoPrompt) {
    $svtChoice = $SvtVariantId.ToString()
} else {
    Write-Host ""
    Write-Host "[PROMPT] Select SVT-AV1 variant to compile:" -ForegroundColor Yellow
Write-Host "  1. svt-av1-hdr       (https://github.com/juliobbv-p/svt-av1-hdr)"
Write-Host "  2. svt-av1-essential (https://github.com/nekotrix/SVT-AV1-Essential)"
Write-Host "  3. 5fish             (https://github.com/5fish/SVT-AV1)"
Write-Host "  4. svt-av1-tritium   (https://github.com/Uranite/svt-av1-tritium)"
Write-Host "  5. svt-av1-skibidi   (https://github.com/Uranite/svt-av1-skibidi)"
Write-Host "  6. mainline          (https://gitlab.com/AOMediaCodec/SVT-AV1)"
$svtChoice = Read-Host "Enter choice (1-6) [Default: 1]"
    if (-not $svtChoice) { $svtChoice = '1' }
}

switch ($svtChoice) {
    '1' { $svtVariant = 'svt-av1-hdr'; $svtRepo = 'https://github.com/juliobbv-p/svt-av1-hdr.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '' }
    '2' { $svtVariant = 'svt-av1-essential'; $svtRepo = 'https://github.com/nekotrix/SVT-AV1-Essential.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '' }
    '3' { $svtVariant = '5fish'; $svtRepo = 'https://github.com/5fish/SVT-AV1.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '-DSVT_LOG_QUIET' }
    '4' { $svtVariant = 'svt-av1-tritium'; $svtRepo = 'https://github.com/Uranite/svt-av1-tritium.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '' }
    '5' { $svtVariant = 'svt-av1-skibidi'; $svtRepo = 'https://github.com/Uranite/svt-av1-skibidi.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '' }
    '6' { $svtVariant = 'mainline'; $svtRepo = 'https://gitlab.com/AOMediaCodec/SVT-AV1.git'; $svtBranch = ''; $svtDir = 'SVT-AV1'; $svtExtraCFlags = '' }
    default {
        Write-Host "[ERROR] Invalid choice." -ForegroundColor Red
        exit 1
    }
}

if ($NoPrompt) {
    $archChoice = $ArchChoiceId.ToString()
} else {
    Write-Host ""
    Write-Host "[PROMPT] Select target architecture for SVT-AV1:" -ForegroundColor Yellow
Write-Host "  1. znver2                   (-march=znver2)"
Write-Host "  2. icelake-server + znver5  (-march=icelake-server -mtune=znver5 -mprefer-vector-width=512)"
Write-Host "  3. native                   (-march=native)"
Write-Host "  4. x86-64-v3                (-march=x86-64-v3)"
Write-Host "  5. skylake                  (-march=skylake)"
Write-Host "  6. haswell                  (-march=haswell)"
$archChoice = Read-Host "Enter choice (1-6) [Default: 1]"
    if (-not $archChoice) { $archChoice = '1' }
}

switch ($archChoice) {
    '1' { $svtArchFlags = "-march=znver2" }
    '2' { $svtArchFlags = "-march=icelake-server -mtune=znver5 -mprefer-vector-width=512" }
    '3' { $svtArchFlags = "-march=native" }
    '4' { $svtArchFlags = "-march=x86-64-v3" }
    '5' { $svtArchFlags = "-march=skylake" }
    '6' { $svtArchFlags = "-march=haswell" }
    default {
        Write-Host "[ERROR] Invalid choice." -ForegroundColor Red
        exit 1
    }
}

$env:CC = 'clang'
$env:CXX = 'clang++'

# Detect NVIDIA GPU generation when CUDA is selected.
# Latest CUDA for Turing and newer
# CUDA 12.9 for Pascal and older
$vsIncludeV143 = $false

if ($vshipBackend -eq 'cuda') {
    $gpuName = Get-CimInstance Win32_VideoController -ErrorAction SilentlyContinue |
    Where-Object { $_.Name -match 'NVIDIA' } |
    Select-Object -ExpandProperty Name -First 1

    if (-not $gpuName) {
        Write-Host "[WARNING] Could not detect NVIDIA GPU. Defaulting to CUDA 12.9 with MSVC v143." -ForegroundColor Yellow
        $vsIncludeV143 = $true
    }

    Write-Host "[INFO] Detected GPU: $gpuName" -ForegroundColor Cyan

    $computeCap = $null
    try {
        $capOutput = & nvidia-smi --query-gpu=compute_cap --format="csv,noheader" 2>$null
        if ($capOutput) {
            $computeCap = [double](($capOutput | Select-Object -First 1).Trim())
        }
    }
    catch { }

    if ($null -ne $computeCap) {
        $isTuringOrNewer = $computeCap -ge 7.5
    }
    else {
        Write-Host "[WARNING] Could not query compute capability via nvidia-smi. Falling back to name parsing." -ForegroundColor Yellow
        $isTuringOrNewer = $gpuName -match 'RTX|GTX 16|TITAN RTX'
    }

    if ($isTuringOrNewer) {
        Write-Host "[INFO] Turing or newer detected. Using latest CUDA version." -ForegroundColor Cyan
        $vsIncludeV143 = $false
    }
    else {
        Write-Host "[INFO] Pascal or older detected. Using CUDA 12.9 with MSVC v143" -ForegroundColor Cyan
        $vsIncludeV143 = $true
    }
}

$depResults = Install-Dependencies -VshipBackend $vshipBackend -VsIncludeV143 $vsIncludeV143
$msysExe = $depResults.MsysExe
$vsPath = $depResults.VsPath

Import-Vcvars -VsPath $vsPath -VsIncludeV143 $vsIncludeV143

if (-not (Test-Path 'lib')) { New-Item -ItemType Directory 'lib' }

if ($vshipBackend -ne 'cuda') {
    Build-Vulkan -VsPath $vsPath
}
if ($enableTQ) {
    Build-Vship -Backend $vshipBackend -VsIncludeV143 $vsIncludeV143 -VsPath $vsPath
}
Build-Dav1d
Build-Opus
if ($vshipBackend -eq 'cuda') {
    Build-NvHeaders -MsysExe $msysExe
}
Build-FFmpeg -VsPath $vsPath -MsysExe $msysExe -Backend $vshipBackend
if ($enableAvm) {
    Build-Avm -NoPrompt $NoPrompt -MsysExe $msysExe
}
if ($enableVvenc) {
    Build-Vvenc -NoPrompt $NoPrompt -MsysExe $msysExe
}
if ($enableVvenc -and $enableTQ) {
    Build-Vvdec -NoPrompt $NoPrompt -MsysExe $msysExe
}
if ($enableX265) {
    Build-X265 -NoPrompt $NoPrompt -MsysExe $msysExe
}
if ($enableX264) {
    Build-X264 -NoPrompt $NoPrompt -MsysExe $msysExe
}
Build-SvtAv1 -Variant $svtVariant -Dir $svtDir -Branch $svtBranch -Repo $svtRepo -ExtraCFlags $svtExtraCFlags -ArchFlags $svtArchFlags -NoPrompt $NoPrompt -MsysExe $msysExe
Build-Xav -Backend $vshipBackend -SvtChoice $svtChoice -enableTQ $enableTQ -enableAvm $enableAvm -enableVvenc $enableVvenc -enableX265 $enableX265 -enableX264 $enableX264

Write-Host "[SUCCESS] Build script finished." -ForegroundColor Green