Create rpng_process_t

This commit is contained in:
twinaphex 2015-02-25 09:38:54 +01:00
parent c07bab40ed
commit 4c7325c07c
4 changed files with 21 additions and 10 deletions

View File

@ -90,7 +90,7 @@ static void deinterlace_pass(uint32_t *data, const struct png_ihdr *ihdr,
}
static bool png_reverse_filter(uint32_t *data, const struct png_ihdr *ihdr,
const uint8_t *inflate_buf, size_t inflate_buf_size,
const uint8_t *inflate_buf, struct rpng_process_t *pngp,
const uint32_t *palette)
{
unsigned i, h;
@ -103,7 +103,7 @@ static bool png_reverse_filter(uint32_t *data, const struct png_ihdr *ihdr,
png_pass_geom(ihdr, ihdr->width, ihdr->height, &bpp, &pitch, &pass_size);
if (inflate_buf_size < pass_size)
if (pngp->total_out < pass_size)
return false;
prev_scanline = (uint8_t*)calloc(1, pitch);
@ -184,7 +184,7 @@ end:
static bool png_reverse_filter_adam7(uint32_t *data,
const struct png_ihdr *ihdr,
const uint8_t *inflate_buf, size_t inflate_buf_size,
const uint8_t *inflate_buf, struct rpng_process_t *pngp,
const uint32_t *palette)
{
unsigned pass;
@ -227,21 +227,21 @@ static bool png_reverse_filter_adam7(uint32_t *data,
png_pass_geom(&tmp_ihdr, pass_width,
pass_height, NULL, NULL, &pass_size);
if (pass_size > inflate_buf_size)
if (pass_size > pngp->total_out)
{
free(tmp_data);
return false;
}
if (!png_reverse_filter(tmp_data,
&tmp_ihdr, inflate_buf, pass_size, palette))
&tmp_ihdr, inflate_buf, pngp, palette))
{
free(tmp_data);
return false;
}
inflate_buf += pass_size;
inflate_buf_size -= pass_size;
pngp->total_out -= pass_size;
deinterlace_pass(data,
ihdr, tmp_data, pass_width, pass_height, &passes[pass]);

View File

@ -215,6 +215,7 @@ bool rpng_load_image_argb(const char *path, uint32_t **data,
char header[8];
z_stream stream = {0};
struct rpng_t rpng = {0};
struct rpng_process_t process = {0};
bool ret = true;
*data = NULL;
@ -337,14 +338,16 @@ bool rpng_load_image_argb(const char *path, uint32_t **data,
if (!*data)
GOTO_END_ERROR();
process.total_out = stream.total_out;
if (rpng.ihdr.interlace == 1)
{
if (!png_reverse_filter_adam7(*data,
&rpng.ihdr, rpng.inflate_buf, stream.total_out, rpng.palette))
&rpng.ihdr, rpng.inflate_buf, &process, rpng.palette))
GOTO_END_ERROR();
}
else if (!png_reverse_filter(*data,
&rpng.ihdr, rpng.inflate_buf, stream.total_out, rpng.palette))
&rpng.ihdr, rpng.inflate_buf, &process, rpng.palette))
GOTO_END_ERROR();
end:

View File

@ -257,6 +257,7 @@ bool rpng_nbio_load_image_argb_iterate(uint8_t *buf, struct rpng_t *rpng)
bool rpng_nbio_load_image_argb_process(struct rpng_t *rpng,
uint32_t **data, unsigned *width, unsigned *height)
{
struct rpng_process_t process = {0};
z_stream stream = {0};
if (inflateInit(&stream) != Z_OK)
@ -296,14 +297,16 @@ bool rpng_nbio_load_image_argb_process(struct rpng_t *rpng,
if (!*data)
return false;
process.total_out = stream.total_out;
if (rpng->ihdr.interlace == 1)
{
if (!png_reverse_filter_adam7(*data,
&rpng->ihdr, rpng->inflate_buf, stream.total_out, rpng->palette))
&rpng->ihdr, rpng->inflate_buf, &process, rpng->palette))
return false;
}
else if (!png_reverse_filter(*data,
&rpng->ihdr, rpng->inflate_buf, stream.total_out, rpng->palette))
&rpng->ihdr, rpng->inflate_buf, &process, rpng->palette))
return false;
return true;

View File

@ -60,6 +60,11 @@ struct png_ihdr
uint8_t interlace;
};
struct rpng_process_t
{
size_t total_out;
};
struct rpng_t
{
bool has_ihdr;