-
Vojtech Moravec authoredVojtech Moravec authored
DataCompressor.java 4.78 KiB
package cz.it4i.qcmp;
import cz.it4i.qcmp.benchmark.CompressionBenchmark;
import cz.it4i.qcmp.cache.QuantizationCacheManager;
import cz.it4i.qcmp.cli.CliConstants;
import cz.it4i.qcmp.cli.CompressionOptionsCLIParser;
import cz.it4i.qcmp.cli.CustomFunctionBase;
import cz.it4i.qcmp.cli.functions.EntropyCalculation;
import cz.it4i.qcmp.compression.ImageCompressor;
import cz.it4i.qcmp.compression.ImageDecompressor;
import cz.it4i.qcmp.fileformat.FileExtensions;
import org.apache.commons.cli.*;
import java.io.IOException;
public class DataCompressor {
public static void main(final String[] args) {
final Options options = CliConstants.getOptions();
final HelpFormatter formatter = new HelpFormatter();
final CommandLineParser parser = new DefaultParser();
final CommandLine cmd;
try {
cmd = parser.parse(options, args);
} catch (final ParseException e) {
if ((args.length > 0) && (args[0].equals("-h") || args[0].equals("--help"))) {
formatter.printHelp(CliConstants.MAIN_HELP, options);
return;
}
if (e.getMessage().startsWith("Missing required option:")) {
System.err.println("Error: Missing required option, see usage below. :^)");
formatter.printHelp(CliConstants.MAIN_HELP, options);
} else {
System.err.println("Error: " + e.getMessage());
}
return;
}
final CompressionOptionsCLIParser compressionOptionsCLIParser = new CompressionOptionsCLIParser(cmd);
// NOTE(Moravec): From this point we need to dispose of possible existing SCIFIO context.
if (compressionOptionsCLIParser.parseError()) {
System.err.println(compressionOptionsCLIParser.getParseError());
ScifioWrapper.dispose();
return;
}
if (compressionOptionsCLIParser.isVerbose()) {
System.out.println(compressionOptionsCLIParser.report());
}
switch (compressionOptionsCLIParser.getMethod()) {
case Compress: {
final ImageCompressor compressor = new ImageCompressor(compressionOptionsCLIParser);
if (!compressor.compress()) {
System.err.println("Errors occurred during compression.");
}
}
break;
case Decompress: {
final ImageDecompressor decompressor = new ImageDecompressor(compressionOptionsCLIParser);
if (!decompressor.decompressToFile()) {
System.err.println("Errors occurred during decompression.");
}
}
break;
case Benchmark: {
CompressionBenchmark.runBenchmark(compressionOptionsCLIParser);
}
break;
case TrainCodebook: {
final ImageCompressor compressor = new ImageCompressor(compressionOptionsCLIParser);
if (!compressor.trainAndSaveCodebook()) {
System.err.println("Errors occurred during training/saving of codebook.");
}
}
break;
case CustomFunction: {
// NOTE(Moravec): Custom function class here |
// V
//CustomFunctionBase customFunction = new MeasurePlaneErrorFunction(parsedCliOptions);
final CustomFunctionBase customFunction = new EntropyCalculation(compressionOptionsCLIParser);
if (!customFunction.run()) {
System.err.println("Errors occurred during custom function.");
}
}
break;
case PrintHelp: {
formatter.printHelp(CliConstants.MAIN_HELP, options);
}
break;
case InspectFile: {
if (compressionOptionsCLIParser.getInputDataInfo().getFilePath().endsWith(FileExtensions.CACHE_FILE_EXT)) {
QuantizationCacheManager.inspectCacheFile(compressionOptionsCLIParser.getInputDataInfo().getFilePath(),
compressionOptionsCLIParser.isVerbose());
} else {
final ImageDecompressor decompressor = new ImageDecompressor(compressionOptionsCLIParser);
try {
System.out.println(decompressor.inspectCompressedFile());
} catch (final IOException e) {
System.err.println("Errors occurred during inspecting file.");
System.err.println(e.getMessage());
e.printStackTrace();
}
}
}
break;
}
ScifioWrapper.dispose();
}
}