1 |
/************************************************************************** |
/***************************************************************************** |
2 |
* |
* |
3 |
* XVID MPEG-4 VIDEO CODEC |
* XVID MPEG-4 VIDEO CODEC |
4 |
* image stuff |
* - image module - |
5 |
* |
* |
6 |
* This program is free software; you can redistribute it and/or modify |
* Copyright(C) 2002 Peter Ross <pross@xvid.org> |
7 |
* it under the terms of the GNU General Public License as published by |
* |
8 |
|
* This file is part of XviD, a free MPEG-4 video encoder/decoder |
9 |
|
* |
10 |
|
* XviD is free software; you can redistribute it and/or modify it |
11 |
|
* under the terms of the GNU General Public License as published by |
12 |
* the Free Software Foundation; either version 2 of the License, or |
* the Free Software Foundation; either version 2 of the License, or |
13 |
* (at your option) any later version. |
* (at your option) any later version. |
14 |
* |
* |
19 |
* |
* |
20 |
* You should have received a copy of the GNU General Public License |
* You should have received a copy of the GNU General Public License |
21 |
* along with this program; if not, write to the Free Software |
* along with this program; if not, write to the Free Software |
22 |
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 |
* |
* |
24 |
*************************************************************************/ |
* Under section 8 of the GNU General Public License, the copyright |
25 |
|
* holders of XVID explicitly forbid distribution in the following |
26 |
/************************************************************************** |
* countries: |
27 |
|
* |
28 |
|
* - Japan |
29 |
|
* - United States of America |
30 |
* |
* |
31 |
* History: |
* Linking XviD statically or dynamically with other modules is making a |
32 |
|
* combined work based on XviD. Thus, the terms and conditions of the |
33 |
|
* GNU General Public License cover the whole combination. |
34 |
* |
* |
35 |
* 01.05.2002 BFRAME image-based u,v interpolation |
* As a special exception, the copyright holders of XviD give you |
36 |
* 22.04.2002 added some B-frame support |
* permission to link XviD with independent modules that communicate with |
37 |
* 14.04.2002 added image_dump_yuvpgm(), added image_mad() |
* XviD solely through the VFW1.1 and DShow interfaces, regardless of the |
38 |
* XVID_CSP_USER input support |
* license terms of these independent modules, and to copy and distribute |
39 |
* 09.04.2002 PSNR calculations |
* the resulting combined work under terms of your choice, provided that |
40 |
* 06.04.2002 removed interlaced edging from U,V blocks (as per spec) |
* every copy of the combined work is accompanied by a complete copy of |
41 |
* 26.03.2002 interlacing support (field-based edging in set_edges) |
* the source code of XviD (the version of XviD used to produce the |
42 |
* 26.01.2002 rgb555, rgb565 |
* combined work), being distributed under the terms of the GNU General |
43 |
* 07.01.2001 commented u,v interpolation (not required for uv-block-based) |
* Public License plus this exception. An independent module is a module |
44 |
* 23.12.2001 removed #ifdefs, added function pointers + init_common() |
* which is not derived from or based on XviD. |
|
* 22.12.2001 cpu #ifdefs |
|
|
* 19.12.2001 image_dump(); useful for debugging |
|
|
* 6.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au> |
|
45 |
* |
* |
46 |
*************************************************************************/ |
* Note that people who make modified versions of XviD are not obligated |
47 |
|
* to grant this special exception for their modified versions; it is |
48 |
|
* their choice whether to do so. The GNU General Public License gives |
49 |
|
* permission to release a modified version without this exception; this |
50 |
|
* exception also makes it possible to release a modified version which |
51 |
|
* carries forward this exception. |
52 |
|
* |
53 |
|
* $Id$ |
54 |
|
* |
55 |
|
****************************************************************************/ |
56 |
|
|
57 |
#include <stdlib.h> |
#include <stdlib.h> |
58 |
#include <string.h> // memcpy, memset |
#include <string.h> /* memcpy, memset */ |
59 |
#include <math.h> |
#include <math.h> |
60 |
|
|
61 |
#include "../portab.h" |
#include "../portab.h" |
62 |
#include "../xvid.h" // XVID_CSP_XXX's |
#include "../xvid.h" /* XVID_CSP_XXX's */ |
63 |
#include "image.h" |
#include "image.h" |
64 |
#include "colorspace.h" |
#include "colorspace.h" |
65 |
#include "interpolate8x8.h" |
#include "interpolate8x8.h" |
166 |
uint32_t edged_width, |
uint32_t edged_width, |
167 |
uint32_t edged_height, |
uint32_t edged_height, |
168 |
uint32_t width, |
uint32_t width, |
169 |
uint32_t height, |
uint32_t height) |
|
uint32_t interlacing) |
|
170 |
{ |
{ |
171 |
const uint32_t edged_width2 = edged_width / 2; |
const uint32_t edged_width2 = edged_width / 2; |
172 |
const uint32_t width2 = width / 2; |
const uint32_t width2 = width / 2; |
179 |
src = image->y; |
src = image->y; |
180 |
|
|
181 |
for (i = 0; i < EDGE_SIZE; i++) { |
for (i = 0; i < EDGE_SIZE; i++) { |
|
// if interlacing, edges contain top-most data from each field |
|
|
if (interlacing && (i & 1)) { |
|
|
memset(dst, *(src + edged_width), EDGE_SIZE); |
|
|
memcpy(dst + EDGE_SIZE, src + edged_width, width); |
|
|
memset(dst + edged_width - EDGE_SIZE, |
|
|
*(src + edged_width + width - 1), EDGE_SIZE); |
|
|
} else { |
|
182 |
memset(dst, *src, EDGE_SIZE); |
memset(dst, *src, EDGE_SIZE); |
183 |
memcpy(dst + EDGE_SIZE, src, width); |
memcpy(dst + EDGE_SIZE, src, width); |
184 |
memset(dst + edged_width - EDGE_SIZE, *(src + width - 1), |
memset(dst + edged_width - EDGE_SIZE, *(src + width - 1), |
185 |
EDGE_SIZE); |
EDGE_SIZE); |
|
} |
|
186 |
dst += edged_width; |
dst += edged_width; |
187 |
} |
} |
188 |
|
|
195 |
|
|
196 |
src -= edged_width; |
src -= edged_width; |
197 |
for (i = 0; i < EDGE_SIZE; i++) { |
for (i = 0; i < EDGE_SIZE; i++) { |
|
// if interlacing, edges contain bottom-most data from each field |
|
|
if (interlacing && !(i & 1)) { |
|
|
memset(dst, *(src - edged_width), EDGE_SIZE); |
|
|
memcpy(dst + EDGE_SIZE, src - edged_width, width); |
|
|
memset(dst + edged_width - EDGE_SIZE, |
|
|
*(src - edged_width + width - 1), EDGE_SIZE); |
|
|
} else { |
|
198 |
memset(dst, *src, EDGE_SIZE); |
memset(dst, *src, EDGE_SIZE); |
199 |
memcpy(dst + EDGE_SIZE, src, width); |
memcpy(dst + EDGE_SIZE, src, width); |
200 |
memset(dst + edged_width - EDGE_SIZE, *(src + width - 1), |
memset(dst + edged_width - EDGE_SIZE, *(src + width - 1), |
201 |
EDGE_SIZE); |
EDGE_SIZE); |
|
} |
|
202 |
dst += edged_width; |
dst += edged_width; |
203 |
} |
} |
204 |
|
|
205 |
|
|
206 |
//U |
/*U */ |
207 |
dst = image->u - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2); |
dst = image->u - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2); |
208 |
src = image->u; |
src = image->u; |
209 |
|
|
231 |
} |
} |
232 |
|
|
233 |
|
|
234 |
// V |
/* V */ |
235 |
dst = image->v - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2); |
dst = image->v - (EDGE_SIZE2 + EDGE_SIZE2 * edged_width2); |
236 |
src = image->v; |
src = image->v; |
237 |
|
|
259 |
} |
} |
260 |
} |
} |
261 |
|
|
262 |
// bframe encoding requires image-based u,v interpolation |
/* bframe encoding requires image-based u,v interpolation */ |
263 |
void |
void |
264 |
image_interpolate(const IMAGE * refn, |
image_interpolate(const IMAGE * refn, |
265 |
IMAGE * refh, |
IMAGE * refh, |
272 |
const uint32_t offset = EDGE_SIZE * (edged_width + 1); |
const uint32_t offset = EDGE_SIZE * (edged_width + 1); |
273 |
const uint32_t stride_add = 7 * edged_width; |
const uint32_t stride_add = 7 * edged_width; |
274 |
|
|
|
#ifdef BFRAMES |
|
|
const uint32_t edged_width2 = edged_width / 2; |
|
|
const uint32_t edged_height2 = edged_height / 2; |
|
|
const uint32_t offset2 = EDGE_SIZE2 * (edged_width2 + 1); |
|
|
const uint32_t stride_add2 = 7 * edged_width2; |
|
|
#endif |
|
|
|
|
275 |
uint8_t *n_ptr, *h_ptr, *v_ptr, *hv_ptr; |
uint8_t *n_ptr, *h_ptr, *v_ptr, *hv_ptr; |
276 |
uint32_t x, y; |
uint32_t x, y; |
277 |
|
|
303 |
n_ptr += stride_add; |
n_ptr += stride_add; |
304 |
} |
} |
305 |
|
|
|
#ifdef BFRAMES |
|
|
n_ptr = refn->u; |
|
|
h_ptr = refh->u; |
|
|
v_ptr = refv->u; |
|
|
hv_ptr = refhv->u; |
|
|
|
|
|
n_ptr -= offset2; |
|
|
h_ptr -= offset2; |
|
|
v_ptr -= offset2; |
|
|
hv_ptr -= offset2; |
|
|
|
|
|
for (y = 0; y < edged_height2; y = y + 8) { |
|
|
for (x = 0; x < edged_width2; x = x + 8) { |
|
|
interpolate8x8_halfpel_h(h_ptr, n_ptr, edged_width2, rounding); |
|
|
interpolate8x8_halfpel_v(v_ptr, n_ptr, edged_width2, rounding); |
|
|
interpolate8x8_halfpel_hv(hv_ptr, n_ptr, edged_width2, rounding); |
|
|
|
|
|
n_ptr += 8; |
|
|
h_ptr += 8; |
|
|
v_ptr += 8; |
|
|
hv_ptr += 8; |
|
|
} |
|
|
h_ptr += stride_add2; |
|
|
v_ptr += stride_add2; |
|
|
hv_ptr += stride_add2; |
|
|
n_ptr += stride_add2; |
|
|
} |
|
|
|
|
|
n_ptr = refn->v; |
|
|
h_ptr = refh->v; |
|
|
v_ptr = refv->v; |
|
|
hv_ptr = refhv->v; |
|
|
|
|
|
n_ptr -= offset2; |
|
|
h_ptr -= offset2; |
|
|
v_ptr -= offset2; |
|
|
hv_ptr -= offset2; |
|
|
|
|
|
for (y = 0; y < edged_height2; y = y + 8) { |
|
|
for (x = 0; x < edged_width2; x = x + 8) { |
|
|
interpolate8x8_halfpel_h(h_ptr, n_ptr, edged_width2, rounding); |
|
|
interpolate8x8_halfpel_v(v_ptr, n_ptr, edged_width2, rounding); |
|
|
interpolate8x8_halfpel_hv(hv_ptr, n_ptr, edged_width2, rounding); |
|
|
|
|
|
n_ptr += 8; |
|
|
h_ptr += 8; |
|
|
v_ptr += 8; |
|
|
hv_ptr += 8; |
|
|
} |
|
|
h_ptr += stride_add2; |
|
|
v_ptr += stride_add2; |
|
|
hv_ptr += stride_add2; |
|
|
n_ptr += stride_add2; |
|
|
} |
|
|
#endif |
|
|
|
|
306 |
/* |
/* |
307 |
interpolate_halfpel_h( |
interpolate_halfpel_h( |
308 |
refh->y - offset, |
refh->y - offset, |
482 |
edged_width / 2, width, height); |
edged_width / 2, width, height); |
483 |
return 0; |
return 0; |
484 |
|
|
485 |
case XVID_CSP_YV12: // u,v swapped |
case XVID_CSP_YV12: /* u,v swapped */ |
486 |
yv12_to_yuv(dst, dst_stride, image->y, image->v, image->u, edged_width, |
yv12_to_yuv(dst, dst_stride, image->y, image->v, image->u, edged_width, |
487 |
edged_width / 2, width, height); |
edged_width / 2, width, height); |
488 |
return 0; |
return 0; |
492 |
edged_width, edged_width / 2, width, height); |
edged_width, edged_width / 2, width, height); |
493 |
return 0; |
return 0; |
494 |
|
|
495 |
case XVID_CSP_YVYU: // u,v swapped |
case XVID_CSP_YVYU: /* u,v swapped */ |
496 |
yv12_to_yuyv(dst, dst_stride, image->y, image->v, image->u, |
yv12_to_yuyv(dst, dst_stride, image->y, image->v, image->u, |
497 |
edged_width, edged_width / 2, width, height); |
edged_width, edged_width / 2, width, height); |
498 |
return 0; |
return 0; |
551 |
return psnr_y; |
return psnr_y; |
552 |
} |
} |
553 |
|
|
554 |
/* |
#if 0 |
555 |
|
|
556 |
#include <stdio.h> |
#include <stdio.h> |
557 |
#include <string.h> |
#include <string.h> |
575 |
} |
} |
576 |
|
|
577 |
|
|
578 |
// dump image+edges to yuv pgm files |
/* dump image+edges to yuv pgm files */ |
579 |
|
|
580 |
int image_dump(IMAGE * image, uint32_t edged_width, uint32_t edged_height, char * path, int number) |
int image_dump(IMAGE * image, uint32_t edged_width, uint32_t edged_height, char * path, int number) |
581 |
{ |
{ |
598 |
|
|
599 |
return 0; |
return 0; |
600 |
} |
} |
601 |
*/ |
#endif |
602 |
|
|
603 |
|
|
604 |
|
|