mirror of https://git.ffmpeg.org/ffmpeg.git
cdxl: fix video decoding for some files
Width is padded for ham encodings too. Signed-off-by: Paul B Mahol <onemda@gmail.com> Signed-off-by: Justin Ruggles <justin.ruggles@gmail.com>
This commit is contained in:
parent
6d10c5bfc0
commit
b5c626fdf3
|
@ -60,27 +60,29 @@ static void import_palette(CDXLVideoContext *c, uint32_t *new_palette)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void bitplanar2chunky(CDXLVideoContext *c, int width,
|
static void bitplanar2chunky(CDXLVideoContext *c, int linesize, uint8_t *out)
|
||||||
int linesize, uint8_t *out)
|
|
||||||
{
|
{
|
||||||
|
int skip = FFALIGN(c->avctx->width, 16) - c->avctx->width;
|
||||||
GetBitContext gb;
|
GetBitContext gb;
|
||||||
int x, y, plane;
|
int x, y, plane;
|
||||||
|
|
||||||
init_get_bits(&gb, c->video, c->video_size * 8);
|
init_get_bits(&gb, c->video, c->video_size * 8);
|
||||||
memset(out, 0, linesize * c->avctx->height);
|
memset(out, 0, linesize * c->avctx->height);
|
||||||
for (plane = 0; plane < c->bpp; plane++)
|
for (plane = 0; plane < c->bpp; plane++) {
|
||||||
for (y = 0; y < c->avctx->height; y++)
|
for (y = 0; y < c->avctx->height; y++) {
|
||||||
for (x = 0; x < width; x++)
|
for (x = 0; x < c->avctx->width; x++)
|
||||||
out[linesize * y + x] |= get_bits1(&gb) << plane;
|
out[linesize * y + x] |= get_bits1(&gb) << plane;
|
||||||
|
skip_bits(&gb, skip);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void cdxl_decode_rgb(CDXLVideoContext *c)
|
static void cdxl_decode_rgb(CDXLVideoContext *c)
|
||||||
{
|
{
|
||||||
uint32_t *new_palette = (uint32_t *)c->frame.data[1];
|
uint32_t *new_palette = (uint32_t *)c->frame.data[1];
|
||||||
int padded_width = FFALIGN(c->avctx->width, 16);
|
|
||||||
|
|
||||||
import_palette(c, new_palette);
|
import_palette(c, new_palette);
|
||||||
bitplanar2chunky(c, padded_width, c->frame.linesize[0], c->frame.data[0]);
|
bitplanar2chunky(c, c->frame.linesize[0], c->frame.data[0]);
|
||||||
}
|
}
|
||||||
|
|
||||||
static void cdxl_decode_ham6(CDXLVideoContext *c)
|
static void cdxl_decode_ham6(CDXLVideoContext *c)
|
||||||
|
@ -94,7 +96,7 @@ static void cdxl_decode_ham6(CDXLVideoContext *c)
|
||||||
out = c->frame.data[0];
|
out = c->frame.data[0];
|
||||||
|
|
||||||
import_palette(c, new_palette);
|
import_palette(c, new_palette);
|
||||||
bitplanar2chunky(c, avctx->width, avctx->width, c->new_video);
|
bitplanar2chunky(c, avctx->width, c->new_video);
|
||||||
|
|
||||||
for (y = 0; y < avctx->height; y++) {
|
for (y = 0; y < avctx->height; y++) {
|
||||||
r = new_palette[0] & 0xFF0000;
|
r = new_palette[0] & 0xFF0000;
|
||||||
|
@ -137,7 +139,7 @@ static void cdxl_decode_ham8(CDXLVideoContext *c)
|
||||||
out = c->frame.data[0];
|
out = c->frame.data[0];
|
||||||
|
|
||||||
import_palette(c, new_palette);
|
import_palette(c, new_palette);
|
||||||
bitplanar2chunky(c, avctx->width, avctx->width, c->new_video);
|
bitplanar2chunky(c, avctx->width, c->new_video);
|
||||||
|
|
||||||
for (y = 0; y < avctx->height; y++) {
|
for (y = 0; y < avctx->height; y++) {
|
||||||
r = new_palette[0] & 0xFF0000;
|
r = new_palette[0] & 0xFF0000;
|
||||||
|
@ -209,16 +211,13 @@ static int cdxl_decode_frame(AVCodecContext *avctx, void *data,
|
||||||
if (w != avctx->width || h != avctx->height)
|
if (w != avctx->width || h != avctx->height)
|
||||||
avcodec_set_dimensions(avctx, w, h);
|
avcodec_set_dimensions(avctx, w, h);
|
||||||
|
|
||||||
|
if (c->video_size < FFALIGN(avctx->width, 16) * avctx->height * c->bpp / 8)
|
||||||
|
return AVERROR_INVALIDDATA;
|
||||||
if (encoding == 0) {
|
if (encoding == 0) {
|
||||||
if (c->video_size < FFALIGN(avctx->width, 16) *
|
|
||||||
avctx->height * c->bpp / 8)
|
|
||||||
return AVERROR_INVALIDDATA;
|
|
||||||
avctx->pix_fmt = PIX_FMT_PAL8;
|
avctx->pix_fmt = PIX_FMT_PAL8;
|
||||||
} else if (encoding == 1 && (c->bpp == 6 || c->bpp == 8)) {
|
} else if (encoding == 1 && (c->bpp == 6 || c->bpp == 8)) {
|
||||||
if (c->palette_size != (1 << (c->bpp - 1)))
|
if (c->palette_size != (1 << (c->bpp - 1)))
|
||||||
return AVERROR_INVALIDDATA;
|
return AVERROR_INVALIDDATA;
|
||||||
if (c->video_size < avctx->width * avctx->height * c->bpp / 8)
|
|
||||||
return AVERROR_INVALIDDATA;
|
|
||||||
avctx->pix_fmt = PIX_FMT_BGR24;
|
avctx->pix_fmt = PIX_FMT_BGR24;
|
||||||
} else {
|
} else {
|
||||||
av_log_ask_for_sample(avctx, "unsupported encoding %d and bpp %d\n",
|
av_log_ask_for_sample(avctx, "unsupported encoding %d and bpp %d\n",
|
||||||
|
|
Loading…
Reference in New Issue