param(
    [Parameter(Mandatory=$true)]
    [string]$InputFile,

    [Parameter(Mandatory=$true)]
    [string]$OutputFile,

    [Parameter(Mandatory=$false)]
    [switch]$Lossless,   # 是否啟用無損壓縮

    [Parameter(Mandatory=$false)]
    [int]$CRF,           # 用戶可手動(dòng)指定 CRF 值

    [Parameter(Mandatory=$false)]
    [int]$TargetSizeMB   # 目標(biāo)大小 (MB)
)

# 檢查 ffmpeg 是否可用
if (-not (Get-Command ffmpeg -ErrorAction SilentlyContinue)) {
    Write-Host "錯(cuò)誤: 未檢測到 ffmpeg，請先安裝并配置到 PATH。" -ForegroundColor Red
    exit 1
}

# 獲取視頻信息
$info = & ffprobe -v error -select_streams v:0 -show_entries stream=width,height,r_frame_rate,bit_rate,duration -of default=noprint_wrappers=1 "$InputFile"

$width   = ($info | Select-String "width="   | ForEach-Object { $_.ToString().Split("=")[1] })
$height  = ($info | Select-String "height="  | ForEach-Object { $_.ToString().Split("=")[1] })
$fpsRaw  = ($info | Select-String "r_frame_rate=" | ForEach-Object { $_.ToString().Split("=")[1] })
$bitrate = ($info | Select-String "bit_rate=" | ForEach-Object { $_.ToString().Split("=")[1] })
$duration = ($info | Select-String "duration=" | ForEach-Object { $_.ToString().Split("=")[1] })

# 計(jì)算幀率
$fpsParts = $fpsRaw -split "/"
if ($fpsParts.Count -eq 2) {
    $fps = [math]::Round([double]$fpsParts[0] / [double]$fpsParts[1], 2)
} else {
    $fps = [double]$fpsRaw
}

Write-Host "檢測到視頻: ${width}x${height}, ${fps}fps, 碼率=${bitrate}bps, 時(shí)長=${duration}s"

# 判斷壓縮模式
if ($Lossless) {
    # 無損壓縮
    $codec = "libx264"
    $preset = "slow"
    $extraParams = "-crf 0"
    $audioCodec = "copy"
    Write-Host "選擇無損壓縮: $codec $preset $extraParams, 音頻=$audioCodec"
} elseif ($TargetSizeMB) {
    # 目標(biāo)大小模式
    $targetBitrate = [math]::Round(($TargetSizeMB * 8 * 1024) / [double]$duration)
    Write-Host "目標(biāo)大小模式: 目標(biāo)=$TargetSizeMB MB, 計(jì)算碼率=$targetBitrate kbps"

    $codec = "libx264"
    $preset = "slow"
    $extraParams = "-b:v ${targetBitrate}k"
    $audioCodec = "aac -b:a 128k"
    Write-Host "選擇目標(biāo)大小壓縮: $codec $preset $extraParams, 音頻=$audioCodec"
} else {
    # 有損壓縮，自動(dòng)推薦 CRF
    if ($CRF) {
        $chosenCRF = $CRF
        Write-Host "用戶指定 CRF=$chosenCRF"
    } else {
        if ([int]$width -ge 1920 -or [int]$bitrate -ge 4000000) {
            $chosenCRF = 23
        } elseif ([int]$width -ge 1280 -or [int]$bitrate -ge 2000000) {
            $chosenCRF = 26
        } else {
            $chosenCRF = 28
        }
        Write-Host "自動(dòng)推薦 CRF=$chosenCRF"
    }

    $codec = "libx264"
    $preset = "slow"
    $extraParams = "-crf $chosenCRF"
    $audioCodec = "aac -b:a 128k"
    Write-Host "選擇有損壓縮: $codec $preset $extraParams, 音頻=$audioCodec"
}

# 執(zhí)行壓縮
$ffmpegCmd = "ffmpeg -i `"$InputFile`" -c:v $codec -preset $preset $extraParams -c:a $audioCodec `"$OutputFile`""
Write-Host "執(zhí)行命令: $ffmpegCmd"
& cmd /c $ffmpegCmd

# 顯示文件大小對比
$origSize = (Get-Item $InputFile).Length
$newSize  = (Get-Item $OutputFile).Length

Write-Host "原始大小: $([math]::Round($origSize/1MB,2)) MB"
Write-Host "壓縮后大小: $([math]::Round($newSize/1MB,2)) MB" -ForegroundColor Green

./auto_compress.ps1 -InputFile "input.mp4" -OutputFile "output.mp4"

./auto_compress.ps1 -InputFile "input.mp4" -OutputFile "output.mp4" -CRF 24

./auto_compress.ps1 -InputFile "input.mp4" -OutputFile "output.mp4" -Lossless

./auto_compress.ps1 -InputFile "input.mp4" -OutputFile "output.mp4" -TargetSizeMB 100