vp9_reconinter.h 5.25 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
void vp9_build_inter_predictors_sby(MACROBLOCKD *xd,
19 20 21 22
                                    int mb_row,
                                    int mb_col,
                                    BLOCK_SIZE_TYPE bsize);

23
void vp9_build_inter_predictors_sbuv(MACROBLOCKD *xd,
24 25 26
                                     int mb_row,
                                     int mb_col,
                                     BLOCK_SIZE_TYPE bsize);
27

28 29 30
void vp9_build_inter_predictors_sb(MACROBLOCKD *mb,
                                   int mb_row, int mb_col,
                                   BLOCK_SIZE_TYPE bsize);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
31

32 33 34
void vp9_setup_interp_filters(MACROBLOCKD *xd,
                              INTERPOLATIONFILTERTYPE filter,
                              VP9_COMMON *cm);
John Koleszar's avatar
John Koleszar committed
35

36
void vp9_setup_scale_factors_for_frame(struct scale_factors *scale,
37
                                       int other_w, int other_h,
38 39
                                       int this_w, int this_h);

40 41 42 43 44 45 46 47 48
void vp9_build_inter_predictor(const uint8_t *src, int src_stride,
                               uint8_t *dst, int dst_stride,
                               const int_mv *mv_q3,
                               const struct scale_factors *scale,
                               int w, int h, int do_avg,
                               const struct subpix_fn_table *subpix);

void vp9_build_inter_predictor_q4(const uint8_t *src, int src_stride,
                                  uint8_t *dst, int dst_stride,
49
                                  const int_mv *mv_q4,
50 51 52
                                  const struct scale_factors *scale,
                                  int w, int h, int do_avg,
                                  const struct subpix_fn_table *subpix);
53

54
static int scaled_buffer_offset(int x_offset, int y_offset, int stride,
55
                                const struct scale_factors *scale) {
56 57 58
  const int x = scale ? scale->scale_value_x(x_offset, scale) : x_offset;
  const int y = scale ? scale->scale_value_y(y_offset, scale) : y_offset;
  return y * stride + x;
59 60 61 62
}

static void setup_pred_plane(struct buf_2d *dst,
                             uint8_t *src, int stride,
63
                             int mi_row, int mi_col,
64 65
                             const struct scale_factors *scale,
                             int subsampling_x, int subsampling_y) {
66 67
  const int x = (MI_SIZE * mi_col) >> subsampling_x;
  const int y = (MI_SIZE * mi_row) >> subsampling_y;
68 69 70 71 72 73 74
  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,
75
                             int mi_row, int mi_col) {
76 77 78 79
  uint8_t *buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
                         src->alpha_buffer};
  int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
                    src->alpha_stride};
80 81 82 83 84 85 86
  int i;

  for (i = 0; i < MAX_MB_PLANE; ++i) {
    struct macroblockd_plane *pd = &xd->plane[i];
    setup_pred_plane(&pd->dst, buffers[i], strides[i], mi_row, mi_col, NULL,
                     pd->subsampling_x, pd->subsampling_y);
  }
87 88
}

89 90 91
static void setup_pre_planes(MACROBLOCKD *xd,
                             const YV12_BUFFER_CONFIG *src0,
                             const YV12_BUFFER_CONFIG *src1,
92
                             int mi_row, int mi_col,
93 94
                             const struct scale_factors *scale,
                             const struct scale_factors *scale_uv) {
95 96 97 98 99 100
  const YV12_BUFFER_CONFIG *srcs[2] = {src0, src1};
  int i, j;

  for (i = 0; i < 2; ++i) {
    const YV12_BUFFER_CONFIG *src = srcs[i];
    if (src) {
101 102 103 104
      uint8_t* buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer,
                             src->alpha_buffer};
      int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride,
                        src->alpha_stride};
105 106 107 108 109 110 111 112 113 114

      for (j = 0; j < MAX_MB_PLANE; ++j) {
        struct macroblockd_plane *pd = &xd->plane[j];
        const struct scale_factors *sf = j ? scale_uv : scale;
        setup_pred_plane(&pd->pre[i],
                         buffers[j], strides[j],
                         mi_row, mi_col, sf ? &sf[i] : NULL,
                         pd->subsampling_x, pd->subsampling_y);
      }
    }
115 116 117
  }
}

118 119 120 121 122 123 124 125 126 127
static void set_scale_factors(MACROBLOCKD *xd,
    int ref0, int ref1,
    struct scale_factors scale_factor[MAX_REF_FRAMES]) {

  xd->scale_factor[0] = scale_factor[ref0 >= 0 ? ref0 : 0];
  xd->scale_factor[1] = scale_factor[ref1 >= 0 ? ref1 : 0];
  xd->scale_factor_uv[0] = xd->scale_factor[0];
  xd->scale_factor_uv[1] = xd->scale_factor[1];
}

128 129
void vp9_setup_scale_factors(VP9_COMMON *cm, int i);

130
#endif  // VP9_COMMON_VP9_RECONINTER_H_