shithub: pokecrystal

ref: 2b7237e299d5018e837bae550d18deeff3c7db7d
dir: /tools/lz/main.c/

View raw version
#include "proto.h"

int main (int argc, char ** argv) {
  struct options options = get_options(argc, argv);
  unsigned short size;
  unsigned char * file_buffer = read_file_into_buffer(options.input, &size);
  struct command * commands;
  if (options.mode & 2) {
    unsigned short original_size = size, remainder;
    commands = get_commands_from_file(file_buffer, &size, &remainder);
    if (!commands) error_exit(1, "invalid command stream");
    if (options.mode == 2) {
      unsigned char * uncompressed = get_uncompressed_data(commands, file_buffer, &size);
      if (!uncompressed) error_exit(1, "output data is too large");
      write_raw_data_to_file(options.output, uncompressed, size);
      free(uncompressed);
    } else
      write_commands_and_padding_to_textfile(options.output, commands, size, file_buffer, original_size - remainder, remainder);
  } else {
    commands = compress(file_buffer, &size, options.method);
    (options.mode ? write_commands_to_textfile : write_commands_to_file)(options.output, commands, size, file_buffer, options.alignment);
  }
  free(file_buffer);
  free(commands);
  return 0;
}

struct command * compress (const unsigned char * data, unsigned short * size, unsigned method) {
  unsigned char * bitflipped = malloc(*size);
  unsigned current;
  for (current = 0; current < *size; current ++) bitflipped[current] = bit_flipping_table[data[current]];
  const struct compressor * compressor = compressors;
  struct command * result;
  if (method < COMPRESSION_METHODS) {
    while (method >= compressor -> methods) method -= (compressor ++) -> methods;
    result = compressor -> function(data, bitflipped, size, method);
  } else {
    struct command * compressed_sequences[COMPRESSION_METHODS];
    unsigned short lengths[COMPRESSION_METHODS];
    unsigned flags = compressor -> methods;
    for (current = 0; current < COMPRESSION_METHODS; current ++) {
      lengths[current] = *size;
      if (!flags) flags = (++ compressor) -> methods;
      compressed_sequences[current] = compressor -> function(data, bitflipped, lengths + current, -- flags);
    }
    result = select_command_sequence(compressed_sequences, lengths, COMPRESSION_METHODS, size);
    for (current = 0; current < COMPRESSION_METHODS; current ++) free(compressed_sequences[current]);
  }
  free(bitflipped);
  return result;
}