reconinter.h 27 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

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

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

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

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

89
      if (w <= 2 || h <= 2) {
90
        av1_convolve_c(src, src_stride, dst, dst_stride, w, h, interp_filters,
91
                       subpel_x, xs, subpel_y, ys, conv_params);
92
93
94
95
96
97
      } 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);
98
99
100
        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 {
101
        av1_convolve(src, src_stride, dst, dst_stride, w, h, interp_filters,
102
103
                     subpel_x, xs, subpel_y, ys, conv_params);
      }
104
    }
105
  }
Jingning Han's avatar
Jingning Han committed
106
107
}

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

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

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

clang-format's avatar
clang-format committed
179
180
181
#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)
182

183
184
#define WEDGE_WEIGHT_BITS 6

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

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

205
206
typedef uint8_t *wedge_masks_type[MAX_WEDGE_TYPES];

207
208
209
210
211
typedef struct {
  int bits;
  const wedge_code_type *codebook;
  uint8_t *signflip;
  int smoother;
212
  wedge_masks_type *masks;
213
214
} wedge_params_type;

215
extern const wedge_params_type wedge_params_lookup[BLOCK_SIZES_ALL];
216

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

237
238
static INLINE int is_any_masked_compound_used(BLOCK_SIZE sb_type) {
  COMPOUND_TYPE comp_type;
239
#if CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
240
  if (sb_type < BLOCK_4X4) return 0;
241
#endif  // CONFIG_CHROMA_2X2 || CONFIG_CHROMA_SUB8X8
242
243
244
245
246
247
248
249
250
251
  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;
252
253
254
}

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

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

static INLINE int get_interintra_wedge_bits(BLOCK_SIZE sb_type) {
265
  return wedge_params_lookup[sb_type].bits;
266
}
267
268

#if CONFIG_COMPOUND_SEGMENT
269
void build_compound_seg_mask(uint8_t *mask, SEG_MASK_TYPE mask_type,
270
271
272
                             const uint8_t *src0, int src0_stride,
                             const uint8_t *src1, int src1_stride,
                             BLOCK_SIZE sb_type, int h, int w);
273
#if CONFIG_HIGHBITDEPTH
274
275
276
277
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);
278
#endif  // CONFIG_HIGHBITDEPTH
279
#endif  // CONFIG_COMPOUND_SEGMENT
280

281
282
283
284
285
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,
286
#if CONFIG_SUPERTX
287
    int wedge_offset_x, int wedge_offset_y,
288
#endif  // CONFIG_SUPERTX
289
    int plane,
290
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
291
    const WarpTypesAllowed *warp_types, int p_col, int p_row, int ref,
292
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
293
    MACROBLOCKD *xd);
294

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

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

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

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

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

#if CONFIG_SUPERTX
374
void av1_build_inter_predictor_sb_sub8x8_extend(const AV1_COMMON *cm,
375
376
377
378
                                                MACROBLOCKD *xd, int mi_row_ori,
                                                int mi_col_ori, int mi_row,
                                                int mi_col, int plane,
                                                BLOCK_SIZE bsize, int block);
379

380
381
382
383
void av1_build_inter_predictor_sb_extend(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                         int mi_row_ori, int mi_col_ori,
                                         int mi_row, int mi_col, int plane,
                                         BLOCK_SIZE bsize);
384
struct macroblockd_plane;
Yaowu Xu's avatar
Yaowu Xu committed
385
void av1_build_masked_inter_predictor_complex(
clang-format's avatar
clang-format committed
386
387
388
389
    MACROBLOCKD *xd, uint8_t *dst, int dst_stride, const uint8_t *pre,
    int pre_stride, int mi_row, int mi_col, int mi_row_ori, int mi_col_ori,
    BLOCK_SIZE bsize, BLOCK_SIZE top_bsize, PARTITION_TYPE partition,
    int plane);
390
#endif  // CONFIG_SUPERTX
Jingning Han's avatar
Jingning Han committed
391

Yaowu Xu's avatar
Yaowu Xu committed
392
void av1_build_inter_predictor(const uint8_t *src, int src_stride, uint8_t *dst,
393
                               int dst_stride, const MV *src_mv,
Yaowu Xu's avatar
Yaowu Xu committed
394
                               const struct scale_factors *sf, int w, int h,
395
                               ConvolveParams *conv_params,
396
                               InterpFilters interp_filters,
397
398
399
400
#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
401
402
                               enum mv_precision precision, int x, int y,
                               const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
403

404
#if CONFIG_HIGHBITDEPTH
405
406
407
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,
408
    InterpFilters interp_filters,
409
410
411
412
413
#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
414
415
416
417
#endif

static INLINE int scaled_buffer_offset(int x_offset, int y_offset, int stride,
                                       const struct scale_factors *sf) {
418
419
420
421
  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
422
423
424
  return y * stride + x;
}

425
426
427
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
428
429
                                    const struct scale_factors *scale,
                                    int subsampling_x, int subsampling_y) {
430
#if CONFIG_CHROMA_SUB8X8
431
432
433
434
435
  // 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;
436
437
438
439
#else
  (void)bsize;
#endif

Jingning Han's avatar
Jingning Han committed
440
441
442
  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);
443
444
445
  dst->buf0 = src;
  dst->width = width;
  dst->height = height;
Jingning Han's avatar
Jingning Han committed
446
447
448
  dst->stride = stride;
}

Yaowu Xu's avatar
Yaowu Xu committed
449
450
451
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
452

Yaowu Xu's avatar
Yaowu Xu committed
453
454
455
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);
456

457
458
// Detect if the block have sub-pixel level motion vectors
// per component.
459
#define CHECK_SUBPEL 0
460
461
static INLINE int has_subpel_mv_component(const MODE_INFO *const mi,
                                          const MACROBLOCKD *const xd,
462
                                          int dir) {
463
#if CHECK_SUBPEL
464
  const MB_MODE_INFO *const mbmi = &mi->mbmi;
465
466
467
  const BLOCK_SIZE bsize = mbmi->sb_type;
  int plane;
  int ref = (dir >> 1);
468
469
470
471
472
#if CONFIG_CB4X4
  const int unify_bsize = 1;
#else
  const int unify_bsize = 0;
#endif
473

474
  if (bsize >= BLOCK_8X8 || unify_bsize) {
475
    if (dir & 0x01) {
clang-format's avatar
clang-format committed
476
      if (mbmi->mv[ref].as_mv.col & SUBPEL_MASK) return 1;
477
    } else {
clang-format's avatar
clang-format committed
478
      if (mbmi->mv[ref].as_mv.row & SUBPEL_MASK) return 1;
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
    }
  } 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
494
            if (mv.col & SUBPEL_MASK) return 1;
495
          } else {
clang-format's avatar
clang-format committed
496
            if (mv.row & SUBPEL_MASK) return 1;
497
498
499
500
501
502
503
          }
        }
      }
    }
  }

  return 0;
504
505
506
507
508
509
#else
  (void)mi;
  (void)xd;
  (void)dir;
  return 1;
#endif
510
511
}

512
513
static INLINE void set_default_interp_filters(
    MB_MODE_INFO *const mbmi, InterpFilter frame_interp_filter) {
514
515
  mbmi->interp_filters =
      av1_broadcast_interp_filter(av1_unswitchable_filter(frame_interp_filter));
516
517
}

Yaowu Xu's avatar
Yaowu Xu committed
518
static INLINE int av1_is_interp_needed(const MACROBLOCKD *const xd) {
519
520
521
522
523
524
525
526
527
528
529
530
  (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) {
531
  MODE_INFO *const mi = xd->mi[0];
532
533
534
535
536
537
538
539
  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;
540
541
542
      }
    }
  }
543
  return 0;
544
}
545

Yue Chen's avatar
Yue Chen committed
546
#if CONFIG_MOTION_VAR
Yaowu Xu's avatar
Yaowu Xu committed
547
const uint8_t *av1_get_obmc_mask(int length);
548
549
void av1_count_overlappable_neighbors(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                      int mi_row, int mi_col);
550
void av1_build_obmc_inter_prediction(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
551
552
553
554
555
                                     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]);
556
void av1_build_prediction_by_above_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
557
558
                                         int mi_row, int mi_col,
                                         uint8_t *tmp_buf[MAX_MB_PLANE],
559
560
                                         int tmp_width[MAX_MB_PLANE],
                                         int tmp_height[MAX_MB_PLANE],
561
                                         int tmp_stride[MAX_MB_PLANE]);
562
void av1_build_prediction_by_left_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
563
564
565
566
567
                                        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]);
568
569
void av1_build_obmc_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                        int mi_row, int mi_col);
570
571
572
573
#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
574
#endif  // CONFIG_MOTION_VAR
575

576
577
#define MASK_MASTER_SIZE ((MAX_WEDGE_SIZE) << 1)
#define MASK_MASTER_STRIDE (MASK_MASTER_SIZE)
578

Yaowu Xu's avatar
Yaowu Xu committed
579
void av1_init_wedge_masks();
580

Yaowu Xu's avatar
Yaowu Xu committed
581
582
583
static INLINE const uint8_t *av1_get_contiguous_soft_mask(int wedge_index,
                                                          int wedge_sign,
                                                          BLOCK_SIZE sb_type) {
584
585
586
  return wedge_params_lookup[sb_type].masks[wedge_sign][wedge_index];
}

Yaowu Xu's avatar
Yaowu Xu committed
587
588
589
590
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);

591
592
593
594
595
596
597
598
599
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);
600
#if CONFIG_INTERINTRA
601
602
603
604
605
606
607
608
609
610
611
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
612
                                         BLOCK_SIZE bsize);
613
614
615
616
617
618
619
620
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
621
622
623
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);
624
#endif  // CONFIG_INTERINTRA
625
// Encoder only
Yaowu Xu's avatar
Yaowu Xu committed
626
void av1_build_inter_predictors_for_planes_single_buf(
clang-format's avatar
clang-format committed
627
628
    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]);
629
630
631
632
633
634
635
void av1_build_wedge_inter_predictor_from_buf(
    MACROBLOCKD *xd, BLOCK_SIZE bsize, int plane_from, int plane_to,
#if CONFIG_SUPERTX
    int wedge_offset_x, int wedge_offset_y,
#endif  // CONFIG_SUPERTX
    uint8_t *ext_dst0[3], int ext_dst_stride0[3], uint8_t *ext_dst1[3],
    int ext_dst_stride1[3]);
636

637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
#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
676
677
678
679
#ifdef __cplusplus
}  // extern "C"
#endif

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