73 lines
1.9 KiB
C#
73 lines
1.9 KiB
C#
namespace ConfusedPolarBear.Plugin.IntroSkipper.Tests;
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Microsoft.Extensions.Logging;
|
|
using Xunit;
|
|
|
|
public class TestBlackFrames
|
|
{
|
|
[FactSkipFFmpegTests]
|
|
public void TestBlackFrameDetection()
|
|
{
|
|
var range = 1e-5;
|
|
|
|
var expected = new List<BlackFrame>();
|
|
expected.AddRange(CreateFrameSequence(2, 3));
|
|
expected.AddRange(CreateFrameSequence(5, 6));
|
|
expected.AddRange(CreateFrameSequence(8, 9.96));
|
|
|
|
var actual = FFmpegWrapper.DetectBlackFrames(queueFile("rainbow.mp4"), new(0, 10), 85);
|
|
|
|
for (var i = 0; i < expected.Count; i++)
|
|
{
|
|
var (e, a) = (expected[i], actual[i]);
|
|
Assert.Equal(e.Percentage, a.Percentage);
|
|
Assert.InRange(a.Time, e.Time - range, e.Time + range);
|
|
}
|
|
}
|
|
|
|
[FactSkipFFmpegTests]
|
|
public void TestEndCreditDetection()
|
|
{
|
|
var range = 1;
|
|
|
|
var analyzer = CreateBlackFrameAnalyzer();
|
|
|
|
var episode = queueFile("credits.mp4");
|
|
episode.Duration = (int)new TimeSpan(0, 5, 30).TotalSeconds;
|
|
|
|
var result = analyzer.AnalyzeMediaFile(episode, 240, 30, 85);
|
|
Assert.NotNull(result);
|
|
Assert.InRange(result.IntroStart, 300 - range, 300 + range);
|
|
}
|
|
|
|
private QueuedEpisode queueFile(string path)
|
|
{
|
|
return new()
|
|
{
|
|
EpisodeId = Guid.NewGuid(),
|
|
Name = path,
|
|
Path = "../../../video/" + path
|
|
};
|
|
}
|
|
|
|
private BlackFrame[] CreateFrameSequence(double start, double end)
|
|
{
|
|
var frames = new List<BlackFrame>();
|
|
|
|
for (var i = start; i < end; i += 0.04)
|
|
{
|
|
frames.Add(new(100, i));
|
|
}
|
|
|
|
return frames.ToArray();
|
|
}
|
|
|
|
private BlackFrameAnalyzer CreateBlackFrameAnalyzer()
|
|
{
|
|
var logger = new LoggerFactory().CreateLogger<BlackFrameAnalyzer>();
|
|
return new(logger);
|
|
}
|
|
}
|