Home | History | Annotate | Download | only in common 
      1 /*
      2  * Copyright (c) 2016, Alliance for Open Media. All rights reserved
      3  *
      4  * This source code is subject to the terms of the BSD 2 Clause License and
      5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
      6  * was not distributed with this source code in the LICENSE file, you can
      7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
      8  * Media Patent License 1.0 was not distributed with this source code in the
      9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
     10  */
     11 
     12 #ifndef AOM_AV1_COMMON_RECONINTRA_H_
     13 #define AOM_AV1_COMMON_RECONINTRA_H_
     14 
     15 #include <stdlib.h>
     16 
     17 #include "aom/aom_integer.h"
     18 #include "av1/common/blockd.h"
     19 #include "av1/common/onyxc_int.h"
     20 
     21 #ifdef __cplusplus
     22 extern "C" {
     23 #endif
     24 
     25 void av1_init_intra_predictors(void);
     26 void av1_predict_intra_block_facade(const AV1_COMMON *cm, MACROBLOCKD *xd,
     27                                     int plane, int blk_col, int blk_row,
     28                                     TX_SIZE tx_size);
     29 void av1_predict_intra_block(const AV1_COMMON *cm, const MACROBLOCKD *xd,
     30                              int bw, int bh, TX_SIZE tx_size,
     31                              PREDICTION_MODE mode, int angle_delta,
     32                              int use_palette,
     33                              FILTER_INTRA_MODE filter_intra_mode,
     34                              const uint8_t *ref, int ref_stride, uint8_t *dst,
     35                              int dst_stride, int aoff, int loff, int plane);
     36 
     37 // Mapping of interintra to intra mode for use in the intra component
     38 static const PREDICTION_MODE interintra_to_intra_mode[INTERINTRA_MODES] = {
     39   DC_PRED, V_PRED, H_PRED, SMOOTH_PRED
     40 };
     41 
     42 // Mapping of intra mode to the interintra mode
     43 static const INTERINTRA_MODE intra_to_interintra_mode[INTRA_MODES] = {
     44   II_DC_PRED, II_V_PRED, II_H_PRED, II_V_PRED,      II_SMOOTH_PRED, II_V_PRED,
     45   II_H_PRED,  II_H_PRED, II_V_PRED, II_SMOOTH_PRED, II_SMOOTH_PRED
     46 };
     47 
     48 #define FILTER_INTRA_SCALE_BITS 4
     49 
     50 static INLINE int av1_is_directional_mode(PREDICTION_MODE mode) {
     51   return mode >= V_PRED && mode <= D67_PRED;
     52 }
     53 
     54 static INLINE int av1_use_angle_delta(BLOCK_SIZE bsize) {
     55   return bsize >= BLOCK_8X8;
     56 }
     57 
     58 static INLINE int av1_allow_intrabc(const AV1_COMMON *const cm) {
     59   return frame_is_intra_only(cm) && cm->allow_screen_content_tools &&
     60          cm->allow_intrabc;
     61 }
     62 
     63 static INLINE int av1_filter_intra_allowed_bsize(const AV1_COMMON *const cm,
     64                                                  BLOCK_SIZE bs) {
     65   if (!cm->seq_params.enable_filter_intra || bs == BLOCK_INVALID) return 0;
     66 
     67   return block_size_wide[bs] <= 32 && block_size_high[bs] <= 32;
     68 }
     69 
     70 static INLINE int av1_filter_intra_allowed(const AV1_COMMON *const cm,
     71                                            const MB_MODE_INFO *mbmi) {
     72   return mbmi->mode == DC_PRED &&
     73          mbmi->palette_mode_info.palette_size[0] == 0 &&
     74          av1_filter_intra_allowed_bsize(cm, mbmi->sb_type);
     75 }
     76 
     77 extern const int8_t av1_filter_intra_taps[FILTER_INTRA_MODES][8][8];
     78 
     79 // Get the shift (up-scaled by 256) in X w.r.t a unit change in Y.
     80 // If angle > 0 && angle < 90, dx = -((int)(256 / t));
     81 // If angle > 90 && angle < 180, dx = (int)(256 / t);
     82 // If angle > 180 && angle < 270, dx = 1;
     83 static INLINE int av1_get_dx(int angle) {
     84   if (angle > 0 && angle < 90) {
     85     return dr_intra_derivative[angle];
     86   } else if (angle > 90 && angle < 180) {
     87     return dr_intra_derivative[180 - angle];
     88   } else {
     89     // In this case, we are not really going to use dx. We may return any value.
     90     return 1;
     91   }
     92 }
     93 
     94 // Get the shift (up-scaled by 256) in Y w.r.t a unit change in X.
     95 // If angle > 0 && angle < 90, dy = 1;
     96 // If angle > 90 && angle < 180, dy = (int)(256 * t);
     97 // If angle > 180 && angle < 270, dy = -((int)(256 * t));
     98 static INLINE int av1_get_dy(int angle) {
     99   if (angle > 90 && angle < 180) {
    100     return dr_intra_derivative[angle - 90];
    101   } else if (angle > 180 && angle < 270) {
    102     return dr_intra_derivative[270 - angle];
    103   } else {
    104     // In this case, we are not really going to use dy. We may return any value.
    105     return 1;
    106   }
    107 }
    108 
    109 static INLINE int av1_use_intra_edge_upsample(int bs0, int bs1, int delta,
    110                                               int type) {
    111   const int d = abs(delta);
    112   const int blk_wh = bs0 + bs1;
    113   if (d == 0 || d >= 40) return 0;
    114   return type ? (blk_wh <= 8) : (blk_wh <= 16);
    115 }
    116 #ifdef __cplusplus
    117 }  // extern "C"
    118 #endif
    119 #endif  // AOM_AV1_COMMON_RECONINTRA_H_
    120