OLD | NEW |
(Empty) | |
| 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 "vp9/common/vp9_extend.h" |
| 12 #include "vpx_mem/vpx_mem.h" |
| 13 |
| 14 static void copy_and_extend_plane(unsigned char *s, /* source */ |
| 15 int sp, /* source pitch */ |
| 16 unsigned char *d, /* destination */ |
| 17 int dp, /* destination pitch */ |
| 18 int h, /* height */ |
| 19 int w, /* width */ |
| 20 int et, /* extend top border */ |
| 21 int el, /* extend left border */ |
| 22 int eb, /* extend bottom border */ |
| 23 int er) { /* extend right border */ |
| 24 int i; |
| 25 unsigned char *src_ptr1, *src_ptr2; |
| 26 unsigned char *dest_ptr1, *dest_ptr2; |
| 27 int linesize; |
| 28 |
| 29 /* copy the left and right most columns out */ |
| 30 src_ptr1 = s; |
| 31 src_ptr2 = s + w - 1; |
| 32 dest_ptr1 = d - el; |
| 33 dest_ptr2 = d + w; |
| 34 |
| 35 for (i = 0; i < h; i++) { |
| 36 vpx_memset(dest_ptr1, src_ptr1[0], el); |
| 37 vpx_memcpy(dest_ptr1 + el, src_ptr1, w); |
| 38 vpx_memset(dest_ptr2, src_ptr2[0], er); |
| 39 src_ptr1 += sp; |
| 40 src_ptr2 += sp; |
| 41 dest_ptr1 += dp; |
| 42 dest_ptr2 += dp; |
| 43 } |
| 44 |
| 45 /* Now copy the top and bottom lines into each line of the respective |
| 46 * borders |
| 47 */ |
| 48 src_ptr1 = d - el; |
| 49 src_ptr2 = d + dp * (h - 1) - el; |
| 50 dest_ptr1 = d + dp * (-et) - el; |
| 51 dest_ptr2 = d + dp * (h) - el; |
| 52 linesize = el + er + w; |
| 53 |
| 54 for (i = 0; i < et; i++) { |
| 55 vpx_memcpy(dest_ptr1, src_ptr1, linesize); |
| 56 dest_ptr1 += dp; |
| 57 } |
| 58 |
| 59 for (i = 0; i < eb; i++) { |
| 60 vpx_memcpy(dest_ptr2, src_ptr2, linesize); |
| 61 dest_ptr2 += dp; |
| 62 } |
| 63 } |
| 64 |
| 65 void vp9_copy_and_extend_frame(YV12_BUFFER_CONFIG *src, |
| 66 YV12_BUFFER_CONFIG *dst) { |
| 67 int et = dst->border; |
| 68 int el = dst->border; |
| 69 int eb = dst->border + dst->y_height - src->y_height; |
| 70 int er = dst->border + dst->y_width - src->y_width; |
| 71 |
| 72 copy_and_extend_plane(src->y_buffer, src->y_stride, |
| 73 dst->y_buffer, dst->y_stride, |
| 74 src->y_height, src->y_width, |
| 75 et, el, eb, er); |
| 76 |
| 77 et = dst->border >> 1; |
| 78 el = dst->border >> 1; |
| 79 eb = (dst->border >> 1) + dst->uv_height - src->uv_height; |
| 80 er = (dst->border >> 1) + dst->uv_width - src->uv_width; |
| 81 |
| 82 copy_and_extend_plane(src->u_buffer, src->uv_stride, |
| 83 dst->u_buffer, dst->uv_stride, |
| 84 src->uv_height, src->uv_width, |
| 85 et, el, eb, er); |
| 86 |
| 87 copy_and_extend_plane(src->v_buffer, src->uv_stride, |
| 88 dst->v_buffer, dst->uv_stride, |
| 89 src->uv_height, src->uv_width, |
| 90 et, el, eb, er); |
| 91 } |
| 92 |
| 93 void vp9_copy_and_extend_frame_with_rect(YV12_BUFFER_CONFIG *src, |
| 94 YV12_BUFFER_CONFIG *dst, |
| 95 int srcy, int srcx, |
| 96 int srch, int srcw) { |
| 97 int et = dst->border; |
| 98 int el = dst->border; |
| 99 int eb = dst->border + dst->y_height - src->y_height; |
| 100 int er = dst->border + dst->y_width - src->y_width; |
| 101 int src_y_offset = srcy * src->y_stride + srcx; |
| 102 int dst_y_offset = srcy * dst->y_stride + srcx; |
| 103 int src_uv_offset = ((srcy * src->uv_stride) >> 1) + (srcx >> 1); |
| 104 int dst_uv_offset = ((srcy * dst->uv_stride) >> 1) + (srcx >> 1); |
| 105 |
| 106 // If the side is not touching the bounder then don't extend. |
| 107 if (srcy) |
| 108 et = 0; |
| 109 if (srcx) |
| 110 el = 0; |
| 111 if (srcy + srch != src->y_height) |
| 112 eb = 0; |
| 113 if (srcx + srcw != src->y_width) |
| 114 er = 0; |
| 115 |
| 116 copy_and_extend_plane(src->y_buffer + src_y_offset, |
| 117 src->y_stride, |
| 118 dst->y_buffer + dst_y_offset, |
| 119 dst->y_stride, |
| 120 srch, srcw, |
| 121 et, el, eb, er); |
| 122 |
| 123 et = (et + 1) >> 1; |
| 124 el = (el + 1) >> 1; |
| 125 eb = (eb + 1) >> 1; |
| 126 er = (er + 1) >> 1; |
| 127 srch = (srch + 1) >> 1; |
| 128 srcw = (srcw + 1) >> 1; |
| 129 |
| 130 copy_and_extend_plane(src->u_buffer + src_uv_offset, |
| 131 src->uv_stride, |
| 132 dst->u_buffer + dst_uv_offset, |
| 133 dst->uv_stride, |
| 134 srch, srcw, |
| 135 et, el, eb, er); |
| 136 |
| 137 copy_and_extend_plane(src->v_buffer + src_uv_offset, |
| 138 src->uv_stride, |
| 139 dst->v_buffer + dst_uv_offset, |
| 140 dst->uv_stride, |
| 141 srch, srcw, |
| 142 et, el, eb, er); |
| 143 } |
| 144 |
| 145 /* note the extension is only for the last row, for intra prediction purpose */ |
| 146 void vp9_extend_mb_row(YV12_BUFFER_CONFIG *ybf, unsigned char *YPtr, |
| 147 unsigned char *UPtr, unsigned char *VPtr) { |
| 148 int i; |
| 149 |
| 150 YPtr += ybf->y_stride * 14; |
| 151 UPtr += ybf->uv_stride * 6; |
| 152 VPtr += ybf->uv_stride * 6; |
| 153 |
| 154 for (i = 0; i < 4; i++) { |
| 155 YPtr[i] = YPtr[-1]; |
| 156 UPtr[i] = UPtr[-1]; |
| 157 VPtr[i] = VPtr[-1]; |
| 158 } |
| 159 |
| 160 YPtr += ybf->y_stride; |
| 161 UPtr += ybf->uv_stride; |
| 162 VPtr += ybf->uv_stride; |
| 163 |
| 164 for (i = 0; i < 4; i++) { |
| 165 YPtr[i] = YPtr[-1]; |
| 166 UPtr[i] = UPtr[-1]; |
| 167 VPtr[i] = VPtr[-1]; |
| 168 } |
| 169 } |
OLD | NEW |