85 lines
3.4 KiB
C#
85 lines
3.4 KiB
C#
using System;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using BrotliCalc.Commands.Base;
|
|
using BrotliCalc.Helpers;
|
|
using BrotliImpl.Transformers;
|
|
using BrotliLib.Numbers;
|
|
|
|
namespace BrotliCalc.Commands{
|
|
class CmdBenchReserializeRebuild : ICommand{
|
|
private const int SkippedRuns = 3;
|
|
private const int CountedRuns = 10;
|
|
|
|
public string FullName => "bench-reserialize-rebuild";
|
|
public string ShortName => "brr";
|
|
|
|
public string ArgumentDesc => "<source-path> <output-file>";
|
|
public IntRange ArgumentCount => IntRange.Only(2);
|
|
|
|
public string Process(string[] args){
|
|
int totalFiles = 0;
|
|
int failedFiles = 0;
|
|
|
|
var items = Brotli.ListPath(args[0]).SelectCompressedFiles().ToArray();
|
|
|
|
using(var progress = new Progress(items.Length))
|
|
using(var table = new Table.CSV(args[1], new []{
|
|
"File", "Quality", "Original Bytes", "Reserialize Bytes", "Rebuild Bytes", "Avg Reserialize Time (ms)", "Avg Rebuild Time (ms)"
|
|
})){
|
|
foreach(var (group, file) in items){
|
|
progress.Start($"Processing {file}");
|
|
|
|
int? originalBytes = file.SizeBytes;
|
|
int? reserializeBytes = null;
|
|
int? rebuildBytes = null;
|
|
long? reserializeTotalTime = 0L;
|
|
long? rebuildTotalTime = 0L;
|
|
|
|
for(int run = 1; run <= SkippedRuns + CountedRuns; run++){
|
|
Stopwatch swReserialize = Stopwatch.StartNew();
|
|
|
|
try{
|
|
reserializeBytes = group.CountBytesAndValidate(file.Reader);
|
|
}catch(Exception e){
|
|
Debug.WriteLine(e.ToString());
|
|
++failedFiles;
|
|
reserializeTotalTime = null;
|
|
rebuildTotalTime = null;
|
|
break;
|
|
}finally{
|
|
swReserialize.Stop();
|
|
}
|
|
|
|
Stopwatch swRebuild = Stopwatch.StartNew();
|
|
|
|
try{
|
|
rebuildBytes = group.CountBytesAndValidate(file.Transforming(new TransformRebuild()));
|
|
}catch(Exception e){
|
|
Debug.WriteLine(e.ToString());
|
|
++failedFiles;
|
|
reserializeTotalTime = null;
|
|
rebuildTotalTime = null;
|
|
break;
|
|
}finally{
|
|
swRebuild.Stop();
|
|
}
|
|
|
|
if (run > SkippedRuns){
|
|
reserializeTotalTime += swReserialize.ElapsedMilliseconds;
|
|
rebuildTotalTime += swRebuild.ElapsedMilliseconds;
|
|
}
|
|
}
|
|
|
|
++totalFiles;
|
|
table.AddRow(file.Name, file.Identifier, originalBytes, reserializeBytes, rebuildBytes, reserializeTotalTime / CountedRuns, rebuildTotalTime / CountedRuns); // subtraction propagates null
|
|
|
|
progress.Finish($"Completed {file}");
|
|
}
|
|
}
|
|
|
|
return $"Processed {totalFiles} file(s) with {failedFiles} error(s).";
|
|
}
|
|
}
|
|
}
|