lzw_decode.py

import sys
from sys import argv
import struct
from struct import *

# taking the compressed file input and the number of bits from command line
# defining the maximum table size
# opening the compressed file
# defining variables
'''
input_file, n = argv[1:]            
maximum_table_size = pow(2,int(n))
file = open(input_file, "rb")
'''
compressed_data = list(map(int,input().split()))
next_code = 256
decompressed_data = ""
string = ""

# Reading the compressed file.
'''
while True:
    rec = file.read(2)
    if len(rec) != 2:
        break
    (data, ) = unpack('>H', rec)
    compressed_data.append(data)
'''
# Building and initializing the dictionary.
dictionary_size = 256
#dictionary = dict([(x, chr(x)) for x in range(dictionary_size)])
dictionary = {i: chr(i) for i in range(dictionary_size)}

# iterating through the codes.
# LZW Decompression algorithm
for code in compressed_data:
    if not (code in dictionary):
        #print(code)
        dictionary[code] = string + (string[0])
    decompressed_data += dictionary[code]
    if not(len(string) == 0):
        dictionary[next_code] = string + (dictionary[code][0])
        next_code += 1
        
    string = dictionary[code]

print(decompressed_data)
'''
# storing the decompressed string into a file.
out = input_file.split(".")[0]
output_file = open(out + "_decoded.txt", "w")
for data in decompressed_data:
    output_file.write(data)
    
output_file.close()
file.close()
'''