VPF::[Python] Сохранение данных, сжатых алгоритмом LZW

Hirurg2605

Дата 10.1.2012, 17:56 (ссылка)

(нет голосов)

Загрузка ...

Новичок

Профиль
Группа: Участник
Сообщений: 6
Регистрация: 10.1.2012

Репутация: нет
Всего: нет

Добрый вечер! При выполнении лабораторной работы у меня возникла проблема с сохранением данных, сжатых алгоритмом LZW. Код алгоритма был найден мной в интернете и в его работе я разобрался, но в моём задании надо на выходе получать файл, а алгоритм выдаёт только список чисел. Если кто-нибудь сталкивался с решением этой задачи, подскажите пожалуйста как правильно сформировать выходной файл. Можно ли записать эти числа в файл построчно или надо как-то побитово заполнять файл. Прошу помочь мне в реализации побитовой записи, т.к. по этой части я совсем чайник smile

Вот то, что у меня имеется сейчас (с построчной записью кодов в десятичной записи):
Примечание. В 42-ой строке вместо ccode должно стоять code smile

а то в сообщении форматирование сбивалось

Код


#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import os

# The initial LZW tables that maps codes to strings and vice-versa.
code_to_str = [chr(i) for i in range(256)]
str_to_code = dict((chr(i), i) for i in range(256))

def compress(seq):
    '''
    Returns an LZW compressed list of the input character sequence.
    '''
    output = []
    table = dict(str_to_code)

    s = ''
    for ch in seq:
        it = s + ch
        if it in table:
            s = it
        else:
            output.append(table[s])
            table[it] = len(table)
            s = ch
    output.append(table[s])
    return output

def decompress(seq):
    '''
    Returns a decompressed list of the LZW compressed input.
    '''
    table = code_to_str[:]
    prevcode = seq[0]

    output = []
    output.append(table[prevcode])

    for code in seq[1:]:
        try:
            entry = table[cсode]
        except IndexError:
            # The lzw special case when code is not yet defined.
            entry = table[prevcode]
            entry += entry[0]
        output.append(entry)
        table.append(table[prevcode] + entry[0])
        prevcode = code
    return output

# MAIN
if len(sys.argv) != 4:
    print 'Usage: LZW.py [c|d] [path]InputFileName [path]OutputFileName'
    sys.exit()
mode = sys.argv[1] # compress/decompress
inputFile = sys.argv[2]
outputFile = sys.argv[3]

f1 = open(inputFile, "r", 1)
s = "".join([line for line in f1])
f1.close()
print s
f2 = open(outputFile, "w", 1)
compressed = compress(s)
for i in range(len(compressed)):
    print compressed[i]
    f2.write(str(compressed[i]))
f2.close

data = open('uncompressed').read()
print data
data = compress(data)
s = "".join([str(line) for line in data])
#print s
print data
s = "".join([line for line in decompress(data)])
print s
#print decompress(data)

Hirurg2605

Дата 11.1.2012, 10:37 (ссылка)

(нет голосов)

Загрузка ...

Новичок

Профиль
Группа: Участник
Сообщений: 6
Регистрация: 10.1.2012

Репутация: нет
Всего: нет

Вот немного доработанный код smile

а то за вчерашний сегодня стыдно

Код


#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import os
import time

# The initial LZW tables that maps codes to strings and vice-versa.
code_to_str = [chr(i) for i in range(256)]
str_to_code = dict((chr(i), i) for i in range(256))

def compress(seq):
    '''
    Returns an LZW compressed list of the input character sequence.
    '''
    output = []
    table = dict(str_to_code)

    s = ''
    for ch in seq:
        it = s + ch
        if it in table:
            s = it
        else:
            output.append(table[s])
            table[it] = len(table)
            s = ch
    output.append(table[s])
    return output

def decompress(seq):
    '''
    Returns a decompressed list of the LZW compressed input.
    '''
    table = code_to_str[:]
    prevcode = seq[0]

    output = []
    output.append(table[prevcode])

    for code in seq[1:]:
        try:
            entry = table[сcode]        #в ccode убрать лишнюю c (сбивается форматирование сообщения на форуме :-) )
        except IndexError:
            # The lzw special case when code is not yet defined.
            entry = table[prevcode]
            entry += entry[0]
        output.append(entry)
        table.append(table[prevcode] + entry[0])
        prevcode = code
    return output

# MAIN
if len(sys.argv) != 4:
    print 'Usage: LZW.py [c|d] [path]InputFileName [path]OutputFileName'
    sys.exit()
mode = sys.argv[1] # compress/decompress
inputFile = sys.argv[2]
outputFile = sys.argv[3]

fileSize = os.path.getsize(inputFile)
print fileSize

f2 = open(outputFile, "w", 1)

if mode == 'c':
    t = time.time()
    f1 = open(inputFile, "r", 1)
    s = "".join([line for line in f1])
    f1.close()
    print 'Вход:', s
    compressed = compress(s)
    outputList = []
    for i in range(len(compressed)):
        #print compressed[i]
        f2.write(str(compressed[i]) + '\n')
        outputList.append(compressed[i])
    f2.close
    print 'Выход:', outputList
    print 'Время сжатия:', time.time() - t
    fileSize = os.path.getsize(outputFile)
    print fileSize

elif mode == 'd':
    t = time.time()
    f1 = open(inputFile, "r", 1)
    s = []
    for line in f1:
        if len(line) > 0:
            s.append(int(line[:len(line)-1]))
    f1.close()
    print 'Вход:', s
    data = decompress(s)
    s = "".join([line for line in data])
    print 'Выход:', s
    f2.write(s)
    f2.close
    print 'Время декодирования: ', time.time() - t
    fileSize = os.path.getsize(outputFile)
    print fileSize

если ничего не подскажите буду пробовать сдать это smile

0 Пользователей читают эту тему (0 Гостей и 0 Скрытых Пользователей)
0 Пользователей:
« Предыдущая тема \| Центр помощи \| Следующая тема »