/*
 * 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_