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
12 #include <stdlib.h>
13 #include <string.h>
14 #include "vpx/vpx_image.h"
15
img_alloc_helper(vpx_image_t * img,vpx_img_fmt_t fmt,unsigned int d_w,unsigned int d_h,unsigned int stride_align,unsigned char * img_data)16 static vpx_image_t *img_alloc_helper(vpx_image_t *img,
17 vpx_img_fmt_t fmt,
18 unsigned int d_w,
19 unsigned int d_h,
20 unsigned int stride_align,
21 unsigned char *img_data)
22 {
23
24 unsigned int h, w, s, xcs, ycs, bps;
25 int align;
26
27 /* Treat align==0 like align==1 */
28 if (!stride_align)
29 stride_align = 1;
30
31 /* Validate alignment (must be power of 2) */
32 if (stride_align & (stride_align - 1))
33 goto fail;
34
35 /* Get sample size for this format */
36 switch (fmt)
37 {
38 case VPX_IMG_FMT_RGB32:
39 case VPX_IMG_FMT_RGB32_LE:
40 case VPX_IMG_FMT_ARGB:
41 case VPX_IMG_FMT_ARGB_LE:
42 bps = 32;
43 break;
44 case VPX_IMG_FMT_RGB24:
45 case VPX_IMG_FMT_BGR24:
46 bps = 24;
47 break;
48 case VPX_IMG_FMT_RGB565:
49 case VPX_IMG_FMT_RGB565_LE:
50 case VPX_IMG_FMT_RGB555:
51 case VPX_IMG_FMT_RGB555_LE:
52 case VPX_IMG_FMT_UYVY:
53 case VPX_IMG_FMT_YUY2:
54 case VPX_IMG_FMT_YVYU:
55 bps = 16;
56 break;
57 case VPX_IMG_FMT_I420:
58 case VPX_IMG_FMT_YV12:
59 case VPX_IMG_FMT_VPXI420:
60 case VPX_IMG_FMT_VPXYV12:
61 bps = 12;
62 break;
63 default:
64 bps = 16;
65 break;
66 }
67
68 /* Get chroma shift values for this format */
69 switch (fmt)
70 {
71 case VPX_IMG_FMT_I420:
72 case VPX_IMG_FMT_YV12:
73 case VPX_IMG_FMT_VPXI420:
74 case VPX_IMG_FMT_VPXYV12:
75 xcs = 1;
76 break;
77 default:
78 xcs = 0;
79 break;
80 }
81
82 switch (fmt)
83 {
84 case VPX_IMG_FMT_I420:
85 case VPX_IMG_FMT_YV12:
86 case VPX_IMG_FMT_VPXI420:
87 case VPX_IMG_FMT_VPXYV12:
88 ycs = 1;
89 break;
90 default:
91 ycs = 0;
92 break;
93 }
94
95 /* Calculate storage sizes given the chroma subsampling */
96 align = (1 << xcs) - 1;
97 w = (d_w + align) & ~align;
98 align = (1 << ycs) - 1;
99 h = (d_h + align) & ~align;
100 s = (fmt & VPX_IMG_FMT_PLANAR) ? w : bps * w / 8;
101 s = (s + stride_align - 1) & ~(stride_align - 1);
102
103 /* Allocate the new image */
104 if (!img)
105 {
106 img = (vpx_image_t *)calloc(1, sizeof(vpx_image_t));
107
108 if (!img)
109 goto fail;
110
111 img->self_allocd = 1;
112 }
113 else
114 {
115 memset(img, 0, sizeof(vpx_image_t));
116 }
117
118 img->img_data = img_data;
119
120 if (!img_data)
121 {
122 img->img_data = malloc((fmt & VPX_IMG_FMT_PLANAR) ? h * w * bps / 8 : h * s);
123 img->img_data_owner = 1;
124 }
125
126 if (!img->img_data)
127 goto fail;
128
129 img->fmt = fmt;
130 img->w = w;
131 img->h = h;
132 img->x_chroma_shift = xcs;
133 img->y_chroma_shift = ycs;
134 img->bps = bps;
135
136 /* Calculate strides */
137 img->stride[VPX_PLANE_Y] = img->stride[VPX_PLANE_ALPHA] = s;
138 img->stride[VPX_PLANE_U] = img->stride[VPX_PLANE_V] = s >> xcs;
139
140 /* Default viewport to entire image */
141 if (!vpx_img_set_rect(img, 0, 0, d_w, d_h))
142 return img;
143
144 fail:
145 vpx_img_free(img);
146 return NULL;
147 }
148
vpx_img_alloc(vpx_image_t * img,vpx_img_fmt_t fmt,unsigned int d_w,unsigned int d_h,unsigned int stride_align)149 vpx_image_t *vpx_img_alloc(vpx_image_t *img,
150 vpx_img_fmt_t fmt,
151 unsigned int d_w,
152 unsigned int d_h,
153 unsigned int stride_align)
154 {
155 return img_alloc_helper(img, fmt, d_w, d_h, stride_align, NULL);
156 }
157
vpx_img_wrap(vpx_image_t * img,vpx_img_fmt_t fmt,unsigned int d_w,unsigned int d_h,unsigned int stride_align,unsigned char * img_data)158 vpx_image_t *vpx_img_wrap(vpx_image_t *img,
159 vpx_img_fmt_t fmt,
160 unsigned int d_w,
161 unsigned int d_h,
162 unsigned int stride_align,
163 unsigned char *img_data)
164 {
165 return img_alloc_helper(img, fmt, d_w, d_h, stride_align, img_data);
166 }
167
vpx_img_set_rect(vpx_image_t * img,unsigned int x,unsigned int y,unsigned int w,unsigned int h)168 int vpx_img_set_rect(vpx_image_t *img,
169 unsigned int x,
170 unsigned int y,
171 unsigned int w,
172 unsigned int h)
173 {
174 unsigned char *data;
175
176 if (x + w <= img->w && y + h <= img->h)
177 {
178 img->d_w = w;
179 img->d_h = h;
180
181 /* Calculate plane pointers */
182 if (!(img->fmt & VPX_IMG_FMT_PLANAR))
183 {
184 img->planes[VPX_PLANE_PACKED] =
185 img->img_data + x * img->bps / 8 + y * img->stride[VPX_PLANE_PACKED];
186 }
187 else
188 {
189 data = img->img_data;
190
191 if (img->fmt & VPX_IMG_FMT_HAS_ALPHA)
192 {
193 img->planes[VPX_PLANE_ALPHA] =
194 data + x + y * img->stride[VPX_PLANE_ALPHA];
195 data += img->h * img->stride[VPX_PLANE_ALPHA];
196 }
197
198 img->planes[VPX_PLANE_Y] = data + x + y * img->stride[VPX_PLANE_Y];
199 data += img->h * img->stride[VPX_PLANE_Y];
200
201 if (!(img->fmt & VPX_IMG_FMT_UV_FLIP))
202 {
203 img->planes[VPX_PLANE_U] = data
204 + (x >> img->x_chroma_shift)
205 + (y >> img->y_chroma_shift) * img->stride[VPX_PLANE_U];
206 data += (img->h >> img->y_chroma_shift) * img->stride[VPX_PLANE_U];
207 img->planes[VPX_PLANE_V] = data
208 + (x >> img->x_chroma_shift)
209 + (y >> img->y_chroma_shift) * img->stride[VPX_PLANE_V];
210 }
211 else
212 {
213 img->planes[VPX_PLANE_V] = data
214 + (x >> img->x_chroma_shift)
215 + (y >> img->y_chroma_shift) * img->stride[VPX_PLANE_V];
216 data += (img->h >> img->y_chroma_shift) * img->stride[VPX_PLANE_V];
217 img->planes[VPX_PLANE_U] = data
218 + (x >> img->x_chroma_shift)
219 + (y >> img->y_chroma_shift) * img->stride[VPX_PLANE_U];
220 }
221 }
222
223 return 0;
224 }
225
226 return -1;
227 }
228
vpx_img_flip(vpx_image_t * img)229 void vpx_img_flip(vpx_image_t *img)
230 {
231 /* Note: In the calculation pointer adjustment calculation, we want the
232 * rhs to be promoted to a signed type. Section 6.3.1.8 of the ISO C99
233 * standard indicates that if the adjustment parameter is unsigned, the
234 * stride parameter will be promoted to unsigned, causing errors when
235 * the lhs is a larger type than the rhs.
236 */
237 img->planes[VPX_PLANE_Y] += (signed)(img->d_h - 1) * img->stride[VPX_PLANE_Y];
238 img->stride[VPX_PLANE_Y] = -img->stride[VPX_PLANE_Y];
239
240 img->planes[VPX_PLANE_U] += (signed)((img->d_h >> img->y_chroma_shift) - 1)
241 * img->stride[VPX_PLANE_U];
242 img->stride[VPX_PLANE_U] = -img->stride[VPX_PLANE_U];
243
244 img->planes[VPX_PLANE_V] += (signed)((img->d_h >> img->y_chroma_shift) - 1)
245 * img->stride[VPX_PLANE_V];
246 img->stride[VPX_PLANE_V] = -img->stride[VPX_PLANE_V];
247
248 img->planes[VPX_PLANE_ALPHA] += (signed)(img->d_h - 1) * img->stride[VPX_PLANE_ALPHA];
249 img->stride[VPX_PLANE_ALPHA] = -img->stride[VPX_PLANE_ALPHA];
250 }
251
vpx_img_free(vpx_image_t * img)252 void vpx_img_free(vpx_image_t *img)
253 {
254 if (img)
255 {
256 if (img->img_data && img->img_data_owner)
257 free(img->img_data);
258
259 if (img->self_allocd)
260 free(img);
261 }
262 }
263