mirror of
https://github.com/babalae/better-genshin-impact.git
synced 2026-03-20 08:29:50 +08:00
500 lines
18 KiB
C#
500 lines
18 KiB
C#
using BetterGenshinImpact.Core.Config;
|
||
using BetterGenshinImpact.GameTask.Common;
|
||
|
||
using BetterGenshinImpact.Helpers;
|
||
using BetterGenshinImpact.View;
|
||
using Fischless.GameCapture;
|
||
using Microsoft.Extensions.Logging;
|
||
using OpenCvSharp;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Diagnostics;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using System.Threading;
|
||
using BetterGenshinImpact.GameTask.GameLoading;
|
||
using Fischless.GameCapture.Graphics;
|
||
using BetterGenshinImpact.Service;
|
||
using Vanara.PInvoke;
|
||
|
||
namespace BetterGenshinImpact.GameTask
|
||
{
|
||
public class TaskTriggerDispatcher : IDisposable
|
||
{
|
||
private readonly ILogger<TaskTriggerDispatcher> _logger = App.GetLogger<TaskTriggerDispatcher>();
|
||
|
||
private static TaskTriggerDispatcher? _instance;
|
||
|
||
private readonly System.Timers.Timer _timer = new();
|
||
private List<ITaskTrigger>? _triggers;
|
||
|
||
public IGameCapture? GameCapture { get; private set; }
|
||
|
||
private static readonly object _locker = new();
|
||
private int _frameIndex = 0;
|
||
|
||
private RECT _gameRect = RECT.Empty;
|
||
private bool _prevGameActive;
|
||
|
||
private DateTime _prevManualGc = DateTime.MinValue;
|
||
|
||
private static readonly object _triggerListLocker = new();
|
||
|
||
private User32.HWINEVENTHOOK _winEventHookMoveSize;
|
||
private User32.HWINEVENTHOOK _winEventHookLocation;
|
||
private User32.WinEventProc _winEventProc;
|
||
private const uint EVENT_SYSTEM_MOVESIZESTART = 0x000A;
|
||
private const uint EVENT_SYSTEM_MOVESIZEEND = 0x000B;
|
||
private const uint EVENT_OBJECT_LOCATIONCHANGE = 0x800B;
|
||
private const uint WINEVENT_SKIPOWNTHREAD = 0x0001;
|
||
private const uint WINEVENT_SKIPOWNPROCESS = 0x0002;
|
||
|
||
public event EventHandler? UiTaskStopTickEvent;
|
||
|
||
public event EventHandler? UiTaskStartTickEvent;
|
||
|
||
public TaskTriggerDispatcher()
|
||
{
|
||
_instance = this;
|
||
_timer.Elapsed += Tick;
|
||
//_timer.Tick += Tick;
|
||
}
|
||
|
||
public static TaskTriggerDispatcher Instance()
|
||
{
|
||
if (_instance == null)
|
||
{
|
||
throw new Exception("请先在启动页启动BetterGI，如果已经启动请重启");
|
||
}
|
||
|
||
return _instance;
|
||
}
|
||
|
||
public static IGameCapture GlobalGameCapture
|
||
{
|
||
get
|
||
{
|
||
_instance = Instance();
|
||
|
||
if (_instance.GameCapture == null)
|
||
{
|
||
throw new Exception("截图器未初始化!");
|
||
}
|
||
|
||
return _instance.GameCapture;
|
||
}
|
||
}
|
||
|
||
public void ClearTriggers()
|
||
{
|
||
lock (_triggerListLocker)
|
||
{
|
||
GameTaskManager.ClearTriggers();
|
||
_triggers?.Clear();
|
||
}
|
||
}
|
||
|
||
public void SetTriggers(List<ITaskTrigger> list)
|
||
{
|
||
lock (_triggerListLocker)
|
||
{
|
||
_triggers = list;
|
||
}
|
||
}
|
||
|
||
public bool AddTrigger(string name, object? externalConfig)
|
||
{
|
||
lock (_triggerListLocker)
|
||
{
|
||
if (GameTaskManager.AddTrigger(name, externalConfig))
|
||
{
|
||
SetTriggers(GameTaskManager.ConvertToTriggerList(true));
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
}
|
||
|
||
public void Start(IntPtr hWnd, CaptureModes mode, int interval = 50)
|
||
{
|
||
// 初始化截图器
|
||
GameCapture = GameCaptureFactory.Create(mode);
|
||
// 激活窗口 保证后面能够正常获取窗口信息
|
||
SystemControl.ActivateWindow(hWnd);
|
||
|
||
// 初始化任务上下文(一定要在初始化触发器前完成)
|
||
TaskContext.Instance().Init(hWnd);
|
||
|
||
// 初始化触发器(一定要在任务上下文初始化完毕后使用)
|
||
_triggers = GameTaskManager.LoadInitialTriggers();
|
||
GameLoadingTrigger.GlobalEnabled = TaskContext.Instance().Config.GenshinStartConfig.AutoEnterGameEnabled;
|
||
|
||
// if (GraphicsCapture.IsHdrEnabled(hWnd))
|
||
// {
|
||
// _logger.LogError("游戏窗口在HDR模式下无法获取正常颜色的截图，请关闭HDR模式！");
|
||
// }
|
||
|
||
// 启动截图
|
||
GameCapture.Start(hWnd,
|
||
new Dictionary<string, object>()
|
||
{
|
||
{ "autoFixWin11BitBlt", OsVersionHelper.IsWindows11_OrGreater && TaskContext.Instance().Config.AutoFixWin11BitBlt }
|
||
}
|
||
);
|
||
|
||
// 使用 SetWinEventHook 监听窗口移动和大小变化事件
|
||
_winEventProc = WinEventCallback;
|
||
var flags = (User32.WINEVENT)(WINEVENT_SKIPOWNPROCESS | WINEVENT_SKIPOWNTHREAD);
|
||
_winEventHookMoveSize = User32.SetWinEventHook(EVENT_SYSTEM_MOVESIZESTART, EVENT_SYSTEM_MOVESIZEEND, default, _winEventProc, 0, 0, flags);
|
||
_winEventHookLocation = User32.SetWinEventHook(EVENT_OBJECT_LOCATIONCHANGE, EVENT_OBJECT_LOCATIONCHANGE, default, _winEventProc, 0, 0, flags);
|
||
|
||
// 启动定时器
|
||
_frameIndex = 0;
|
||
_timer.Interval = interval;
|
||
if (!_timer.Enabled)
|
||
{
|
||
_timer.Start();
|
||
}
|
||
}
|
||
|
||
public void Stop()
|
||
{
|
||
_timer.Stop();
|
||
GameCapture?.Stop();
|
||
_gameRect = RECT.Empty;
|
||
_prevGameActive = false;
|
||
PictureInPictureService.Hide(resetManual: true);
|
||
if (_winEventHookMoveSize != default)
|
||
{
|
||
User32.UnhookWinEvent(_winEventHookMoveSize);
|
||
_winEventHookMoveSize = default;
|
||
}
|
||
if (_winEventHookLocation != default)
|
||
{
|
||
User32.UnhookWinEvent(_winEventHookLocation);
|
||
_winEventHookLocation = default;
|
||
}
|
||
}
|
||
|
||
public void StartTimer()
|
||
{
|
||
if (!_timer.Enabled)
|
||
{
|
||
_timer.Start();
|
||
}
|
||
}
|
||
|
||
public void StopTimer()
|
||
{
|
||
if (_timer.Enabled)
|
||
{
|
||
_timer.Stop();
|
||
}
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
Stop();
|
||
}
|
||
|
||
public void Tick(object? sender, EventArgs e)
|
||
{
|
||
var hasLock = false;
|
||
try
|
||
{
|
||
Monitor.TryEnter(_locker, ref hasLock);
|
||
if (!hasLock)
|
||
{
|
||
// 正在执行时跳过
|
||
return;
|
||
}
|
||
|
||
// 检查截图器是否初始化
|
||
var maskWindow = MaskWindow.Instance();
|
||
if (GameCapture == null || !GameCapture.IsCapturing)
|
||
{
|
||
if (!TaskContext.Instance().SystemInfo.GameProcess.HasExited)
|
||
{
|
||
_logger.LogError("截图器未初始化!");
|
||
}
|
||
else
|
||
{
|
||
_logger.LogInformation("游戏已退出，BetterGI 自动停止截图器");
|
||
}
|
||
|
||
PictureInPictureService.Hide(resetManual: true);
|
||
UiTaskStopTickEvent?.Invoke(sender, e);
|
||
maskWindow.Invoke(maskWindow.HideSelf);
|
||
return;
|
||
}
|
||
|
||
// 检查游戏是否在前台
|
||
var hasBackgroundTriggerToRun = false;
|
||
var autoSkipConfig = TaskContext.Instance().Config.AutoSkipConfig;
|
||
var shouldShowPictureInPicture = autoSkipConfig.Enabled
|
||
&& autoSkipConfig.PictureInPictureEnabled
|
||
&& !PictureInPictureService.IsManuallyClosed
|
||
&& TaskControl.TaskSemaphore.CurrentCount == 1; // 没有任务持有锁（也就是没有任务正在运行）
|
||
var active = SystemControl.IsGenshinImpactActive();
|
||
if (!active)
|
||
{
|
||
// 检查游戏是否已结束
|
||
if (TaskContext.Instance().SystemInfo.GameProcess.HasExited)
|
||
{
|
||
_logger.LogInformation("游戏已退出，BetterGI 自动停止截图器");
|
||
UiTaskStopTickEvent?.Invoke(sender, e);
|
||
return;
|
||
}
|
||
|
||
if (_prevGameActive)
|
||
{
|
||
Debug.WriteLine("游戏窗口不在前台, 不再进行截屏");
|
||
}
|
||
|
||
if (!TaskContext.Instance().Config.MaskWindowConfig.UseSubform)
|
||
{
|
||
var pName = SystemControl.GetActiveProcessName();
|
||
if (pName != "BetterGI" && pName != "YuanShen" && pName != "GenshinImpact" && pName != "Genshin Impact Cloud Game")
|
||
{
|
||
maskWindow.Invoke(() => { maskWindow.HideSelf(); });
|
||
}
|
||
}
|
||
|
||
_prevGameActive = active;
|
||
|
||
if (_triggers != null)
|
||
{
|
||
lock (_triggerListLocker)
|
||
{
|
||
var exclusive = _triggers.FirstOrDefault(t => t is { IsEnabled: true, IsExclusive: true });
|
||
if (exclusive != null)
|
||
{
|
||
hasBackgroundTriggerToRun = exclusive.IsBackgroundRunning;
|
||
}
|
||
else
|
||
{
|
||
hasBackgroundTriggerToRun = _triggers.Any(t => t is { IsEnabled: true, IsBackgroundRunning: true });
|
||
}
|
||
}
|
||
}
|
||
if (!hasBackgroundTriggerToRun && shouldShowPictureInPicture)
|
||
{
|
||
hasBackgroundTriggerToRun = true;
|
||
}
|
||
|
||
if (!hasBackgroundTriggerToRun)
|
||
{
|
||
// 没有后台运行的触发器，这次不再进行截图
|
||
PictureInPictureService.Hide();
|
||
return;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
PictureInPictureService.Hide(resetManual: true);
|
||
if (!TaskContext.Instance().Config.MaskWindowConfig.UseSubform)
|
||
{
|
||
// if (!_prevGameActive)
|
||
// {
|
||
maskWindow.Invoke(() =>
|
||
{
|
||
if (maskWindow.IsExist())
|
||
{
|
||
maskWindow.Show();
|
||
if (!_prevGameActive)
|
||
{
|
||
maskWindow.BringToTop();
|
||
}
|
||
}
|
||
});
|
||
// }
|
||
}
|
||
|
||
_prevGameActive = active;
|
||
// // 移动游戏窗口的时候同步遮罩窗口的位置,此时不进行捕获
|
||
if (SyncMaskWindowPosition())
|
||
{
|
||
return;
|
||
}
|
||
}
|
||
|
||
if (_triggers == null || !_triggers.Exists(t => t.IsEnabled))
|
||
{
|
||
// Debug.WriteLine("没有可用的触发器且不处于仅截屏状态, 不再进行截屏");
|
||
return;
|
||
}
|
||
|
||
// 帧序号自增 1分钟后归零(MaxFrameIndexSecond)
|
||
_frameIndex = (_frameIndex + 1) % (int)(CaptureContent.MaxFrameIndexSecond * 1000d / _timer.Interval);
|
||
|
||
var speedTimer = new SpeedTimer();
|
||
// 捕获游戏画面
|
||
var bitmap = GameCapture.Capture();
|
||
speedTimer.Record("截图");
|
||
|
||
if (bitmap == null)
|
||
{
|
||
_logger.LogWarning("截图失败!");
|
||
return;
|
||
}
|
||
|
||
if (shouldShowPictureInPicture && !active)
|
||
{
|
||
PictureInPictureService.Update(bitmap);
|
||
}
|
||
else
|
||
{
|
||
PictureInPictureService.Hide();
|
||
}
|
||
|
||
// 循环执行所有触发器 有独占状态的触发器的时候只执行独占触发器
|
||
var content = new CaptureContent(bitmap, _frameIndex, _timer.Interval);
|
||
|
||
lock (_triggerListLocker)
|
||
{
|
||
var exclusiveTrigger = _triggers!.FirstOrDefault(t => t is { IsEnabled: true, IsExclusive: true });
|
||
if (exclusiveTrigger != null)
|
||
{
|
||
exclusiveTrigger.OnCapture(content);
|
||
speedTimer.Record(exclusiveTrigger.Name);
|
||
}
|
||
else
|
||
{
|
||
var runningTriggers = _triggers!.Where(t => t.IsEnabled);
|
||
if (hasBackgroundTriggerToRun)
|
||
{
|
||
runningTriggers = runningTriggers.Where(t => t.IsBackgroundRunning);
|
||
}
|
||
|
||
foreach (var trigger in runningTriggers)
|
||
{
|
||
trigger.OnCapture(content);
|
||
speedTimer.Record(trigger.Name);
|
||
}
|
||
}
|
||
}
|
||
|
||
speedTimer.DebugPrint();
|
||
content.Dispose();
|
||
}
|
||
finally
|
||
{
|
||
if ((DateTime.Now - _prevManualGc).TotalSeconds > 2)
|
||
{
|
||
GC.Collect();
|
||
_prevManualGc = DateTime.Now;
|
||
}
|
||
|
||
if (hasLock)
|
||
{
|
||
Monitor.Exit(_locker);
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// / 移动游戏窗口的时候同步遮罩窗口的位置
|
||
/// </summary>
|
||
/// <returns></returns>
|
||
private bool SyncMaskWindowPosition()
|
||
{
|
||
var hWnd = TaskContext.Instance().GameHandle;
|
||
var currentRect = SystemControl.GetCaptureRect(hWnd);
|
||
if (_gameRect == RECT.Empty)
|
||
{
|
||
_gameRect = new RECT(currentRect);
|
||
}
|
||
else if (_gameRect != currentRect)
|
||
{
|
||
// 后面大概可以取消掉这个判断，支持随意移动变化窗口 —— 不支持 需要考虑的问题太多了
|
||
if ((_gameRect.Width != currentRect.Width || _gameRect.Height != currentRect.Height)
|
||
&& !SizeIsZero(_gameRect) && !SizeIsZero(currentRect))
|
||
{
|
||
_logger.LogError("► 游戏窗口大小发生变化 {W}x{H}->{CW}x{CH}, 自动重启截图器中...", _gameRect.Width, _gameRect.Height, currentRect.Width, currentRect.Height);
|
||
UiTaskStopTickEvent?.Invoke(null, EventArgs.Empty);
|
||
UiTaskStartTickEvent?.Invoke(null, EventArgs.Empty);
|
||
_logger.LogInformation("► 游戏窗口大小发生变化，截图器重启完成！");
|
||
}
|
||
|
||
_gameRect = new RECT(currentRect);
|
||
TaskContext.Instance().SystemInfo.CaptureAreaRect = currentRect;
|
||
MaskWindow.Instance().RefreshPosition();
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
private bool SizeIsZero(RECT rect)
|
||
{
|
||
return rect.Width == 0 || rect.Height == 0;
|
||
}
|
||
|
||
private void WinEventCallback(User32.HWINEVENTHOOK hWinEventHook, uint @event, HWND hwnd, int idObject, int idChild, uint dwEventThread, uint dwmsEventTime)
|
||
{
|
||
var target = TaskContext.Instance().GameHandle;
|
||
if (target == IntPtr.Zero)
|
||
{
|
||
return;
|
||
}
|
||
if (idObject != 0)
|
||
{
|
||
return;
|
||
}
|
||
var hwndPtr = hwnd.DangerousGetHandle();
|
||
if (hwndPtr == target)
|
||
{
|
||
SyncMaskWindowPosition();
|
||
}
|
||
}
|
||
|
||
public void TakeScreenshot()
|
||
{
|
||
try
|
||
{
|
||
var path = Global.Absolute($@"log\screenshot\");
|
||
if (!Directory.Exists(path))
|
||
{
|
||
Directory.CreateDirectory(path);
|
||
}
|
||
|
||
Mat mat;
|
||
try
|
||
{
|
||
mat = TaskControl.CaptureGameImage(GameCapture);
|
||
}
|
||
catch (Exception)
|
||
{
|
||
_logger.LogInformation("截图失败，未获取到图像");
|
||
return;
|
||
}
|
||
var name = $@"{DateTime.Now:yyyyMMddHHmmssffff}.png";
|
||
var savePath = Global.Absolute($@"log\screenshot\{name}");
|
||
if (TaskContext.Instance().Config.CommonConfig.ScreenshotUidCoverEnabled)
|
||
{
|
||
var assetScale = TaskContext.Instance().SystemInfo.ScaleTo1080PRatio;
|
||
var rect = new Rect((int)(mat.Width - MaskWindowConfig.UidCoverRightBottomRect.X * assetScale),
|
||
(int)(mat.Height - MaskWindowConfig.UidCoverRightBottomRect.Y * assetScale),
|
||
(int)(MaskWindowConfig.UidCoverRightBottomRect.Width * assetScale),
|
||
(int)(MaskWindowConfig.UidCoverRightBottomRect.Height * assetScale));
|
||
mat.Rectangle(rect, Scalar.White, -1);
|
||
Cv2.ImWrite(savePath, mat);
|
||
}
|
||
else
|
||
{
|
||
Cv2.ImWrite(savePath, mat);
|
||
}
|
||
|
||
mat.Dispose();
|
||
|
||
_logger.LogInformation("截图已保存: {Name}", name);
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogError("截图保存失败: {Message}", e.Message);
|
||
_logger.LogDebug("截图保存失败: {StackTrace}", e.StackTrace);
|
||
}
|
||
}
|
||
}
|
||
}
|