1 /* 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. 3 * 4 * Use of this source code is governed by a BSD-style license 5 * that can be found in the LICENSE file in the root of the source 6 * tree. An additional intellectual property rights grant can be found 7 * in the file PATENTS. All contributing project authors may 8 * be found in the AUTHORS file in the root of the source tree. 9 */ 10 11 #ifndef VPX_VP9_ENCODER_VP9_BLOCK_H_ 12 #define VPX_VP9_ENCODER_VP9_BLOCK_H_ 13 14 #include "vpx_util/vpx_thread.h" 15 16 #include "vp9/common/vp9_entropymv.h" 17 #include "vp9/common/vp9_entropy.h" 18 19 #ifdef __cplusplus 20 extern "C" { 21 #endif 22 23 typedef struct { 24 unsigned int sse; 25 int sum; 26 unsigned int var; 27 } diff; 28 29 struct macroblock_plane { 30 DECLARE_ALIGNED(16, int16_t, src_diff[64 * 64]); 31 tran_low_t *qcoeff; 32 tran_low_t *coeff; 33 uint16_t *eobs; 34 struct buf_2d src; 35 36 // Quantizer setings 37 DECLARE_ALIGNED(16, int16_t, round_fp[8]); 38 int16_t *quant_fp; 39 int16_t *quant; 40 int16_t *quant_shift; 41 int16_t *zbin; 42 int16_t *round; 43 44 int64_t quant_thred[2]; 45 }; 46 47 /* The [2] dimension is for whether we skip the EOB node (i.e. if previous 48 * coefficient in this block was zero) or not. */ 49 typedef unsigned int vp9_coeff_cost[PLANE_TYPES][REF_TYPES][COEF_BANDS][2] 50 [COEFF_CONTEXTS][ENTROPY_TOKENS]; 51 52 typedef struct { 53 int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REF_CANDIDATES]; 54 uint8_t mode_context[MAX_REF_FRAMES]; 55 } MB_MODE_INFO_EXT; 56 57 typedef struct { 58 int col_min; 59 int col_max; 60 int row_min; 61 int row_max; 62 } MvLimits; 63 64 typedef struct macroblock MACROBLOCK; 65 struct macroblock { 66 // cf. https://bugs.chromium.org/p/webm/issues/detail?id=1054 67 #if defined(_MSC_VER) && _MSC_VER < 1900 68 int64_t bsse[MAX_MB_PLANE << 2]; 69 #endif 70 71 struct macroblock_plane plane[MAX_MB_PLANE]; 72 73 MACROBLOCKD e_mbd; 74 MB_MODE_INFO_EXT *mbmi_ext; 75 MB_MODE_INFO_EXT *mbmi_ext_base; 76 int skip_block; 77 int select_tx_size; 78 int skip_recode; 79 int skip_optimize; 80 int q_index; 81 int block_qcoeff_opt; 82 int block_tx_domain; 83 84 // The equivalent error at the current rdmult of one whole bit (not one 85 // bitcost unit). 86 int errorperbit; 87 // The equivalend SAD error of one (whole) bit at the current quantizer 88 // for large blocks. 89 int sadperbit16; 90 // The equivalend SAD error of one (whole) bit at the current quantizer 91 // for sub-8x8 blocks. 92 int sadperbit4; 93 int rddiv; 94 int rdmult; 95 int cb_rdmult; 96 int segment_id; 97 int mb_energy; 98 99 // These are set to their default values at the beginning, and then adjusted 100 // further in the encoding process. 101 BLOCK_SIZE min_partition_size; 102 BLOCK_SIZE max_partition_size; 103 104 int mv_best_ref_index[MAX_REF_FRAMES]; 105 unsigned int max_mv_context[MAX_REF_FRAMES]; 106 unsigned int source_variance; 107 unsigned int pred_sse[MAX_REF_FRAMES]; 108 int pred_mv_sad[MAX_REF_FRAMES]; 109 110 int nmvjointcost[MV_JOINTS]; 111 int *nmvcost[2]; 112 int *nmvcost_hp[2]; 113 int **mvcost; 114 115 int nmvjointsadcost[MV_JOINTS]; 116 int *nmvsadcost[2]; 117 int *nmvsadcost_hp[2]; 118 int **mvsadcost; 119 120 // sharpness is used to disable skip mode and change rd_mult 121 int sharpness; 122 123 // aq mode is used to adjust rd based on segment. 124 int adjust_rdmult_by_segment; 125 126 // These define limits to motion vector components to prevent them 127 // from extending outside the UMV borders 128 MvLimits mv_limits; 129 130 // Notes transform blocks where no coefficents are coded. 131 // Set during mode selection. Read during block encoding. 132 uint8_t zcoeff_blk[TX_SIZES][256]; 133 134 // Accumulate the tx block eobs in a partition block. 135 int32_t sum_y_eobs[TX_SIZES]; 136 137 int skip; 138 139 int encode_breakout; 140 141 // note that token_costs is the cost when eob node is skipped 142 vp9_coeff_cost token_costs[TX_SIZES]; 143 144 int optimize; 145 146 // indicate if it is in the rd search loop or encoding process 147 int use_lp32x32fdct; 148 int skip_encode; 149 150 // In first pass, intra prediction is done based on source pixels 151 // at tile boundaries 152 int fp_src_pred; 153 154 // use fast quantization process 155 int quant_fp; 156 157 // skip forward transform and quantization 158 uint8_t skip_txfm[MAX_MB_PLANE << 2]; 159 #define SKIP_TXFM_NONE 0 160 #define SKIP_TXFM_AC_DC 1 161 #define SKIP_TXFM_AC_ONLY 2 162 163 // cf. https://bugs.chromium.org/p/webm/issues/detail?id=1054 164 #if !defined(_MSC_VER) || _MSC_VER >= 1900 165 int64_t bsse[MAX_MB_PLANE << 2]; 166 #endif 167 168 // Used to store sub partition's choices. 169 MV pred_mv[MAX_REF_FRAMES]; 170 171 // Strong color activity detection. Used in RTC coding mode to enhance 172 // the visual quality at the boundary of moving color objects. 173 uint8_t color_sensitivity[2]; 174 175 uint8_t sb_is_skin; 176 177 uint8_t skip_low_source_sad; 178 179 uint8_t lowvar_highsumdiff; 180 181 uint8_t last_sb_high_content; 182 183 int sb_use_mv_part; 184 185 int sb_mvcol_part; 186 187 int sb_mvrow_part; 188 189 int sb_pickmode_part; 190 191 int zero_temp_sad_source; 192 193 // For each superblock: saves the content value (e.g., low/high sad/sumdiff) 194 // based on source sad, prior to encoding the frame. 195 uint8_t content_state_sb; 196 197 // Used to save the status of whether a block has a low variance in 198 // choose_partitioning. 0 for 64x64, 1~2 for 64x32, 3~4 for 32x64, 5~8 for 199 // 32x32, 9~24 for 16x16. 200 uint8_t variance_low[25]; 201 202 uint8_t arf_frame_usage; 203 uint8_t lastgolden_frame_usage; 204 205 void (*fwd_txfm4x4)(const int16_t *input, tran_low_t *output, int stride); 206 void (*inv_txfm_add)(const tran_low_t *input, uint8_t *dest, int stride, 207 int eob); 208 #if CONFIG_VP9_HIGHBITDEPTH 209 void (*highbd_inv_txfm_add)(const tran_low_t *input, uint16_t *dest, 210 int stride, int eob, int bd); 211 #endif 212 DECLARE_ALIGNED(16, uint8_t, est_pred[64 * 64]); 213 214 struct scale_factors *me_sf; 215 }; 216 217 #ifdef __cplusplus 218 } // extern "C" 219 #endif 220 221 #endif // VPX_VP9_ENCODER_VP9_BLOCK_H_ 222