blob: 36c795c56e6e5dd00dc4172153b35b509c5d9da1 [file] [log] [blame] [edit]
/*
* Copyright (c) 2021, Alliance for Open Media. All rights reserved
*
* This source code is subject to the terms of the BSD 3-Clause Clear License
* and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear
* License was not distributed with this source code in the LICENSE file, you
* can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the
* Alliance for Open Media Patent License 1.0 was not distributed with this
* source code in the PATENTS file, you can obtain it at
* aomedia.org/license/patent-license/.
*/
#ifndef AOM_AV1_DECODER_DECODER_H_
#define AOM_AV1_DECODER_DECODER_H_
#include "config/aom_config.h"
#include "aom/aom_codec.h"
#include "aom_dsp/bitreader.h"
#include "aom_scale/yv12config.h"
#include "aom_util/aom_thread.h"
#include "av1/common/av1_common_int.h"
#include "av1/common/thread_common.h"
#include "av1/decoder/dthread.h"
#if CONFIG_ACCOUNTING
#include "av1/decoder/accounting.h"
#endif
#if CONFIG_INSPECTION
#include "av1/decoder/inspection.h"
#endif
#ifdef __cplusplus
extern "C" {
#endif
/*!
* \brief Contains coding block data required by the decoder.
*
* This includes:
* - Coding block info that is common between encoder and decoder.
* - Other coding block info only needed by the decoder.
* Contrast this with a similar struct MACROBLOCK on encoder side.
* This data is also common between ThreadData and AV1Decoder structs.
*/
typedef struct DecoderCodingBlock {
/*!
* Coding block info that is common between encoder and decoder.
*/
DECLARE_ALIGNED(32, MACROBLOCKD, xd);
/*!
* True if the at least one of the coding blocks decoded was corrupted.
*/
int corrupted;
/*!
* Pointer to 'mc_buf' inside 'pbi->td' (single-threaded decoding) or
* 'pbi->thread_data[i].td' (multi-threaded decoding).
*/
uint16_t *mc_buf[2];
/*!
* Pointer to 'dqcoeff' inside 'td->cb_buffer_base' or 'pbi->cb_buffer_base'
* with appropriate offset for the current superblock, for each plane.
*/
tran_low_t *dqcoeff_block[MAX_MB_PLANE];
/*!
* cb_offset[p] is the offset into the dqcoeff_block[p] for the current coding
* block, for each plane 'p'.
*/
uint16_t cb_offset[MAX_MB_PLANE];
/*!
* Pointer to 'eob_data' inside 'td->cb_buffer_base' or 'pbi->cb_buffer_base'
* with appropriate offset for the current superblock, for each plane.
*/
eob_info *eob_data[MAX_MB_PLANE];
/*!
* txb_offset[p] is the offset into the eob_data[p] for the current coding
* block, for each plane 'p'.
*/
uint16_t txb_offset[MAX_MB_PLANE];
/*!
* ref_mv_count[i] specifies the number of number of motion vector candidates
* in xd->ref_mv_stack[i].
*/
uint8_t ref_mv_count[MODE_CTX_REF_FRAMES];
} DecoderCodingBlock;
/*!\cond */
typedef void (*decode_block_visitor_fn_t)(const AV1_COMMON *const cm,
DecoderCodingBlock *dcb,
aom_reader *const r, const int plane,
const int row, const int col,
const TX_SIZE tx_size);
typedef void (*predict_inter_block_visitor_fn_t)(AV1_COMMON *const cm,
DecoderCodingBlock *dcb,
BLOCK_SIZE bsize);
typedef void (*cfl_store_inter_block_visitor_fn_t)(AV1_COMMON *const cm,
MACROBLOCKD *const xd);
typedef struct ThreadData {
DecoderCodingBlock dcb;
// Coding block buffer for the current superblock.
// Used only for single-threaded decoding and multi-threaded decoding with
// row_mt == 1 cases.
// See also: similar buffer in 'AV1Decoder'.
CB_BUFFER cb_buffer_base;
aom_reader *bit_reader;
// Motion compensation buffer used to get a prediction buffer with extended
// borders. One buffer for each of the two possible references.
uint16_t *mc_buf[2];
// Allocated size of 'mc_buf'.
int32_t mc_buf_size;
CONV_BUF_TYPE *tmp_conv_dst;
uint16_t *tmp_obmc_bufs[2];
decode_block_visitor_fn_t read_coeffs_tx_intra_block_visit;
decode_block_visitor_fn_t predict_and_recon_intra_block_visit;
decode_block_visitor_fn_t read_coeffs_tx_inter_block_visit;
#if CONFIG_CROSS_CHROMA_TX
decode_block_visitor_fn_t inverse_cctx_block_visit;
#endif // CONFIG_CROSS_CHROMA_TX
decode_block_visitor_fn_t inverse_tx_inter_block_visit;
predict_inter_block_visitor_fn_t predict_inter_block_visit;
cfl_store_inter_block_visitor_fn_t cfl_store_inter_block_visit;
#if CONFIG_REF_MV_BANK
REF_MV_BANK ref_mv_bank;
#endif // CONFIG_REF_MV_BANK
#if CONFIG_WARP_REF_LIST
WARP_PARAM_BANK warp_param_bank;
#endif // CONFIG_WARP_REF_LIST
} ThreadData;
typedef struct AV1DecRowMTJobInfo {
int tile_row;
int tile_col;
int mi_row;
} AV1DecRowMTJobInfo;
typedef struct AV1DecRowMTSyncData {
#if CONFIG_MULTITHREAD
pthread_mutex_t *mutex_;
pthread_cond_t *cond_;
#endif
int allocated_sb_rows;
int *cur_sb_col;
int sync_range;
int mi_rows;
int mi_cols;
int mi_rows_parse_done;
int mi_rows_decode_started;
int num_threads_working;
} AV1DecRowMTSync;
typedef struct AV1DecRowMTInfo {
int tile_rows_start;
int tile_rows_end;
int tile_cols_start;
int tile_cols_end;
int start_tile;
int end_tile;
int mi_rows_to_decode;
// Invariant:
// mi_rows_parse_done >= mi_rows_decode_started.
// mi_rows_parse_done and mi_rows_decode_started are both initialized to 0.
// mi_rows_parse_done is incremented freely. mi_rows_decode_started may only
// be incremented to catch up with mi_rows_parse_done but is not allowed to
// surpass mi_rows_parse_done.
//
// When mi_rows_decode_started reaches mi_rows_to_decode, there are no more
// decode jobs.
// Indicates the progress of the bit-stream parsing of superblocks.
// Initialized to 0. Incremented by sb_mi_size when parse sb row is done.
int mi_rows_parse_done;
// Indicates the progress of the decoding of superblocks.
// Initialized to 0. Incremented by sb_mi_size when decode sb row is started.
int mi_rows_decode_started;
// Boolean: Initialized to 0 (false). Set to 1 (true) on error to abort
// decoding.
int row_mt_exit;
} AV1DecRowMTInfo;
typedef struct TileDataDec {
TileInfo tile_info;
aom_reader bit_reader;
DECLARE_ALIGNED(16, FRAME_CONTEXT, tctx);
AV1DecRowMTSync dec_row_mt_sync;
} TileDataDec;
typedef struct TileBufferDec {
const uint8_t *data;
size_t size;
} TileBufferDec;
typedef struct DataBuffer {
const uint8_t *data;
size_t size;
} DataBuffer;
typedef struct EXTERNAL_REFERENCES {
YV12_BUFFER_CONFIG refs[MAX_EXTERNAL_REFERENCES];
int num;
} EXTERNAL_REFERENCES;
typedef struct TileJobsDec {
TileBufferDec *tile_buffer;
TileDataDec *tile_data;
} TileJobsDec;
typedef struct AV1DecTileMTData {
#if CONFIG_MULTITHREAD
pthread_mutex_t *job_mutex;
#endif
TileJobsDec *job_queue;
int jobs_enqueued;
int jobs_dequeued;
int alloc_tile_rows;
int alloc_tile_cols;
} AV1DecTileMT;
/*!
* \Holds subgop related info.
*/
typedef struct {
unsigned char disp_frame_idx[MAX_SUBGOP_STATS_SIZE];
int show_existing_frame[MAX_SUBGOP_STATS_SIZE];
int show_frame[MAX_SUBGOP_STATS_SIZE];
int qindex[MAX_SUBGOP_STATS_SIZE];
int refresh_frame_flags[MAX_SUBGOP_STATS_SIZE];
unsigned int ref_frame_map[MAX_SUBGOP_STATS_SIZE][REF_FRAMES];
unsigned char stat_count;
} SubGOPStatsDec;
typedef struct AV1Decoder {
DecoderCodingBlock dcb;
DECLARE_ALIGNED(32, AV1_COMMON, common);
AVxWorker lf_worker;
AV1LfSync lf_row_sync;
AV1LrSync lr_row_sync;
AV1LrStruct lr_ctxt;
AVxWorker *tile_workers;
int num_workers;
DecWorkerData *thread_data;
ThreadData td;
TileDataDec *tile_data;
int allocated_tiles;
TileBufferDec tile_buffers[MAX_TILE_ROWS][MAX_TILE_COLS];
AV1DecTileMT tile_mt_info;
// Each time the decoder is called, we expect to receive a full temporal unit.
// This can contain up to one shown frame per spatial layer in the current
// operating point (note that some layers may be entirely omitted).
// If the 'output_all_layers' option is true, we save all of these shown
// frames so that they can be returned to the application. If the
// 'output_all_layers' option is false, then we only output one image per
// temporal unit.
//
// Note: The saved buffers are released at the start of the next time the
// application calls aom_codec_decode().
int output_all_layers;
RefCntBuffer *output_frames[MAX_NUM_SPATIAL_LAYERS];
size_t num_output_frames; // How many frames are queued up so far?
// In order to properly support random-access decoding, we need
// to behave slightly differently for the very first frame we decode.
// So we track whether this is the first frame or not.
int decoding_first_frame;
int max_threads;
int inv_tile_order;
int need_resync; // wait for key/intra-only frame.
int reset_decoder_state;
int tile_size_bytes;
int tile_col_size_bytes;
int dec_tile_row, dec_tile_col; // always -1 for non-VR tile encoding
#if CONFIG_ACCOUNTING
int acct_enabled;
Accounting accounting;
#endif
int sequence_header_ready;
int sequence_header_changed;
#if CONFIG_INSPECTION
aom_inspect_cb inspect_cb;
void *inspect_ctx;
#endif
int operating_point;
int current_operating_point;
int seen_frame_header;
// The expected start_tile (tg_start syntax element) of the next tile group.
int next_start_tile;
// State if the camera frame header is already decoded while
// large_scale_tile = 1.
int camera_frame_header_ready;
size_t frame_header_size;
DataBuffer obu_size_hdr;
int output_frame_width_in_tiles_minus_1;
int output_frame_height_in_tiles_minus_1;
int tile_count_minus_1;
uint32_t coded_tile_data_size;
unsigned int ext_tile_debug; // for ext-tile software debug & testing
// Decoder has 3 modes of operation:
// (1) Single-threaded decoding.
// (2) Multi-threaded decoding with each tile decoded in parallel.
// (3) In addition to (2), each thread decodes 1 superblock row in parallel.
// row_mt = 1 triggers mode (3) above, while row_mt = 0, will trigger mode (1)
// or (2) depending on 'max_threads'.
unsigned int row_mt;
EXTERNAL_REFERENCES ext_refs;
YV12_BUFFER_CONFIG tile_list_outbuf;
// Coding block buffer for the current frame.
// Allocated and used only for multi-threaded decoding with 'row_mt == 0'.
// See also: similar buffer in 'ThreadData' struct.
CB_BUFFER *cb_buffer_base;
// Allocated size of 'cb_buffer_base'. Currently same as the number of
// superblocks in the coded frame.
int cb_buffer_alloc_size;
int allocated_row_mt_sync_rows;
#if CONFIG_MULTITHREAD
pthread_mutex_t *row_mt_mutex_;
pthread_cond_t *row_mt_cond_;
#endif
AV1DecRowMTInfo frame_row_mt_info;
aom_metadata_array_t *metadata;
int context_update_tile_id;
int skip_loop_filter;
int skip_film_grain;
int is_annexb;
int valid_for_referencing[REF_FRAMES];
int is_fwd_kf_present;
int is_arf_frame_present;
int num_tile_groups;
aom_s_frame_info sframe_info;
unsigned int enable_subgop_stats;
SubGOPStatsDec subgop_stats;
} AV1Decoder;
// Returns 0 on success. Sets pbi->common.error.error_code to a nonzero error
// code and returns a nonzero value on failure.
int av1_receive_compressed_data(struct AV1Decoder *pbi, size_t size,
const uint8_t **psource);
// Get the frame at a particular index in the output queue
int av1_get_raw_frame(AV1Decoder *pbi, size_t index, YV12_BUFFER_CONFIG **sd,
aom_film_grain_t **grain_params);
int av1_get_frame_to_show(struct AV1Decoder *pbi, YV12_BUFFER_CONFIG *frame);
aom_codec_err_t av1_copy_reference_dec(struct AV1Decoder *pbi, int idx,
YV12_BUFFER_CONFIG *sd);
aom_codec_err_t av1_set_reference_dec(AV1_COMMON *cm, int idx,
int use_external_ref,
YV12_BUFFER_CONFIG *sd);
aom_codec_err_t av1_copy_new_frame_dec(AV1_COMMON *cm,
YV12_BUFFER_CONFIG *new_frame,
YV12_BUFFER_CONFIG *sd);
struct AV1Decoder *av1_decoder_create(BufferPool *const pool);
void av1_decoder_remove(struct AV1Decoder *pbi);
void av1_dealloc_dec_jobs(struct AV1DecTileMTData *tile_mt_info);
void av1_dec_row_mt_dealloc(AV1DecRowMTSync *dec_row_mt_sync);
void av1_dec_free_cb_buf(AV1Decoder *pbi);
static INLINE void decrease_ref_count(RefCntBuffer *const buf,
BufferPool *const pool) {
if (buf != NULL) {
--buf->ref_count;
// Reference counts should never become negative. If this assertion fails,
// there is a bug in our reference count management.
assert(buf->ref_count >= 0);
// A worker may only get a free framebuffer index when calling get_free_fb.
// But the raw frame buffer is not set up until we finish decoding header.
// So if any error happens during decoding header, frame_bufs[idx] will not
// have a valid raw frame buffer.
if (buf->ref_count == 0 && buf->raw_frame_buffer.data) {
pool->release_fb_cb(pool->cb_priv, &buf->raw_frame_buffer);
buf->raw_frame_buffer.data = NULL;
buf->raw_frame_buffer.size = 0;
buf->raw_frame_buffer.priv = NULL;
}
}
}
#define ACCT_STR __func__
static INLINE int av1_read_uniform(aom_reader *r, int n) {
const int l = get_unsigned_bits(n);
const int m = (1 << l) - n;
const int v = aom_read_literal(r, l - 1, ACCT_STR);
assert(l != 0);
if (v < m)
return v;
else
return (v << 1) - m + aom_read_literal(r, 1, ACCT_STR);
}
typedef void (*palette_visitor_fn_t)(MACROBLOCKD *const xd, int plane,
aom_reader *r);
void av1_visit_palette(AV1Decoder *const pbi, MACROBLOCKD *const xd,
aom_reader *r, palette_visitor_fn_t visit);
typedef void (*block_visitor_fn_t)(AV1Decoder *const pbi, ThreadData *const td,
int mi_row, int mi_col, aom_reader *r,
PARTITION_TYPE partition, BLOCK_SIZE bsize,
PARTITION_TREE *parent, int index);
/*!\endcond */
#ifdef __cplusplus
} // extern "C"
#endif
#endif // AOM_AV1_DECODER_DECODER_H_