• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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   int16_t *quant_fp;
38   int16_t *round_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 mb_energy;
97 
98   // These are set to their default values at the beginning, and then adjusted
99   // further in the encoding process.
100   BLOCK_SIZE min_partition_size;
101   BLOCK_SIZE max_partition_size;
102 
103   int mv_best_ref_index[MAX_REF_FRAMES];
104   unsigned int max_mv_context[MAX_REF_FRAMES];
105   unsigned int source_variance;
106   unsigned int pred_sse[MAX_REF_FRAMES];
107   int pred_mv_sad[MAX_REF_FRAMES];
108 
109   int nmvjointcost[MV_JOINTS];
110   int *nmvcost[2];
111   int *nmvcost_hp[2];
112   int **mvcost;
113 
114   int nmvjointsadcost[MV_JOINTS];
115   int *nmvsadcost[2];
116   int *nmvsadcost_hp[2];
117   int **mvsadcost;
118 
119   // sharpness is used to disable skip mode and change rd_mult
120   int sharpness;
121 
122   // aq mode is used to adjust rd based on segment.
123   int adjust_rdmult_by_segment;
124 
125   // These define limits to motion vector components to prevent them
126   // from extending outside the UMV borders
127   MvLimits mv_limits;
128 
129   // Notes transform blocks where no coefficents are coded.
130   // Set during mode selection. Read during block encoding.
131   uint8_t zcoeff_blk[TX_SIZES][256];
132 
133   // Accumulate the tx block eobs in a partition block.
134   int32_t sum_y_eobs[TX_SIZES];
135 
136   int skip;
137 
138   int encode_breakout;
139 
140   // note that token_costs is the cost when eob node is skipped
141   vp9_coeff_cost token_costs[TX_SIZES];
142 
143   int optimize;
144 
145   // indicate if it is in the rd search loop or encoding process
146   int use_lp32x32fdct;
147   int skip_encode;
148 
149   // In first pass, intra prediction is done based on source pixels
150   // at tile boundaries
151   int fp_src_pred;
152 
153   // use fast quantization process
154   int quant_fp;
155 
156   // skip forward transform and quantization
157   uint8_t skip_txfm[MAX_MB_PLANE << 2];
158 #define SKIP_TXFM_NONE 0
159 #define SKIP_TXFM_AC_DC 1
160 #define SKIP_TXFM_AC_ONLY 2
161 
162 // cf. https://bugs.chromium.org/p/webm/issues/detail?id=1054
163 #if !defined(_MSC_VER) || _MSC_VER >= 1900
164   int64_t bsse[MAX_MB_PLANE << 2];
165 #endif
166 
167   // Used to store sub partition's choices.
168   MV pred_mv[MAX_REF_FRAMES];
169 
170   // Strong color activity detection. Used in RTC coding mode to enhance
171   // the visual quality at the boundary of moving color objects.
172   uint8_t color_sensitivity[2];
173 
174   uint8_t sb_is_skin;
175 
176   uint8_t skip_low_source_sad;
177 
178   uint8_t lowvar_highsumdiff;
179 
180   uint8_t last_sb_high_content;
181 
182   int sb_use_mv_part;
183 
184   int sb_mvcol_part;
185 
186   int sb_mvrow_part;
187 
188   int sb_pickmode_part;
189 
190   int zero_temp_sad_source;
191 
192   // For each superblock: saves the content value (e.g., low/high sad/sumdiff)
193   // based on source sad, prior to encoding the frame.
194   uint8_t content_state_sb;
195 
196   // Used to save the status of whether a block has a low variance in
197   // choose_partitioning. 0 for 64x64, 1~2 for 64x32, 3~4 for 32x64, 5~8 for
198   // 32x32, 9~24 for 16x16.
199   uint8_t variance_low[25];
200 
201   uint8_t arf_frame_usage;
202   uint8_t lastgolden_frame_usage;
203 
204   void (*fwd_txfm4x4)(const int16_t *input, tran_low_t *output, int stride);
205   void (*inv_txfm_add)(const tran_low_t *input, uint8_t *dest, int stride,
206                        int eob);
207 #if CONFIG_VP9_HIGHBITDEPTH
208   void (*highbd_inv_txfm_add)(const tran_low_t *input, uint16_t *dest,
209                               int stride, int eob, int bd);
210 #endif
211 #if CONFIG_ML_VAR_PARTITION
212   DECLARE_ALIGNED(16, uint8_t, est_pred[64 * 64]);
213 #endif  // CONFIG_ML_VAR_PARTITION
214 
215   struct scale_factors *me_sf;
216 };
217 
218 #ifdef __cplusplus
219 }  // extern "C"
220 #endif
221 
222 #endif  // VPX_VP9_ENCODER_VP9_BLOCK_H_
223