12#ifndef AOM_AV1_ENCODER_NONRD_OPT_H_
13#define AOM_AV1_ENCODER_NONRD_OPT_H_
15#include "av1/encoder/rdopt_utils.h"
16#include "av1/encoder/rdopt.h"
18#define RTC_INTER_MODES (4)
19#define RTC_INTRA_MODES (4)
20#define RTC_MODES (AOMMAX(RTC_INTER_MODES, RTC_INTRA_MODES))
21#define CALC_BIASED_RDCOST(rdcost) (7 * (rdcost) >> 3)
22#define NUM_COMP_INTER_MODES_RT (6)
23#define NUM_INTER_MODES 12
24#define CAP_TX_SIZE_FOR_BSIZE_GT32(tx_mode_search_type, bsize) \
25 (((tx_mode_search_type) != ONLY_4X4 && (bsize) > BLOCK_32X32) ? true : false)
26#define TX_SIZE_FOR_BSIZE_GT32 (TX_16X16)
27#define FILTER_SEARCH_SIZE 2
28#if !CONFIG_REALTIME_ONLY
29#define MOTION_MODE_SEARCH_SIZE 2
32extern int g_pick_inter_mode_cnt;
41 PRED_BUFFER *best_pred;
42 PREDICTION_MODE best_mode;
45 MV_REFERENCE_FRAME best_ref_frame;
46 MV_REFERENCE_FRAME best_second_ref_frame;
47 uint8_t best_mode_skip_txfm;
48 uint8_t best_mode_initial_skip_flag;
49 int_interpfilters best_pred_filter;
50 MOTION_MODE best_motion_mode;
51 WarpedMotionParams wm_params;
53 PALETTE_MODE_INFO pmi;
58 MV_REFERENCE_FRAME ref_frame;
59 PREDICTION_MODE pred_mode;
63 MV_REFERENCE_FRAME ref_frame[2];
64 PREDICTION_MODE pred_mode;
67struct estimate_block_intra_args {
73 unsigned int best_sad;
74 bool prune_mode_based_on_sad;
89 int64_t
uv_dist[RTC_INTER_MODES][REF_FRAMES];
91 struct buf_2d
yv12_mb[REF_FRAMES][MAX_MB_PLANE];
93 unsigned int vars[RTC_INTER_MODES][REF_FRAMES];
108static const uint8_t b_width_log2_lookup[BLOCK_SIZES] = { 0, 0, 1, 1, 1, 2,
111static const uint8_t b_height_log2_lookup[BLOCK_SIZES] = { 0, 1, 0, 1, 2, 1,
115static const PREDICTION_MODE intra_mode_list[] = { DC_PRED, V_PRED, H_PRED,
118static const PREDICTION_MODE inter_mode_list[] = { NEARESTMV, NEARMV, GLOBALMV,
121static const THR_MODES mode_idx[REF_FRAMES][RTC_MODES] = {
122 { THR_DC, THR_V_PRED, THR_H_PRED, THR_SMOOTH },
123 { THR_NEARESTMV, THR_NEARMV, THR_GLOBALMV, THR_NEWMV },
124 { THR_NEARESTL2, THR_NEARL2, THR_GLOBALL2, THR_NEWL2 },
125 { THR_NEARESTL3, THR_NEARL3, THR_GLOBALL3, THR_NEWL3 },
126 { THR_NEARESTG, THR_NEARG, THR_GLOBALG, THR_NEWG },
127 { THR_NEARESTB, THR_NEARB, THR_GLOBALB, THR_NEWB },
128 { THR_NEARESTA2, THR_NEARA2, THR_GLOBALA2, THR_NEWA2 },
129 { THR_NEARESTA, THR_NEARA, THR_GLOBALA, THR_NEWA },
134static const REF_MODE ref_mode_set[NUM_INTER_MODES] = {
135 { LAST_FRAME, NEARESTMV }, { LAST_FRAME, NEARMV },
136 { LAST_FRAME, GLOBALMV }, { LAST_FRAME, NEWMV },
137 { GOLDEN_FRAME, NEARESTMV }, { GOLDEN_FRAME, NEARMV },
138 { GOLDEN_FRAME, GLOBALMV }, { GOLDEN_FRAME, NEWMV },
139 { ALTREF_FRAME, NEARESTMV }, { ALTREF_FRAME, NEARMV },
140 { ALTREF_FRAME, GLOBALMV }, { ALTREF_FRAME, NEWMV },
143static const COMP_REF_MODE comp_ref_mode_set[NUM_COMP_INTER_MODES_RT] = {
144 { { LAST_FRAME, GOLDEN_FRAME }, GLOBAL_GLOBALMV },
145 { { LAST_FRAME, GOLDEN_FRAME }, NEAREST_NEARESTMV },
146 { { LAST_FRAME, LAST2_FRAME }, GLOBAL_GLOBALMV },
147 { { LAST_FRAME, LAST2_FRAME }, NEAREST_NEARESTMV },
148 { { LAST_FRAME, ALTREF_FRAME }, GLOBAL_GLOBALMV },
149 { { LAST_FRAME, ALTREF_FRAME }, NEAREST_NEARESTMV },
152static const int_interpfilters filters_ref_set[9] = {
153 [0].as_filters = { EIGHTTAP_REGULAR, EIGHTTAP_REGULAR },
154 [1].as_filters = { EIGHTTAP_SMOOTH, EIGHTTAP_SMOOTH },
155 [2].as_filters = { EIGHTTAP_REGULAR, EIGHTTAP_SMOOTH },
156 [3].as_filters = { EIGHTTAP_SMOOTH, EIGHTTAP_REGULAR },
157 [4].as_filters = { MULTITAP_SHARP, MULTITAP_SHARP },
158 [5].as_filters = { EIGHTTAP_REGULAR, MULTITAP_SHARP },
159 [6].as_filters = { MULTITAP_SHARP, EIGHTTAP_REGULAR },
160 [7].as_filters = { EIGHTTAP_SMOOTH, MULTITAP_SHARP },
161 [8].as_filters = { MULTITAP_SHARP, EIGHTTAP_SMOOTH }
166 INTER_NEAREST = (1 << NEARESTMV),
167 INTER_NEAREST_NEW = (1 << NEARESTMV) | (1 << NEWMV),
168 INTER_NEAREST_NEAR = (1 << NEARESTMV) | (1 << NEARMV),
169 INTER_NEAR_NEW = (1 << NEARMV) | (1 << NEWMV),
175DECLARE_ALIGNED(16,
static const int16_t, default_scan_8x8_transpose[64]) = {
176 0, 8, 1, 2, 9, 16, 24, 17, 10, 3, 4, 11, 18, 25, 32, 40,
177 33, 26, 19, 12, 5, 6, 13, 20, 27, 34, 41, 48, 56, 49, 42, 35,
178 28, 21, 14, 7, 15, 22, 29, 36, 43, 50, 57, 58, 51, 44, 37, 30,
179 23, 31, 38, 45, 52, 59, 60, 53, 46, 39, 47, 54, 61, 62, 55, 63
189DECLARE_ALIGNED(16,
static const int16_t,
190 av1_default_iscan_8x8_transpose[64]) = {
191 0, 2, 3, 9, 10, 20, 21, 35, 1, 4, 8, 11, 19, 22, 34, 36,
192 5, 7, 12, 18, 23, 33, 37, 48, 6, 13, 17, 24, 32, 38, 47, 49,
193 14, 16, 25, 31, 39, 46, 50, 57, 15, 26, 30, 40, 45, 51, 56, 58,
194 27, 29, 41, 44, 52, 55, 59, 62, 28, 42, 43, 53, 54, 60, 61, 63
200DECLARE_ALIGNED(16,
static const int16_t,
201 default_scan_lp_16x16_transpose[256]) = {
202 0, 8, 2, 4, 10, 16, 24, 18, 12, 6, 64, 14, 20, 26, 32,
203 40, 34, 28, 22, 72, 66, 68, 74, 80, 30, 36, 42, 48, 56, 50,
204 44, 38, 88, 82, 76, 70, 128, 78, 84, 90, 96, 46, 52, 58, 1,
205 9, 3, 60, 54, 104, 98, 92, 86, 136, 130, 132, 138, 144, 94, 100,
206 106, 112, 62, 5, 11, 17, 25, 19, 13, 7, 120, 114, 108, 102, 152,
207 146, 140, 134, 192, 142, 148, 154, 160, 110, 116, 122, 65, 15, 21, 27,
208 33, 41, 35, 29, 23, 73, 67, 124, 118, 168, 162, 156, 150, 200, 194,
209 196, 202, 208, 158, 164, 170, 176, 126, 69, 75, 81, 31, 37, 43, 49,
210 57, 51, 45, 39, 89, 83, 77, 71, 184, 178, 172, 166, 216, 210, 204,
211 198, 206, 212, 218, 224, 174, 180, 186, 129, 79, 85, 91, 97, 47, 53,
212 59, 61, 55, 105, 99, 93, 87, 137, 131, 188, 182, 232, 226, 220, 214,
213 222, 228, 234, 240, 190, 133, 139, 145, 95, 101, 107, 113, 63, 121, 115,
214 109, 103, 153, 147, 141, 135, 248, 242, 236, 230, 238, 244, 250, 193, 143,
215 149, 155, 161, 111, 117, 123, 125, 119, 169, 163, 157, 151, 201, 195, 252,
216 246, 254, 197, 203, 209, 159, 165, 171, 177, 127, 185, 179, 173, 167, 217,
217 211, 205, 199, 207, 213, 219, 225, 175, 181, 187, 189, 183, 233, 227, 221,
218 215, 223, 229, 235, 241, 191, 249, 243, 237, 231, 239, 245, 251, 253, 247,
222#if CONFIG_AV1_HIGHBITDEPTH
227DECLARE_ALIGNED(16,
static const int16_t,
228 default_scan_fp_16x16_transpose[256]) = {
229 0, 4, 2, 8, 6, 16, 20, 18, 12, 10, 64, 14, 24, 22, 32,
230 36, 34, 28, 26, 68, 66, 72, 70, 80, 30, 40, 38, 48, 52, 50,
231 44, 42, 84, 82, 76, 74, 128, 78, 88, 86, 96, 46, 56, 54, 1,
232 5, 3, 60, 58, 100, 98, 92, 90, 132, 130, 136, 134, 144, 94, 104,
233 102, 112, 62, 9, 7, 17, 21, 19, 13, 11, 116, 114, 108, 106, 148,
234 146, 140, 138, 192, 142, 152, 150, 160, 110, 120, 118, 65, 15, 25, 23,
235 33, 37, 35, 29, 27, 69, 67, 124, 122, 164, 162, 156, 154, 196, 194,
236 200, 198, 208, 158, 168, 166, 176, 126, 73, 71, 81, 31, 41, 39, 49,
237 53, 51, 45, 43, 85, 83, 77, 75, 180, 178, 172, 170, 212, 210, 204,
238 202, 206, 216, 214, 224, 174, 184, 182, 129, 79, 89, 87, 97, 47, 57,
239 55, 61, 59, 101, 99, 93, 91, 133, 131, 188, 186, 228, 226, 220, 218,
240 222, 232, 230, 240, 190, 137, 135, 145, 95, 105, 103, 113, 63, 117, 115,
241 109, 107, 149, 147, 141, 139, 244, 242, 236, 234, 238, 248, 246, 193, 143,
242 153, 151, 161, 111, 121, 119, 125, 123, 165, 163, 157, 155, 197, 195, 252,
243 250, 254, 201, 199, 209, 159, 169, 167, 177, 127, 181, 179, 173, 171, 213,
244 211, 205, 203, 207, 217, 215, 225, 175, 185, 183, 189, 187, 229, 227, 221,
245 219, 223, 233, 231, 241, 191, 245, 243, 237, 235, 239, 249, 247, 253, 251,
256DECLARE_ALIGNED(16,
static const int16_t,
257 av1_default_iscan_lp_16x16_transpose[256]) = {
258 0, 44, 2, 46, 3, 63, 9, 69, 1, 45, 4, 64, 8, 68, 11,
259 87, 5, 65, 7, 67, 12, 88, 18, 94, 6, 66, 13, 89, 17, 93,
260 24, 116, 14, 90, 16, 92, 25, 117, 31, 123, 15, 91, 26, 118, 30,
261 122, 41, 148, 27, 119, 29, 121, 42, 149, 48, 152, 28, 120, 43, 150,
262 47, 151, 62, 177, 10, 86, 20, 96, 21, 113, 35, 127, 19, 95, 22,
263 114, 34, 126, 37, 144, 23, 115, 33, 125, 38, 145, 52, 156, 32, 124,
264 39, 146, 51, 155, 58, 173, 40, 147, 50, 154, 59, 174, 73, 181, 49,
265 153, 60, 175, 72, 180, 83, 198, 61, 176, 71, 179, 84, 199, 98, 202,
266 70, 178, 85, 200, 97, 201, 112, 219, 36, 143, 54, 158, 55, 170, 77,
267 185, 53, 157, 56, 171, 76, 184, 79, 194, 57, 172, 75, 183, 80, 195,
268 102, 206, 74, 182, 81, 196, 101, 205, 108, 215, 82, 197, 100, 204, 109,
269 216, 131, 223, 99, 203, 110, 217, 130, 222, 140, 232, 111, 218, 129, 221,
270 141, 233, 160, 236, 128, 220, 142, 234, 159, 235, 169, 245, 78, 193, 104,
271 208, 105, 212, 135, 227, 103, 207, 106, 213, 134, 226, 136, 228, 107, 214,
272 133, 225, 137, 229, 164, 240, 132, 224, 138, 230, 163, 239, 165, 241, 139,
273 231, 162, 238, 166, 242, 189, 249, 161, 237, 167, 243, 188, 248, 190, 250,
274 168, 244, 187, 247, 191, 251, 210, 254, 186, 246, 192, 252, 209, 253, 211,
278#if CONFIG_AV1_HIGHBITDEPTH
285DECLARE_ALIGNED(16,
static const int16_t,
286 av1_default_iscan_fp_16x16_transpose[256]) = {
287 0, 44, 2, 46, 1, 45, 4, 64, 3, 63, 9, 69, 8, 68, 11,
288 87, 5, 65, 7, 67, 6, 66, 13, 89, 12, 88, 18, 94, 17, 93,
289 24, 116, 14, 90, 16, 92, 15, 91, 26, 118, 25, 117, 31, 123, 30,
290 122, 41, 148, 27, 119, 29, 121, 28, 120, 43, 150, 42, 149, 48, 152,
291 47, 151, 62, 177, 10, 86, 20, 96, 19, 95, 22, 114, 21, 113, 35,
292 127, 34, 126, 37, 144, 23, 115, 33, 125, 32, 124, 39, 146, 38, 145,
293 52, 156, 51, 155, 58, 173, 40, 147, 50, 154, 49, 153, 60, 175, 59,
294 174, 73, 181, 72, 180, 83, 198, 61, 176, 71, 179, 70, 178, 85, 200,
295 84, 199, 98, 202, 97, 201, 112, 219, 36, 143, 54, 158, 53, 157, 56,
296 171, 55, 170, 77, 185, 76, 184, 79, 194, 57, 172, 75, 183, 74, 182,
297 81, 196, 80, 195, 102, 206, 101, 205, 108, 215, 82, 197, 100, 204, 99,
298 203, 110, 217, 109, 216, 131, 223, 130, 222, 140, 232, 111, 218, 129, 221,
299 128, 220, 142, 234, 141, 233, 160, 236, 159, 235, 169, 245, 78, 193, 104,
300 208, 103, 207, 106, 213, 105, 212, 135, 227, 134, 226, 136, 228, 107, 214,
301 133, 225, 132, 224, 138, 230, 137, 229, 164, 240, 163, 239, 165, 241, 139,
302 231, 162, 238, 161, 237, 167, 243, 166, 242, 189, 249, 188, 248, 190, 250,
303 168, 244, 187, 247, 186, 246, 192, 252, 191, 251, 210, 254, 209, 253, 211,
315DECLARE_ALIGNED(16,
static const int16_t,
316 av1_fast_idtx_scan_4x4[16]) = { 0, 1, 4, 8, 5, 2, 3, 6,
317 9, 12, 13, 10, 7, 11, 14, 15 };
320DECLARE_ALIGNED(16,
static const int16_t,
321 av1_fast_idtx_iscan_4x4[16]) = { 0, 1, 5, 6, 2, 4, 7, 12,
322 3, 8, 11, 13, 9, 10, 14, 15 };
324static const SCAN_ORDER av1_fast_idtx_scan_order_4x4 = {
325 av1_fast_idtx_scan_4x4, av1_fast_idtx_iscan_4x4
329DECLARE_ALIGNED(16,
static const int16_t, av1_fast_idtx_scan_8x8[64]) = {
330 0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32, 25, 18, 11, 4, 5,
331 12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13, 6, 7, 14, 21, 28,
332 35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51,
333 58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63
337DECLARE_ALIGNED(16,
static const int16_t, av1_fast_idtx_iscan_8x8[64]) = {
338 0, 1, 5, 6, 14, 15, 27, 28, 2, 4, 7, 13, 16, 26, 29, 42,
339 3, 8, 12, 17, 25, 30, 41, 43, 9, 11, 18, 24, 31, 40, 44, 53,
340 10, 19, 23, 32, 39, 45, 52, 54, 20, 22, 33, 38, 46, 51, 55, 60,
341 21, 34, 37, 47, 50, 56, 59, 61, 35, 36, 48, 49, 57, 58, 62, 63
344static const SCAN_ORDER av1_fast_idtx_scan_order_8x8 = {
345 av1_fast_idtx_scan_8x8, av1_fast_idtx_iscan_8x8
349DECLARE_ALIGNED(16,
static const int16_t, av1_fast_idtx_scan_16x16[256]) = {
350 0, 1, 16, 32, 17, 2, 3, 18, 33, 48, 64, 49, 34, 19, 4,
351 5, 20, 35, 50, 65, 80, 96, 81, 66, 51, 36, 21, 6, 7, 22,
352 37, 52, 67, 82, 97, 112, 128, 113, 98, 83, 68, 53, 38, 23, 8,
353 9, 24, 39, 54, 69, 84, 99, 114, 129, 144, 160, 145, 130, 115, 100,
354 85, 70, 55, 40, 25, 10, 11, 26, 41, 56, 71, 86, 101, 116, 131,
355 146, 161, 176, 192, 177, 162, 147, 132, 117, 102, 87, 72, 57, 42, 27,
356 12, 13, 28, 43, 58, 73, 88, 103, 118, 133, 148, 163, 178, 193, 208,
357 224, 209, 194, 179, 164, 149, 134, 119, 104, 89, 74, 59, 44, 29, 14,
358 15, 30, 45, 60, 75, 90, 105, 120, 135, 150, 165, 180, 195, 210, 225,
359 240, 241, 226, 211, 196, 181, 166, 151, 136, 121, 106, 91, 76, 61, 46,
360 31, 47, 62, 77, 92, 107, 122, 137, 152, 167, 182, 197, 212, 227, 242,
361 243, 228, 213, 198, 183, 168, 153, 138, 123, 108, 93, 78, 63, 79, 94,
362 109, 124, 139, 154, 169, 184, 199, 214, 229, 244, 245, 230, 215, 200, 185,
363 170, 155, 140, 125, 110, 95, 111, 126, 141, 156, 171, 186, 201, 216, 231,
364 246, 247, 232, 217, 202, 187, 172, 157, 142, 127, 143, 158, 173, 188, 203,
365 218, 233, 248, 249, 234, 219, 204, 189, 174, 159, 175, 190, 205, 220, 235,
366 250, 251, 236, 221, 206, 191, 207, 222, 237, 252, 253, 238, 223, 239, 254,
371DECLARE_ALIGNED(16,
static const int16_t, av1_fast_idtx_iscan_16x16[256]) = {
372 0, 1, 5, 6, 14, 15, 27, 28, 44, 45, 65, 66, 90, 91, 119,
373 120, 2, 4, 7, 13, 16, 26, 29, 43, 46, 64, 67, 89, 92, 118,
374 121, 150, 3, 8, 12, 17, 25, 30, 42, 47, 63, 68, 88, 93, 117,
375 122, 149, 151, 9, 11, 18, 24, 31, 41, 48, 62, 69, 87, 94, 116,
376 123, 148, 152, 177, 10, 19, 23, 32, 40, 49, 61, 70, 86, 95, 115,
377 124, 147, 153, 176, 178, 20, 22, 33, 39, 50, 60, 71, 85, 96, 114,
378 125, 146, 154, 175, 179, 200, 21, 34, 38, 51, 59, 72, 84, 97, 113,
379 126, 145, 155, 174, 180, 199, 201, 35, 37, 52, 58, 73, 83, 98, 112,
380 127, 144, 156, 173, 181, 198, 202, 219, 36, 53, 57, 74, 82, 99, 111,
381 128, 143, 157, 172, 182, 197, 203, 218, 220, 54, 56, 75, 81, 100, 110,
382 129, 142, 158, 171, 183, 196, 204, 217, 221, 234, 55, 76, 80, 101, 109,
383 130, 141, 159, 170, 184, 195, 205, 216, 222, 233, 235, 77, 79, 102, 108,
384 131, 140, 160, 169, 185, 194, 206, 215, 223, 232, 236, 245, 78, 103, 107,
385 132, 139, 161, 168, 186, 193, 207, 214, 224, 231, 237, 244, 246, 104, 106,
386 133, 138, 162, 167, 187, 192, 208, 213, 225, 230, 238, 243, 247, 252, 105,
387 134, 137, 163, 166, 188, 191, 209, 212, 226, 229, 239, 242, 248, 251, 253,
388 135, 136, 164, 165, 189, 190, 210, 211, 227, 228, 240, 241, 249, 250, 254,
393static INLINE
int get_model_rd_flag(
const AV1_COMP *cpi,
const MACROBLOCKD *xd,
395 const AV1_COMMON *
const cm = &cpi->
common;
396 const int large_block = bsize >= BLOCK_32X32;
399 !cyclic_refresh_segment_id_boosted(xd->
mi[0]->
segment_id) &&
426 MV_REFERENCE_FRAME ref_frame,
427 int_mv frame_mv[MB_MODE_COUNT][REF_FRAMES],
428 struct buf_2d yv12_mb[8][MAX_MB_PLANE],
430 int force_skip_low_temp_var,
432 AV1_COMMON *
const cm = &cpi->
common;
433 MACROBLOCKD *
const xd = &x->
e_mbd;
436 const YV12_BUFFER_CONFIG *yv12 = get_ref_frame_yv12_buf(cm, ref_frame);
437 const int num_planes = av1_num_planes(cm);
442 frame_mv[NEWMV][ref_frame].as_int = INVALID_MV;
444 assert(yv12 != NULL);
446 const struct scale_factors *
const sf =
447 get_ref_scale_factors_const(cm, ref_frame);
448 av1_setup_pred_block(xd, yv12_mb[ref_frame], yv12, sf, sf, num_planes);
449 av1_find_mv_refs(cm, xd, mbmi, ref_frame, mbmi_ext->
ref_mv_count,
454 av1_copy_usable_ref_mv_stack_and_weight(xd, mbmi_ext, ref_frame);
455 av1_find_best_ref_mvs_from_stack(
457 &frame_mv[NEARESTMV][ref_frame], &frame_mv[NEARMV][ref_frame], 0);
458 frame_mv[GLOBALMV][ref_frame] = mbmi_ext->
global_mvs[ref_frame];
460 if (!av1_is_scaled(sf) && bsize >= BLOCK_8X8 && !skip_pred_mv &&
461 !(force_skip_low_temp_var && ref_frame != LAST_FRAME)) {
462 av1_mv_pred(cpi, x, yv12_mb[ref_frame][0].buf, yv12->y_stride, ref_frame,
467 av1_count_overlappable_neighbors(cm, xd);
473 PREDICTION_MODE pred_mode,
474 MV_REFERENCE_FRAME ref_frame0,
475 MV_REFERENCE_FRAME ref_frame1,
476 const AV1_COMMON *cm) {
479 mbmi->
mode = pred_mode;
483 pmi->palette_size[PLANE_TYPE_Y] = 0;
484 pmi->palette_size[PLANE_TYPE_UV] = 0;
486 mbmi->
mv[0].as_int = mbmi->
mv[1].as_int = 0;
493static INLINE
void init_estimate_block_intra_args(
497 args->mode = DC_PRED;
500 args->best_sad = UINT_MAX;
501 args->prune_mode_based_on_sad =
false;
504static INLINE
int get_pred_buffer(PRED_BUFFER *p,
int len) {
505 for (
int buf_idx = 0; buf_idx < len; buf_idx++) {
506 if (!p[buf_idx].in_use) {
507 p[buf_idx].in_use = 1;
514static INLINE
void free_pred_buffer(PRED_BUFFER *p) {
515 if (p != NULL) p->in_use = 0;
518#if CONFIG_INTERNAL_STATS
519static INLINE
void store_coding_context_nonrd(
MACROBLOCK *x,
520 PICK_MODE_CONTEXT *ctx,
523static INLINE
void store_coding_context_nonrd(
MACROBLOCK *x,
524 PICK_MODE_CONTEXT *ctx) {
526 MACROBLOCKD *
const xd = &x->
e_mbd;
534#if CONFIG_INTERNAL_STATS
535 ctx->best_mode_index = mode_index;
537 ctx->mic = *xd->
mi[0];
539 av1_copy_mbmi_ext_to_mbmi_ext_frame(&ctx->mbmi_ext_best, &x->
mbmi_ext,
543void av1_block_yrd(
MACROBLOCK *x, RD_STATS *this_rdc,
int *skippable,
544 BLOCK_SIZE bsize, TX_SIZE tx_size);
546void av1_block_yrd_idtx(
MACROBLOCK *x,
const uint8_t *
const pred_buf,
547 int pred_stride, RD_STATS *this_rdc,
int *skippable,
548 BLOCK_SIZE bsize, TX_SIZE tx_size);
550int64_t av1_model_rd_for_sb_uv(
AV1_COMP *cpi, BLOCK_SIZE plane_bsize,
552 RD_STATS *this_rdc,
int start_plane,
555void av1_estimate_block_intra(
int plane,
int block,
int row,
int col,
556 BLOCK_SIZE plane_bsize, TX_SIZE tx_size,
560 int best_early_term,
unsigned int ref_cost_intra,
561 int reuse_prediction,
struct buf_2d *orig_dst,
562 PRED_BUFFER *tmp_buffers,
563 PRED_BUFFER **this_mode_pred, RD_STATS *best_rdc,
564 BEST_PICKMODE *best_pickmode,
565 PICK_MODE_CONTEXT *ctx);
struct macroblock MACROBLOCK
Encoder's parameters related to the current coding block.
struct AV1_COMP AV1_COMP
Top level encoder structure.
@ AOM_CBR
Definition aom_encoder.h:185
static void find_predictors(AV1_COMP *cpi, MACROBLOCK *x, MV_REFERENCE_FRAME ref_frame, int_mv frame_mv[MB_MODE_COUNT][REF_FRAMES], struct buf_2d yv12_mb[8][3], BLOCK_SIZE bsize, int force_skip_low_temp_var, int skip_pred_mv)
Finds predicted motion vectors for a block.
Definition nonrd_opt.h:425
FeatureFlags features
Definition av1_common_int.h:907
CommonQuantParams quant_params
Definition av1_common_int.h:924
RateControlCfg rc_cfg
Definition encoder.h:937
Top level encoder structure.
Definition encoder.h:2815
AV1EncoderConfig oxcf
Definition encoder.h:2863
AV1_COMMON common
Definition encoder.h:2858
int base_qindex
Definition av1_common_int.h:615
InterpFilter interp_filter
Definition av1_common_int.h:409
bool switchable_motion_mode
Definition av1_common_int.h:407
bool allow_high_precision_mv
Definition av1_common_int.h:369
Structure to store parameters and statistics used in non-rd inter mode evaluation.
Definition nonrd_opt.h:81
int_mv frame_mv[MB_MODE_COUNT][REF_FRAMES]
Array to hold motion vector for all modes and reference frames.
Definition nonrd_opt.h:97
unsigned int ref_costs_single[REF_FRAMES]
Array to hold ref cost of single reference mode for all ref frames.
Definition nonrd_opt.h:95
int64_t uv_dist[(4)][REF_FRAMES]
Distortion of chroma planes for all modes and reference frames.
Definition nonrd_opt.h:89
RD_STATS this_rdc
Structure to RD cost of current mode.
Definition nonrd_opt.h:85
unsigned int vars[(4)][REF_FRAMES]
Array to hold variance of all modes and reference frames.
Definition nonrd_opt.h:93
uint8_t mode_checked[MB_MODE_COUNT][REF_FRAMES]
Array to hold flags of evaluated modes for each reference frame.
Definition nonrd_opt.h:105
int_mv frame_mv_best[MB_MODE_COUNT][REF_FRAMES]
Array to hold best mv for all modes and reference frames.
Definition nonrd_opt.h:99
int single_inter_mode_costs[(4)][REF_FRAMES]
Array to hold inter mode cost of single ref mode for all ref frames.
Definition nonrd_opt.h:101
RD_STATS best_rdc
Pointer to the RD Cost for the best mode found so far.
Definition nonrd_opt.h:87
int use_ref_frame_mask[REF_FRAMES]
Array to hold use reference frame mask for each reference frame.
Definition nonrd_opt.h:103
BEST_PICKMODE best_pickmode
Structure to hold best inter mode data.
Definition nonrd_opt.h:83
struct buf_2d yv12_mb[REF_FRAMES][3]
Buffer to hold predicted block for all reference frames and planes.
Definition nonrd_opt.h:91
Extended mode info derived from mbmi.
Definition block.h:222
int_mv global_mvs[REF_FRAMES]
Global mvs.
Definition block.h:231
int16_t mode_context[MODE_CTX_REF_FRAMES]
Context used to encode the current mode.
Definition block.h:233
uint8_t ref_mv_count[MODE_CTX_REF_FRAMES]
Number of ref mvs in the drl.
Definition block.h:229
Stores the prediction/txfm mode of the current coding block.
Definition blockd.h:222
int_mv mv[2]
The motion vectors used by the current inter mode.
Definition blockd.h:244
PREDICTION_MODE mode
The prediction mode used.
Definition blockd.h:232
UV_PREDICTION_MODE uv_mode
The UV mode when intra is used.
Definition blockd.h:234
PALETTE_MODE_INFO palette_mode_info
Stores the size and colors of palette mode.
Definition blockd.h:280
uint8_t segment_id
The segment id.
Definition blockd.h:310
uint8_t ref_mv_idx
Which ref_mv to use.
Definition blockd.h:314
MV_REFERENCE_FRAME ref_frame[2]
The reference frames for the MV.
Definition blockd.h:246
FILTER_INTRA_MODE_INFO filter_intra_mode_info
The type of filter intra mode used (if applicable).
Definition blockd.h:274
MOTION_MODE motion_mode
The motion mode used by the inter prediction.
Definition blockd.h:250
uint8_t num_proj_ref
Number of samples used by warp causal.
Definition blockd.h:252
INTERINTRA_MODE interintra_mode
The type of intra mode used by inter-intra.
Definition blockd.h:259
enum aom_rc_mode mode
Definition encoder.h:604
Stores various encoding/search decisions related to txfm search.
Definition block.h:526
uint8_t skip_txfm
Whether to skip transform and quantization on a partition block level.
Definition block.h:528
MACROBLOCKD e_mbd
Decoder's view of current coding block.
Definition block.h:896
int pred_mv1_sad[REF_FRAMES]
The sad of the 2nd mv ref (near).
Definition block.h:1115
int pred_mv0_sad[REF_FRAMES]
The sad of the 1st mv ref (nearest).
Definition block.h:1113
TxfmSearchInfo txfm_search_info
Results of the txfm searches that have been done.
Definition block.h:1311
int pred_mv_sad[REF_FRAMES]
Sum absolute distortion of the predicted mv for each ref frame.
Definition block.h:1105
MB_MODE_INFO_EXT mbmi_ext
Derived coding information.
Definition block.h:903
uint16_t weight[MODE_CTX_REF_FRAMES][MAX_REF_MV_STACK_SIZE]
Definition blockd.h:781
CANDIDATE_MV ref_mv_stack[MODE_CTX_REF_FRAMES][MAX_REF_MV_STACK_SIZE]
Definition blockd.h:776
MB_MODE_INFO ** mi
Definition blockd.h:617