| /* |
| * 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 AV1_COMMON_AV1_CONVOLVE_H_ |
| #define AV1_COMMON_AV1_CONVOLVE_H_ |
| #include "av1/common/filter.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| typedef enum CONVOLVE_OPT { |
| // indicate the results in dst buf is rounded by FILTER_BITS or not |
| CONVOLVE_OPT_ROUND, |
| CONVOLVE_OPT_NO_ROUND, |
| } CONVOLVE_OPT; |
| |
| typedef int32_t CONV_BUF_TYPE; |
| |
| typedef struct ConvolveParams { |
| int ref; |
| int do_average; |
| CONVOLVE_OPT round; |
| CONV_BUF_TYPE *dst; |
| int dst_stride; |
| int round_0; |
| int round_1; |
| int plane; |
| int do_post_rounding; |
| #if CONFIG_JNT_COMP |
| int use_jnt_comp_avg; |
| int fwd_offset; |
| int bck_offset; |
| #endif |
| } ConvolveParams; |
| |
| static INLINE ConvolveParams get_conv_params(int ref, int do_average, |
| int plane) { |
| ConvolveParams conv_params; |
| conv_params.ref = ref; |
| conv_params.do_average = do_average; |
| conv_params.round = CONVOLVE_OPT_ROUND; |
| conv_params.plane = plane; |
| conv_params.do_post_rounding = 0; |
| return conv_params; |
| } |
| |
| #if CONFIG_DUAL_FILTER && USE_EXTRA_FILTER |
| static INLINE void av1_convolve_filter_params_fixup_1212( |
| const InterpFilterParams *params_x, InterpFilterParams *params_y) { |
| if (params_x->interp_filter == MULTITAP_SHARP && |
| params_y->interp_filter == MULTITAP_SHARP) { |
| // Avoid two directions both using 12-tap filter. |
| // This will reduce hardware implementation cost. |
| *params_y = av1_get_interp_filter_params(EIGHTTAP_SHARP); |
| } |
| } |
| #endif |
| |
| static INLINE void av1_get_convolve_filter_params(InterpFilters interp_filters, |
| int avoid_1212, |
| InterpFilterParams *params_x, |
| InterpFilterParams *params_y |
| #if CONFIG_SHORT_FILTER |
| , |
| int w, int h |
| #endif |
| ) { |
| #if CONFIG_DUAL_FILTER |
| InterpFilter filter_x = av1_extract_interp_filter(interp_filters, 1); |
| InterpFilter filter_y = av1_extract_interp_filter(interp_filters, 0); |
| #else |
| InterpFilter filter_x = av1_extract_interp_filter(interp_filters, 0); |
| InterpFilter filter_y = av1_extract_interp_filter(interp_filters, 0); |
| #endif |
| #if CONFIG_SHORT_FILTER |
| *params_x = av1_get_interp_filter_params_with_block_size(filter_x, w); |
| *params_y = av1_get_interp_filter_params_with_block_size(filter_y, h); |
| #else |
| *params_x = av1_get_interp_filter_params(filter_x); |
| *params_y = av1_get_interp_filter_params(filter_y); |
| #endif |
| |
| if (avoid_1212) { |
| #if CONFIG_DUAL_FILTER && USE_EXTRA_FILTER |
| convolve_filter_params_fixup_1212(params_x, params_y); |
| #endif |
| } |
| } |
| |
| struct AV1Common; |
| void av1_convolve_init(struct AV1Common *cm); |
| |
| void av1_convolve_2d_facade(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, |
| InterpFilters interp_filters, const int subpel_x_q4, |
| int x_step_q4, const int subpel_y_q4, int y_step_q4, |
| int scaled, ConvolveParams *conv_params); |
| |
| static INLINE ConvolveParams get_conv_params_no_round(int ref, int do_average, |
| int plane, int32_t *dst, |
| int dst_stride) { |
| ConvolveParams conv_params; |
| conv_params.ref = ref; |
| conv_params.do_average = do_average; |
| conv_params.round = CONVOLVE_OPT_NO_ROUND; |
| conv_params.round_0 = 5; |
| conv_params.round_1 = 0; |
| conv_params.dst = dst; |
| conv_params.dst_stride = dst_stride; |
| conv_params.plane = plane; |
| conv_params.do_post_rounding = 0; |
| return conv_params; |
| } |
| |
| #if CONFIG_HIGHBITDEPTH |
| void av1_highbd_convolve_2d_facade(const uint8_t *src8, int src_stride, |
| uint8_t *dst, int dst_stride, int w, int h, |
| InterpFilters interp_filters, |
| const int subpel_x_q4, int x_step_q4, |
| const int subpel_y_q4, int y_step_q4, |
| int scaled, ConvolveParams *conv_params, |
| int bd); |
| #endif |
| |
| void av1_convolve(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, InterpFilters interp_filters, |
| const int subpel_x, int xstep, const int subpel_y, int ystep, |
| ConvolveParams *conv_params); |
| |
| void av1_convolve_c(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, InterpFilters interp_filters, |
| const int subpel_x, int xstep, const int subpel_y, |
| int ystep, ConvolveParams *conv_params); |
| |
| void av1_convolve_scale(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, |
| InterpFilters interp_filters, const int subpel_x, |
| int xstep, const int subpel_y, int ystep, |
| ConvolveParams *conv_params); |
| |
| #if CONFIG_HIGHBITDEPTH |
| void av1_highbd_convolve(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, |
| InterpFilters interp_filters, const int subpel_x, |
| int xstep, const int subpel_y, int ystep, int avg, |
| int bd); |
| |
| void av1_highbd_convolve_scale(const uint8_t *src, int src_stride, uint8_t *dst, |
| int dst_stride, int w, int h, |
| InterpFilters interp_filters, const int subpel_x, |
| int xstep, const int subpel_y, int ystep, |
| int avg, int bd); |
| #endif // CONFIG_HIGHBITDEPTH |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AV1_COMMON_AV1_CONVOLVE_H_ |