|
1 /* |
|
2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
|
3 * |
|
4 * Use of this source code is governed by a BSD-style license |
|
5 * that can be found in the LICENSE file in the root of the source |
|
6 * 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. |
|
9 */ |
|
10 |
|
11 #ifndef VP9_COMMON_VP9_RECONINTER_H_ |
|
12 #define VP9_COMMON_VP9_RECONINTER_H_ |
|
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, int mi_row, int mi_col, |
|
19 BLOCK_SIZE bsize); |
|
20 |
|
21 void vp9_build_inter_predictors_sbuv(MACROBLOCKD *xd, int mi_row, int mi_col, |
|
22 BLOCK_SIZE bsize); |
|
23 |
|
24 void vp9_build_inter_predictors_sb(MACROBLOCKD *xd, int mi_row, int mi_col, |
|
25 BLOCK_SIZE bsize); |
|
26 |
|
27 void vp9_setup_interp_filters(MACROBLOCKD *xd, |
|
28 INTERPOLATION_TYPE filter, |
|
29 VP9_COMMON *cm); |
|
30 |
|
31 void vp9_build_inter_predictor(const uint8_t *src, int src_stride, |
|
32 uint8_t *dst, int dst_stride, |
|
33 const MV *mv_q3, |
|
34 const struct scale_factors *scale, |
|
35 int w, int h, int do_avg, |
|
36 const struct subpix_fn_table *subpix, |
|
37 enum mv_precision precision); |
|
38 |
|
39 static int scaled_buffer_offset(int x_offset, int y_offset, int stride, |
|
40 const struct scale_factors *scale) { |
|
41 const int x = scale ? scale->sfc->scale_value_x(x_offset, scale->sfc) : |
|
42 x_offset; |
|
43 const int y = scale ? scale->sfc->scale_value_y(y_offset, scale->sfc) : |
|
44 y_offset; |
|
45 return y * stride + x; |
|
46 } |
|
47 |
|
48 static void setup_pred_plane(struct buf_2d *dst, |
|
49 uint8_t *src, int stride, |
|
50 int mi_row, int mi_col, |
|
51 const struct scale_factors *scale, |
|
52 int subsampling_x, int subsampling_y) { |
|
53 const int x = (MI_SIZE * mi_col) >> subsampling_x; |
|
54 const int y = (MI_SIZE * mi_row) >> subsampling_y; |
|
55 dst->buf = src + scaled_buffer_offset(x, y, stride, scale); |
|
56 dst->stride = stride; |
|
57 } |
|
58 |
|
59 // TODO(jkoleszar): audit all uses of this that don't set mb_row, mb_col |
|
60 static void setup_dst_planes(MACROBLOCKD *xd, |
|
61 const YV12_BUFFER_CONFIG *src, |
|
62 int mi_row, int mi_col) { |
|
63 uint8_t *buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer, |
|
64 src->alpha_buffer}; |
|
65 int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride, |
|
66 src->alpha_stride}; |
|
67 int i; |
|
68 |
|
69 for (i = 0; i < MAX_MB_PLANE; ++i) { |
|
70 struct macroblockd_plane *pd = &xd->plane[i]; |
|
71 setup_pred_plane(&pd->dst, buffers[i], strides[i], mi_row, mi_col, NULL, |
|
72 pd->subsampling_x, pd->subsampling_y); |
|
73 } |
|
74 } |
|
75 |
|
76 static void setup_pre_planes(MACROBLOCKD *xd, int i, |
|
77 const YV12_BUFFER_CONFIG *src, |
|
78 int mi_row, int mi_col, |
|
79 const struct scale_factors *sf) { |
|
80 if (src) { |
|
81 int j; |
|
82 uint8_t* buffers[4] = {src->y_buffer, src->u_buffer, src->v_buffer, |
|
83 src->alpha_buffer}; |
|
84 int strides[4] = {src->y_stride, src->uv_stride, src->uv_stride, |
|
85 src->alpha_stride}; |
|
86 |
|
87 for (j = 0; j < MAX_MB_PLANE; ++j) { |
|
88 struct macroblockd_plane *pd = &xd->plane[j]; |
|
89 setup_pred_plane(&pd->pre[i], buffers[j], strides[j], |
|
90 mi_row, mi_col, sf, pd->subsampling_x, pd->subsampling_y); |
|
91 } |
|
92 } |
|
93 } |
|
94 |
|
95 static void set_scale_factors(MACROBLOCKD *xd, int ref0, int ref1, |
|
96 struct scale_factors sf[MAX_REF_FRAMES]) { |
|
97 xd->scale_factor[0] = sf[ref0 >= 0 ? ref0 : 0]; |
|
98 xd->scale_factor[1] = sf[ref1 >= 0 ? ref1 : 0]; |
|
99 } |
|
100 |
|
101 void vp9_setup_scale_factors(VP9_COMMON *cm, int i); |
|
102 |
|
103 #endif // VP9_COMMON_VP9_RECONINTER_H_ |