| /* |
| * 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_COMMON_RECONINTRA_H_ |
| #define AOM_AV1_COMMON_RECONINTRA_H_ |
| |
| #include <stdlib.h> |
| |
| #include "aom/aom_integer.h" |
| #include "av1/common/blockd.h" |
| #include "av1/common/onyxc_int.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| // Calculate the number of rows / columns available in the reference |
| // frame. |
| int av1_intra_top_available(const MACROBLOCKD *xd, int plane); |
| int av1_intra_left_available(const MACROBLOCKD *xd, int plane); |
| |
| // Calculate the number of rows / columns unavailable in the reference |
| // frame. av1_intra_right_unavailable indicates the number of columns |
| // unavailable on the top of the block (starting from the right), |
| // av1_intra_bottom_unavailable indicates the number of rows unavailable |
| // to the left of the block (starting from bottom). |
| int av1_intra_right_unavailable(const MACROBLOCKD *xd, int plane, |
| TX_SIZE tx_size); |
| int av1_intra_bottom_unavailable(const MACROBLOCKD *xd, int plane, |
| TX_SIZE tx_size); |
| |
| // Equivalent to memmove, but looks at the bit-depth and converts the |
| // pointer to dst16 (and the amount of data moved) if in high bitdepth mode. |
| void av1_bd_memmove(uint8_t *dst, const uint8_t *ref, size_t n, bool is_hbd); |
| |
| // Equivalent to memset, but looks at the bit-depth and copies the value |
| // every uint16_t space if in high bitdepth mode. |
| void av1_bd_memset(uint8_t *dst, int c, size_t n, bool is_hbd); |
| |
| // Extends the intra-predictor to have a border region consisting of the |
| // reference frame; border region is to the top-left and assumed to be |
| // offset negatively from the passed in pointer (which points to the start |
| // of the regular intra-prediction). Note that it is possible for the |
| // bit-depth to be 8 and is_hbd to be true, if high-bitdepth pipeline |
| // is forced on. |
| void av1_extend_intra_border(const uint8_t *ref, int ref_stride, uint8_t *dst, |
| int dst_stride, int top_rows_available, |
| int right_cols_unavailable, |
| int left_cols_available, |
| int bottom_rows_unavailable, int width, int height, |
| int border, aom_bit_depth_t bd, bool is_hbd); |
| void av1_init_intra_predictors(void); |
| void av1_predict_intra_block_facade(const AV1_COMMON *cm, MACROBLOCKD *xd, |
| int plane, int blk_col, int blk_row, |
| TX_SIZE tx_size); |
| void av1_predict_intra_block(const AV1_COMMON *cm, const MACROBLOCKD *xd, |
| int wpx, int hpx, TX_SIZE tx_size, |
| PREDICTION_MODE mode, int angle_delta, |
| int use_palette, |
| FILTER_INTRA_MODE filter_intra_mode, |
| #if CONFIG_ADAPT_FILTER_INTRA |
| ADAPT_FILTER_INTRA_MODE adapt_filter_intra_mode, |
| #endif |
| #if CONFIG_DERIVED_INTRA_MODE |
| int use_derived_intra_mode, |
| #endif // CONFIG_DERIVED_INTRA_MODE |
| const uint8_t *ref, int ref_stride, uint8_t *dst, |
| int dst_stride, int col_off, int row_off, |
| int plane); |
| |
| // Mapping of interintra to intra mode for use in the intra component |
| static const PREDICTION_MODE interintra_to_intra_mode[INTERINTRA_MODES] = { |
| DC_PRED, V_PRED, H_PRED, SMOOTH_PRED, |
| #if CONFIG_ILLUM_MCOMP |
| DC_PRED |
| #endif // CONFIG_ILLUM_MCOMP |
| }; |
| |
| // Mapping of intra mode to the interintra mode |
| static const INTERINTRA_MODE intra_to_interintra_mode[INTRA_MODES] = { |
| II_DC_PRED, II_V_PRED, II_H_PRED, II_V_PRED, II_SMOOTH_PRED, II_V_PRED, |
| II_H_PRED, II_H_PRED, II_V_PRED, II_SMOOTH_PRED, II_SMOOTH_PRED |
| }; |
| |
| #define FILTER_INTRA_SCALE_BITS 4 |
| |
| static INLINE int av1_use_angle_delta(BLOCK_SIZE bsize) { |
| return bsize >= BLOCK_8X8; |
| } |
| |
| static INLINE int av1_allow_intrabc(const AV1_COMMON *const cm) { |
| return frame_is_intra_only(cm) && cm->allow_screen_content_tools && |
| cm->allow_intrabc; |
| } |
| |
| static INLINE int av1_filter_intra_allowed_bsize(const AV1_COMMON *const cm, |
| BLOCK_SIZE bs) { |
| if (!cm->seq_params.enable_filter_intra || bs == BLOCK_INVALID) return 0; |
| |
| return block_size_wide[bs] <= 32 && block_size_high[bs] <= 32; |
| } |
| |
| static INLINE int av1_filter_intra_allowed(const AV1_COMMON *const cm, |
| const MB_MODE_INFO *mbmi) { |
| return mbmi->mode == DC_PRED && |
| mbmi->palette_mode_info.palette_size[0] == 0 && |
| av1_filter_intra_allowed_bsize(cm, mbmi->sb_type); |
| } |
| |
| #if CONFIG_ADAPT_FILTER_INTRA |
| static INLINE int av1_adapt_filter_intra_allowed_bsize( |
| const AV1_COMMON *const cm, BLOCK_SIZE bs) { |
| if (!cm->seq_params.enable_adapt_filter_intra || bs == BLOCK_INVALID) |
| return 0; |
| |
| return block_size_wide[bs] <= 128 && block_size_high[bs] <= 128; |
| } |
| |
| static INLINE int av1_adapt_filter_intra_allowed(const AV1_COMMON *const cm, |
| const MB_MODE_INFO *mbmi) { |
| return mbmi->mode == DC_PRED && |
| mbmi->palette_mode_info.palette_size[0] == 0 && |
| mbmi->filter_intra_mode_info.use_filter_intra == 0 && |
| av1_adapt_filter_intra_allowed_bsize(cm, mbmi->sb_type); |
| } |
| #endif // CONFIG_ADAPT_FILTER_INTRA |
| |
| extern const int8_t av1_filter_intra_taps[FILTER_INTRA_MODES][8][8]; |
| |
| static const int16_t dr_intra_derivative[90] = { |
| // More evenly spread out angles and limited to 10-bit |
| // Values that are 0 will never be used |
| // Approx angle |
| 0, 0, 0, // |
| 1023, 0, 0, // 3, ... |
| 547, 0, 0, // 6, ... |
| 372, 0, 0, 0, 0, // 9, ... |
| 273, 0, 0, // 14, ... |
| 215, 0, 0, // 17, ... |
| 178, 0, 0, // 20, ... |
| 151, 0, 0, // 23, ... (113 & 203 are base angles) |
| 132, 0, 0, // 26, ... |
| 116, 0, 0, // 29, ... |
| 102, 0, 0, 0, // 32, ... |
| 90, 0, 0, // 36, ... |
| 80, 0, 0, // 39, ... |
| 71, 0, 0, // 42, ... |
| 64, 0, 0, // 45, ... (45 & 135 are base angles) |
| 57, 0, 0, // 48, ... |
| 51, 0, 0, // 51, ... |
| 45, 0, 0, 0, // 54, ... |
| 40, 0, 0, // 58, ... |
| 35, 0, 0, // 61, ... |
| 31, 0, 0, // 64, ... |
| 27, 0, 0, // 67, ... (67 & 157 are base angles) |
| 23, 0, 0, // 70, ... |
| 19, 0, 0, // 73, ... |
| 15, 0, 0, 0, 0, // 76, ... |
| 11, 0, 0, // 81, ... |
| 7, 0, 0, // 84, ... |
| 3, 0, 0, // 87, ... |
| }; |
| |
| // Get the shift (up-scaled by 256) in X w.r.t a unit change in Y. |
| // If angle > 0 && angle < 90, dx = -((int)(256 / t)); |
| // If angle > 90 && angle < 180, dx = (int)(256 / t); |
| // If angle > 180 && angle < 270, dx = 1; |
| static INLINE int av1_get_dx(int angle) { |
| #if CONFIG_DERIVED_INTRA_MODE |
| if ((angle > 0 && angle < 90) || (angle > 90 && angle < 180)) { |
| if (angle > 90) angle = 180 - angle; |
| int dx = dr_intra_derivative[angle]; |
| if (!dx) dx = (int)round(64 / tan(angle * PI / 180)); |
| return dx; |
| } else { |
| return 1; |
| } |
| #endif // CONFIG_DERIVED_INTRA_MODE |
| |
| if (angle > 0 && angle < 90) { |
| return dr_intra_derivative[angle]; |
| } else if (angle > 90 && angle < 180) { |
| return dr_intra_derivative[180 - angle]; |
| } else { |
| // In this case, we are not really going to use dx. We may return any value. |
| return 1; |
| } |
| } |
| |
| // Get the shift (up-scaled by 256) in Y w.r.t a unit change in X. |
| // If angle > 0 && angle < 90, dy = 1; |
| // If angle > 90 && angle < 180, dy = (int)(256 * t); |
| // If angle > 180 && angle < 270, dy = -((int)(256 * t)); |
| static INLINE int av1_get_dy(int angle) { |
| #if CONFIG_DERIVED_INTRA_MODE |
| if ((angle > 90 && angle < 180) || (angle > 180 && angle < 270)) { |
| if (angle > 90 && angle < 180) { |
| angle = angle - 90; |
| } else { |
| angle = 270 - angle; |
| } |
| int dy = dr_intra_derivative[angle]; |
| if (!dy) dy = (int)round(64 / tan(angle * PI / 180)); |
| return dy; |
| } else { |
| return 1; |
| } |
| #endif // CONFIG_DERIVED_INTRA_MODE |
| |
| if (angle > 90 && angle < 180) { |
| return dr_intra_derivative[angle - 90]; |
| } else if (angle > 180 && angle < 270) { |
| return dr_intra_derivative[270 - angle]; |
| } else { |
| // In this case, we are not really going to use dy. We may return any value. |
| return 1; |
| } |
| } |
| |
| static INLINE int av1_use_intra_edge_upsample(int bs0, int bs1, int delta, |
| int type) { |
| const int d = abs(delta); |
| const int blk_wh = bs0 + bs1; |
| if (d == 0 || d >= 40) return 0; |
| return type ? (blk_wh <= 8) : (blk_wh <= 16); |
| } |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| #endif // AOM_AV1_COMMON_RECONINTRA_H_ |