mirror of
https://github.com/vasi/pixz
synced 2024-11-16 21:26:04 +00:00
Use dynamic block sizes
This commit is contained in:
parent
1505bc8c86
commit
dedbc6d23f
43
read.c
43
read.c
@ -25,6 +25,7 @@ static void wanted_free(wanted_t *w);
|
||||
|
||||
typedef struct {
|
||||
uint8_t *input, *output;
|
||||
size_t incap, outcap;
|
||||
size_t insize, outsize;
|
||||
off_t uoffset; // uncompressed offset
|
||||
} io_block_t;
|
||||
@ -52,9 +53,8 @@ static void tar_write_last(void);
|
||||
#pragma mark DECLARE UTILS
|
||||
|
||||
static lzma_vli gFileIndexOffset = 0;
|
||||
static size_t gBlockInSize = 0, gBlockOutSize = 0;
|
||||
|
||||
static void set_block_sizes(void);
|
||||
static void check_capacity(io_block_t *ib, size_t incap, size_t outcap);
|
||||
|
||||
|
||||
#pragma mark MAIN
|
||||
@ -64,7 +64,6 @@ void pixz_read(bool verify, size_t nspecs, char **specs) {
|
||||
if (verify)
|
||||
gFileIndexOffset = read_file_index(0);
|
||||
wanted_files(nspecs, specs);
|
||||
set_block_sizes();
|
||||
|
||||
#if DEBUG
|
||||
for (wanted_t *w = gWantedFiles; w; w = w->next)
|
||||
@ -135,8 +134,8 @@ void pixz_read(bool verify, size_t nspecs, char **specs) {
|
||||
|
||||
static void *block_create(void) {
|
||||
io_block_t *ib = malloc(sizeof(io_block_t));
|
||||
ib->input = malloc(gBlockInSize);
|
||||
ib->output = malloc(gBlockOutSize);
|
||||
ib->incap = ib->outcap = 0;
|
||||
ib->input = ib->output = NULL;
|
||||
return ib;
|
||||
}
|
||||
|
||||
@ -150,25 +149,6 @@ static void block_free(void* data) {
|
||||
|
||||
#pragma mark SETUP
|
||||
|
||||
static void set_block_sizes() {
|
||||
lzma_index_iter iter;
|
||||
lzma_index_iter_init(&iter, gIndex);
|
||||
while (!lzma_index_iter_next(&iter, LZMA_INDEX_ITER_BLOCK)) {
|
||||
// exclude the file index block
|
||||
lzma_vli off = iter.block.compressed_file_offset;
|
||||
if (gFileIndexOffset && off == gFileIndexOffset)
|
||||
continue;
|
||||
|
||||
size_t in = iter.block.total_size,
|
||||
out = iter.block.uncompressed_size;
|
||||
if (out > gBlockOutSize)
|
||||
gBlockOutSize = out;
|
||||
if (in > gBlockInSize)
|
||||
gBlockInSize = in;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
static void wanted_free(wanted_t *w) {
|
||||
for (wanted_t *w = gWantedFiles; w; ) {
|
||||
wanted_t *tmp = w->next;
|
||||
@ -244,6 +224,17 @@ static void wanted_files(size_t count, char **specs) {
|
||||
|
||||
#pragma mark THREADS
|
||||
|
||||
static void check_capacity(io_block_t *ib, size_t incap, size_t outcap) {
|
||||
if (incap > ib->incap) {
|
||||
ib->incap = incap;
|
||||
ib->input = malloc(incap);
|
||||
}
|
||||
if (outcap > ib->outcap) {
|
||||
ib->outcap = outcap;
|
||||
ib->output = malloc(outcap);
|
||||
}
|
||||
}
|
||||
|
||||
static void read_thread(void) {
|
||||
off_t offset = ftello(gInFile);
|
||||
wanted_t *w = gWantedFiles;
|
||||
@ -273,6 +264,8 @@ static void read_thread(void) {
|
||||
pipeline_item_t *pi;
|
||||
queue_pop(gPipelineStartQ, (void**)&pi);
|
||||
io_block_t *ib = (io_block_t*)(pi->data);
|
||||
check_capacity(ib, iter.block.unpadded_size,
|
||||
iter.block.uncompressed_size);
|
||||
|
||||
// Seek if needed, and get the data
|
||||
if (offset != boffset) {
|
||||
@ -310,7 +303,7 @@ static void decode_thread(size_t thnum) {
|
||||
|
||||
stream.avail_in = ib->insize - block.header_size;
|
||||
stream.next_in = ib->input + block.header_size;
|
||||
stream.avail_out = gBlockOutSize;
|
||||
stream.avail_out = ib->outcap;
|
||||
stream.next_out = ib->output;
|
||||
|
||||
lzma_ret err = LZMA_OK;
|
||||
|
Loading…
Reference in New Issue
Block a user