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 <assert.h> 12#include "./vpx_config.h" 13#include "vpx/vpx_integer.h" 14#include "vpx_mem/vpx_mem.h" 15#include "vpx_scale/yv12config.h" 16 17static void extend_plane(uint8_t *const src, int src_stride, 18 int width, int height, 19 int extend_top, int extend_left, 20 int extend_bottom, int extend_right) { 21 int i; 22 const int linesize = extend_left + extend_right + width; 23 24 /* copy the left and right most columns out */ 25 uint8_t *src_ptr1 = src; 26 uint8_t *src_ptr2 = src + width - 1; 27 uint8_t *dst_ptr1 = src - extend_left; 28 uint8_t *dst_ptr2 = src + width; 29 30 for (i = 0; i < height; ++i) { 31 vpx_memset(dst_ptr1, src_ptr1[0], extend_left); 32 vpx_memset(dst_ptr2, src_ptr2[0], extend_right); 33 src_ptr1 += src_stride; 34 src_ptr2 += src_stride; 35 dst_ptr1 += src_stride; 36 dst_ptr2 += src_stride; 37 } 38 39 /* Now copy the top and bottom lines into each line of the respective 40 * borders 41 */ 42 src_ptr1 = src - extend_left; 43 src_ptr2 = src + src_stride * (height - 1) - extend_left; 44 dst_ptr1 = src + src_stride * -extend_top - extend_left; 45 dst_ptr2 = src + src_stride * height - extend_left; 46 47 for (i = 0; i < extend_top; ++i) { 48 vpx_memcpy(dst_ptr1, src_ptr1, linesize); 49 dst_ptr1 += src_stride; 50 } 51 52 for (i = 0; i < extend_bottom; ++i) { 53 vpx_memcpy(dst_ptr2, src_ptr2, linesize); 54 dst_ptr2 += src_stride; 55 } 56} 57 58void vp8_yv12_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf) { 59 assert(ybf->y_height - ybf->y_crop_height < 16); 60 assert(ybf->y_width - ybf->y_crop_width < 16); 61 assert(ybf->y_height - ybf->y_crop_height >= 0); 62 assert(ybf->y_width - ybf->y_crop_width >= 0); 63 64 extend_plane(ybf->y_buffer, ybf->y_stride, 65 ybf->y_crop_width, ybf->y_crop_height, 66 ybf->border, ybf->border, 67 ybf->border + ybf->y_height - ybf->y_crop_height, 68 ybf->border + ybf->y_width - ybf->y_crop_width); 69 70 extend_plane(ybf->u_buffer, ybf->uv_stride, 71 (ybf->y_crop_width + 1) / 2, (ybf->y_crop_height + 1) / 2, 72 ybf->border / 2, ybf->border / 2, 73 (ybf->border + ybf->y_height - ybf->y_crop_height + 1) / 2, 74 (ybf->border + ybf->y_width - ybf->y_crop_width + 1) / 2); 75 76 extend_plane(ybf->v_buffer, ybf->uv_stride, 77 (ybf->y_crop_width + 1) / 2, (ybf->y_crop_height + 1) / 2, 78 ybf->border / 2, ybf->border / 2, 79 (ybf->border + ybf->y_height - ybf->y_crop_height + 1) / 2, 80 (ybf->border + ybf->y_width - ybf->y_crop_width + 1) / 2); 81} 82 83#if CONFIG_VP9 84static void extend_frame(YV12_BUFFER_CONFIG *const ybf, int ext_size) { 85 const int c_w = ybf->uv_crop_width; 86 const int c_h = ybf->uv_crop_height; 87 const int c_ext_size = ext_size >> 1; 88 const int c_et = c_ext_size; 89 const int c_el = c_ext_size; 90 const int c_eb = c_ext_size + ybf->uv_height - ybf->uv_crop_height; 91 const int c_er = c_ext_size + ybf->uv_width - ybf->uv_crop_width; 92 93 assert(ybf->y_height - ybf->y_crop_height < 16); 94 assert(ybf->y_width - ybf->y_crop_width < 16); 95 assert(ybf->y_height - ybf->y_crop_height >= 0); 96 assert(ybf->y_width - ybf->y_crop_width >= 0); 97 98 extend_plane(ybf->y_buffer, ybf->y_stride, 99 ybf->y_crop_width, ybf->y_crop_height, 100 ext_size, ext_size, 101 ext_size + ybf->y_height - ybf->y_crop_height, 102 ext_size + ybf->y_width - ybf->y_crop_width); 103 104 extend_plane(ybf->u_buffer, ybf->uv_stride, 105 c_w, c_h, c_et, c_el, c_eb, c_er); 106 107 extend_plane(ybf->v_buffer, ybf->uv_stride, 108 c_w, c_h, c_et, c_el, c_eb, c_er); 109} 110 111void vp9_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf) { 112 extend_frame(ybf, ybf->border); 113} 114 115void vp9_extend_frame_inner_borders_c(YV12_BUFFER_CONFIG *ybf) { 116 const int inner_bw = (ybf->border > VP9INNERBORDERINPIXELS) ? 117 VP9INNERBORDERINPIXELS : ybf->border; 118 extend_frame(ybf, inner_bw); 119} 120#endif // CONFIG_VP9 121 122// Copies the source image into the destination image and updates the 123// destination's UMV borders. 124// Note: The frames are assumed to be identical in size. 125void vp8_yv12_copy_frame_c(const YV12_BUFFER_CONFIG *src_ybc, 126 YV12_BUFFER_CONFIG *dst_ybc) { 127 int row; 128 const uint8_t *src = src_ybc->y_buffer; 129 uint8_t *dst = dst_ybc->y_buffer; 130 131#if 0 132 /* These assertions are valid in the codec, but the libvpx-tester uses 133 * this code slightly differently. 134 */ 135 assert(src_ybc->y_width == dst_ybc->y_width); 136 assert(src_ybc->y_height == dst_ybc->y_height); 137#endif 138 139 for (row = 0; row < src_ybc->y_height; ++row) { 140 vpx_memcpy(dst, src, src_ybc->y_width); 141 src += src_ybc->y_stride; 142 dst += dst_ybc->y_stride; 143 } 144 145 src = src_ybc->u_buffer; 146 dst = dst_ybc->u_buffer; 147 148 for (row = 0; row < src_ybc->uv_height; ++row) { 149 vpx_memcpy(dst, src, src_ybc->uv_width); 150 src += src_ybc->uv_stride; 151 dst += dst_ybc->uv_stride; 152 } 153 154 src = src_ybc->v_buffer; 155 dst = dst_ybc->v_buffer; 156 157 for (row = 0; row < src_ybc->uv_height; ++row) { 158 vpx_memcpy(dst, src, src_ybc->uv_width); 159 src += src_ybc->uv_stride; 160 dst += dst_ybc->uv_stride; 161 } 162 163 vp8_yv12_extend_frame_borders_c(dst_ybc); 164} 165 166void vpx_yv12_copy_y_c(const YV12_BUFFER_CONFIG *src_ybc, 167 YV12_BUFFER_CONFIG *dst_ybc) { 168 int row; 169 const uint8_t *src = src_ybc->y_buffer; 170 uint8_t *dst = dst_ybc->y_buffer; 171 172 for (row = 0; row < src_ybc->y_height; ++row) { 173 vpx_memcpy(dst, src, src_ybc->y_width); 174 src += src_ybc->y_stride; 175 dst += dst_ybc->y_stride; 176 } 177} 178