218 lines
8.8 KiB
C#
218 lines
8.8 KiB
C#
using Spine.Exporters;
|
|
using Spine;
|
|
using SpineViewer.Extensions;
|
|
using SpineViewer.Models;
|
|
using SpineViewer.Resources;
|
|
using SpineViewer.Services;
|
|
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using System.IO;
|
|
using System.Collections.Immutable;
|
|
using SpineViewer.ViewModels.MainWindow;
|
|
|
|
namespace SpineViewer.ViewModels.Exporters
|
|
{
|
|
public class FFmpegVideoExporterViewModel(MainWindowViewModel vmMain) : VideoExporterViewModel(vmMain)
|
|
{
|
|
public static ImmutableArray<FFmpegVideoExporter.VideoFormat> VideoFormatOptions { get; } = Enum.GetValues<FFmpegVideoExporter.VideoFormat>().ToImmutableArray();
|
|
|
|
public FFmpegVideoExporter.VideoFormat Format
|
|
{
|
|
get => _format;
|
|
set
|
|
{
|
|
if (!SetProperty(ref _format, value))
|
|
return;
|
|
OnPropertyChanged(nameof(EnableParamLoop));
|
|
OnPropertyChanged(nameof(EnableParamQuality));
|
|
OnPropertyChanged(nameof(EnableParamLossless));
|
|
OnPropertyChanged(nameof(EnableParamApngPred));
|
|
OnPropertyChanged(nameof(EnableParamCrf));
|
|
OnPropertyChanged(nameof(EnableParamProfile));
|
|
}
|
|
}
|
|
protected FFmpegVideoExporter.VideoFormat _format = FFmpegVideoExporter.VideoFormat.Mp4;
|
|
|
|
public bool Loop { get => _loop; set => SetProperty(ref _loop, value); }
|
|
protected bool _loop = true;
|
|
|
|
public bool EnableParamLoop =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Gif ||
|
|
_format == FFmpegVideoExporter.VideoFormat.Webp ||
|
|
_format == FFmpegVideoExporter.VideoFormat.Apng;
|
|
|
|
public int Quality { get => _quality; set => SetProperty(ref _quality, Math.Clamp(value, 0, 100)); }
|
|
protected int _quality = 75;
|
|
|
|
public bool EnableParamQuality =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Webp;
|
|
|
|
public bool Lossless { get => _lossless; set => SetProperty(ref _lossless, value); }
|
|
protected bool _lossless = false;
|
|
|
|
public bool EnableParamLossless =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Webp;
|
|
|
|
public int ApngPred { get => _apngPred; set => SetProperty(ref _apngPred, Math.Clamp(value, 0, 5)); }
|
|
protected int _apngPred = 5;
|
|
|
|
public bool EnableParamApngPred =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Apng;
|
|
|
|
public int Crf { get => _crf; set => SetProperty(ref _crf, Math.Clamp(value, 0, 63)); }
|
|
protected int _crf = 23;
|
|
|
|
public bool EnableParamCrf =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Mp4 ||
|
|
_format == FFmpegVideoExporter.VideoFormat.Webm ||
|
|
_format == FFmpegVideoExporter.VideoFormat.Mkv;
|
|
|
|
public int Profile { get => _profile; set => SetProperty(ref _profile, Math.Clamp(value, -1, 5)); }
|
|
protected int _profile = 5;
|
|
|
|
public bool EnableParamProfile =>
|
|
_format == FFmpegVideoExporter.VideoFormat.Mov;
|
|
|
|
private string FormatSuffix => $".{_format.ToString().ToLower()}";
|
|
|
|
protected override void Export(SpineObjectModel[] models)
|
|
{
|
|
if (!DialogService.ShowFFmpegVideoExporterDialog(this)) return;
|
|
SpineObject[] spines = models.Select(m => m.GetSpineObject()).ToArray();
|
|
ProgressService.RunAsync((pr, ct) => ExportTask(spines, pr, ct), AppResource.Str_FFmpegVideoExporterTitle);
|
|
foreach (var sp in spines) sp.Dispose();
|
|
}
|
|
|
|
private void ExportTask(SpineObject[] spines, IProgressReporter pr, CancellationToken ct)
|
|
{
|
|
if (spines.Length <= 0) return;
|
|
|
|
var timestamp = DateTime.Now.ToString("yyMMddHHmmss");
|
|
using var exporter = new FFmpegVideoExporter(_renderer.Resolution.X + _margin * 2, _renderer.Resolution.Y + _margin * 2)
|
|
{
|
|
BackgroundColor = new(_backgroundColor.R, _backgroundColor.G, _backgroundColor.B, _backgroundColor.A),
|
|
Fps = _fps,
|
|
Speed = _speed,
|
|
KeepLast = _keepLast,
|
|
Format = _format,
|
|
Loop = _loop,
|
|
Quality = _quality,
|
|
Lossless = _lossless,
|
|
ApngPred = _apngPred,
|
|
Crf = _crf,
|
|
Profile = _profile,
|
|
};
|
|
|
|
// 非自动分辨率则直接用预览画面的视区参数
|
|
if (!_autoResolution)
|
|
{
|
|
using var view = _renderer.GetView();
|
|
var bounds = view.GetBounds().GetCanvasBounds(_renderer.Resolution, _margin);
|
|
exporter.Size = bounds.Size;
|
|
exporter.Center = bounds.Position + bounds.Size / 2;
|
|
exporter.Rotation = view.Rotation;
|
|
}
|
|
|
|
// BUG: FFmpeg 导出时对 RenderTexture 的频繁资源申请释放似乎使 SFML 库内部出现问题, 会卡死所有使用 SFML 的地方, 包括渲染线程
|
|
// 所以临时把渲染线程停掉, 只让此处使用 SFML 资源, 这个问题或许和多个线程同时使用渲染资源有关
|
|
_vmMain.SFMLRendererViewModel.StopRender();
|
|
|
|
if (_exportSingle)
|
|
{
|
|
var filename = $"video_{timestamp}_{Guid.NewGuid().ToString()[..6]}_{_fps}{FormatSuffix}";
|
|
var output = Path.Combine(_outputDir!, filename);
|
|
|
|
if (_autoResolution) SetAutoResolutionAnimated(exporter, spines);
|
|
|
|
// 如果时长是一个负数值则使用所有动画时长的最大值
|
|
exporter.Duration = _duration < 0 ? spines.Select(sp => sp.GetAnimationMaxDuration()).DefaultIfEmpty(0).Max() : _duration;
|
|
|
|
exporter.ProgressReporter = (total, done, text) =>
|
|
{
|
|
pr.Total = total;
|
|
pr.Done = done;
|
|
pr.ProgressText = text;
|
|
_vmMain.ProgressValue = pr.Done / pr.Total;
|
|
};
|
|
|
|
_vmMain.ProgressState = System.Windows.Shell.TaskbarItemProgressState.Normal;
|
|
_vmMain.ProgressValue = 0;
|
|
try
|
|
{
|
|
exporter.Export(output, ct, spines);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Trace(ex.ToString());
|
|
_logger.Error("Failed to export {0}, {1}", output, ex.Message);
|
|
}
|
|
_vmMain.ProgressState = System.Windows.Shell.TaskbarItemProgressState.None;
|
|
}
|
|
else
|
|
{
|
|
// 统计总帧数
|
|
int totalFrameCount = 0;
|
|
if (_duration < 0)
|
|
{
|
|
foreach (var sp in spines)
|
|
{
|
|
exporter.Duration = sp.GetAnimationMaxDuration();
|
|
totalFrameCount += exporter.GetFrameCount();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
exporter.Duration = _duration;
|
|
totalFrameCount = exporter.GetFrameCount() * spines.Length;
|
|
}
|
|
|
|
pr.Total = totalFrameCount;
|
|
pr.Done = 0;
|
|
|
|
exporter.ProgressReporter = (total, done, text) =>
|
|
{
|
|
pr.Done++;
|
|
pr.ProgressText = text;
|
|
_vmMain.ProgressValue = pr.Done / pr.Total;
|
|
};
|
|
|
|
_vmMain.ProgressState = System.Windows.Shell.TaskbarItemProgressState.Normal;
|
|
_vmMain.ProgressValue = 0;
|
|
foreach (var sp in spines)
|
|
{
|
|
if (ct.IsCancellationRequested)
|
|
{
|
|
_logger.Info("Export cancelled");
|
|
break;
|
|
}
|
|
|
|
if (_autoResolution) SetAutoResolutionAnimated(exporter, sp);
|
|
|
|
// 如果时长是负数则需要每次都设置成动画的时长值, 否则前面统计帧数时已经设置过时长值
|
|
if (_duration < 0) exporter.Duration = sp.GetAnimationMaxDuration();
|
|
|
|
var filename = $"{sp.Name}_{timestamp}_{Guid.NewGuid().ToString()[..6]}_{_fps}{FormatSuffix}";
|
|
var output = Path.Combine(_outputDir ?? sp.AssetsDir, filename);
|
|
|
|
try
|
|
{
|
|
exporter.Export(output, ct, sp);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.Trace(ex.ToString());
|
|
_logger.Error("Failed to export {0}, {1}", output, ex.Message);
|
|
}
|
|
}
|
|
_vmMain.ProgressState = System.Windows.Shell.TaskbarItemProgressState.None;
|
|
}
|
|
|
|
_vmMain.SFMLRendererViewModel.StartRender();
|
|
}
|
|
}
|
|
}
|