| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved. |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. 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 www.aomedia.org/license/patent. |
| */ |
| |
| #ifndef AOM_AV1_ENCODER_VAR_BASED_PART_H_ |
| #define AOM_AV1_ENCODER_VAR_BASED_PART_H_ |
| |
| #include <stdio.h> |
| |
| #include "config/aom_config.h" |
| #include "config/aom_dsp_rtcd.h" |
| #include "config/av1_rtcd.h" |
| |
| #include "av1/encoder/encoder.h" |
| |
| // Calculate block index x and y from split level and index |
| #define GET_BLK_IDX_X(idx, level) (((idx) & (0x01)) << (level)) |
| #define GET_BLK_IDX_Y(idx, level) (((idx) >> (0x01)) << (level)) |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #define QINDEX_LARGE_BLOCK_THR \ |
| 100 // Use increased thresholds for midres for speed 9 when qindex is above |
| // this threshold |
| |
| #define CALC_CHROMA_THRESH_FOR_ZEROMV_SKIP(thresh_exit_part) \ |
| ((3 * (thresh_exit_part)) >> 2) |
| /*!\brief Set the thresholds for variance based partition. |
| * |
| * Set the variance split thresholds for following the block sizes: |
| * 0 - threshold_128x128, 1 - threshold_64x64, 2 - threshold_32x32, |
| * 3 - vbp_threshold_16x16. 4 - vbp_threshold_8x8 (to split to 4x4 partition) is |
| * currently only used on key frame. The thresholds are based om Q, resolution, |
| * noise level, and content state. |
| * |
| * \ingroup variance_partition |
| * \callgraph |
| * \callergraph |
| * |
| * \param[in] cpi Top level encoder structure |
| * \param[in] q q index |
| * \param[in] content_lowsumdiff Low sumdiff flag for superblock |
| * |
| * \remark Returns the set of thresholds in \c cpi->vbp_info.thresholds. |
| */ |
| void av1_set_variance_partition_thresholds(AV1_COMP *cpi, int q, |
| int content_lowsumdiff); |
| |
| /*!\brief Variance based partition selection. |
| * |
| * Select the partitioning based on the variance of the residual signal, |
| * residual generated as the difference between the source and prediction. |
| * The prediction is the reconstructed LAST or reconstructed GOLDEN, whichever |
| * has lower y sad. For LAST, option exists (speed feature) to use motion |
| * compensation based on superblock motion via int_pro_motion_estimation. For |
| * key frames reference is fixed 128 level, so variance is the source variance. |
| * The variance is computed for downsampled inputs (8x8 or 4x4 downsampled), |
| * and selection is done top-down via as set of partition thresholds. defined |
| * for each block level, and set based on Q, resolution, noise level, and |
| * content state. |
| * |
| * \ingroup variance_partition |
| * \callgraph |
| * \callergraph |
| * |
| * \param[in] cpi Top level encoder structure |
| * \param[in] tile Pointer to TileInfo |
| * \param[in] td Pointer to ThreadData |
| * \param[in] x Pointer to MACROBLOCK |
| * \param[in] mi_row Row coordinate of the superblock in a step |
| size of MI_SIZE |
| * \param[in] mi_col Column coordinate of the super block in a step |
| size of MI_SIZE |
| * |
| * \return Returns the partition in \c xd->mi[0]->sb_type. Also sets the low |
| * temporal variance flag and the color sensitivity flag (both used in |
| * nonrd_pickmode). |
| */ |
| int av1_choose_var_based_partitioning(AV1_COMP *cpi, const TileInfo *const tile, |
| ThreadData *td, MACROBLOCK *x, int mi_row, |
| int mi_col); |
| |
| // Read out the block's temporal variance for 64x64 SB case. |
| int av1_get_force_skip_low_temp_var_small_sb(const uint8_t *variance_low, |
| int mi_row, int mi_col, |
| BLOCK_SIZE bsize); |
| // Read out the block's temporal variance for 128x128 SB case. |
| int av1_get_force_skip_low_temp_var(const uint8_t *variance_low, int mi_row, |
| int mi_col, BLOCK_SIZE bsize); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AOM_AV1_ENCODER_VAR_BASED_PART_H_ |