diff options
author | Vittorio Giovara <vittorio.giovara@gmail.com> | 2015-01-27 23:56:19 +0300 |
---|---|---|
committer | Vittorio Giovara <vittorio.giovara@gmail.com> | 2015-01-29 18:04:46 +0300 |
commit | 598f7d046cbf306706623210c5baafa3b7cd1df3 (patch) | |
tree | 1d11a206caebdf9e816f7e84aeec08d106df008e /libavcodec/dnxhddec.c | |
parent | 41e03e284ee7b0d4caa3a5d28681ad46e3105d65 (diff) |
DNxHD: Simplify pixel format detection
Error out in case of unknown values.
Diffstat (limited to 'libavcodec/dnxhddec.c')
-rw-r--r-- | libavcodec/dnxhddec.c | 45 |
1 files changed, 20 insertions, 25 deletions
diff --git a/libavcodec/dnxhddec.c b/libavcodec/dnxhddec.c index 4f714d86af..9b1e48158d 100644 --- a/libavcodec/dnxhddec.c +++ b/libavcodec/dnxhddec.c @@ -133,35 +133,30 @@ static int dnxhd_decode_header(DNXHDContext *ctx, AVFrame *frame, av_dlog(ctx->avctx, "width %d, height %d\n", ctx->width, ctx->height); - ctx->is_444 = 0; - if (buf[0x4] == 0x2) { - ctx->avctx->pix_fmt = AV_PIX_FMT_YUV444P10; - ctx->avctx->bits_per_raw_sample = 10; - if (ctx->bit_depth != 10) { - ff_blockdsp_init(&ctx->bdsp, ctx->avctx); - ff_idctdsp_init(&ctx->idsp, ctx->avctx); - ctx->bit_depth = 10; + if (!ctx->bit_depth) { + ff_blockdsp_init(&ctx->bdsp, ctx->avctx); + ff_idctdsp_init(&ctx->idsp, ctx->avctx); + } + if (buf[0x21] == 0x58) { /* 10 bit */ + ctx->bit_depth = ctx->avctx->bits_per_raw_sample = 10; + + if (buf[0x4] == 0x2) { ctx->decode_dct_block = dnxhd_decode_dct_block_10_444; - } - ctx->is_444 = 1; - } else if (buf[0x21] & 0x40) { - ctx->avctx->pix_fmt = AV_PIX_FMT_YUV422P10; - ctx->avctx->bits_per_raw_sample = 10; - if (ctx->bit_depth != 10) { - ff_blockdsp_init(&ctx->bdsp, ctx->avctx); - ff_idctdsp_init(&ctx->idsp, ctx->avctx); - ctx->bit_depth = 10; + ctx->avctx->pix_fmt = AV_PIX_FMT_YUV444P10; + ctx->is_444 = 1; + } else { ctx->decode_dct_block = dnxhd_decode_dct_block_10; + ctx->avctx->pix_fmt = AV_PIX_FMT_YUV422P10; } - } else { + } else if (buf[0x21] == 0x38) { /* 8 bit */ + ctx->bit_depth = ctx->avctx->bits_per_raw_sample = 8; + ctx->avctx->pix_fmt = AV_PIX_FMT_YUV422P; - ctx->avctx->bits_per_raw_sample = 8; - if (ctx->bit_depth != 8) { - ff_blockdsp_init(&ctx->bdsp, ctx->avctx); - ff_idctdsp_init(&ctx->idsp, ctx->avctx); - ctx->bit_depth = 8; - ctx->decode_dct_block = dnxhd_decode_dct_block_8; - } + ctx->decode_dct_block = dnxhd_decode_dct_block_8; + } else { + av_log(ctx->avctx, AV_LOG_ERROR, "invalid bit depth value (%d).\n", + buf[0x21]); + return AVERROR_INVALIDDATA; } cid = AV_RB32(buf + 0x28); |