| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent. |
| */ |
| |
| #ifndef AV1_DECODER_DECODER_H_ |
| #define 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/thread_common.h" |
| #include "av1/common/onyxc_int.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 |
| |
| typedef struct ThreadData { |
| aom_reader *bit_reader; |
| DECLARE_ALIGNED(32, MACROBLOCKD, xd); |
| /* dqcoeff are shared by all the planes. So planes must be decoded serially */ |
| DECLARE_ALIGNED(32, tran_low_t, dqcoeff[MAX_TX_SQUARE]); |
| CB_BUFFER cb_buffer_base; |
| uint8_t *mc_buf[2]; |
| int32_t mc_buf_size; |
| } ThreadData; |
| |
| typedef struct TileDataDec { |
| TileInfo tile_info; |
| aom_reader bit_reader; |
| DECLARE_ALIGNED(16, FRAME_CONTEXT, tctx); |
| } 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; |
| |
| typedef struct AV1Decoder { |
| DECLARE_ALIGNED(32, MACROBLOCKD, mb); |
| |
| DECLARE_ALIGNED(32, AV1_COMMON, common); |
| |
| int refresh_frame_flags; |
| |
| // TODO(hkuang): Combine this with cur_buf in macroblockd as they are |
| // the same. |
| RefCntBuffer *cur_buf; // Current decoding frame buffer. |
| |
| AVxWorker *frame_worker_owner; // frame_worker that owns this pbi. |
| 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; |
| YV12_BUFFER_CONFIG *output_frames[MAX_NUM_SPATIAL_LAYERS]; |
| size_t output_frame_index[MAX_NUM_SPATIAL_LAYERS]; // Buffer pool indices |
| 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 allow_lowbitdepth; |
| int max_threads; |
| int inv_tile_order; |
| int need_resync; // wait for key/intra-only frame. |
| int hold_ref_buf; // hold the reference buffer. |
| |
| 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 |
| size_t uncomp_hdr_size; // Size of the uncompressed header |
| int tg_size; // Number of tiles in the current tilegroup |
| int tg_start; // First tile in the current tilegroup |
| int tg_size_bit_offset; |
| int sequence_header_ready; |
| #if CONFIG_INSPECTION |
| aom_inspect_cb inspect_cb; |
| void *inspect_ctx; |
| #endif |
| int operating_point; |
| int current_operating_point; |
| int seen_frame_header; |
| |
| // 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 |
| unsigned int row_mt; |
| EXTERNAL_REFERENCES ext_refs; |
| size_t tile_list_size; |
| uint8_t *tile_list_output; |
| size_t buffer_sz; |
| } AV1Decoder; |
| |
| int av1_receive_compressed_data(struct AV1Decoder *pbi, size_t size, |
| const uint8_t **dest); |
| |
| // 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_jobs_sync); |
| |
| static INLINE void decrease_ref_count(int idx, RefCntBuffer *const frame_bufs, |
| BufferPool *const pool) { |
| if (idx >= 0) { |
| --frame_bufs[idx].ref_count; |
| // A worker may only get a free framebuffer index when calling get_free_fb. |
| // But the private buffer is not set up until finish decoding header. |
| // So any error happens during decoding header, the frame_bufs will not |
| // have valid priv buffer. |
| if (frame_bufs[idx].ref_count == 0 && |
| frame_bufs[idx].raw_frame_buffer.priv) { |
| pool->release_fb_cb(pool->cb_priv, &frame_bufs[idx].raw_frame_buffer); |
| } |
| } |
| } |
| |
| static INLINE int dec_is_ref_frame_buf(AV1Decoder *const pbi, |
| RefCntBuffer *frame_buf) { |
| AV1_COMMON *const cm = &pbi->common; |
| int i; |
| for (i = 0; i < INTER_REFS_PER_FRAME; ++i) { |
| RefBuffer *const ref_frame = &cm->frame_refs[i]; |
| if (ref_frame->idx == INVALID_IDX) continue; |
| if (frame_buf == &cm->buffer_pool->frame_bufs[ref_frame->idx]) break; |
| } |
| return (i < INTER_REFS_PER_FRAME); |
| } |
| |
| #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, int mi_row, |
| int mi_col, aom_reader *r, BLOCK_SIZE bsize, |
| palette_visitor_fn_t visit); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AV1_DECODER_DECODER_H_ |