[svn] / branches / release-1_3-branch / xvidcore / src / image / image.c Repository:
ViewVC logotype

Annotation of /branches/release-1_3-branch/xvidcore/src/image/image.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1665 - (view) (download)
Original Path: trunk/xvidcore/src/image/image.c

1 : edgomez 851 /**************************************************************************
2 : albeu 315 *
3 : edgomez 1382 * XVID MPEG-4 VIDEO CODEC
4 :     * - Image management functions -
5 : albeu 315 *
6 : suxen_drol 1397 * Copyright(C) 2001-2004 Peter Ross <pross@xvid.org>
7 : albeu 315 *
8 : edgomez 1382 * This program is free software ; you can redistribute it and/or modify
9 :     * it under the terms of the GNU General Public License as published by
10 :     * the Free Software Foundation ; either version 2 of the License, or
11 :     * (at your option) any later version.
12 : albeu 315 *
13 : edgomez 1382 * This program is distributed in the hope that it will be useful,
14 :     * but WITHOUT ANY WARRANTY ; without even the implied warranty of
15 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 :     * GNU General Public License for more details.
17 : albeu 315 *
18 : edgomez 1382 * You should have received a copy of the GNU General Public License
19 :     * along with this program ; if not, write to the Free Software
20 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21 : albeu 315 *
22 : syskin 1665 * $Id: image.c,v 1.34 2005-12-17 12:04:52 syskin Exp $
23 : albeu 315 *
24 : edgomez 1382 ****************************************************************************/
25 : albeu 315
26 :     #include <stdlib.h>
27 : edgomez 1382 #include <string.h> /* memcpy, memset */
28 : albeu 315 #include <math.h>
29 :     #include "../portab.h"
30 : edgomez 1382 #include "../global.h"
31 :     #include "../xvid.h"
32 : albeu 315 #include "image.h"
33 :     #include "colorspace.h"
34 :     #include "interpolate8x8.h"
35 :     #include "../utils/mem_align.h"
36 : edgomez 1424 #include "../motion/sad.h"
37 : albeu 315
38 : edgomez 1382 #include "font.h"
39 : edgomez 851
40 : albeu 315 #define SAFETY 64
41 :     #define EDGE_SIZE2 (EDGE_SIZE/2)
42 :    
43 :    
44 :     int32_t
45 :     image_create(IMAGE * image,
46 :     uint32_t edged_width,
47 :     uint32_t edged_height)
48 :     {
49 :     const uint32_t edged_width2 = edged_width / 2;
50 :     const uint32_t edged_height2 = edged_height / 2;
51 :    
52 :     image->y =
53 :     xvid_malloc(edged_width * (edged_height + 1) + SAFETY, CACHE_LINE);
54 :     if (image->y == NULL) {
55 :     return -1;
56 :     }
57 : edgomez 1382 memset(image->y, 0, edged_width * (edged_height + 1) + SAFETY);
58 : albeu 315
59 :     image->u = xvid_malloc(edged_width2 * edged_height2 + SAFETY, CACHE_LINE);
60 :     if (image->u == NULL) {
61 :     xvid_free(image->y);
62 : edgomez 1382 image->y = NULL;
63 : albeu 315 return -1;
64 :     }
65 : edgomez 1382 memset(image->u, 0, edged_width2 * edged_height2 + SAFETY);
66 :    
67 : albeu 315 image->v = xvid_malloc(edged_width2 * edged_height2 + SAFETY, CACHE_LINE);
68 :     if (image->v == NULL) {
69 :     xvid_free(image->u);
70 : edgomez 1382 image->u = NULL;
71 : albeu 315 xvid_free(image->y);
72 : edgomez 1382 image->y = NULL;
73 : albeu 315 return -1;
74 :     }
75 : edgomez 1382 memset(image->v, 0, edged_width2 * edged_height2 + SAFETY);
76 : albeu 315
77 :     image->y += EDGE_SIZE * edged_width + EDGE_SIZE;
78 :     image->u += EDGE_SIZE2 * edged_width2 + EDGE_SIZE2;
79 :     image->v += EDGE_SIZE2 * edged_width2 + EDGE_SIZE2;
80 :    
81 :     return 0;
82 :     }
83 :    
84 :    
85 :    
86 :     void
87 :     image_destroy(IMAGE * image,
88 :     uint32_t edged_width,
89 :     uint32_t edged_height)
90 :     {
91 :     const uint32_t edged_width2 = edged_width / 2;
92 :    
93 :     if (image->y) {
94 :     xvid_free(image->y - (EDGE_SIZE * edged_width + EDGE_SIZE));
95 : edgomez 1382 image->y = NULL;
96 : albeu 315 }
97 :     if (image->u) {
98 :     xvid_free(image->u - (EDGE_SIZE2 * edged_width2 + EDGE_SIZE2));
99 : edgomez 1382 image->u = NULL;
100 : albeu 315 }
101 :     if (image->v) {
102 :     xvid_free(image->v - (EDGE_SIZE2 * edged_width2 + EDGE_SIZE2));
103 : edgomez 1382 image->v = NULL;
104 : albeu 315 }
105 :     }
106 :    
107 :    
108 :     void
109 :     image_swap(IMAGE * image1,
110 :     IMAGE * image2)
111 :     {
112 : edgomez 1382 SWAP(uint8_t*, image1->y, image2->y);
113 :     SWAP(uint8_t*, image1->u, image2->u);
114 :     SWAP(uint8_t*, image1->v, image2->v);
115 : albeu 315 }
116 :    
117 :    
118 :     void
119 :     image_copy(IMAGE * image1,
120 :     IMAGE * image2,
121 :     uint32_t edged_width,
122 :     uint32_t height)
123 :     {
124 :     memcpy(image1->y, image2->y, edged_width * height);
125 :     memcpy(image1->u, image2->u, edged_width * height / 4);
126 :     memcpy(image1->v, image2->v, edged_width * height / 4);
127 :     }
128 :    
129 : edgomez 1382 /* setedges bug was fixed in this BS version */
130 :     #define SETEDGES_BUG_BEFORE 18
131 : albeu 315
132 :     void
133 :     image_setedges(IMAGE * image,
134 :     uint32_t edged_width,
135 :     uint32_t edged_height,
136 :     uint32_t width,
137 : edgomez 1382 uint32_t height,
138 :     int bs_version)
139 : albeu 315 {
140 :     const uint32_t edged_width2 = edged_width / 2;
141 : edgomez 1382 uint32_t width2;
142 : albeu 315 uint32_t i;
143 :     uint8_t *dst;
144 :     uint8_t *src;
145 :    
146 :     dst = image->y - (EDGE_SIZE + EDGE_SIZE * edged_width);
147 :     src = image->y;
148 :    
149 : edgomez 1382 /* According to the Standard Clause 7.6.4, padding is done starting at 16
150 :     * pixel width and height multiples. This was not respected in old xvids */
151 :     if (bs_version == 0 || bs_version >= SETEDGES_BUG_BEFORE) {
152 :     width = (width+15)&~15;
153 :     height = (height+15)&~15;
154 :     }
155 :    
156 :     width2 = width/2;
157 :    
158 : albeu 315 for (i = 0; i < EDGE_SIZE; i++) {
159 : edgomez 851 memset(dst, *src, EDGE_SIZE);
160 :     memcpy(dst + EDGE_SIZE, src, width);
161 :     memset(dst + edged_width - EDGE_SIZE, *(src + width - 1),
162 :     EDGE_SIZE);
163 : albeu 315 dst += edged_width;
164 :     }
165 :    
166 :     for (i = 0; i < height; i++) {
167 :     memset(dst, *src, EDGE_SIZE);
168 :     memset(dst + edged_width - EDGE_SIZE, src[width - 1], EDGE_SIZE);
169 :     dst += edged_width;
170 :     src += edged_width;
171 :     }
172 :    
173 :     src -= edged_width;
174 :     for (i = 0; i < EDGE_SIZE; i++) {
175 : edgomez 851 memset(dst, *src, EDGE_SIZE);
176 :     memcpy(dst + EDGE_SIZE, src, width);
177 :     memset(dst + edged_width - EDGE_SIZE, *(src + width - 1),
178 : albeu 315 EDGE_SIZE);
179 :     dst += edged_width;
180 :     }
181 :    
182 :    
183 : edgomez 1382 /* U */
184 : albeu 315 dst = image->u - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2);
185 :     src = image->u;
186 :    
187 :     for (i = 0; i < EDGE_SIZE2; i++) {
188 :     memset(dst, *src, EDGE_SIZE2);
189 :     memcpy(dst + EDGE_SIZE2, src, width2);
190 :     memset(dst + edged_width2 - EDGE_SIZE2, *(src + width2 - 1),
191 :     EDGE_SIZE2);
192 :     dst += edged_width2;
193 :     }
194 :    
195 :     for (i = 0; i < height / 2; i++) {
196 :     memset(dst, *src, EDGE_SIZE2);
197 :     memset(dst + edged_width2 - EDGE_SIZE2, src[width2 - 1], EDGE_SIZE2);
198 :     dst += edged_width2;
199 :     src += edged_width2;
200 :     }
201 :     src -= edged_width2;
202 :     for (i = 0; i < EDGE_SIZE2; i++) {
203 :     memset(dst, *src, EDGE_SIZE2);
204 :     memcpy(dst + EDGE_SIZE2, src, width2);
205 :     memset(dst + edged_width2 - EDGE_SIZE2, *(src + width2 - 1),
206 :     EDGE_SIZE2);
207 :     dst += edged_width2;
208 :     }
209 :    
210 :    
211 : edgomez 1382 /* V */
212 : albeu 315 dst = image->v - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2);
213 :     src = image->v;
214 :    
215 :     for (i = 0; i < EDGE_SIZE2; i++) {
216 :     memset(dst, *src, EDGE_SIZE2);
217 :     memcpy(dst + EDGE_SIZE2, src, width2);
218 :     memset(dst + edged_width2 - EDGE_SIZE2, *(src + width2 - 1),
219 :     EDGE_SIZE2);
220 :     dst += edged_width2;
221 :     }
222 :    
223 :     for (i = 0; i < height / 2; i++) {
224 :     memset(dst, *src, EDGE_SIZE2);
225 :     memset(dst + edged_width2 - EDGE_SIZE2, src[width2 - 1], EDGE_SIZE2);
226 :     dst += edged_width2;
227 :     src += edged_width2;
228 :     }
229 :     src -= edged_width2;
230 :     for (i = 0; i < EDGE_SIZE2; i++) {
231 :     memset(dst, *src, EDGE_SIZE2);
232 :     memcpy(dst + EDGE_SIZE2, src, width2);
233 :     memset(dst + edged_width2 - EDGE_SIZE2, *(src + width2 - 1),
234 :     EDGE_SIZE2);
235 :     dst += edged_width2;
236 :     }
237 :     }
238 :    
239 :     void
240 : syskin 1665 image_interpolate(const uint8_t * refn,
241 :     uint8_t * refh,
242 :     uint8_t * refv,
243 :     uint8_t * refhv,
244 : albeu 315 uint32_t edged_width,
245 :     uint32_t edged_height,
246 : edgomez 851 uint32_t quarterpel,
247 : albeu 315 uint32_t rounding)
248 :     {
249 : edgomez 1382 const uint32_t offset = EDGE_SIZE2 * (edged_width + 1); /* we only interpolate half of the edge area */
250 : albeu 315 const uint32_t stride_add = 7 * edged_width;
251 : syskin 1664
252 : syskin 1665 uint8_t *n_ptr;
253 :     uint8_t *h_ptr, *v_ptr, *hv_ptr;
254 : albeu 315 uint32_t x, y;
255 :    
256 : syskin 1665 n_ptr = (uint8_t*)refn;
257 :     h_ptr = refh;
258 :     v_ptr = refhv;
259 : albeu 315
260 :     n_ptr -= offset;
261 :     h_ptr -= offset;
262 :     v_ptr -= offset;
263 :    
264 : edgomez 1382 /* Note we initialize the hv pointer later, as we can optimize code a bit
265 :     * doing it down to up in quarterpel and up to down in halfpel */
266 : edgomez 851 if(quarterpel) {
267 : edgomez 1382
268 : edgomez 851 for (y = 0; y < (edged_height - EDGE_SIZE); y += 8) {
269 :     for (x = 0; x < (edged_width - EDGE_SIZE); x += 8) {
270 :     interpolate8x8_6tap_lowpass_h(h_ptr, n_ptr, edged_width, rounding);
271 :     interpolate8x8_6tap_lowpass_v(v_ptr, n_ptr, edged_width, rounding);
272 : albeu 315
273 : edgomez 851 n_ptr += 8;
274 :     h_ptr += 8;
275 :     v_ptr += 8;
276 :     }
277 : edgomez 1382
278 : edgomez 851 n_ptr += EDGE_SIZE;
279 :     h_ptr += EDGE_SIZE;
280 :     v_ptr += EDGE_SIZE;
281 :    
282 :     h_ptr += stride_add;
283 :     v_ptr += stride_add;
284 :     n_ptr += stride_add;
285 :     }
286 :    
287 : syskin 1665 h_ptr = refh + (edged_height - EDGE_SIZE - EDGE_SIZE2)*edged_width - EDGE_SIZE2;
288 :     hv_ptr = refhv + (edged_height - EDGE_SIZE - EDGE_SIZE2)*edged_width - EDGE_SIZE2;
289 : edgomez 851
290 :     for (y = 0; y < (edged_height - EDGE_SIZE); y = y + 8) {
291 : edgomez 1382 hv_ptr -= stride_add;
292 :     h_ptr -= stride_add;
293 :     hv_ptr -= EDGE_SIZE;
294 :     h_ptr -= EDGE_SIZE;
295 :    
296 : edgomez 851 for (x = 0; x < (edged_width - EDGE_SIZE); x = x + 8) {
297 : edgomez 1382 hv_ptr -= 8;
298 :     h_ptr -= 8;
299 : edgomez 851 interpolate8x8_6tap_lowpass_v(hv_ptr, h_ptr, edged_width, rounding);
300 :     }
301 :     }
302 : edgomez 1382 } else {
303 : edgomez 851
304 : syskin 1665 hv_ptr = refhv;
305 : edgomez 1382 hv_ptr -= offset;
306 :    
307 : edgomez 851 for (y = 0; y < (edged_height - EDGE_SIZE); y += 8) {
308 :     for (x = 0; x < (edged_width - EDGE_SIZE); x += 8) {
309 :     interpolate8x8_halfpel_h(h_ptr, n_ptr, edged_width, rounding);
310 :     interpolate8x8_halfpel_v(v_ptr, n_ptr, edged_width, rounding);
311 :     interpolate8x8_halfpel_hv(hv_ptr, n_ptr, edged_width, rounding);
312 :    
313 :     n_ptr += 8;
314 :     h_ptr += 8;
315 :     v_ptr += 8;
316 :     hv_ptr += 8;
317 :     }
318 : edgomez 1382
319 : edgomez 851 h_ptr += EDGE_SIZE;
320 :     v_ptr += EDGE_SIZE;
321 :     hv_ptr += EDGE_SIZE;
322 :     n_ptr += EDGE_SIZE;
323 :    
324 :     h_ptr += stride_add;
325 :     v_ptr += stride_add;
326 :     hv_ptr += stride_add;
327 :     n_ptr += stride_add;
328 :     }
329 :     }
330 : albeu 315 }
331 :    
332 :    
333 : edgomez 851 /*
334 :     chroma optimize filter, invented by mf
335 :     a chroma pixel is average from the surrounding pixels, when the
336 :     correpsonding luma pixels are pure black or white.
337 :     */
338 :    
339 :     void
340 :     image_chroma_optimize(IMAGE * img, int width, int height, int edged_width)
341 :     {
342 :     int x,y;
343 :     int pixels = 0;
344 :    
345 :     for (y = 1; y < height/2 - 1; y++)
346 :     for (x = 1; x < width/2 - 1; x++)
347 :     {
348 :     #define IS_PURE(a) ((a)<=16||(a)>=235)
349 :     #define IMG_Y(Y,X) img->y[(Y)*edged_width + (X)]
350 :     #define IMG_U(Y,X) img->u[(Y)*edged_width/2 + (X)]
351 :     #define IMG_V(Y,X) img->v[(Y)*edged_width/2 + (X)]
352 :    
353 : edgomez 1382 if (IS_PURE(IMG_Y(y*2 ,x*2 )) &&
354 : edgomez 851 IS_PURE(IMG_Y(y*2 ,x*2+1)) &&
355 : edgomez 1382 IS_PURE(IMG_Y(y*2+1,x*2 )) &&
356 : edgomez 851 IS_PURE(IMG_Y(y*2+1,x*2+1)))
357 :     {
358 :     IMG_U(y,x) = (IMG_U(y,x-1) + IMG_U(y-1, x) + IMG_U(y, x+1) + IMG_U(y+1, x)) / 4;
359 :     IMG_V(y,x) = (IMG_V(y,x-1) + IMG_V(y-1, x) + IMG_V(y, x+1) + IMG_V(y+1, x)) / 4;
360 :     pixels++;
361 :     }
362 :    
363 :     #undef IS_PURE
364 :     #undef IMG_Y
365 :     #undef IMG_U
366 :     #undef IMG_V
367 :     }
368 : edgomez 1382
369 :     DPRINTF(XVID_DEBUG_DEBUG,"chroma_optimized_pixels = %i/%i\n", pixels, width*height/4);
370 : edgomez 851 }
371 :    
372 :    
373 :    
374 :    
375 :    
376 :     /*
377 :     perform safe packed colorspace conversion, by splitting
378 :     the image up into an optimized area (pixel width divisible by 16),
379 :     and two unoptimized/plain-c areas (pixel width divisible by 2)
380 :     */
381 :    
382 : edgomez 1382 static void
383 : edgomez 851 safe_packed_conv(uint8_t * x_ptr, int x_stride,
384 :     uint8_t * y_ptr, uint8_t * u_ptr, uint8_t * v_ptr,
385 :     int y_stride, int uv_stride,
386 :     int width, int height, int vflip,
387 :     packedFunc * func_opt, packedFunc func_c, int size)
388 :     {
389 :     int width_opt, width_c;
390 :    
391 :     if (func_opt != func_c && x_stride < size*((width+15)/16)*16)
392 :     {
393 :     width_opt = width & (~15);
394 :     width_c = width - width_opt;
395 :     }
396 :     else
397 :     {
398 :     width_opt = width;
399 :     width_c = 0;
400 :     }
401 :    
402 :     func_opt(x_ptr, x_stride,
403 :     y_ptr, u_ptr, v_ptr, y_stride, uv_stride,
404 :     width_opt, height, vflip);
405 :    
406 :     if (width_c)
407 :     {
408 :     func_c(x_ptr + size*width_opt, x_stride,
409 :     y_ptr + width_opt, u_ptr + width_opt/2, v_ptr + width_opt/2,
410 :     y_stride, uv_stride, width_c, height, vflip);
411 :     }
412 :     }
413 :    
414 :    
415 :    
416 : albeu 315 int
417 :     image_input(IMAGE * image,
418 :     uint32_t width,
419 :     int height,
420 :     uint32_t edged_width,
421 : edgomez 1382 uint8_t * src[4],
422 :     int src_stride[4],
423 : edgomez 851 int csp,
424 :     int interlacing)
425 : albeu 315 {
426 : edgomez 851 const int edged_width2 = edged_width/2;
427 :     const int width2 = width/2;
428 :     const int height2 = height/2;
429 : edgomez 1382 #if 0
430 :     const int height_signed = (csp & XVID_CSP_VFLIP) ? -height : height;
431 :     #endif
432 : albeu 315
433 :     switch (csp & ~XVID_CSP_VFLIP) {
434 :     case XVID_CSP_RGB555:
435 : edgomez 851 safe_packed_conv(
436 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
437 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
438 :     interlacing?rgb555i_to_yv12 :rgb555_to_yv12,
439 :     interlacing?rgb555i_to_yv12_c:rgb555_to_yv12_c, 2);
440 :     break;
441 : albeu 315
442 :     case XVID_CSP_RGB565:
443 : edgomez 851 safe_packed_conv(
444 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
445 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
446 :     interlacing?rgb565i_to_yv12 :rgb565_to_yv12,
447 :     interlacing?rgb565i_to_yv12_c:rgb565_to_yv12_c, 2);
448 :     break;
449 : albeu 315
450 :    
451 : edgomez 1382 case XVID_CSP_BGR:
452 : edgomez 851 safe_packed_conv(
453 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
454 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
455 :     interlacing?bgri_to_yv12 :bgr_to_yv12,
456 :     interlacing?bgri_to_yv12_c:bgr_to_yv12_c, 3);
457 :     break;
458 : albeu 315
459 : edgomez 1382 case XVID_CSP_BGRA:
460 : edgomez 851 safe_packed_conv(
461 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
462 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
463 :     interlacing?bgrai_to_yv12 :bgra_to_yv12,
464 :     interlacing?bgrai_to_yv12_c:bgra_to_yv12_c, 4);
465 :     break;
466 : albeu 315
467 : edgomez 851 case XVID_CSP_ABGR :
468 :     safe_packed_conv(
469 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
470 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
471 :     interlacing?abgri_to_yv12 :abgr_to_yv12,
472 :     interlacing?abgri_to_yv12_c:abgr_to_yv12_c, 4);
473 :     break;
474 : albeu 315
475 : edgomez 851 case XVID_CSP_RGBA :
476 :     safe_packed_conv(
477 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
478 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
479 :     interlacing?rgbai_to_yv12 :rgba_to_yv12,
480 :     interlacing?rgbai_to_yv12_c:rgba_to_yv12_c, 4);
481 :     break;
482 : edgomez 1382
483 :     case XVID_CSP_ARGB:
484 :     safe_packed_conv(
485 :     src[0], src_stride[0], image->y, image->u, image->v,
486 :     edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
487 :     interlacing?argbi_to_yv12 : argb_to_yv12,
488 :     interlacing?argbi_to_yv12_c: argb_to_yv12_c, 4);
489 :     break;
490 : albeu 315
491 :     case XVID_CSP_YUY2:
492 : edgomez 851 safe_packed_conv(
493 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
494 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
495 :     interlacing?yuyvi_to_yv12 :yuyv_to_yv12,
496 :     interlacing?yuyvi_to_yv12_c:yuyv_to_yv12_c, 2);
497 :     break;
498 : albeu 315
499 :     case XVID_CSP_YVYU: /* u/v swapped */
500 : edgomez 851 safe_packed_conv(
501 : edgomez 1382 src[0], src_stride[0], image->y, image->v, image->u,
502 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
503 :     interlacing?yuyvi_to_yv12 :yuyv_to_yv12,
504 :     interlacing?yuyvi_to_yv12_c:yuyv_to_yv12_c, 2);
505 :     break;
506 : albeu 315
507 :     case XVID_CSP_UYVY:
508 : edgomez 851 safe_packed_conv(
509 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
510 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
511 :     interlacing?uyvyi_to_yv12 :uyvy_to_yv12,
512 :     interlacing?uyvyi_to_yv12_c:uyvy_to_yv12_c, 2);
513 :     break;
514 : albeu 315
515 : edgomez 1382 case XVID_CSP_I420: /* YCbCr == YUV == internal colorspace for MPEG */
516 : edgomez 851 yv12_to_yv12(image->y, image->u, image->v, edged_width, edged_width2,
517 : edgomez 1382 src[0], src[0] + src_stride[0]*height, src[0] + src_stride[0]*height + (src_stride[0]/2)*height2,
518 :     src_stride[0], src_stride[0]/2, width, height, (csp & XVID_CSP_VFLIP));
519 :     break;
520 :    
521 :     case XVID_CSP_YV12: /* YCrCb == YVA == U and V plane swapped */
522 : edgomez 851 yv12_to_yv12(image->y, image->v, image->u, edged_width, edged_width2,
523 : edgomez 1382 src[0], src[0] + src_stride[0]*height, src[0] + src_stride[0]*height + (src_stride[0]/2)*height2,
524 :     src_stride[0], src_stride[0]/2, width, height, (csp & XVID_CSP_VFLIP));
525 : edgomez 851 break;
526 :    
527 : edgomez 1382 case XVID_CSP_PLANAR: /* YCbCr with arbitrary pointers and different strides for Y and UV */
528 :     yv12_to_yv12(image->y, image->u, image->v, edged_width, edged_width2,
529 :     src[0], src[1], src[2], src_stride[0], src_stride[1], /* v: dst_stride[2] not yet supported */
530 :     width, height, (csp & XVID_CSP_VFLIP));
531 : edgomez 851 break;
532 : albeu 315
533 :     case XVID_CSP_NULL:
534 :     break;
535 :    
536 : edgomez 851 default :
537 :     return -1;
538 : albeu 315 }
539 :    
540 : edgomez 851
541 :     /* pad out image when the width and/or height is not a multiple of 16 */
542 :    
543 :     if (width & 15)
544 :     {
545 :     int i;
546 :     int pad_width = 16 - (width&15);
547 :     for (i = 0; i < height; i++)
548 :     {
549 : edgomez 1382 memset(image->y + i*edged_width + width,
550 : edgomez 851 *(image->y + i*edged_width + width - 1), pad_width);
551 :     }
552 :     for (i = 0; i < height/2; i++)
553 :     {
554 : edgomez 1382 memset(image->u + i*edged_width2 + width2,
555 : edgomez 851 *(image->u + i*edged_width2 + width2 - 1),pad_width/2);
556 : edgomez 1382 memset(image->v + i*edged_width2 + width2,
557 : edgomez 851 *(image->v + i*edged_width2 + width2 - 1),pad_width/2);
558 :     }
559 :     }
560 :    
561 :     if (height & 15)
562 :     {
563 : edgomez 1382 int pad_height = 16 - (height&15);
564 : edgomez 851 int length = ((width+15)/16)*16;
565 :     int i;
566 :     for (i = 0; i < pad_height; i++)
567 :     {
568 :     memcpy(image->y + (height+i)*edged_width,
569 :     image->y + (height-1)*edged_width,length);
570 :     }
571 :    
572 :     for (i = 0; i < pad_height/2; i++)
573 :     {
574 :     memcpy(image->u + (height2+i)*edged_width2,
575 :     image->u + (height2-1)*edged_width2,length/2);
576 :     memcpy(image->v + (height2+i)*edged_width2,
577 :     image->v + (height2-1)*edged_width2,length/2);
578 :     }
579 :     }
580 :    
581 :     /*
582 :     if (interlacing)
583 :     image_printf(image, edged_width, height, 5,5, "[i]");
584 :     image_dump_yuvpgm(image, edged_width, ((width+15)/16)*16, ((height+15)/16)*16, "\\encode.pgm");
585 :     */
586 :     return 0;
587 : albeu 315 }
588 :    
589 :    
590 :    
591 :     int
592 :     image_output(IMAGE * image,
593 :     uint32_t width,
594 :     int height,
595 :     uint32_t edged_width,
596 : edgomez 1382 uint8_t * dst[4],
597 : Skal 1617 int dst_stride[4],
598 : edgomez 851 int csp,
599 :     int interlacing)
600 : albeu 315 {
601 : edgomez 851 const int edged_width2 = edged_width/2;
602 :     int height2 = height/2;
603 :    
604 :     /*
605 :     if (interlacing)
606 :     image_printf(image, edged_width, height, 5,100, "[i]=%i,%i",width,height);
607 :     image_dump_yuvpgm(image, edged_width, width, height, "\\decode.pgm");
608 :     */
609 :    
610 : albeu 315 switch (csp & ~XVID_CSP_VFLIP) {
611 :     case XVID_CSP_RGB555:
612 : edgomez 851 safe_packed_conv(
613 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
614 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
615 :     interlacing?yv12_to_rgb555i :yv12_to_rgb555,
616 :     interlacing?yv12_to_rgb555i_c:yv12_to_rgb555_c, 2);
617 : albeu 315 return 0;
618 :    
619 :     case XVID_CSP_RGB565:
620 : edgomez 851 safe_packed_conv(
621 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
622 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
623 :     interlacing?yv12_to_rgb565i :yv12_to_rgb565,
624 :     interlacing?yv12_to_rgb565i_c:yv12_to_rgb565_c, 2);
625 : albeu 315 return 0;
626 :    
627 : edgomez 1382 case XVID_CSP_BGR:
628 : edgomez 851 safe_packed_conv(
629 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
630 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
631 :     interlacing?yv12_to_bgri :yv12_to_bgr,
632 :     interlacing?yv12_to_bgri_c:yv12_to_bgr_c, 3);
633 : albeu 315 return 0;
634 :    
635 : edgomez 1382 case XVID_CSP_BGRA:
636 : edgomez 851 safe_packed_conv(
637 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
638 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
639 :     interlacing?yv12_to_bgrai :yv12_to_bgra,
640 :     interlacing?yv12_to_bgrai_c:yv12_to_bgra_c, 4);
641 : albeu 315 return 0;
642 :    
643 : edgomez 851 case XVID_CSP_ABGR:
644 :     safe_packed_conv(
645 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
646 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
647 :     interlacing?yv12_to_abgri :yv12_to_abgr,
648 :     interlacing?yv12_to_abgri_c:yv12_to_abgr_c, 4);
649 : albeu 315 return 0;
650 :    
651 : edgomez 851 case XVID_CSP_RGBA:
652 :     safe_packed_conv(
653 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
654 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
655 :     interlacing?yv12_to_rgbai :yv12_to_rgba,
656 :     interlacing?yv12_to_rgbai_c:yv12_to_rgba_c, 4);
657 : albeu 315 return 0;
658 :    
659 : edgomez 1382 case XVID_CSP_ARGB:
660 :     safe_packed_conv(
661 :     dst[0], dst_stride[0], image->y, image->u, image->v,
662 :     edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
663 :     interlacing?yv12_to_argbi :yv12_to_argb,
664 :     interlacing?yv12_to_argbi_c:yv12_to_argb_c, 4);
665 :     return 0;
666 :    
667 : albeu 315 case XVID_CSP_YUY2:
668 : edgomez 851 safe_packed_conv(
669 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
670 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
671 :     interlacing?yv12_to_yuyvi :yv12_to_yuyv,
672 :     interlacing?yv12_to_yuyvi_c:yv12_to_yuyv_c, 2);
673 : albeu 315 return 0;
674 :    
675 : edgomez 1382 case XVID_CSP_YVYU: /* u,v swapped */
676 : edgomez 851 safe_packed_conv(
677 : edgomez 1382 dst[0], dst_stride[0], image->y, image->v, image->u,
678 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
679 :     interlacing?yv12_to_yuyvi :yv12_to_yuyv,
680 :     interlacing?yv12_to_yuyvi_c:yv12_to_yuyv_c, 2);
681 : albeu 315 return 0;
682 :    
683 :     case XVID_CSP_UYVY:
684 : edgomez 851 safe_packed_conv(
685 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
686 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
687 :     interlacing?yv12_to_uyvyi :yv12_to_uyvy,
688 :     interlacing?yv12_to_uyvyi_c:yv12_to_uyvy_c, 2);
689 : albeu 315 return 0;
690 :    
691 : edgomez 1382 case XVID_CSP_I420: /* YCbCr == YUV == internal colorspace for MPEG */
692 :     yv12_to_yv12(dst[0], dst[0] + dst_stride[0]*height, dst[0] + dst_stride[0]*height + (dst_stride[0]/2)*height2,
693 :     dst_stride[0], dst_stride[0]/2,
694 : edgomez 851 image->y, image->u, image->v, edged_width, edged_width2,
695 :     width, height, (csp & XVID_CSP_VFLIP));
696 :     return 0;
697 :    
698 : edgomez 1382 case XVID_CSP_YV12: /* YCrCb == YVU == U and V plane swapped */
699 :     yv12_to_yv12(dst[0], dst[0] + dst_stride[0]*height, dst[0] + dst_stride[0]*height + (dst_stride[0]/2)*height2,
700 :     dst_stride[0], dst_stride[0]/2,
701 : edgomez 851 image->y, image->v, image->u, edged_width, edged_width2,
702 :     width, height, (csp & XVID_CSP_VFLIP));
703 :     return 0;
704 :    
705 : edgomez 1382 case XVID_CSP_PLANAR: /* YCbCr with arbitrary pointers and different strides for Y and UV */
706 :     yv12_to_yv12(dst[0], dst[1], dst[2],
707 :     dst_stride[0], dst_stride[1], /* v: dst_stride[2] not yet supported */
708 :     image->y, image->u, image->v, edged_width, edged_width2,
709 :     width, height, (csp & XVID_CSP_VFLIP));
710 : albeu 315 return 0;
711 :    
712 : edgomez 1382 case XVID_CSP_INTERNAL :
713 :     dst[0] = image->y;
714 :     dst[1] = image->u;
715 :     dst[2] = image->v;
716 :     dst_stride[0] = edged_width;
717 :     dst_stride[1] = edged_width/2;
718 :     dst_stride[2] = edged_width/2;
719 :     return 0;
720 :    
721 : albeu 315 case XVID_CSP_NULL:
722 : edgomez 1382 case XVID_CSP_SLICE:
723 : albeu 315 return 0;
724 :    
725 :     }
726 :    
727 :     return -1;
728 :     }
729 :    
730 :     float
731 :     image_psnr(IMAGE * orig_image,
732 :     IMAGE * recon_image,
733 :     uint16_t stride,
734 :     uint16_t width,
735 :     uint16_t height)
736 :     {
737 :     int32_t diff, x, y, quad = 0;
738 :     uint8_t *orig = orig_image->y;
739 :     uint8_t *recon = recon_image->y;
740 :     float psnr_y;
741 :    
742 :     for (y = 0; y < height; y++) {
743 :     for (x = 0; x < width; x++) {
744 :     diff = *(orig + x) - *(recon + x);
745 :     quad += diff * diff;
746 :     }
747 :     orig += stride;
748 :     recon += stride;
749 :     }
750 :    
751 :     psnr_y = (float) quad / (float) (width * height);
752 :    
753 :     if (psnr_y) {
754 :     psnr_y = (float) (255 * 255) / psnr_y;
755 :     psnr_y = 10 * (float) log10(psnr_y);
756 :     } else
757 :     psnr_y = (float) 99.99;
758 :    
759 :     return psnr_y;
760 :     }
761 :    
762 :    
763 : edgomez 851 float sse_to_PSNR(long sse, int pixels)
764 :     {
765 :     if (sse==0)
766 :     return 99.99F;
767 :    
768 : edgomez 1382 return 48.131F - 10*(float)log10((float)sse/(float)(pixels)); /* log10(255*255)=4.8131 */
769 : edgomez 851
770 :     }
771 :    
772 : edgomez 1424 long plane_sse(uint8_t *orig,
773 :     uint8_t *recon,
774 :     uint16_t stride,
775 :     uint16_t width,
776 :     uint16_t height)
777 : edgomez 851 {
778 : edgomez 1424 int y, bwidth, bheight;
779 :     long sse = 0;
780 : edgomez 851
781 : edgomez 1424 bwidth = width & (~0x07);
782 :     bheight = height & (~0x07);
783 :    
784 :     /* Compute the 8x8 integer part */
785 :     for (y = 0; y<bheight; y += 8) {
786 :     int x;
787 :    
788 :     /* Compute sse for the band */
789 :     for (x = 0; x<bwidth; x += 8)
790 :     sse += sse8_8bit(orig + x, recon + x, stride);
791 :    
792 :     /* remaining pixels of the 8 pixels high band */
793 :     for (x = bwidth; x < width; x++) {
794 :     int diff;
795 :     diff = *(orig + 0*stride + x) - *(recon + 0*stride + x);
796 :     sse += diff * diff;
797 :     diff = *(orig + 1*stride + x) - *(recon + 1*stride + x);
798 :     sse += diff * diff;
799 :     diff = *(orig + 2*stride + x) - *(recon + 2*stride + x);
800 :     sse += diff * diff;
801 :     diff = *(orig + 3*stride + x) - *(recon + 3*stride + x);
802 :     sse += diff * diff;
803 :     diff = *(orig + 4*stride + x) - *(recon + 4*stride + x);
804 :     sse += diff * diff;
805 :     diff = *(orig + 5*stride + x) - *(recon + 5*stride + x);
806 :     sse += diff * diff;
807 :     diff = *(orig + 6*stride + x) - *(recon + 6*stride + x);
808 :     sse += diff * diff;
809 :     diff = *(orig + 7*stride + x) - *(recon + 7*stride + x);
810 :     sse += diff * diff;
811 :     }
812 :    
813 :     orig += 8*stride;
814 :     recon += 8*stride;
815 :     }
816 :    
817 :     /* Compute the down rectangle sse */
818 :     for (y = bheight; y < height; y++) {
819 :     int x;
820 : edgomez 851 for (x = 0; x < width; x++) {
821 : edgomez 1424 int diff;
822 : edgomez 851 diff = *(orig + x) - *(recon + x);
823 :     sse += diff * diff;
824 :     }
825 :     orig += stride;
826 :     recon += stride;
827 :     }
828 : edgomez 1424
829 :     return (sse);
830 : edgomez 851 }
831 :    
832 : edgomez 1382 #if 0
833 : edgomez 851
834 : albeu 315 #include <stdio.h>
835 :     #include <string.h>
836 :    
837 :     int image_dump_pgm(uint8_t * bmp, uint32_t width, uint32_t height, char * filename)
838 :     {
839 :     FILE * f;
840 :     char hdr[1024];
841 : edgomez 1382
842 : albeu 315 f = fopen(filename, "wb");
843 :     if ( f == NULL)
844 :     {
845 :     return -1;
846 :     }
847 :     sprintf(hdr, "P5\n#xvid\n%i %i\n255\n", width, height);
848 :     fwrite(hdr, strlen(hdr), 1, f);
849 :     fwrite(bmp, width, height, f);
850 :     fclose(f);
851 :    
852 :     return 0;
853 :     }
854 :    
855 :    
856 : edgomez 1382 /* dump image+edges to yuv pgm files */
857 : albeu 315
858 :     int image_dump(IMAGE * image, uint32_t edged_width, uint32_t edged_height, char * path, int number)
859 :     {
860 :     char filename[1024];
861 :    
862 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'y');
863 :     image_dump_pgm(
864 :     image->y - (EDGE_SIZE * edged_width + EDGE_SIZE),
865 :     edged_width, edged_height, filename);
866 :    
867 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'u');
868 :     image_dump_pgm(
869 :     image->u - (EDGE_SIZE2 * edged_width / 2 + EDGE_SIZE2),
870 :     edged_width / 2, edged_height / 2, filename);
871 :    
872 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'v');
873 :     image_dump_pgm(
874 :     image->v - (EDGE_SIZE2 * edged_width / 2 + EDGE_SIZE2),
875 :     edged_width / 2, edged_height / 2, filename);
876 :    
877 :     return 0;
878 :     }
879 : edgomez 1382 #endif
880 : albeu 315
881 :    
882 :    
883 :     /* dump image to yuvpgm file */
884 :    
885 :     #include <stdio.h>
886 :    
887 :     int
888 :     image_dump_yuvpgm(const IMAGE * image,
889 :     const uint32_t edged_width,
890 :     const uint32_t width,
891 :     const uint32_t height,
892 :     char *filename)
893 :     {
894 :     FILE *f;
895 :     char hdr[1024];
896 :     uint32_t i;
897 :     uint8_t *bmp1;
898 :     uint8_t *bmp2;
899 :    
900 :    
901 :     f = fopen(filename, "wb");
902 :     if (f == NULL) {
903 :     return -1;
904 :     }
905 :     sprintf(hdr, "P5\n#xvid\n%i %i\n255\n", width, (3 * height) / 2);
906 :     fwrite(hdr, strlen(hdr), 1, f);
907 :    
908 :     bmp1 = image->y;
909 :     for (i = 0; i < height; i++) {
910 :     fwrite(bmp1, width, 1, f);
911 :     bmp1 += edged_width;
912 :     }
913 :    
914 :     bmp1 = image->u;
915 :     bmp2 = image->v;
916 :     for (i = 0; i < height / 2; i++) {
917 :     fwrite(bmp1, width / 2, 1, f);
918 :     fwrite(bmp2, width / 2, 1, f);
919 :     bmp1 += edged_width / 2;
920 :     bmp2 += edged_width / 2;
921 :     }
922 :    
923 :     fclose(f);
924 :     return 0;
925 :     }
926 :    
927 :    
928 :     float
929 :     image_mad(const IMAGE * img1,
930 :     const IMAGE * img2,
931 :     uint32_t stride,
932 :     uint32_t width,
933 :     uint32_t height)
934 :     {
935 :     const uint32_t stride2 = stride / 2;
936 :     const uint32_t width2 = width / 2;
937 :     const uint32_t height2 = height / 2;
938 :    
939 :     uint32_t x, y;
940 :     uint32_t sum = 0;
941 :    
942 :     for (y = 0; y < height; y++)
943 :     for (x = 0; x < width; x++)
944 : edgomez 1382 sum += abs(img1->y[x + y * stride] - img2->y[x + y * stride]);
945 : albeu 315
946 :     for (y = 0; y < height2; y++)
947 :     for (x = 0; x < width2; x++)
948 : edgomez 1382 sum += abs(img1->u[x + y * stride2] - img2->u[x + y * stride2]);
949 : albeu 315
950 :     for (y = 0; y < height2; y++)
951 :     for (x = 0; x < width2; x++)
952 : edgomez 1382 sum += abs(img1->v[x + y * stride2] - img2->v[x + y * stride2]);
953 : albeu 315
954 :     return (float) sum / (width * height * 3 / 2);
955 :     }
956 :    
957 :     void
958 : suxen_drol 1631 output_slice(IMAGE * cur, int stride, int width, xvid_image_t* out_frm, int mbx, int mby,int mbl) {
959 : albeu 315 uint8_t *dY,*dU,*dV,*sY,*sU,*sV;
960 : suxen_drol 1631 int stride2 = stride >> 1;
961 : albeu 315 int w = mbl << 4, w2,i;
962 :    
963 :     if(w > width)
964 :     w = width;
965 :     w2 = w >> 1;
966 : suxen_drol 323
967 : edgomez 1382 dY = (uint8_t*)out_frm->plane[0] + (mby << 4) * out_frm->stride[0] + (mbx << 4);
968 :     dU = (uint8_t*)out_frm->plane[1] + (mby << 3) * out_frm->stride[1] + (mbx << 3);
969 :     dV = (uint8_t*)out_frm->plane[2] + (mby << 3) * out_frm->stride[2] + (mbx << 3);
970 : suxen_drol 1631 sY = cur->y + (mby << 4) * stride + (mbx << 4);
971 :     sU = cur->u + (mby << 3) * stride2 + (mbx << 3);
972 :     sV = cur->v + (mby << 3) * stride2 + (mbx << 3);
973 : albeu 315
974 :     for(i = 0 ; i < 16 ; i++) {
975 :     memcpy(dY,sY,w);
976 : edgomez 1382 dY += out_frm->stride[0];
977 : suxen_drol 1631 sY += stride;
978 : suxen_drol 323 }
979 : albeu 315 for(i = 0 ; i < 8 ; i++) {
980 :     memcpy(dU,sU,w2);
981 : edgomez 1382 dU += out_frm->stride[1];
982 : suxen_drol 1631 sU += stride2;
983 : albeu 315 }
984 :     for(i = 0 ; i < 8 ; i++) {
985 :     memcpy(dV,sV,w2);
986 : edgomez 1382 dV += out_frm->stride[2];
987 : suxen_drol 1631 sV += stride2;
988 : albeu 315 }
989 :     }
990 : edgomez 851
991 :    
992 :     void
993 :     image_clear(IMAGE * img, int width, int height, int edged_width,
994 :     int y, int u, int v)
995 :     {
996 :     uint8_t * p;
997 :     int i;
998 :    
999 :     p = img->y;
1000 :     for (i = 0; i < height; i++) {
1001 :     memset(p, y, width);
1002 :     p += edged_width;
1003 :     }
1004 :    
1005 :     p = img->u;
1006 :     for (i = 0; i < height/2; i++) {
1007 :     memset(p, u, width/2);
1008 :     p += edged_width/2;
1009 :     }
1010 :    
1011 :     p = img->v;
1012 :     for (i = 0; i < height/2; i++) {
1013 :     memset(p, v, width/2);
1014 :     p += edged_width/2;
1015 :     }
1016 :     }

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4