public void StreamingDecompressionSimpleRead(int readCount)
{
var data = new byte[] { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
var tempStream = new MemoryStream();
using (var compressionStream = new CompressionStream(tempStream))
compressionStream.Write(data, 0, data.Length);
tempStream.Seek(0, SeekOrigin.Begin);
var buffer = new byte[data.Length];
using (var decompressionStream = new DecompressionStream(tempStream))
{
int bytesRead;
int totalBytesRead = 0;
while ((bytesRead = decompressionStream.Read(buffer, totalBytesRead, Math.Min(readCount, buffer.Length - totalBytesRead))) > 0)
{
Assert.LessOrEqual(bytesRead, readCount);
totalBytesRead += bytesRead;
}
Assert.AreEqual(data.Length, totalBytesRead);
}
Assert.AreEqual(data, buffer);
}
/// <summary>まとめて読む用</summary>
/// <param name="Values">読み込み結果を格納する配列
/// 要素数がVector(long).Countの倍数になってないとデータが壊れる</param>
/// <returns>読み込んだ要素数</returns>
public int Read(long[] Values)
{
if (!Readable)
{
return(0);
}
int ValuesCursor = 0;
while (ValuesCursor < Values.Length)
{
int ReadBytes = zip.Read(MemoryMarshal.Cast <long, byte>(Values.AsSpan(ValuesCursor, Math.Min(BufSize, Values.Length - ValuesCursor))));
if (ReadBytes < sizeof(long))
{
Readable = false; break;
}
int ReadElements = ReadBytes / sizeof(long);
//XORを取って圧縮しやすくしたのを元に戻す
var TakeDiffSpan = MemoryMarshal.Cast <long, Vector <long> >(Values.AsSpan(ValuesCursor, ReadElements));
var Before = LastRead;
for (int i = 0; i < TakeDiffSpan.Length; i++)
{
Before = (TakeDiffSpan[i] ^= Before);
}
LastRead = Before;
ValuesCursor += ReadElements;
}
return(ValuesCursor);
}
public void RoundTrip_StreamingToStreaming(
[CombinatorialValues(false, true)] bool useDict, [CombinatorialValues(false, true)] bool advanced,
[CombinatorialValues(1, 2, 7, 101, 1024, 65535, DataGenerator.LargeBufferSize,
DataGenerator.LargeBufferSize + 1)]
int zstdBufferSize,
[CombinatorialValues(1, 2, 7, 101, 1024, 65535, DataGenerator.LargeBufferSize,
DataGenerator.LargeBufferSize + 1)]
int copyBufferSize)
{
var dict = useDict ? TrainDict() : null;
var testStream = DataGenerator.GetLargeStream(DataFill.Sequential);
const int offset = 1;
var buffer = new byte[copyBufferSize + offset + 1];
var tempStream = new MemoryStream();
using (var compressionStream =
new CompressionStream(tempStream, Compressor.DefaultCompressionLevel, zstdBufferSize))
{
compressionStream.LoadDictionary(dict);
if (advanced)
{
compressionStream.SetParameter(ZSTD_cParameter.ZSTD_c_windowLog, 11);
compressionStream.SetParameter(ZSTD_cParameter.ZSTD_c_checksumFlag, 1);
}
int bytesRead;
while ((bytesRead = testStream.Read(buffer, offset, copyBufferSize)) > 0)
{
compressionStream.Write(buffer, offset, bytesRead);
}
}
tempStream.Seek(0, SeekOrigin.Begin);
var resultStream = new MemoryStream();
using (var decompressionStream = new DecompressionStream(tempStream, zstdBufferSize))
{
decompressionStream.LoadDictionary(dict);
if (advanced)
{
decompressionStream.SetParameter(ZSTD_dParameter.ZSTD_d_windowLogMax, 11);
}
int bytesRead;
while ((bytesRead = decompressionStream.Read(buffer, offset, copyBufferSize)) > 0)
{
resultStream.Write(buffer, offset, bytesRead);
}
}
Assert.True(testStream.ToArray().SequenceEqual(resultStream.ToArray()));
}
public void StreamingCompressionZeroAndOneByte()
{
var data = new byte[] { 0, 0, 0, 1, 2, 3, 4, 0, 0, 0 };
var tempStream = new MemoryStream();
using (var compressionStream = new CompressionStream(tempStream))
{
compressionStream.Write(data, 0, 0);
compressionStream.Write(ReadOnlySpan <byte> .Empty);
compressionStream.WriteAsync(data, 0, 0).GetAwaiter().GetResult();
compressionStream.WriteAsync(ReadOnlyMemory <byte> .Empty).GetAwaiter().GetResult();
compressionStream.Write(data, 3, 1);
compressionStream.Write(new ReadOnlySpan <byte>(data, 4, 1));
compressionStream.Flush();
compressionStream.WriteAsync(data, 5, 1).GetAwaiter().GetResult();
compressionStream.WriteAsync(new ReadOnlyMemory <byte>(data, 6, 1)).GetAwaiter().GetResult();
compressionStream.FlushAsync().GetAwaiter().GetResult();
}
tempStream.Seek(0, SeekOrigin.Begin);
var result = new byte[data.Length];
using (var decompressionStream = new DecompressionStream(tempStream))
{
Assert.AreEqual(0, decompressionStream.Read(result, 0, 0));
Assert.AreEqual(0, decompressionStream.Read(Span <byte> .Empty));
Assert.AreEqual(0, decompressionStream.ReadAsync(result, 0, 0).GetAwaiter().GetResult());
Assert.AreEqual(0, decompressionStream.ReadAsync(Memory <byte> .Empty).GetAwaiter().GetResult());
Assert.AreEqual(1, decompressionStream.Read(result, 3, 1));
Assert.AreEqual(1, decompressionStream.Read(new Span <byte>(result, 4, 1)));
Assert.AreEqual(1, decompressionStream.ReadAsync(result, 5, 1).GetAwaiter().GetResult());
Assert.AreEqual(1, decompressionStream.ReadAsync(new Memory <byte>(result, 6, 1)).GetAwaiter().GetResult());
}
Assert.AreEqual(data, result);
}
public void RoundTrip_StreamingToStreaming(
[Values(false, true)] bool useDict, [Values(false, true)] bool advanced,
[Values(1, 2, 7, 101, 1024, 65535, DataGenerator.LargeBufferSize, DataGenerator.LargeBufferSize + 1)] int zstdBufferSize,
[Values(1, 2, 7, 101, 1024, 65535, DataGenerator.LargeBufferSize, DataGenerator.LargeBufferSize + 1)] int copyBufferSize)
{
var dict = useDict ? TrainDict() : null;
var testStream = DataGenerator.GetLargeStream(DataFill.Sequential);
const int offset = 1;
var buffer = new byte[copyBufferSize + offset + 1];
var tempStream = new MemoryStream();
using (var compressionOptions = new CompressionOptions(dict, advanced ? new Dictionary <ZSTD_cParameter, int> {
{ ZSTD_cParameter.ZSTD_c_windowLog, 11 }, { ZSTD_cParameter.ZSTD_c_checksumFlag, 1 }, { ZSTD_cParameter.ZSTD_c_nbWorkers, 4 }
} : null))
using (var compressionStream = new CompressionStream(tempStream, compressionOptions, zstdBufferSize))
{
int bytesRead;
while ((bytesRead = testStream.Read(buffer, offset, copyBufferSize)) > 0)
{
compressionStream.Write(buffer, offset, bytesRead);
}
}
tempStream.Seek(0, SeekOrigin.Begin);
var resultStream = new MemoryStream();
using (var decompressionOptions = new DecompressionOptions(dict, advanced ? new Dictionary <ZSTD_dParameter, int> {
{ ZSTD_dParameter.ZSTD_d_windowLogMax, 11 }
} : null))
using (var decompressionStream = new DecompressionStream(tempStream, decompressionOptions, zstdBufferSize))
{
int bytesRead;
while ((bytesRead = decompressionStream.Read(buffer, offset, copyBufferSize)) > 0)
{
resultStream.Write(buffer, offset, bytesRead);
}
}
Assert.AreEqual(testStream.ToArray(), resultStream.ToArray());
}
public void RoundTrip_StreamingToStreaming_Stress([Values(true, false)] bool useDict, [Values(true, false)] bool async)
{
long i = 0;
var dict = useDict ? TrainDict() : null;
var compressionOptions = new CompressionOptions(dict);
var decompressionOptions = new DecompressionOptions(dict);
Enumerable.Range(0, 10000)
.AsParallel()
.WithDegreeOfParallelism(Environment.ProcessorCount * 4)
.ForAll(n =>
{
var testStream = DataGenerator.GetSmallStream(DataFill.Sequential);
var cBuffer = new byte[1 + (int)(n % (testStream.Length * 11))];
var dBuffer = new byte[1 + (int)(n % (testStream.Length * 13))];
var tempStream = new MemoryStream();
using (var compressionStream = new CompressionStream(tempStream, compressionOptions, 1 + (int)(n % (testStream.Length * 17))))
{
int bytesRead;
int offset = n % cBuffer.Length;
while ((bytesRead = testStream.Read(cBuffer, offset, cBuffer.Length - offset)) > 0)
{
if (async)
{
compressionStream.WriteAsync(cBuffer, offset, bytesRead).GetAwaiter().GetResult();
}
else
{
compressionStream.Write(cBuffer, offset, bytesRead);
}
if (Interlocked.Increment(ref i) % 100 == 0)
{
GC.Collect(GC.MaxGeneration, GCCollectionMode.Forced, true, true);
}
}
}
tempStream.Seek(0, SeekOrigin.Begin);
var resultStream = new MemoryStream();
using (var decompressionStream = new DecompressionStream(tempStream, decompressionOptions, 1 + (int)(n % (testStream.Length * 19))))
{
int bytesRead;
int offset = n % dBuffer.Length;
while ((bytesRead = async ? decompressionStream.ReadAsync(dBuffer, offset, dBuffer.Length - offset).GetAwaiter().GetResult() : decompressionStream.Read(dBuffer, offset, dBuffer.Length - offset)) > 0)
{
resultStream.Write(dBuffer, offset, bytesRead);
if (Interlocked.Increment(ref i) % 100 == 0)
{
GC.Collect(GC.MaxGeneration, GCCollectionMode.Forced, true, true);
}
}
}
Assert.AreEqual(testStream.ToArray(), resultStream.ToArray());
});
GC.KeepAlive(compressionOptions);
GC.KeepAlive(decompressionOptions);
}