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 
extend_plane(uint8_t * const src,int src_stride,int width,int height,int extend_top,int extend_left,int extend_bottom,int extend_right)17 static 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 
vp8_yv12_extend_frame_borders_c(YV12_BUFFER_CONFIG * ybf)58 void 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
extend_frame(YV12_BUFFER_CONFIG * const ybf,int ext_size)84 static 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 
vp9_extend_frame_borders_c(YV12_BUFFER_CONFIG * ybf)111 void vp9_extend_frame_borders_c(YV12_BUFFER_CONFIG *ybf) {
112   extend_frame(ybf, ybf->border);
113 }
114 
vp9_extend_frame_inner_borders_c(YV12_BUFFER_CONFIG * ybf)115 void 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.
vp8_yv12_copy_frame_c(const YV12_BUFFER_CONFIG * src_ybc,YV12_BUFFER_CONFIG * dst_ybc)125 void 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 
vpx_yv12_copy_y_c(const YV12_BUFFER_CONFIG * src_ybc,YV12_BUFFER_CONFIG * dst_ybc)166 void 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