/* * 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. */ #include <math.h> #include <stdbool.h> #include <string.h> #include "config/aom_dsp_rtcd.h" #include "config/aom_scale_rtcd.h" #include "aom/aom_integer.h" #include "av1/common/av1_common_int.h" #include "av1/common/reconinter.h" #include "av1/encoder/encoder.h" #include "av1/encoder/ethread.h" #include "av1/encoder/pickcdef.h" #include "av1/encoder/mcomp.h" // Get primary and secondary filter strength for the given strength index and // search method static inline void get_cdef_filter_strengths(CDEF_PICK_METHOD pick_method, int *pri_strength, int *sec_strength, int strength_idx) { … } // Store CDEF filter strength calculated from strength index for given search // method #define STORE_CDEF_FILTER_STRENGTH(cdef_strength, pick_method, strength_idx) … /* Search for the best strength to add as an option, knowing we already selected nb_strengths options. */ static uint64_t search_one(int *lev, int nb_strengths, uint64_t mse[][TOTAL_STRENGTHS], int sb_count, CDEF_PICK_METHOD pick_method) { … } /* Search for the best luma+chroma strength to add as an option, knowing we already selected nb_strengths options. */ static uint64_t search_one_dual(int *lev0, int *lev1, int nb_strengths, uint64_t (**mse)[TOTAL_STRENGTHS], int sb_count, CDEF_PICK_METHOD pick_method) { … } /* Search for the set of strengths that minimizes mse. */ static uint64_t joint_strength_search(int *best_lev, int nb_strengths, uint64_t mse[][TOTAL_STRENGTHS], int sb_count, CDEF_PICK_METHOD pick_method) { … } /* Search for the set of luma+chroma strengths that minimizes mse. */ static uint64_t joint_strength_search_dual(int *best_lev0, int *best_lev1, int nb_strengths, uint64_t (**mse)[TOTAL_STRENGTHS], int sb_count, CDEF_PICK_METHOD pick_method) { … } static inline void init_src_params(int *src_stride, int *width, int *height, int *width_log2, int *height_log2, BLOCK_SIZE bsize) { … } #if CONFIG_AV1_HIGHBITDEPTH /* Compute MSE only on the blocks we filtered. */ static uint64_t compute_cdef_dist_highbd(void *dst, int dstride, uint16_t *src, cdef_list *dlist, int cdef_count, BLOCK_SIZE bsize, int coeff_shift, int row, int col) { assert(bsize == BLOCK_4X4 || bsize == BLOCK_4X8 || bsize == BLOCK_8X4 || bsize == BLOCK_8X8); uint64_t sum = 0; int bi, bx, by; uint16_t *dst16 = CONVERT_TO_SHORTPTR((uint8_t *)dst); uint16_t *dst_buff = &dst16[row * dstride + col]; int src_stride, width, height, width_log2, height_log2; init_src_params(&src_stride, &width, &height, &width_log2, &height_log2, bsize); for (bi = 0; bi < cdef_count; bi++) { by = dlist[bi].by; bx = dlist[bi].bx; sum += aom_mse_wxh_16bit_highbd( &dst_buff[(by << height_log2) * dstride + (bx << width_log2)], dstride, &src[bi << (height_log2 + width_log2)], src_stride, width, height); } return sum >> 2 * coeff_shift; } #endif // Checks dual and quad block processing is applicable for block widths 8 and 4 // respectively. static inline int is_dual_or_quad_applicable(cdef_list *dlist, int width, int cdef_count, int bi, int iter) { … } static uint64_t compute_cdef_dist(void *dst, int dstride, uint16_t *src, cdef_list *dlist, int cdef_count, BLOCK_SIZE bsize, int coeff_shift, int row, int col) { … } // Fill the boundary regions of the block with CDEF_VERY_LARGE, only if the // region is outside frame boundary static inline void fill_borders_for_fbs_on_frame_boundary( uint16_t *inbuf, int hfilt_size, int vfilt_size, bool is_fb_on_frm_left_boundary, bool is_fb_on_frm_right_boundary, bool is_fb_on_frm_top_boundary, bool is_fb_on_frm_bottom_boundary) { … } // Calculate the number of 8x8/4x4 filter units for which SSE can be calculated // after CDEF filtering in single function call static AOM_FORCE_INLINE int get_error_calc_width_in_filt_units( cdef_list *dlist, int cdef_count, int bi, int subsampling_x, int subsampling_y) { … } // Returns the block error after CDEF filtering for a given strength static inline uint64_t get_filt_error( const CdefSearchCtx *cdef_search_ctx, const struct macroblockd_plane *pd, cdef_list *dlist, int dir[CDEF_NBLOCKS][CDEF_NBLOCKS], int *dirinit, int var[CDEF_NBLOCKS][CDEF_NBLOCKS], uint16_t *in, uint8_t *ref_buffer, int ref_stride, int row, int col, int pri_strength, int sec_strength, int cdef_count, int pli, int coeff_shift, BLOCK_SIZE bs) { … } // Calculates MSE at block level. // Inputs: // cdef_search_ctx: Pointer to the structure containing parameters related to // CDEF search context. // fbr: Row index in units of 64x64 block // fbc: Column index in units of 64x64 block // Returns: // Nothing will be returned. Contents of cdef_search_ctx will be modified. void av1_cdef_mse_calc_block(CdefSearchCtx *cdef_search_ctx, struct aom_internal_error_info *error_info, int fbr, int fbc, int sb_count) { … } // MSE calculation at frame level. // Inputs: // cdef_search_ctx: Pointer to the structure containing parameters related to // CDEF search context. // Returns: // Nothing will be returned. Contents of cdef_search_ctx will be modified. static void cdef_mse_calc_frame(CdefSearchCtx *cdef_search_ctx, struct aom_internal_error_info *error_info) { … } // Allocates memory for members of CdefSearchCtx. // Inputs: // cdef_search_ctx: Pointer to the structure containing parameters // related to CDEF search context. // Returns: // Nothing will be returned. Contents of cdef_search_ctx will be modified. static void cdef_alloc_data(AV1_COMMON *cm, CdefSearchCtx *cdef_search_ctx) { … } // Deallocates the memory allocated for members of CdefSearchCtx. // Inputs: // cdef_search_ctx: Pointer to the structure containing parameters // related to CDEF search context. // Returns: // Nothing will be returned. void av1_cdef_dealloc_data(CdefSearchCtx *cdef_search_ctx) { … } // Initialize the parameters related to CDEF search context. // Inputs: // frame: Pointer to compressed frame buffer // ref: Pointer to the frame buffer holding the source frame // cm: Pointer to top level common structure // xd: Pointer to common current coding block structure // cdef_search_ctx: Pointer to the structure containing parameters related to // CDEF search context. // pick_method: Search method used to select CDEF parameters // Returns: // Nothing will be returned. Contents of cdef_search_ctx will be modified. static inline void cdef_params_init(const YV12_BUFFER_CONFIG *frame, const YV12_BUFFER_CONFIG *ref, AV1_COMMON *cm, MACROBLOCKD *xd, CdefSearchCtx *cdef_search_ctx, CDEF_PICK_METHOD pick_method) { … } void av1_pick_cdef_from_qp(AV1_COMMON *const cm, int skip_cdef, int is_screen_content) { … } void av1_cdef_search(AV1_COMP *cpi) { … }
Codebase Browser
chromium