| /* |
| * 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. |
| */ |
| |
| |
| #include <limits.h> |
| #include "vpx_config.h" |
| #include "onyx_int.h" |
| #include "mr_dissim.h" |
| #include "vpx_mem/vpx_mem.h" |
| #include "rdopt.h" |
| |
| void vp8_cal_low_res_mb_cols(VP8_COMP *cpi) |
| { |
| int low_res_w; |
| |
| /* Support arbitrary down-sampling factor */ |
| unsigned int iw = cpi->oxcf.Width*cpi->oxcf.mr_down_sampling_factor.den |
| + cpi->oxcf.mr_down_sampling_factor.num - 1; |
| |
| low_res_w = iw/cpi->oxcf.mr_down_sampling_factor.num; |
| cpi->mr_low_res_mb_cols = ((low_res_w + 15) >> 4); |
| } |
| |
| #define GET_MV(x) \ |
| if(x->mbmi.ref_frame !=INTRA_FRAME) \ |
| { \ |
| mvx[cnt] = x->mbmi.mv.as_mv.row; \ |
| mvy[cnt] = x->mbmi.mv.as_mv.col; \ |
| cnt++; \ |
| } |
| |
| #define GET_MV_SIGN(x) \ |
| if(x->mbmi.ref_frame !=INTRA_FRAME) \ |
| { \ |
| mvx[cnt] = x->mbmi.mv.as_mv.row; \ |
| mvy[cnt] = x->mbmi.mv.as_mv.col; \ |
| if (cm->ref_frame_sign_bias[x->mbmi.ref_frame] \ |
| != cm->ref_frame_sign_bias[tmp->mbmi.ref_frame]) \ |
| { \ |
| mvx[cnt] *= -1; \ |
| mvy[cnt] *= -1; \ |
| } \ |
| cnt++; \ |
| } |
| |
| void vp8_cal_dissimilarity(VP8_COMP *cpi) |
| { |
| VP8_COMMON *cm = &cpi->common; |
| int i; |
| |
| /* Note: The first row & first column in mip are outside the frame, which |
| * were initialized to all 0.(ref_frame, mode, mv...) |
| * Their ref_frame = 0 means they won't be counted in the following |
| * calculation. |
| */ |
| if (cpi->oxcf.mr_total_resolutions >1 |
| && cpi->oxcf.mr_encoder_id < (cpi->oxcf.mr_total_resolutions - 1)) |
| { |
| /* Store info for show/no-show frames for supporting alt_ref. |
| * If parent frame is alt_ref, child has one too. |
| */ |
| LOWER_RES_FRAME_INFO* store_info |
| = (LOWER_RES_FRAME_INFO*)cpi->oxcf.mr_low_res_mode_info; |
| |
| store_info->frame_type = cm->frame_type; |
| |
| if(cm->frame_type != KEY_FRAME) |
| { |
| store_info->is_frame_dropped = 0; |
| for (i = 1; i < MAX_REF_FRAMES; i++) |
| store_info->low_res_ref_frames[i] = cpi->current_ref_frames[i]; |
| } |
| |
| if(cm->frame_type != KEY_FRAME) |
| { |
| int mb_row; |
| int mb_col; |
| /* Point to beginning of allocated MODE_INFO arrays. */ |
| MODE_INFO *tmp = cm->mip + cm->mode_info_stride; |
| LOWER_RES_MB_INFO* store_mode_info = store_info->mb_info; |
| |
| for (mb_row = 0; mb_row < cm->mb_rows; mb_row ++) |
| { |
| tmp++; |
| for (mb_col = 0; mb_col < cm->mb_cols; mb_col ++) |
| { |
| int dissim = INT_MAX; |
| |
| if(tmp->mbmi.ref_frame !=INTRA_FRAME) |
| { |
| int mvx[8]; |
| int mvy[8]; |
| int mmvx; |
| int mmvy; |
| int cnt=0; |
| const MODE_INFO *here = tmp; |
| const MODE_INFO *above = here - cm->mode_info_stride; |
| const MODE_INFO *left = here - 1; |
| const MODE_INFO *aboveleft = above - 1; |
| const MODE_INFO *aboveright = NULL; |
| const MODE_INFO *right = NULL; |
| const MODE_INFO *belowleft = NULL; |
| const MODE_INFO *below = NULL; |
| const MODE_INFO *belowright = NULL; |
| |
| /* If alternate reference frame is used, we have to |
| * check sign of MV. */ |
| if(cpi->oxcf.play_alternate) |
| { |
| /* Gather mv of neighboring MBs */ |
| GET_MV_SIGN(above) |
| GET_MV_SIGN(left) |
| GET_MV_SIGN(aboveleft) |
| |
| if(mb_col < (cm->mb_cols-1)) |
| { |
| right = here + 1; |
| aboveright = above + 1; |
| GET_MV_SIGN(right) |
| GET_MV_SIGN(aboveright) |
| } |
| |
| if(mb_row < (cm->mb_rows-1)) |
| { |
| below = here + cm->mode_info_stride; |
| belowleft = below - 1; |
| GET_MV_SIGN(below) |
| GET_MV_SIGN(belowleft) |
| } |
| |
| if(mb_col < (cm->mb_cols-1) |
| && mb_row < (cm->mb_rows-1)) |
| { |
| belowright = below + 1; |
| GET_MV_SIGN(belowright) |
| } |
| }else |
| { |
| /* No alt_ref and gather mv of neighboring MBs */ |
| GET_MV(above) |
| GET_MV(left) |
| GET_MV(aboveleft) |
| |
| if(mb_col < (cm->mb_cols-1)) |
| { |
| right = here + 1; |
| aboveright = above + 1; |
| GET_MV(right) |
| GET_MV(aboveright) |
| } |
| |
| if(mb_row < (cm->mb_rows-1)) |
| { |
| below = here + cm->mode_info_stride; |
| belowleft = below - 1; |
| GET_MV(below) |
| GET_MV(belowleft) |
| } |
| |
| if(mb_col < (cm->mb_cols-1) |
| && mb_row < (cm->mb_rows-1)) |
| { |
| belowright = below + 1; |
| GET_MV(belowright) |
| } |
| } |
| |
| if (cnt > 0) |
| { |
| int max_mvx = mvx[0]; |
| int min_mvx = mvx[0]; |
| int max_mvy = mvy[0]; |
| int min_mvy = mvy[0]; |
| int i; |
| |
| if (cnt > 1) |
| { |
| for (i=1; i< cnt; i++) |
| { |
| if (mvx[i] > max_mvx) max_mvx = mvx[i]; |
| else if (mvx[i] < min_mvx) min_mvx = mvx[i]; |
| if (mvy[i] > max_mvy) max_mvy = mvy[i]; |
| else if (mvy[i] < min_mvy) min_mvy = mvy[i]; |
| } |
| } |
| |
| mmvx = MAX(abs(min_mvx - here->mbmi.mv.as_mv.row), |
| abs(max_mvx - here->mbmi.mv.as_mv.row)); |
| mmvy = MAX(abs(min_mvy - here->mbmi.mv.as_mv.col), |
| abs(max_mvy - here->mbmi.mv.as_mv.col)); |
| dissim = MAX(mmvx, mmvy); |
| } |
| } |
| |
| /* Store mode info for next resolution encoding */ |
| store_mode_info->mode = tmp->mbmi.mode; |
| store_mode_info->ref_frame = tmp->mbmi.ref_frame; |
| store_mode_info->mv.as_int = tmp->mbmi.mv.as_int; |
| store_mode_info->dissim = dissim; |
| tmp++; |
| store_mode_info++; |
| } |
| } |
| } |
| } |
| } |
| |
| /* This function is called only when this frame is dropped at current |
| resolution level. */ |
| void vp8_store_drop_frame_info(VP8_COMP *cpi) |
| { |
| /* If the frame is dropped in lower-resolution encoding, this information |
| is passed to higher resolution level so that the encoder knows there |
| is no mode & motion info available. |
| */ |
| if (cpi->oxcf.mr_total_resolutions >1 |
| && cpi->oxcf.mr_encoder_id < (cpi->oxcf.mr_total_resolutions - 1)) |
| { |
| /* Store info for show/no-show frames for supporting alt_ref. |
| * If parent frame is alt_ref, child has one too. |
| */ |
| LOWER_RES_FRAME_INFO* store_info |
| = (LOWER_RES_FRAME_INFO*)cpi->oxcf.mr_low_res_mode_info; |
| |
| /* Set frame_type to be INTER_FRAME since we won't drop key frame. */ |
| store_info->frame_type = INTER_FRAME; |
| store_info->is_frame_dropped = 1; |
| } |
| } |
