95 lines
3.1 KiB
C#
95 lines
3.1 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Threading.Tasks;
|
|
using Content.Client.Viewport;
|
|
using Content.Shared.Input;
|
|
using Robust.Client.Graphics;
|
|
using Robust.Client.Input;
|
|
using Robust.Client.State;
|
|
using Robust.Shared.ContentPack;
|
|
using Robust.Shared.Input.Binding;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Log;
|
|
using Robust.Shared.Utility;
|
|
using SixLabors.ImageSharp;
|
|
using SixLabors.ImageSharp.PixelFormats;
|
|
|
|
namespace Content.Client.Screenshot
|
|
{
|
|
internal sealed class ScreenshotHook : IScreenshotHook
|
|
{
|
|
private static readonly ResPath BaseScreenshotPath = new("/Screenshots");
|
|
|
|
[Dependency] private readonly IInputManager _inputManager = default!;
|
|
[Dependency] private readonly IClyde _clyde = default!;
|
|
[Dependency] private readonly IResourceManager _resourceManager = default!;
|
|
[Dependency] private readonly IStateManager _stateManager = default!;
|
|
|
|
public void Initialize()
|
|
{
|
|
_inputManager.SetInputCommand(ContentKeyFunctions.TakeScreenshot, InputCmdHandler.FromDelegate(_ =>
|
|
{
|
|
_clyde.Screenshot(ScreenshotType.Final, Take);
|
|
}));
|
|
|
|
_inputManager.SetInputCommand(ContentKeyFunctions.TakeScreenshotNoUI, InputCmdHandler.FromDelegate(_ =>
|
|
{
|
|
if (_stateManager.CurrentState is IMainViewportState state)
|
|
{
|
|
state.Viewport.Viewport.Screenshot(Take);
|
|
}
|
|
else
|
|
{
|
|
Logger.InfoS("screenshot", "Can't take no-UI screenshot: current state is not GameScreen");
|
|
}
|
|
}));
|
|
}
|
|
|
|
private async void Take<T>(Image<T> screenshot) where T : unmanaged, IPixel<T>
|
|
{
|
|
var time = DateTime.Now.ToString("yyyy-M-dd_HH.mm.ss");
|
|
|
|
if (!_resourceManager.UserData.IsDir(BaseScreenshotPath))
|
|
{
|
|
_resourceManager.UserData.CreateDir(BaseScreenshotPath);
|
|
}
|
|
|
|
for (var i = 0; i < 5; i++)
|
|
{
|
|
try
|
|
{
|
|
var filename = time;
|
|
|
|
if (i != 0)
|
|
{
|
|
filename = $"{filename}-{i}";
|
|
}
|
|
|
|
await using var file =
|
|
_resourceManager.UserData.Open(BaseScreenshotPath / $"{filename}.png", FileMode.CreateNew, FileAccess.Write, FileShare.None);
|
|
|
|
await Task.Run(() =>
|
|
{
|
|
// Saving takes forever, so don't hang the game on it.
|
|
screenshot.SaveAsPng(file);
|
|
});
|
|
|
|
Logger.InfoS("screenshot", "Screenshot taken as {0}.png", filename);
|
|
return;
|
|
}
|
|
catch (IOException e)
|
|
{
|
|
Logger.WarningS("screenshot", "Failed to save screenshot, retrying?:\n{0}", e);
|
|
}
|
|
}
|
|
|
|
Logger.ErrorS("screenshot", "Unable to save screenshot.");
|
|
}
|
|
}
|
|
|
|
public interface IScreenshotHook
|
|
{
|
|
void Initialize();
|
|
}
|
|
}
|