| /* |
| * 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_PICKRST_H_ |
| #define AOM_AV1_ENCODER_PICKRST_H_ |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #include "av1/encoder/encoder.h" |
| |
| struct yv12_buffer_config; |
| struct AV1_COMP; |
| |
| // Enable extra debugging for loop restoration costing? |
| // |
| // If this is set to 1, then we record not just the selected LR parameters, but |
| // also the values which the search process thinks they should be delta-coded |
| // against. Then, when writing out the bitstream, we verify this information, |
| // to help ensure that the search code is costing things properly |
| #define DEBUG_LR_COSTING 0 |
| |
| #if DEBUG_LR_COSTING |
| #define MAX_LR_UNITS_W 64 |
| #define MAX_LR_UNITS_H 64 |
| |
| // Storage for reference parameters. |
| // |
| // The storage size is determined by: |
| // * This is always written and then checked within the same frame encode pass, |
| // so we do not need to buffer multiple frames of data |
| // * The parameters can be different per plane within one frame |
| // * The relevant set of ref parameters can differ between the search where |
| // we set the frame restoration mode to RESTORE_WIENER, and the search where |
| // we set it to RESTORE_SWITCHABLE. |
| // So we need to store at least two sets of Wiener params and two sets of |
| // SGR params, and the easiest way to do this is to index by |
| // frame_restoration_type |
| extern RestorationUnitInfo lr_ref_params[RESTORE_TYPES][MAX_MB_PLANE] |
| [MAX_LR_UNITS_W * MAX_LR_UNITS_H]; |
| #endif // DEBUG_LR_COSTING |
| |
| static const uint8_t g_shuffle_stats_data[16] = { |
| 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, |
| }; |
| |
| static const uint8_t g_shuffle_stats_highbd_data[32] = { |
| 0, 1, 2, 3, 2, 3, 4, 5, 4, 5, 6, 7, 6, 7, 8, 9, |
| 0, 1, 2, 3, 2, 3, 4, 5, 4, 5, 6, 7, 6, 7, 8, 9, |
| }; |
| |
| static inline uint8_t find_average(const uint8_t *src, int h_start, int h_end, |
| int v_start, int v_end, int stride) { |
| uint64_t sum = 0; |
| for (int i = v_start; i < v_end; i++) { |
| for (int j = h_start; j < h_end; j++) { |
| sum += src[i * stride + j]; |
| } |
| } |
| uint64_t avg = sum / ((v_end - v_start) * (h_end - h_start)); |
| return (uint8_t)avg; |
| } |
| |
| #if CONFIG_AV1_HIGHBITDEPTH |
| static inline uint16_t find_average_highbd(const uint16_t *src, int h_start, |
| int h_end, int v_start, int v_end, |
| int stride) { |
| uint64_t sum = 0; |
| for (int i = v_start; i < v_end; i++) { |
| for (int j = h_start; j < h_end; j++) { |
| sum += src[i * stride + j]; |
| } |
| } |
| uint64_t avg = sum / ((v_end - v_start) * (h_end - h_start)); |
| return (uint16_t)avg; |
| } |
| #endif |
| |
| /*!\brief Algorithm for AV1 loop restoration search and estimation. |
| * |
| * \ingroup in_loop_restoration |
| * This function determines proper restoration filter types and |
| * associated parameters for each restoration unit in a frame. |
| * |
| * \param[in] sd Source frame buffer |
| * \param[in,out] cpi Top-level encoder structure |
| * |
| * \remark Nothing is returned. Instead, chosen restoration filter |
| * types and parameters are stored per plane in the \c rst_info structure |
| * of type \ref RestorationInfo inside \c cpi->common: |
| * \arg \c rst_info[ \c 0 ]: Chosen parameters for Y plane |
| * \arg \c rst_info[ \c 1 ]: Chosen parameters for U plane if it exists |
| * \arg \c rst_info[ \c 2 ]: Chosen parameters for V plane if it exists |
| * \par |
| * The following fields in each \c rst_info[ \c p], \c p = 0, 1, 2 |
| * are populated: |
| * \arg \c rst_info[ \c p ].\c frame_restoration_type |
| * \arg \c rst_info[ \c p ].\c unit_info[ \c u ], |
| * for each \c u in 0, 1, ..., \c n( \c p ) - 1, |
| * where \c n( \c p ) is the number of restoration units in plane \c p. |
| * \par |
| * The following fields in each \c rst_info[ \c p ].\c unit_info[ \c u ], |
| * \c p = 0, 1, 2 and \c u = 0, 1, ..., \c n( \c p ) - 1, of type |
| * \ref RestorationUnitInfo are populated: |
| * \arg \c rst_info[ \c p ].\c unit_info[ \c u ].\c restoration_type |
| * \arg \c rst_info[ \c p ].\c unit_info[ \c u ].\c wiener_info OR |
| * \c rst_info[ \c p ].\c unit_info[ \c u ].\c sgrproj_info OR |
| * neither, depending on |
| * \c rst_info[ \c p ].\c unit_info[ \c u ].\c restoration_type |
| * |
| */ |
| void av1_pick_filter_restoration(const YV12_BUFFER_CONFIG *sd, AV1_COMP *cpi); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AOM_AV1_ENCODER_PICKRST_H_ |