Skip to content
Snippets Groups Projects
Commit b5d7ab5a authored by Vojtech Moravec's avatar Vojtech Moravec
Browse files

Renamec the benchmark classes.

parent 3a547be1
Branches
No related tags found
No related merge requests found
......@@ -6,13 +6,13 @@ public class CompressionBenchmark {
public static void runBenchmark(final ParsedCliOptions options) {
switch (options.getQuantizationType()) {
case Scalar: {
ScalarQuantizationBenchmark sqBenchmark = new ScalarQuantizationBenchmark(options);
SQBenchmark sqBenchmark = new SQBenchmark(options);
sqBenchmark.startBenchmark();
}
break;
case Vector1D:
case Vector2D: {
VectorQuantizationBenchmark vqBenchmark = new VectorQuantizationBenchmark(options);
VQBenchmark vqBenchmark = new VQBenchmark(options);
vqBenchmark.startBenchmark(options.getVectorDimension());
}
break;
......
package azgracompress.benchmark;
import azgracompress.U16;
import azgracompress.cache.QuantizationCacheManager;
import azgracompress.cli.ParsedCliOptions;
import azgracompress.data.V3i;
import azgracompress.quantization.QTrainIteration;
import azgracompress.cache.QuantizationCacheManager;
import azgracompress.quantization.scalar.LloydMaxU16ScalarQuantization;
import azgracompress.quantization.scalar.SQCodebook;
import azgracompress.quantization.scalar.ScalarQuantizer;
......@@ -14,15 +13,8 @@ import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
public class ScalarQuantizationBenchmark extends BenchmarkBase {
public ScalarQuantizationBenchmark(final String inputFile,
final String outputDirectory,
final int[] planes,
final V3i rawImageDims) {
super(inputFile, outputDirectory, planes, rawImageDims);
}
public ScalarQuantizationBenchmark(final ParsedCliOptions options) {
public class SQBenchmark extends BenchmarkBase {
public SQBenchmark(final ParsedCliOptions options) {
super(options);
}
......
......@@ -10,15 +10,15 @@ import azgracompress.quantization.vector.VectorQuantizer;
import java.io.File;
public class VectorQuantizationBenchmark extends BenchmarkBase {
public class VQBenchmark extends BenchmarkBase {
final static V2i DEFAULT_QVECTOR = new V2i(3, 3);
public VectorQuantizationBenchmark(String inputFile, String outputDirectory, int[] planes, V3i rawImageDims) {
public VQBenchmark(String inputFile, String outputDirectory, int[] planes, V3i rawImageDims) {
super(inputFile, outputDirectory, planes, rawImageDims);
}
public VectorQuantizationBenchmark(final ParsedCliOptions options) {
public VQBenchmark(final ParsedCliOptions options) {
super(options);
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment