AOMedia AV1 Codec
nonrd_opt.h
1 /*
2  * Copyright (c) 2022, Alliance for Open Media. All rights reserved
3  *
4  * This source code is subject to the terms of the BSD 2 Clause License and
5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6  * was not distributed with this source code in the LICENSE file, you can
7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8  * Media Patent License 1.0 was not distributed with this source code in the
9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10  */
11 
12 #ifndef AOM_AV1_ENCODER_NONRD_OPT_H_
13 #define AOM_AV1_ENCODER_NONRD_OPT_H_
14 
15 #include "av1/encoder/rdopt_utils.h"
16 
17 #define RTC_INTER_MODES (4)
18 #define RTC_INTRA_MODES (4)
19 #define RTC_MODES (AOMMAX(RTC_INTER_MODES, RTC_INTRA_MODES))
20 
21 static const PREDICTION_MODE intra_mode_list[] = { DC_PRED, V_PRED, H_PRED,
22  SMOOTH_PRED };
23 
24 static const PREDICTION_MODE inter_mode_list[] = { NEARESTMV, NEARMV, GLOBALMV,
25  NEWMV };
26 
27 static const THR_MODES mode_idx[REF_FRAMES][RTC_MODES] = {
28  { THR_DC, THR_V_PRED, THR_H_PRED, THR_SMOOTH },
29  { THR_NEARESTMV, THR_NEARMV, THR_GLOBALMV, THR_NEWMV },
30  { THR_NEARESTL2, THR_NEARL2, THR_GLOBALL2, THR_NEWL2 },
31  { THR_NEARESTL3, THR_NEARL3, THR_GLOBALL3, THR_NEWL3 },
32  { THR_NEARESTG, THR_NEARG, THR_GLOBALG, THR_NEWG },
33  { THR_NEARESTB, THR_NEARB, THR_GLOBALB, THR_NEWB },
34  { THR_NEARESTA2, THR_NEARA2, THR_GLOBALA2, THR_NEWA2 },
35  { THR_NEARESTA, THR_NEARA, THR_GLOBALA, THR_NEWA },
36 };
37 
38 // Indicates the blocks for which RD model should be based on special logic
39 static INLINE int get_model_rd_flag(const AV1_COMP *cpi, const MACROBLOCKD *xd,
40  BLOCK_SIZE bsize) {
41  const int large_block = bsize >= BLOCK_32X32;
42  const AV1_COMMON *const cm = &cpi->common;
43  return cpi->oxcf.rc_cfg.mode == AOM_CBR && large_block &&
44  !cyclic_refresh_segment_id_boosted(xd->mi[0]->segment_id) &&
46  cm->seq_params->bit_depth == AOM_BITS_8;
47 }
74 static INLINE void find_predictors(
75  AV1_COMP *cpi, MACROBLOCK *x, MV_REFERENCE_FRAME ref_frame,
76  int_mv frame_mv[MB_MODE_COUNT][REF_FRAMES], TileDataEnc *tile_data,
77  struct buf_2d yv12_mb[8][MAX_MB_PLANE], BLOCK_SIZE bsize,
78  int force_skip_low_temp_var, int skip_pred_mv) {
79  AV1_COMMON *const cm = &cpi->common;
80  MACROBLOCKD *const xd = &x->e_mbd;
81  MB_MODE_INFO *const mbmi = xd->mi[0];
82  MB_MODE_INFO_EXT *const mbmi_ext = &x->mbmi_ext;
83  const YV12_BUFFER_CONFIG *yv12 = get_ref_frame_yv12_buf(cm, ref_frame);
84  const int num_planes = av1_num_planes(cm);
85  (void)tile_data;
86 
87  x->pred_mv_sad[ref_frame] = INT_MAX;
88  x->pred_mv0_sad[ref_frame] = INT_MAX;
89  x->pred_mv1_sad[ref_frame] = INT_MAX;
90  frame_mv[NEWMV][ref_frame].as_int = INVALID_MV;
91  // TODO(kyslov) this needs various further optimizations. to be continued..
92  assert(yv12 != NULL);
93  if (yv12 != NULL) {
94  const struct scale_factors *const sf =
95  get_ref_scale_factors_const(cm, ref_frame);
96  av1_setup_pred_block(xd, yv12_mb[ref_frame], yv12, sf, sf, num_planes);
97  av1_find_mv_refs(cm, xd, mbmi, ref_frame, mbmi_ext->ref_mv_count,
98  xd->ref_mv_stack, xd->weight, NULL, mbmi_ext->global_mvs,
99  mbmi_ext->mode_context);
100  // TODO(Ravi): Populate mbmi_ext->ref_mv_stack[ref_frame][4] and
101  // mbmi_ext->weight[ref_frame][4] inside av1_find_mv_refs.
102  av1_copy_usable_ref_mv_stack_and_weight(xd, mbmi_ext, ref_frame);
103  av1_find_best_ref_mvs_from_stack(
104  cm->features.allow_high_precision_mv, mbmi_ext, ref_frame,
105  &frame_mv[NEARESTMV][ref_frame], &frame_mv[NEARMV][ref_frame], 0);
106  frame_mv[GLOBALMV][ref_frame] = mbmi_ext->global_mvs[ref_frame];
107  // Early exit for non-LAST frame if force_skip_low_temp_var is set.
108  if (!av1_is_scaled(sf) && bsize >= BLOCK_8X8 && !skip_pred_mv &&
109  !(force_skip_low_temp_var && ref_frame != LAST_FRAME)) {
110  av1_mv_pred(cpi, x, yv12_mb[ref_frame][0].buf, yv12->y_stride, ref_frame,
111  bsize);
112  }
113  }
115  av1_count_overlappable_neighbors(cm, xd);
116  }
117  mbmi->num_proj_ref = 1;
118 }
119 
120 #endif // AOM_AV1_ENCODER_NONRD_OPT_H_
Extended mode info derived from mbmi.
Definition: block.h:193
AV1_COMMON common
Definition: encoder.h:2802
RateControlCfg rc_cfg
Definition: encoder.h:937
int pred_mv1_sad[REF_FRAMES]
The sad of the 2nd mv ref (near).
Definition: block.h:1086
FeatureFlags features
Definition: av1_common_int.h:905
Definition: aom_encoder.h:185
int pred_mv0_sad[REF_FRAMES]
The sad of the 1st mv ref (nearest).
Definition: block.h:1084
AV1EncoderConfig oxcf
Definition: encoder.h:2807
uint8_t segment_id
The segment id.
Definition: blockd.h:310
CommonQuantParams quant_params
Definition: av1_common_int.h:922
enum aom_rc_mode mode
Definition: encoder.h:604
bool switchable_motion_mode
Definition: av1_common_int.h:405
Definition: aom_codec.h:319
uint8_t ref_mv_count[MODE_CTX_REF_FRAMES]
Number of ref mvs in the drl.
Definition: block.h:200
CANDIDATE_MV ref_mv_stack[MODE_CTX_REF_FRAMES][MAX_REF_MV_STACK_SIZE]
Definition: blockd.h:780
int pred_mv_sad[REF_FRAMES]
Sum absolute distortion of the predicted mv for each ref frame.
Definition: block.h:1076
SequenceHeader * seq_params
Definition: av1_common_int.h:976
bool allow_high_precision_mv
Definition: av1_common_int.h:367
YV12 frame buffer data structure.
Definition: yv12config.h:39
int_mv global_mvs[REF_FRAMES]
Global mvs.
Definition: block.h:202
MB_MODE_INFO ** mi
Definition: blockd.h:621
Variables related to current coding block.
Definition: blockd.h:574
Top level encoder structure.
Definition: encoder.h:2759
static void find_predictors(AV1_COMP *cpi, MACROBLOCK *x, MV_REFERENCE_FRAME ref_frame, int_mv frame_mv[MB_MODE_COUNT][REF_FRAMES], TileDataEnc *tile_data, 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:74
Top level common structure used by both encoder and decoder.
Definition: av1_common_int.h:750
uint8_t num_proj_ref
Number of samples used by warp causal.
Definition: blockd.h:252
Stores the prediction/txfm mode of the current coding block.
Definition: blockd.h:222
int16_t mode_context[MODE_CTX_REF_FRAMES]
Context used to encode the current mode.
Definition: block.h:204
int base_qindex
Definition: av1_common_int.h:613
uint16_t weight[MODE_CTX_REF_FRAMES][MAX_REF_MV_STACK_SIZE]
Definition: blockd.h:785
Encoder's parameters related to the current coding block.
Definition: block.h:849
MB_MODE_INFO_EXT mbmi_ext
Derived coding information.
Definition: block.h:874
MACROBLOCKD e_mbd
Decoder's view of current coding block.
Definition: block.h:867