418 lines
13 KiB
C
418 lines
13 KiB
C
/*
|
|
* 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.
|
|
*/
|
|
|
|
/*!\defgroup aom_decoder AOMedia AOM/AV1 Decoder
|
|
* \ingroup aom
|
|
*
|
|
* @{
|
|
*/
|
|
/*!\file
|
|
* \brief Provides definitions for using AOM or AV1 within the aom Decoder
|
|
* interface.
|
|
*/
|
|
#ifndef AOM_AOM_AOMDX_H_
|
|
#define AOM_AOM_AOMDX_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* Include controls common to both the encoder and decoder */
|
|
#include "aom/aom.h"
|
|
|
|
/*!\name Algorithm interface for AV1
|
|
*
|
|
* This interface provides the capability to decode AV1 streams.
|
|
* @{
|
|
*/
|
|
|
|
/*!\brief A single instance of the AV1 decoder.
|
|
*\deprecated This access mechanism is provided for backwards compatibility;
|
|
* prefer aom_codec_av1_dx().
|
|
*/
|
|
extern aom_codec_iface_t aom_codec_av1_dx_algo;
|
|
/*!\brief The interface to the AV1 decoder.
|
|
*/
|
|
extern aom_codec_iface_t *aom_codec_av1_dx(void);
|
|
|
|
/*!@} - end algorithm interface member group */
|
|
|
|
/** Data structure that stores bit accounting for debug
|
|
*/
|
|
typedef struct Accounting Accounting;
|
|
|
|
#ifndef AOM_INSPECTION_H_
|
|
/** Callback that inspects decoder frame data.
|
|
*/
|
|
typedef void (*aom_inspect_cb)(void *decoder, void *ctx);
|
|
|
|
#endif
|
|
|
|
/*!\brief Structure to hold inspection callback and context.
|
|
*
|
|
* Defines a structure to hold the inspection callback function and calling
|
|
* context.
|
|
*/
|
|
typedef struct aom_inspect_init {
|
|
/*! Inspection callback. */
|
|
aom_inspect_cb inspect_cb;
|
|
|
|
/*! Inspection context. */
|
|
void *inspect_ctx;
|
|
} aom_inspect_init;
|
|
|
|
/*!\brief Structure to collect a buffer index when inspecting.
|
|
*
|
|
* Defines a structure to hold the buffer and return an index
|
|
* when calling decode from inspect. This enables us to decode
|
|
* non showable sub frames.
|
|
*/
|
|
typedef struct {
|
|
/*! Pointer for new position in compressed buffer after decoding 1 OBU. */
|
|
const unsigned char *buf;
|
|
/*! Index into reference buffer array to see result of decoding 1 OBU. */
|
|
int idx;
|
|
/*! Is a show existing frame. */
|
|
int show_existing;
|
|
} Av1DecodeReturn;
|
|
|
|
/*!\brief Structure to hold a tile's start address and size in the bitstream.
|
|
*
|
|
* Defines a structure to hold a tile's start address and size in the bitstream.
|
|
*/
|
|
typedef struct aom_tile_data {
|
|
/*! Tile data size. */
|
|
size_t coded_tile_data_size;
|
|
/*! Tile's start address. */
|
|
const void *coded_tile_data;
|
|
/*! Extra size information. */
|
|
size_t extra_size;
|
|
} aom_tile_data;
|
|
|
|
/*!\brief Structure to hold the external reference frame pointer.
|
|
*
|
|
* Define a structure to hold the external reference frame pointer.
|
|
*/
|
|
typedef struct av1_ext_ref_frame {
|
|
/*! Start pointer of external references. */
|
|
aom_image_t *img;
|
|
/*! Number of available external references. */
|
|
int num;
|
|
} av1_ext_ref_frame_t;
|
|
|
|
/*!\enum aom_dec_control_id
|
|
* \brief AOM decoder control functions
|
|
*
|
|
* This set of macros define the control functions available for the AOM
|
|
* decoder interface.
|
|
*
|
|
* \sa #aom_codec_control(aom_codec_ctx_t *ctx, int ctrl_id, ...)
|
|
*/
|
|
enum aom_dec_control_id {
|
|
/*!\brief Codec control function to get info on which reference frames were
|
|
* updated by the last decode, int* parameter
|
|
*/
|
|
AOMD_GET_LAST_REF_UPDATES = AOM_DECODER_CTRL_ID_START,
|
|
|
|
/*!\brief Codec control function to check if the indicated frame is
|
|
corrupted, int* parameter
|
|
*/
|
|
AOMD_GET_FRAME_CORRUPTED,
|
|
|
|
/*!\brief Codec control function to get info on which reference frames were
|
|
* used by the last decode, int* parameter
|
|
*/
|
|
AOMD_GET_LAST_REF_USED,
|
|
|
|
/*!\brief Codec control function to get the dimensions that the current
|
|
* frame is decoded at, int* parameter. This may be different to the
|
|
* intended display size for the frame as specified in the wrapper or frame
|
|
* header (see AV1D_GET_DISPLAY_SIZE).
|
|
*/
|
|
AV1D_GET_FRAME_SIZE,
|
|
|
|
/*!\brief Codec control function to get the current frame's intended display
|
|
* dimensions (as specified in the wrapper or frame header), int* parameter.
|
|
* This may be different to the decoded dimensions of this frame (see
|
|
* AV1D_GET_FRAME_SIZE).
|
|
*/
|
|
AV1D_GET_DISPLAY_SIZE,
|
|
|
|
/*!\brief Codec control function to get the bit depth of the stream,
|
|
* unsigned int* parameter
|
|
*/
|
|
AV1D_GET_BIT_DEPTH,
|
|
|
|
/*!\brief Codec control function to get the image format of the stream,
|
|
* aom_img_fmt_t* parameter
|
|
*/
|
|
AV1D_GET_IMG_FORMAT,
|
|
|
|
/*!\brief Codec control function to get the size of the tile, unsigned int
|
|
parameter */
|
|
AV1D_GET_TILE_SIZE,
|
|
|
|
/*!\brief Codec control function to get the tile count in a tile list, int*
|
|
* parameter
|
|
*/
|
|
AV1D_GET_TILE_COUNT,
|
|
|
|
/*!\brief Codec control function to set the byte alignment of the planes in
|
|
* the reference buffers, int parameter
|
|
*
|
|
* Valid values are power of 2, from 32 to 1024. A value of 0 sets
|
|
* legacy alignment. I.e. Y plane is aligned to 32 bytes, U plane directly
|
|
* follows Y plane, and V plane directly follows U plane. Default value is 0.
|
|
*/
|
|
AV1_SET_BYTE_ALIGNMENT,
|
|
|
|
/*!\brief Codec control function to invert the decoding order to from right to
|
|
* left, int parameter
|
|
*
|
|
* The function is used in a test to confirm the decoding independence of tile
|
|
* columns. The function may be used in application where this order
|
|
* of decoding is desired. int parameter
|
|
*
|
|
* TODO(yaowu): Rework the unit test that uses this control, and in a future
|
|
* release, this test-only control shall be removed.
|
|
*/
|
|
AV1_INVERT_TILE_DECODE_ORDER,
|
|
|
|
/*!\brief Codec control function to set the skip loop filter flag, int
|
|
* parameter
|
|
*
|
|
* Valid values are integers. The decoder will skip the loop filter
|
|
* when its value is set to nonzero. If the loop filter is skipped the
|
|
* decoder may accumulate decode artifacts. The default value is 0.
|
|
*/
|
|
AV1_SET_SKIP_LOOP_FILTER,
|
|
|
|
/*!\brief Codec control function to retrieve a pointer to the Accounting
|
|
* struct, takes Accounting** as parameter
|
|
*
|
|
* If called before a frame has been decoded, this returns AOM_CODEC_ERROR.
|
|
* The caller should ensure that AOM_CODEC_OK is returned before attempting
|
|
* to dereference the Accounting pointer.
|
|
*
|
|
* \attention When compiled without --enable-accounting, this returns
|
|
* AOM_CODEC_INCAPABLE.
|
|
*/
|
|
AV1_GET_ACCOUNTING,
|
|
|
|
/*!\brief Codec control function to get last decoded frame quantizer,
|
|
* int* parameter
|
|
*
|
|
* Returned value uses internal quantizer scale defined by the codec.
|
|
*/
|
|
AOMD_GET_LAST_QUANTIZER,
|
|
|
|
/*!\brief Codec control function to set the range of tile decoding, int
|
|
* parameter
|
|
*
|
|
* A value that is greater and equal to zero indicates only the specific
|
|
* row/column is decoded. A value that is -1 indicates the whole row/column
|
|
* is decoded. A special case is both values are -1 that means the whole
|
|
* frame is decoded.
|
|
*/
|
|
AV1_SET_DECODE_TILE_ROW,
|
|
AV1_SET_DECODE_TILE_COL,
|
|
|
|
/*!\brief Codec control function to set the tile coding mode, int parameter
|
|
*
|
|
* - 0 = tiles are coded in normal tile mode
|
|
* - 1 = tiles are coded in large-scale tile mode
|
|
*/
|
|
AV1_SET_TILE_MODE,
|
|
|
|
/*!\brief Codec control function to get the frame header information of an
|
|
* encoded frame, unsigned int* parameter
|
|
*/
|
|
AV1D_GET_FRAME_HEADER_INFO,
|
|
|
|
/*!\brief Codec control function to get the start address and size of a
|
|
* tile in the coded bitstream, aom_tile_data* parameter.
|
|
*/
|
|
AV1D_GET_TILE_DATA,
|
|
|
|
/*!\brief Codec control function to set the external references' pointers in
|
|
* the decoder, av1_ext_ref_frame_t* parameter.
|
|
*
|
|
* This is used while decoding the tile list OBU in large-scale tile coding
|
|
* mode.
|
|
*/
|
|
AV1D_SET_EXT_REF_PTR,
|
|
|
|
/*!\brief Codec control function to enable the ext-tile software debug and
|
|
* testing code in the decoder, unsigned int parameter
|
|
*/
|
|
AV1D_EXT_TILE_DEBUG,
|
|
|
|
/*!\brief Codec control function to enable the row based multi-threading of
|
|
* decoding, unsigned int parameter
|
|
*
|
|
* - 0 = disabled
|
|
* - 1 = enabled (default)
|
|
*/
|
|
AV1D_SET_ROW_MT,
|
|
|
|
/*!\brief Codec control function to indicate whether bitstream is in
|
|
* Annex-B format, unsigned int parameter
|
|
*/
|
|
AV1D_SET_IS_ANNEXB,
|
|
|
|
/*!\brief Codec control function to indicate which operating point to use,
|
|
* int parameter
|
|
*
|
|
* A scalable stream may define multiple operating points, each of which
|
|
* defines a set of temporal and spatial layers to be processed. The
|
|
* operating point index may take a value between 0 and
|
|
* operating_points_cnt_minus_1 (which is at most 31).
|
|
*/
|
|
AV1D_SET_OPERATING_POINT,
|
|
|
|
/*!\brief Codec control function to indicate whether to output one frame per
|
|
* temporal unit (the default), or one frame per spatial layer. int parameter
|
|
*
|
|
* In a scalable stream, each temporal unit corresponds to a single "frame"
|
|
* of video, and within a temporal unit there may be multiple spatial layers
|
|
* with different versions of that frame.
|
|
* For video playback, only the highest-quality version (within the
|
|
* selected operating point) is needed, but for some use cases it is useful
|
|
* to have access to multiple versions of a frame when they are available.
|
|
*/
|
|
AV1D_SET_OUTPUT_ALL_LAYERS,
|
|
|
|
/*!\brief Codec control function to set an aom_inspect_cb callback that is
|
|
* invoked each time a frame is decoded, aom_inspect_init* parameter
|
|
*
|
|
* \attention When compiled without --enable-inspection, this
|
|
* returns AOM_CODEC_INCAPABLE.
|
|
*/
|
|
AV1_SET_INSPECTION_CALLBACK,
|
|
|
|
/*!\brief Codec control function to set the skip film grain flag, int
|
|
* parameter
|
|
*
|
|
* Valid values are integers. The decoder will skip the film grain when its
|
|
* value is set to nonzero. The default value is 0.
|
|
*/
|
|
AV1D_SET_SKIP_FILM_GRAIN,
|
|
|
|
AOM_DECODER_CTRL_ID_MAX,
|
|
|
|
AOMD_GET_FWD_KF_PRESENT,
|
|
|
|
/*!\brief Codec control function to get the frame flags of the previous frame
|
|
* decoded. This will return a flag of type aom_codec_frame_flags_t.
|
|
*/
|
|
AOMD_GET_FRAME_FLAGS,
|
|
};
|
|
|
|
/*!\cond */
|
|
/*!\brief AOM decoder control function parameter type
|
|
*
|
|
* Defines the data types that AOMD control functions take.
|
|
*
|
|
* \note Additional common controls are defined in aom.h.
|
|
*
|
|
* \note For each control ID "X", a macro-define of
|
|
* AOM_CTRL_X is provided. It is used at compile time to determine
|
|
* if the control ID is supported by the libaom library available,
|
|
* when the libaom version cannot be controlled.
|
|
*/
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_LAST_REF_UPDATES, int *)
|
|
#define AOM_CTRL_AOMD_GET_LAST_REF_UPDATES
|
|
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_FRAME_CORRUPTED, int *)
|
|
#define AOM_CTRL_AOMD_GET_FRAME_CORRUPTED
|
|
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_LAST_REF_USED, int *)
|
|
#define AOM_CTRL_AOMD_GET_LAST_REF_USED
|
|
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_LAST_QUANTIZER, int *)
|
|
#define AOM_CTRL_AOMD_GET_LAST_QUANTIZER
|
|
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_FWD_KF_PRESENT, int *)
|
|
#define AOM_CTRL_AOMD_GET_FWD_KF_PRESENT
|
|
|
|
AOM_CTRL_USE_TYPE(AOMD_GET_FRAME_FLAGS, int *)
|
|
#define AOM_CTRL_AOMD_GET_FRAME_FLAGS
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_DISPLAY_SIZE, int *)
|
|
#define AOM_CTRL_AV1D_GET_DISPLAY_SIZE
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_BIT_DEPTH, unsigned int *)
|
|
#define AOM_CTRL_AV1D_GET_BIT_DEPTH
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_IMG_FORMAT, aom_img_fmt_t *)
|
|
#define AOM_CTRL_AV1D_GET_IMG_FORMAT
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_TILE_SIZE, unsigned int *)
|
|
#define AOM_CTRL_AV1D_GET_TILE_SIZE
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_TILE_COUNT, unsigned int *)
|
|
#define AOM_CTRL_AV1D_GET_TILE_COUNT
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_FRAME_SIZE, int *)
|
|
#define AOM_CTRL_AV1D_GET_FRAME_SIZE
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_INVERT_TILE_DECODE_ORDER, int)
|
|
#define AOM_CTRL_AV1_INVERT_TILE_DECODE_ORDER
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_GET_ACCOUNTING, Accounting **)
|
|
#define AOM_CTRL_AV1_GET_ACCOUNTING
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_SET_DECODE_TILE_ROW, int)
|
|
#define AOM_CTRL_AV1_SET_DECODE_TILE_ROW
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_SET_DECODE_TILE_COL, int)
|
|
#define AOM_CTRL_AV1_SET_DECODE_TILE_COL
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_SET_TILE_MODE, unsigned int)
|
|
#define AOM_CTRL_AV1_SET_TILE_MODE
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_FRAME_HEADER_INFO, aom_tile_data *)
|
|
#define AOM_CTRL_AV1D_GET_FRAME_HEADER_INFO
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_GET_TILE_DATA, aom_tile_data *)
|
|
#define AOM_CTRL_AV1D_GET_TILE_DATA
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_EXT_REF_PTR, av1_ext_ref_frame_t *)
|
|
#define AOM_CTRL_AV1D_SET_EXT_REF_PTR
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_EXT_TILE_DEBUG, unsigned int)
|
|
#define AOM_CTRL_AV1D_EXT_TILE_DEBUG
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_ROW_MT, unsigned int)
|
|
#define AOM_CTRL_AV1D_SET_ROW_MT
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_SKIP_FILM_GRAIN, int)
|
|
#define AOM_CTRL_AV1D_SET_SKIP_FILM_GRAIN
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_IS_ANNEXB, unsigned int)
|
|
#define AOM_CTRL_AV1D_SET_IS_ANNEXB
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_OPERATING_POINT, int)
|
|
#define AOM_CTRL_AV1D_SET_OPERATING_POINT
|
|
|
|
AOM_CTRL_USE_TYPE(AV1D_SET_OUTPUT_ALL_LAYERS, int)
|
|
#define AOM_CTRL_AV1D_SET_OUTPUT_ALL_LAYERS
|
|
|
|
AOM_CTRL_USE_TYPE(AV1_SET_INSPECTION_CALLBACK, aom_inspect_init *)
|
|
#define AOM_CTRL_AV1_SET_INSPECTION_CALLBACK
|
|
/*!\endcond */
|
|
/*! @} - end defgroup aom_decoder */
|
|
#ifdef __cplusplus
|
|
} // extern "C"
|
|
#endif
|
|
|
|
#endif // AOM_AOM_AOMDX_H_
|