blob: cea1e14f75cae32dd738515146d30a5591009653 [file] [log] [blame] [edit]
/*
* 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_