vp9_reconinter.h 4.06 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
7
 *  in the file PATENTS.  All contributing project authors may
8
 *  be found in the AUTHORS file in the root of the source tree.
John Koleszar's avatar
John Koleszar committed
9 10
 */

11 12
#ifndef VP9_COMMON_VP9_RECONINTER_H_
#define VP9_COMMON_VP9_RECONINTER_H_
John Koleszar's avatar
John Koleszar committed
13

14
#include "vpx/vpx_integer.h"
15
#include "vp9/common/vp9_onyxc_int.h"
16

17
struct subpix_fn_table;
18 19
void vp9_build_inter_predictors_sby(MACROBLOCKD *xd, int mi_row, int mi_col,
                                    BLOCK_SIZE bsize);
20

21 22
void vp9_build_inter_predictors_sbuv(MACROBLOCKD *xd, int mi_row, int mi_col,
                                     BLOCK_SIZE bsize);
23

24 25
void vp9_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col,
                                   BLOCK_SIZE bsize);
26 27 28

void vp9_dec_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col,
                                       BLOCK_SIZE bsize);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
29

30 31
void vp9_build_inter_predictor(const uint8_t *src, int src_stride,
                               uint8_t *dst, int dst_stride,
32
                               const MV *mv_q3,
33
                               const struct scale_factors *sf,
34
                               int w, int h, int do_avg,
35
                               const struct subpix_fn_table *subpix,
36 37
                               enum mv_precision precision,
                               int x, int y);
38

39
static int scaled_buffer_offset(int x_offset, int y_offset, int stride,
40 41 42
                                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;
43
  return y * stride + x;
44 45 46 47
}

static void setup_pred_plane(struct buf_2d *dst,
                             uint8_t *src, int stride,
48
                             int mi_row, int mi_col,
49 50
                             const struct scale_factors *scale,
                             int subsampling_x, int subsampling_y) {
51 52
  const int x = (MI_SIZE * mi_col) >> subsampling_x;
  const int y = (MI_SIZE * mi_row) >> subsampling_y;
53 54 55 56 57 58 59
  dst->buf = src + scaled_buffer_offset(x, y, stride, scale);
  dst->stride = stride;
}

// TODO(jkoleszar): audit all uses of this that don't set mb_row, mb_col
static void setup_dst_planes(MACROBLOCKD *xd,
                             const YV12_BUFFER_CONFIG *src,
60
                             int mi_row, int mi_col) {
61 62 63 64
  uint8_t *const buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
                               src->alpha_buffer};
  const int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
                          src->alpha_stride};
65 66 67
  int i;

  for (i = 0; i < MAX_MB_PLANE; ++i) {
68
    struct macroblockd_plane *const pd = &xd->plane[i];
69 70 71
    setup_pred_plane(&pd->dst, buffers[i], strides[i], mi_row, mi_col, NULL,
                     pd->subsampling_x, pd->subsampling_y);
  }
72 73
}

74
static void setup_pre_planes(MACROBLOCKD *xd, int idx,
75
                             const YV12_BUFFER_CONFIG *src,
76
                             int mi_row, int mi_col,
77
                             const struct scale_factors *sf) {
78 79 80 81 82 83
  if (src != NULL) {
    int i;
    uint8_t *const buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
                                 src->alpha_buffer};
    const int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
                            src->alpha_stride};
84

85 86 87 88
    for (i = 0; i < MAX_MB_PLANE; ++i) {
      struct macroblockd_plane *const pd = &xd->plane[i];
      setup_pred_plane(&pd->pre[idx], buffers[i], strides[i], mi_row, mi_col,
                       sf, pd->subsampling_x, pd->subsampling_y);
89
    }
90 91 92
  }
}

93 94
static void set_scale_factors(VP9_COMMON *cm, MACROBLOCKD *xd,
                              int ref0, int ref1) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
95 96
  xd->block_refs[0] = &cm->frame_refs[ref0 >= 0 ? ref0 : 0];
  xd->block_refs[1] = &cm->frame_refs[ref1 >= 0 ? ref1 : 0];
97 98
}

99
#endif  // VP9_COMMON_VP9_RECONINTER_H_