| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #ifndef AOM_AV1_ENCODER_MV_PREC_H_ |
| #define AOM_AV1_ENCODER_MV_PREC_H_ |
| |
| #include "av1/encoder/encoder.h" |
| #include "av1/encoder/speed_features.h" |
| |
| // Q threshold for high precision mv. |
| #define HIGH_PRECISION_MV_QTHRESH 128 |
| void av1_collect_mv_stats(AV1_COMP *cpi, int current_q); |
| |
| static AOM_INLINE int av1_frame_allows_smart_mv(const AV1_COMP *cpi) { |
| const int gf_group_index = cpi->gf_group.index; |
| const int gf_update_type = cpi->gf_group.update_type[gf_group_index]; |
| return !frame_is_intra_only(&cpi->common) && |
| !(gf_update_type == INTNL_OVERLAY_UPDATE || |
| gf_update_type == KFFLT_OVERLAY_UPDATE || |
| gf_update_type == OVERLAY_UPDATE); |
| } |
| |
| #if CONFIG_FLEX_MVRES |
| static AOM_INLINE void av1_set_high_precision_mv(AV1_COMP *cpi, |
| MvSubpelPrecision precision) { |
| FeatureFlags *features = &cpi->common.features; |
| features->fr_mv_precision = precision; |
| features->use_pb_mv_precision = 0; |
| } |
| #else |
| static AOM_INLINE void av1_set_high_precision_mv( |
| AV1_COMP *cpi, int allow_high_precision_mv, |
| int cur_frame_force_integer_mv) { |
| MvCosts *const mv_costs = &cpi->td.mb.mv_costs; |
| const int copy_hp = cpi->common.features.allow_high_precision_mv = |
| allow_high_precision_mv && !cur_frame_force_integer_mv; |
| |
| mv_costs->nmv_cost[0] = &mv_costs->nmv_cost_alloc[0][MV_MAX]; |
| mv_costs->nmv_cost[1] = &mv_costs->nmv_cost_alloc[1][MV_MAX]; |
| mv_costs->nmv_cost_hp[0] = &mv_costs->nmv_cost_hp_alloc[0][MV_MAX]; |
| mv_costs->nmv_cost_hp[1] = &mv_costs->nmv_cost_hp_alloc[1][MV_MAX]; |
| mv_costs->mv_cost_stack = |
| copy_hp ? mv_costs->nmv_cost_hp : mv_costs->nmv_cost; |
| #if CONFIG_ADAPTIVE_MVD |
| mv_costs->amvd_nmv_cost[0] = &mv_costs->amvd_nmv_cost_alloc[0][MV_MAX]; |
| mv_costs->amvd_nmv_cost[1] = &mv_costs->amvd_nmv_cost_alloc[1][MV_MAX]; |
| mv_costs->amvd_nmv_cost_hp[0] = &mv_costs->amvd_nmv_cost_hp_alloc[0][MV_MAX]; |
| mv_costs->amvd_nmv_cost_hp[1] = &mv_costs->amvd_nmv_cost_hp_alloc[1][MV_MAX]; |
| mv_costs->amvd_mv_cost_stack = |
| copy_hp ? mv_costs->amvd_nmv_cost_hp : mv_costs->amvd_nmv_cost; |
| #endif // CONFIG_ADAPTIVE_MVD |
| } |
| #endif |
| |
| void av1_pick_and_set_high_precision_mv(AV1_COMP *cpi, int qindex); |
| |
| #endif // AOM_AV1_ENCODER_MV_PREC_H_ |