reconinter.h 26.3 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1
/*
Yaowu Xu's avatar
Yaowu Xu committed
2
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Jingning Han's avatar
Jingning Han committed
3
 *
Yaowu Xu's avatar
Yaowu Xu committed
4 5 6 7 8 9
 * 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.
Jingning Han's avatar
Jingning Han committed
10 11
 */

Yaowu Xu's avatar
Yaowu Xu committed
12 13
#ifndef AV1_COMMON_RECONINTER_H_
#define AV1_COMMON_RECONINTER_H_
Jingning Han's avatar
Jingning Han committed
14

15 16
#include "av1/common/filter.h"
#include "av1/common/onyxc_int.h"
17
#include "av1/common/convolve.h"
18 19 20
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
#include "av1/common/warped_motion.h"
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
Yaowu Xu's avatar
Yaowu Xu committed
21
#include "aom/aom_integer.h"
Jingning Han's avatar
Jingning Han committed
22

23
#if CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
24
#define WARP_WM_NEIGHBORS_WITH_OBMC 0
25 26
#endif  // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION

27
#if CONFIG_MOTION_VAR && CONFIG_GLOBAL_MOTION
28
#define WARP_GM_NEIGHBORS_WITH_OBMC 0
29 30
#endif  // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION

31 32 33
#define AOM_LEFT_TOP_MARGIN_SCALED \
  ((AOM_BORDER_IN_PIXELS - AOM_INTERP_EXTEND) << SCALE_SUBPEL_BITS)

Jingning Han's avatar
Jingning Han committed
34 35 36 37
#ifdef __cplusplus
extern "C" {
#endif

38 39 40
static INLINE int has_scale(int xs, int ys) {
  return xs != SCALE_SUBPEL_SHIFTS || ys != SCALE_SUBPEL_SHIFTS;
}
41

Jingning Han's avatar
Jingning Han committed
42
static INLINE void inter_predictor(const uint8_t *src, int src_stride,
43 44 45
                                   uint8_t *dst, int dst_stride, int subpel_x,
                                   int subpel_y, const struct scale_factors *sf,
                                   int w, int h, ConvolveParams *conv_params,
46 47
                                   InterpFilters interp_filters, int xs,
                                   int ys) {
48
  assert(conv_params->do_average == 0 || conv_params->do_average == 1);
Alex Converse's avatar
Alex Converse committed
49
  assert(sf);
50
  if (has_scale(xs, ys)) {
51 52
    // TODO(afergs, debargha): Use a different scale convolve function
    // that uses higher precision for subpel_x, subpel_y, xs, ys
53 54 55
    if (conv_params->round == CONVOLVE_OPT_NO_ROUND) {
#if CONFIG_CONVOLVE_ROUND
      av1_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
56 57
                             interp_filters, subpel_x, xs, subpel_y, ys, 1,
                             conv_params);
58 59 60 61 62 63
      conv_params->do_post_rounding = 1;
#else
      assert(0);
#endif  // CONFIG_CONVOLVE_ROUND
    } else {
      assert(conv_params->round == CONVOLVE_OPT_ROUND);
64
      av1_convolve_scale(src, src_stride, dst, dst_stride, w, h, interp_filters,
65 66
                         subpel_x, xs, subpel_y, ys, conv_params);
    }
67 68 69 70 71 72 73 74 75 76
  } else {
    subpel_x >>= SCALE_EXTRA_BITS;
    subpel_y >>= SCALE_EXTRA_BITS;
    xs >>= SCALE_EXTRA_BITS;
    ys >>= SCALE_EXTRA_BITS;
    assert(subpel_x < SUBPEL_SHIFTS);
    assert(subpel_y < SUBPEL_SHIFTS);
    assert(xs <= SUBPEL_SHIFTS);
    assert(ys <= SUBPEL_SHIFTS);
    if (conv_params->round == CONVOLVE_OPT_NO_ROUND) {
Angie Chiang's avatar
Angie Chiang committed
77
#if CONFIG_CONVOLVE_ROUND
78
      av1_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
79 80
                             interp_filters, subpel_x, xs, subpel_y, ys, 0,
                             conv_params);
81
      conv_params->do_post_rounding = 1;
82
#else
83
      assert(0);
84 85
#endif  // CONFIG_CONVOLVE_ROUND
    } else {
86
      assert(conv_params->round == CONVOLVE_OPT_ROUND);
87 88 89 90 91

      InterpFilterParams filter_params_x, filter_params_y;
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y);

92
      if (w <= 2 || h <= 2) {
93
        av1_convolve_c(src, src_stride, dst, dst_stride, w, h, interp_filters,
94
                       subpel_x, xs, subpel_y, ys, conv_params);
95 96 97 98 99 100
      } else if (filter_params_x.taps == SUBPEL_TAPS &&
                 filter_params_y.taps == SUBPEL_TAPS) {
        const int16_t *kernel_x =
            av1_get_interp_filter_subpel_kernel(filter_params_x, subpel_x);
        const int16_t *kernel_y =
            av1_get_interp_filter_subpel_kernel(filter_params_y, subpel_y);
101 102 103
        sf->predict[subpel_x != 0][subpel_y != 0][conv_params->do_average](
            src, src_stride, dst, dst_stride, kernel_x, xs, kernel_y, ys, w, h);
      } else {
104
        av1_convolve(src, src_stride, dst, dst_stride, w, h, interp_filters,
105 106
                     subpel_x, xs, subpel_y, ys, conv_params);
      }
107
    }
108
  }
Jingning Han's avatar
Jingning Han committed
109 110
}

111
#if CONFIG_HIGHBITDEPTH
112 113
static INLINE void highbd_inter_predictor(const uint8_t *src, int src_stride,
                                          uint8_t *dst, int dst_stride,
114
                                          int subpel_x, int subpel_y,
clang-format's avatar
clang-format committed
115
                                          const struct scale_factors *sf, int w,
116
                                          int h, ConvolveParams *conv_params,
117 118
                                          InterpFilters interp_filters, int xs,
                                          int ys, int bd) {
119 120
  const int avg = conv_params->do_average;
  assert(avg == 0 || avg == 1);
121

122
  if (has_scale(xs, ys)) {
123 124 125
    if (conv_params->round == CONVOLVE_OPT_NO_ROUND) {
#if CONFIG_CONVOLVE_ROUND
      av1_highbd_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
126 127
                                    interp_filters, subpel_x, xs, subpel_y, ys,
                                    1, conv_params, bd);
128 129 130 131 132 133
      conv_params->do_post_rounding = 1;
#else
      assert(0);
#endif  // CONFIG_CONVOLVE_ROUND
    } else {
      av1_highbd_convolve_scale(src, src_stride, dst, dst_stride, w, h,
134
                                interp_filters, subpel_x, xs, subpel_y, ys, avg,
135 136
                                bd);
    }
137 138 139 140 141 142 143 144 145 146
  } else {
    subpel_x >>= SCALE_EXTRA_BITS;
    subpel_y >>= SCALE_EXTRA_BITS;
    xs >>= SCALE_EXTRA_BITS;
    ys >>= SCALE_EXTRA_BITS;
    assert(subpel_x < SUBPEL_SHIFTS);
    assert(subpel_y < SUBPEL_SHIFTS);
    assert(xs <= SUBPEL_SHIFTS);
    assert(ys <= SUBPEL_SHIFTS);
    if (conv_params->round == CONVOLVE_OPT_NO_ROUND) {
147
#if CONFIG_CONVOLVE_ROUND
148
      av1_highbd_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
149 150
                                    interp_filters, subpel_x, xs, subpel_y, ys,
                                    0, conv_params, bd);
151
      conv_params->do_post_rounding = 1;
152
#else
153
      assert(0);
154 155
#endif  // CONFIG_CONVOLVE_ROUND
    } else {
156 157 158 159 160 161 162 163 164 165
      InterpFilterParams filter_params_x, filter_params_y;
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y);

      if (filter_params_x.taps == SUBPEL_TAPS &&
          filter_params_y.taps == SUBPEL_TAPS && w > 2 && h > 2) {
        const int16_t *kernel_x =
            av1_get_interp_filter_subpel_kernel(filter_params_x, subpel_x);
        const int16_t *kernel_y =
            av1_get_interp_filter_subpel_kernel(filter_params_y, subpel_y);
166 167 168 169 170
        sf->highbd_predict[subpel_x != 0][subpel_y != 0][avg](
            src, src_stride, dst, dst_stride, kernel_x, xs, kernel_y, ys, w, h,
            bd);
      } else {
        av1_highbd_convolve(src, src_stride, dst, dst_stride, w, h,
171 172
                            interp_filters, subpel_x, xs, subpel_y, ys, avg,
                            bd);
173
      }
174
    }
175
  }
Jingning Han's avatar
Jingning Han committed
176
}
177
#endif  // CONFIG_HIGHBITDEPTH
Jingning Han's avatar
Jingning Han committed
178

179
// Set to (1 << 5) if the 32-ary codebooks are used for any bock size
clang-format's avatar
clang-format committed
180
#define MAX_WEDGE_TYPES (1 << 4)
181

clang-format's avatar
clang-format committed
182 183 184
#define MAX_WEDGE_SIZE_LOG2 5  // 32x32
#define MAX_WEDGE_SIZE (1 << MAX_WEDGE_SIZE_LOG2)
#define MAX_WEDGE_SQUARE (MAX_WEDGE_SIZE * MAX_WEDGE_SIZE)
185

186 187
#define WEDGE_WEIGHT_BITS 6

clang-format's avatar
clang-format committed
188
#define WEDGE_NONE -1
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207

// Angles are with respect to horizontal anti-clockwise
typedef enum {
  WEDGE_HORIZONTAL = 0,
  WEDGE_VERTICAL = 1,
  WEDGE_OBLIQUE27 = 2,
  WEDGE_OBLIQUE63 = 3,
  WEDGE_OBLIQUE117 = 4,
  WEDGE_OBLIQUE153 = 5,
  WEDGE_DIRECTIONS
} WedgeDirectionType;

// 3-tuple: {direction, x_offset, y_offset}
typedef struct {
  WedgeDirectionType direction;
  int x_offset;
  int y_offset;
} wedge_code_type;

208 209
typedef uint8_t *wedge_masks_type[MAX_WEDGE_TYPES];

210 211 212 213 214
typedef struct {
  int bits;
  const wedge_code_type *codebook;
  uint8_t *signflip;
  int smoother;
215
  wedge_masks_type *masks;
216 217
} wedge_params_type;

218
extern const wedge_params_type wedge_params_lookup[BLOCK_SIZES_ALL];
219

220 221
static INLINE int is_interinter_compound_used(COMPOUND_TYPE type,
                                              BLOCK_SIZE sb_type) {
222
  (void)sb_type;
223
  switch (type) {
224
#if CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
225
    case COMPOUND_AVERAGE: return sb_type >= BLOCK_4X4;
226
#else   // CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
227
    case COMPOUND_AVERAGE: return 1;
228
#endif  // CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
229
#if CONFIG_WEDGE
230
    case COMPOUND_WEDGE: return wedge_params_lookup[sb_type].bits > 0;
231
#endif  // CONFIG_WEDGE
232
#if CONFIG_COMPOUND_SEGMENT
233 234
    case COMPOUND_SEG:
      return AOMMIN(block_size_wide[sb_type], block_size_high[sb_type]) >= 8;
235 236 237
#endif  // CONFIG_COMPOUND_SEGMENT
    default: assert(0); return 0;
  }
238
}
239

240 241
static INLINE int is_any_masked_compound_used(BLOCK_SIZE sb_type) {
  COMPOUND_TYPE comp_type;
242
#if CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
243
  if (sb_type < BLOCK_4X4) return 0;
244
#endif  // CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
245 246 247 248 249 250 251 252 253 254
  for (comp_type = 0; comp_type < COMPOUND_TYPES; comp_type++) {
    if (is_masked_compound_type(comp_type) &&
        is_interinter_compound_used(comp_type, sb_type))
      return 1;
  }
  return 0;
}

static INLINE int get_wedge_bits_lookup(BLOCK_SIZE sb_type) {
  return wedge_params_lookup[sb_type].bits;
255 256 257
}

static INLINE int get_interinter_wedge_bits(BLOCK_SIZE sb_type) {
258
  const int wbits = wedge_params_lookup[sb_type].bits;
259 260 261 262
  return (wbits > 0) ? wbits + 1 : 0;
}

static INLINE int is_interintra_wedge_used(BLOCK_SIZE sb_type) {
clang-format's avatar
clang-format committed
263
  (void)sb_type;
264
  return wedge_params_lookup[sb_type].bits > 0;
265 266 267
}

static INLINE int get_interintra_wedge_bits(BLOCK_SIZE sb_type) {
268
  return wedge_params_lookup[sb_type].bits;
269
}
270 271

#if CONFIG_COMPOUND_SEGMENT
272
void build_compound_seg_mask(uint8_t *mask, SEG_MASK_TYPE mask_type,
273 274 275
                             const uint8_t *src0, int src0_stride,
                             const uint8_t *src1, int src1_stride,
                             BLOCK_SIZE sb_type, int h, int w);
276
#if CONFIG_HIGHBITDEPTH
277 278 279 280
void build_compound_seg_mask_highbd(uint8_t *mask, SEG_MASK_TYPE mask_type,
                                    const uint8_t *src0, int src0_stride,
                                    const uint8_t *src1, int src1_stride,
                                    BLOCK_SIZE sb_type, int h, int w, int bd);
281
#endif  // CONFIG_HIGHBITDEPTH
282
#endif  // CONFIG_COMPOUND_SEGMENT
283

284 285 286 287 288
void av1_make_masked_inter_predictor(
    const uint8_t *pre, int pre_stride, uint8_t *dst, int dst_stride,
    const int subpel_x, const int subpel_y, const struct scale_factors *sf,
    int w, int h, ConvolveParams *conv_params, InterpFilters interp_filters,
    int xs, int ys,
Yi Luo's avatar
Yi Luo committed
289
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION || CONFIG_COMPOUND_SEGMENT
290
    int plane,
Yi Luo's avatar
Yi Luo committed
291
#endif
292
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
293
    const WarpTypesAllowed *warp_types, int p_col, int p_row, int ref,
294
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
295
    MACROBLOCKD *xd);
296

Jingning Han's avatar
Jingning Han committed
297 298 299 300 301
static INLINE int round_mv_comp_q4(int value) {
  return (value < 0 ? value - 2 : value + 2) / 4;
}

static MV mi_mv_pred_q4(const MODE_INFO *mi, int idx) {
clang-format's avatar
clang-format committed
302 303 304 305 306 307 308 309
  MV res = {
    round_mv_comp_q4(
        mi->bmi[0].as_mv[idx].as_mv.row + mi->bmi[1].as_mv[idx].as_mv.row +
        mi->bmi[2].as_mv[idx].as_mv.row + mi->bmi[3].as_mv[idx].as_mv.row),
    round_mv_comp_q4(
        mi->bmi[0].as_mv[idx].as_mv.col + mi->bmi[1].as_mv[idx].as_mv.col +
        mi->bmi[2].as_mv[idx].as_mv.col + mi->bmi[3].as_mv[idx].as_mv.col)
  };
Jingning Han's avatar
Jingning Han committed
310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325
  return res;
}

static INLINE int round_mv_comp_q2(int value) {
  return (value < 0 ? value - 1 : value + 1) / 2;
}

static MV mi_mv_pred_q2(const MODE_INFO *mi, int idx, int block0, int block1) {
  MV res = { round_mv_comp_q2(mi->bmi[block0].as_mv[idx].as_mv.row +
                              mi->bmi[block1].as_mv[idx].as_mv.row),
             round_mv_comp_q2(mi->bmi[block0].as_mv[idx].as_mv.col +
                              mi->bmi[block1].as_mv[idx].as_mv.col) };
  return res;
}

// TODO(jkoleszar): yet another mv clamping function :-(
Johann's avatar
Johann committed
326
static INLINE MV clamp_mv_to_umv_border_sb(const MACROBLOCKD *xd,
clang-format's avatar
clang-format committed
327 328
                                           const MV *src_mv, int bw, int bh,
                                           int ss_x, int ss_y) {
Jingning Han's avatar
Jingning Han committed
329 330 331
  // If the MV points so far into the UMV border that no visible pixels
  // are used for reconstruction, the subpel part of the MV can be
  // discarded and the MV limited to 16 pixels with equivalent results.
Yaowu Xu's avatar
Yaowu Xu committed
332
  const int spel_left = (AOM_INTERP_EXTEND + bw) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
333
  const int spel_right = spel_left - SUBPEL_SHIFTS;
Yaowu Xu's avatar
Yaowu Xu committed
334
  const int spel_top = (AOM_INTERP_EXTEND + bh) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
335
  const int spel_bottom = spel_top - SUBPEL_SHIFTS;
clang-format's avatar
clang-format committed
336 337
  MV clamped_mv = { src_mv->row * (1 << (1 - ss_y)),
                    src_mv->col * (1 << (1 - ss_x)) };
Jingning Han's avatar
Jingning Han committed
338 339 340
  assert(ss_x <= 1);
  assert(ss_y <= 1);

clang-format's avatar
clang-format committed
341
  clamp_mv(&clamped_mv, xd->mb_to_left_edge * (1 << (1 - ss_x)) - spel_left,
Jingning Han's avatar
Jingning Han committed
342 343 344 345 346 347 348
           xd->mb_to_right_edge * (1 << (1 - ss_x)) + spel_right,
           xd->mb_to_top_edge * (1 << (1 - ss_y)) - spel_top,
           xd->mb_to_bottom_edge * (1 << (1 - ss_y)) + spel_bottom);

  return clamped_mv;
}

Johann's avatar
Johann committed
349 350
static INLINE MV average_split_mvs(const struct macroblockd_plane *pd,
                                   const MODE_INFO *mi, int ref, int block) {
Jingning Han's avatar
Jingning Han committed
351
  const int ss_idx = ((pd->subsampling_x > 0) << 1) | (pd->subsampling_y > 0);
clang-format's avatar
clang-format committed
352
  MV res = { 0, 0 };
Jingning Han's avatar
Jingning Han committed
353
  switch (ss_idx) {
clang-format's avatar
clang-format committed
354 355 356 357 358
    case 0: res = mi->bmi[block].as_mv[ref].as_mv; break;
    case 1: res = mi_mv_pred_q2(mi, ref, block, block + 2); break;
    case 2: res = mi_mv_pred_q2(mi, ref, block, block + 1); break;
    case 3: res = mi_mv_pred_q4(mi, ref); break;
    default: assert(ss_idx <= 3 && ss_idx >= 0);
Jingning Han's avatar
Jingning Han committed
359 360 361 362
  }
  return res;
}

363 364 365
void av1_build_inter_predictors_sby(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                    int mi_row, int mi_col, BUFFER_SET *ctx,
                                    BLOCK_SIZE bsize);
Jingning Han's avatar
Jingning Han committed
366

367 368 369
void av1_build_inter_predictors_sbuv(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                     int mi_row, int mi_col, BUFFER_SET *ctx,
                                     BLOCK_SIZE bsize);
Jingning Han's avatar
Jingning Han committed
370

371 372 373
void av1_build_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                   int mi_row, int mi_col, BUFFER_SET *ctx,
                                   BLOCK_SIZE bsize);
374

Yaowu Xu's avatar
Yaowu Xu committed
375
void av1_build_inter_predictor(const uint8_t *src, int src_stride, uint8_t *dst,
376
                               int dst_stride, const MV *src_mv,
Yaowu Xu's avatar
Yaowu Xu committed
377
                               const struct scale_factors *sf, int w, int h,
378
                               ConvolveParams *conv_params,
379
                               InterpFilters interp_filters,
380 381 382 383
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
                               const WarpTypesAllowed *warp_types, int p_col,
                               int p_row, int plane, int ref,
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
384 385
                               enum mv_precision precision, int x, int y,
                               const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
386

387
#if CONFIG_HIGHBITDEPTH
388 389 390
void av1_highbd_build_inter_predictor(
    const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride,
    const MV *mv_q3, const struct scale_factors *sf, int w, int h, int do_avg,
391
    InterpFilters interp_filters,
392 393 394 395 396
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
    const WarpTypesAllowed *warp_types, int p_col, int p_row,
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
    int plane, enum mv_precision precision, int x, int y,
    const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
397 398 399 400
#endif

static INLINE int scaled_buffer_offset(int x_offset, int y_offset, int stride,
                                       const struct scale_factors *sf) {
401 402 403 404
  const int x =
      sf ? sf->scale_value_x(x_offset, sf) >> SCALE_EXTRA_BITS : x_offset;
  const int y =
      sf ? sf->scale_value_y(y_offset, sf) >> SCALE_EXTRA_BITS : y_offset;
Jingning Han's avatar
Jingning Han committed
405 406 407
  return y * stride + x;
}

408 409 410
static INLINE void setup_pred_plane(struct buf_2d *dst, BLOCK_SIZE bsize,
                                    uint8_t *src, int width, int height,
                                    int stride, int mi_row, int mi_col,
Jingning Han's avatar
Jingning Han committed
411 412
                                    const struct scale_factors *scale,
                                    int subsampling_x, int subsampling_y) {
413
#if CONFIG_CHROMA_SUB8X8
414 415 416 417 418
  // Offset the buffer pointer
  if (subsampling_y && (mi_row & 0x01) && (mi_size_high[bsize] == 1))
    mi_row -= 1;
  if (subsampling_x && (mi_col & 0x01) && (mi_size_wide[bsize] == 1))
    mi_col -= 1;
419 420 421 422
#else
  (void)bsize;
#endif

Jingning Han's avatar
Jingning Han committed
423 424 425
  const int x = (MI_SIZE * mi_col) >> subsampling_x;
  const int y = (MI_SIZE * mi_row) >> subsampling_y;
  dst->buf = src + scaled_buffer_offset(x, y, stride, scale);
426 427 428
  dst->buf0 = src;
  dst->width = width;
  dst->height = height;
Jingning Han's avatar
Jingning Han committed
429 430 431
  dst->stride = stride;
}

Yaowu Xu's avatar
Yaowu Xu committed
432 433 434
void av1_setup_dst_planes(struct macroblockd_plane *planes, BLOCK_SIZE bsize,
                          const YV12_BUFFER_CONFIG *src, int mi_row,
                          int mi_col);
Jingning Han's avatar
Jingning Han committed
435

Yaowu Xu's avatar
Yaowu Xu committed
436 437 438
void av1_setup_pre_planes(MACROBLOCKD *xd, int idx,
                          const YV12_BUFFER_CONFIG *src, int mi_row, int mi_col,
                          const struct scale_factors *sf);
439

440 441
// Detect if the block have sub-pixel level motion vectors
// per component.
442
#define CHECK_SUBPEL 0
443 444
static INLINE int has_subpel_mv_component(const MODE_INFO *const mi,
                                          const MACROBLOCKD *const xd,
445
                                          int dir) {
446
#if CHECK_SUBPEL
447
  const MB_MODE_INFO *const mbmi = &mi->mbmi;
448 449 450
  const BLOCK_SIZE bsize = mbmi->sb_type;
  int plane;
  int ref = (dir >> 1);
451 452 453 454 455
#if CONFIG_CB4X4
  const int unify_bsize = 1;
#else
  const int unify_bsize = 0;
#endif
456

457
  if (bsize >= BLOCK_8X8 || unify_bsize) {
458
    if (dir & 0x01) {
clang-format's avatar
clang-format committed
459
      if (mbmi->mv[ref].as_mv.col & SUBPEL_MASK) return 1;
460
    } else {
clang-format's avatar
clang-format committed
461
      if (mbmi->mv[ref].as_mv.row & SUBPEL_MASK) return 1;
462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
    }
  } else {
    for (plane = 0; plane < MAX_MB_PLANE; ++plane) {
      const PARTITION_TYPE bp = BLOCK_8X8 - bsize;
      const struct macroblockd_plane *const pd = &xd->plane[plane];
      const int have_vsplit = bp != PARTITION_HORZ;
      const int have_hsplit = bp != PARTITION_VERT;
      const int num_4x4_w = 2 >> ((!have_vsplit) | pd->subsampling_x);
      const int num_4x4_h = 2 >> ((!have_hsplit) | pd->subsampling_y);

      int x, y;
      for (y = 0; y < num_4x4_h; ++y) {
        for (x = 0; x < num_4x4_w; ++x) {
          const MV mv = average_split_mvs(pd, mi, ref, y * 2 + x);
          if (dir & 0x01) {
clang-format's avatar
clang-format committed
477
            if (mv.col & SUBPEL_MASK) return 1;
478
          } else {
clang-format's avatar
clang-format committed
479
            if (mv.row & SUBPEL_MASK) return 1;
480 481 482 483 484 485 486
          }
        }
      }
    }
  }

  return 0;
487 488 489 490 491 492
#else
  (void)mi;
  (void)xd;
  (void)dir;
  return 1;
#endif
493 494
}

495 496
static INLINE void set_default_interp_filters(
    MB_MODE_INFO *const mbmi, InterpFilter frame_interp_filter) {
497 498
  mbmi->interp_filters =
      av1_broadcast_interp_filter(av1_unswitchable_filter(frame_interp_filter));
499 500
}

Yaowu Xu's avatar
Yaowu Xu committed
501
static INLINE int av1_is_interp_needed(const MACROBLOCKD *const xd) {
502 503 504 505 506 507 508 509 510 511 512 513
  (void)xd;
#if CONFIG_WARPED_MOTION
  const MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi;
  if (mbmi->motion_mode == WARPED_CAUSAL) return 0;
#endif  // CONFIG_WARPED_MOTION
#if CONFIG_GLOBAL_MOTION
  if (is_nontrans_global_motion(xd)) return 0;
#endif  // CONFIG_GLOBAL_MOTION
  return 1;
}

static INLINE int av1_is_interp_search_needed(const MACROBLOCKD *const xd) {
514
  MODE_INFO *const mi = xd->mi[0];
515 516 517 518 519 520 521 522
  const int is_compound = has_second_ref(&mi->mbmi);
  int ref;
  for (ref = 0; ref < 1 + is_compound; ++ref) {
    int row_col;
    for (row_col = 0; row_col < 2; ++row_col) {
      const int dir = (ref << 1) + row_col;
      if (has_subpel_mv_component(mi, xd, dir)) {
        return 1;
523 524 525
      }
    }
  }
526
  return 0;
527
}
528

Yue Chen's avatar
Yue Chen committed
529
#if CONFIG_MOTION_VAR
Yaowu Xu's avatar
Yaowu Xu committed
530
const uint8_t *av1_get_obmc_mask(int length);
531 532
void av1_count_overlappable_neighbors(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                      int mi_row, int mi_col);
533
void av1_build_obmc_inter_prediction(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
534 535 536 537 538
                                     int mi_row, int mi_col,
                                     uint8_t *above[MAX_MB_PLANE],
                                     int above_stride[MAX_MB_PLANE],
                                     uint8_t *left[MAX_MB_PLANE],
                                     int left_stride[MAX_MB_PLANE]);
539
void av1_build_prediction_by_above_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
540 541
                                         int mi_row, int mi_col,
                                         uint8_t *tmp_buf[MAX_MB_PLANE],
542 543
                                         int tmp_width[MAX_MB_PLANE],
                                         int tmp_height[MAX_MB_PLANE],
544
                                         int tmp_stride[MAX_MB_PLANE]);
545
void av1_build_prediction_by_left_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
546 547 548 549 550
                                        int mi_row, int mi_col,
                                        uint8_t *tmp_buf[MAX_MB_PLANE],
                                        int tmp_width[MAX_MB_PLANE],
                                        int tmp_height[MAX_MB_PLANE],
                                        int tmp_stride[MAX_MB_PLANE]);
551 552
void av1_build_obmc_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                        int mi_row, int mi_col);
553 554 555 556
#if CONFIG_NCOBMC
void av1_build_ncobmc_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                          int mi_row, int mi_col);
#endif
Yue Chen's avatar
Yue Chen committed
557
#endif  // CONFIG_MOTION_VAR
558

559 560
#define MASK_MASTER_SIZE ((MAX_WEDGE_SIZE) << 1)
#define MASK_MASTER_STRIDE (MASK_MASTER_SIZE)
561

Yaowu Xu's avatar
Yaowu Xu committed
562
void av1_init_wedge_masks();
563

Yaowu Xu's avatar
Yaowu Xu committed
564 565 566
static INLINE const uint8_t *av1_get_contiguous_soft_mask(int wedge_index,
                                                          int wedge_sign,
                                                          BLOCK_SIZE sb_type) {
567 568 569
  return wedge_params_lookup[sb_type].masks[wedge_sign][wedge_index];
}

Yaowu Xu's avatar
Yaowu Xu committed
570 571 572 573
const uint8_t *av1_get_soft_mask(int wedge_index, int wedge_sign,
                                 BLOCK_SIZE sb_type, int wedge_offset_x,
                                 int wedge_offset_y);

574 575 576 577 578 579 580 581 582
const uint8_t *av1_get_compound_type_mask_inverse(
    const INTERINTER_COMPOUND_DATA *const comp_data,
#if CONFIG_COMPOUND_SEGMENT
    uint8_t *mask_buffer, int h, int w, int stride,
#endif
    BLOCK_SIZE sb_type);

const uint8_t *av1_get_compound_type_mask(
    const INTERINTER_COMPOUND_DATA *const comp_data, BLOCK_SIZE sb_type);
583
#if CONFIG_INTERINTRA
584 585 586 587 588 589 590 591 592 593 594
void av1_build_interintra_predictors(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                     uint8_t *ypred, uint8_t *upred,
                                     uint8_t *vpred, int ystride, int ustride,
                                     int vstride, BUFFER_SET *ctx,
                                     BLOCK_SIZE bsize);
void av1_build_interintra_predictors_sby(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                         uint8_t *ypred, int ystride,
                                         BUFFER_SET *ctx, BLOCK_SIZE bsize);
void av1_build_interintra_predictors_sbc(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                         uint8_t *upred, int ustride,
                                         BUFFER_SET *ctx, int plane,
Yaowu Xu's avatar
Yaowu Xu committed
595
                                         BLOCK_SIZE bsize);
596 597 598 599 600 601 602 603
void av1_build_interintra_predictors_sbuv(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                          uint8_t *upred, uint8_t *vpred,
                                          int ustride, int vstride,
                                          BUFFER_SET *ctx, BLOCK_SIZE bsize);

void av1_build_intra_predictors_for_interintra(
    const AV1_COMMON *cm, MACROBLOCKD *xd, BLOCK_SIZE bsize, int plane,
    BUFFER_SET *ctx, uint8_t *intra_pred, int intra_stride);
Yaowu Xu's avatar
Yaowu Xu committed
604 605 606
void av1_combine_interintra(MACROBLOCKD *xd, BLOCK_SIZE bsize, int plane,
                            const uint8_t *inter_pred, int inter_stride,
                            const uint8_t *intra_pred, int intra_stride);
607
#endif  // CONFIG_INTERINTRA
608
// Encoder only
Yaowu Xu's avatar
Yaowu Xu committed
609
void av1_build_inter_predictors_for_planes_single_buf(
clang-format's avatar
clang-format committed
610 611
    MACROBLOCKD *xd, BLOCK_SIZE bsize, int plane_from, int plane_to, int mi_row,
    int mi_col, int ref, uint8_t *ext_dst[3], int ext_dst_stride[3]);
612 613 614 615 616 617
void av1_build_wedge_inter_predictor_from_buf(MACROBLOCKD *xd, BLOCK_SIZE bsize,
                                              int plane_from, int plane_to,
                                              uint8_t *ext_dst0[3],
                                              int ext_dst_stride0[3],
                                              uint8_t *ext_dst1[3],
                                              int ext_dst_stride1[3]);
618

619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657
#if CONFIG_NCOBMC_ADAPT_WEIGHT
#define ASSIGN_ALIGNED_PTRS(p, a, s) \
  p[0] = a;                          \
  p[1] = a + s;                      \
  p[2] = a + 2 * s;

#define ASSIGN_ALIGNED_PTRS_HBD(p, a, s, l) \
  p[0] = CONVERT_TO_BYTEPTR(a);             \
  p[1] = CONVERT_TO_BYTEPTR(a + s * l);     \
  p[2] = CONVERT_TO_BYTEPTR(a + 2 * s * l);

void alloc_ncobmc_pred_buffer(MACROBLOCKD *const xd);
void free_ncobmc_pred_buffer(MACROBLOCKD *const xd);
void set_sb_mi_boundaries(const AV1_COMMON *const cm, MACROBLOCKD *const xd,
                          const int mi_row, const int mi_col);

void reset_xd_boundary(MACROBLOCKD *xd, int mi_row, int bh, int mi_col, int bw,
                       int mi_rows, int mi_cols);

void get_pred_from_intrpl_buf(MACROBLOCKD *xd, int mi_row, int mi_col,
                              BLOCK_SIZE bsize, int plane);

void build_ncobmc_intrpl_pred(const AV1_COMMON *const cm, MACROBLOCKD *xd,
                              int plane, int pxl_row, int pxl_col,
                              BLOCK_SIZE bsize, uint8_t *preds[][MAX_MB_PLANE],
                              int ps[MAX_MB_PLANE],  // pred buffer strides
                              int mode);

void av1_get_ext_blk_preds(const AV1_COMMON *cm, MACROBLOCKD *xd, int bsize,
                           int mi_row, int mi_col,
                           uint8_t *dst_buf[][MAX_MB_PLANE],
                           int dst_stride[MAX_MB_PLANE]);

void av1_get_ori_blk_pred(const AV1_COMMON *cm, MACROBLOCKD *xd, int bsize,
                          int mi_row, int mi_col,
                          uint8_t *dst_buf[MAX_MB_PLANE],
                          int dst_stride[MAX_MB_PLANE]);
#endif  // CONFIG_NCOBMC_ADAPT_WEIGHT

Jingning Han's avatar
Jingning Han committed
658 659 660 661
#ifdef __cplusplus
}  // extern "C"
#endif

Yaowu Xu's avatar
Yaowu Xu committed
662
#endif  // AV1_COMMON_RECONINTER_H_