lzss

#!/usr/bin/env python3

#
# LZSS - Compress a file with Final Fantasy VII LZSS compression
#
# Copyright (C) Christian Bauer <www.cebix.net>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#

__version__ = "1.4"

import sys
import os
import struct

import ff7


# Print usage information and exit.
def usage(exitcode, error = None):
    print("Usage: %s [OPTION...] <fromfile> <tofile>" % os.path.basename(sys.argv[0]))
    print("  -V, --version                   Display version information and exit")
    print("  -?, --help                      Show this help message")

    if error is not None:
        print("\nError:", error, file=sys.stderr)

    sys.exit(exitcode)


# Parse command line arguments
inputFileName = None
outputFileName = None

for arg in sys.argv[1:]:
    if arg == "--version" or arg == "-V":
        print("LZSS", __version__)
        sys.exit(0)
    elif arg == "--help" or arg == "-?":
        usage(0)
    elif arg[0] == "-":
        usage(64, "Invalid option '%s'" % arg)
    else:
        if inputFileName is None:
            inputFileName = arg
        elif outputFileName is None:
            outputFileName = arg
        else:
            usage(64, "Unexpected extra argument '%s'" % arg)

if inputFileName is None:
    usage(64, "No input file specified")
if outputFileName is None:
    usage(64, "No output file specified")

# Read the input file
try:
    inputFile = open(inputFileName, "rb")
except IOError as e:
    print("Error opening file '%s': %s" % (inputFileName, e.strerror), file=sys.stderr)
    sys.exit(1)

data = inputFile.read()

# Compress it
data = ff7.compressLzss(data)

# Write data to output file
try:
    outputFile = open(outputFileName, "wb")
except IOError as e:
    print("Error creating file '%s': %s" % (outputFileName, e.strerror), file=sys.stderr)
    sys.exit(1)

outputFile.write(struct.pack("<L", len(data)))
outputFile.write(data)
outputFile.close()