[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 1664 - (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 1664 * $Id: image.c,v 1.33 2005-12-17 11:24:32 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 : edgomez 1382 /* bframe encoding requires image-based u,v interpolation */
240 : albeu 315 void
241 :     image_interpolate(const IMAGE * refn,
242 :     IMAGE * refh,
243 :     IMAGE * refv,
244 :     IMAGE * refhv,
245 :     uint32_t edged_width,
246 :     uint32_t edged_height,
247 : edgomez 851 uint32_t quarterpel,
248 : albeu 315 uint32_t rounding)
249 :     {
250 : edgomez 1382 const uint32_t offset = EDGE_SIZE2 * (edged_width + 1); /* we only interpolate half of the edge area */
251 : albeu 315 const uint32_t stride_add = 7 * edged_width;
252 : syskin 1664
253 : albeu 315 uint8_t *n_ptr, *h_ptr, *v_ptr, *hv_ptr;
254 :     uint32_t x, y;
255 :    
256 :    
257 :     n_ptr = refn->y;
258 :     h_ptr = refh->y;
259 :     v_ptr = refv->y;
260 :    
261 :     n_ptr -= offset;
262 :     h_ptr -= offset;
263 :     v_ptr -= offset;
264 :    
265 : edgomez 1382 /* Note we initialize the hv pointer later, as we can optimize code a bit
266 :     * doing it down to up in quarterpel and up to down in halfpel */
267 : edgomez 851 if(quarterpel) {
268 : edgomez 1382
269 : edgomez 851 for (y = 0; y < (edged_height - EDGE_SIZE); y += 8) {
270 :     for (x = 0; x < (edged_width - EDGE_SIZE); x += 8) {
271 :     interpolate8x8_6tap_lowpass_h(h_ptr, n_ptr, edged_width, rounding);
272 :     interpolate8x8_6tap_lowpass_v(v_ptr, n_ptr, edged_width, rounding);
273 : albeu 315
274 : edgomez 851 n_ptr += 8;
275 :     h_ptr += 8;
276 :     v_ptr += 8;
277 :     }
278 : edgomez 1382
279 : edgomez 851 n_ptr += EDGE_SIZE;
280 :     h_ptr += EDGE_SIZE;
281 :     v_ptr += EDGE_SIZE;
282 :    
283 :     h_ptr += stride_add;
284 :     v_ptr += stride_add;
285 :     n_ptr += stride_add;
286 :     }
287 :    
288 : edgomez 1382 h_ptr = refh->y + (edged_height - EDGE_SIZE - EDGE_SIZE2)*edged_width - EDGE_SIZE2;
289 :     hv_ptr = refhv->y + (edged_height - EDGE_SIZE - EDGE_SIZE2)*edged_width - EDGE_SIZE2;
290 : edgomez 851
291 :     for (y = 0; y < (edged_height - EDGE_SIZE); y = y + 8) {
292 : edgomez 1382 hv_ptr -= stride_add;
293 :     h_ptr -= stride_add;
294 :     hv_ptr -= EDGE_SIZE;
295 :     h_ptr -= EDGE_SIZE;
296 :    
297 : edgomez 851 for (x = 0; x < (edged_width - EDGE_SIZE); x = x + 8) {
298 : edgomez 1382 hv_ptr -= 8;
299 :     h_ptr -= 8;
300 : edgomez 851 interpolate8x8_6tap_lowpass_v(hv_ptr, h_ptr, edged_width, rounding);
301 :     }
302 :     }
303 : edgomez 1382 } else {
304 : edgomez 851
305 : edgomez 1382 hv_ptr = refhv->y;
306 :     hv_ptr -= offset;
307 :    
308 : edgomez 851 for (y = 0; y < (edged_height - EDGE_SIZE); y += 8) {
309 :     for (x = 0; x < (edged_width - EDGE_SIZE); x += 8) {
310 :     interpolate8x8_halfpel_h(h_ptr, n_ptr, edged_width, rounding);
311 :     interpolate8x8_halfpel_v(v_ptr, n_ptr, edged_width, rounding);
312 :     interpolate8x8_halfpel_hv(hv_ptr, n_ptr, edged_width, rounding);
313 :    
314 :     n_ptr += 8;
315 :     h_ptr += 8;
316 :     v_ptr += 8;
317 :     hv_ptr += 8;
318 :     }
319 : edgomez 1382
320 : edgomez 851 h_ptr += EDGE_SIZE;
321 :     v_ptr += EDGE_SIZE;
322 :     hv_ptr += EDGE_SIZE;
323 :     n_ptr += EDGE_SIZE;
324 :    
325 :     h_ptr += stride_add;
326 :     v_ptr += stride_add;
327 :     hv_ptr += stride_add;
328 :     n_ptr += stride_add;
329 :     }
330 :     }
331 : albeu 315 }
332 :    
333 :    
334 : edgomez 851 /*
335 :     chroma optimize filter, invented by mf
336 :     a chroma pixel is average from the surrounding pixels, when the
337 :     correpsonding luma pixels are pure black or white.
338 :     */
339 :    
340 :     void
341 :     image_chroma_optimize(IMAGE * img, int width, int height, int edged_width)
342 :     {
343 :     int x,y;
344 :     int pixels = 0;
345 :    
346 :     for (y = 1; y < height/2 - 1; y++)
347 :     for (x = 1; x < width/2 - 1; x++)
348 :     {
349 :     #define IS_PURE(a) ((a)<=16||(a)>=235)
350 :     #define IMG_Y(Y,X) img->y[(Y)*edged_width + (X)]
351 :     #define IMG_U(Y,X) img->u[(Y)*edged_width/2 + (X)]
352 :     #define IMG_V(Y,X) img->v[(Y)*edged_width/2 + (X)]
353 :    
354 : edgomez 1382 if (IS_PURE(IMG_Y(y*2 ,x*2 )) &&
355 : edgomez 851 IS_PURE(IMG_Y(y*2 ,x*2+1)) &&
356 : edgomez 1382 IS_PURE(IMG_Y(y*2+1,x*2 )) &&
357 : edgomez 851 IS_PURE(IMG_Y(y*2+1,x*2+1)))
358 :     {
359 :     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;
360 :     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;
361 :     pixels++;
362 :     }
363 :    
364 :     #undef IS_PURE
365 :     #undef IMG_Y
366 :     #undef IMG_U
367 :     #undef IMG_V
368 :     }
369 : edgomez 1382
370 :     DPRINTF(XVID_DEBUG_DEBUG,"chroma_optimized_pixels = %i/%i\n", pixels, width*height/4);
371 : edgomez 851 }
372 :    
373 :    
374 :    
375 :    
376 :    
377 :     /*
378 :     perform safe packed colorspace conversion, by splitting
379 :     the image up into an optimized area (pixel width divisible by 16),
380 :     and two unoptimized/plain-c areas (pixel width divisible by 2)
381 :     */
382 :    
383 : edgomez 1382 static void
384 : edgomez 851 safe_packed_conv(uint8_t * x_ptr, int x_stride,
385 :     uint8_t * y_ptr, uint8_t * u_ptr, uint8_t * v_ptr,
386 :     int y_stride, int uv_stride,
387 :     int width, int height, int vflip,
388 :     packedFunc * func_opt, packedFunc func_c, int size)
389 :     {
390 :     int width_opt, width_c;
391 :    
392 :     if (func_opt != func_c && x_stride < size*((width+15)/16)*16)
393 :     {
394 :     width_opt = width & (~15);
395 :     width_c = width - width_opt;
396 :     }
397 :     else
398 :     {
399 :     width_opt = width;
400 :     width_c = 0;
401 :     }
402 :    
403 :     func_opt(x_ptr, x_stride,
404 :     y_ptr, u_ptr, v_ptr, y_stride, uv_stride,
405 :     width_opt, height, vflip);
406 :    
407 :     if (width_c)
408 :     {
409 :     func_c(x_ptr + size*width_opt, x_stride,
410 :     y_ptr + width_opt, u_ptr + width_opt/2, v_ptr + width_opt/2,
411 :     y_stride, uv_stride, width_c, height, vflip);
412 :     }
413 :     }
414 :    
415 :    
416 :    
417 : albeu 315 int
418 :     image_input(IMAGE * image,
419 :     uint32_t width,
420 :     int height,
421 :     uint32_t edged_width,
422 : edgomez 1382 uint8_t * src[4],
423 :     int src_stride[4],
424 : edgomez 851 int csp,
425 :     int interlacing)
426 : albeu 315 {
427 : edgomez 851 const int edged_width2 = edged_width/2;
428 :     const int width2 = width/2;
429 :     const int height2 = height/2;
430 : edgomez 1382 #if 0
431 :     const int height_signed = (csp & XVID_CSP_VFLIP) ? -height : height;
432 :     #endif
433 : albeu 315
434 :     switch (csp & ~XVID_CSP_VFLIP) {
435 :     case XVID_CSP_RGB555:
436 : edgomez 851 safe_packed_conv(
437 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
438 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
439 :     interlacing?rgb555i_to_yv12 :rgb555_to_yv12,
440 :     interlacing?rgb555i_to_yv12_c:rgb555_to_yv12_c, 2);
441 :     break;
442 : albeu 315
443 :     case XVID_CSP_RGB565:
444 : edgomez 851 safe_packed_conv(
445 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
446 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
447 :     interlacing?rgb565i_to_yv12 :rgb565_to_yv12,
448 :     interlacing?rgb565i_to_yv12_c:rgb565_to_yv12_c, 2);
449 :     break;
450 : albeu 315
451 :    
452 : edgomez 1382 case XVID_CSP_BGR:
453 : edgomez 851 safe_packed_conv(
454 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
455 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
456 :     interlacing?bgri_to_yv12 :bgr_to_yv12,
457 :     interlacing?bgri_to_yv12_c:bgr_to_yv12_c, 3);
458 :     break;
459 : albeu 315
460 : edgomez 1382 case XVID_CSP_BGRA:
461 : edgomez 851 safe_packed_conv(
462 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
463 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
464 :     interlacing?bgrai_to_yv12 :bgra_to_yv12,
465 :     interlacing?bgrai_to_yv12_c:bgra_to_yv12_c, 4);
466 :     break;
467 : albeu 315
468 : edgomez 851 case XVID_CSP_ABGR :
469 :     safe_packed_conv(
470 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
471 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
472 :     interlacing?abgri_to_yv12 :abgr_to_yv12,
473 :     interlacing?abgri_to_yv12_c:abgr_to_yv12_c, 4);
474 :     break;
475 : albeu 315
476 : edgomez 851 case XVID_CSP_RGBA :
477 :     safe_packed_conv(
478 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
479 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
480 :     interlacing?rgbai_to_yv12 :rgba_to_yv12,
481 :     interlacing?rgbai_to_yv12_c:rgba_to_yv12_c, 4);
482 :     break;
483 : edgomez 1382
484 :     case XVID_CSP_ARGB:
485 :     safe_packed_conv(
486 :     src[0], src_stride[0], image->y, image->u, image->v,
487 :     edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
488 :     interlacing?argbi_to_yv12 : argb_to_yv12,
489 :     interlacing?argbi_to_yv12_c: argb_to_yv12_c, 4);
490 :     break;
491 : albeu 315
492 :     case XVID_CSP_YUY2:
493 : edgomez 851 safe_packed_conv(
494 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
495 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
496 :     interlacing?yuyvi_to_yv12 :yuyv_to_yv12,
497 :     interlacing?yuyvi_to_yv12_c:yuyv_to_yv12_c, 2);
498 :     break;
499 : albeu 315
500 :     case XVID_CSP_YVYU: /* u/v swapped */
501 : edgomez 851 safe_packed_conv(
502 : edgomez 1382 src[0], src_stride[0], image->y, image->v, image->u,
503 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
504 :     interlacing?yuyvi_to_yv12 :yuyv_to_yv12,
505 :     interlacing?yuyvi_to_yv12_c:yuyv_to_yv12_c, 2);
506 :     break;
507 : albeu 315
508 :     case XVID_CSP_UYVY:
509 : edgomez 851 safe_packed_conv(
510 : edgomez 1382 src[0], src_stride[0], image->y, image->u, image->v,
511 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
512 :     interlacing?uyvyi_to_yv12 :uyvy_to_yv12,
513 :     interlacing?uyvyi_to_yv12_c:uyvy_to_yv12_c, 2);
514 :     break;
515 : albeu 315
516 : edgomez 1382 case XVID_CSP_I420: /* YCbCr == YUV == internal colorspace for MPEG */
517 : edgomez 851 yv12_to_yv12(image->y, image->u, image->v, edged_width, edged_width2,
518 : edgomez 1382 src[0], src[0] + src_stride[0]*height, src[0] + src_stride[0]*height + (src_stride[0]/2)*height2,
519 :     src_stride[0], src_stride[0]/2, width, height, (csp & XVID_CSP_VFLIP));
520 :     break;
521 :    
522 :     case XVID_CSP_YV12: /* YCrCb == YVA == U and V plane swapped */
523 : edgomez 851 yv12_to_yv12(image->y, image->v, image->u, edged_width, edged_width2,
524 : edgomez 1382 src[0], src[0] + src_stride[0]*height, src[0] + src_stride[0]*height + (src_stride[0]/2)*height2,
525 :     src_stride[0], src_stride[0]/2, width, height, (csp & XVID_CSP_VFLIP));
526 : edgomez 851 break;
527 :    
528 : edgomez 1382 case XVID_CSP_PLANAR: /* YCbCr with arbitrary pointers and different strides for Y and UV */
529 :     yv12_to_yv12(image->y, image->u, image->v, edged_width, edged_width2,
530 :     src[0], src[1], src[2], src_stride[0], src_stride[1], /* v: dst_stride[2] not yet supported */
531 :     width, height, (csp & XVID_CSP_VFLIP));
532 : edgomez 851 break;
533 : albeu 315
534 :     case XVID_CSP_NULL:
535 :     break;
536 :    
537 : edgomez 851 default :
538 :     return -1;
539 : albeu 315 }
540 :    
541 : edgomez 851
542 :     /* pad out image when the width and/or height is not a multiple of 16 */
543 :    
544 :     if (width & 15)
545 :     {
546 :     int i;
547 :     int pad_width = 16 - (width&15);
548 :     for (i = 0; i < height; i++)
549 :     {
550 : edgomez 1382 memset(image->y + i*edged_width + width,
551 : edgomez 851 *(image->y + i*edged_width + width - 1), pad_width);
552 :     }
553 :     for (i = 0; i < height/2; i++)
554 :     {
555 : edgomez 1382 memset(image->u + i*edged_width2 + width2,
556 : edgomez 851 *(image->u + i*edged_width2 + width2 - 1),pad_width/2);
557 : edgomez 1382 memset(image->v + i*edged_width2 + width2,
558 : edgomez 851 *(image->v + i*edged_width2 + width2 - 1),pad_width/2);
559 :     }
560 :     }
561 :    
562 :     if (height & 15)
563 :     {
564 : edgomez 1382 int pad_height = 16 - (height&15);
565 : edgomez 851 int length = ((width+15)/16)*16;
566 :     int i;
567 :     for (i = 0; i < pad_height; i++)
568 :     {
569 :     memcpy(image->y + (height+i)*edged_width,
570 :     image->y + (height-1)*edged_width,length);
571 :     }
572 :    
573 :     for (i = 0; i < pad_height/2; i++)
574 :     {
575 :     memcpy(image->u + (height2+i)*edged_width2,
576 :     image->u + (height2-1)*edged_width2,length/2);
577 :     memcpy(image->v + (height2+i)*edged_width2,
578 :     image->v + (height2-1)*edged_width2,length/2);
579 :     }
580 :     }
581 :    
582 :     /*
583 :     if (interlacing)
584 :     image_printf(image, edged_width, height, 5,5, "[i]");
585 :     image_dump_yuvpgm(image, edged_width, ((width+15)/16)*16, ((height+15)/16)*16, "\\encode.pgm");
586 :     */
587 :     return 0;
588 : albeu 315 }
589 :    
590 :    
591 :    
592 :     int
593 :     image_output(IMAGE * image,
594 :     uint32_t width,
595 :     int height,
596 :     uint32_t edged_width,
597 : edgomez 1382 uint8_t * dst[4],
598 : Skal 1617 int dst_stride[4],
599 : edgomez 851 int csp,
600 :     int interlacing)
601 : albeu 315 {
602 : edgomez 851 const int edged_width2 = edged_width/2;
603 :     int height2 = height/2;
604 :    
605 :     /*
606 :     if (interlacing)
607 :     image_printf(image, edged_width, height, 5,100, "[i]=%i,%i",width,height);
608 :     image_dump_yuvpgm(image, edged_width, width, height, "\\decode.pgm");
609 :     */
610 :    
611 : albeu 315 switch (csp & ~XVID_CSP_VFLIP) {
612 :     case XVID_CSP_RGB555:
613 : edgomez 851 safe_packed_conv(
614 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
615 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
616 :     interlacing?yv12_to_rgb555i :yv12_to_rgb555,
617 :     interlacing?yv12_to_rgb555i_c:yv12_to_rgb555_c, 2);
618 : albeu 315 return 0;
619 :    
620 :     case XVID_CSP_RGB565:
621 : edgomez 851 safe_packed_conv(
622 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
623 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
624 :     interlacing?yv12_to_rgb565i :yv12_to_rgb565,
625 :     interlacing?yv12_to_rgb565i_c:yv12_to_rgb565_c, 2);
626 : albeu 315 return 0;
627 :    
628 : edgomez 1382 case XVID_CSP_BGR:
629 : edgomez 851 safe_packed_conv(
630 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
631 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
632 :     interlacing?yv12_to_bgri :yv12_to_bgr,
633 :     interlacing?yv12_to_bgri_c:yv12_to_bgr_c, 3);
634 : albeu 315 return 0;
635 :    
636 : edgomez 1382 case XVID_CSP_BGRA:
637 : edgomez 851 safe_packed_conv(
638 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
639 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
640 :     interlacing?yv12_to_bgrai :yv12_to_bgra,
641 :     interlacing?yv12_to_bgrai_c:yv12_to_bgra_c, 4);
642 : albeu 315 return 0;
643 :    
644 : edgomez 851 case XVID_CSP_ABGR:
645 :     safe_packed_conv(
646 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
647 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
648 :     interlacing?yv12_to_abgri :yv12_to_abgr,
649 :     interlacing?yv12_to_abgri_c:yv12_to_abgr_c, 4);
650 : albeu 315 return 0;
651 :    
652 : edgomez 851 case XVID_CSP_RGBA:
653 :     safe_packed_conv(
654 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
655 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
656 :     interlacing?yv12_to_rgbai :yv12_to_rgba,
657 :     interlacing?yv12_to_rgbai_c:yv12_to_rgba_c, 4);
658 : albeu 315 return 0;
659 :    
660 : edgomez 1382 case XVID_CSP_ARGB:
661 :     safe_packed_conv(
662 :     dst[0], dst_stride[0], image->y, image->u, image->v,
663 :     edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
664 :     interlacing?yv12_to_argbi :yv12_to_argb,
665 :     interlacing?yv12_to_argbi_c:yv12_to_argb_c, 4);
666 :     return 0;
667 :    
668 : albeu 315 case XVID_CSP_YUY2:
669 : edgomez 851 safe_packed_conv(
670 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
671 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
672 :     interlacing?yv12_to_yuyvi :yv12_to_yuyv,
673 :     interlacing?yv12_to_yuyvi_c:yv12_to_yuyv_c, 2);
674 : albeu 315 return 0;
675 :    
676 : edgomez 1382 case XVID_CSP_YVYU: /* u,v swapped */
677 : edgomez 851 safe_packed_conv(
678 : edgomez 1382 dst[0], dst_stride[0], image->y, image->v, image->u,
679 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
680 :     interlacing?yv12_to_yuyvi :yv12_to_yuyv,
681 :     interlacing?yv12_to_yuyvi_c:yv12_to_yuyv_c, 2);
682 : albeu 315 return 0;
683 :    
684 :     case XVID_CSP_UYVY:
685 : edgomez 851 safe_packed_conv(
686 : edgomez 1382 dst[0], dst_stride[0], image->y, image->u, image->v,
687 : edgomez 851 edged_width, edged_width2, width, height, (csp & XVID_CSP_VFLIP),
688 :     interlacing?yv12_to_uyvyi :yv12_to_uyvy,
689 :     interlacing?yv12_to_uyvyi_c:yv12_to_uyvy_c, 2);
690 : albeu 315 return 0;
691 :    
692 : edgomez 1382 case XVID_CSP_I420: /* YCbCr == YUV == internal colorspace for MPEG */
693 :     yv12_to_yv12(dst[0], dst[0] + dst_stride[0]*height, dst[0] + dst_stride[0]*height + (dst_stride[0]/2)*height2,
694 :     dst_stride[0], dst_stride[0]/2,
695 : edgomez 851 image->y, image->u, image->v, edged_width, edged_width2,
696 :     width, height, (csp & XVID_CSP_VFLIP));
697 :     return 0;
698 :    
699 : edgomez 1382 case XVID_CSP_YV12: /* YCrCb == YVU == U and V plane swapped */
700 :     yv12_to_yv12(dst[0], dst[0] + dst_stride[0]*height, dst[0] + dst_stride[0]*height + (dst_stride[0]/2)*height2,
701 :     dst_stride[0], dst_stride[0]/2,
702 : edgomez 851 image->y, image->v, image->u, edged_width, edged_width2,
703 :     width, height, (csp & XVID_CSP_VFLIP));
704 :     return 0;
705 :    
706 : edgomez 1382 case XVID_CSP_PLANAR: /* YCbCr with arbitrary pointers and different strides for Y and UV */
707 :     yv12_to_yv12(dst[0], dst[1], dst[2],
708 :     dst_stride[0], dst_stride[1], /* v: dst_stride[2] not yet supported */
709 :     image->y, image->u, image->v, edged_width, edged_width2,
710 :     width, height, (csp & XVID_CSP_VFLIP));
711 : albeu 315 return 0;
712 :    
713 : edgomez 1382 case XVID_CSP_INTERNAL :
714 :     dst[0] = image->y;
715 :     dst[1] = image->u;
716 :     dst[2] = image->v;
717 :     dst_stride[0] = edged_width;
718 :     dst_stride[1] = edged_width/2;
719 :     dst_stride[2] = edged_width/2;
720 :     return 0;
721 :    
722 : albeu 315 case XVID_CSP_NULL:
723 : edgomez 1382 case XVID_CSP_SLICE:
724 : albeu 315 return 0;
725 :    
726 :     }
727 :    
728 :     return -1;
729 :     }
730 :    
731 :     float
732 :     image_psnr(IMAGE * orig_image,
733 :     IMAGE * recon_image,
734 :     uint16_t stride,
735 :     uint16_t width,
736 :     uint16_t height)
737 :     {
738 :     int32_t diff, x, y, quad = 0;
739 :     uint8_t *orig = orig_image->y;
740 :     uint8_t *recon = recon_image->y;
741 :     float psnr_y;
742 :    
743 :     for (y = 0; y < height; y++) {
744 :     for (x = 0; x < width; x++) {
745 :     diff = *(orig + x) - *(recon + x);
746 :     quad += diff * diff;
747 :     }
748 :     orig += stride;
749 :     recon += stride;
750 :     }
751 :    
752 :     psnr_y = (float) quad / (float) (width * height);
753 :    
754 :     if (psnr_y) {
755 :     psnr_y = (float) (255 * 255) / psnr_y;
756 :     psnr_y = 10 * (float) log10(psnr_y);
757 :     } else
758 :     psnr_y = (float) 99.99;
759 :    
760 :     return psnr_y;
761 :     }
762 :    
763 :    
764 : edgomez 851 float sse_to_PSNR(long sse, int pixels)
765 :     {
766 :     if (sse==0)
767 :     return 99.99F;
768 :    
769 : edgomez 1382 return 48.131F - 10*(float)log10((float)sse/(float)(pixels)); /* log10(255*255)=4.8131 */
770 : edgomez 851
771 :     }
772 :    
773 : edgomez 1424 long plane_sse(uint8_t *orig,
774 :     uint8_t *recon,
775 :     uint16_t stride,
776 :     uint16_t width,
777 :     uint16_t height)
778 : edgomez 851 {
779 : edgomez 1424 int y, bwidth, bheight;
780 :     long sse = 0;
781 : edgomez 851
782 : edgomez 1424 bwidth = width & (~0x07);
783 :     bheight = height & (~0x07);
784 :    
785 :     /* Compute the 8x8 integer part */
786 :     for (y = 0; y<bheight; y += 8) {
787 :     int x;
788 :    
789 :     /* Compute sse for the band */
790 :     for (x = 0; x<bwidth; x += 8)
791 :     sse += sse8_8bit(orig + x, recon + x, stride);
792 :    
793 :     /* remaining pixels of the 8 pixels high band */
794 :     for (x = bwidth; x < width; x++) {
795 :     int diff;
796 :     diff = *(orig + 0*stride + x) - *(recon + 0*stride + x);
797 :     sse += diff * diff;
798 :     diff = *(orig + 1*stride + x) - *(recon + 1*stride + x);
799 :     sse += diff * diff;
800 :     diff = *(orig + 2*stride + x) - *(recon + 2*stride + x);
801 :     sse += diff * diff;
802 :     diff = *(orig + 3*stride + x) - *(recon + 3*stride + x);
803 :     sse += diff * diff;
804 :     diff = *(orig + 4*stride + x) - *(recon + 4*stride + x);
805 :     sse += diff * diff;
806 :     diff = *(orig + 5*stride + x) - *(recon + 5*stride + x);
807 :     sse += diff * diff;
808 :     diff = *(orig + 6*stride + x) - *(recon + 6*stride + x);
809 :     sse += diff * diff;
810 :     diff = *(orig + 7*stride + x) - *(recon + 7*stride + x);
811 :     sse += diff * diff;
812 :     }
813 :    
814 :     orig += 8*stride;
815 :     recon += 8*stride;
816 :     }
817 :    
818 :     /* Compute the down rectangle sse */
819 :     for (y = bheight; y < height; y++) {
820 :     int x;
821 : edgomez 851 for (x = 0; x < width; x++) {
822 : edgomez 1424 int diff;
823 : edgomez 851 diff = *(orig + x) - *(recon + x);
824 :     sse += diff * diff;
825 :     }
826 :     orig += stride;
827 :     recon += stride;
828 :     }
829 : edgomez 1424
830 :     return (sse);
831 : edgomez 851 }
832 :    
833 : edgomez 1382 #if 0
834 : edgomez 851
835 : albeu 315 #include <stdio.h>
836 :     #include <string.h>
837 :    
838 :     int image_dump_pgm(uint8_t * bmp, uint32_t width, uint32_t height, char * filename)
839 :     {
840 :     FILE * f;
841 :     char hdr[1024];
842 : edgomez 1382
843 : albeu 315 f = fopen(filename, "wb");
844 :     if ( f == NULL)
845 :     {
846 :     return -1;
847 :     }
848 :     sprintf(hdr, "P5\n#xvid\n%i %i\n255\n", width, height);
849 :     fwrite(hdr, strlen(hdr), 1, f);
850 :     fwrite(bmp, width, height, f);
851 :     fclose(f);
852 :    
853 :     return 0;
854 :     }
855 :    
856 :    
857 : edgomez 1382 /* dump image+edges to yuv pgm files */
858 : albeu 315
859 :     int image_dump(IMAGE * image, uint32_t edged_width, uint32_t edged_height, char * path, int number)
860 :     {
861 :     char filename[1024];
862 :    
863 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'y');
864 :     image_dump_pgm(
865 :     image->y - (EDGE_SIZE * edged_width + EDGE_SIZE),
866 :     edged_width, edged_height, filename);
867 :    
868 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'u');
869 :     image_dump_pgm(
870 :     image->u - (EDGE_SIZE2 * edged_width / 2 + EDGE_SIZE2),
871 :     edged_width / 2, edged_height / 2, filename);
872 :    
873 :     sprintf(filename, "%s_%i_%c.pgm", path, number, 'v');
874 :     image_dump_pgm(
875 :     image->v - (EDGE_SIZE2 * edged_width / 2 + EDGE_SIZE2),
876 :     edged_width / 2, edged_height / 2, filename);
877 :    
878 :     return 0;
879 :     }
880 : edgomez 1382 #endif
881 : albeu 315
882 :    
883 :    
884 :     /* dump image to yuvpgm file */
885 :    
886 :     #include <stdio.h>
887 :    
888 :     int
889 :     image_dump_yuvpgm(const IMAGE * image,
890 :     const uint32_t edged_width,
891 :     const uint32_t width,
892 :     const uint32_t height,
893 :     char *filename)
894 :     {
895 :     FILE *f;
896 :     char hdr[1024];
897 :     uint32_t i;
898 :     uint8_t *bmp1;
899 :     uint8_t *bmp2;
900 :    
901 :    
902 :     f = fopen(filename, "wb");
903 :     if (f == NULL) {
904 :     return -1;
905 :     }
906 :     sprintf(hdr, "P5\n#xvid\n%i %i\n255\n", width, (3 * height) / 2);
907 :     fwrite(hdr, strlen(hdr), 1, f);
908 :    
909 :     bmp1 = image->y;
910 :     for (i = 0; i < height; i++) {
911 :     fwrite(bmp1, width, 1, f);
912 :     bmp1 += edged_width;
913 :     }
914 :    
915 :     bmp1 = image->u;
916 :     bmp2 = image->v;
917 :     for (i = 0; i < height / 2; i++) {
918 :     fwrite(bmp1, width / 2, 1, f);
919 :     fwrite(bmp2, width / 2, 1, f);
920 :     bmp1 += edged_width / 2;
921 :     bmp2 += edged_width / 2;
922 :     }
923 :    
924 :     fclose(f);
925 :     return 0;
926 :     }
927 :    
928 :    
929 :     float
930 :     image_mad(const IMAGE * img1,
931 :     const IMAGE * img2,
932 :     uint32_t stride,
933 :     uint32_t width,
934 :     uint32_t height)
935 :     {
936 :     const uint32_t stride2 = stride / 2;
937 :     const uint32_t width2 = width / 2;
938 :     const uint32_t height2 = height / 2;
939 :    
940 :     uint32_t x, y;
941 :     uint32_t sum = 0;
942 :    
943 :     for (y = 0; y < height; y++)
944 :     for (x = 0; x < width; x++)
945 : edgomez 1382 sum += abs(img1->y[x + y * stride] - img2->y[x + y * stride]);
946 : albeu 315
947 :     for (y = 0; y < height2; y++)
948 :     for (x = 0; x < width2; x++)
949 : edgomez 1382 sum += abs(img1->u[x + y * stride2] - img2->u[x + y * stride2]);
950 : albeu 315
951 :     for (y = 0; y < height2; y++)
952 :     for (x = 0; x < width2; x++)
953 : edgomez 1382 sum += abs(img1->v[x + y * stride2] - img2->v[x + y * stride2]);
954 : albeu 315
955 :     return (float) sum / (width * height * 3 / 2);
956 :     }
957 :    
958 :     void
959 : suxen_drol 1631 output_slice(IMAGE * cur, int stride, int width, xvid_image_t* out_frm, int mbx, int mby,int mbl) {
960 : albeu 315 uint8_t *dY,*dU,*dV,*sY,*sU,*sV;
961 : suxen_drol 1631 int stride2 = stride >> 1;
962 : albeu 315 int w = mbl << 4, w2,i;
963 :    
964 :     if(w > width)
965 :     w = width;
966 :     w2 = w >> 1;
967 : suxen_drol 323
968 : edgomez 1382 dY = (uint8_t*)out_frm->plane[0] + (mby << 4) * out_frm->stride[0] + (mbx << 4);
969 :     dU = (uint8_t*)out_frm->plane[1] + (mby << 3) * out_frm->stride[1] + (mbx << 3);
970 :     dV = (uint8_t*)out_frm->plane[2] + (mby << 3) * out_frm->stride[2] + (mbx << 3);
971 : suxen_drol 1631 sY = cur->y + (mby << 4) * stride + (mbx << 4);
972 :     sU = cur->u + (mby << 3) * stride2 + (mbx << 3);
973 :     sV = cur->v + (mby << 3) * stride2 + (mbx << 3);
974 : albeu 315
975 :     for(i = 0 ; i < 16 ; i++) {
976 :     memcpy(dY,sY,w);
977 : edgomez 1382 dY += out_frm->stride[0];
978 : suxen_drol 1631 sY += stride;
979 : suxen_drol 323 }
980 : albeu 315 for(i = 0 ; i < 8 ; i++) {
981 :     memcpy(dU,sU,w2);
982 : edgomez 1382 dU += out_frm->stride[1];
983 : suxen_drol 1631 sU += stride2;
984 : albeu 315 }
985 :     for(i = 0 ; i < 8 ; i++) {
986 :     memcpy(dV,sV,w2);
987 : edgomez 1382 dV += out_frm->stride[2];
988 : suxen_drol 1631 sV += stride2;
989 : albeu 315 }
990 :     }
991 : edgomez 851
992 :    
993 :     void
994 :     image_clear(IMAGE * img, int width, int height, int edged_width,
995 :     int y, int u, int v)
996 :     {
997 :     uint8_t * p;
998 :     int i;
999 :    
1000 :     p = img->y;
1001 :     for (i = 0; i < height; i++) {
1002 :     memset(p, y, width);
1003 :     p += edged_width;
1004 :     }
1005 :    
1006 :     p = img->u;
1007 :     for (i = 0; i < height/2; i++) {
1008 :     memset(p, u, width/2);
1009 :     p += edged_width/2;
1010 :     }
1011 :    
1012 :     p = img->v;
1013 :     for (i = 0; i < height/2; i++) {
1014 :     memset(p, v, width/2);
1015 :     p += edged_width/2;
1016 :     }
1017 :     }

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