/* * Copyright (c) 2010 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #ifndef VP9_COMMON_VP9_RECONINTER_H_ #define VP9_COMMON_VP9_RECONINTER_H_ #include "vpx/vpx_integer.h" #include "vp9/common/vp9_onyxc_int.h" struct subpix_fn_table; void vp9_build_inter_predictors_sby(MACROBLOCKD *xd, int mb_row, int mb_col, BLOCK_SIZE_TYPE bsize); void vp9_build_inter_predictors_sbuv(MACROBLOCKD *xd, int mb_row, int mb_col, BLOCK_SIZE_TYPE bsize); void vp9_build_inter_predictors_sb(MACROBLOCKD *mb, int mb_row, int mb_col, BLOCK_SIZE_TYPE bsize); void vp9_setup_interp_filters(MACROBLOCKD *xd, INTERPOLATIONFILTERTYPE filter, VP9_COMMON *cm); void vp9_setup_scale_factors_for_frame(struct scale_factors *scale, YV12_BUFFER_CONFIG *other, int this_w, int this_h); 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, const int_mv *mv_q4, const struct scale_factors *scale, int w, int h, int do_avg, const struct subpix_fn_table *subpix); static int scale_value_x_with_scaling(int val, const struct scale_factors *scale) { return val * scale->x_num / scale->x_den; } static int scale_value_y_with_scaling(int val, const struct scale_factors *scale) { return val * scale->y_num / scale->y_den; } static int unscaled_value(int val, const struct scale_factors *scale) { (void) scale; return val; } static int scaled_buffer_offset(int x_offset, int y_offset, int stride, const struct scale_factors *scale) { if (scale) return scale->scale_value_y(y_offset, scale) * stride + scale->scale_value_x(x_offset, scale); return y_offset * stride + x_offset; } static void setup_pred_plane(struct buf_2d *dst, uint8_t *src, int stride, int mi_row, int mi_col, const struct scale_factors *scale, int subsampling_x, int subsampling_y) { 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); 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, int mi_row, int mi_col) { setup_pred_plane(&xd->plane[0].dst, src->y_buffer, src->y_stride, mi_row, mi_col, NULL, xd->plane[0].subsampling_x, xd->plane[0].subsampling_y); setup_pred_plane(&xd->plane[1].dst, src->u_buffer, src->uv_stride, mi_row, mi_col, NULL, xd->plane[1].subsampling_x, xd->plane[1].subsampling_y); setup_pred_plane(&xd->plane[2].dst, src->v_buffer, src->uv_stride, mi_row, mi_col, NULL, xd->plane[2].subsampling_x, xd->plane[2].subsampling_y); } static void setup_pre_planes(MACROBLOCKD *xd, const YV12_BUFFER_CONFIG *src0, const YV12_BUFFER_CONFIG *src1, int mi_row, int mi_col, const struct scale_factors *scale, const struct scale_factors *scale_uv) { int i; for (i = 0; i < 2; i++) { const YV12_BUFFER_CONFIG *src = i ? src1 : src0; if (!src) continue; setup_pred_plane(&xd->plane[0].pre[i], src->y_buffer, src->y_stride, mi_row, mi_col, scale ? scale + i : NULL, xd->plane[0].subsampling_x, xd->plane[0].subsampling_y); setup_pred_plane(&xd->plane[1].pre[i], src->u_buffer, src->uv_stride, mi_row, mi_col, scale_uv ? scale_uv + i : NULL, xd->plane[1].subsampling_x, xd->plane[1].subsampling_y); setup_pred_plane(&xd->plane[2].pre[i], src->v_buffer, src->uv_stride, mi_row, mi_col, scale_uv ? scale_uv + i : NULL, xd->plane[2].subsampling_x, xd->plane[2].subsampling_y); } } static void setup_pred_block(YV12_BUFFER_CONFIG *dst, const YV12_BUFFER_CONFIG *src, int mi_row, int mi_col, const struct scale_factors *scale, const struct scale_factors *scale_uv) { const int recon_y_stride = src->y_stride; const int recon_uv_stride = src->uv_stride; int recon_yoffset; int recon_uvoffset; if (scale) { recon_yoffset = scaled_buffer_offset(MI_SIZE * mi_col, MI_SIZE * mi_row, recon_y_stride, scale); recon_uvoffset = scaled_buffer_offset(MI_UV_SIZE * mi_col, MI_UV_SIZE * mi_row, recon_uv_stride, scale_uv); } else { recon_yoffset = MI_SIZE * mi_row * recon_y_stride + MI_SIZE * mi_col; recon_uvoffset = MI_UV_SIZE * mi_row * recon_uv_stride + MI_UV_SIZE * mi_col; } *dst = *src; dst->y_buffer += recon_yoffset; dst->u_buffer += recon_uvoffset; dst->v_buffer += recon_uvoffset; } 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]; } static void set_offsets_with_scaling(struct scale_factors *scale, int row, int col) { const int x_q4 = 16 * col; const int y_q4 = 16 * row; scale->x_offset_q4 = (x_q4 * scale->x_num / scale->x_den) & 0xf; scale->y_offset_q4 = (y_q4 * scale->y_num / scale->y_den) & 0xf; } static void set_offsets_without_scaling(struct scale_factors *scale, int row, int col) { scale->x_offset_q4 = 0; scale->y_offset_q4 = 0; } static int_mv32 motion_vector_q3_to_q4_with_scaling( const int_mv *src_mv, const struct scale_factors *scale) { // returns mv * scale + offset int_mv32 result; const int32_t mv_row_q4 = src_mv->as_mv.row << 1; const int32_t mv_col_q4 = src_mv->as_mv.col << 1; /* TODO(jkoleszar): make fixed point, or as a second multiply? */ result.as_mv.row = mv_row_q4 * scale->y_num / scale->y_den + scale->y_offset_q4; result.as_mv.col = mv_col_q4 * scale->x_num / scale->x_den + scale->x_offset_q4; return result; } static int_mv32 motion_vector_q3_to_q4_without_scaling( const int_mv *src_mv, const struct scale_factors *scale) { // returns mv * scale + offset int_mv32 result; result.as_mv.row = src_mv->as_mv.row << 1; result.as_mv.col = src_mv->as_mv.col << 1; return result; } static int32_t motion_vector_component_q4_with_scaling(int mv_q4, int num, int den, int offset_q4) { // returns the scaled and offset value of the mv component. /* TODO(jkoleszar): make fixed point, or as a second multiply? */ return mv_q4 * num / den + offset_q4; } static int32_t motion_vector_component_q4_without_scaling(int mv_q4, int num, int den, int offset_q4) { // returns the scaled and offset value of the mv component. (void)num; (void)den; (void)offset_q4; return mv_q4; } #endif // VP9_COMMON_VP9_RECONINTER_H_