wabbajack/Wabbajack.Hashing.PHash.Test/FileLoadingTests.cs

58 lines
2.0 KiB
C#
Raw Normal View History

using System.IO;
using System.Threading;
2021-09-27 12:42:46 +00:00
using System.Threading.Tasks;
using FluentAssertions;
2021-09-27 12:42:46 +00:00
using Shipwreck.Phash;
using Wabbajack.DTOs.Texture;
using Wabbajack.Paths;
using Wabbajack.Paths.IO;
using Xunit;
2021-10-23 16:51:17 +00:00
namespace Wabbajack.Hashing.PHash.Test;
public class FileLoadingTests
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
[Theory]
[InlineData("test-dxt5.dds", 1.0f)]
[InlineData("test-dxt5-recompressed.dds", 1f)]
[InlineData("test-dxt5-small-bc7.dds", 0.983f)]
[InlineData("test-dxt5-small-bc7-vflip.dds", 0.189f)]
public async Task LoadAllFiles(string file, float difference)
2021-09-27 12:42:46 +00:00
{
2021-10-23 16:51:17 +00:00
var baseState =
await ImageLoader.Load("TestData/test-dxt5.dds".ToRelativePath().RelativeTo(KnownFolders.EntryPoint));
var state = await ImageLoader.Load("TestData".ToRelativePath().Combine(file)
.RelativeTo(KnownFolders.EntryPoint));
Assert.NotEqual(DXGI_FORMAT.UNKNOWN, baseState.Format);
2021-09-27 12:42:46 +00:00
2021-10-23 16:51:17 +00:00
Assert.Equal(difference,
ImagePhash.GetCrossCorrelation(
new Digest {Coefficients = baseState.PerceptualHash.Data},
new Digest {Coefficients = state.PerceptualHash.Data}),
1.0);
2021-09-27 12:42:46 +00:00
}
[Fact]
public async Task CanConvertCubeMaps()
{
// File used here via re-upload permissions found on the mod's Nexus page:
// https://www.nexusmods.com/fallout4/mods/43458?tab=description
// Used for testing purposes only
var path = "TestData/WindowDisabled_CGPlayerHouseCube.dds".ToRelativePath().RelativeTo(KnownFolders.EntryPoint);
var baseState = await ImageLoader.Load(path);
baseState.Height.Should().Be(128);
baseState.Width.Should().Be(128);
//baseState.Frames.Should().Be(6);
using var ms = new MemoryStream();
await using var ins = path.Open(FileMode.Open, FileAccess.Read, FileShare.Read);
await ImageLoader.Recompress(ins, 128, 128, DXGI_FORMAT.BC1_UNORM, ms, CancellationToken.None, leaveOpen:true);
ms.Length.Should().Be(ins.Length);
}
2021-09-27 12:42:46 +00:00
}