123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213 |
- /*
- * This file is part of FFmpeg.
- *
- * FFmpeg is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * FFmpeg is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with FFmpeg; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- */
- #ifndef AVCODEC_CBS_VP9_H
- #define AVCODEC_CBS_VP9_H
- #include <stddef.h>
- #include <stdint.h>
- #include "cbs.h"
- // Miscellaneous constants (section 3).
- enum {
- VP9_REFS_PER_FRAME = 3,
- VP9_MIN_TILE_WIDTH_B64 = 4,
- VP9_MAX_TILE_WIDTH_B64 = 64,
- VP9_NUM_REF_FRAMES = 8,
- VP9_MAX_REF_FRAMES = 4,
- VP9_MAX_SEGMENTS = 8,
- VP9_SEG_LVL_MAX = 4,
- };
- // Frame types (section 7.2).
- enum {
- VP9_KEY_FRAME = 0,
- VP9_NON_KEY_FRAME = 1,
- };
- // Frame sync bytes (section 7.2.1).
- enum {
- VP9_FRAME_SYNC_0 = 0x49,
- VP9_FRAME_SYNC_1 = 0x83,
- VP9_FRAME_SYNC_2 = 0x42,
- };
- // Color space values (section 7.2.2).
- enum {
- VP9_CS_UNKNOWN = 0,
- VP9_CS_BT_601 = 1,
- VP9_CS_BT_709 = 2,
- VP9_CS_SMPTE_170 = 3,
- VP9_CS_SMPTE_240 = 4,
- VP9_CS_BT_2020 = 5,
- VP9_CS_RESERVED = 6,
- VP9_CS_RGB = 7,
- };
- // Reference frame types (section 7.4.12).
- enum {
- VP9_INTRA_FRAME = 0,
- VP9_LAST_FRAME = 1,
- VP9_GOLDEN_FRAME = 2,
- VP9_ALTREF_FRAME = 3,
- };
- // Superframe properties (section B.3).
- enum {
- VP9_MAX_FRAMES_IN_SUPERFRAME = 8,
- VP9_SUPERFRAME_MARKER = 6,
- };
- typedef struct VP9RawFrameHeader {
- uint8_t frame_marker;
- uint8_t profile_low_bit;
- uint8_t profile_high_bit;
- uint8_t show_existing_frame;
- uint8_t frame_to_show_map_idx;
- uint8_t frame_type;
- uint8_t show_frame;
- uint8_t error_resilient_mode;
- // Color config.
- uint8_t ten_or_twelve_bit;
- uint8_t color_space;
- uint8_t color_range;
- uint8_t subsampling_x;
- uint8_t subsampling_y;
- uint8_t refresh_frame_flags;
- uint8_t intra_only;
- uint8_t reset_frame_context;
- uint8_t ref_frame_idx[VP9_REFS_PER_FRAME];
- uint8_t ref_frame_sign_bias[VP9_MAX_REF_FRAMES];
- uint8_t allow_high_precision_mv;
- uint8_t refresh_frame_context;
- uint8_t frame_parallel_decoding_mode;
- uint8_t frame_context_idx;
- // Frame/render size.
- uint8_t found_ref[VP9_REFS_PER_FRAME];
- uint16_t frame_width_minus_1;
- uint16_t frame_height_minus_1;
- uint8_t render_and_frame_size_different;
- uint16_t render_width_minus_1;
- uint16_t render_height_minus_1;
- // Interpolation filter.
- uint8_t is_filter_switchable;
- uint8_t raw_interpolation_filter_type;
- // Loop filter params.
- uint8_t loop_filter_level;
- uint8_t loop_filter_sharpness;
- uint8_t loop_filter_delta_enabled;
- uint8_t loop_filter_delta_update;
- uint8_t update_ref_delta[VP9_MAX_REF_FRAMES];
- int8_t loop_filter_ref_deltas[VP9_MAX_REF_FRAMES];
- uint8_t update_mode_delta[2];
- int8_t loop_filter_mode_deltas[2];
- // Quantization params.
- uint8_t base_q_idx;
- int8_t delta_q_y_dc;
- int8_t delta_q_uv_dc;
- int8_t delta_q_uv_ac;
- // Segmentation params.
- uint8_t segmentation_enabled;
- uint8_t segmentation_update_map;
- uint8_t segmentation_tree_probs[7];
- uint8_t segmentation_temporal_update;
- uint8_t segmentation_pred_prob[3];
- uint8_t segmentation_update_data;
- uint8_t segmentation_abs_or_delta_update;
- uint8_t feature_enabled[VP9_MAX_SEGMENTS][VP9_SEG_LVL_MAX];
- uint8_t feature_value[VP9_MAX_SEGMENTS][VP9_SEG_LVL_MAX];
- uint8_t feature_sign[VP9_MAX_SEGMENTS][VP9_SEG_LVL_MAX];
- // Tile info.
- uint8_t tile_cols_log2;
- uint8_t tile_rows_log2;
- uint16_t header_size_in_bytes;
- } VP9RawFrameHeader;
- typedef struct VP9RawFrame {
- VP9RawFrameHeader header;
- uint8_t *data;
- size_t data_size;
- AVBufferRef *data_ref;
- } VP9RawFrame;
- typedef struct VP9RawSuperframeIndex {
- uint8_t superframe_marker;
- uint8_t bytes_per_framesize_minus_1;
- uint8_t frames_in_superframe_minus_1;
- uint32_t frame_sizes[VP9_MAX_FRAMES_IN_SUPERFRAME];
- } VP9RawSuperframeIndex;
- typedef struct VP9RawSuperframe {
- VP9RawFrame frames[VP9_MAX_FRAMES_IN_SUPERFRAME];
- VP9RawSuperframeIndex index;
- } VP9RawSuperframe;
- typedef struct VP9ReferenceFrameState {
- int frame_width; // RefFrameWidth
- int frame_height; // RefFrameHeight
- int subsampling_x; // RefSubsamplingX
- int subsampling_y; // RefSubsamplingY
- int bit_depth; // RefBitDepth
- } VP9ReferenceFrameState;
- typedef struct CodedBitstreamVP9Context {
- int profile;
- // Frame dimensions in 8x8 mode info blocks.
- uint16_t mi_cols;
- uint16_t mi_rows;
- // Frame dimensions in 64x64 superblocks.
- uint16_t sb64_cols;
- uint16_t sb64_rows;
- int frame_width;
- int frame_height;
- uint8_t subsampling_x;
- uint8_t subsampling_y;
- int bit_depth;
- VP9ReferenceFrameState ref[VP9_NUM_REF_FRAMES];
- } CodedBitstreamVP9Context;
- #endif /* AVCODEC_CBS_VP9_H */
|