2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
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.
11 #ifndef VPX_VP9_ENCODER_VP9_BLOCK_H_
12 #define VPX_VP9_ENCODER_VP9_BLOCK_H_
14 #include "vpx_util/vpx_thread.h"
16 #include "vp9/common/vp9_entropymv.h"
17 #include "vp9/common/vp9_entropy.h"
29 struct macroblock_plane {
30 DECLARE_ALIGNED(16, int16_t, src_diff[64 * 64]);
37 DECLARE_ALIGNED(16, int16_t, round_fp[8]);
44 int64_t quant_thred[2];
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];
53 int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REF_CANDIDATES];
54 uint8_t mode_context[MAX_REF_FRAMES];
64 typedef struct macroblock 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];
71 struct macroblock_plane plane[MAX_MB_PLANE];
74 MB_MODE_INFO_EXT *mbmi_ext;
75 MB_MODE_INFO_EXT *mbmi_ext_base;
84 // The equivalent error at the current rdmult of one whole bit (not one
87 // The equivalend SAD error of one (whole) bit at the current quantizer
90 // The equivalend SAD error of one (whole) bit at the current quantizer
91 // for sub-8x8 blocks.
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;
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];
110 int nmvjointcost[MV_JOINTS];
115 int nmvjointsadcost[MV_JOINTS];
117 int *nmvsadcost_hp[2];
120 // sharpness is used to disable skip mode and change rd_mult
123 // aq mode is used to adjust rd based on segment.
124 int adjust_rdmult_by_segment;
126 // These define limits to motion vector components to prevent them
127 // from extending outside the UMV borders
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];
134 // Accumulate the tx block eobs in a partition block.
135 int32_t sum_y_eobs[TX_SIZES];
141 // note that token_costs is the cost when eob node is skipped
142 vp9_coeff_cost token_costs[TX_SIZES];
146 // indicate if it is in the rd search loop or encoding process
150 // In first pass, intra prediction is done based on source pixels
151 // at tile boundaries
154 // use fast quantization process
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
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];
168 // Used to store sub partition's choices.
169 MV pred_mv[MAX_REF_FRAMES];
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];
177 uint8_t skip_low_source_sad;
179 uint8_t lowvar_highsumdiff;
181 uint8_t last_sb_high_content;
189 int sb_pickmode_part;
191 int zero_temp_sad_source;
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;
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];
202 uint8_t arf_frame_usage;
203 uint8_t lastgolden_frame_usage;
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,
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);
212 DECLARE_ALIGNED(16, uint8_t, est_pred[64 * 64]);
214 struct scale_factors *me_sf;
221 #endif // VPX_VP9_ENCODER_VP9_BLOCK_H_