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 #include "vpx_mem/vpx_mem.h" 12 13 #include "vp9/common/vp9_common.h" 14 #include "vp9/common/vp9_extend.h" 15 16 static void copy_and_extend_plane(const uint8_t *src, int src_pitch, 17 uint8_t *dst, int dst_pitch, 18 int w, int h, 19 int extend_top, int extend_left, 20 int extend_bottom, int extend_right) { 21 int i, linesize; 22 23 // copy the left and right most columns out 24 const uint8_t *src_ptr1 = src; 25 const uint8_t *src_ptr2 = src + w - 1; 26 uint8_t *dst_ptr1 = dst - extend_left; 27 uint8_t *dst_ptr2 = dst + w; 28 29 for (i = 0; i < h; i++) { 30 vpx_memset(dst_ptr1, src_ptr1[0], extend_left); 31 vpx_memcpy(dst_ptr1 + extend_left, src_ptr1, w); 32 vpx_memset(dst_ptr2, src_ptr2[0], extend_right); 33 src_ptr1 += src_pitch; 34 src_ptr2 += src_pitch; 35 dst_ptr1 += dst_pitch; 36 dst_ptr2 += dst_pitch; 37 } 38 39 // Now copy the top and bottom lines into each line of the respective 40 // borders 41 src_ptr1 = dst - extend_left; 42 src_ptr2 = dst + dst_pitch * (h - 1) - extend_left; 43 dst_ptr1 = dst + dst_pitch * (-extend_top) - extend_left; 44 dst_ptr2 = dst + dst_pitch * (h) - extend_left; 45 linesize = extend_left + extend_right + w; 46 47 for (i = 0; i < extend_top; i++) { 48 vpx_memcpy(dst_ptr1, src_ptr1, linesize); 49 dst_ptr1 += dst_pitch; 50 } 51 52 for (i = 0; i < extend_bottom; i++) { 53 vpx_memcpy(dst_ptr2, src_ptr2, linesize); 54 dst_ptr2 += dst_pitch; 55 } 56 } 57 58 void vp9_copy_and_extend_frame(const YV12_BUFFER_CONFIG *src, 59 YV12_BUFFER_CONFIG *dst) { 60 // Extend src frame in buffer 61 // Altref filtering assumes 16 pixel extension 62 const int et_y = 16; 63 const int el_y = 16; 64 // Motion estimation may use src block variance with the block size up 65 // to 64x64, so the right and bottom need to be extended to 64 multiple 66 // or up to 16, whichever is greater. 67 const int eb_y = MAX(ALIGN_POWER_OF_TWO(src->y_width, 6) - src->y_width, 68 16); 69 const int er_y = MAX(ALIGN_POWER_OF_TWO(src->y_height, 6) - src->y_height, 70 16); 71 const int uv_width_subsampling = (src->uv_width != src->y_width); 72 const int uv_height_subsampling = (src->uv_height != src->y_height); 73 const int et_uv = et_y >> uv_height_subsampling; 74 const int el_uv = el_y >> uv_width_subsampling; 75 const int eb_uv = eb_y >> uv_height_subsampling; 76 const int er_uv = er_y >> uv_width_subsampling; 77 78 #if CONFIG_ALPHA 79 const int et_a = dst->border >> (dst->alpha_height != dst->y_height); 80 const int el_a = dst->border >> (dst->alpha_width != dst->y_width); 81 const int eb_a = et_a + dst->alpha_height - src->alpha_height; 82 const int er_a = el_a + dst->alpha_width - src->alpha_width; 83 84 copy_and_extend_plane(src->alpha_buffer, src->alpha_stride, 85 dst->alpha_buffer, dst->alpha_stride, 86 src->alpha_width, src->alpha_height, 87 et_a, el_a, eb_a, er_a); 88 #endif 89 90 copy_and_extend_plane(src->y_buffer, src->y_stride, 91 dst->y_buffer, dst->y_stride, 92 src->y_width, src->y_height, 93 et_y, el_y, eb_y, er_y); 94 95 copy_and_extend_plane(src->u_buffer, src->uv_stride, 96 dst->u_buffer, dst->uv_stride, 97 src->uv_width, src->uv_height, 98 et_uv, el_uv, eb_uv, er_uv); 99 100 copy_and_extend_plane(src->v_buffer, src->uv_stride, 101 dst->v_buffer, dst->uv_stride, 102 src->uv_width, src->uv_height, 103 et_uv, el_uv, eb_uv, er_uv); 104 } 105 106 void vp9_copy_and_extend_frame_with_rect(const YV12_BUFFER_CONFIG *src, 107 YV12_BUFFER_CONFIG *dst, 108 int srcy, int srcx, 109 int srch, int srcw) { 110 // If the side is not touching the bounder then don't extend. 111 const int et_y = srcy ? 0 : dst->border; 112 const int el_y = srcx ? 0 : dst->border; 113 const int eb_y = srcy + srch != src->y_height ? 0 : 114 dst->border + dst->y_height - src->y_height; 115 const int er_y = srcx + srcw != src->y_width ? 0 : 116 dst->border + dst->y_width - src->y_width; 117 const int src_y_offset = srcy * src->y_stride + srcx; 118 const int dst_y_offset = srcy * dst->y_stride + srcx; 119 120 const int et_uv = ROUND_POWER_OF_TWO(et_y, 1); 121 const int el_uv = ROUND_POWER_OF_TWO(el_y, 1); 122 const int eb_uv = ROUND_POWER_OF_TWO(eb_y, 1); 123 const int er_uv = ROUND_POWER_OF_TWO(er_y, 1); 124 const int src_uv_offset = ((srcy * src->uv_stride) >> 1) + (srcx >> 1); 125 const int dst_uv_offset = ((srcy * dst->uv_stride) >> 1) + (srcx >> 1); 126 const int srch_uv = ROUND_POWER_OF_TWO(srch, 1); 127 const int srcw_uv = ROUND_POWER_OF_TWO(srcw, 1); 128 129 copy_and_extend_plane(src->y_buffer + src_y_offset, src->y_stride, 130 dst->y_buffer + dst_y_offset, dst->y_stride, 131 srcw, srch, 132 et_y, el_y, eb_y, er_y); 133 134 copy_and_extend_plane(src->u_buffer + src_uv_offset, src->uv_stride, 135 dst->u_buffer + dst_uv_offset, dst->uv_stride, 136 srcw_uv, srch_uv, 137 et_uv, el_uv, eb_uv, er_uv); 138 139 copy_and_extend_plane(src->v_buffer + src_uv_offset, src->uv_stride, 140 dst->v_buffer + dst_uv_offset, dst->uv_stride, 141 srcw_uv, srch_uv, 142 et_uv, el_uv, eb_uv, er_uv); 143 } 144
