def _test_decompress(self, fmt, z, b):
d = zopfli.ZopfliDecompressor(fmt)
self.assertEqual(d.decompress(z) + d.flush(), b)
self.assertEqual(d.unused_data, b'')
self.assertEqual(d.unconsumed_tail, b'')
if sys.version_info >= (3, 3):
self.assertTrue(d.eof)
def zopfli_decode(data, out=None):
"""Decompress Zopfli."""
d = zopfli.ZopfliDecompressor(zopfli.ZOPFLI_FORMAT_ZLIB)
return d.decompress(data) + d.flush()
def test_unknown(self):
with self.assertRaises(ValueError):
zopfli.ZopfliCompressor(-1)
with self.assertRaises(ValueError):
zopfli.ZopfliDecompressor(-1)
def main():
parser = argparse.ArgumentParser(
description='Compresses ILBM IFF file with Deflate algorithm.')
parser.add_argument('-m',
'--method',
type=str,
choices=['none', 'deflate'],
default='deflate',
help='Compression method to use.')
parser.add_argument('-q',
'--quiet',
action='store_true',
help='Silence out diagnostic messages.')
parser.add_argument(
'-f',
'--force',
action='store_true',
help='If the output file exists, the tool will overwrite it.')
parser.add_argument('input',
metavar='INPUT',
type=str,
help='Input ILBM IFF file name.')
parser.add_argument('output',
metavar='OUTPUT',
type=str,
nargs='?',
help='Output ILBM IFF file name.')
args = parser.parse_args()
logLevel = [logging.INFO, logging.WARNING][args.quiet]
logging.basicConfig(level=logLevel, format="%(levelname)s: %(message)s")
if args.output is None:
args.output = args.input
if not os.path.isfile(args.input):
raise SystemExit('File "%s" does not exists!' % args.input)
if os.path.exists(args.output) and not args.force:
raise SystemExit('File "%s" already exists (use "-f" to override).' %
args.output)
ilbm = ILBM()
ilbm.ChunkBlackList.append('CRNG')
if ilbm.load(args.input):
bmhd = ilbm.get('BMHD')
body = ilbm.get('BODY')
size = ((bmhd.data.w + 15) & ~15) / 8 * bmhd.data.h * bmhd.data.nPlanes
payload = body.data.read()
logging.info(bmhd.data)
logging.info('BODY size before compression: %d/%d' %
(len(payload), size))
if bmhd.data.compression in [0, 1, 254, 255]:
if bmhd.data.compression == 1:
payload = UnRLE(payload)
if bmhd.data.compression == 254:
decompressor = zopfli.ZopfliDecompressor(
zopfli.ZOPFLI_FORMAT_DEFLATE)
payload = decompressor.decompress(
payload) + decompressor.flush()
compression = 0
if args.method == 'deflate':
compressor = zopfli.ZopfliCompressor(
zopfli.ZOPFLI_FORMAT_DEFLATE)
payload = compressor.compress(payload) + compressor.flush()
compression = 254
if args.method == 'none':
compression = 0
body.data = payload
bmhd.data = bmhd.data._replace(compression=compression)
logging.info('BODY size after compression: %d/%d' %
(len(body.data), size))
ilbm.save(args.output)
else:
logging.warning('Unknown compression: %d' % bmhd.data.compression)
