29 #define LONG_BITSTREAM_READER
47 for (
i = 0;
i < 64;
i++)
48 dst[
i] = permutation[
src[
i]];
51 #define ALPHA_SHIFT_16_TO_10(alpha_val) (alpha_val >> 6)
52 #define ALPHA_SHIFT_8_TO_10(alpha_val) ((alpha_val << 2) | (alpha_val >> 6))
53 #define ALPHA_SHIFT_16_TO_12(alpha_val) (alpha_val >> 4)
54 #define ALPHA_SHIFT_8_TO_12(alpha_val) ((alpha_val << 4) | (alpha_val >> 4))
57 const int num_bits,
const int decode_precision) {
58 const int mask = (1 << num_bits) - 1;
59 int i, idx,
val, alpha_val;
75 alpha_val = (alpha_val +
val) &
mask;
77 if (decode_precision == 10) {
83 if (decode_precision == 10) {
89 if (idx >= num_coeffs)
95 if (idx +
val > num_coeffs)
96 val = num_coeffs - idx;
98 for (
i = 0;
i <
val;
i++) {
99 if (decode_precision == 10) {
106 for (
i = 0;
i <
val;
i++) {
107 if (decode_precision == 10) {
114 }
while (idx < num_coeffs);
120 if (num_bits == 16) {
130 if (num_bits == 16) {
146 case MKTAG(
'a',
'p',
'c',
'o'):
149 case MKTAG(
'a',
'p',
'c',
's'):
152 case MKTAG(
'a',
'p',
'c',
'n'):
155 case MKTAG(
'a',
'p',
'c',
'h'):
158 case MKTAG(
'a',
'p',
'4',
'h'):
162 case MKTAG(
'a',
'p',
'4',
'x'):
172 av_log(avctx,
AV_LOG_DEBUG,
"Auto bitdepth precision. Use 10b decoding based on codec tag.\n");
174 av_log(avctx,
AV_LOG_DEBUG,
"Auto bitdepth precision. Use 12b decoding based on codec tag.\n");
185 ctx->prodsp.idct_permutation_type);
209 ff_dlog(avctx,
"header size %d\n", hdr_size);
210 if (hdr_size > data_size) {
234 ctx->frame_type = (buf[12] >> 2) & 3;
235 ctx->alpha_info = buf[17] & 0xf;
237 if (
ctx->alpha_info > 2) {
243 ff_dlog(avctx,
"frame type %d\n",
ctx->frame_type);
245 if (
ctx->frame_type == 0) {
246 ctx->scan =
ctx->progressive_scan;
248 ctx->scan =
ctx->interlaced_scan;
249 ctx->frame->interlaced_frame = 1;
250 ctx->frame->top_field_first =
ctx->frame_type == 1;
253 if (
ctx->alpha_info) {
277 if(buf + data_size - ptr < 64) {
284 memset(
ctx->qmat_luma, 4, 64);
288 if(buf + data_size - ptr < 64) {
292 permute(
ctx->qmat_chroma,
ctx->prodsp.idct_permutation, ptr);
294 memcpy(
ctx->qmat_chroma,
ctx->qmat_luma, 64);
303 int i, hdr_size, slice_count;
304 unsigned pic_data_size;
305 int log2_slice_mb_width, log2_slice_mb_height;
306 int slice_mb_count, mb_x, mb_y;
307 const uint8_t *data_ptr, *index_ptr;
309 hdr_size = buf[0] >> 3;
310 if (hdr_size < 8 || hdr_size > buf_size) {
315 pic_data_size =
AV_RB32(buf + 1);
316 if (pic_data_size > buf_size) {
321 log2_slice_mb_width = buf[7] >> 4;
322 log2_slice_mb_height = buf[7] & 0xF;
323 if (log2_slice_mb_width > 3 || log2_slice_mb_height) {
325 1 << log2_slice_mb_width, 1 << log2_slice_mb_height);
329 ctx->mb_width = (avctx->
width + 15) >> 4;
331 ctx->mb_height = (avctx->
height + 31) >> 5;
333 ctx->mb_height = (avctx->
height + 15) >> 4;
337 slice_count =
ctx->mb_height * ((
ctx->mb_width >> log2_slice_mb_width) +
340 if (
ctx->slice_count != slice_count || !
ctx->slices) {
342 ctx->slice_count = 0;
346 ctx->slice_count = slice_count;
352 if (hdr_size + slice_count*2 > buf_size) {
358 index_ptr = buf + hdr_size;
359 data_ptr = index_ptr + slice_count*2;
361 slice_mb_count = 1 << log2_slice_mb_width;
365 for (
i = 0;
i < slice_count;
i++) {
368 slice->
data = data_ptr;
369 data_ptr +=
AV_RB16(index_ptr +
i*2);
371 while (
ctx->mb_width - mb_x < slice_mb_count)
372 slice_mb_count >>= 1;
384 mb_x += slice_mb_count;
385 if (mb_x ==
ctx->mb_width) {
386 slice_mb_count = 1 << log2_slice_mb_width;
390 if (data_ptr > buf + buf_size) {
396 if (mb_x || mb_y !=
ctx->mb_height) {
398 mb_y,
ctx->mb_height);
402 return pic_data_size;
405 #define DECODE_CODEWORD(val, codebook, SKIP) \
407 unsigned int rice_order, exp_order, switch_bits; \
408 unsigned int q, buf, bits; \
410 UPDATE_CACHE(re, gb); \
411 buf = GET_CACHE(re, gb); \
414 switch_bits = codebook & 3; \
415 rice_order = codebook >> 5; \
416 exp_order = (codebook >> 2) & 7; \
418 q = 31 - av_log2(buf); \
420 if (q > switch_bits) { \
421 bits = exp_order - switch_bits + (q<<1); \
422 if (bits > FFMIN(MIN_CACHE_BITS, 31)) \
423 return AVERROR_INVALIDDATA; \
424 val = SHOW_UBITS(re, gb, bits) - (1 << exp_order) + \
425 ((switch_bits + 1) << rice_order); \
426 SKIP(re, gb, bits); \
427 } else if (rice_order) { \
428 SKIP_BITS(re, gb, q+1); \
429 val = (q << rice_order) + SHOW_UBITS(re, gb, rice_order); \
430 SKIP(re, gb, rice_order); \
437 #define TOSIGNED(x) (((x) >> 1) ^ (-((x) & 1)))
439 #define FIRST_DC_CB 0xB8
444 int blocks_per_slice)
459 for (
i = 1;
i < blocks_per_slice;
i++,
out += 64) {
463 prev_dc += (((
code + 1) >> 1) ^ sign) - sign;
471 static const uint8_t run_to_cb[16] = { 0x06, 0x06, 0x05, 0x05, 0x04, 0x29, 0x29, 0x29, 0x29, 0x28, 0x28, 0x28, 0x28, 0x28, 0x28, 0x4C };
472 static const uint8_t lev_to_cb[10] = { 0x04, 0x0A, 0x05, 0x06, 0x04, 0x28, 0x28, 0x28, 0x28, 0x4C };
475 int16_t *
out,
int blocks_per_slice)
478 int block_mask, sign;
480 int max_coeffs,
i, bits_left;
481 int log2_block_count =
av_log2(blocks_per_slice);
488 max_coeffs = 64 << log2_block_count;
489 block_mask = blocks_per_slice - 1;
491 for (
pos = block_mask;;) {
493 if (bits_left <= 0 || (bits_left < 32 && !
SHOW_UBITS(
re, gb, bits_left)))
498 if (
pos >= max_coeffs) {
506 i =
pos >> log2_block_count;
510 out[((
pos & block_mask) << 6) +
ctx->scan[
i]] = ((
level ^ sign) - sign);
518 uint16_t *dst,
int dst_stride,
519 const uint8_t *buf,
unsigned buf_size,
526 int i, blocks_per_slice = slice->
mb_count<<2;
529 for (
i = 0;
i < blocks_per_slice;
i++)
530 ctx->bdsp.clear_block(blocks+(
i<<6));
541 ctx->prodsp.idct_put(dst, dst_stride,
block+(0<<6), qmat);
542 ctx->prodsp.idct_put(dst +8, dst_stride,
block+(1<<6), qmat);
543 ctx->prodsp.idct_put(dst+4*dst_stride , dst_stride,
block+(2<<6), qmat);
544 ctx->prodsp.idct_put(dst+4*dst_stride+8, dst_stride,
block+(3<<6), qmat);
552 uint16_t *dst,
int dst_stride,
553 const uint8_t *buf,
unsigned buf_size,
554 const int16_t *qmat,
int log2_blocks_per_mb)
560 int i, j, blocks_per_slice = slice->
mb_count << log2_blocks_per_mb;
563 for (
i = 0;
i < blocks_per_slice;
i++)
564 ctx->bdsp.clear_block(blocks+(
i<<6));
575 for (j = 0; j < log2_blocks_per_mb; j++) {
576 ctx->prodsp.idct_put(dst, dst_stride,
block+(0<<6), qmat);
577 ctx->prodsp.idct_put(dst+4*dst_stride, dst_stride,
block+(1<<6), qmat);
589 uint16_t *dst,
int dst_stride,
590 const uint8_t *buf,
int buf_size,
591 int blocks_per_slice)
598 for (
i = 0;
i < blocks_per_slice<<2;
i++)
599 ctx->bdsp.clear_block(blocks+(
i<<6));
603 if (
ctx->alpha_info == 2) {
604 ctx->unpack_alpha(&gb, blocks, blocks_per_slice * 4 * 64, 16);
606 ctx->unpack_alpha(&gb, blocks, blocks_per_slice * 4 * 64, 8);
611 for (
i = 0;
i < 16;
i++) {
612 memcpy(dst,
block, 16 * blocks_per_slice *
sizeof(*dst));
613 dst += dst_stride >> 1;
614 block += 16 * blocks_per_slice;
624 int i, hdr_size, qscale, log2_chroma_blocks_per_mb;
625 int luma_stride, chroma_stride;
626 int y_data_size, u_data_size, v_data_size, a_data_size,
offset;
627 uint8_t *dest_y, *dest_u, *dest_v;
632 uint16_t val_no_chroma;
639 hdr_size = buf[0] >> 3;
640 qscale =
av_clip(buf[1], 1, 224);
641 qscale = qscale > 128 ? qscale - 96 << 2: qscale;
642 y_data_size =
AV_RB16(buf + 2);
643 u_data_size =
AV_RB16(buf + 4);
644 v_data_size = slice->
data_size - y_data_size - u_data_size - hdr_size;
645 if (hdr_size > 7) v_data_size =
AV_RB16(buf + 6);
646 a_data_size = slice->
data_size - y_data_size - u_data_size -
647 v_data_size - hdr_size;
649 if (y_data_size < 0 || u_data_size < 0 || v_data_size < 0
650 || hdr_size+y_data_size+u_data_size+v_data_size > slice->
data_size){
657 for (
i = 0;
i < 64;
i++) {
658 qmat_luma_scaled [
i] =
ctx->qmat_luma [
i] * qscale;
659 qmat_chroma_scaled[
i] =
ctx->qmat_chroma[
i] * qscale;
662 if (
ctx->frame_type == 0) {
666 luma_stride = pic->
linesize[0] << 1;
667 chroma_stride = pic->
linesize[1] << 1;
673 log2_chroma_blocks_per_mb = 2;
676 log2_chroma_blocks_per_mb = 1;
681 dest_u = pic->
data[1] + (slice->
mb_y << 4) * chroma_stride + (slice->
mb_x << mb_x_shift);
682 dest_v = pic->
data[2] + (slice->
mb_y << 4) * chroma_stride + (slice->
mb_x << mb_x_shift);
684 if (
ctx->frame_type &&
ctx->first_field ^
ctx->frame->top_field_first) {
692 buf, y_data_size, qmat_luma_scaled);
698 buf + y_data_size, u_data_size,
699 qmat_chroma_scaled, log2_chroma_blocks_per_mb);
704 buf + y_data_size + u_data_size, v_data_size,
705 qmat_chroma_scaled, log2_chroma_blocks_per_mb);
710 size_t mb_max_x = slice->
mb_count << (mb_x_shift - 1);
715 val_no_chroma = 511 * 4;
717 for (
i = 0;
i < 16; ++
i)
718 for (j = 0; j < mb_max_x; ++j) {
719 *(uint16_t*)(dest_u + (
i * chroma_stride) + (j << 1)) = val_no_chroma;
720 *(uint16_t*)(dest_v + (
i * chroma_stride) + (j << 1)) = val_no_chroma;
725 if (
ctx->alpha_info && pic->
data[3] && a_data_size) {
728 buf + y_data_size + u_data_size + v_data_size,
744 for (
i = 0;
i <
ctx->slice_count;
i++)
749 if (error < ctx->slice_count)
752 return ctx->slices[0].ret;
762 int buf_size = avpkt->
size;
763 int frame_hdr_size, pic_size, ret;
772 ctx->frame->key_frame = 1;
773 ctx->first_field = 1;
779 if (frame_hdr_size < 0)
780 return frame_hdr_size;
782 buf += frame_hdr_size;
783 buf_size -= frame_hdr_size;
792 if (
ctx->first_field)
802 buf_size -= pic_size;
804 if (
ctx->frame_type && buf_size > 0 &&
ctx->first_field) {
805 ctx->first_field = 0;
static double val(void *priv, double ch)
Libavcodec external API header.
#define FF_PROFILE_PRORES_STANDARD
#define FF_PROFILE_UNKNOWN
#define FF_PROFILE_PRORES_PROXY
#define FF_PROFILE_PRORES_4444
#define FF_PROFILE_PRORES_LT
#define FF_PROFILE_PRORES_XQ
#define FF_PROFILE_PRORES_HQ
static av_cold int init(AVCodecContext *avctx)
#define flags(name, subs,...)
#define MKTAG(a, b, c, d)
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
#define FF_DECODE_ERROR_INVALID_BITSTREAM
bitstream reader API header.
#define CLOSE_READER(name, gb)
static int get_bits_left(GetBitContext *gb)
#define SHOW_UBITS(name, gb, num)
static unsigned int get_bits1(GetBitContext *s)
#define SHOW_SBITS(name, gb, num)
#define OPEN_READER(name, gb)
#define SKIP_BITS(name, gb, num)
#define UPDATE_CACHE(name, gb)
#define LAST_SKIP_BITS(name, gb, num)
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
#define AV_CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
#define AV_LOG_WARNING
Something somehow does not look correct.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
void * av_mallocz_array(size_t nmemb, size_t size)
Allocate a memory block for an array with av_mallocz().
@ AV_PICTURE_TYPE_I
Intra.
av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation, enum idct_permutation_type perm_type)
av_cold void ff_blockdsp_init(BlockDSPContext *c, AVCodecContext *avctx)
int ff_set_dimensions(AVCodecContext *s, int width, int height)
Check that the provided frame dimensions are valid and set them on the codec context.
common internal API header
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
static const AVProfile profiles[]
static const uint16_t mask[17]
#define LOCAL_ALIGNED_32(t, v,...)
#define LOCAL_ALIGNED_16(t, v,...)
#define AV_PIX_FMT_YUV444P12
@ AVCOL_RANGE_MPEG
Narrow or limited range content.
#define AV_PIX_FMT_YUVA444P10
#define AV_PIX_FMT_YUV422P12
#define AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUVA422P10
#define AV_PIX_FMT_YUVA422P12
#define AV_PIX_FMT_YUVA444P12
#define AV_PIX_FMT_YUV444P10
const AVProfile ff_prores_profiles[]
const uint8_t ff_prores_progressive_scan[64]
const uint8_t ff_prores_interlaced_scan[64]
#define ALPHA_SHIFT_16_TO_12(alpha_val)
static av_always_inline int decode_dc_coeffs(GetBitContext *gb, int16_t *out, int blocks_per_slice)
#define DECODE_CODEWORD(val, codebook, SKIP)
static av_always_inline int decode_ac_coeffs(AVCodecContext *avctx, GetBitContext *gb, int16_t *out, int blocks_per_slice)
static void unpack_alpha_10(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits)
static void decode_slice_alpha(ProresContext *ctx, uint16_t *dst, int dst_stride, const uint8_t *buf, int buf_size, int blocks_per_slice)
Decode alpha slice plane.
static av_cold int decode_close(AVCodecContext *avctx)
static const uint8_t run_to_cb[16]
static void unpack_alpha_12(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits)
static av_cold int decode_init(AVCodecContext *avctx)
AVCodec ff_prores_decoder
static int decode_picture(AVCodecContext *avctx)
#define ALPHA_SHIFT_16_TO_10(alpha_val)
static const uint8_t dc_codebook[7]
static void unpack_alpha(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits, const int decode_precision)
static int decode_slice_luma(AVCodecContext *avctx, SliceContext *slice, uint16_t *dst, int dst_stride, const uint8_t *buf, unsigned buf_size, const int16_t *qmat)
static int decode_frame_header(ProresContext *ctx, const uint8_t *buf, const int data_size, AVCodecContext *avctx)
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
static const uint8_t lev_to_cb[10]
static void permute(uint8_t *dst, const uint8_t *src, const uint8_t permutation[64])
static int decode_slice_thread(AVCodecContext *avctx, void *arg, int jobnr, int threadnr)
#define ALPHA_SHIFT_8_TO_10(alpha_val)
static int decode_slice_chroma(AVCodecContext *avctx, SliceContext *slice, uint16_t *dst, int dst_stride, const uint8_t *buf, unsigned buf_size, const int16_t *qmat, int log2_blocks_per_mb)
static int decode_picture_header(AVCodecContext *avctx, const uint8_t *buf, const int buf_size)
#define ALPHA_SHIFT_8_TO_12(alpha_val)
av_cold int ff_proresdsp_init(ProresDSPContext *dsp, AVCodecContext *avctx)
FF_ENABLE_DEPRECATION_WARNINGS int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
main external API structure.
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
int width
picture width / height.
enum AVColorRange color_range
MPEG vs JPEG YUV range.
unsigned int codec_tag
fourcc (LSB first, so "ABCD" -> ('D'<<24) + ('C'<<16) + ('B'<<8) + 'A').
enum AVColorPrimaries color_primaries
Chromaticity coordinates of the source primaries.
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
enum AVColorSpace colorspace
YUV colorspace type.
int skip_alpha
Skip processing alpha if supported by codec.
enum AVColorTransferCharacteristic color_trc
Color Transfer Characteristic.
int flags
AV_CODEC_FLAG_*.
int(* execute2)(struct AVCodecContext *c, int(*func)(struct AVCodecContext *c2, void *arg, int jobnr, int threadnr), void *arg2, int *ret, int count)
The codec may call this to execute several independent things.
const char * name
Name of the codec implementation.
This structure describes decoded (raw) audio or video data.
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
This structure stores compressed data.
static void error(const char *err)
static const uint8_t offset[127][2]