reconinter.h 24.6 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
#include "av1/common/warped_motion.h"
Yaowu Xu's avatar
Yaowu Xu committed
19
#include "aom/aom_integer.h"
Jingning Han's avatar
Jingning Han committed
20

21
#if CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
22
#define WARP_WM_NEIGHBORS_WITH_OBMC 0
23
24
#endif  // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION

25
#if CONFIG_MOTION_VAR
26
#define WARP_GM_NEIGHBORS_WITH_OBMC 0
27
28
#endif  // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION

29
30
31
#define AOM_LEFT_TOP_MARGIN_SCALED \
  ((AOM_BORDER_IN_PIXELS - AOM_INTERP_EXTEND) << SCALE_SUBPEL_BITS)

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

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

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

      InterpFilterParams filter_params_x, filter_params_y;
Zhijie Yang's avatar
Zhijie Yang committed
87
88
89
90
#if CONFIG_SHORT_FILTER
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y, w, h);
#else
91
92
93
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y);

Zhijie Yang's avatar
Zhijie Yang committed
94
95
#endif

96
      if (w <= 2 || h <= 2) {
97
        av1_convolve_c(src, src_stride, dst, dst_stride, w, h, interp_filters,
98
                       subpel_x, xs, subpel_y, ys, conv_params);
99
100
101
102
103
104
      } 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);
105
106
107
        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 {
108
        av1_convolve(src, src_stride, dst, dst_stride, w, h, interp_filters,
109
110
                     subpel_x, xs, subpel_y, ys, conv_params);
      }
111
    }
112
  }
Jingning Han's avatar
Jingning Han committed
113
114
}

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

126
  if (has_scale(xs, ys)) {
127
128
129
    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,
130
131
                                    interp_filters, subpel_x, xs, subpel_y, ys,
                                    1, conv_params, bd);
132
133
134
135
136
137
      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,
138
                                interp_filters, subpel_x, xs, subpel_y, ys, avg,
139
140
                                bd);
    }
141
142
143
144
145
146
147
148
149
150
  } 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) {
151
#if CONFIG_CONVOLVE_ROUND
152
      av1_highbd_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
153
154
                                    interp_filters, subpel_x, xs, subpel_y, ys,
                                    0, conv_params, bd);
155
      conv_params->do_post_rounding = 1;
156
#else
157
      assert(0);
158
159
#endif  // CONFIG_CONVOLVE_ROUND
    } else {
160
      InterpFilterParams filter_params_x, filter_params_y;
Zhijie Yang's avatar
Zhijie Yang committed
161
162
163
164
#if CONFIG_SHORT_FILTER
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y, w, h);
#else
165
166
      av1_get_convolve_filter_params(interp_filters, 0, &filter_params_x,
                                     &filter_params_y);
Zhijie Yang's avatar
Zhijie Yang committed
167
#endif
168
169
170
171
172
173
174

      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);
175
176
177
178
179
        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,
180
181
                            interp_filters, subpel_x, xs, subpel_y, ys, avg,
                            bd);
182
      }
183
    }
184
  }
Jingning Han's avatar
Jingning Han committed
185
}
186
#endif  // CONFIG_HIGHBITDEPTH
Jingning Han's avatar
Jingning Han committed
187

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

clang-format's avatar
clang-format committed
191
192
193
#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)
194

195
196
#define WEDGE_WEIGHT_BITS 6

clang-format's avatar
clang-format committed
197
#define WEDGE_NONE -1
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216

// 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;

217
218
typedef uint8_t *wedge_masks_type[MAX_WEDGE_TYPES];

219
220
221
222
223
typedef struct {
  int bits;
  const wedge_code_type *codebook;
  uint8_t *signflip;
  int smoother;
224
  wedge_masks_type *masks;
225
226
} wedge_params_type;

227
extern const wedge_params_type wedge_params_lookup[BLOCK_SIZES_ALL];
228

229
230
static INLINE int is_interinter_compound_used(COMPOUND_TYPE type,
                                              BLOCK_SIZE sb_type) {
231
  (void)sb_type;
232
  switch (type) {
233
    case COMPOUND_AVERAGE: return sb_type >= BLOCK_4X4;
234
#if CONFIG_WEDGE
235
    case COMPOUND_WEDGE: return wedge_params_lookup[sb_type].bits > 0;
236
#endif  // CONFIG_WEDGE
237
#if CONFIG_COMPOUND_SEGMENT
238
239
    case COMPOUND_SEG:
      return AOMMIN(block_size_wide[sb_type], block_size_high[sb_type]) >= 8;
240
241
242
#endif  // CONFIG_COMPOUND_SEGMENT
    default: assert(0); return 0;
  }
243
}
244

245
246
static INLINE int is_any_masked_compound_used(BLOCK_SIZE sb_type) {
  COMPOUND_TYPE comp_type;
247
  if (sb_type < BLOCK_4X4) return 0;
248
249
250
251
252
253
254
255
256
257
  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;
258
259
260
}

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

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

static INLINE int get_interintra_wedge_bits(BLOCK_SIZE sb_type) {
271
  return wedge_params_lookup[sb_type].bits;
272
}
273
274

#if CONFIG_COMPOUND_SEGMENT
275
void build_compound_seg_mask(uint8_t *mask, SEG_MASK_TYPE mask_type,
276
277
278
                             const uint8_t *src0, int src0_stride,
                             const uint8_t *src1, int src1_stride,
                             BLOCK_SIZE sb_type, int h, int w);
279
#if CONFIG_HIGHBITDEPTH
280
281
282
283
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);
284
#endif  // CONFIG_HIGHBITDEPTH
285
#endif  // CONFIG_COMPOUND_SEGMENT
286

287
288
289
290
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,
291
292
    int xs, int ys, int plane, const WarpTypesAllowed *warp_types, int p_col,
    int p_row, int ref, MACROBLOCKD *xd);
293

Jingning Han's avatar
Jingning Han committed
294
295
296
297
298
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
299
300
301
302
303
304
305
306
  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
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
  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
323
static INLINE MV clamp_mv_to_umv_border_sb(const MACROBLOCKD *xd,
clang-format's avatar
clang-format committed
324
325
                                           const MV *src_mv, int bw, int bh,
                                           int ss_x, int ss_y) {
Jingning Han's avatar
Jingning Han committed
326
327
328
  // 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
329
  const int spel_left = (AOM_INTERP_EXTEND + bw) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
330
  const int spel_right = spel_left - SUBPEL_SHIFTS;
Yaowu Xu's avatar
Yaowu Xu committed
331
  const int spel_top = (AOM_INTERP_EXTEND + bh) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
332
  const int spel_bottom = spel_top - SUBPEL_SHIFTS;
clang-format's avatar
clang-format committed
333
334
  MV clamped_mv = { src_mv->row * (1 << (1 - ss_y)),
                    src_mv->col * (1 << (1 - ss_x)) };
Jingning Han's avatar
Jingning Han committed
335
336
337
  assert(ss_x <= 1);
  assert(ss_y <= 1);

clang-format's avatar
clang-format committed
338
  clamp_mv(&clamped_mv, xd->mb_to_left_edge * (1 << (1 - ss_x)) - spel_left,
Jingning Han's avatar
Jingning Han committed
339
340
341
342
343
344
345
           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
346
347
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
348
  const int ss_idx = ((pd->subsampling_x > 0) << 1) | (pd->subsampling_y > 0);
clang-format's avatar
clang-format committed
349
  MV res = { 0, 0 };
Jingning Han's avatar
Jingning Han committed
350
  switch (ss_idx) {
clang-format's avatar
clang-format committed
351
352
353
354
355
    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
356
357
358
359
  }
  return res;
}

360
361
362
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
363

364
365
366
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
367

368
369
370
void av1_build_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                   int mi_row, int mi_col, BUFFER_SET *ctx,
                                   BLOCK_SIZE bsize);
371

372
373
374
375
376
377
void av1_build_inter_predictor(
    const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride,
    const MV *src_mv, const struct scale_factors *sf, int w, int h,
    ConvolveParams *conv_params, InterpFilters interp_filters,
    const WarpTypesAllowed *warp_types, int p_col, int p_row, int plane,
    int ref, enum mv_precision precision, int x, int y, const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
378

379
#if CONFIG_HIGHBITDEPTH
380
381
382
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,
383
384
    InterpFilters interp_filters, const WarpTypesAllowed *warp_types, int p_col,
    int p_row, int plane, enum mv_precision precision, int x, int y,
385
    const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
386
387
388
389
#endif

static INLINE int scaled_buffer_offset(int x_offset, int y_offset, int stride,
                                       const struct scale_factors *sf) {
390
391
392
393
  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
394
395
396
  return y * stride + x;
}

397
398
399
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
400
401
                                    const struct scale_factors *scale,
                                    int subsampling_x, int subsampling_y) {
402
403
404
405
406
  // 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;
407

Jingning Han's avatar
Jingning Han committed
408
409
410
  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);
411
412
413
  dst->buf0 = src;
  dst->width = width;
  dst->height = height;
Jingning Han's avatar
Jingning Han committed
414
415
416
  dst->stride = stride;
}

Yaowu Xu's avatar
Yaowu Xu committed
417
418
419
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
420

Yaowu Xu's avatar
Yaowu Xu committed
421
422
423
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);
424

425
426
// Detect if the block have sub-pixel level motion vectors
// per component.
427
#define CHECK_SUBPEL 0
428
429
static INLINE int has_subpel_mv_component(const MODE_INFO *const mi,
                                          const MACROBLOCKD *const xd,
430
                                          int dir) {
431
#if CHECK_SUBPEL
432
  const MB_MODE_INFO *const mbmi = &mi->mbmi;
433
434
435
436
  const BLOCK_SIZE bsize = mbmi->sb_type;
  int plane;
  int ref = (dir >> 1);

437
438
  if (dir & 0x01) {
    if (mbmi->mv[ref].as_mv.col & SUBPEL_MASK) return 1;
439
  } else {
440
    if (mbmi->mv[ref].as_mv.row & SUBPEL_MASK) return 1;
441
442
443
  }

  return 0;
444
445
446
447
448
449
#else
  (void)mi;
  (void)xd;
  (void)dir;
  return 1;
#endif
450
451
}

452
453
static INLINE void set_default_interp_filters(
    MB_MODE_INFO *const mbmi, InterpFilter frame_interp_filter) {
454
455
  mbmi->interp_filters =
      av1_broadcast_interp_filter(av1_unswitchable_filter(frame_interp_filter));
456
457
}

Yaowu Xu's avatar
Yaowu Xu committed
458
static INLINE int av1_is_interp_needed(const MACROBLOCKD *const xd) {
459
460
461
462
463
464
465
466
467
468
  (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 (is_nontrans_global_motion(xd)) return 0;
  return 1;
}

static INLINE int av1_is_interp_search_needed(const MACROBLOCKD *const xd) {
469
  MODE_INFO *const mi = xd->mi[0];
470
471
472
473
474
475
476
477
  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;
478
479
480
      }
    }
  }
481
  return 0;
482
}
483

Yue Chen's avatar
Yue Chen committed
484
#if CONFIG_MOTION_VAR
Yaowu Xu's avatar
Yaowu Xu committed
485
const uint8_t *av1_get_obmc_mask(int length);
486
487
void av1_count_overlappable_neighbors(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                      int mi_row, int mi_col);
488
void av1_build_obmc_inter_prediction(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
489
490
491
492
493
                                     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]);
494
void av1_build_prediction_by_above_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
495
496
                                         int mi_row, int mi_col,
                                         uint8_t *tmp_buf[MAX_MB_PLANE],
497
498
                                         int tmp_width[MAX_MB_PLANE],
                                         int tmp_height[MAX_MB_PLANE],
499
                                         int tmp_stride[MAX_MB_PLANE]);
500
void av1_build_prediction_by_left_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
501
502
503
504
505
                                        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]);
506
507
void av1_build_obmc_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                        int mi_row, int mi_col);
508
509
510
511
#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
512
#endif  // CONFIG_MOTION_VAR
513

514
515
#define MASK_MASTER_SIZE ((MAX_WEDGE_SIZE) << 1)
#define MASK_MASTER_STRIDE (MASK_MASTER_SIZE)
516

Yaowu Xu's avatar
Yaowu Xu committed
517
void av1_init_wedge_masks();
518

Yaowu Xu's avatar
Yaowu Xu committed
519
520
521
static INLINE const uint8_t *av1_get_contiguous_soft_mask(int wedge_index,
                                                          int wedge_sign,
                                                          BLOCK_SIZE sb_type) {
522
523
524
  return wedge_params_lookup[sb_type].masks[wedge_sign][wedge_index];
}

Yaowu Xu's avatar
Yaowu Xu committed
525
526
527
528
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);

529
530
531
532
533
534
535
536
537
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);
538
#if CONFIG_INTERINTRA
539
540
541
542
543
544
545
546
547
548
549
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
550
                                         BLOCK_SIZE bsize);
551
552
553
554
555
556
557
558
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
559
560
561
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);
562
#endif  // CONFIG_INTERINTRA
563
// Encoder only
Yaowu Xu's avatar
Yaowu Xu committed
564
void av1_build_inter_predictors_for_planes_single_buf(
clang-format's avatar
clang-format committed
565
566
    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]);
567
568
569
570
571
572
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]);
573

574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
#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
613
614
615
616
#ifdef __cplusplus
}  // extern "C"
#endif

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