feat: 设置 pv 后端

Author: clansty

MaiChartManager/Controllers/ModController.cs

@@ -72,7 +72,7 @@ public void InstallMelonLoader()
             return;
         }
 
-        var zipFile = Path.Combine(Path.GetDirectoryName(Application.ExecutablePath), @"AquaMai\MelonLoader.x64.zip");
+        var zipFile = Path.Combine(StaticSettings.exeDir, @"AquaMai\MelonLoader.x64.zip");
         var zip = ZipFile.OpenRead(zipFile);
         foreach (var entry in zip.Entries)
         {
@@ -95,7 +95,7 @@ public record InstallAquaMaiRequest(GameEdition Version);
     [HttpPost]
     public void InstallAquaMai(InstallAquaMaiRequest request)
     {
-        var src = Path.Combine(Path.GetDirectoryName(Application.ExecutablePath), "AquaMai", request.Version.ToString(), "AquaMai.dll");
+        var src = Path.Combine(StaticSettings.exeDir, "AquaMai", request.Version.ToString(), "AquaMai.dll");
         var dest = Path.Combine(StaticSettings.GamePath, @"Mods\AquaMai.dll");
         Directory.CreateDirectory(Path.GetDirectoryName(dest));
         System.IO.File.Copy(src, dest, true);

MaiChartManager/Controllers/MusicController.cs

@@ -1,11 +1,14 @@
 using System.IO.Compression;
+using System.Text.Json;
+using System.Text.Json.Serialization;
 using AssetStudio;
 using MaiChartManager.Models;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.VisualBasic.FileIO;
 using Sitreamai;
 using Sitreamai.Models;
 using Standart.Hash.xxHash;
+using Xabe.FFmpeg;
 
 namespace MaiChartManager.Controllers;
 
@@ -236,6 +239,112 @@ public void SetAudio(int id, [FromForm] float padding, IFormFile file, IFormFile
         StaticSettings.AcbAwb[$"music{id:000000}.awb"] = targetAwbPath;
     }
 
+    public enum SetMovieEventType
+    {
+        Progress,
+        Success,
+        Error
+    }
+
+    [HttpPut]
+    [DisableRequestSizeLimit]
+    public async Task SetMovie(int id, [FromForm] float padding, IFormFile file)
+    {
+        Response.Headers.Append("Content-Type", "text/event-stream");
+        var tmpDir = Directory.CreateTempSubdirectory();
+        logger.LogInformation("Temp dir: {tmpDir}", tmpDir.FullName);
+        // Convert vp9
+        string outVideoPath;
+        try
+        {
+            var srcFilePath = Path.Combine(tmpDir.FullName, Path.GetFileName(file.FileName));
+            var srcFileStream = System.IO.File.OpenWrite(srcFilePath);
+            await file.CopyToAsync(srcFileStream);
+            await srcFileStream.DisposeAsync();
+
+            var srcMedia = await FFmpeg.GetMediaInfo(srcFilePath);
+            var conversion = FFmpeg.Conversions.New()
+                .UseMultiThread(true)
+                .AddParameter("-cpu-used 5");
+            if (padding < 0)
+            {
+                conversion.SetSeek(TimeSpan.FromSeconds(-padding));
+            }
+            else if (padding > 0)
+            {
+                var blankPath = Path.Combine(tmpDir.FullName, "blank.mp4");
+                var blank = FFmpeg.Conversions.New()
+                    .SetOutputTime(TimeSpan.FromSeconds(padding))
+                    .SetInputFormat(Format.lavfi)
+                    .AddParameter("-i color=c=black:s=720x720:r=1")
+                    .UseMultiThread(true)
+                    .SetOutput(blankPath);
+                logger.LogInformation("About to run FFMpeg with params: {params}", blank.Build());
+                await blank.Start();
+                var blankVideoInfo = await FFmpeg.GetMediaInfo(blankPath);
+                conversion.AddStream(blankVideoInfo.VideoStreams.First().SetCodec(VideoCodec.vp9));
+            }
+
+            outVideoPath = Path.Combine(tmpDir.FullName, "out.ivf");
+            conversion
+                .AddStream(srcMedia.VideoStreams.First().SetCodec(VideoCodec.vp9))
+                .SetOutput(outVideoPath);
+            logger.LogInformation("About to run FFMpeg with params: {params}", conversion.Build());
+            conversion.OnProgress += async (sender, args) =>
+            {
+                logger.LogInformation("FFMpeg progress: {progress}", args.Percent);
+                await Response.WriteAsync($"event: {SetMovieEventType.Progress}\ndata: {args.Percent}\n\n");
+                await Response.Body.FlushAsync();
+            };
+            await conversion.Start();
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e, "Failed to convert video");
+            SentrySdk.CaptureException(e);
+            await Response.WriteAsync($"event: {SetMovieEventType.Error}\ndata: 视频转换为 VP9 失败：{e.Message}\n\n");
+            await Response.Body.FlushAsync();
+            return;
+        }
+
+        // Convert ivf to usm
+        var outputFile = Path.Combine(tmpDir.FullName, "out.usm");
+        try
+        {
+            await WannaCRI.WannaCRI.CreateUsmAsync(outVideoPath);
+            if (!System.IO.File.Exists(outputFile) || new FileInfo(outputFile).Length == 0)
+            {
+                throw new Exception("Output file not found or empty");
+            }
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e, "Failed to convert ivf to usm");
+            SentrySdk.CaptureException(e);
+            await Response.WriteAsync($"event: {SetMovieEventType.Error}\ndata: 视频转换为 USM 失败：{e.Message}\n\n");
+            await Response.Body.FlushAsync();
+            return;
+        }
+
+        try
+        {
+            var targetPath = Path.Combine(StaticSettings.StreamingAssets, settings.AssetDir, $@"MovieData\{id:000000}.dat");
+            Directory.CreateDirectory(Path.GetDirectoryName(targetPath));
+            FileSystem.CopyFile(outputFile, targetPath, true);
+
+            StaticSettings.MovieDataMap[id] = targetPath;
+            await Response.WriteAsync($"event: {SetMovieEventType.Success}\n\n");
+            await Response.Body.FlushAsync();
+        }
+        catch (Exception e)
+        {
+            logger.LogError(e, "Failed to copy movie data");
+            SentrySdk.CaptureException(e);
+            await Response.WriteAsync($"event: {SetMovieEventType.Error}\ndata: 复制文件失败：{e.Message}\n\n");
+            await Response.Body.FlushAsync();
+        }
+    }
+
     [HttpPost]
     public void RequestCopyTo(int id)
     {