2011-06-26 13 views
16

Potrzebuję kompresji i dekompresji LZMA w locie. Otrzymuję duży plik za pośrednictwem qnetworkmanager w Qt i muszę go rozpakować podczas pobierania strumienia danych.Kompresja C++ lzma i dekompresja dużego strumienia przez części

Kiedy otrzymuję część strumienia danych, muszę ją rozpakować, dołączyć do pliku, a następnie zwolnić pamięć używaną podczas procesu. Jaki jest najlepszy sposób na zrobienie tego?

Teraz próbuję z xz-utils, czystym cappi, może ktoś może zaproponować lepszy sposób?

następujący kod na podstawie this przykład

UPD2:

extern "C" void *lz_alloc(void *opaque, size_t nmemb, size_t size) 
{ 
void *p = NULL; 
try{ 
    p = new char [size]; 
} 
catch(std::bad_alloc &ba) 
{ 
    p = NULL; 
} 
return p; 
} 

extern "C" void lz_free(void *opaque, void *ptr) 
{ 
delete [] (char*)ptr; 
} 


QByteArray lzCompress(QByteArray data) 
{ 
QByteArray arr; 
lzma_check check = LZMA_CHECK_CRC64; 
lzma_stream strm = LZMA_STREAM_INIT; /* alloc and init lzma_stream struct */ 
lzma_allocator al; 
al.alloc = lz_alloc; 
al.free = lz_free; 
strm.allocator = &al; 
byte *in_buf; 
byte out_buf [OUT_BUF_MAX]; 
size_t in_len; /* length of useful data in in_buf */ 
size_t out_len; /* length of useful data in out_buf */ 
lzma_ret ret_xz; 

/* initialize xz encoder */ 
ret_xz = lzma_easy_encoder (&strm, 9 | LZMA_PRESET_EXTREME, check); 
if (ret_xz != LZMA_OK) { 
    return QByteArray(); 
} 

in_len = data.size(); 
in_buf = (byte*)data.data(); 
strm.next_in = in_buf; 
strm.avail_in = in_len; 

do { 
    strm.next_out = out_buf; 
    strm.avail_out = OUT_BUF_MAX; 
    ret_xz = lzma_code (&strm, LZMA_FINISH); 

    out_len = OUT_BUF_MAX - strm.avail_out; 
    arr.append((char*)out_buf, out_len); 
    out_buf[0] = 0; 
} while (strm.avail_out == 0); 
lzma_end (&strm); 
return arr; 
} 

mam sleeped kilka godzin, a teraz myślę jaśniej, stały mój zły kod, aktualizowany go (to zachowują się tak samo qCompress działa)

UPD3:

kod dekompresji (qUncompress jak zachowanie)

QByteArray lzUncompress(QByteArray data) 
{ 
lzma_stream strm = LZMA_STREAM_INIT; /* alloc and init lzma_stream struct */ 
const uint32_t flags = LZMA_TELL_UNSUPPORTED_CHECK | LZMA_CONCATENATED; 
const uint64_t memory_limit = UINT64_MAX; /* no memory limit */ 
byte *in_buf; 
uint8_t out_buf [OUT_BUF_MAX]; 
size_t in_len; /* length of useful data in in_buf */ 
size_t out_len; /* length of useful data in out_buf */ 
lzma_ret ret_xz; 
QByteArray arr; 

ret_xz = lzma_stream_decoder (&strm, memory_limit, flags); 
if (ret_xz != LZMA_OK) { 
    return QByteArray(); 
} 

in_len = data.size(); 
in_buf = (byte*)data.data(); 

strm.next_in = in_buf; 
strm.avail_in = in_len; 
do { 
    strm.next_out = out_buf; 
    strm.avail_out = OUT_BUF_MAX; 
    ret_xz = lzma_code (&strm, LZMA_FINISH); 

    out_len = OUT_BUF_MAX - strm.avail_out; 
    arr.append((char*)out_buf, out_len); 
    out_buf[0] = 0; 
} while (strm.avail_out == 0); 
lzma_end (&strm); 
return arr; 
} 

UPD4:

podstawowa klasa strumień dekompresji, po kodzie tylko rozpakować strumień xz pobranego z serwera http na bieżąco, co dokładnie muszę:

class lz_stream_decompressor : public QObject 
{ 
Q_OBJECT 
public: 
lz_stream_decompressor(QNetworkReply *r, QNetworkAccessManager *q, const QString &str, unsigned long sz): flags(LZMA_TELL_UNSUPPORTED_CHECK | LZMA_CONCATENATED), memory_limit(UINT64_MAX), state(0), total_upd_size(sz) 
{ 
    repl = r; 
    qnm = q; 
    path = str; 
    strm.next_in = NULL; 
    strm.avail_in = 0; 
    strm.total_in = 0; 
    strm.next_out = NULL; 
    strm.avail_out = 0; 
    strm.total_out = 0; 
    strm.allocator = NULL; 
    strm.internal = NULL; 
    strm.reserved_ptr1 = NULL; 
    strm.reserved_ptr2 = NULL; 
    strm.reserved_ptr3 = NULL; 
    strm.reserved_ptr4 = NULL; 
    strm.reserved_int1 = 0; 
    strm.reserved_int2 = 0; 
    strm.reserved_int3 = 0; 
    strm.reserved_int4 = 0; 
    strm.reserved_enum1 = LZMA_RESERVED_ENUM; 
    strm.reserved_enum2 = LZMA_RESERVED_ENUM; 
    ret_xz = lzma_stream_decoder (&strm, memory_limit, flags); 
    if (ret_xz != LZMA_OK) 
    { 
     state = -1; 
     repl->abort(); 
    } 
    else 
    { 
     connect(repl, SIGNAL(downloadProgress(qint64,qint64)), SLOT(handle_new_data(qint64,qint64))); 
     connect(q, SIGNAL(finished(QNetworkReply*)), SLOT(compressed_file_request_finished(QNetworkReply*))); 
     QFile(path).rename(path + ".tmp"); 
    } 
} 
~lz_stream_decompressor() 
{ 
/*  if(repl) 
     delete repl; */ 
    lzma_end (&strm); 
} 
const short get_state() 
{ 
    return state; 
} 
signals: 
void finished(); 

public slots: 
void handle_new_data(qint64 bytesReceived, qint64 bytesTotal); 
void compressed_file_request_finished(QNetworkReply*); 
private: 
QNetworkReply *repl; 
QNetworkAccessManager *qnm; 
lzma_stream strm; 
const uint32_t flags; 
const uint64_t memory_limit; /* no memory limit */ 
short state; 
byte *in_buf; 
byte out_buf [OUT_BUF_MAX]; 
size_t in_len; /* length of useful data in in_buf */ 
size_t out_len; /* length of useful data in out_buf */ 
lzma_ret ret_xz; 
QString path; 
unsigned long &total_upd_size; 
}; 

i realizacja:

void lz_stream_decompressor::handle_new_data(qint64 bytesReceived, qint64 bytesTotal) 
{ 
if(repl->error() != QNetworkReply::NoError) 
{//TODO: handle error here 
    QFile(path).remove(); 
    QFile(path + ".tmp").rename(path); 
    return; 
} 
total_upd_size -= repl->bytesAvailable(); 
QByteArray data = repl->readAll(); 
in_len = data.size(); 
in_buf = (byte*)data.data(); 
strm.next_in = in_buf; 
strm.avail_in = in_len; 

do { 
    strm.next_out = out_buf; 
    strm.avail_out = OUT_BUF_MAX; 
    ret_xz = lzma_code (&strm, LZMA_RUN); 
    out_len = OUT_BUF_MAX - strm.avail_out; 
    QFile file(path); 
    if(file.open(QIODevice::WriteOnly | QIODevice::Append)) 
    { 
     file.write(QByteArray((char*)out_buf, (int)out_len)); 
     file.close(); 
    } 
    out_buf[0] = 0; 
} while (strm.avail_out == 0); 
} 

void lz_stream_decompressor::compressed_file_request_finished(QNetworkReply* repl) 
{ 
if(repl->error() != QNetworkReply::NoError) 
{//TODO: handle error here 
    QFile(path).remove(); 
    QFile(path + ".tmp").rename(path); 
    emit finished(); 
    return; 
} 
total_upd_size -= repl->bytesAvailable(); 
QByteArray data = repl->readAll(); 
in_len = data.size(); 
in_buf = (byte*)data.data(); 
strm.next_in = in_buf; 
strm.avail_in = in_len; 

do { 
    strm.next_out = out_buf; 
    strm.avail_out = OUT_BUF_MAX; 
    ret_xz = lzma_code (&strm, LZMA_FINISH); 
    out_len = OUT_BUF_MAX - strm.avail_out; 
    QFile file(path); 
    if(file.open(QIODevice::WriteOnly | QIODevice::Append)) 
    { 
     file.write(QByteArray((char*)out_buf, (int)out_len)); 
     file.close(); 
    } 
    out_buf[0] = 0; 
} while (strm.avail_out == 0); 
repl->deleteLater(); 
QFile(path + ".tmp").remove(); 
emit finished(); 
} 

wszystko to na przykładzie z pierwszego linku, musisz zastąpić skomentowane części kodu swoim kodem, aby zrobić coś z nieskompresowanymi danymi.

chciałbym zobaczyć jakieś sugestie do tego kodu

trzeba także połączyć „compressed_file_request_finished” gniazdo do sygnału qnetworkmanager zakończone do końca nieskompresowane dane.

UPD5:

stałe lzCompress i lzUncompress, wygląda działa dobrze teraz, nie wiem na temat korzystania LZMA_FULL_FLUSH w handle_new_data, jak czytam to co muszę, ale nadal nie jest pewny, teraz dostosowania istniejącego kodu aby skorzystać z tej ...

UPD6:

trzeba też coś takiego:

/* read/write buffer sizes */ 
#define IN_BUF_MAX 409600 
#define OUT_BUF_MAX 409600 
/* analogous to xz CLI options: -0 to -9 */ 
#define COMPRESSION_LEVEL 7 

/* boolean setting, analogous to xz CLI option: -e */ 
#define COMPRESSION_EXTREME true 

w zakresie widoczności umożliwiającym działanie tego kodu.

UPD7:

zaktualizowany kod, wszystkie przetestowane i działa, i odkryli, że liblzma nie całkowicie bezpieczny wątku, próbowałem zrobić wielowątkowe kompresji listy plików. i bardzo często się zawiesza.

+1

Dzięki za wiele aktualizacji i kompletne rozwiązanie! –

+0

Czy udało Ci się to zrobić za pomocą LZMA? –

Odpowiedz

7

Na tej stronie znajdziesz SDK lzma który zapewnia kodów źródłowych w różnych językach, a niektóre pliki binarne: http://www.7-zip.org/sdk.html

Masz dwa rozwiązania:

  • użyć kodu źródłowego C++ do dekompresji dochodzące Przepływ
  • Użyj plików binarnych dekodera jako narzędzia zewnętrznego w swojej aplikacji
+0

thx dla informacji, obecnie znalazłem inne rozwiązanie, pisząc klasę dekompresora strumieniowego na podstawie qnetworkmanager z qt i liblzma z xz-utils (przykłady kodu z xz-utils wygląda na łatwiejsze dla mnie), będę umieszczać kod tutaj, kiedy zrobię . – sss123next

+0

@Patrice Bernassola jakieś sugestie? – sss123next

+0

Czy masz problem z kodem, który opublikowałeś? –