public void OpusArchiveEncoderTest()
{
ExtendedArchiveWriter writer = new ExtendedArchiveWriter("opusencoder");
writer.Files.Add(new Subfile(new FileSource(WavTestFile), "audio.wav", "arc"));
writer.Write("opusencodertest.ppx");
ExtendedArchive arc = new ExtendedArchive("opusencodertest.ppx");
var subfile = arc.Files.First();
Assert.AreEqual(ArchiveFileType.OpusAudio, subfile.Type);
Assert.IsTrue(subfile.Name == "audio.opus", $"Internal name did not switch to \"audio.opus\". Actual: {subfile.Name}");
Assert.IsTrue(subfile.EmulatedName == "audio.wav", $"Emulated name did stay as \"audio.wav\". Actual: {subfile.EmulatedName}");
Assert.IsTrue(subfile.ArchiveName == "arc", $"Archive name did not stay as \"arc\". Actual: {subfile.ArchiveName}");
Assert.IsTrue(subfile.EmulatedArchiveName == "arc.pp", $"Emulated archive name did not change to \"arc.pp\". Actual: {subfile.EmulatedArchiveName}");
using (OpusEncoder decoder = new OpusEncoder(subfile.GetRawStream()))
{
Stream decoded = decoder.Decode();
//Ensure it can be read
using (WaveReader wavreader = new WaveReader(decoded)) { }
}
File.Delete("opusencodertest.ppx");
}
public void OpusDecoderTest()
{
using (Stream opus = File.OpenRead(OpusTestFile))
using (OpusEncoder decoder = new OpusEncoder(opus))
{
Stream decoded = decoder.Decode();
//Ensure it can be read
using (WaveReader wavreader = new WaveReader(decoded)) { }
}
}
public void OpusEncoderTest()
{
using (Stream source = File.OpenRead(WavTestFile))
{
using (OpusEncoder encoder = new OpusEncoder(source))
{
Stream encoded = encoder.Encode();
using (OpusEncoder decoder = new OpusEncoder(encoded))
{
Stream decoded = decoder.Decode();
//Ensure it can be read
using (WaveReader wavreader = new WaveReader(decoded)) { }
}
}
}
}
/// <summary>
/// Handler for any pipe requests.
/// </summary>
/// <param name="request">The command to execute.</param>
/// <param name="argument">Any additional arguments.</param>
/// <param name="handler">The streamhandler to use.</param>
private static void Server_OnRequest(string request, string argument, StreamHandler handler)
{
if (request == "ready")
{
//Notify the AA2 instance that we are ready
if (IsLoaded)
{
Console.WriteLine("Connected to pipe");
}
handler.WriteString(IsLoaded.ToString());
}
else if (request == "matchfiles")
{
//Send a list of all loaded .pp files
if (LogFiles)
{
Console.WriteLine("!!LOADED FILELIST!!");
}
var loadedPP = Cache.LoadedFileReferences.Keys.Select(x => x.Archive).Distinct();
foreach (string pp in loadedPP)
{
handler.WriteString(pp + ".pp");
if (LogFiles)
{
Console.WriteLine(pp);
}
}
handler.WriteString("");
}
else if (request == "load")
{
//Transfer the file
lock (Cache.LoadLock)
{
string[] splitNames = argument.Replace("data/", "").Split('/');
FileEntry entry = new FileEntry(splitNames[0], splitNames[1]);
Logger.LogFile(argument);
//Ensure we have the file
if (!Cache.ReferenceMd5Sums.ContainsKey(entry))
{
//We don't have the file
handler.WriteString("NotAvailable");
if (LogFiles)
{
Console.WriteLine("!" + argument);
}
return;
}
if (LogFiles)
{
Console.WriteLine(argument);
}
//Write the data to the pipe
var fileMd5 = Cache.ReferenceMd5Sums[entry];
if (!Cache.LoadedFiles.TryGetValue(fileMd5, out var cachedFile))
{
//Console.WriteLine("Cache miss");
var chunk = Cache.LoadedFileReferences[entry];
chunk.Allocate();
// wait for the file to be available
while (true)
{
if (Cache.LoadedFiles.TryGetValue(fileMd5, out cachedFile) && cachedFile.Ready)
{
break;
}
Thread.Sleep(50);
}
}
while (!cachedFile.Ready)
{
Thread.Sleep(50);
}
using var compressedDataRef = cachedFile.GetMemory();
using var buffer = MemoryPool <byte> .Shared.Rent((int) cachedFile.UncompressedSize * 2);
ZstdDecompressor.DecompressData(compressedDataRef.Memory.Span, buffer.Memory.Span, out int uncompressedSize);
//Console.WriteLine($"Expected: {Utility.GetBytesReadable(cachedFile.UncompressedSize)} Actual: {Utility.GetBytesReadable(uncompressedSize)}");
if (entry.File.EndsWith("wav"))
{
using (var inputStream = new ReadOnlyMemoryStream(buffer.Memory.Slice(0, (int)uncompressedSize)))
//using (var rentedSpan = MemoryPool<byte>.Shared.Rent(48_000_000))
using (var outputStream = new MemoryStream())
{
OpusEncoder.Decode(inputStream, outputStream, false);
outputStream.Position = 0;
//Console.WriteLine($"Decompressed wav size: {Utility.GetBytesReadable(outputStream.Length)}");
handler.WriteString(outputStream.Length.ToString());
outputStream.CopyTo(handler.BaseStream);
}
}
else
{
handler.WriteString(uncompressedSize.ToString());
handler.BaseStream.Write(buffer.Memory.Slice(0, (int)uncompressedSize).Span);
}
}
}
else
{
//Unknown command
//Ignore instead of throwing exception
Console.WriteLine("Unknown request: " + request + " [:] " + argument);
}
}
