# 适用于Windows系统
# @时间    : 2024-06-28
# @作者    : 三巧(https://blog.csdn.net/qq_39124701)
# @文件名  : 设置小蓝熊的CPU亲和性、CPU优先级再设置法环的CPU亲和性.ps1
# @使用方法: 打开记事本，将所有代码复制到记事本中，保存文件时候修改文件后缀为".ps1"。
#            右键该文件，点击"使用 PowerShell 运行"即可。



# 定义设置CPU亲和性的函数
function Set-ProcessCpuAffinity {
    param(
        [Diagnostics.Process]$Process,
        [int[]]$Affinity
    )
    
    # 获取Process的句柄
    $handle = $Process.Handle
    $processAffinityMask = 0
    $systemAffinityMask = 0

    # 循环设置CPU亲和性掩码
    for ($i = 0; $i -lt $Affinity.Length; $i++) {
        $cpu = $Affinity[$i]
        $processAffinityMask = $processAffinityMask -bor [int](1 -shl $cpu)
        $systemAffinityMask = $systemAffinityMask -bor [int](1 -shl $cpu)
    }

    # 设置进程亲和性
    $success = [Kernel32]::SetProcessAffinityMask($handle, $processAffinityMask)
    if (-not $success) {
        Write-Error "Unable to set process affinity."
    }

    # 设置系统亲和性
    $success = [Kernel32]::SetProcessAffinityMask($handle, $systemAffinityMask)
    if (-not $success) {
        Write-Error "Unable to set system affinity."
    }
}

# 导入Kernel32.dll
Add-Type -TypeDefinition @"
    using System;
    using System.Runtime.InteropServices;

    public class Kernel32 {
        [DllImport("kernel32.dll", SetLastError = true)]
        public static extern bool SetProcessAffinityMask(IntPtr handle, IntPtr affinityMask);
    }
"@

# 使用WMI获取逻辑处理器数  
$logicalProcessors = (Get-WmiObject Win32_ComputerSystem).NumberOfLogicalProcessors  



# 定义要设置CPU亲和性的CPU编号数组（EAC）
# $cpuAffinity = 1, 2  # 例如，只使用CPU 1和CPU 2
$cpuAffinity = 0  # 只使用CPU0

# 获取已经运行的EasyAntiCheat_EOS.exe进程对象
$eac = Get-Process -Name "EasyAntiCheat_EOS" -ErrorAction SilentlyContinue

if ($null -ne $eac) {
    # 调用函数设置CPU亲和性
    Set-ProcessCpuAffinity -Process $eac -Affinity $cpuAffinity


    # 定义要设置的优先级（Idle：空闲）
    $priorityClass = [System.Diagnostics.ProcessPriorityClass]::Idle
    # 设置进程的优先级
    $eac.PriorityClass = $priorityClass
    # 显示设置结果
    if ($eac.PriorityClass -eq $priorityClass) {
        Write-Host "Priority class set to $([enum]::GetName([System.Diagnostics.ProcessPriorityClass], $priorityClass))"
    } else {
        Write-Error "Failed to set priority class."
    }
} else {
    Write-Host "EasyAntiCheat_EOS.exe is not running."
}

# 定义要设置CPU亲和性的CPU编号数组（法环）
# $cpuAffinity = 1,2,3,4,5,6,7,8,9,10,11  #（写死数字的方法只适用于12核心的处理器）
$cpuAffinity = 1..($logicalProcessors - 1) # 从1开始，也就是除了 CPU0 都使用

# 获取已经运行的eldenring.exe进程对象
$er = Get-Process -Name "eldenring" -ErrorAction SilentlyContinue

if ($null -ne $er) {
    # 调用函数设置CPU亲和性
    Set-ProcessCpuAffinity -Process $er -Affinity $cpuAffinity
} else {
    Write-Host "eldenring.exe is not running."
}


Write-Host ""
Write-Host "Script execution completed successfully!"
Read-Host -Prompt "Press Enter to quit"