diff options
Diffstat (limited to 'vp9/common/vp9_findnearmv.h')
-rw-r--r-- | vp9/common/vp9_findnearmv.h | 183 |
1 files changed, 183 insertions, 0 deletions
diff --git a/vp9/common/vp9_findnearmv.h b/vp9/common/vp9_findnearmv.h new file mode 100644 index 000000000..759bdbaff --- /dev/null +++ b/vp9/common/vp9_findnearmv.h @@ -0,0 +1,183 @@ +/* + * 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 __INC_FINDNEARMV_H +#define __INC_FINDNEARMV_H + +#include "vp9_mv.h" +#include "vp9_blockd.h" +#include "vp9_treecoder.h" +#include "vp9_onyxc_int.h" + +/* check a list of motion vectors by sad score using a number rows of pixels + * above and a number cols of pixels in the left to select the one with best + * score to use as ref motion vector + */ +void vp9_find_best_ref_mvs(MACROBLOCKD *xd, + unsigned char *ref_y_buffer, + int ref_y_stride, + int_mv *mvlist, + int_mv *best_mv, + int_mv *nearest, + int_mv *near); + +static void mv_bias(int refmb_ref_frame_sign_bias, int refframe, int_mv *mvp, const int *ref_frame_sign_bias) { + MV xmv; + xmv = mvp->as_mv; + + if (refmb_ref_frame_sign_bias != ref_frame_sign_bias[refframe]) { + xmv.row *= -1; + xmv.col *= -1; + } + + mvp->as_mv = xmv; +} + +#define LEFT_TOP_MARGIN (16 << 3) +#define RIGHT_BOTTOM_MARGIN (16 << 3) + +static void clamp_mv(int_mv *mv, + int mb_to_left_edge, + int mb_to_right_edge, + int mb_to_top_edge, + int mb_to_bottom_edge) { + mv->as_mv.col = (mv->as_mv.col < mb_to_left_edge) ? + mb_to_left_edge : mv->as_mv.col; + mv->as_mv.col = (mv->as_mv.col > mb_to_right_edge) ? + mb_to_right_edge : mv->as_mv.col; + mv->as_mv.row = (mv->as_mv.row < mb_to_top_edge) ? + mb_to_top_edge : mv->as_mv.row; + mv->as_mv.row = (mv->as_mv.row > mb_to_bottom_edge) ? + mb_to_bottom_edge : mv->as_mv.row; +} + +static void clamp_mv2(int_mv *mv, const MACROBLOCKD *xd) { + clamp_mv(mv, + xd->mb_to_left_edge - LEFT_TOP_MARGIN, + xd->mb_to_right_edge + RIGHT_BOTTOM_MARGIN, + xd->mb_to_top_edge - LEFT_TOP_MARGIN, + xd->mb_to_bottom_edge + RIGHT_BOTTOM_MARGIN); +} + +static unsigned int check_mv_bounds(int_mv *mv, + int mb_to_left_edge, + int mb_to_right_edge, + int mb_to_top_edge, + int mb_to_bottom_edge) { + return (mv->as_mv.col < mb_to_left_edge) || + (mv->as_mv.col > mb_to_right_edge) || + (mv->as_mv.row < mb_to_top_edge) || + (mv->as_mv.row > mb_to_bottom_edge); +} + +vp9_prob *vp9_mv_ref_probs(VP9_COMMON *pc, + vp9_prob p[VP9_MVREFS - 1], + const int context); + +extern const unsigned char vp9_mbsplit_offset[4][16]; + +static int left_block_mv(const MODE_INFO *cur_mb, int b) { + if (!(b & 3)) { + /* On L edge, get from MB to left of us */ + --cur_mb; + + if (cur_mb->mbmi.mode != SPLITMV) + return cur_mb->mbmi.mv[0].as_int; + b += 4; + } + + return (cur_mb->bmi + b - 1)->as_mv.first.as_int; +} + +static int left_block_second_mv(const MODE_INFO *cur_mb, int b) { + if (!(b & 3)) { + /* On L edge, get from MB to left of us */ + --cur_mb; + + if (cur_mb->mbmi.mode != SPLITMV) + return cur_mb->mbmi.second_ref_frame > 0 ? + cur_mb->mbmi.mv[1].as_int : cur_mb->mbmi.mv[0].as_int; + b += 4; + } + + return cur_mb->mbmi.second_ref_frame > 0 ? + (cur_mb->bmi + b - 1)->as_mv.second.as_int : + (cur_mb->bmi + b - 1)->as_mv.first.as_int; +} + +static int above_block_mv(const MODE_INFO *cur_mb, int b, int mi_stride) { + if (!(b >> 2)) { + /* On top edge, get from MB above us */ + cur_mb -= mi_stride; + + if (cur_mb->mbmi.mode != SPLITMV) + return cur_mb->mbmi.mv[0].as_int; + b += 16; + } + + return (cur_mb->bmi + b - 4)->as_mv.first.as_int; +} + +static int above_block_second_mv(const MODE_INFO *cur_mb, int b, int mi_stride) { + if (!(b >> 2)) { + /* On top edge, get from MB above us */ + cur_mb -= mi_stride; + + if (cur_mb->mbmi.mode != SPLITMV) + return cur_mb->mbmi.second_ref_frame > 0 ? + cur_mb->mbmi.mv[1].as_int : cur_mb->mbmi.mv[0].as_int; + b += 16; + } + + return cur_mb->mbmi.second_ref_frame > 0 ? + (cur_mb->bmi + b - 4)->as_mv.second.as_int : + (cur_mb->bmi + b - 4)->as_mv.first.as_int; +} + +static B_PREDICTION_MODE left_block_mode(const MODE_INFO *cur_mb, int b) { + if (!(b & 3)) { + /* On L edge, get from MB to left of us */ + --cur_mb; + + if (cur_mb->mbmi.mode < I8X8_PRED) { + return pred_mode_conv(cur_mb->mbmi.mode); + } else if (cur_mb->mbmi.mode == I8X8_PRED) { + return pred_mode_conv((cur_mb->bmi + 3 + b)->as_mode.first); + } else if (cur_mb->mbmi.mode == B_PRED) { + return ((cur_mb->bmi + 3 + b)->as_mode.first); + } else { + return B_DC_PRED; + } + } + return (cur_mb->bmi + b - 1)->as_mode.first; +} + +static B_PREDICTION_MODE above_block_mode(const MODE_INFO *cur_mb, + int b, int mi_stride) { + if (!(b >> 2)) { + /* On top edge, get from MB above us */ + cur_mb -= mi_stride; + + if (cur_mb->mbmi.mode < I8X8_PRED) { + return pred_mode_conv(cur_mb->mbmi.mode); + } else if (cur_mb->mbmi.mode == I8X8_PRED) { + return pred_mode_conv((cur_mb->bmi + 12 + b)->as_mode.first); + } else if (cur_mb->mbmi.mode == B_PRED) { + return ((cur_mb->bmi + 12 + b)->as_mode.first); + } else { + return B_DC_PRED; + } + } + + return (cur_mb->bmi + b - 4)->as_mode.first; +} + +#endif |