chiyotsai | c666b1f | 2019-12-20 10:44:58 -0800 | [diff] [blame] | 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 |
chiyotsai | 51bd648 | 2019-12-20 10:49:34 -0800 | [diff] [blame] | 20 | #if !CONFIG_REALTIME_ONLY |
| 21 | void av1_collect_mv_stats(AV1_COMP *cpi, int current_q); |
| 22 | |
| 23 | static AOM_INLINE int av1_frame_allows_smart_mv(const AV1_COMP *cpi) { |
Mufaddal Chakera | ab20d37 | 2021-03-17 12:18:34 +0530 | [diff] [blame] | 24 | const int gf_group_index = cpi->gf_frame_index; |
Mufaddal Chakera | 8ee04fa | 2021-03-17 13:33:18 +0530 | [diff] [blame] | 25 | const int gf_update_type = cpi->ppi->gf_group.update_type[gf_group_index]; |
chiyotsai | 51bd648 | 2019-12-20 10:49:34 -0800 | [diff] [blame] | 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 |
chiyotsai | c666b1f | 2019-12-20 10:44:58 -0800 | [diff] [blame] | 31 | |
| 32 | static AOM_INLINE void av1_set_high_precision_mv( |
| 33 | AV1_COMP *cpi, int allow_high_precision_mv, |
| 34 | int cur_frame_force_integer_mv) { |
Fyodor Kyslov | 648c650 | 2021-02-02 18:41:10 -0800 | [diff] [blame] | 35 | MvCosts *const mv_costs = cpi->td.mb.mv_costs; |
Jayasanker J | c144305 | 2021-07-21 13:18:08 +0530 | [diff] [blame] | 36 | // Avoid accessing 'mv_costs' when it is not allocated. |
| 37 | if (mv_costs == NULL) return; |
| 38 | |
Urvang Joshi | b6409e9 | 2020-03-23 11:23:27 -0700 | [diff] [blame] | 39 | const int copy_hp = cpi->common.features.allow_high_precision_mv = |
chiyotsai | 8b7cef8 | 2020-01-21 16:52:54 -0800 | [diff] [blame] | 40 | allow_high_precision_mv && !cur_frame_force_integer_mv; |
chiyotsai | c95e364 | 2020-04-10 13:17:06 -0700 | [diff] [blame] | 41 | |
chiyotsai | 9a06d18 | 2020-05-01 17:12:12 -0700 | [diff] [blame] | 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; |
chiyotsai | c666b1f | 2019-12-20 10:44:58 -0800 | [diff] [blame] | 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_ |