mirror of
https://github.com/babalae/better-genshin-impact.git
synced 2026-04-04 11:15:18 +08:00
485 lines
17 KiB
C#
485 lines
17 KiB
C#
using BetterGenshinImpact.Core.Config;
|
||
using BetterGenshinImpact.GameTask.AutoDomain;
|
||
using BetterGenshinImpact.GameTask.AutoFight;
|
||
using BetterGenshinImpact.GameTask.AutoGeniusInvokation;
|
||
using BetterGenshinImpact.GameTask.AutoWood;
|
||
using BetterGenshinImpact.GameTask.Model;
|
||
using BetterGenshinImpact.GameTask.Model.Enum;
|
||
using BetterGenshinImpact.Genshin.Settings;
|
||
using BetterGenshinImpact.Helpers;
|
||
using BetterGenshinImpact.View;
|
||
using Fischless.GameCapture;
|
||
using Microsoft.Extensions.Logging;
|
||
using OpenCvSharp;
|
||
using OpenCvSharp.Extensions;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Diagnostics;
|
||
using System.Drawing;
|
||
using System.Drawing.Imaging;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
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;
|
||
|
||
/// <summary>
|
||
/// 捕获结果队列
|
||
/// </summary>
|
||
private Bitmap _bitmap = new(10, 10);
|
||
|
||
/// <summary>
|
||
/// 仅捕获模式
|
||
/// </summary>
|
||
private DispatcherCaptureModeEnum _dispatcherCacheCaptureMode = DispatcherCaptureModeEnum.OnlyTrigger;
|
||
|
||
private static readonly object _bitmapLocker = new();
|
||
|
||
public event EventHandler UiTaskStopTickEvent;
|
||
|
||
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 Start(IntPtr hWnd, CaptureModes mode, int interval = 50)
|
||
{
|
||
// 初始化截图器
|
||
GameCapture = GameCaptureFactory.Create(mode);
|
||
// 激活窗口 保证后面能够正常获取窗口信息
|
||
SystemControl.ActivateWindow(hWnd);
|
||
|
||
// 初始化任务上下文(一定要在初始化触发器前完成)
|
||
TaskContext.Instance().Init(hWnd);
|
||
PrintSystemInfo();
|
||
|
||
// 初始化触发器(一定要在任务上下文初始化完毕后使用)
|
||
_triggers = GameTaskManager.LoadTriggers();
|
||
|
||
// 启动截图
|
||
GameCapture.Start(hWnd,
|
||
new Dictionary<string, object>()
|
||
{
|
||
{ "useBitmapCache", TaskContext.Instance().Config.WgcUseBitmapCache }
|
||
}
|
||
);
|
||
|
||
// 捕获模式初始化配置
|
||
if (TaskContext.Instance().Config.CommonConfig.ScreenshotEnabled)
|
||
{
|
||
_dispatcherCacheCaptureMode = DispatcherCaptureModeEnum.CacheCaptureWithTrigger;
|
||
}
|
||
|
||
// 读取游戏注册表配置
|
||
ReadGameSettings();
|
||
|
||
// 启动定时器
|
||
_frameIndex = 0;
|
||
_timer.Interval = interval;
|
||
if (!_timer.Enabled)
|
||
{
|
||
_timer.Start();
|
||
}
|
||
}
|
||
|
||
private void ReadGameSettings()
|
||
{
|
||
try
|
||
{
|
||
SettingsContainer settings = new();
|
||
TaskContext.Instance().GameSettings = settings;
|
||
var lang = settings.Language?.TextLang;
|
||
if (lang != null && lang != TextLanguage.SimplifiedChinese)
|
||
{
|
||
_logger.LogWarning("当前游戏语言{Lang}不是简体中文，部分功能可能无法正常使用。The game language is not Simplified Chinese, some functions may not work properly", lang);
|
||
}
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogWarning("游戏注册表配置信息读取失败：" + e.Source + "\r\n--" + Environment.NewLine + e.StackTrace + "\r\n---" + Environment.NewLine + e.Message);
|
||
}
|
||
}
|
||
|
||
private void PrintSystemInfo()
|
||
{
|
||
var systemInfo = TaskContext.Instance().SystemInfo;
|
||
var width = systemInfo.GameScreenSize.Width;
|
||
var height = systemInfo.GameScreenSize.Height;
|
||
var dpiScale = TaskContext.Instance().DpiScale;
|
||
_logger.LogInformation("当前游戏分辨率{Width}x{Height}，素材缩放比率{Scale}，DPI缩放{Dpi}",
|
||
width, height, systemInfo.AssetScale.ToString("F"), dpiScale);
|
||
|
||
if (width * 9 != height * 16)
|
||
{
|
||
_logger.LogWarning("当前游戏分辨率不是16:9，部分功能可能无法正常使用");
|
||
}
|
||
}
|
||
|
||
public void Stop()
|
||
{
|
||
_timer.Stop();
|
||
GameCapture?.Stop();
|
||
_gameRect = RECT.Empty;
|
||
_prevGameActive = false;
|
||
}
|
||
|
||
public void StartTimer()
|
||
{
|
||
if (!_timer.Enabled)
|
||
{
|
||
_timer.Start();
|
||
}
|
||
}
|
||
|
||
public void StopTimer()
|
||
{
|
||
if (_timer.Enabled)
|
||
{
|
||
_timer.Stop();
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 启动独立任务
|
||
/// </summary>
|
||
public void StartIndependentTask(IndependentTaskEnum taskType, BaseTaskParam param)
|
||
{
|
||
if (!_timer.Enabled)
|
||
{
|
||
throw new Exception("请先在启动页启动BetterGI，如果已经启动请重启");
|
||
}
|
||
|
||
var maskWindow = MaskWindow.Instance();
|
||
maskWindow.LogBox.IsHitTestVisible = false;
|
||
maskWindow.Invoke(() => { maskWindow.Show(); });
|
||
if (taskType == IndependentTaskEnum.AutoGeniusInvokation)
|
||
{
|
||
AutoGeniusInvokationTask.Start((GeniusInvokationTaskParam)param);
|
||
}
|
||
else if (taskType == IndependentTaskEnum.AutoWood)
|
||
{
|
||
Task.Run(() => { new AutoWoodTask().Start((WoodTaskParam)param); });
|
||
}
|
||
else if (taskType == IndependentTaskEnum.AutoFight)
|
||
{
|
||
Task.Run(() => { new AutoFightTask((AutoFightParam)param).Start(); });
|
||
}
|
||
else if (taskType == IndependentTaskEnum.AutoDomain)
|
||
{
|
||
Task.Run(() => { new AutoDomainTask((AutoDomainParam)param).Start(); });
|
||
}
|
||
}
|
||
|
||
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 自动停止截图器");
|
||
}
|
||
UiTaskStopTickEvent.Invoke(sender, e);
|
||
maskWindow.Invoke(maskWindow.Hide);
|
||
return;
|
||
}
|
||
|
||
// 检查游戏是否在前台
|
||
var active = SystemControl.IsGenshinImpactActive();
|
||
if (!active)
|
||
{
|
||
// 检查游戏是否已结束
|
||
if (TaskContext.Instance().SystemInfo.GameProcess.HasExited)
|
||
{
|
||
_logger.LogInformation("游戏已退出，BetterGI 自动停止截图器");
|
||
UiTaskStopTickEvent.Invoke(sender, e);
|
||
return;
|
||
}
|
||
|
||
if (_prevGameActive)
|
||
{
|
||
Debug.WriteLine("游戏窗口不在前台, 不再进行截屏");
|
||
}
|
||
|
||
var pName = SystemControl.GetActiveProcessName();
|
||
if (pName != "BetterGI" && pName != "YuanShen" && pName != "GenshinImpact" && pName != "Genshin Impact Cloud Game")
|
||
{
|
||
maskWindow.Invoke(() => { maskWindow.Hide(); });
|
||
}
|
||
|
||
_prevGameActive = active;
|
||
return;
|
||
}
|
||
else
|
||
{
|
||
if (!_prevGameActive)
|
||
{
|
||
maskWindow.Invoke(() =>
|
||
{
|
||
if (!maskWindow.IsClosed)
|
||
{
|
||
maskWindow.Show();
|
||
}
|
||
});
|
||
}
|
||
|
||
_prevGameActive = active;
|
||
// 移动游戏窗口的时候同步遮罩窗口的位置,此时不进行捕获
|
||
if (SyncMaskWindowPosition())
|
||
{
|
||
return;
|
||
}
|
||
}
|
||
|
||
// 帧序号自增 1分钟后归零(MaxFrameIndexSecond)
|
||
_frameIndex = (_frameIndex + 1) % (int)(CaptureContent.MaxFrameIndexSecond * 1000d / _timer.Interval);
|
||
|
||
if (_dispatcherCacheCaptureMode == DispatcherCaptureModeEnum.OnlyTrigger
|
||
&& (_triggers == null || !_triggers.Exists(t => t.IsEnabled)))
|
||
{
|
||
// Debug.WriteLine("没有可用的触发器且不处于仅截屏状态, 不再进行截屏");
|
||
return;
|
||
}
|
||
|
||
var speedTimer = new SpeedTimer();
|
||
// 捕获游戏画面
|
||
var bitmap = GameCapture.Capture();
|
||
speedTimer.Record("截图");
|
||
|
||
if (bitmap == null)
|
||
{
|
||
_logger.LogWarning("截图失败!");
|
||
return;
|
||
}
|
||
|
||
if (IsOnlyCacheCapture(bitmap))
|
||
{
|
||
return;
|
||
}
|
||
|
||
// 循环执行所有触发器 有独占状态的触发器的时候只执行独占触发器
|
||
var content = new CaptureContent(bitmap, _frameIndex, _timer.Interval, this);
|
||
var exclusiveTrigger = _triggers.FirstOrDefault(t => t is { IsEnabled: true, IsExclusive: true });
|
||
if (exclusiveTrigger != null)
|
||
{
|
||
exclusiveTrigger.OnCapture(content);
|
||
speedTimer.Record(exclusiveTrigger.Name);
|
||
}
|
||
else
|
||
{
|
||
foreach (var trigger in _triggers.Where(trigger => trigger.IsEnabled))
|
||
{
|
||
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);
|
||
}
|
||
|
||
_gameRect = new RECT(currentRect);
|
||
double scale = TaskContext.Instance().DpiScale;
|
||
TaskContext.Instance().SystemInfo.CaptureAreaRect = currentRect;
|
||
MaskWindow.Instance().RefreshPosition(currentRect, scale);
|
||
return true;
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
private bool SizeIsZero(RECT rect)
|
||
{
|
||
return rect.Width == 0 || rect.Height == 0;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 是否仅缓存截图
|
||
/// </summary>
|
||
/// <param name="bitmap"></param>
|
||
/// <returns></returns>
|
||
private bool IsOnlyCacheCapture(Bitmap bitmap)
|
||
{
|
||
lock (_bitmapLocker)
|
||
{
|
||
if (_dispatcherCacheCaptureMode is DispatcherCaptureModeEnum.OnlyCacheCapture or DispatcherCaptureModeEnum.CacheCaptureWithTrigger)
|
||
{
|
||
_bitmap = new Bitmap(bitmap);
|
||
if (_dispatcherCacheCaptureMode == DispatcherCaptureModeEnum.OnlyCacheCapture)
|
||
{
|
||
return true;
|
||
}
|
||
}
|
||
|
||
return false;
|
||
}
|
||
}
|
||
|
||
public void SetCacheCaptureMode(DispatcherCaptureModeEnum mode)
|
||
{
|
||
_dispatcherCacheCaptureMode = mode;
|
||
}
|
||
|
||
public DispatcherCaptureModeEnum GetCacheCaptureMode()
|
||
{
|
||
return _dispatcherCacheCaptureMode;
|
||
}
|
||
|
||
public Bitmap GetLastCaptureBitmap()
|
||
{
|
||
lock (_bitmapLocker)
|
||
{
|
||
return new Bitmap(_bitmap);
|
||
}
|
||
}
|
||
|
||
public CaptureContent GetLastCaptureContent()
|
||
{
|
||
var bitmap = GetLastCaptureBitmap();
|
||
return new CaptureContent(bitmap, _frameIndex, _timer.Interval, this);
|
||
}
|
||
|
||
public void TakeScreenshot()
|
||
{
|
||
if (_dispatcherCacheCaptureMode is DispatcherCaptureModeEnum.OnlyCacheCapture or DispatcherCaptureModeEnum.CacheCaptureWithTrigger)
|
||
{
|
||
try
|
||
{
|
||
var path = Global.Absolute($@"log\screenshot\");
|
||
if (!Directory.Exists(path))
|
||
{
|
||
Directory.CreateDirectory(path);
|
||
}
|
||
|
||
var bitmap = GetLastCaptureBitmap();
|
||
var name = $@"{DateTime.Now:yyyyMMddHHmmssffff}.png";
|
||
var savePath = Global.Absolute($@"log\screenshot\{name}");
|
||
|
||
if (TaskContext.Instance().Config.CommonConfig.ScreenshotUidCoverEnabled)
|
||
{
|
||
var mat = bitmap.ToMat();
|
||
var rect = TaskContext.Instance().Config.MaskWindowConfig.UidCoverRect;
|
||
mat.Rectangle(rect, Scalar.White, -1);
|
||
Cv2.ImWrite(savePath, mat);
|
||
}
|
||
else
|
||
{
|
||
bitmap.Save(savePath, ImageFormat.Png);
|
||
}
|
||
|
||
_logger.LogInformation("截图已保存: {Name}", name);
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
_logger.LogError("截图保存失败: {Message}", e.Message);
|
||
_logger.LogDebug("截图保存失败: {StackTrace}", e.StackTrace);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
_logger.LogWarning("当前不处于截图模式，无法保存截图");
|
||
}
|
||
}
|
||
}
|
||
}
|