29 #define LONG_BITSTREAM_READER 45 for (i = 0; i < 64; i++)
46 dst[i] = permutation[src[i]];
49 #define ALPHA_SHIFT_16_TO_10(alpha_val) (alpha_val >> 6) 50 #define ALPHA_SHIFT_8_TO_10(alpha_val) ((alpha_val << 2) | (alpha_val >> 6)) 51 #define ALPHA_SHIFT_16_TO_12(alpha_val) (alpha_val >> 4) 52 #define ALPHA_SHIFT_8_TO_12(alpha_val) ((alpha_val << 4) | (alpha_val >> 4)) 55 const int num_bits,
const int decode_precision) {
56 const int mask = (1 << num_bits) - 1;
57 int i, idx,
val, alpha_val;
67 val =
get_bits(gb, num_bits == 16 ? 7 : 4);
73 alpha_val = (alpha_val +
val) & mask;
75 if (decode_precision == 10) {
81 if (decode_precision == 10) {
87 if (idx >= num_coeffs)
93 if (idx + val > num_coeffs)
94 val = num_coeffs - idx;
96 for (i = 0; i <
val; i++) {
97 if (decode_precision == 10) {
104 for (i = 0; i <
val; i++) {
105 if (decode_precision == 10) {
112 }
while (idx < num_coeffs);
118 if (num_bits == 16) {
128 if (num_bits == 16) {
144 case MKTAG(
'a',
'p',
'c',
'o'):
147 case MKTAG(
'a',
'p',
'c',
's'):
150 case MKTAG(
'a',
'p',
'c',
'n'):
153 case MKTAG(
'a',
'p',
'c',
'h'):
156 case MKTAG(
'a',
'p',
'4',
'h'):
160 case MKTAG(
'a',
'p',
'4',
'x'):
170 av_log(avctx,
AV_LOG_DEBUG,
"Auto bitdepth precision. Use 10b decoding based on codec tag.\n");
172 av_log(avctx,
AV_LOG_DEBUG,
"Auto bitdepth precision. Use 12b decoding based on codec tag.\n");
207 ff_dlog(avctx,
"header size %d\n", hdr_size);
208 if (hdr_size > data_size) {
214 ff_dlog(avctx,
"%.4s version %d\n", buf+4, version);
223 if (width != avctx->
width || height != avctx->
height) {
272 ff_dlog(avctx,
"flags %x\n", flags);
275 if(buf + data_size - ptr < 64) {
286 if(buf + data_size - ptr < 64) {
301 int i, hdr_size, slice_count;
302 unsigned pic_data_size;
303 int log2_slice_mb_width, log2_slice_mb_height;
304 int slice_mb_count, mb_x, mb_y;
305 const uint8_t *data_ptr, *index_ptr;
307 hdr_size = buf[0] >> 3;
308 if (hdr_size < 8 || hdr_size > buf_size) {
313 pic_data_size =
AV_RB32(buf + 1);
314 if (pic_data_size > buf_size) {
319 log2_slice_mb_width = buf[7] >> 4;
320 log2_slice_mb_height = buf[7] & 0xF;
321 if (log2_slice_mb_width > 3 || log2_slice_mb_height) {
323 1 << log2_slice_mb_width, 1 << log2_slice_mb_height);
336 av_popcount(ctx->
mb_width & (1 << log2_slice_mb_width) - 1));
350 if (hdr_size + slice_count*2 > buf_size) {
356 index_ptr = buf + hdr_size;
357 data_ptr = index_ptr + slice_count*2;
359 slice_mb_count = 1 << log2_slice_mb_width;
363 for (i = 0; i < slice_count; i++) {
366 slice->
data = data_ptr;
367 data_ptr +=
AV_RB16(index_ptr + i*2);
369 while (ctx->
mb_width - mb_x < slice_mb_count)
370 slice_mb_count >>= 1;
382 mb_x += slice_mb_count;
384 slice_mb_count = 1 << log2_slice_mb_width;
388 if (data_ptr > buf + buf_size) {
400 return pic_data_size;
403 #define DECODE_CODEWORD(val, codebook, SKIP) \ 405 unsigned int rice_order, exp_order, switch_bits; \ 406 unsigned int q, buf, bits; \ 408 UPDATE_CACHE(re, gb); \ 409 buf = GET_CACHE(re, gb); \ 412 switch_bits = codebook & 3; \ 413 rice_order = codebook >> 5; \ 414 exp_order = (codebook >> 2) & 7; \ 416 q = 31 - av_log2(buf); \ 418 if (q > switch_bits) { \ 419 bits = exp_order - switch_bits + (q<<1); \ 420 if (bits > FFMIN(MIN_CACHE_BITS, 31)) \ 421 return AVERROR_INVALIDDATA; \ 422 val = SHOW_UBITS(re, gb, bits) - (1 << exp_order) + \ 423 ((switch_bits + 1) << rice_order); \ 424 SKIP(re, gb, bits); \ 425 } else if (rice_order) { \ 426 SKIP_BITS(re, gb, q+1); \ 427 val = (q << rice_order) + SHOW_UBITS(re, gb, rice_order); \ 428 SKIP(re, gb, rice_order); \ 435 #define TOSIGNED(x) (((x) >> 1) ^ (-((x) & 1))) 437 #define FIRST_DC_CB 0xB8 442 int blocks_per_slice)
457 for (i = 1; i < blocks_per_slice; i++, out += 64) {
459 if(code) sign ^= -(code & 1);
461 prev_dc += (((code + 1) >> 1) ^ sign) - sign;
469 static const uint8_t run_to_cb[16] = { 0x06, 0x06, 0x05, 0x05, 0x04, 0x29, 0x29, 0x29, 0x29, 0x28, 0x28, 0x28, 0x28, 0x28, 0x28, 0x4C };
470 static const uint8_t lev_to_cb[10] = { 0x04, 0x0A, 0x05, 0x06, 0x04, 0x28, 0x28, 0x28, 0x28, 0x4C };
473 int16_t *
out,
int blocks_per_slice)
476 int block_mask, sign;
478 int max_coeffs,
i, bits_left;
479 int log2_block_count =
av_log2(blocks_per_slice);
486 max_coeffs = 64 << log2_block_count;
487 block_mask = blocks_per_slice - 1;
489 for (pos = block_mask;;) {
491 if (!bits_left || (bits_left < 32 && !
SHOW_UBITS(
re, gb, bits_left)))
496 if (pos >= max_coeffs) {
504 i = pos >> log2_block_count;
508 out[((pos & block_mask) << 6) + ctx->
scan[
i]] = ((level ^ sign) - sign);
516 uint16_t *dst,
int dst_stride,
517 const uint8_t *buf,
unsigned buf_size,
524 int i, blocks_per_slice = slice->
mb_count<<2;
527 for (i = 0; i < blocks_per_slice; i++)
538 for (i = 0; i < slice->
mb_count; i++) {
541 ctx->
prodsp.
idct_put(dst+4*dst_stride , dst_stride, block+(2<<6), qmat);
542 ctx->
prodsp.
idct_put(dst+4*dst_stride+8, dst_stride, block+(3<<6), qmat);
550 uint16_t *dst,
int dst_stride,
551 const uint8_t *buf,
unsigned buf_size,
552 const int16_t *qmat,
int log2_blocks_per_mb)
558 int i, j, blocks_per_slice = slice->
mb_count << log2_blocks_per_mb;
561 for (i = 0; i < blocks_per_slice; i++)
572 for (i = 0; i < slice->
mb_count; i++) {
573 for (j = 0; j < log2_blocks_per_mb; j++) {
575 ctx->
prodsp.
idct_put(dst+4*dst_stride, dst_stride, block+(1<<6), qmat);
587 uint16_t *dst,
int dst_stride,
588 const uint8_t *buf,
int buf_size,
589 int blocks_per_slice)
596 for (i = 0; i < blocks_per_slice<<2; i++)
602 ctx->
unpack_alpha(&gb, blocks, blocks_per_slice * 4 * 64, 16);
604 ctx->
unpack_alpha(&gb, blocks, blocks_per_slice * 4 * 64, 8);
609 for (i = 0; i < 16; i++) {
610 memcpy(dst, block, 16 * blocks_per_slice *
sizeof(*dst));
611 dst += dst_stride >> 1;
612 block += 16 * blocks_per_slice;
622 int i, hdr_size, qscale, log2_chroma_blocks_per_mb;
623 int luma_stride, chroma_stride;
624 int y_data_size, u_data_size, v_data_size, a_data_size;
625 uint8_t *dest_y, *dest_u, *dest_v, *dest_a;
630 uint16_t val_no_chroma;
637 hdr_size = buf[0] >> 3;
638 qscale = av_clip(buf[1], 1, 224);
639 qscale = qscale > 128 ? qscale - 96 << 2: qscale;
640 y_data_size =
AV_RB16(buf + 2);
641 u_data_size =
AV_RB16(buf + 4);
642 v_data_size = slice->
data_size - y_data_size - u_data_size - hdr_size;
643 if (hdr_size > 7) v_data_size =
AV_RB16(buf + 6);
644 a_data_size = slice->
data_size - y_data_size - u_data_size -
645 v_data_size - hdr_size;
647 if (y_data_size < 0 || u_data_size < 0 || v_data_size < 0
648 || hdr_size+y_data_size+u_data_size+v_data_size > slice->
data_size){
655 for (i = 0; i < 64; i++) {
656 qmat_luma_scaled [
i] = ctx->
qmat_luma [
i] * qscale;
664 luma_stride = pic->
linesize[0] << 1;
665 chroma_stride = pic->
linesize[1] << 1;
671 log2_chroma_blocks_per_mb = 2;
674 log2_chroma_blocks_per_mb = 1;
677 dest_y = pic->
data[0] + (slice->
mb_y << 4) * luma_stride + (slice->
mb_x << 5);
678 dest_u = pic->
data[1] + (slice->
mb_y << 4) * chroma_stride + (slice->
mb_x << mb_x_shift);
679 dest_v = pic->
data[2] + (slice->
mb_y << 4) * chroma_stride + (slice->
mb_x << mb_x_shift);
680 dest_a = pic->
data[3] + (slice->
mb_y << 4) * luma_stride + (slice->
mb_x << 5);
690 buf, y_data_size, qmat_luma_scaled);
696 buf + y_data_size, u_data_size,
697 qmat_chroma_scaled, log2_chroma_blocks_per_mb);
702 buf + y_data_size + u_data_size, v_data_size,
703 qmat_chroma_scaled, log2_chroma_blocks_per_mb);
708 size_t mb_max_x = slice->
mb_count << (mb_x_shift - 1);
713 val_no_chroma = 511 * 4;
715 for (i = 0; i < 16; ++
i)
716 for (j = 0; j < mb_max_x; ++j) {
717 *(uint16_t*)(dest_u + (i * chroma_stride) + (j << 1)) = val_no_chroma;
718 *(uint16_t*)(dest_v + (i * chroma_stride) + (j << 1)) = val_no_chroma;
725 buf + y_data_size + u_data_size + v_data_size,
745 if (error < ctx->slice_count)
758 int buf_size = avpkt->
size;
759 int frame_hdr_size, pic_size, ret;
775 if (frame_hdr_size < 0)
776 return frame_hdr_size;
778 buf += frame_hdr_size;
779 buf_size -= frame_hdr_size;
798 buf_size -= pic_size;
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
This structure describes decoded (raw) audio or video data.
#define AV_PIX_FMT_YUVA422P10
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
#define AV_LOG_WARNING
Something somehow does not look correct.
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.
static av_cold int init(AVCodecContext *avctx)
static int decode_slice_thread(AVCodecContext *avctx, void *arg, int jobnr, int threadnr)
void(* clear_block)(int16_t *block)
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)
enum AVColorRange color_range
MPEG vs JPEG YUV range.
av_cold void ff_blockdsp_init(BlockDSPContext *c, AVCodecContext *avctx)
#define ALPHA_SHIFT_16_TO_12(alpha_val)
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
#define FF_PROFILE_PRORES_LT
unsigned mb_height
height of the current picture in mb
int idct_permutation_type
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
static void error(const char *err)
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
#define FF_PROFILE_PRORES_XQ
static int decode_picture_header(AVCodecContext *avctx, const uint8_t *buf, const int buf_size)
Multithreading support functions.
static av_cold int decode_init(AVCodecContext *avctx)
#define ALPHA_SHIFT_8_TO_12(alpha_val)
static void unpack_alpha_12(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits)
AVCodec ff_prores_decoder
bitstream reader API header.
int interlaced_frame
The content of the picture is interlaced.
#define AV_PIX_FMT_YUV422P12
#define AV_CODEC_FLAG_GRAY
Only decode/encode grayscale.
const AVProfile ff_prores_profiles[]
int slice_count
number of slices in the current picture
static int get_bits_left(GetBitContext *gb)
unsigned mb_width
width of the current picture in mb
#define UPDATE_CACHE(name, gb)
#define i(width, name, range_min, range_max)
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
static void unpack_alpha_10(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits)
static const uint16_t mask[17]
int skip_alpha
Skip processing alpha if supported by codec.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
int flags
AV_CODEC_FLAG_*.
const char * name
Name of the codec implementation.
#define AV_PIX_FMT_YUV444P10
#define CLOSE_READER(name, gb)
#define AV_CODEC_CAP_FRAME_THREADS
Codec supports frame-level multithreading.
#define SKIP_BITS(name, gb, num)
common internal API header
static void unpack_alpha(GetBitContext *gb, uint16_t *dst, int num_coeffs, const int num_bits, const int decode_precision)
enum AVPictureType pict_type
Picture type of the frame.
av_cold void ff_init_scantable_permutation(uint8_t *idct_permutation, enum idct_permutation_type perm_type)
#define AV_PIX_FMT_YUVA444P12
int width
picture width / height.
#define FF_PROFILE_PRORES_STANDARD
#define FF_PROFILE_UNKNOWN
uint8_t idct_permutation[64]
enum AVColorPrimaries color_primaries
Chromaticity coordinates of the source primaries.
#define LAST_SKIP_BITS(name, gb, num)
static const uint8_t lev_to_cb[10]
#define AV_PIX_FMT_YUVA444P10
uint8_t interlaced_scan[64]
#define ALPHA_SHIFT_16_TO_10(alpha_val)
#define SHOW_UBITS(name, gb, num)
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
#define AV_CODEC_CAP_SLICE_THREADS
Codec supports slice-based (or partition-based) multithreading.
#define FF_PROFILE_PRORES_4444
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 uint8_t ff_prores_interlaced_scan[64]
Libavcodec external API header.
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
int ff_thread_get_buffer(AVCodecContext *avctx, ThreadFrame *f, int flags)
Wrapper around get_buffer() for frame-multithreaded codecs.
main external API structure.
const uint8_t ff_prores_progressive_scan[64]
static void permute(uint8_t *dst, const uint8_t *src, const uint8_t permutation[64])
unsigned int codec_tag
fourcc (LSB first, so "ABCD" -> ('D'<<24) + ('C'<<16) + ('B'<<8) + 'A').
#define OPEN_READER(name, gb)
static av_always_inline int decode_ac_coeffs(AVCodecContext *avctx, GetBitContext *gb, int16_t *out, int blocks_per_slice)
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
static unsigned int get_bits1(GetBitContext *s)
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 const uint8_t run_to_cb[16]
static const AVProfile profiles[]
uint8_t progressive_scan[64]
enum AVColorSpace colorspace
YUV colorspace type.
enum AVColorTransferCharacteristic color_trc
Color Transfer Characteristic.
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
#define FF_PROFILE_PRORES_HQ
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
#define DECODE_CODEWORD(val, codebook, SKIP)
#define FF_DECODE_ERROR_INVALID_BITSTREAM
int decode_error_flags
decode error flags of the frame, set to a combination of FF_DECODE_ERROR_xxx flags if the decoder pro...
#define flags(name, subs,...)
#define AV_PIX_FMT_YUV422P10
#define AV_PIX_FMT_YUV444P12
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
the normal 219*2^(n-8) "MPEG" YUV ranges
#define LOCAL_ALIGNED_32(t, v,...)
#define SHOW_SBITS(name, gb, num)
common internal api header.
static const uint8_t dc_codebook[7]
#define ALPHA_SHIFT_8_TO_10(alpha_val)
int top_field_first
If the content is interlaced, is top field displayed first.
av_cold int ff_proresdsp_init(ProresDSPContext *dsp, AVCodecContext *avctx)
void(* unpack_alpha)(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.
int key_frame
1 -> keyframe, 0-> not
static int decode_frame_header(ProresContext *ctx, const uint8_t *buf, const int data_size, AVCodecContext *avctx)
static av_always_inline int decode_dc_coeffs(GetBitContext *gb, int16_t *out, int blocks_per_slice)
#define FF_PROFILE_PRORES_PROXY
static av_cold int decode_close(AVCodecContext *avctx)
#define LOCAL_ALIGNED_16(t, v,...)
int frame_type
0 = progressive, 1 = tff, 2 = bff
#define MKTAG(a, b, c, d)
static double val(void *priv, double ch)
This structure stores compressed data.
static int decode_picture(AVCodecContext *avctx)
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
void(* idct_put)(uint16_t *out, ptrdiff_t linesize, int16_t *block, const int16_t *qmat)
#define AV_PIX_FMT_YUVA422P12
void * av_mallocz_array(size_t nmemb, size_t size)
Allocate a memory block for an array with av_mallocz().