def decompress(compf, f): """ Decompress a file. """ # read alphabet alphabet = cPickle.load(compf) print "alphabet:\n%r" % alphabet count = 0 while True: header = compf.read(8) if header == '': break print "block %r:" % count nbits, I = struct.unpack('>2I', header) freqs = load_freqs(compf) #print freqs nbyte = (nbits+7) / 8 huff_encode = to01(compf.read(nbyte))[:nbits] #print "huff_encode:\n%r" % huff_encode coding, root = huff.generate_coding(freqs) #print "coding:\n%r" % coding huff_decode = huff.decode(huff_encode, root) mtf_decode = mtf.decode(alphabet, huff_decode) bwt_decode = bwt.decode(mtf_decode, I, reverse = True) content = ''.join(bwt_decode) f.write(content) print "nbits = %r, I = %r, length = %r" % (nbits, I, len(content)) count = count + 1
def test_mix(self): seq = generator(16*1024) encode, I = bwt.encode(seq) decode = bwt.decode(encode, I, True) self.assertEqual(list(seq), decode)
def test_decode(self): seq = 'caraab' decode = bwt.decode(seq, 1, True) self.assertEqual(list('abraca'), decode)