1407bdc243
This patch adds column-based tiling. The idea is to make each tile independently decodable (after reading the common frame header) and also independendly encodable (minus within-frame cost adjustments in the RD loop) to speed-up hardware & software en/decoders if they used multi-threading. Column-based tiling has the added advantage (over other tiling methods) that it minimizes realtime use-case latency, since all threads can start encoding data as soon as the first SB-row worth of data is available to the encoder. There is some test code that does random tile ordering in the decoder, to confirm that each tile is indeed independently decodable from other tiles in the same frame. At tile edges, all contexts assume default values (i.e. 0, 0 motion vector, no coefficients, DC intra4x4 mode), and motion vector search and ordering do not cross tiles in the same frame. t log Tile independence is not maintained between frames ATM, i.e. tile 0 of frame 1 is free to use motion vectors that point into any tile of frame 0. We support 1 (i.e. no tiling), 2 or 4 column-tiles. The loopfilter crosses tile boundaries. I discussed this briefly with Aki and he says that's OK. An in-loop loopfilter would need to do some sync between tile threads, but that shouldn't be a big issue. Resuls: with tiling disabled, we go up slightly because of improved edge use in the intra4x4 prediction. With 2 tiles, we lose about ~1% on derf, ~0.35% on HD and ~0.55% on STD/HD. With 4 tiles, we lose another ~1.5% on derf ~0.77% on HD and ~0.85% on STD/HD. Most of this loss is concentrated in the low-bitrate end of clips, and most of it is because of the loss of edges at tile boundaries and the resulting loss of intra predictors. TODO: - more tiles (perhaps allow row-based tiling also, and max. 8 tiles)? - maybe optionally (for EC purposes), motion vectors themselves should not cross tile edges, or we should emulate such borders as if they were off-frame, to limit error propagation to within one tile only. This doesn't have to be the default behaviour but could be an optional bitstream flag. Change-Id: I5951c3a0742a767b20bc9fb5af685d9892c2c96f
307 lines
9.0 KiB
C
307 lines
9.0 KiB
C
/*
|
|
* Copyright (c) 2010 The WebM project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef VP9_COMMON_VP9_ONYXC_INT_H_
|
|
#define VP9_COMMON_VP9_ONYXC_INT_H_
|
|
|
|
#include "vpx_config.h"
|
|
#include "vpx/internal/vpx_codec_internal.h"
|
|
#include "vp9_rtcd.h"
|
|
#include "vp9/common/vp9_loopfilter.h"
|
|
#include "vp9/common/vp9_entropymv.h"
|
|
#include "vp9/common/vp9_entropy.h"
|
|
#include "vp9/common/vp9_entropymode.h"
|
|
#if CONFIG_POSTPROC
|
|
#include "vp9/common/vp9_postproc.h"
|
|
#endif
|
|
|
|
/*#ifdef PACKET_TESTING*/
|
|
#include "vp9/common/vp9_header.h"
|
|
/*#endif*/
|
|
|
|
/* Create/destroy static data structures. */
|
|
|
|
void vp9_initialize_common(void);
|
|
|
|
#define MINQ 0
|
|
|
|
#define MAXQ 255
|
|
#define QINDEX_BITS 8
|
|
|
|
#define QINDEX_RANGE (MAXQ + 1)
|
|
|
|
#define NUM_REF_FRAMES 3
|
|
#define NUM_REF_FRAMES_LG2 2
|
|
#define NUM_YV12_BUFFERS (NUM_REF_FRAMES + 1)
|
|
|
|
#define NUM_FRAME_CONTEXTS_LG2 2
|
|
#define NUM_FRAME_CONTEXTS (1 << NUM_FRAME_CONTEXTS_LG2)
|
|
|
|
#define COMP_PRED_CONTEXTS 2
|
|
|
|
typedef struct frame_contexts {
|
|
vp9_prob bmode_prob[VP9_NKF_BINTRAMODES - 1];
|
|
vp9_prob ymode_prob[VP9_YMODES - 1]; /* interframe intra mode probs */
|
|
vp9_prob sb_ymode_prob[VP9_I32X32_MODES - 1];
|
|
vp9_prob uv_mode_prob[VP9_YMODES][VP9_UV_MODES - 1];
|
|
vp9_prob i8x8_mode_prob[VP9_I8X8_MODES - 1];
|
|
vp9_prob sub_mv_ref_prob[SUBMVREF_COUNT][VP9_SUBMVREFS - 1];
|
|
vp9_prob mbsplit_prob[VP9_NUMMBSPLITS - 1];
|
|
vp9_coeff_probs coef_probs_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_probs hybrid_coef_probs_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_probs coef_probs_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_probs hybrid_coef_probs_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_probs coef_probs_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_probs hybrid_coef_probs_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_probs coef_probs_32x32[BLOCK_TYPES_32X32];
|
|
|
|
nmv_context nmvc;
|
|
nmv_context pre_nmvc;
|
|
vp9_prob pre_bmode_prob[VP9_NKF_BINTRAMODES - 1];
|
|
vp9_prob pre_ymode_prob[VP9_YMODES - 1]; /* interframe intra mode probs */
|
|
vp9_prob pre_sb_ymode_prob[VP9_I32X32_MODES - 1];
|
|
vp9_prob pre_uv_mode_prob[VP9_YMODES][VP9_UV_MODES - 1];
|
|
vp9_prob pre_i8x8_mode_prob[VP9_I8X8_MODES - 1];
|
|
vp9_prob pre_sub_mv_ref_prob[SUBMVREF_COUNT][VP9_SUBMVREFS - 1];
|
|
vp9_prob pre_mbsplit_prob[VP9_NUMMBSPLITS - 1];
|
|
unsigned int bmode_counts[VP9_NKF_BINTRAMODES];
|
|
unsigned int ymode_counts[VP9_YMODES]; /* interframe intra mode probs */
|
|
unsigned int sb_ymode_counts[VP9_I32X32_MODES];
|
|
unsigned int uv_mode_counts[VP9_YMODES][VP9_UV_MODES];
|
|
unsigned int i8x8_mode_counts[VP9_I8X8_MODES]; /* interframe intra probs */
|
|
unsigned int sub_mv_ref_counts[SUBMVREF_COUNT][VP9_SUBMVREFS];
|
|
unsigned int mbsplit_counts[VP9_NUMMBSPLITS];
|
|
|
|
vp9_coeff_probs pre_coef_probs_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_probs pre_hybrid_coef_probs_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_probs pre_coef_probs_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_probs pre_hybrid_coef_probs_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_probs pre_coef_probs_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_probs pre_hybrid_coef_probs_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_probs pre_coef_probs_32x32[BLOCK_TYPES_32X32];
|
|
|
|
vp9_coeff_count coef_counts_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_count hybrid_coef_counts_4x4[BLOCK_TYPES_4X4];
|
|
vp9_coeff_count coef_counts_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_count hybrid_coef_counts_8x8[BLOCK_TYPES_8X8];
|
|
vp9_coeff_count coef_counts_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_count hybrid_coef_counts_16x16[BLOCK_TYPES_16X16];
|
|
vp9_coeff_count coef_counts_32x32[BLOCK_TYPES_32X32];
|
|
|
|
nmv_context_counts NMVcount;
|
|
vp9_prob switchable_interp_prob[VP9_SWITCHABLE_FILTERS + 1]
|
|
[VP9_SWITCHABLE_FILTERS - 1];
|
|
#if CONFIG_COMP_INTERINTRA_PRED
|
|
unsigned int interintra_counts[2];
|
|
vp9_prob interintra_prob;
|
|
vp9_prob pre_interintra_prob;
|
|
#endif
|
|
|
|
int vp9_mode_contexts[INTER_MODE_CONTEXTS][4];
|
|
unsigned int mv_ref_ct[INTER_MODE_CONTEXTS][4][2];
|
|
} FRAME_CONTEXT;
|
|
|
|
typedef enum {
|
|
RECON_CLAMP_REQUIRED = 0,
|
|
RECON_CLAMP_NOTREQUIRED = 1
|
|
} CLAMP_TYPE;
|
|
|
|
typedef enum {
|
|
SINGLE_PREDICTION_ONLY = 0,
|
|
COMP_PREDICTION_ONLY = 1,
|
|
HYBRID_PREDICTION = 2,
|
|
NB_PREDICTION_TYPES = 3,
|
|
} COMPPREDMODE_TYPE;
|
|
|
|
typedef enum {
|
|
ONLY_4X4 = 0,
|
|
ALLOW_8X8 = 1,
|
|
ALLOW_16X16 = 2,
|
|
ALLOW_32X32 = 3,
|
|
TX_MODE_SELECT = 4,
|
|
NB_TXFM_MODES = 5,
|
|
} TXFM_MODE;
|
|
|
|
typedef struct VP9Common {
|
|
struct vpx_internal_error_info error;
|
|
|
|
DECLARE_ALIGNED(16, int16_t, Y1dequant[QINDEX_RANGE][16]);
|
|
DECLARE_ALIGNED(16, int16_t, Y2dequant[QINDEX_RANGE][16]);
|
|
DECLARE_ALIGNED(16, int16_t, UVdequant[QINDEX_RANGE][16]);
|
|
|
|
int Width;
|
|
int Height;
|
|
int horiz_scale;
|
|
int vert_scale;
|
|
|
|
YUV_TYPE clr_type;
|
|
CLAMP_TYPE clamp_type;
|
|
|
|
YV12_BUFFER_CONFIG *frame_to_show;
|
|
|
|
YV12_BUFFER_CONFIG yv12_fb[NUM_YV12_BUFFERS];
|
|
int fb_idx_ref_cnt[NUM_YV12_BUFFERS]; /* reference counts */
|
|
int ref_frame_map[NUM_REF_FRAMES]; /* maps fb_idx to reference slot */
|
|
|
|
/* TODO(jkoleszar): could expand active_ref_idx to 4, with 0 as intra, and
|
|
* roll new_fb_idx into it.
|
|
*/
|
|
int active_ref_idx[3]; /* each frame can reference 3 buffers */
|
|
int new_fb_idx;
|
|
|
|
YV12_BUFFER_CONFIG post_proc_buffer;
|
|
YV12_BUFFER_CONFIG temp_scale_frame;
|
|
|
|
|
|
FRAME_TYPE last_frame_type; /* Save last frame's frame type for motion search. */
|
|
FRAME_TYPE frame_type;
|
|
|
|
int show_frame;
|
|
|
|
int frame_flags;
|
|
int MBs;
|
|
int mb_rows;
|
|
int mb_cols;
|
|
int mode_info_stride;
|
|
|
|
/* profile settings */
|
|
int experimental;
|
|
int mb_no_coeff_skip;
|
|
TXFM_MODE txfm_mode;
|
|
COMPPREDMODE_TYPE comp_pred_mode;
|
|
int no_lpf;
|
|
int use_bilinear_mc_filter;
|
|
int full_pixel;
|
|
|
|
int base_qindex;
|
|
int last_kf_gf_q; /* Q used on the last GF or KF */
|
|
|
|
int y1dc_delta_q;
|
|
int y2dc_delta_q;
|
|
int y2ac_delta_q;
|
|
int uvdc_delta_q;
|
|
int uvac_delta_q;
|
|
|
|
unsigned int frames_since_golden;
|
|
unsigned int frames_till_alt_ref_frame;
|
|
|
|
/* We allocate a MODE_INFO struct for each macroblock, together with
|
|
an extra row on top and column on the left to simplify prediction. */
|
|
|
|
MODE_INFO *mip; /* Base of allocated array */
|
|
MODE_INFO *mi; /* Corresponds to upper left visible macroblock */
|
|
MODE_INFO *prev_mip; /* MODE_INFO array 'mip' from last decoded frame */
|
|
MODE_INFO *prev_mi; /* 'mi' from last frame (points into prev_mip) */
|
|
|
|
|
|
// Persistent mb segment id map used in prediction.
|
|
unsigned char *last_frame_seg_map;
|
|
|
|
INTERPOLATIONFILTERTYPE mcomp_filter_type;
|
|
LOOPFILTERTYPE filter_type;
|
|
|
|
loop_filter_info_n lf_info;
|
|
|
|
int filter_level;
|
|
int last_sharpness_level;
|
|
int sharpness_level;
|
|
|
|
int refresh_entropy_probs; /* Two state 0 = NO, 1 = YES */
|
|
|
|
int ref_frame_sign_bias[MAX_REF_FRAMES]; /* Two state 0, 1 */
|
|
|
|
/* Y,U,V,Y2 */
|
|
ENTROPY_CONTEXT_PLANES *above_context; /* row of context for each plane */
|
|
ENTROPY_CONTEXT_PLANES left_context[4]; /* (up to) 4 contexts "" */
|
|
|
|
/* keyframe block modes are predicted by their above, left neighbors */
|
|
|
|
vp9_prob kf_bmode_prob[VP9_KF_BINTRAMODES]
|
|
[VP9_KF_BINTRAMODES]
|
|
[VP9_KF_BINTRAMODES - 1];
|
|
vp9_prob kf_ymode_prob[8][VP9_YMODES - 1]; /* keyframe "" */
|
|
vp9_prob sb_kf_ymode_prob[8][VP9_I32X32_MODES - 1];
|
|
int kf_ymode_probs_index;
|
|
int kf_ymode_probs_update;
|
|
vp9_prob kf_uv_mode_prob[VP9_YMODES] [VP9_UV_MODES - 1];
|
|
|
|
vp9_prob prob_intra_coded;
|
|
vp9_prob prob_last_coded;
|
|
vp9_prob prob_gf_coded;
|
|
vp9_prob sb32_coded;
|
|
vp9_prob sb64_coded;
|
|
|
|
// Context probabilities when using predictive coding of segment id
|
|
vp9_prob segment_pred_probs[PREDICTION_PROBS];
|
|
unsigned char temporal_update;
|
|
|
|
// Context probabilities for reference frame prediction
|
|
unsigned char ref_scores[MAX_REF_FRAMES];
|
|
vp9_prob ref_pred_probs[PREDICTION_PROBS];
|
|
vp9_prob mod_refprobs[MAX_REF_FRAMES][PREDICTION_PROBS];
|
|
|
|
vp9_prob prob_comppred[COMP_PRED_CONTEXTS];
|
|
|
|
// FIXME contextualize
|
|
vp9_prob prob_tx[TX_SIZE_MAX_SB - 1];
|
|
|
|
vp9_prob mbskip_pred_probs[MBSKIP_CONTEXTS];
|
|
|
|
FRAME_CONTEXT fc; /* this frame entropy */
|
|
FRAME_CONTEXT frame_contexts[NUM_FRAME_CONTEXTS];
|
|
unsigned int frame_context_idx; /* Context to use/update */
|
|
|
|
unsigned int current_video_frame;
|
|
int near_boffset[3];
|
|
int version;
|
|
|
|
#ifdef PACKET_TESTING
|
|
VP9_HEADER oh;
|
|
#endif
|
|
double bitrate;
|
|
double framerate;
|
|
|
|
#if CONFIG_POSTPROC
|
|
struct postproc_state postproc_state;
|
|
#endif
|
|
|
|
#if CONFIG_COMP_INTERINTRA_PRED
|
|
int use_interintra;
|
|
#endif
|
|
|
|
int error_resilient_mode;
|
|
int frame_parallel_decoding_mode;
|
|
|
|
int tile_columns;
|
|
int cur_tile_mb_col_start, cur_tile_mb_col_end, cur_tile_idx;
|
|
} VP9_COMMON;
|
|
|
|
static int get_free_fb(VP9_COMMON *cm) {
|
|
int i;
|
|
for (i = 0; i < NUM_YV12_BUFFERS; i++)
|
|
if (cm->fb_idx_ref_cnt[i] == 0)
|
|
break;
|
|
|
|
assert(i < NUM_YV12_BUFFERS);
|
|
cm->fb_idx_ref_cnt[i] = 1;
|
|
return i;
|
|
}
|
|
|
|
static void ref_cnt_fb(int *buf, int *idx, int new_idx) {
|
|
if (buf[*idx] > 0)
|
|
buf[*idx]--;
|
|
|
|
*idx = new_idx;
|
|
|
|
buf[new_idx]++;
|
|
}
|
|
|
|
#endif // VP9_COMMON_VP9_ONYXC_INT_H_
|