reconinter.h 33 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 35 36
#ifdef __cplusplus
extern "C" {
#endif

static INLINE void inter_predictor(const uint8_t *src, int src_stride,
                                   uint8_t *dst, int dst_stride,
clang-format's avatar
clang-format committed
37 38
                                   const int subpel_x, const int subpel_y,
                                   const struct scale_factors *sf, int w, int h,
39
                                   ConvolveParams *conv_params,
40
#if CONFIG_DUAL_FILTER
James Zern's avatar
James Zern committed
41
                                   const InterpFilter *interp_filter,
42
#else
James Zern's avatar
James Zern committed
43
                                   const InterpFilter interp_filter,
44
#endif
Jingning Han's avatar
Jingning Han committed
45
                                   int xs, int ys) {
46
#if CONFIG_DUAL_FILTER
47 48 49 50
  InterpFilter filter_x = av1_get_plane_interp_filter(
      interp_filter[1 + 2 * conv_params->ref], conv_params->plane);
  InterpFilter filter_y = av1_get_plane_interp_filter(
      interp_filter[0 + 2 * conv_params->ref], conv_params->plane);
51
  InterpFilterParams interp_filter_params_x =
52
      av1_get_interp_filter_params(filter_x);
53
  InterpFilterParams interp_filter_params_y =
54
      av1_get_interp_filter_params(filter_y);
55
#else
56
  InterpFilterParams interp_filter_params =
Yaowu Xu's avatar
Yaowu Xu committed
57
      av1_get_interp_filter_params(interp_filter);
58 59
#endif

Alex Converse's avatar
Alex Converse committed
60
  assert(sf);
61 62
#if CONFIG_DUAL_FILTER
  if (interp_filter_params_x.taps == SUBPEL_TAPS &&
Angie Chiang's avatar
Angie Chiang committed
63
      interp_filter_params_y.taps == SUBPEL_TAPS && w > 2 && h > 2 &&
64
      conv_params->round == CONVOLVE_OPT_ROUND && xs == 16 && ys == 16) {
65
    const int16_t *kernel_x =
Yaowu Xu's avatar
Yaowu Xu committed
66
        av1_get_interp_filter_subpel_kernel(interp_filter_params_x, subpel_x);
67
    const int16_t *kernel_y =
Yaowu Xu's avatar
Yaowu Xu committed
68
        av1_get_interp_filter_subpel_kernel(interp_filter_params_y, subpel_y);
69
#else
Angie Chiang's avatar
Angie Chiang committed
70
  if (interp_filter_params.taps == SUBPEL_TAPS && w > 2 && h > 2 &&
71
      conv_params->round == CONVOLVE_OPT_ROUND && xs == 16 && ys == 16) {
72
    const int16_t *kernel_x =
Yaowu Xu's avatar
Yaowu Xu committed
73
        av1_get_interp_filter_subpel_kernel(interp_filter_params, subpel_x);
74
    const int16_t *kernel_y =
Yaowu Xu's avatar
Yaowu Xu committed
75
        av1_get_interp_filter_subpel_kernel(interp_filter_params, subpel_y);
76
#endif
77
    sf->predict[subpel_x != 0][subpel_y != 0][conv_params->ref](
clang-format's avatar
clang-format committed
78
        src, src_stride, dst, dst_stride, kernel_x, xs, kernel_y, ys, w, h);
79
  } else {
Angie Chiang's avatar
Angie Chiang committed
80 81 82 83
// ref_idx > 0 means this is the second reference frame
// first reference frame's prediction result is already in dst
// therefore we need to average the first and second results
#if CONFIG_CONVOLVE_ROUND
84
    if (conv_params->round == CONVOLVE_OPT_NO_ROUND && xs == 16 && ys == 16)
Angie Chiang's avatar
Angie Chiang committed
85
      av1_convolve_2d_facade(src, src_stride, dst, dst_stride, w, h,
86 87 88 89 90 91
#if CONFIG_DUAL_FILTER
                             interp_filter,
#else
                             &interp_filter,
#endif
                             subpel_x, xs, subpel_y, ys, conv_params);
Angie Chiang's avatar
Angie Chiang committed
92 93
    else
#endif
94 95 96 97 98 99 100 101 102 103 104
    {
      if (xs == 16 && ys == 16) {
        av1_convolve(src, src_stride, dst, dst_stride, w, h, interp_filter,
                     subpel_x, xs, subpel_y, ys, conv_params);
      } else {
        // If xs == 16 || ys == 16 scaling is happening and the SSE2
        // instructions don't support scaling; use the C versions to be safe.
        av1_convolve_c(src, src_stride, dst, dst_stride, w, h, interp_filter,
                       subpel_x, xs, subpel_y, ys, conv_params);
      }
    }
105
  }
Jingning Han's avatar
Jingning Han committed
106 107
}

108
#if CONFIG_HIGHBITDEPTH
109 110 111 112
static INLINE void highbd_inter_predictor(const uint8_t *src, int src_stride,
                                          uint8_t *dst, int dst_stride,
                                          const int subpel_x,
                                          const int subpel_y,
clang-format's avatar
clang-format committed
113 114
                                          const struct scale_factors *sf, int w,
                                          int h, int ref,
115
#if CONFIG_DUAL_FILTER
James Zern's avatar
James Zern committed
116
                                          const InterpFilter *interp_filter,
117
#else
James Zern's avatar
James Zern committed
118
                                          const InterpFilter interp_filter,
119
#endif
120
                                          int xs, int ys, int bd) {
121 122
#if CONFIG_DUAL_FILTER
  InterpFilterParams interp_filter_params_x =
Yaowu Xu's avatar
Yaowu Xu committed
123
      av1_get_interp_filter_params(interp_filter[1 + 2 * ref]);
124
  InterpFilterParams interp_filter_params_y =
Yaowu Xu's avatar
Yaowu Xu committed
125
      av1_get_interp_filter_params(interp_filter[0 + 2 * ref]);
126
#else
127
  InterpFilterParams interp_filter_params =
Yaowu Xu's avatar
Yaowu Xu committed
128
      av1_get_interp_filter_params(interp_filter);
129 130 131 132
#endif

#if CONFIG_DUAL_FILTER
  if (interp_filter_params_x.taps == SUBPEL_TAPS &&
clang-format's avatar
clang-format committed
133
      interp_filter_params_y.taps == SUBPEL_TAPS && w > 2 && h > 2) {
134
    const int16_t *kernel_x =
Yaowu Xu's avatar
Yaowu Xu committed
135
        av1_get_interp_filter_subpel_kernel(interp_filter_params_x, subpel_x);
136
    const int16_t *kernel_y =
Yaowu Xu's avatar
Yaowu Xu committed
137
        av1_get_interp_filter_subpel_kernel(interp_filter_params_y, subpel_y);
138
#else
139
  if (interp_filter_params.taps == SUBPEL_TAPS && w > 2 && h > 2) {
140
    const int16_t *kernel_x =
Yaowu Xu's avatar
Yaowu Xu committed
141
        av1_get_interp_filter_subpel_kernel(interp_filter_params, subpel_x);
142
    const int16_t *kernel_y =
Yaowu Xu's avatar
Yaowu Xu committed
143
        av1_get_interp_filter_subpel_kernel(interp_filter_params, subpel_y);
144
#endif  // CONFIG_DUAL_FILTER
145
    sf->highbd_predict[subpel_x != 0][subpel_y != 0][ref](
clang-format's avatar
clang-format committed
146
        src, src_stride, dst, dst_stride, kernel_x, xs, kernel_y, ys, w, h, bd);
147
  } else {
148 149 150 151
    // ref > 0 means this is the second reference frame
    // first reference frame's prediction result is already in dst
    // therefore we need to average the first and second results
    int avg = ref > 0;
Yaowu Xu's avatar
Yaowu Xu committed
152 153
    av1_highbd_convolve(src, src_stride, dst, dst_stride, w, h, interp_filter,
                        subpel_x, xs, subpel_y, ys, avg, bd);
154
  }
Jingning Han's avatar
Jingning Han committed
155
}
156
#endif  // CONFIG_HIGHBITDEPTH
Jingning Han's avatar
Jingning Han committed
157

158
#if CONFIG_EXT_INTER
159
// Set to (1 << 5) if the 32-ary codebooks are used for any bock size
clang-format's avatar
clang-format committed
160
#define MAX_WEDGE_TYPES (1 << 4)
161

clang-format's avatar
clang-format committed
162 163 164
#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)
165

166 167
#define WEDGE_WEIGHT_BITS 6

clang-format's avatar
clang-format committed
168
#define WEDGE_NONE -1
169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187

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

188 189
typedef uint8_t *wedge_masks_type[MAX_WEDGE_TYPES];

190 191 192 193 194
typedef struct {
  int bits;
  const wedge_code_type *codebook;
  uint8_t *signflip;
  int smoother;
195
  wedge_masks_type *masks;
196 197 198 199
} wedge_params_type;

extern const wedge_params_type wedge_params_lookup[BLOCK_SIZES];

200 201
static INLINE int is_interinter_compound_used(COMPOUND_TYPE type,
                                              BLOCK_SIZE sb_type) {
202
  (void)sb_type;
203
  switch (type) {
204 205
    case COMPOUND_AVERAGE: return 1;
#if CONFIG_WEDGE
206
    case COMPOUND_WEDGE: return wedge_params_lookup[sb_type].bits > 0;
207
#endif  // CONFIG_WEDGE
208 209 210 211 212
#if CONFIG_COMPOUND_SEGMENT
    case COMPOUND_SEG: return sb_type >= BLOCK_8X8;
#endif  // CONFIG_COMPOUND_SEGMENT
    default: assert(0); return 0;
  }
213
}
214

215 216 217 218 219 220 221 222 223 224 225 226
static INLINE int is_any_masked_compound_used(BLOCK_SIZE sb_type) {
  COMPOUND_TYPE comp_type;
  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;
227 228 229
}

static INLINE int get_interinter_wedge_bits(BLOCK_SIZE sb_type) {
230
  const int wbits = wedge_params_lookup[sb_type].bits;
231 232 233 234
  return (wbits > 0) ? wbits + 1 : 0;
}

static INLINE int is_interintra_wedge_used(BLOCK_SIZE sb_type) {
clang-format's avatar
clang-format committed
235
  (void)sb_type;
236
  return wedge_params_lookup[sb_type].bits > 0;
237 238 239
}

static INLINE int get_interintra_wedge_bits(BLOCK_SIZE sb_type) {
240
  return wedge_params_lookup[sb_type].bits;
241
}
242 243

#if CONFIG_COMPOUND_SEGMENT
244
void build_compound_seg_mask(uint8_t *mask, SEG_MASK_TYPE mask_type,
245 246 247
                             const uint8_t *src0, int src0_stride,
                             const uint8_t *src1, int src1_stride,
                             BLOCK_SIZE sb_type, int h, int w);
248
#if CONFIG_HIGHBITDEPTH
249 250 251 252
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);
253
#endif  // CONFIG_HIGHBITDEPTH
254
#endif  // CONFIG_COMPOUND_SEGMENT
255 256
#endif  // CONFIG_EXT_INTER

257
void build_inter_predictors(MACROBLOCKD *xd, int plane,
Yue Chen's avatar
Yue Chen committed
258
#if CONFIG_MOTION_VAR
259
                            int mi_col_offset, int mi_row_offset,
Yue Chen's avatar
Yue Chen committed
260
#endif  // CONFIG_MOTION_VAR
clang-format's avatar
clang-format committed
261 262
                            int block, int bw, int bh, int x, int y, int w,
                            int h,
263 264 265 266 267
#if CONFIG_SUPERTX && CONFIG_EXT_INTER
                            int wedge_offset_x, int wedge_offset_y,
#endif  // CONFIG_SUPERTX && CONFIG_EXT_INTER
                            int mi_x, int mi_y);

268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
// This function will determine whether or not to create a warped
// prediction and return the appropriate motion model depending
// on the configuration. Behavior will change with different
// combinations of GLOBAL_MOTION, WARPED_MOTION and MOTION_VAR.
static INLINE int allow_warp(const MODE_INFO *const mi,
                             const WarpTypesAllowed *const warp_types,
#if CONFIG_GLOBAL_MOTION
                             const WarpedMotionParams *const gm_params,
#endif  // CONFIG_GLOBAL_MOTION
#if CONFIG_MOTION_VAR
                             int mi_col_offset, int mi_row_offset,
#endif  // CONFIG_MOTION_VAR
                             WarpedMotionParams *final_warp_params) {
  const MB_MODE_INFO *const mbmi = &mi->mbmi;
  set_default_warp_params(final_warp_params);

// Only global motion configured
#if CONFIG_GLOBAL_MOTION && !CONFIG_WARPED_MOTION && !CONFIG_MOTION_VAR
  (void)mbmi;
  if (warp_types->global_warp_allowed) {
    memcpy(final_warp_params, gm_params, sizeof(*final_warp_params));
    return 1;
  }
#endif  // CONFIG_GLOBAL_MOTION && !CONFIG_WARPED_MOTION && !CONFIG_MOTION_VAR

// Only warped motion configured
#if CONFIG_WARPED_MOTION && !CONFIG_GLOBAL_MOTION && !CONFIG_MOTION_VAR
  if (warp_types->local_warp_allowed) {
    memcpy(final_warp_params, &mbmi->wm_params[0], sizeof(*final_warp_params));
    return 1;
  }
#endif  // CONFIG_WARPED_MOTION && !CONFIG_GLOBAL_MOTION && !CONFIG_MOTION_VAR

// Warped and global motion configured
#if CONFIG_GLOBAL_MOTION && CONFIG_WARPED_MOTION && !CONFIG_MOTION_VAR
  // When both are enabled, warped will take priority. The global parameters
  // will only be used to compute projection samples to find the warped model.
  // Note that, if SEPARATE_GLOBAL_MOTION is enabled and a block chooses
  // global, it will not be possible to select WARPED_CAUSAL.
  if (warp_types->local_warp_allowed) {
    memcpy(final_warp_params, &mbmi->wm_params[0], sizeof(*final_warp_params));
    return 1;
  } else if (warp_types->global_warp_allowed) {
    memcpy(final_warp_params, gm_params, sizeof(*final_warp_params));
    return 1;
  }
#endif  // CONFIG_GLOBAL_MOTION && CONFIG_WARPED_MOTION && !CONFIG_MOTION_VAR

// Motion var and global motion configured
#if CONFIG_GLOBAL_MOTION && CONFIG_MOTION_VAR && !CONFIG_WARPED_MOTION
  // We warp if either case is true:
  //   1.) We are predicting a block which uses global motion
  //   2.) We are predicting a neighboring block of a block using OBMC,
  //       the neighboring block uses global motion, and we have enabled
  //       WARP_GM_NEIGHBORS_WITH_OBMC
  const int build_for_obmc = !(mi_col_offset == 0 && mi_row_offset == 0);
  (void)mbmi;
  if (warp_types->global_warp_allowed &&
      (WARP_GM_NEIGHBORS_WITH_OBMC || !build_for_obmc)) {
    memcpy(final_warp_params, gm_params, sizeof(*final_warp_params));
    return 1;
  }
#endif  // CONFIG_GLOBAL_MOTION && CONFIG_MOTION_VAR && !CONFIG_WARPED_MOTION

// Motion var and warped motion configured
#if CONFIG_WARPED_MOTION && CONFIG_MOTION_VAR && !CONFIG_GLOBAL_MOTION
  // We warp if either case is true:
  //   1.) We are predicting a block with motion mode WARPED_CAUSAL
  //   2.) We are predicting a neighboring block of a block using OBMC,
  //       the neighboring block has mode WARPED_CAUSAL, and we have enabled
  //       WARP_WM_NEIGHBORS_WITH_OBMC
  const int build_for_obmc = !(mi_col_offset == 0 && mi_row_offset == 0);
  if (warp_types->local_warp_allowed) {
    if ((build_for_obmc && WARP_WM_NEIGHBORS_WITH_OBMC) || (!build_for_obmc)) {
      memcpy(final_warp_params, &mbmi->wm_params[0],
             sizeof(*final_warp_params));
      return 1;
    }
  }
#endif  // CONFIG_WARPED_MOTION && CONFIG_MOTION_VAR && !CONFIG_GLOBAL_MOTION

// Motion var, warped motion and global motion all configured
#if CONFIG_WARPED_MOTION && CONFIG_MOTION_VAR && CONFIG_GLOBAL_MOTION
  const int build_for_obmc = !(mi_col_offset == 0 && mi_row_offset == 0);
  if (warp_types->local_warp_allowed) {
    if ((build_for_obmc && WARP_WM_NEIGHBORS_WITH_OBMC) || (!build_for_obmc)) {
      memcpy(final_warp_params, &mbmi->wm_params[0],
             sizeof(*final_warp_params));
      return 1;
    }
  } else if (warp_types->global_warp_allowed &&
             (WARP_GM_NEIGHBORS_WITH_OBMC || !build_for_obmc)) {
    memcpy(final_warp_params, gm_params, sizeof(*final_warp_params));
    return 1;
  }
#endif  // CONFIG_WARPED_MOTION && CONFIG_MOTION_VAR && CONFIG_GLOBAL_MOTION

  return 0;
}
#endif  // CONFIG_GLOBAL_MOTION ||CONFIG_WARPED_MOTION

Yaowu Xu's avatar
Yaowu Xu committed
370
static INLINE void av1_make_inter_predictor(
clang-format's avatar
clang-format committed
371 372
    const uint8_t *src, int src_stride, uint8_t *dst, int dst_stride,
    const int subpel_x, const int subpel_y, const struct scale_factors *sf,
373
    int w, int h, ConvolveParams *conv_params,
374
#if CONFIG_DUAL_FILTER
James Zern's avatar
James Zern committed
375
    const InterpFilter *interp_filter,
376
#else
James Zern's avatar
James Zern committed
377
    const InterpFilter interp_filter,
378
#endif
379 380 381 382
#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
383 384 385
#if CONFIG_MOTION_VAR
    int mi_col_offset, int mi_row_offset,
#endif
clang-format's avatar
clang-format committed
386 387
    int xs, int ys, const MACROBLOCKD *xd) {
  (void)xd;
388

389
#if CONFIG_MOTION_VAR
390 391 392 393
  const MODE_INFO *mi = xd->mi[mi_col_offset + xd->mi_stride * mi_row_offset];
#else
  const MODE_INFO *mi = xd->mi[0];
  (void)mi;
394
#endif  // CONFIG_MOTION_VAR
395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410

// Make sure the selected motion mode is valid for this configuration
#if CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
  assert_motion_mode_valid(mi->mbmi.motion_mode,
#if CONFIG_GLOBAL_MOTION && SEPARATE_GLOBAL_MOTION
                           0, xd->global_motion,
#endif  // CONFIG_GLOBAL_MOTION && SEPARATE_GLOBAL_MOTION
                           mi);
#endif  // CONFIG MOTION_VAR || CONFIG_WARPED_MOTION

#if CONFIG_WARPED_MOTION || CONFIG_GLOBAL_MOTION
  WarpedMotionParams final_warp_params;
  const int do_warp = allow_warp(mi, warp_types,
#if CONFIG_GLOBAL_MOTION
                                 &xd->global_motion[mi->mbmi.ref_frame[ref]],
#endif  // CONFIG_GLOBAL_MOTION
411
#if CONFIG_MOTION_VAR
412 413 414 415
                                 mi_col_offset, mi_row_offset,
#endif  // CONFIG_MOTION_VAR
                                 &final_warp_params);
  if (do_warp) {
416 417
    const struct macroblockd_plane *const pd = &xd->plane[plane];
    const struct buf_2d *const pre_buf = &pd->pre[ref];
418
    av1_warp_plane(&final_warp_params,
419
#if CONFIG_HIGHBITDEPTH
420
                   xd->cur_buf->flags & YV12_FLAG_HIGHBITDEPTH, xd->bd,
421
#endif  // CONFIG_HIGHBITDEPTH
422 423 424 425 426
                   pre_buf->buf0, pre_buf->width, pre_buf->height,
                   pre_buf->stride, dst, p_col, p_row, w, h, dst_stride,
                   pd->subsampling_x, pd->subsampling_y, xs, ys, ref);
    return;
  }
427
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
428
#if CONFIG_HIGHBITDEPTH
429
  if (xd->cur_buf->flags & YV12_FLAG_HIGHBITDEPTH) {
clang-format's avatar
clang-format committed
430
    highbd_inter_predictor(src, src_stride, dst, dst_stride, subpel_x, subpel_y,
431 432
                           sf, w, h, conv_params->ref, interp_filter, xs, ys,
                           xd->bd);
433 434
    return;
  }
435
#endif  // CONFIG_HIGHBITDEPTH
436 437
  inter_predictor(src, src_stride, dst, dst_stride, subpel_x, subpel_y, sf, w,
                  h, conv_params, interp_filter, xs, ys);
438 439 440
}

#if CONFIG_EXT_INTER
Yaowu Xu's avatar
Yaowu Xu committed
441 442 443 444 445
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,
446
#if CONFIG_DUAL_FILTER
James Zern's avatar
James Zern committed
447
                                     const InterpFilter *interp_filter,
448
#else
James Zern's avatar
James Zern committed
449
                                     const InterpFilter interp_filter,
450
#endif
Yaowu Xu's avatar
Yaowu Xu committed
451
                                     int xs, int ys,
452
#if CONFIG_SUPERTX
Yaowu Xu's avatar
Yaowu Xu committed
453
                                     int wedge_offset_x, int wedge_offset_y,
454
#endif  // CONFIG_SUPERTX
455
                                     int plane,
456 457 458 459
#if CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
                                     const WarpTypesAllowed *warp_types,
                                     int p_col, int p_row, int ref,
#endif  // CONFIG_GLOBAL_MOTION || CONFIG_WARPED_MOTION
460
                                     MACROBLOCKD *xd);
461 462
#endif  // CONFIG_EXT_INTER

Jingning Han's avatar
Jingning Han committed
463 464 465 466 467
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
468 469 470 471 472 473 474 475
  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
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
  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
492
static INLINE MV clamp_mv_to_umv_border_sb(const MACROBLOCKD *xd,
clang-format's avatar
clang-format committed
493 494
                                           const MV *src_mv, int bw, int bh,
                                           int ss_x, int ss_y) {
Jingning Han's avatar
Jingning Han committed
495 496 497
  // 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
498
  const int spel_left = (AOM_INTERP_EXTEND + bw) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
499
  const int spel_right = spel_left - SUBPEL_SHIFTS;
Yaowu Xu's avatar
Yaowu Xu committed
500
  const int spel_top = (AOM_INTERP_EXTEND + bh) << SUBPEL_BITS;
Jingning Han's avatar
Jingning Han committed
501
  const int spel_bottom = spel_top - SUBPEL_SHIFTS;
clang-format's avatar
clang-format committed
502 503
  MV clamped_mv = { src_mv->row * (1 << (1 - ss_y)),
                    src_mv->col * (1 << (1 - ss_x)) };
Jingning Han's avatar
Jingning Han committed
504 505 506
  assert(ss_x <= 1);
  assert(ss_y <= 1);

clang-format's avatar
clang-format committed
507
  clamp_mv(&clamped_mv, xd->mb_to_left_edge * (1 << (1 - ss_x)) - spel_left,
Jingning Han's avatar
Jingning Han committed
508 509 510 511 512 513 514
           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
515 516
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
517
  const int ss_idx = ((pd->subsampling_x > 0) << 1) | (pd->subsampling_y > 0);
clang-format's avatar
clang-format committed
518
  MV res = { 0, 0 };
Jingning Han's avatar
Jingning Han committed
519
  switch (ss_idx) {
clang-format's avatar
clang-format committed
520 521 522 523 524
    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
525 526 527 528
  }
  return res;
}

Yaowu Xu's avatar
Yaowu Xu committed
529 530
void av1_build_inter_predictor_sub8x8(MACROBLOCKD *xd, int plane, int i, int ir,
                                      int ic, int mi_row, int mi_col);
531

Yaowu Xu's avatar
Yaowu Xu committed
532
void av1_build_inter_predictors_sby(MACROBLOCKD *xd, int mi_row, int mi_col,
533
                                    BUFFER_SET *ctx, BLOCK_SIZE bsize);
Jingning Han's avatar
Jingning Han committed
534

Yaowu Xu's avatar
Yaowu Xu committed
535
void av1_build_inter_predictors_sbuv(MACROBLOCKD *xd, int mi_row, int mi_col,
536
                                     BUFFER_SET *ctx, BLOCK_SIZE bsize);
Jingning Han's avatar
Jingning Han committed
537

Yaowu Xu's avatar
Yaowu Xu committed
538
void av1_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col,
539
                                   BUFFER_SET *ctx, BLOCK_SIZE bsize);
540 541

#if CONFIG_SUPERTX
Yaowu Xu's avatar
Yaowu Xu committed
542
void av1_build_inter_predictors_sb_sub8x8_extend(MACROBLOCKD *xd,
543
#if CONFIG_EXT_INTER
Yaowu Xu's avatar
Yaowu Xu committed
544
                                                 int mi_row_ori, int mi_col_ori,
545
#endif  // CONFIG_EXT_INTER
Yaowu Xu's avatar
Yaowu Xu committed
546 547
                                                 int mi_row, int mi_col,
                                                 BLOCK_SIZE bsize, int block);
548

Yaowu Xu's avatar
Yaowu Xu committed
549
void av1_build_inter_predictors_sb_extend(MACROBLOCKD *xd,
550
#if CONFIG_EXT_INTER
Yaowu Xu's avatar
Yaowu Xu committed
551
                                          int mi_row_ori, int mi_col_ori,
552
#endif  // CONFIG_EXT_INTER
Yaowu Xu's avatar
Yaowu Xu committed
553 554
                                          int mi_row, int mi_col,
                                          BLOCK_SIZE bsize);
555
struct macroblockd_plane;
Yaowu Xu's avatar
Yaowu Xu committed
556
void av1_build_masked_inter_predictor_complex(
clang-format's avatar
clang-format committed
557 558 559 560
    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);
561
#endif  // CONFIG_SUPERTX
Jingning Han's avatar
Jingning Han committed
562

Yaowu Xu's avatar
Yaowu Xu committed
563
void av1_build_inter_predictor(const uint8_t *src, int src_stride, uint8_t *dst,
564
                               int dst_stride, const MV *src_mv,
Yaowu Xu's avatar
Yaowu Xu committed
565
                               const struct scale_factors *sf, int w, int h,
566
                               ConvolveParams *conv_params,
567
#if CONFIG_DUAL_FILTER
James Zern's avatar
James Zern committed
568
                               const InterpFilter *interp_filter,
569
#else
James Zern's avatar
James Zern committed
570
                               const InterpFilter interp_filter,
571
#endif
572 573 574 575
#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
576 577
                               enum mv_precision precision, int x, int y,
                               const MACROBLOCKD *xd);
Jingning Han's avatar
Jingning Han committed
578

579
#if CONFIG_HIGHBITDEPTH
580 581 582
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,
583
#if CONFIG_DUAL_FILTER
584
    const InterpFilter *interp_filter,
585
#else
586
    const InterpFilter interp_filter,
587
#endif
588 589 590 591 592
#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
593 594 595 596 597 598 599 600 601
#endif

static INLINE int scaled_buffer_offset(int x_offset, int y_offset, int stride,
                                       const struct scale_factors *sf) {
  const int x = sf ? sf->scale_value_x(x_offset, sf) : x_offset;
  const int y = sf ? sf->scale_value_y(y_offset, sf) : y_offset;
  return y * stride + x;
}

602 603 604
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
605 606
                                    const struct scale_factors *scale,
                                    int subsampling_x, int subsampling_y) {
607 608 609 610 611 612 613 614 615 616
#if CONFIG_CHROMA_SUB8X8
  if (bsize < BLOCK_8X8) {
    // Offset the buffer pointer
    if (subsampling_y && (mi_row & 0x01)) mi_row -= 1;
    if (subsampling_x && (mi_col & 0x01)) mi_col -= 1;
  }
#else
  (void)bsize;
#endif

Jingning Han's avatar
Jingning Han committed
617 618 619
  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);
620 621 622
  dst->buf0 = src;
  dst->width = width;
  dst->height = height;
Jingning Han's avatar
Jingning Han committed
623 624 625
  dst->stride = stride;
}

Yaowu Xu's avatar
Yaowu Xu committed
626
void av1_setup_dst_planes(struct macroblockd_plane planes[MAX_MB_PLANE],
627 628
                          BLOCK_SIZE bsize, const YV12_BUFFER_CONFIG *src,
                          int mi_row, int mi_col);
Jingning Han's avatar
Jingning Han committed
629

Yaowu Xu's avatar
Yaowu Xu committed
630 631 632
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);
633

634 635
// Detect if the block have sub-pixel level motion vectors
// per component.
636
#define CHECK_SUBPEL 0
637 638
static INLINE int has_subpel_mv_component(const MODE_INFO *const mi,
                                          const MACROBLOCKD *const xd,
639
                                          int dir) {
640
#if CHECK_SUBPEL
641
  const MB_MODE_INFO *const mbmi = &mi->mbmi;
642 643 644
  const BLOCK_SIZE bsize = mbmi->sb_type;
  int plane;
  int ref = (dir >> 1);
645 646 647 648 649
#if CONFIG_CB4X4
  const int unify_bsize = 1;
#else
  const int unify_bsize = 0;
#endif
650

651
  if (bsize >= BLOCK_8X8 || unify_bsize) {
652
    if (dir & 0x01) {
clang-format's avatar
clang-format committed
653
      if (mbmi->mv[ref].as_mv.col & SUBPEL_MASK) return 1;
654
    } else {
clang-format's avatar
clang-format committed
655
      if (mbmi->mv[ref].as_mv.row & SUBPEL_MASK) return 1;
656 657 658 659 660 661 662 663 664 665 666 667 668 669 670
    }
  } 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
671
            if (mv.col & SUBPEL_MASK) return 1;
672
          } else {
clang-format's avatar
clang-format committed
673
            if (mv.row & SUBPEL_MASK) return 1;
674 675 676 677 678 679 680
          }
        }
      }
    }
  }

  return 0;
681 682 683 684 685 686
#else
  (void)mi;
  (void)xd;
  (void)dir;
  return 1;
#endif
687 688
}

689 690 691 692 693 694 695 696 697 698 699 700 701 702
static INLINE void set_default_interp_filters(
    MB_MODE_INFO *const mbmi, InterpFilter frame_interp_filter) {
#if CONFIG_DUAL_FILTER
  int dir;
  for (dir = 0; dir < 4; ++dir)
    mbmi->interp_filter[dir] = frame_interp_filter == SWITCHABLE
                                   ? EIGHTTAP_REGULAR
                                   : frame_interp_filter;
#else
  mbmi->interp_filter = frame_interp_filter == SWITCHABLE ? EIGHTTAP_REGULAR
                                                          : frame_interp_filter;
#endif  // CONFIG_DUAL_FILTER
}

Yaowu Xu's avatar
Yaowu Xu committed
703
static INLINE int av1_is_interp_needed(const MACROBLOCKD *const xd) {
704 705 706 707 708 709 710 711 712 713 714 715
  (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) {
716
  MODE_INFO *const mi = xd->mi[0];
717 718 719 720 721 722 723 724
  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;
725 726 727
      }
    }
  }
728
  return 0;
729
}
730

Yue Chen's avatar
Yue Chen committed
731
#if CONFIG_MOTION_VAR
Yaowu Xu's avatar
Yaowu Xu committed
732
const uint8_t *av1_get_obmc_mask(int length);
733 734
void av1_count_overlappable_neighbors(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                      int mi_row, int mi_col);
735
void av1_build_obmc_inter_prediction(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
736 737 738 739 740
                                     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]);
741
void av1_build_prediction_by_above_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
742 743
                                         int mi_row, int mi_col,
                                         uint8_t *tmp_buf[MAX_MB_PLANE],
744 745
                                         int tmp_width[MAX_MB_PLANE],
                                         int tmp_height[MAX_MB_PLANE],
746
                                         int tmp_stride[MAX_MB_PLANE]);
747
void av1_build_prediction_by_left_preds(const AV1_COMMON *cm, MACROBLOCKD *xd,
Yaowu Xu's avatar
Yaowu Xu committed
748 749 750 751 752
                                        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]);
753 754
void av1_build_obmc_inter_predictors_sb(const AV1_COMMON *cm, MACROBLOCKD *xd,
                                        int mi_row, int mi_col);
755 756 757 758
#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
759
#endif  // CONFIG_MOTION_VAR
760

761
#if CONFIG_EXT_INTER
762 763
#define MASK_MASTER_SIZE ((MAX_WEDGE_SIZE) << 1)
#define MASK_MASTER_STRIDE (MASK_MASTER_SIZE)
764

Yaowu Xu's avatar
Yaowu Xu committed
765
void av1_init_wedge_masks();
766

Yaowu Xu's avatar
Yaowu Xu committed
767 768 769
static INLINE const uint8_t *av1_get_contiguous_soft_mask(int wedge_index,
                                                          int wedge_sign,
                                                          BLOCK_SIZE sb_type) {
770 771 772
  return wedge_params_lookup[sb_type].masks[wedge_sign][wedge_index];
}

Yaowu Xu's avatar
Yaowu Xu committed
773 774 775 776
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);

777 778 779 780 781 782 783 784 785
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);
786

Yaowu Xu's avatar
Yaowu Xu committed
787 788 789
void av1_build_interintra_predictors(MACROBLOCKD *xd, uint8_t *ypred,
                                     uint8_t *upred, uint8_t *vpred,
                                     int ystride, int ustride, int vstride,
790
                                     BUFFER_SET *ctx, BLOCK_SIZE bsize);
Yaowu Xu's avatar
Yaowu Xu committed
791
void av1_build_interintra_predictors_sby(MACROBLOCKD *xd, uint8_t *ypred,
792
                                         int ystride, BUFFER_SET *ctx,
Yaowu Xu's avatar
Yaowu Xu committed
793
                                         BLOCK_SIZE bsize);
794 795 796
void av1_build_interintra_predictors_sbc(MACROBLOCKD *xd, uint8_t *upred,
                                         int ustride, BUFFER_SET *ctx,
                                         int plane, BLOCK_SIZE bsize);
Yaowu Xu's avatar
Yaowu Xu committed
797 798
void av1_build_interintra_predictors_sbuv(MACROBLOCKD *xd, uint8_t *upred,
                                          uint8_t *vpred, int ustride,
799 800
                                          int vstride, BUFFER_SET *ctx,
                                          BLOCK_SIZE bsize);
Yaowu Xu's avatar
Yaowu Xu committed
801 802 803

void av1_build_intra_predictors_for_interintra(MACROBLOCKD *xd,
                                               BLOCK_SIZE bsize, int plane,
804
                                               BUFFER_SET *ctx,
Yaowu Xu's avatar
Yaowu Xu committed
805 806 807 808 809
                                               uint8_t *intra_pred,
                                               int intra_stride);
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);
810

811
// Encoder only
Yaowu Xu's avatar
Yaowu Xu committed
812
void av1_build_inter_predictors_for_planes_single_buf(
clang-format's avatar
clang-format committed
813 814
    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]);
815 816 817 818 819 820 821
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]);
822 823
#endif  // CONFIG_EXT_INTER

Jingning Han's avatar
Jingning Han committed
824 825 826 827
#ifdef __cplusplus
}  // extern "C"
#endif

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