xref: /aosp_15_r20/external/libaom/av1/encoder/mv_prec.h (revision 77c1e3ccc04c968bd2bc212e87364f250e820521)
1 /*
2  * Copyright (c) 2019, 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_MV_PREC_H_
13 #define AOM_AV1_ENCODER_MV_PREC_H_
14 
15 #include "av1/encoder/encoder.h"
16 #include "av1/encoder/speed_features.h"
17 
18 // Q threshold for high precision mv.
19 #define HIGH_PRECISION_MV_QTHRESH 128
20 #if !CONFIG_REALTIME_ONLY
21 void av1_collect_mv_stats(AV1_COMP *cpi, int current_q);
22 
av1_frame_allows_smart_mv(const AV1_COMP * cpi)23 static inline int av1_frame_allows_smart_mv(const AV1_COMP *cpi) {
24   const int gf_group_index = cpi->gf_frame_index;
25   const int gf_update_type = cpi->ppi->gf_group.update_type[gf_group_index];
26   return !frame_is_intra_only(&cpi->common) &&
27          !(gf_update_type == INTNL_OVERLAY_UPDATE ||
28            gf_update_type == OVERLAY_UPDATE);
29 }
30 #endif  // !CONFIG_REALTIME_ONLY
31 
av1_set_high_precision_mv(AV1_COMP * cpi,int allow_high_precision_mv,int cur_frame_force_integer_mv)32 static inline void av1_set_high_precision_mv(AV1_COMP *cpi,
33                                              int allow_high_precision_mv,
34                                              int cur_frame_force_integer_mv) {
35   MvCosts *const mv_costs = cpi->td.mb.mv_costs;
36   // Avoid accessing 'mv_costs' when it is not allocated.
37   if (mv_costs == NULL) return;
38 
39   const int copy_hp = cpi->common.features.allow_high_precision_mv =
40       allow_high_precision_mv && !cur_frame_force_integer_mv;
41 
42   mv_costs->nmv_cost[0] = &mv_costs->nmv_cost_alloc[0][MV_MAX];
43   mv_costs->nmv_cost[1] = &mv_costs->nmv_cost_alloc[1][MV_MAX];
44   mv_costs->nmv_cost_hp[0] = &mv_costs->nmv_cost_hp_alloc[0][MV_MAX];
45   mv_costs->nmv_cost_hp[1] = &mv_costs->nmv_cost_hp_alloc[1][MV_MAX];
46   mv_costs->mv_cost_stack =
47       copy_hp ? mv_costs->nmv_cost_hp : mv_costs->nmv_cost;
48 }
49 
50 void av1_pick_and_set_high_precision_mv(AV1_COMP *cpi, int qindex);
51 
52 #endif  // AOM_AV1_ENCODER_MV_PREC_H_
53