Skip to content
Snippets Groups Projects
DataCompressor.java 4.25 KiB
Newer Older
  • Learn to ignore specific revisions
  • Vojtech Moravec's avatar
    Vojtech Moravec committed
    package azgracompress;
    
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
    import azgracompress.benchmark.CompressionBenchmark;
    
    import azgracompress.cache.QuantizationCacheManager;
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
    import azgracompress.cli.CliConstants;
    
    import azgracompress.cli.CustomFunctionBase;
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
    import azgracompress.cli.ParsedCliOptions;
    
    import azgracompress.cli.functions.EntropyCalculation;
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
    import azgracompress.compression.ImageCompressor;
    import azgracompress.compression.ImageDecompressor;
    
    import azgracompress.fileformat.FileExtensions;
    
    import org.apache.commons.cli.*;
    
    import java.io.IOException;
    
    public class DataCompressor {
    
        public static void main(String[] args) {
    
            Options options = CliConstants.getOptions();
    
    
            HelpFormatter formatter = new HelpFormatter();
            CommandLineParser parser = new DefaultParser();
            CommandLine cmd;
            try {
                cmd = parser.parse(options, args);
            } catch (ParseException e) {
                if ((args.length > 0) && (args[0].equals("-h") || args[0].equals("--help"))) {
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
                    formatter.printHelp(CliConstants.MAIN_HELP, options);
    
                System.err.println("Error: Missing required option, see usage below. :^)");
                formatter.printHelp(CliConstants.MAIN_HELP, options);
    
                return;
            }
    
            ParsedCliOptions parsedCliOptions = new ParsedCliOptions(cmd);
    
            // NOTE(Moravec): From this point we need to dispose of possible existing SCIFIO context.
    
            if (parsedCliOptions.parseError()) {
                System.err.println(parsedCliOptions.getParseError());
    
                ScifioWrapper.dispose();
    
            if (parsedCliOptions.isVerbose()) {
                System.out.println(parsedCliOptions.report());
            }
    
    
            switch (parsedCliOptions.getMethod()) {
    
                    ImageCompressor compressor = new ImageCompressor(parsedCliOptions);
    
                    if (!compressor.compress()) {
    
                        System.err.println("Errors occurred during compression.");
                    }
    
                break;
    
                    ImageDecompressor decompressor = new ImageDecompressor(parsedCliOptions);
    
                    if (!decompressor.decompressToFile()) {
    
                        System.err.println("Errors occurred during decompression.");
    
                case Benchmark: {
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
                    CompressionBenchmark.runBenchmark(parsedCliOptions);
    
                break;
    
                case TrainCodebook: {
                    ImageCompressor compressor = new ImageCompressor(parsedCliOptions);
                    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);
                    CustomFunctionBase customFunction = new EntropyCalculation(parsedCliOptions);
    
                    if (!customFunction.run()) {
                        System.err.println("Errors occurred during custom function.");
                    }
                }
    
                break;
    
    Vojtech Moravec's avatar
    Vojtech Moravec committed
                    formatter.printHelp(CliConstants.MAIN_HELP, options);
    
                    if (parsedCliOptions.getInputDataInfo().getFilePath().endsWith(FileExtensions.CACHE_FILE_EXT)) {
                        QuantizationCacheManager.inspectCacheFile(parsedCliOptions.getInputDataInfo().getFilePath());
                    } else {
                        ImageDecompressor decompressor = new ImageDecompressor(parsedCliOptions);
                        try {
                            System.out.println(decompressor.inspectCompressedFile());
                        } catch (IOException e) {
                            System.err.println("Errors occurred during inspecting file.");
                            System.err.println(e.getMessage());
                            e.printStackTrace();
                        }
    
                break;
    
            ScifioWrapper.dispose();