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 12 #ifndef VP9_COMMON_VP9_FINDNEARMV_H_ 13 #define VP9_COMMON_VP9_FINDNEARMV_H_ 14 15 #include "vp9/common/vp9_mv.h" 16 #include "vp9/common/vp9_blockd.h" 17 #include "vp9/common/vp9_treecoder.h" 18 #include "vp9/common/vp9_onyxc_int.h" 19 20 #define LEFT_TOP_MARGIN ((VP9BORDERINPIXELS - VP9_INTERP_EXTEND) << 3) 21 #define RIGHT_BOTTOM_MARGIN ((VP9BORDERINPIXELS - VP9_INTERP_EXTEND) << 3) 22 23 // check a list of motion vectors by sad score using a number rows of pixels 24 // above and a number cols of pixels in the left to select the one with best 25 // score to use as ref motion vector 26 void vp9_find_best_ref_mvs(MACROBLOCKD *xd, int allow_hp, 27 int_mv *mvlist, int_mv *nearest, int_mv *near); 28 29 // TODO(jingning): this mv clamping function should be block size dependent. 30 static void clamp_mv2(MV *mv, const MACROBLOCKD *xd) { 31 clamp_mv(mv, xd->mb_to_left_edge - LEFT_TOP_MARGIN, 32 xd->mb_to_right_edge + RIGHT_BOTTOM_MARGIN, 33 xd->mb_to_top_edge - LEFT_TOP_MARGIN, 34 xd->mb_to_bottom_edge + RIGHT_BOTTOM_MARGIN); 35 } 36 37 void vp9_append_sub8x8_mvs_for_idx(VP9_COMMON *cm, MACROBLOCKD *xd, 38 const TileInfo *const tile, 39 int_mv *dst_nearest, 40 int_mv *dst_near, 41 int block_idx, int ref_idx, 42 int mi_row, int mi_col); 43 44 static MB_PREDICTION_MODE left_block_mode(const MODE_INFO *cur_mi, 45 const MODE_INFO *left_mi, int b) { 46 if (b == 0 || b == 2) { 47 if (!left_mi || is_inter_block(&left_mi->mbmi)) 48 return DC_PRED; 49 50 return left_mi->mbmi.sb_type < BLOCK_8X8 ? left_mi->bmi[b + 1].as_mode 51 : left_mi->mbmi.mode; 52 } else { 53 assert(b == 1 || b == 3); 54 return cur_mi->bmi[b - 1].as_mode; 55 } 56 } 57 58 static MB_PREDICTION_MODE above_block_mode(const MODE_INFO *cur_mi, 59 const MODE_INFO *above_mi, int b) { 60 if (b == 0 || b == 1) { 61 if (!above_mi || is_inter_block(&above_mi->mbmi)) 62 return DC_PRED; 63 64 return above_mi->mbmi.sb_type < BLOCK_8X8 ? above_mi->bmi[b + 2].as_mode 65 : above_mi->mbmi.mode; 66 } else { 67 assert(b == 2 || b == 3); 68 return cur_mi->bmi[b - 2].as_mode; 69 } 70 } 71 72 #endif // VP9_COMMON_VP9_FINDNEARMV_H_ 73