projects / libav.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Do not use a fake libavcodec/swscale.h, but always use the real one
[libav.git] / libavcodec / imgconvert.c
CommitLineData
de6d9b64
FB		 1/*
2 * Misc image convertion routines
524c6b63 3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard.
de6d9b64 4 *
b78e7197
DB		 5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
ff4ec49e
FB		 8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
b78e7197 10 * version 2.1 of the License, or (at your option) any later version.
de6d9b64 11 *
b78e7197 12 * FFmpeg is distributed in the hope that it will be useful,
de6d9b64 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
ff4ec49e
FB		 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
de6d9b64 16 *
ff4ec49e 17 * You should have received a copy of the GNU Lesser General Public
b78e7197 18 * License along with FFmpeg; if not, write to the Free Software
5509bffa 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
de6d9b64 20 */
983e3246
MN		 21
22/**
1ab3d669 23 * @file imgconvert.c
983e3246
MN		 24 * Misc image convertion routines.
25 */
26
c50c0bc8
FB		 27/* TODO:
28 * - write 'ffimg' program to test all the image related stuff
29 * - move all api to slice based system
30 * - integrate deinterlacing, postprocessing and scaling in the conversion process
31 */
983e3246 32
de6d9b64 33#include "avcodec.h"
85c242d8 34#include "dsputil.h"
de6d9b64 35
54329dd5 36#ifdef USE_FASTMEMCPY
f4bd289a 37#include "libvo/fastmemcpy.h"
54329dd5 38#endif
5981f4e6
F		 39
40#ifdef HAVE_MMX
41#include "i386/mmx.h"
42#endif
524c6b63 43
7e7e5940
FB		 44#define xglue(x, y) x ## y
45#define glue(x, y) xglue(x, y)
46
b6147995
FB		 47#define FF_COLOR_RGB 0 /* RGB color space */
48#define FF_COLOR_GRAY 1 /* gray color space */
49#define FF_COLOR_YUV 2 /* YUV color space. 16 <= Y <= 235, 16 <= U, V <= 240 */
50#define FF_COLOR_YUV_JPEG 3 /* YUV color space. 0 <= Y <= 255, 0 <= U, V <= 255 */
51
7e7e5940
FB		 52#define FF_PIXEL_PLANAR 0 /* each channel has one component in AVPicture */
53#define FF_PIXEL_PACKED 1 /* only one components containing all the channels */
54#define FF_PIXEL_PALETTE 2 /* one components containing indexes for a palette */
55
524c6b63
FB		 56typedef struct PixFmtInfo {
57 const char *name;
7e7e5940
FB		 58 uint8_t nb_channels; /* number of channels (including alpha) */
59 uint8_t color_type; /* color type (see FF_COLOR_xxx constants) */
60 uint8_t pixel_type; /* pixel storage type (see FF_PIXEL_xxx constants) */
0c1a9eda 61 uint8_t is_alpha : 1; /* true if alpha can be specified */
7e7e5940
FB		 62 uint8_t x_chroma_shift; /* X chroma subsampling factor is 2 ^ shift */
63 uint8_t y_chroma_shift; /* Y chroma subsampling factor is 2 ^ shift */
64 uint8_t depth; /* bit depth of the color components */
524c6b63
FB		 65} PixFmtInfo;
66
67/* this table gives more information about formats */
62a05b5b 68static const PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {
524c6b63
FB		 69 /* YUV formats */
70 [PIX_FMT_YUV420P] = {
ef9f7306 71 .name = "yuv420p",
7e7e5940 72 .nb_channels = 3,
b6147995 73 .color_type = FF_COLOR_YUV,
7e7e5940 74 .pixel_type = FF_PIXEL_PLANAR,
b6147995 75 .depth = 8,
115329f1 76 .x_chroma_shift = 1, .y_chroma_shift = 1,
524c6b63
FB		 77 },
78 [PIX_FMT_YUV422P] = {
ef9f7306 79 .name = "yuv422p",
7e7e5940 80 .nb_channels = 3,
b6147995 81 .color_type = FF_COLOR_YUV,
7e7e5940 82 .pixel_type = FF_PIXEL_PLANAR,
b6147995 83 .depth = 8,
115329f1 84 .x_chroma_shift = 1, .y_chroma_shift = 0,
524c6b63
FB		 85 },
86 [PIX_FMT_YUV444P] = {
ef9f7306 87 .name = "yuv444p",
7e7e5940 88 .nb_channels = 3,
b6147995 89 .color_type = FF_COLOR_YUV,
7e7e5940 90 .pixel_type = FF_PIXEL_PLANAR,
b6147995 91 .depth = 8,
115329f1 92 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 93 },
94 [PIX_FMT_YUV422] = {
ef9f7306 95 .name = "yuv422",
7e7e5940 96 .nb_channels = 1,
b6147995 97 .color_type = FF_COLOR_YUV,
7e7e5940 98 .pixel_type = FF_PIXEL_PACKED,
b6147995 99 .depth = 8,
ef9f7306 100 .x_chroma_shift = 1, .y_chroma_shift = 0,
524c6b63 101 },
ebb177dd
TK		 102 [PIX_FMT_UYVY422] = {
103 .name = "uyvy422",
104 .nb_channels = 1,
105 .color_type = FF_COLOR_YUV,
106 .pixel_type = FF_PIXEL_PACKED,
107 .depth = 8,
108 .x_chroma_shift = 1, .y_chroma_shift = 0,
109 },
524c6b63 110 [PIX_FMT_YUV410P] = {
ef9f7306 111 .name = "yuv410p",
7e7e5940 112 .nb_channels = 3,
b6147995 113 .color_type = FF_COLOR_YUV,
7e7e5940 114 .pixel_type = FF_PIXEL_PLANAR,
b6147995 115 .depth = 8,
ef9f7306 116 .x_chroma_shift = 2, .y_chroma_shift = 2,
524c6b63
FB		 117 },
118 [PIX_FMT_YUV411P] = {
ef9f7306 119 .name = "yuv411p",
7e7e5940 120 .nb_channels = 3,
b6147995 121 .color_type = FF_COLOR_YUV,
7e7e5940 122 .pixel_type = FF_PIXEL_PLANAR,
b6147995 123 .depth = 8,
ef9f7306 124 .x_chroma_shift = 2, .y_chroma_shift = 0,
524c6b63
FB		 125 },
126
b6147995
FB		 127 /* JPEG YUV */
128 [PIX_FMT_YUVJ420P] = {
129 .name = "yuvj420p",
7e7e5940 130 .nb_channels = 3,
b6147995 131 .color_type = FF_COLOR_YUV_JPEG,
7e7e5940 132 .pixel_type = FF_PIXEL_PLANAR,
b6147995 133 .depth = 8,
115329f1 134 .x_chroma_shift = 1, .y_chroma_shift = 1,
b6147995
FB		 135 },
136 [PIX_FMT_YUVJ422P] = {
137 .name = "yuvj422p",
7e7e5940 138 .nb_channels = 3,
b6147995 139 .color_type = FF_COLOR_YUV_JPEG,
7e7e5940 140 .pixel_type = FF_PIXEL_PLANAR,
b6147995 141 .depth = 8,
115329f1 142 .x_chroma_shift = 1, .y_chroma_shift = 0,
b6147995
FB		 143 },
144 [PIX_FMT_YUVJ444P] = {
145 .name = "yuvj444p",
7e7e5940 146 .nb_channels = 3,
b6147995 147 .color_type = FF_COLOR_YUV_JPEG,
7e7e5940 148 .pixel_type = FF_PIXEL_PLANAR,
b6147995 149 .depth = 8,
115329f1 150 .x_chroma_shift = 0, .y_chroma_shift = 0,
b6147995
FB		 151 },
152
524c6b63
FB		 153 /* RGB formats */
154 [PIX_FMT_RGB24] = {
ef9f7306 155 .name = "rgb24",
7e7e5940 156 .nb_channels = 3,
b6147995 157 .color_type = FF_COLOR_RGB,
7e7e5940 158 .pixel_type = FF_PIXEL_PACKED,
b6147995 159 .depth = 8,
2cbb7820 160 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 161 },
162 [PIX_FMT_BGR24] = {
ef9f7306 163 .name = "bgr24",
7e7e5940 164 .nb_channels = 3,
b6147995 165 .color_type = FF_COLOR_RGB,
7e7e5940 166 .pixel_type = FF_PIXEL_PACKED,
b6147995 167 .depth = 8,
2cbb7820 168 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 169 },
170 [PIX_FMT_RGBA32] = {
ef9f7306 171 .name = "rgba32",
7e7e5940 172 .nb_channels = 4, .is_alpha = 1,
b6147995 173 .color_type = FF_COLOR_RGB,
7e7e5940 174 .pixel_type = FF_PIXEL_PACKED,
b6147995 175 .depth = 8,
2cbb7820 176 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 177 },
178 [PIX_FMT_RGB565] = {
ef9f7306 179 .name = "rgb565",
7e7e5940 180 .nb_channels = 3,
b6147995 181 .color_type = FF_COLOR_RGB,
7e7e5940 182 .pixel_type = FF_PIXEL_PACKED,
b6147995 183 .depth = 5,
2cbb7820 184 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 185 },
186 [PIX_FMT_RGB555] = {
ef9f7306 187 .name = "rgb555",
7e7e5940 188 .nb_channels = 4, .is_alpha = 1,
b6147995 189 .color_type = FF_COLOR_RGB,
7e7e5940 190 .pixel_type = FF_PIXEL_PACKED,
b6147995 191 .depth = 5,
2cbb7820 192 .x_chroma_shift = 0, .y_chroma_shift = 0,
524c6b63
FB		 193 },
194
195 /* gray / mono formats */
196 [PIX_FMT_GRAY8] = {
ef9f7306 197 .name = "gray",
7e7e5940 198 .nb_channels = 1,
b6147995 199 .color_type = FF_COLOR_GRAY,
7e7e5940 200 .pixel_type = FF_PIXEL_PLANAR,
b6147995 201 .depth = 8,
524c6b63
FB		 202 },
203 [PIX_FMT_MONOWHITE] = {
ef9f7306 204 .name = "monow",
7e7e5940 205 .nb_channels = 1,
b6147995 206 .color_type = FF_COLOR_GRAY,
7e7e5940 207 .pixel_type = FF_PIXEL_PLANAR,
b6147995 208 .depth = 1,
524c6b63
FB		 209 },
210 [PIX_FMT_MONOBLACK] = {
ef9f7306 211 .name = "monob",
7e7e5940 212 .nb_channels = 1,
b6147995 213 .color_type = FF_COLOR_GRAY,
7e7e5940 214 .pixel_type = FF_PIXEL_PLANAR,
b6147995 215 .depth = 1,
524c6b63 216 },
7e6d70d0
FB		 217
218 /* paletted formats */
219 [PIX_FMT_PAL8] = {
220 .name = "pal8",
7e7e5940 221 .nb_channels = 4, .is_alpha = 1,
b6147995 222 .color_type = FF_COLOR_RGB,
7e7e5940 223 .pixel_type = FF_PIXEL_PALETTE,
b6147995 224 .depth = 8,
7e6d70d0 225 },
eab895aa
TK		 226 [PIX_FMT_XVMC_MPEG2_MC] = {
227 .name = "xvmcmc",
228 },
229 [PIX_FMT_XVMC_MPEG2_IDCT] = {
230 .name = "xvmcidct",
231 },
f02be79d
RS		 232 [PIX_FMT_UYVY411] = {
233 .name = "uyvy411",
234 .nb_channels = 1,
235 .color_type = FF_COLOR_YUV,
236 .pixel_type = FF_PIXEL_PACKED,
237 .depth = 8,
238 .x_chroma_shift = 2, .y_chroma_shift = 0,
239 },
00b2fa86
LA		 240 [PIX_FMT_BGR32] = {
241 .name = "bgr32",
242 .nb_channels = 4, .is_alpha = 1,
243 .color_type = FF_COLOR_RGB,
244 .pixel_type = FF_PIXEL_PACKED,
245 .depth = 8,
246 .x_chroma_shift = 0, .y_chroma_shift = 0,
247 },
248 [PIX_FMT_BGR565] = {
249 .name = "bgr565",
250 .nb_channels = 3,
251 .color_type = FF_COLOR_RGB,
252 .pixel_type = FF_PIXEL_PACKED,
253 .depth = 5,
254 .x_chroma_shift = 0, .y_chroma_shift = 0,
255 },
256 [PIX_FMT_BGR555] = {
257 .name = "bgr555",
258 .nb_channels = 4, .is_alpha = 1,
259 .color_type = FF_COLOR_RGB,
260 .pixel_type = FF_PIXEL_PACKED,
261 .depth = 5,
262 .x_chroma_shift = 0, .y_chroma_shift = 0,
263 },
264 [PIX_FMT_RGB8] = {
265 .name = "rgb8",
266 .nb_channels = 1,
267 .color_type = FF_COLOR_RGB,
268 .pixel_type = FF_PIXEL_PACKED,
269 .depth = 8,
270 .x_chroma_shift = 0, .y_chroma_shift = 0,
271 },
272 [PIX_FMT_RGB4] = {
273 .name = "rgb4",
274 .nb_channels = 1,
275 .color_type = FF_COLOR_RGB,
276 .pixel_type = FF_PIXEL_PACKED,
277 .depth = 4,
278 .x_chroma_shift = 0, .y_chroma_shift = 0,
279 },
280 [PIX_FMT_RGB4_BYTE] = {
281 .name = "rgb4_byte",
282 .nb_channels = 1,
283 .color_type = FF_COLOR_RGB,
284 .pixel_type = FF_PIXEL_PACKED,
285 .depth = 8,
286 .x_chroma_shift = 0, .y_chroma_shift = 0,
287 },
288 [PIX_FMT_BGR8] = {
289 .name = "bgr8",
290 .nb_channels = 1,
291 .color_type = FF_COLOR_RGB,
292 .pixel_type = FF_PIXEL_PACKED,
293 .depth = 8,
294 .x_chroma_shift = 0, .y_chroma_shift = 0,
295 },
296 [PIX_FMT_BGR4] = {
297 .name = "bgr4",
298 .nb_channels = 1,
299 .color_type = FF_COLOR_RGB,
300 .pixel_type = FF_PIXEL_PACKED,
301 .depth = 4,
302 .x_chroma_shift = 0, .y_chroma_shift = 0,
303 },
304 [PIX_FMT_BGR4_BYTE] = {
305 .name = "bgr4_byte",
306 .nb_channels = 1,
307 .color_type = FF_COLOR_RGB,
308 .pixel_type = FF_PIXEL_PACKED,
309 .depth = 8,
310 .x_chroma_shift = 0, .y_chroma_shift = 0,
311 },
312 [PIX_FMT_NV12] = {
313 .name = "nv12",
314 .nb_channels = 2,
315 .color_type = FF_COLOR_YUV,
316 .pixel_type = FF_PIXEL_PLANAR,
317 .depth = 8,
318 .x_chroma_shift = 1, .y_chroma_shift = 1,
319 },
320 [PIX_FMT_NV21] = {
321 .name = "nv12",
322 .nb_channels = 2,
323 .color_type = FF_COLOR_YUV,
324 .pixel_type = FF_PIXEL_PLANAR,
325 .depth = 8,
326 .x_chroma_shift = 1, .y_chroma_shift = 1,
327 },
328
329 [PIX_FMT_BGR32_1] = {
330 .name = "bgr32_1",
331 .nb_channels = 4, .is_alpha = 1,
332 .color_type = FF_COLOR_RGB,
333 .pixel_type = FF_PIXEL_PACKED,
334 .depth = 8,
335 .x_chroma_shift = 0, .y_chroma_shift = 0,
336 },
337 [PIX_FMT_RGB32_1] = {
338 .name = "rgb32_1",
339 .nb_channels = 4, .is_alpha = 1,
340 .color_type = FF_COLOR_RGB,
341 .pixel_type = FF_PIXEL_PACKED,
342 .depth = 8,
343 .x_chroma_shift = 0, .y_chroma_shift = 0,
344 },
524c6b63
FB		 345};
346
347void avcodec_get_chroma_sub_sample(int pix_fmt, int *h_shift, int *v_shift)
348{
b6147995
FB		 349 *h_shift = pix_fmt_info[pix_fmt].x_chroma_shift;
350 *v_shift = pix_fmt_info[pix_fmt].y_chroma_shift;
524c6b63
FB		 351}
352
353const char *avcodec_get_pix_fmt_name(int pix_fmt)
354{
355 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB)
356 return "???";
357 else
358 return pix_fmt_info[pix_fmt].name;
359}
360
63167088
RS		 361enum PixelFormat avcodec_get_pix_fmt(const char* name)
362{
115329f1
DB		 363 int i;
364
63167088
RS		 365 for (i=0; i < PIX_FMT_NB; i++)
366 if (!strcmp(pix_fmt_info[i].name, name))
bb270c08 367 break;
63167088
RS		 368 return i;
369}
370
2a877875 371/* Picture field are filled with 'ptr' addresses. Also return size */
0c1a9eda 372int avpicture_fill(AVPicture *picture, uint8_t *ptr,
bb270c08 373 int pix_fmt, int width, int height)
2a877875 374{
4c7e8619 375 int size, w2, h2, size2;
62a05b5b 376 const PixFmtInfo *pinfo;
115329f1 377
0ecca7a4
MN		 378 if(avcodec_check_dimensions(NULL, width, height))
379 goto fail;
380
4c7e8619 381 pinfo = &pix_fmt_info[pix_fmt];
2a877875
FB		 382 size = width * height;
383 switch(pix_fmt) {
384 case PIX_FMT_YUV420P:
4c7e8619
FB		 385 case PIX_FMT_YUV422P:
386 case PIX_FMT_YUV444P:
387 case PIX_FMT_YUV410P:
388 case PIX_FMT_YUV411P:
c50c0bc8
FB		 389 case PIX_FMT_YUVJ420P:
390 case PIX_FMT_YUVJ422P:
391 case PIX_FMT_YUVJ444P:
4c7e8619
FB		 392 w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;
393 h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;
394 size2 = w2 * h2;
2a877875
FB		 395 picture->data[0] = ptr;
396 picture->data[1] = picture->data[0] + size;
4c7e8619 397 picture->data[2] = picture->data[1] + size2;
2a877875 398 picture->linesize[0] = width;
4c7e8619
FB		 399 picture->linesize[1] = w2;
400 picture->linesize[2] = w2;
401 return size + 2 * size2;
00b2fa86
LA		 402 case PIX_FMT_NV12:
403 case PIX_FMT_NV21:
404 w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;
405 h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;
406 size2 = w2 * h2 * 2;
407 picture->data[0] = ptr;
408 picture->data[1] = picture->data[0] + size;
409 picture->data[2] = NULL;
410 picture->linesize[0] = width;
411 picture->linesize[1] = w2;
412 picture->linesize[2] = 0;
413 return size + 2 * size2;
2a877875
FB		 414 case PIX_FMT_RGB24:
415 case PIX_FMT_BGR24:
416 picture->data[0] = ptr;
417 picture->data[1] = NULL;
418 picture->data[2] = NULL;
419 picture->linesize[0] = width * 3;
420 return size * 3;
2a877875 421 case PIX_FMT_RGBA32:
00b2fa86
LA		 422 case PIX_FMT_BGR32:
423 case PIX_FMT_RGB32_1:
424 case PIX_FMT_BGR32_1:
2a877875
FB		 425 picture->data[0] = ptr;
426 picture->data[1] = NULL;
427 picture->data[2] = NULL;
428 picture->linesize[0] = width * 4;
429 return size * 4;
00b2fa86
LA		 430 case PIX_FMT_BGR555:
431 case PIX_FMT_BGR565:
2a877875
FB		 432 case PIX_FMT_RGB555:
433 case PIX_FMT_RGB565:
434 case PIX_FMT_YUV422:
435 picture->data[0] = ptr;
436 picture->data[1] = NULL;
437 picture->data[2] = NULL;
438 picture->linesize[0] = width * 2;
439 return size * 2;
ebb177dd
TK		 440 case PIX_FMT_UYVY422:
441 picture->data[0] = ptr;
442 picture->data[1] = NULL;
443 picture->data[2] = NULL;
444 picture->linesize[0] = width * 2;
445 return size * 2;
f02be79d
RS		 446 case PIX_FMT_UYVY411:
447 picture->data[0] = ptr;
448 picture->data[1] = NULL;
449 picture->data[2] = NULL;
450 picture->linesize[0] = width + width/2;
451 return size + size/2;
00b2fa86
LA		 452 case PIX_FMT_RGB8:
453 case PIX_FMT_BGR8:
454 case PIX_FMT_RGB4_BYTE:
455 case PIX_FMT_BGR4_BYTE:
2a877875
FB		 456 case PIX_FMT_GRAY8:
457 picture->data[0] = ptr;
458 picture->data[1] = NULL;
459 picture->data[2] = NULL;
460 picture->linesize[0] = width;
461 return size;
00b2fa86
LA		 462 case PIX_FMT_RGB4:
463 case PIX_FMT_BGR4:
464 picture->data[0] = ptr;
465 picture->data[1] = NULL;
466 picture->data[2] = NULL;
467 picture->linesize[0] = width / 2;
468 return size / 2;
2a877875
FB		 469 case PIX_FMT_MONOWHITE:
470 case PIX_FMT_MONOBLACK:
471 picture->data[0] = ptr;
472 picture->data[1] = NULL;
473 picture->data[2] = NULL;
474 picture->linesize[0] = (width + 7) >> 3;
475 return picture->linesize[0] * height;
7e6d70d0
FB		 476 case PIX_FMT_PAL8:
477 size2 = (size + 3) & ~3;
478 picture->data[0] = ptr;
479 picture->data[1] = ptr + size2; /* palette is stored here as 256 32 bit words */
480 picture->data[2] = NULL;
481 picture->linesize[0] = width;
482 picture->linesize[1] = 4;
483 return size2 + 256 * 4;
2a877875 484 default:
0ecca7a4 485fail:
2a877875
FB		 486 picture->data[0] = NULL;
487 picture->data[1] = NULL;
488 picture->data[2] = NULL;
7e6d70d0 489 picture->data[3] = NULL;
2a877875
FB		 490 return -1;
491 }
492}
493
da64ecc3 494int avpicture_layout(const AVPicture* src, int pix_fmt, int width, int height,
63167088
RS		 495 unsigned char *dest, int dest_size)
496{
62a05b5b 497 const PixFmtInfo* pf = &pix_fmt_info[pix_fmt];
63167088 498 int i, j, w, h, data_planes;
115329f1 499 const unsigned char* s;
63167088
RS		 500 int size = avpicture_get_size(pix_fmt, width, height);
501
0ecca7a4 502 if (size > dest_size || size < 0)
63167088
RS		 503 return -1;
504
affd55a1 505 if (pf->pixel_type == FF_PIXEL_PACKED || pf->pixel_type == FF_PIXEL_PALETTE) {
115329f1
DB		 506 if (pix_fmt == PIX_FMT_YUV422 ||
507 pix_fmt == PIX_FMT_UYVY422 ||
00b2fa86
LA		 508 pix_fmt == PIX_FMT_BGR565 ||
509 pix_fmt == PIX_FMT_BGR565 ||
ebb177dd
TK		 510 pix_fmt == PIX_FMT_RGB565 ||
511 pix_fmt == PIX_FMT_RGB555)
512 w = width * 2;
bb270c08
DB		 513 else if (pix_fmt == PIX_FMT_UYVY411)
514 w = width + width/2;
515 else if (pix_fmt == PIX_FMT_PAL8)
516 w = width;
517 else
518 w = width * (pf->depth * pf->nb_channels / 8);
519
520 data_planes = 1;
521 h = height;
63167088
RS		 522 } else {
523 data_planes = pf->nb_channels;
bb270c08
DB		 524 w = (width*pf->depth + 7)/8;
525 h = height;
63167088 526 }
115329f1 527
63167088
RS		 528 for (i=0; i<data_planes; i++) {
529 if (i == 1) {
bb270c08
DB		 530 w = width >> pf->x_chroma_shift;
531 h = height >> pf->y_chroma_shift;
532 }
63167088 533 s = src->data[i];
bb270c08
DB		 534 for(j=0; j<h; j++) {
535 memcpy(dest, s, w);
536 dest += w;
537 s += src->linesize[i];
538 }
63167088 539 }
115329f1 540
affd55a1 541 if (pf->pixel_type == FF_PIXEL_PALETTE)
bb270c08 542 memcpy((unsigned char *)(((size_t)dest + 3) & ~3), src->data[1], 256 * 4);
115329f1 543
63167088
RS		 544 return size;
545}
546
2a877875
FB		 547int avpicture_get_size(int pix_fmt, int width, int height)
548{
549 AVPicture dummy_pict;
550 return avpicture_fill(&dummy_pict, NULL, pix_fmt, width, height);
551}
552
b6147995 553/**
115329f1 554 * compute the loss when converting from a pixel format to another
b6147995
FB		 555 */
556int avcodec_get_pix_fmt_loss(int dst_pix_fmt, int src_pix_fmt,
557 int has_alpha)
558{
559 const PixFmtInfo *pf, *ps;
560 int loss;
561
562 ps = &pix_fmt_info[src_pix_fmt];
563 pf = &pix_fmt_info[dst_pix_fmt];
564
565 /* compute loss */
566 loss = 0;
567 pf = &pix_fmt_info[dst_pix_fmt];
0a9ad8d1
FB		 568 if (pf->depth < ps->depth ||
569 (dst_pix_fmt == PIX_FMT_RGB555 && src_pix_fmt == PIX_FMT_RGB565))
b6147995 570 loss |= FF_LOSS_DEPTH;
0a9ad8d1
FB		 571 if (pf->x_chroma_shift > ps->x_chroma_shift ||
572 pf->y_chroma_shift > ps->y_chroma_shift)
b6147995
FB		 573 loss |= FF_LOSS_RESOLUTION;
574 switch(pf->color_type) {
575 case FF_COLOR_RGB:
576 if (ps->color_type != FF_COLOR_RGB &&
577 ps->color_type != FF_COLOR_GRAY)
578 loss |= FF_LOSS_COLORSPACE;
579 break;
580 case FF_COLOR_GRAY:
581 if (ps->color_type != FF_COLOR_GRAY)
582 loss |= FF_LOSS_COLORSPACE;
583 break;
584 case FF_COLOR_YUV:
585 if (ps->color_type != FF_COLOR_YUV)
586 loss |= FF_LOSS_COLORSPACE;
587 break;
588 case FF_COLOR_YUV_JPEG:
589 if (ps->color_type != FF_COLOR_YUV_JPEG &&
115329f1 590 ps->color_type != FF_COLOR_YUV &&
0a9ad8d1 591 ps->color_type != FF_COLOR_GRAY)
b6147995
FB		 592 loss |= FF_LOSS_COLORSPACE;
593 break;
594 default:
595 /* fail safe test */
596 if (ps->color_type != pf->color_type)
597 loss |= FF_LOSS_COLORSPACE;
598 break;
599 }
600 if (pf->color_type == FF_COLOR_GRAY &&
601 ps->color_type != FF_COLOR_GRAY)
602 loss |= FF_LOSS_CHROMA;
603 if (!pf->is_alpha && (ps->is_alpha && has_alpha))
604 loss |= FF_LOSS_ALPHA;
115329f1 605 if (pf->pixel_type == FF_PIXEL_PALETTE &&
7e7e5940 606 (ps->pixel_type != FF_PIXEL_PALETTE && ps->color_type != FF_COLOR_GRAY))
b6147995
FB		 607 loss |= FF_LOSS_COLORQUANT;
608 return loss;
609}
610
611static int avg_bits_per_pixel(int pix_fmt)
612{
613 int bits;
614 const PixFmtInfo *pf;
615
616 pf = &pix_fmt_info[pix_fmt];
7e7e5940
FB		 617 switch(pf->pixel_type) {
618 case FF_PIXEL_PACKED:
b6147995 619 switch(pix_fmt) {
7e7e5940 620 case PIX_FMT_YUV422:
ebb177dd 621 case PIX_FMT_UYVY422:
b6147995
FB		 622 case PIX_FMT_RGB565:
623 case PIX_FMT_RGB555:
00b2fa86
LA		 624 case PIX_FMT_BGR565:
625 case PIX_FMT_BGR555:
b6147995 626 bits = 16;
b6147995 627 break;
bb270c08
DB		 628 case PIX_FMT_UYVY411:
629 bits = 12;
630 break;
b6147995 631 default:
7e7e5940 632 bits = pf->depth * pf->nb_channels;
b6147995
FB		 633 break;
634 }
7e7e5940
FB		 635 break;
636 case FF_PIXEL_PLANAR:
637 if (pf->x_chroma_shift == 0 && pf->y_chroma_shift == 0) {
638 bits = pf->depth * pf->nb_channels;
639 } else {
115329f1 640 bits = pf->depth + ((2 * pf->depth) >>
7e7e5940
FB		 641 (pf->x_chroma_shift + pf->y_chroma_shift));
642 }
643 break;
644 case FF_PIXEL_PALETTE:
645 bits = 8;
646 break;
647 default:
648 bits = -1;
649 break;
b6147995
FB		 650 }
651 return bits;
652}
653
115329f1 654static int avcodec_find_best_pix_fmt1(int pix_fmt_mask,
b6147995
FB		 655 int src_pix_fmt,
656 int has_alpha,
657 int loss_mask)
658{
659 int dist, i, loss, min_dist, dst_pix_fmt;
660
661 /* find exact color match with smallest size */
662 dst_pix_fmt = -1;
663 min_dist = 0x7fffffff;
664 for(i = 0;i < PIX_FMT_NB; i++) {
665 if (pix_fmt_mask & (1 << i)) {
666 loss = avcodec_get_pix_fmt_loss(i, src_pix_fmt, has_alpha) & loss_mask;
667 if (loss == 0) {
668 dist = avg_bits_per_pixel(i);
669 if (dist < min_dist) {
670 min_dist = dist;
671 dst_pix_fmt = i;
672 }
673 }
674 }
675 }
676 return dst_pix_fmt;
677}
678
115329f1
DB		 679/**
680 * find best pixel format to convert to. Return -1 if none found
b6147995
FB		 681 */
682int avcodec_find_best_pix_fmt(int pix_fmt_mask, int src_pix_fmt,
683 int has_alpha, int *loss_ptr)
684{
685 int dst_pix_fmt, loss_mask, i;
686 static const int loss_mask_order[] = {
687 ~0, /* no loss first */
688 ~FF_LOSS_ALPHA,
689 ~FF_LOSS_RESOLUTION,
690 ~(FF_LOSS_COLORSPACE | FF_LOSS_RESOLUTION),
691 ~FF_LOSS_COLORQUANT,
692 ~FF_LOSS_DEPTH,
693 0,
694 };
695
696 /* try with successive loss */
697 i = 0;
698 for(;;) {
699 loss_mask = loss_mask_order[i++];
115329f1 700 dst_pix_fmt = avcodec_find_best_pix_fmt1(pix_fmt_mask, src_pix_fmt,
b6147995
FB		 701 has_alpha, loss_mask);
702 if (dst_pix_fmt >= 0)
703 goto found;
704 if (loss_mask == 0)
705 break;
706 }
707 return -1;
708 found:
709 if (loss_ptr)
710 *loss_ptr = avcodec_get_pix_fmt_loss(dst_pix_fmt, src_pix_fmt, has_alpha);
711 return dst_pix_fmt;
712}
713
54009d42 714void ff_img_copy_plane(uint8_t *dst, int dst_wrap,
e352ff08
FB		 715 const uint8_t *src, int src_wrap,
716 int width, int height)
7e7e5940 717{
115329f1 718 if((!dst) || (!src))
5b2ad9f5 719 return;
7e7e5940
FB		 720 for(;height > 0; height--) {
721 memcpy(dst, src, width);
722 dst += dst_wrap;
723 src += src_wrap;
724 }
725}
726
0469baf1
FB		 727/**
728 * Copy image 'src' to 'dst'.
729 */
da64ecc3 730void img_copy(AVPicture *dst, const AVPicture *src,
7e7e5940
FB		 731 int pix_fmt, int width, int height)
732{
733 int bwidth, bits, i;
62a05b5b 734 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
115329f1 735
7e7e5940
FB		 736 pf = &pix_fmt_info[pix_fmt];
737 switch(pf->pixel_type) {
738 case FF_PIXEL_PACKED:
739 switch(pix_fmt) {
740 case PIX_FMT_YUV422:
ebb177dd 741 case PIX_FMT_UYVY422:
7e7e5940
FB		 742 case PIX_FMT_RGB565:
743 case PIX_FMT_RGB555:
00b2fa86
LA		 744 case PIX_FMT_BGR565:
745 case PIX_FMT_BGR555:
7e7e5940
FB		 746 bits = 16;
747 break;
bb270c08
DB		 748 case PIX_FMT_UYVY411:
749 bits = 12;
750 break;
7e7e5940
FB		 751 default:
752 bits = pf->depth * pf->nb_channels;
753 break;
754 }
755 bwidth = (width * bits + 7) >> 3;
54009d42 756 ff_img_copy_plane(dst->data[0], dst->linesize[0],
7e7e5940
FB		 757 src->data[0], src->linesize[0],
758 bwidth, height);
759 break;
760 case FF_PIXEL_PLANAR:
761 for(i = 0; i < pf->nb_channels; i++) {
762 int w, h;
763 w = width;
764 h = height;
765 if (i == 1 || i == 2) {
766 w >>= pf->x_chroma_shift;
767 h >>= pf->y_chroma_shift;
768 }
769 bwidth = (w * pf->depth + 7) >> 3;
54009d42 770 ff_img_copy_plane(dst->data[i], dst->linesize[i],
7e7e5940
FB		 771 src->data[i], src->linesize[i],
772 bwidth, h);
773 }
774 break;
775 case FF_PIXEL_PALETTE:
54009d42 776 ff_img_copy_plane(dst->data[0], dst->linesize[0],
7e7e5940
FB		 777 src->data[0], src->linesize[0],
778 width, height);
779 /* copy the palette */
54009d42 780 ff_img_copy_plane(dst->data[1], dst->linesize[1],
7e7e5940
FB		 781 src->data[1], src->linesize[1],
782 4, 256);
783 break;
784 }
785}
2a877875 786
de6d9b64
FB		 787/* XXX: totally non optimized */
788
da64ecc3 789static void yuv422_to_yuv420p(AVPicture *dst, const AVPicture *src,
524c6b63 790 int width, int height)
de6d9b64 791{
c50c0bc8
FB		 792 const uint8_t *p, *p1;
793 uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
d4f5d74a 794 int w;
115329f1 795
c50c0bc8
FB		 796 p1 = src->data[0];
797 lum1 = dst->data[0];
0a05e494
FB		 798 cb1 = dst->data[1];
799 cr1 = dst->data[2];
c50c0bc8 800
d4f5d74a 801 for(;height >= 1; height -= 2) {
c50c0bc8
FB		 802 p = p1;
803 lum = lum1;
804 cb = cb1;
805 cr = cr1;
d4f5d74a 806 for(w = width; w >= 2; w -= 2) {
e78df699
MN		 807 lum[0] = p[0];
808 cb[0] = p[1];
809 lum[1] = p[2];
810 cr[0] = p[3];
de6d9b64
FB		 811 p += 4;
812 lum += 2;
813 cb++;
814 cr++;
815 }
d4f5d74a 816 if (w) {
e78df699 817 lum[0] = p[0];
d4f5d74a
GM		 818 cb[0] = p[1];
819 cr[0] = p[3];
820 cb++;
821 cr++;
de6d9b64 822 }
c50c0bc8
FB		 823 p1 += src->linesize[0];
824 lum1 += dst->linesize[0];
d4f5d74a
GM		 825 if (height>1) {
826 p = p1;
827 lum = lum1;
828 for(w = width; w >= 2; w -= 2) {
829 lum[0] = p[0];
830 lum[1] = p[2];
831 p += 4;
832 lum += 2;
833 }
834 if (w) {
835 lum[0] = p[0];
836 }
837 p1 += src->linesize[0];
838 lum1 += dst->linesize[0];
839 }
c50c0bc8
FB		 840 cb1 += dst->linesize[1];
841 cr1 += dst->linesize[2];
842 }
843}
844
ebb177dd
TK		 845static void uyvy422_to_yuv420p(AVPicture *dst, const AVPicture *src,
846 int width, int height)
847{
848 const uint8_t *p, *p1;
849 uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
850 int w;
115329f1 851
ebb177dd 852 p1 = src->data[0];
115329f1 853
ebb177dd
TK		 854 lum1 = dst->data[0];
855 cb1 = dst->data[1];
856 cr1 = dst->data[2];
857
858 for(;height >= 1; height -= 2) {
859 p = p1;
860 lum = lum1;
861 cb = cb1;
862 cr = cr1;
863 for(w = width; w >= 2; w -= 2) {
864 lum[0] = p[1];
865 cb[0] = p[0];
866 lum[1] = p[3];
867 cr[0] = p[2];
868 p += 4;
869 lum += 2;
870 cb++;
871 cr++;
872 }
873 if (w) {
874 lum[0] = p[1];
875 cb[0] = p[0];
876 cr[0] = p[2];
877 cb++;
878 cr++;
879 }
880 p1 += src->linesize[0];
881 lum1 += dst->linesize[0];
882 if (height>1) {
883 p = p1;
884 lum = lum1;
885 for(w = width; w >= 2; w -= 2) {
886 lum[0] = p[1];
887 lum[1] = p[3];
888 p += 4;
889 lum += 2;
890 }
891 if (w) {
892 lum[0] = p[1];
893 }
894 p1 += src->linesize[0];
895 lum1 += dst->linesize[0];
896 }
897 cb1 += dst->linesize[1];
898 cr1 += dst->linesize[2];
899 }
900}
901
902
903static void uyvy422_to_yuv422p(AVPicture *dst, const AVPicture *src,
904 int width, int height)
905{
906 const uint8_t *p, *p1;
907 uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
908 int w;
909
910 p1 = src->data[0];
911 lum1 = dst->data[0];
912 cb1 = dst->data[1];
913 cr1 = dst->data[2];
914 for(;height > 0; height--) {
915 p = p1;
916 lum = lum1;
917 cb = cb1;
918 cr = cr1;
919 for(w = width; w >= 2; w -= 2) {
920 lum[0] = p[1];
921 cb[0] = p[0];
922 lum[1] = p[3];
923 cr[0] = p[2];
924 p += 4;
925 lum += 2;
926 cb++;
927 cr++;
928 }
929 p1 += src->linesize[0];
930 lum1 += dst->linesize[0];
931 cb1 += dst->linesize[1];
932 cr1 += dst->linesize[2];
933 }
934}
935
936
da64ecc3 937static void yuv422_to_yuv422p(AVPicture *dst, const AVPicture *src,
c50c0bc8
FB		 938 int width, int height)
939{
940 const uint8_t *p, *p1;
941 uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
942 int w;
943
944 p1 = src->data[0];
945 lum1 = dst->data[0];
0a05e494
FB		 946 cb1 = dst->data[1];
947 cr1 = dst->data[2];
948 for(;height > 0; height--) {
c50c0bc8
FB		 949 p = p1;
950 lum = lum1;
951 cb = cb1;
952 cr = cr1;
953 for(w = width; w >= 2; w -= 2) {
954 lum[0] = p[0];
955 cb[0] = p[1];
956 lum[1] = p[2];
957 cr[0] = p[3];
958 p += 4;
959 lum += 2;
960 cb++;
961 cr++;
962 }
963 p1 += src->linesize[0];
964 lum1 += dst->linesize[0];
965 cb1 += dst->linesize[1];
966 cr1 += dst->linesize[2];
de6d9b64
FB		 967 }
968}
969
da64ecc3 970static void yuv422p_to_yuv422(AVPicture *dst, const AVPicture *src,
c50c0bc8
FB		 971 int width, int height)
972{
973 uint8_t *p, *p1;
974 const uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
975 int w;
976
977 p1 = dst->data[0];
978 lum1 = src->data[0];
0a05e494
FB		 979 cb1 = src->data[1];
980 cr1 = src->data[2];
981 for(;height > 0; height--) {
c50c0bc8
FB		 982 p = p1;
983 lum = lum1;
984 cb = cb1;
985 cr = cr1;
986 for(w = width; w >= 2; w -= 2) {
987 p[0] = lum[0];
988 p[1] = cb[0];
989 p[2] = lum[1];
990 p[3] = cr[0];
991 p += 4;
992 lum += 2;
993 cb++;
994 cr++;
995 }
0a05e494
FB		 996 p1 += dst->linesize[0];
997 lum1 += src->linesize[0];
998 cb1 += src->linesize[1];
999 cr1 += src->linesize[2];
c50c0bc8
FB		 1000 }
1001}
1002
ebb177dd
TK		 1003static void yuv422p_to_uyvy422(AVPicture *dst, const AVPicture *src,
1004 int width, int height)
1005{
1006 uint8_t *p, *p1;
1007 const uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
1008 int w;
1009
1010 p1 = dst->data[0];
1011 lum1 = src->data[0];
1012 cb1 = src->data[1];
1013 cr1 = src->data[2];
1014 for(;height > 0; height--) {
1015 p = p1;
1016 lum = lum1;
1017 cb = cb1;
1018 cr = cr1;
1019 for(w = width; w >= 2; w -= 2) {
1020 p[1] = lum[0];
1021 p[0] = cb[0];
1022 p[3] = lum[1];
1023 p[2] = cr[0];
1024 p += 4;
1025 lum += 2;
1026 cb++;
1027 cr++;
1028 }
1029 p1 += dst->linesize[0];
1030 lum1 += src->linesize[0];
1031 cb1 += src->linesize[1];
1032 cr1 += src->linesize[2];
1033 }
1034}
1035
f02be79d
RS		 1036static void uyvy411_to_yuv411p(AVPicture *dst, const AVPicture *src,
1037 int width, int height)
1038{
1039 const uint8_t *p, *p1;
1040 uint8_t *lum, *cr, *cb, *lum1, *cr1, *cb1;
1041 int w;
1042
1043 p1 = src->data[0];
1044 lum1 = dst->data[0];
1045 cb1 = dst->data[1];
1046 cr1 = dst->data[2];
1047 for(;height > 0; height--) {
1048 p = p1;
1049 lum = lum1;
1050 cb = cb1;
1051 cr = cr1;
1052 for(w = width; w >= 4; w -= 4) {
1053 cb[0] = p[0];
bb270c08 1054 lum[0] = p[1];
f02be79d
RS		 1055 lum[1] = p[2];
1056 cr[0] = p[3];
bb270c08
DB		 1057 lum[2] = p[4];
1058 lum[3] = p[5];
f02be79d
RS		 1059 p += 6;
1060 lum += 4;
1061 cb++;
1062 cr++;
1063 }
1064 p1 += src->linesize[0];
1065 lum1 += dst->linesize[0];
1066 cb1 += dst->linesize[1];
1067 cr1 += dst->linesize[2];
1068 }
1069}
ebb177dd
TK		 1070
1071
d7e2f57f
MN		 1072static void yuv420p_to_yuv422(AVPicture *dst, const AVPicture *src,
1073 int width, int height)
1074{
1075 int w, h;
1076 uint8_t *line1, *line2, *linesrc = dst->data[0];
1077 uint8_t *lum1, *lum2, *lumsrc = src->data[0];
1078 uint8_t *cb1, *cb2 = src->data[1];
1079 uint8_t *cr1, *cr2 = src->data[2];
115329f1 1080
d7e2f57f
MN		 1081 for(h = height / 2; h--;) {
1082 line1 = linesrc;
1083 line2 = linesrc + dst->linesize[0];
115329f1 1084
d7e2f57f
MN		 1085 lum1 = lumsrc;
1086 lum2 = lumsrc + src->linesize[0];
115329f1 1087
d7e2f57f
MN		 1088 cb1 = cb2;
1089 cr1 = cr2;
115329f1 1090
d7e2f57f 1091 for(w = width / 2; w--;) {
115329f1
DB		 1092 *line1++ = *lum1++; *line2++ = *lum2++;
1093 *line1++ = *line2++ = *cb1++;
1094 *line1++ = *lum1++; *line2++ = *lum2++;
d7e2f57f
MN		 1095 *line1++ = *line2++ = *cr1++;
1096 }
115329f1 1097
d7e2f57f
MN		 1098 linesrc += dst->linesize[0] * 2;
1099 lumsrc += src->linesize[0] * 2;
1100 cb2 += src->linesize[1];
1101 cr2 += src->linesize[2];
1102 }
1103}
1104
bac65165
LA		 1105static void yuv420p_to_uyvy422(AVPicture *dst, const AVPicture *src,
1106 int width, int height)
1107{
1108 int w, h;
1109 uint8_t *line1, *line2, *linesrc = dst->data[0];
1110 uint8_t *lum1, *lum2, *lumsrc = src->data[0];
1111 uint8_t *cb1, *cb2 = src->data[1];
1112 uint8_t *cr1, *cr2 = src->data[2];
115329f1 1113
bac65165
LA		 1114 for(h = height / 2; h--;) {
1115 line1 = linesrc;
1116 line2 = linesrc + dst->linesize[0];
115329f1 1117
bac65165
LA		 1118 lum1 = lumsrc;
1119 lum2 = lumsrc + src->linesize[0];
115329f1 1120
bac65165
LA		 1121 cb1 = cb2;
1122 cr1 = cr2;
115329f1 1123
bac65165 1124 for(w = width / 2; w--;) {
115329f1
DB		 1125 *line1++ = *line2++ = *cb1++;
1126 *line1++ = *lum1++; *line2++ = *lum2++;
bac65165 1127 *line1++ = *line2++ = *cr1++;
115329f1 1128 *line1++ = *lum1++; *line2++ = *lum2++;
bac65165 1129 }
115329f1 1130
bac65165
LA		 1131 linesrc += dst->linesize[0] * 2;
1132 lumsrc += src->linesize[0] * 2;
1133 cb2 += src->linesize[1];
1134 cr2 += src->linesize[2];
1135 }
1136}
1137
c50c0bc8 1138#define SCALEBITS 10
de6d9b64 1139#define ONE_HALF (1 << (SCALEBITS - 1))
bb270c08 1140#define FIX(x) ((int) ((x) * (1<<SCALEBITS) + 0.5))
de6d9b64 1141
c50c0bc8
FB		 1142#define YUV_TO_RGB1_CCIR(cb1, cr1)\
1143{\
1144 cb = (cb1) - 128;\
1145 cr = (cr1) - 128;\
1146 r_add = FIX(1.40200*255.0/224.0) * cr + ONE_HALF;\
1147 g_add = - FIX(0.34414*255.0/224.0) * cb - FIX(0.71414*255.0/224.0) * cr + \
1148 ONE_HALF;\
1149 b_add = FIX(1.77200*255.0/224.0) * cb + ONE_HALF;\
1150}
b6147995 1151
c50c0bc8
FB		 1152#define YUV_TO_RGB2_CCIR(r, g, b, y1)\
1153{\
1154 y = ((y1) - 16) * FIX(255.0/219.0);\
1155 r = cm[(y + r_add) >> SCALEBITS];\
1156 g = cm[(y + g_add) >> SCALEBITS];\
1157 b = cm[(y + b_add) >> SCALEBITS];\
1158}
1159
1160#define YUV_TO_RGB1(cb1, cr1)\
1161{\
1162 cb = (cb1) - 128;\
1163 cr = (cr1) - 128;\
1164 r_add = FIX(1.40200) * cr + ONE_HALF;\
1165 g_add = - FIX(0.34414) * cb - FIX(0.71414) * cr + ONE_HALF;\
1166 b_add = FIX(1.77200) * cb + ONE_HALF;\
1167}
b6147995
FB		 1168
1169#define YUV_TO_RGB2(r, g, b, y1)\
1170{\
c50c0bc8
FB		 1171 y = (y1) << SCALEBITS;\
1172 r = cm[(y + r_add) >> SCALEBITS];\
1173 g = cm[(y + g_add) >> SCALEBITS];\
1174 b = cm[(y + b_add) >> SCALEBITS];\
b6147995
FB		 1175}
1176
c50c0bc8 1177#define Y_CCIR_TO_JPEG(y)\
b6147995
FB		 1178 cm[((y) * FIX(255.0/219.0) + (ONE_HALF - 16 * FIX(255.0/219.0))) >> SCALEBITS]
1179
c50c0bc8 1180#define Y_JPEG_TO_CCIR(y)\
b6147995
FB		 1181 (((y) * FIX(219.0/255.0) + (ONE_HALF + (16 << SCALEBITS))) >> SCALEBITS)
1182
c50c0bc8
FB		 1183#define C_CCIR_TO_JPEG(y)\
1184 cm[(((y) - 128) * FIX(127.0/112.0) + (ONE_HALF + (128 << SCALEBITS))) >> SCALEBITS]
1185
1186/* NOTE: the clamp is really necessary! */
4cfbf61b
FH		 1187static inline int C_JPEG_TO_CCIR(int y) {
1188 y = (((y - 128) * FIX(112.0/127.0) + (ONE_HALF + (128 << SCALEBITS))) >> SCALEBITS);
1189 if (y < 16)
bb270c08 1190 y = 16;
4cfbf61b
FH		 1191 return y;
1192}
1193
c50c0bc8 1194
b6147995
FB		 1195#define RGB_TO_Y(r, g, b) \
1196((FIX(0.29900) * (r) + FIX(0.58700) * (g) + \
1197 FIX(0.11400) * (b) + ONE_HALF) >> SCALEBITS)
1198
c50c0bc8 1199#define RGB_TO_U(r1, g1, b1, shift)\
b6147995 1200(((- FIX(0.16874) * r1 - FIX(0.33126) * g1 + \
c50c0bc8 1201 FIX(0.50000) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
b6147995 1202
c50c0bc8 1203#define RGB_TO_V(r1, g1, b1, shift)\
b6147995 1204(((FIX(0.50000) * r1 - FIX(0.41869) * g1 - \
c50c0bc8 1205 FIX(0.08131) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
b6147995
FB		 1206
1207#define RGB_TO_Y_CCIR(r, g, b) \
1208((FIX(0.29900*219.0/255.0) * (r) + FIX(0.58700*219.0/255.0) * (g) + \
1209 FIX(0.11400*219.0/255.0) * (b) + (ONE_HALF + (16 << SCALEBITS))) >> SCALEBITS)
1210
c50c0bc8 1211#define RGB_TO_U_CCIR(r1, g1, b1, shift)\
b6147995 1212(((- FIX(0.16874*224.0/255.0) * r1 - FIX(0.33126*224.0/255.0) * g1 + \
c50c0bc8 1213 FIX(0.50000*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
b6147995 1214
c50c0bc8 1215#define RGB_TO_V_CCIR(r1, g1, b1, shift)\
b6147995 1216(((FIX(0.50000*224.0/255.0) * r1 - FIX(0.41869*224.0/255.0) * g1 - \
c50c0bc8 1217 FIX(0.08131*224.0/255.0) * b1 + (ONE_HALF << shift) - 1) >> (SCALEBITS + shift)) + 128)
b6147995 1218
c50c0bc8
FB		 1219static uint8_t y_ccir_to_jpeg[256];
1220static uint8_t y_jpeg_to_ccir[256];
1221static uint8_t c_ccir_to_jpeg[256];
1222static uint8_t c_jpeg_to_ccir[256];
1223
1224/* init various conversion tables */
1225static void img_convert_init(void)
b6147995 1226{
c50c0bc8 1227 int i;
b6147995
FB		 1228 uint8_t *cm = cropTbl + MAX_NEG_CROP;
1229
c50c0bc8
FB		 1230 for(i = 0;i < 256; i++) {
1231 y_ccir_to_jpeg[i] = Y_CCIR_TO_JPEG(i);
1232 y_jpeg_to_ccir[i] = Y_JPEG_TO_CCIR(i);
1233 c_ccir_to_jpeg[i] = C_CCIR_TO_JPEG(i);
1234 c_jpeg_to_ccir[i] = C_JPEG_TO_CCIR(i);
b6147995
FB		 1235 }
1236}
1237
c50c0bc8 1238/* apply to each pixel the given table */
115329f1 1239static void img_apply_table(uint8_t *dst, int dst_wrap,
c50c0bc8
FB		 1240 const uint8_t *src, int src_wrap,
1241 int width, int height, const uint8_t *table1)
b6147995
FB		 1242{
1243 int n;
1244 const uint8_t *s;
1245 uint8_t *d;
c50c0bc8 1246 const uint8_t *table;
b6147995 1247
c50c0bc8 1248 table = table1;
b6147995
FB		 1249 for(;height > 0; height--) {
1250 s = src;
1251 d = dst;
1252 n = width;
1253 while (n >= 4) {
c50c0bc8
FB		 1254 d[0] = table[s[0]];
1255 d[1] = table[s[1]];
1256 d[2] = table[s[2]];
1257 d[3] = table[s[3]];
b6147995
FB		 1258 d += 4;
1259 s += 4;
1260 n -= 4;
1261 }
1262 while (n > 0) {
c50c0bc8 1263 d[0] = table[s[0]];
b6147995
FB		 1264 d++;
1265 s++;
1266 n--;
1267 }
1268 dst += dst_wrap;
1269 src += src_wrap;
1270 }
1271}
1272
85c242d8 1273/* XXX: use generic filter ? */
e352ff08
FB		 1274/* XXX: in most cases, the sampling position is incorrect */
1275
1276/* 4x1 -> 1x1 */
115329f1 1277static void shrink41(uint8_t *dst, int dst_wrap,
e352ff08
FB		 1278 const uint8_t *src, int src_wrap,
1279 int width, int height)
1280{
1281 int w;
1282 const uint8_t *s;
1283 uint8_t *d;
1284
1285 for(;height > 0; height--) {
1286 s = src;
1287 d = dst;
1288 for(w = width;w > 0; w--) {
1289 d[0] = (s[0] + s[1] + s[2] + s[3] + 2) >> 2;
1290 s += 4;
1291 d++;
1292 }
1293 src += src_wrap;
1294 dst += dst_wrap;
1295 }
1296}
1297
1298/* 2x1 -> 1x1 */
115329f1 1299static void shrink21(uint8_t *dst, int dst_wrap,
e352ff08
FB		 1300 const uint8_t *src, int src_wrap,
1301 int width, int height)
1302{
1303 int w;
1304 const uint8_t *s;
1305 uint8_t *d;
1306
1307 for(;height > 0; height--) {
1308 s = src;
1309 d = dst;
1310 for(w = width;w > 0; w--) {
1311 d[0] = (s[0] + s[1]) >> 1;
1312 s += 2;
1313 d++;
1314 }
1315 src += src_wrap;
1316 dst += dst_wrap;
1317 }
1318}
1319
85c242d8 1320/* 1x2 -> 1x1 */
115329f1 1321static void shrink12(uint8_t *dst, int dst_wrap,
e352ff08
FB		 1322 const uint8_t *src, int src_wrap,
1323 int width, int height)
85c242d8
FB		 1324{
1325 int w;
e352ff08
FB		 1326 uint8_t *d;
1327 const uint8_t *s1, *s2;
85c242d8
FB		 1328
1329 for(;height > 0; height--) {
1330 s1 = src;
1331 s2 = s1 + src_wrap;
1332 d = dst;
1333 for(w = width;w >= 4; w-=4) {
1334 d[0] = (s1[0] + s2[0]) >> 1;
1335 d[1] = (s1[1] + s2[1]) >> 1;
1336 d[2] = (s1[2] + s2[2]) >> 1;
1337 d[3] = (s1[3] + s2[3]) >> 1;
1338 s1 += 4;
1339 s2 += 4;
1340 d += 4;
1341 }
1342 for(;w > 0; w--) {
1343 d[0] = (s1[0] + s2[0]) >> 1;
1344 s1++;
1345 s2++;
1346 d++;
1347 }
1348 src += 2 * src_wrap;
1349 dst += dst_wrap;
1350 }
1351}
1352
1353/* 2x2 -> 1x1 */
54009d42 1354void ff_shrink22(uint8_t *dst, int dst_wrap,
e352ff08 1355 const uint8_t *src, int src_wrap,
85c242d8
FB		 1356 int width, int height)
1357{
1358 int w;
e352ff08
FB		 1359 const uint8_t *s1, *s2;
1360 uint8_t *d;
85c242d8
FB		 1361
1362 for(;height > 0; height--) {
1363 s1 = src;
1364 s2 = s1 + src_wrap;
1365 d = dst;
1366 for(w = width;w >= 4; w-=4) {
0a9ad8d1
FB		 1367 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
1368 d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
1369 d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
1370 d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
85c242d8
FB		 1371 s1 += 8;
1372 s2 += 8;
1373 d += 4;
1374 }
1375 for(;w > 0; w--) {
0a9ad8d1 1376 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
85c242d8
FB		 1377 s1 += 2;
1378 s2 += 2;
1379 d++;
1380 }
1381 src += 2 * src_wrap;
1382 dst += dst_wrap;
1383 }
1384}
1385
e352ff08 1386/* 4x4 -> 1x1 */
54009d42 1387void ff_shrink44(uint8_t *dst, int dst_wrap,
e352ff08 1388 const uint8_t *src, int src_wrap,
6742d95d
FR		 1389 int width, int height)
1390{
1391 int w;
e352ff08
FB		 1392 const uint8_t *s1, *s2, *s3, *s4;
1393 uint8_t *d;
6742d95d
FR		 1394
1395 for(;height > 0; height--) {
1396 s1 = src;
e352ff08
FB		 1397 s2 = s1 + src_wrap;
1398 s3 = s2 + src_wrap;
1399 s4 = s3 + src_wrap;
6742d95d 1400 d = dst;
e352ff08
FB		 1401 for(w = width;w > 0; w--) {
1402 d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
1403 s2[0] + s2[1] + s2[2] + s2[3] +
1404 s3[0] + s3[1] + s3[2] + s3[3] +
1405 s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
1406 s1 += 4;
1407 s2 += 4;
1408 s3 += 4;
1409 s4 += 4;
6742d95d
FR		 1410 d++;
1411 }
e352ff08
FB		 1412 src += 4 * src_wrap;
1413 dst += dst_wrap;
1414 }
1415}
1416
54009d42
MN		 1417/* 8x8 -> 1x1 */
1418void ff_shrink88(uint8_t *dst, int dst_wrap,
1419 const uint8_t *src, int src_wrap,
1420 int width, int height)
1421{
1422 int w, i;
1423
1424 for(;height > 0; height--) {
1425 for(w = width;w > 0; w--) {
1426 int tmp=0;
1427 for(i=0; i<8; i++){
1428 tmp += src[0] + src[1] + src[2] + src[3] + src[4] + src[5] + src[6] + src[7];
1429 src += src_wrap;
1430 }
1431 *(dst++) = (tmp + 32)>>6;
1432 src += 8 - 8*src_wrap;
1433 }
1434 src += 8*src_wrap - 8*width;
1435 dst += dst_wrap - width;
1436 }
1437}
1438
e352ff08
FB		 1439static void grow21_line(uint8_t *dst, const uint8_t *src,
1440 int width)
1441{
1442 int w;
1443 const uint8_t *s1;
1444 uint8_t *d;
1445
1446 s1 = src;
1447 d = dst;
1448 for(w = width;w >= 4; w-=4) {
1449 d[1] = d[0] = s1[0];
1450 d[3] = d[2] = s1[1];
1451 s1 += 2;
1452 d += 4;
1453 }
1454 for(;w >= 2; w -= 2) {
1455 d[1] = d[0] = s1[0];
1456 s1 ++;
1457 d += 2;
1458 }
1459 /* only needed if width is not a multiple of two */
1460 /* XXX: veryfy that */
1461 if (w) {
1462 d[0] = s1[0];
1463 }
1464}
1465
1466static void grow41_line(uint8_t *dst, const uint8_t *src,
1467 int width)
1468{
1469 int w, v;
1470 const uint8_t *s1;
1471 uint8_t *d;
1472
1473 s1 = src;
1474 d = dst;
1475 for(w = width;w >= 4; w-=4) {
1476 v = s1[0];
1477 d[0] = v;
1478 d[1] = v;
1479 d[2] = v;
1480 d[3] = v;
1481 s1 ++;
1482 d += 4;
1483 }
1484}
1485
1486/* 1x1 -> 2x1 */
1487static void grow21(uint8_t *dst, int dst_wrap,
1488 const uint8_t *src, int src_wrap,
1489 int width, int height)
1490{
1491 for(;height > 0; height--) {
1492 grow21_line(dst, src, width);
1493 src += src_wrap;
1494 dst += dst_wrap;
1495 }
1496}
1497
1498/* 1x1 -> 2x2 */
1499static void grow22(uint8_t *dst, int dst_wrap,
1500 const uint8_t *src, int src_wrap,
1501 int width, int height)
1502{
1503 for(;height > 0; height--) {
1504 grow21_line(dst, src, width);
6742d95d
FR		 1505 if (height%2)
1506 src += src_wrap;
1507 dst += dst_wrap;
1508 }
1509}
1510
e352ff08
FB		 1511/* 1x1 -> 4x1 */
1512static void grow41(uint8_t *dst, int dst_wrap,
1513 const uint8_t *src, int src_wrap,
1514 int width, int height)
1515{
1516 for(;height > 0; height--) {
1517 grow41_line(dst, src, width);
1518 src += src_wrap;
1519 dst += dst_wrap;
1520 }
1521}
1522
1523/* 1x1 -> 4x4 */
1524static void grow44(uint8_t *dst, int dst_wrap,
1525 const uint8_t *src, int src_wrap,
1526 int width, int height)
1527{
1528 for(;height > 0; height--) {
1529 grow41_line(dst, src, width);
1530 if ((height & 3) == 1)
1531 src += src_wrap;
1532 dst += dst_wrap;
1533 }
1534}
1535
524c6b63 1536/* 1x2 -> 2x1 */
115329f1 1537static void conv411(uint8_t *dst, int dst_wrap,
e352ff08 1538 const uint8_t *src, int src_wrap,
789587d5
FB		 1539 int width, int height)
1540{
1541 int w, c;
e352ff08
FB		 1542 const uint8_t *s1, *s2;
1543 uint8_t *d;
789587d5 1544
50643575
MN		 1545 width>>=1;
1546
524c6b63 1547 for(;height > 0; height--) {
789587d5
FB		 1548 s1 = src;
1549 s2 = src + src_wrap;
1550 d = dst;
1551 for(w = width;w > 0; w--) {
1552 c = (s1[0] + s2[0]) >> 1;
1553 d[0] = c;
1554 d[1] = c;
1555 s1++;
1556 s2++;
1557 d += 2;
1558 }
1559 src += src_wrap * 2;
1560 dst += dst_wrap;
1561 }
1562}
1563
7e7e5940
FB		 1564/* XXX: add jpeg quantize code */
1565
1566#define TRANSP_INDEX (6*6*6)
1567
1568/* this is maybe slow, but allows for extensions */
1569static inline unsigned char gif_clut_index(uint8_t r, uint8_t g, uint8_t b)
85c242d8 1570{
7e7e5940 1571 return ((((r)/47)%6)*6*6+(((g)/47)%6)*6+(((b)/47)%6));
85c242d8
FB		 1572}
1573
7e7e5940
FB		 1574static void build_rgb_palette(uint8_t *palette, int has_alpha)
1575{
1576 uint32_t *pal;
1577 static const uint8_t pal_value[6] = { 0x00, 0x33, 0x66, 0x99, 0xcc, 0xff };
1578 int i, r, g, b;
1579
1580 pal = (uint32_t *)palette;
1581 i = 0;
1582 for(r = 0; r < 6; r++) {
1583 for(g = 0; g < 6; g++) {
1584 for(b = 0; b < 6; b++) {
115329f1 1585 pal[i++] = (0xff << 24) | (pal_value[r] << 16) |
7e7e5940
FB		 1586 (pal_value[g] << 8) | pal_value[b];
1587 }
1588 }
1589 }
1590 if (has_alpha)
1591 pal[i++] = 0;
1592 while (i < 256)
1593 pal[i++] = 0xff000000;
524c6b63
FB		 1594}
1595
1596/* copy bit n to bits 0 ... n - 1 */
1597static inline unsigned int bitcopy_n(unsigned int a, int n)
b71472eb 1598{
524c6b63
FB		 1599 int mask;
1600 mask = (1 << n) - 1;
1601 return (a & (0xff & ~mask)) | ((-((a >> n) & 1)) & mask);
1602}
1603
1604/* rgb555 handling */
1605
7e7e5940
FB		 1606#define RGB_NAME rgb555
1607
524c6b63
FB		 1608#define RGB_IN(r, g, b, s)\
1609{\
0c1a9eda 1610 unsigned int v = ((const uint16_t *)(s))[0];\
524c6b63
FB		 1611 r = bitcopy_n(v >> (10 - 3), 3);\
1612 g = bitcopy_n(v >> (5 - 3), 3);\
1613 b = bitcopy_n(v << 3, 3);\
1614}
1615
7e7e5940 1616#define RGBA_IN(r, g, b, a, s)\
524c6b63 1617{\
7e7e5940
FB		 1618 unsigned int v = ((const uint16_t *)(s))[0];\
1619 r = bitcopy_n(v >> (10 - 3), 3);\
1620 g = bitcopy_n(v >> (5 - 3), 3);\
1621 b = bitcopy_n(v << 3, 3);\
b5ff5e22 1622 a = (-(v >> 15)) & 0xff;\
7e7e5940
FB		 1623}
1624
1625#define RGBA_OUT(d, r, g, b, a)\
1626{\
1627 ((uint16_t *)(d))[0] = ((r >> 3) << 10) | ((g >> 3) << 5) | (b >> 3) | \
1628 ((a << 8) & 0x8000);\
524c6b63
FB		 1629}
1630
1631#define BPP 2
1632
7e7e5940 1633#include "imgconvert_template.h"
524c6b63
FB		 1634
1635/* rgb565 handling */
1636
7e7e5940
FB		 1637#define RGB_NAME rgb565
1638
524c6b63
FB		 1639#define RGB_IN(r, g, b, s)\
1640{\
0c1a9eda 1641 unsigned int v = ((const uint16_t *)(s))[0];\
524c6b63
FB		 1642 r = bitcopy_n(v >> (11 - 3), 3);\
1643 g = bitcopy_n(v >> (5 - 2), 2);\
1644 b = bitcopy_n(v << 3, 3);\
1645}
1646
1647#define RGB_OUT(d, r, g, b)\
1648{\
0c1a9eda 1649 ((uint16_t *)(d))[0] = ((r >> 3) << 11) | ((g >> 2) << 5) | (b >> 3);\
524c6b63
FB		 1650}
1651
1652#define BPP 2
1653
7e7e5940 1654#include "imgconvert_template.h"
524c6b63
FB		 1655
1656/* bgr24 handling */
1657
7e7e5940
FB		 1658#define RGB_NAME bgr24
1659
524c6b63
FB		 1660#define RGB_IN(r, g, b, s)\
1661{\
1662 b = (s)[0];\
1663 g = (s)[1];\
1664 r = (s)[2];\
1665}
1666
1667#define RGB_OUT(d, r, g, b)\
1668{\
1669 (d)[0] = b;\
1670 (d)[1] = g;\
1671 (d)[2] = r;\
1672}
1673
1674#define BPP 3
1675
7e7e5940 1676#include "imgconvert_template.h"
524c6b63
FB		 1677
1678#undef RGB_IN
1679#undef RGB_OUT
1680#undef BPP
1681
1682/* rgb24 handling */
1683
7e7e5940
FB		 1684#define RGB_NAME rgb24
1685#define FMT_RGB24
1686
524c6b63
FB		 1687#define RGB_IN(r, g, b, s)\
1688{\
1689 r = (s)[0];\
1690 g = (s)[1];\
1691 b = (s)[2];\
1692}
1693
1694#define RGB_OUT(d, r, g, b)\
1695{\
1696 (d)[0] = r;\
1697 (d)[1] = g;\
1698 (d)[2] = b;\
1699}
1700
1701#define BPP 3
1702
7e7e5940 1703#include "imgconvert_template.h"
524c6b63
FB		 1704
1705/* rgba32 handling */
1706
7e7e5940
FB		 1707#define RGB_NAME rgba32
1708#define FMT_RGBA32
1709
524c6b63
FB		 1710#define RGB_IN(r, g, b, s)\
1711{\
0c1a9eda 1712 unsigned int v = ((const uint32_t *)(s))[0];\
524c6b63
FB		 1713 r = (v >> 16) & 0xff;\
1714 g = (v >> 8) & 0xff;\
1715 b = v & 0xff;\
1716}
1717
7e7e5940 1718#define RGBA_IN(r, g, b, a, s)\
524c6b63 1719{\
7e7e5940
FB		 1720 unsigned int v = ((const uint32_t *)(s))[0];\
1721 a = (v >> 24) & 0xff;\
1722 r = (v >> 16) & 0xff;\
1723 g = (v >> 8) & 0xff;\
1724 b = v & 0xff;\
524c6b63
FB		 1725}
1726
7e7e5940
FB		 1727#define RGBA_OUT(d, r, g, b, a)\
1728{\
1729 ((uint32_t *)(d))[0] = (a << 24) | (r << 16) | (g << 8) | b;\
b71472eb
PG		 1730}
1731
7e7e5940 1732#define BPP 4
524c6b63 1733
7e7e5940 1734#include "imgconvert_template.h"
b71472eb 1735
da64ecc3 1736static void mono_to_gray(AVPicture *dst, const AVPicture *src,
2a877875 1737 int width, int height, int xor_mask)
524c6b63
FB		 1738{
1739 const unsigned char *p;
1740 unsigned char *q;
1741 int v, dst_wrap, src_wrap;
1742 int y, w;
1743
1744 p = src->data[0];
1745 src_wrap = src->linesize[0] - ((width + 7) >> 3);
1746
1747 q = dst->data[0];
2a877875 1748 dst_wrap = dst->linesize[0] - width;
524c6b63 1749 for(y=0;y<height;y++) {
115329f1 1750 w = width;
524c6b63 1751 while (w >= 8) {
2a877875
FB		 1752 v = *p++ ^ xor_mask;
1753 q[0] = -(v >> 7);
1754 q[1] = -((v >> 6) & 1);
1755 q[2] = -((v >> 5) & 1);
1756 q[3] = -((v >> 4) & 1);
1757 q[4] = -((v >> 3) & 1);
1758 q[5] = -((v >> 2) & 1);
1759 q[6] = -((v >> 1) & 1);
1760 q[7] = -((v >> 0) & 1);
524c6b63 1761 w -= 8;
2a877875 1762 q += 8;
524c6b63
FB		 1763 }
1764 if (w > 0) {
2a877875 1765 v = *p++ ^ xor_mask;
524c6b63 1766 do {
2a877875
FB		 1767 q[0] = -((v >> 7) & 1);
1768 q++;
524c6b63
FB		 1769 v <<= 1;
1770 } while (--w);
85c242d8 1771 }
524c6b63
FB		 1772 p += src_wrap;
1773 q += dst_wrap;
85c242d8
FB		 1774 }
1775}
1776
da64ecc3 1777static void monowhite_to_gray(AVPicture *dst, const AVPicture *src,
524c6b63
FB		 1778 int width, int height)
1779{
2a877875
FB		 1780 mono_to_gray(dst, src, width, height, 0xff);
1781}
524c6b63 1782
da64ecc3 1783static void monoblack_to_gray(AVPicture *dst, const AVPicture *src,
2a877875
FB		 1784 int width, int height)
1785{
1786 mono_to_gray(dst, src, width, height, 0x00);
1787}
524c6b63 1788
da64ecc3 1789static void gray_to_mono(AVPicture *dst, const AVPicture *src,
2a877875
FB		 1790 int width, int height, int xor_mask)
1791{
1792 int n;
0c1a9eda
ZK		 1793 const uint8_t *s;
1794 uint8_t *d;
2a877875
FB		 1795 int j, b, v, n1, src_wrap, dst_wrap, y;
1796
1797 s = src->data[0];
1798 src_wrap = src->linesize[0] - width;
1799
1800 d = dst->data[0];
1801 dst_wrap = dst->linesize[0] - ((width + 7) >> 3);
524c6b63
FB		 1802
1803 for(y=0;y<height;y++) {
2a877875
FB		 1804 n = width;
1805 while (n >= 8) {
1806 v = 0;
1807 for(j=0;j<8;j++) {
1808 b = s[0];
1809 s++;
1810 v = (v << 1) | (b >> 7);
1811 }
1812 d[0] = v ^ xor_mask;
1813 d++;
1814 n -= 8;
524c6b63 1815 }
2a877875
FB		 1816 if (n > 0) {
1817 n1 = n;
1818 v = 0;
1819 while (n > 0) {
1820 b = s[0];
1821 s++;
1822 v = (v << 1) | (b >> 7);
1823 n--;
1824 }
1825 d[0] = (v << (8 - (n1 & 7))) ^ xor_mask;
1826 d++;
524c6b63 1827 }
2a877875
FB		 1828 s += src_wrap;
1829 d += dst_wrap;
524c6b63
FB		 1830 }
1831}
1832
da64ecc3 1833static void gray_to_monowhite(AVPicture *dst, const AVPicture *src,
2a877875
FB		 1834 int width, int height)
1835{
1836 gray_to_mono(dst, src, width, height, 0xff);
1837}
1838
da64ecc3 1839static void gray_to_monoblack(AVPicture *dst, const AVPicture *src,
2a877875
FB		 1840 int width, int height)
1841{
1842 gray_to_mono(dst, src, width, height, 0x00);
1843}
1844
524c6b63 1845typedef struct ConvertEntry {
da64ecc3 1846 void (*convert)(AVPicture *dst,
bb270c08 1847 const AVPicture *src, int width, int height);
524c6b63
FB		 1848} ConvertEntry;
1849
c50c0bc8
FB		 1850/* Add each new convertion function in this table. In order to be able
1851 to convert from any format to any format, the following constraints
1852 must be satisfied:
1853
115329f1 1854 - all FF_COLOR_RGB formats must convert to and from PIX_FMT_RGB24
c50c0bc8
FB		 1855
1856 - all FF_COLOR_GRAY formats must convert to and from PIX_FMT_GRAY8
1857
1858 - all FF_COLOR_RGB formats with alpha must convert to and from PIX_FMT_RGBA32
1859
e352ff08 1860 - PIX_FMT_YUV444P and PIX_FMT_YUVJ444P must convert to and from
c50c0bc8
FB		 1861 PIX_FMT_RGB24.
1862
1863 - PIX_FMT_422 must convert to and from PIX_FMT_422P.
e352ff08
FB		 1864
1865 The other conversion functions are just optimisations for common cases.
524c6b63 1866*/
62a05b5b 1867static const ConvertEntry convert_table[PIX_FMT_NB][PIX_FMT_NB] = {
524c6b63 1868 [PIX_FMT_YUV420P] = {
d7e2f57f
MN		 1869 [PIX_FMT_YUV422] = {
1870 .convert = yuv420p_to_yuv422,
1871 },
115329f1 1872 [PIX_FMT_RGB555] = {
ef9f7306 1873 .convert = yuv420p_to_rgb555
524c6b63 1874 },
115329f1 1875 [PIX_FMT_RGB565] = {
ef9f7306 1876 .convert = yuv420p_to_rgb565
524c6b63 1877 },
115329f1 1878 [PIX_FMT_BGR24] = {
ef9f7306 1879 .convert = yuv420p_to_bgr24
524c6b63 1880 },
115329f1 1881 [PIX_FMT_RGB24] = {
ef9f7306 1882 .convert = yuv420p_to_rgb24
524c6b63 1883 },
115329f1 1884 [PIX_FMT_RGBA32] = {
ef9f7306 1885 .convert = yuv420p_to_rgba32
524c6b63 1886 },
bb270c08 1887 [PIX_FMT_UYVY422] = {
bac65165
LA		 1888 .convert = yuv420p_to_uyvy422,
1889 },
524c6b63 1890 },
115329f1
DB		 1891 [PIX_FMT_YUV422P] = {
1892 [PIX_FMT_YUV422] = {
c50c0bc8
FB		 1893 .convert = yuv422p_to_yuv422,
1894 },
115329f1 1895 [PIX_FMT_UYVY422] = {
ebb177dd
TK		 1896 .convert = yuv422p_to_uyvy422,
1897 },
c50c0bc8 1898 },
115329f1
DB		 1899 [PIX_FMT_YUV444P] = {
1900 [PIX_FMT_RGB24] = {
c50c0bc8
FB		 1901 .convert = yuv444p_to_rgb24
1902 },
1903 },
1904 [PIX_FMT_YUVJ420P] = {
115329f1 1905 [PIX_FMT_RGB555] = {
c50c0bc8 1906 .convert = yuvj420p_to_rgb555
524c6b63 1907 },
115329f1 1908 [PIX_FMT_RGB565] = {
c50c0bc8 1909 .convert = yuvj420p_to_rgb565
524c6b63 1910 },
115329f1 1911 [PIX_FMT_BGR24] = {
c50c0bc8 1912 .convert = yuvj420p_to_bgr24
524c6b63 1913 },
115329f1 1914 [PIX_FMT_RGB24] = {
c50c0bc8 1915 .convert = yuvj420p_to_rgb24
524c6b63 1916 },
115329f1 1917 [PIX_FMT_RGBA32] = {
c50c0bc8
FB		 1918 .convert = yuvj420p_to_rgba32
1919 },
1920 },
115329f1
DB		 1921 [PIX_FMT_YUVJ444P] = {
1922 [PIX_FMT_RGB24] = {
c50c0bc8 1923 .convert = yuvj444p_to_rgb24
524c6b63
FB		 1924 },
1925 },
115329f1
DB		 1926 [PIX_FMT_YUV422] = {
1927 [PIX_FMT_YUV420P] = {
ef9f7306 1928 .convert = yuv422_to_yuv420p,
524c6b63 1929 },
115329f1 1930 [PIX_FMT_YUV422P] = {
c50c0bc8
FB		 1931 .convert = yuv422_to_yuv422p,
1932 },
524c6b63 1933 },
115329f1
DB		 1934 [PIX_FMT_UYVY422] = {
1935 [PIX_FMT_YUV420P] = {
ebb177dd
TK		 1936 .convert = uyvy422_to_yuv420p,
1937 },
115329f1 1938 [PIX_FMT_YUV422P] = {
ebb177dd
TK		 1939 .convert = uyvy422_to_yuv422p,
1940 },
1941 },
524c6b63 1942 [PIX_FMT_RGB24] = {
115329f1 1943 [PIX_FMT_YUV420P] = {
ef9f7306 1944 .convert = rgb24_to_yuv420p
524c6b63 1945 },
115329f1 1946 [PIX_FMT_RGB565] = {
ef9f7306 1947 .convert = rgb24_to_rgb565
524c6b63 1948 },
115329f1 1949 [PIX_FMT_RGB555] = {
ef9f7306 1950 .convert = rgb24_to_rgb555
524c6b63 1951 },
115329f1 1952 [PIX_FMT_RGBA32] = {
7e7e5940
FB		 1953 .convert = rgb24_to_rgba32
1954 },
115329f1 1955 [PIX_FMT_BGR24] = {
7e7e5940
FB		 1956 .convert = rgb24_to_bgr24
1957 },
115329f1 1958 [PIX_FMT_GRAY8] = {
ef9f7306 1959 .convert = rgb24_to_gray
524c6b63 1960 },
7e7e5940 1961 [PIX_FMT_PAL8] = {
7e6d70d0
FB		 1962 .convert = rgb24_to_pal8
1963 },
115329f1 1964 [PIX_FMT_YUV444P] = {
c50c0bc8
FB		 1965 .convert = rgb24_to_yuv444p
1966 },
115329f1 1967 [PIX_FMT_YUVJ420P] = {
c50c0bc8
FB		 1968 .convert = rgb24_to_yuvj420p
1969 },
115329f1 1970 [PIX_FMT_YUVJ444P] = {
c50c0bc8
FB		 1971 .convert = rgb24_to_yuvj444p
1972 },
524c6b63
FB		 1973 },
1974 [PIX_FMT_RGBA32] = {
115329f1 1975 [PIX_FMT_RGB24] = {
7e7e5940
FB		 1976 .convert = rgba32_to_rgb24
1977 },
115329f1 1978 [PIX_FMT_RGB555] = {
7e7e5940
FB		 1979 .convert = rgba32_to_rgb555
1980 },
115329f1 1981 [PIX_FMT_PAL8] = {
7e7e5940
FB		 1982 .convert = rgba32_to_pal8
1983 },
115329f1 1984 [PIX_FMT_YUV420P] = {
ef9f7306 1985 .convert = rgba32_to_yuv420p
524c6b63 1986 },
115329f1 1987 [PIX_FMT_GRAY8] = {
69572401
FB		 1988 .convert = rgba32_to_gray
1989 },
524c6b63
FB		 1990 },
1991 [PIX_FMT_BGR24] = {
115329f1 1992 [PIX_FMT_RGB24] = {
7e7e5940
FB		 1993 .convert = bgr24_to_rgb24
1994 },
115329f1 1995 [PIX_FMT_YUV420P] = {
ef9f7306 1996 .convert = bgr24_to_yuv420p
524c6b63 1997 },
115329f1 1998 [PIX_FMT_GRAY8] = {
69572401
FB		 1999 .convert = bgr24_to_gray
2000 },
524c6b63
FB		 2001 },
2002 [PIX_FMT_RGB555] = {
115329f1 2003 [PIX_FMT_RGB24] = {
7e7e5940
FB		 2004 .convert = rgb555_to_rgb24
2005 },
115329f1 2006 [PIX_FMT_RGBA32] = {
7e7e5940
FB		 2007 .convert = rgb555_to_rgba32
2008 },
115329f1 2009 [PIX_FMT_YUV420P] = {
ef9f7306 2010 .convert = rgb555_to_yuv420p
524c6b63 2011 },
115329f1 2012 [PIX_FMT_GRAY8] = {
69572401
FB		 2013 .convert = rgb555_to_gray
2014 },
524c6b63
FB		 2015 },
2016 [PIX_FMT_RGB565] = {
115329f1 2017 [PIX_FMT_RGB24] = {
7e7e5940
FB		 2018 .convert = rgb565_to_rgb24
2019 },
115329f1 2020 [PIX_FMT_YUV420P] = {
ef9f7306 2021 .convert = rgb565_to_yuv420p
524c6b63 2022 },
115329f1 2023 [PIX_FMT_GRAY8] = {
69572401
FB		 2024 .convert = rgb565_to_gray
2025 },
524c6b63
FB		 2026 },
2027 [PIX_FMT_GRAY8] = {
115329f1 2028 [PIX_FMT_RGB555] = {
69572401
FB		 2029 .convert = gray_to_rgb555
2030 },
115329f1 2031 [PIX_FMT_RGB565] = {
69572401
FB		 2032 .convert = gray_to_rgb565
2033 },
115329f1 2034 [PIX_FMT_RGB24] = {
ef9f7306 2035 .convert = gray_to_rgb24
524c6b63 2036 },
115329f1 2037 [PIX_FMT_BGR24] = {
69572401
FB		 2038 .convert = gray_to_bgr24
2039 },
115329f1 2040 [PIX_FMT_RGBA32] = {
69572401
FB		 2041 .convert = gray_to_rgba32
2042 },
115329f1 2043 [PIX_FMT_MONOWHITE] = {
ef9f7306 2044 .convert = gray_to_monowhite
2a877875 2045 },
115329f1 2046 [PIX_FMT_MONOBLACK] = {
ef9f7306 2047 .convert = gray_to_monoblack
2a877875 2048 },
524c6b63
FB		 2049 },
2050 [PIX_FMT_MONOWHITE] = {
115329f1 2051 [PIX_FMT_GRAY8] = {
ef9f7306 2052 .convert = monowhite_to_gray
524c6b63
FB		 2053 },
2054 },
2055 [PIX_FMT_MONOBLACK] = {
115329f1 2056 [PIX_FMT_GRAY8] = {
ef9f7306 2057 .convert = monoblack_to_gray
524c6b63
FB		 2058 },
2059 },
7e6d70d0 2060 [PIX_FMT_PAL8] = {
115329f1 2061 [PIX_FMT_RGB555] = {
7e6d70d0
FB		 2062 .convert = pal8_to_rgb555
2063 },
115329f1 2064 [PIX_FMT_RGB565] = {
7e6d70d0
FB		 2065 .convert = pal8_to_rgb565
2066 },
115329f1 2067 [PIX_FMT_BGR24] = {
7e6d70d0
FB		 2068 .convert = pal8_to_bgr24
2069 },
115329f1 2070 [PIX_FMT_RGB24] = {
7e6d70d0
FB		 2071 .convert = pal8_to_rgb24
2072 },
115329f1 2073 [PIX_FMT_RGBA32] = {
7e6d70d0
FB		 2074 .convert = pal8_to_rgba32
2075 },
2076 },
115329f1
DB		 2077 [PIX_FMT_UYVY411] = {
2078 [PIX_FMT_YUV411P] = {
f02be79d
RS		 2079 .convert = uyvy411_to_yuv411p,
2080 },
2081 },
2082
524c6b63
FB		 2083};
2084
75917b88 2085int avpicture_alloc(AVPicture *picture,
524c6b63
FB		 2086 int pix_fmt, int width, int height)
2087{
2d5545c3 2088 int size;
524c6b63
FB		 2089 void *ptr;
2090
2091 size = avpicture_get_size(pix_fmt, width, height);
0ecca7a4
MN		 2092 if(size<0)
2093 goto fail;
524c6b63
FB		 2094 ptr = av_malloc(size);
2095 if (!ptr)
2096 goto fail;
2097 avpicture_fill(picture, ptr, pix_fmt, width, height);
2098 return 0;
2099 fail:
2100 memset(picture, 0, sizeof(AVPicture));
2101 return -1;
2102}
2103
75917b88 2104void avpicture_free(AVPicture *picture)
524c6b63 2105{
8e1e6f31 2106 av_free(picture->data[0]);
524c6b63
FB		 2107}
2108
c50c0bc8 2109/* return true if yuv planar */
62a05b5b 2110static inline int is_yuv_planar(const PixFmtInfo *ps)
c50c0bc8
FB		 2111{
2112 return (ps->color_type == FF_COLOR_YUV ||
115329f1 2113 ps->color_type == FF_COLOR_YUV_JPEG) &&
7e7e5940 2114 ps->pixel_type == FF_PIXEL_PLANAR;
c50c0bc8
FB		 2115}
2116
f2651e7a
BC		 2117/**
2118 * Crop image top and left side
2119 */
2120int img_crop(AVPicture *dst, const AVPicture *src,
2121 int pix_fmt, int top_band, int left_band)
2122{
2123 int y_shift;
2124 int x_shift;
2125
2126 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB || !is_yuv_planar(&pix_fmt_info[pix_fmt]))
2127 return -1;
2128
2129 y_shift = pix_fmt_info[pix_fmt].y_chroma_shift;
2130 x_shift = pix_fmt_info[pix_fmt].x_chroma_shift;
2131
2132 dst->data[0] = src->data[0] + (top_band * src->linesize[0]) + left_band;
2133 dst->data[1] = src->data[1] + ((top_band >> y_shift) * src->linesize[1]) + (left_band >> x_shift);
2134 dst->data[2] = src->data[2] + ((top_band >> y_shift) * src->linesize[2]) + (left_band >> x_shift);
2135
2136 dst->linesize[0] = src->linesize[0];
2137 dst->linesize[1] = src->linesize[1];
2138 dst->linesize[2] = src->linesize[2];
2139 return 0;
2140}
2141
5341c209
LA		 2142/**
2143 * Pad image
2144 */
2145int img_pad(AVPicture *dst, const AVPicture *src, int height, int width, int pix_fmt,
2146 int padtop, int padbottom, int padleft, int padright, int *color)
2147{
2148 uint8_t *optr, *iptr;
2149 int y_shift;
2150 int x_shift;
2151 int yheight;
2152 int i, y;
2153
2154 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB || !is_yuv_planar(&pix_fmt_info[pix_fmt]))
2155 return -1;
2156
2157 for (i = 0; i < 3; i++) {
2158 x_shift = i ? pix_fmt_info[pix_fmt].x_chroma_shift : 0;
2159 y_shift = i ? pix_fmt_info[pix_fmt].y_chroma_shift : 0;
2160
2161 if (padtop || padleft) {
2162 memset(dst->data[i], color[i], dst->linesize[i] * (padtop >> y_shift) + (padleft >> x_shift));
2163 }
2164
2165 if (padleft || padright || src) {
2166 if (src) { /* first line */
2167 iptr = src->data[i];
2168 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) + (padleft >> x_shift);
2169 memcpy(optr, iptr, src->linesize[i]);
2170 iptr += src->linesize[i];
2171 }
2172 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) + (dst->linesize[i] - (padright >> x_shift));
2173 yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
2174 for (y = 0; y < yheight; y++) {
2175 memset(optr, color[i], (padleft + padright) >> x_shift);
2176 if (src) {
2177 memcpy(optr + ((padleft + padright) >> x_shift), iptr, src->linesize[i]);
2178 iptr += src->linesize[i];
2179 }
2180 optr += dst->linesize[i];
2181 }
2182 }
2183
2184 if (padbottom || padright) {
2185 optr = dst->data[i] + dst->linesize[i] * ((height - padbottom) >> y_shift) - (padright >> x_shift);
2186 memset(optr, color[i], dst->linesize[i] * (padbottom >> y_shift) + (padright >> x_shift));
2187 }
2188 }
2189 return 0;
2190}
2191
790c9ca7 2192#ifndef CONFIG_SWSCALER
85c242d8
FB		 2193/* XXX: always use linesize. Return -1 if not supported */
2194int img_convert(AVPicture *dst, int dst_pix_fmt,
115329f1 2195 const AVPicture *src, int src_pix_fmt,
524c6b63 2196 int src_width, int src_height)
85c242d8 2197{
c50c0bc8 2198 static int inited;
2a877875 2199 int i, ret, dst_width, dst_height, int_pix_fmt;
62a05b5b
SH		 2200 const PixFmtInfo *src_pix, *dst_pix;
2201 const ConvertEntry *ce;
524c6b63
FB		 2202 AVPicture tmp1, *tmp = &tmp1;
2203
2204 if (src_pix_fmt < 0 || src_pix_fmt >= PIX_FMT_NB ||
2205 dst_pix_fmt < 0 || dst_pix_fmt >= PIX_FMT_NB)
2206 return -1;
2207 if (src_width <= 0 || src_height <= 0)
2208 return 0;
69572401 2209
c50c0bc8
FB		 2210 if (!inited) {
2211 inited = 1;
2212 img_convert_init();
2213 }
2214
524c6b63
FB		 2215 dst_width = src_width;
2216 dst_height = src_height;
69572401 2217
524c6b63
FB		 2218 dst_pix = &pix_fmt_info[dst_pix_fmt];
2219 src_pix = &pix_fmt_info[src_pix_fmt];
2220 if (src_pix_fmt == dst_pix_fmt) {
7e7e5940
FB		 2221 /* no conversion needed: just copy */
2222 img_copy(dst, src, dst_pix_fmt, dst_width, dst_height);
524c6b63
FB		 2223 return 0;
2224 }
2225
2226 ce = &convert_table[src_pix_fmt][dst_pix_fmt];
2227 if (ce->convert) {
ebb177dd 2228 /* specific conversion routine */
524c6b63
FB		 2229 ce->convert(dst, src, dst_width, dst_height);
2230 return 0;
2231 }
2232
524c6b63 2233 /* gray to YUV */
c50c0bc8 2234 if (is_yuv_planar(dst_pix) &&
b6147995 2235 src_pix_fmt == PIX_FMT_GRAY8) {
524c6b63
FB		 2236 int w, h, y;
2237 uint8_t *d;
2238
b6147995 2239 if (dst_pix->color_type == FF_COLOR_YUV_JPEG) {
54009d42 2240 ff_img_copy_plane(dst->data[0], dst->linesize[0],
b6147995
FB		 2241 src->data[0], src->linesize[0],
2242 dst_width, dst_height);
2243 } else {
c50c0bc8
FB		 2244 img_apply_table(dst->data[0], dst->linesize[0],
2245 src->data[0], src->linesize[0],
2246 dst_width, dst_height,
2247 y_jpeg_to_ccir);
b6147995 2248 }
524c6b63
FB		 2249 /* fill U and V with 128 */
2250 w = dst_width;
2251 h = dst_height;
2252 w >>= dst_pix->x_chroma_shift;
2253 h >>= dst_pix->y_chroma_shift;
2254 for(i = 1; i <= 2; i++) {
2255 d = dst->data[i];
2a877875
FB		 2256 for(y = 0; y< h; y++) {
2257 memset(d, 128, w);
524c6b63
FB		 2258 d += dst->linesize[i];
2259 }
b71472eb 2260 }
524c6b63
FB		 2261 return 0;
2262 }
2263
2264 /* YUV to gray */
115329f1 2265 if (is_yuv_planar(src_pix) &&
b6147995
FB		 2266 dst_pix_fmt == PIX_FMT_GRAY8) {
2267 if (src_pix->color_type == FF_COLOR_YUV_JPEG) {
54009d42 2268 ff_img_copy_plane(dst->data[0], dst->linesize[0],
b6147995
FB		 2269 src->data[0], src->linesize[0],
2270 dst_width, dst_height);
2271 } else {
c50c0bc8
FB		 2272 img_apply_table(dst->data[0], dst->linesize[0],
2273 src->data[0], src->linesize[0],
2274 dst_width, dst_height,
2275 y_ccir_to_jpeg);
b6147995 2276 }
524c6b63
FB		 2277 return 0;
2278 }
2279
c50c0bc8
FB		 2280 /* YUV to YUV planar */
2281 if (is_yuv_planar(dst_pix) && is_yuv_planar(src_pix)) {
e352ff08 2282 int x_shift, y_shift, w, h, xy_shift;
115329f1 2283 void (*resize_func)(uint8_t *dst, int dst_wrap,
e352ff08 2284 const uint8_t *src, int src_wrap,
524c6b63
FB		 2285 int width, int height);
2286
2287 /* compute chroma size of the smallest dimensions */
2288 w = dst_width;
2289 h = dst_height;
2290 if (dst_pix->x_chroma_shift >= src_pix->x_chroma_shift)
2291 w >>= dst_pix->x_chroma_shift;
2292 else
2293 w >>= src_pix->x_chroma_shift;
2294 if (dst_pix->y_chroma_shift >= src_pix->y_chroma_shift)
2295 h >>= dst_pix->y_chroma_shift;
2296 else
2297 h >>= src_pix->y_chroma_shift;
2298
2299 x_shift = (dst_pix->x_chroma_shift - src_pix->x_chroma_shift);
2300 y_shift = (dst_pix->y_chroma_shift - src_pix->y_chroma_shift);
e352ff08
FB		 2301 xy_shift = ((x_shift & 0xf) << 4) | (y_shift & 0xf);
2302 /* there must be filters for conversion at least from and to
2303 YUV444 format */
2304 switch(xy_shift) {
2305 case 0x00:
54009d42 2306 resize_func = ff_img_copy_plane;
e352ff08
FB		 2307 break;
2308 case 0x10:
2309 resize_func = shrink21;
2310 break;
2311 case 0x20:
2312 resize_func = shrink41;
2313 break;
2314 case 0x01:
2315 resize_func = shrink12;
2316 break;
2317 case 0x11:
54009d42 2318 resize_func = ff_shrink22;
e352ff08
FB		 2319 break;
2320 case 0x22:
54009d42 2321 resize_func = ff_shrink44;
e352ff08
FB		 2322 break;
2323 case 0xf0:
2324 resize_func = grow21;
2325 break;
2326 case 0xe0:
2327 resize_func = grow41;
2328 break;
2329 case 0xff:
524c6b63 2330 resize_func = grow22;
e352ff08
FB		 2331 break;
2332 case 0xee:
2333 resize_func = grow44;
2334 break;
2335 case 0xf1:
524c6b63 2336 resize_func = conv411;
e352ff08
FB		 2337 break;
2338 default:
524c6b63 2339 /* currently not handled */
e352ff08 2340 goto no_chroma_filter;
85c242d8 2341 }
524c6b63 2342
54009d42 2343 ff_img_copy_plane(dst->data[0], dst->linesize[0],
7e7e5940
FB		 2344 src->data[0], src->linesize[0],
2345 dst_width, dst_height);
566986ee 2346
524c6b63 2347 for(i = 1;i <= 2; i++)
566986ee
MK		 2348 resize_func(dst->data[i], dst->linesize[i],
2349 src->data[i], src->linesize[i],
185fdc54 2350 dst_width>>dst_pix->x_chroma_shift, dst_height>>dst_pix->y_chroma_shift);
c50c0bc8
FB		 2351 /* if yuv color space conversion is needed, we do it here on
2352 the destination image */
2353 if (dst_pix->color_type != src_pix->color_type) {
2354 const uint8_t *y_table, *c_table;
2355 if (dst_pix->color_type == FF_COLOR_YUV) {
2356 y_table = y_jpeg_to_ccir;
2357 c_table = c_jpeg_to_ccir;
2358 } else {
2359 y_table = y_ccir_to_jpeg;
2360 c_table = c_ccir_to_jpeg;
2361 }
2362 img_apply_table(dst->data[0], dst->linesize[0],
2363 dst->data[0], dst->linesize[0],
2364 dst_width, dst_height,
2365 y_table);
2366
2367 for(i = 1;i <= 2; i++)
2368 img_apply_table(dst->data[i], dst->linesize[i],
2369 dst->data[i], dst->linesize[i],
115329f1 2370 dst_width>>dst_pix->x_chroma_shift,
c50c0bc8
FB		 2371 dst_height>>dst_pix->y_chroma_shift,
2372 c_table);
2373 }
2374 return 0;
85c242d8 2375 }
e352ff08 2376 no_chroma_filter:
524c6b63 2377
2a877875 2378 /* try to use an intermediate format */
c50c0bc8
FB		 2379 if (src_pix_fmt == PIX_FMT_YUV422 ||
2380 dst_pix_fmt == PIX_FMT_YUV422) {
2381 /* specific case: convert to YUV422P first */
2382 int_pix_fmt = PIX_FMT_YUV422P;
ebb177dd
TK		 2383 } else if (src_pix_fmt == PIX_FMT_UYVY422 ||
2384 dst_pix_fmt == PIX_FMT_UYVY422) {
2385 /* specific case: convert to YUV422P first */
2386 int_pix_fmt = PIX_FMT_YUV422P;
f02be79d
RS		 2387 } else if (src_pix_fmt == PIX_FMT_UYVY411 ||
2388 dst_pix_fmt == PIX_FMT_UYVY411) {
2389 /* specific case: convert to YUV411P first */
2390 int_pix_fmt = PIX_FMT_YUV411P;
c50c0bc8 2391 } else if ((src_pix->color_type == FF_COLOR_GRAY &&
115329f1 2392 src_pix_fmt != PIX_FMT_GRAY8) ||
c50c0bc8
FB		 2393 (dst_pix->color_type == FF_COLOR_GRAY &&
2394 dst_pix_fmt != PIX_FMT_GRAY8)) {
2395 /* gray8 is the normalized format */
2a877875 2396 int_pix_fmt = PIX_FMT_GRAY8;
115329f1 2397 } else if ((is_yuv_planar(src_pix) &&
c50c0bc8
FB		 2398 src_pix_fmt != PIX_FMT_YUV444P &&
2399 src_pix_fmt != PIX_FMT_YUVJ444P)) {
2400 /* yuv444 is the normalized format */
2401 if (src_pix->color_type == FF_COLOR_YUV_JPEG)
2402 int_pix_fmt = PIX_FMT_YUVJ444P;
2403 else
2404 int_pix_fmt = PIX_FMT_YUV444P;
115329f1 2405 } else if ((is_yuv_planar(dst_pix) &&
c50c0bc8
FB		 2406 dst_pix_fmt != PIX_FMT_YUV444P &&
2407 dst_pix_fmt != PIX_FMT_YUVJ444P)) {
2408 /* yuv444 is the normalized format */
2409 if (dst_pix->color_type == FF_COLOR_YUV_JPEG)
2410 int_pix_fmt = PIX_FMT_YUVJ444P;
2411 else
2412 int_pix_fmt = PIX_FMT_YUV444P;
2a877875 2413 } else {
c50c0bc8
FB		 2414 /* the two formats are rgb or gray8 or yuv[j]444p */
2415 if (src_pix->is_alpha && dst_pix->is_alpha)
2416 int_pix_fmt = PIX_FMT_RGBA32;
2417 else
2418 int_pix_fmt = PIX_FMT_RGB24;
2a877875
FB		 2419 }
2420 if (avpicture_alloc(tmp, int_pix_fmt, dst_width, dst_height) < 0)
2421 return -1;
2422 ret = -1;
2423 if (img_convert(tmp, int_pix_fmt,
2424 src, src_pix_fmt, src_width, src_height) < 0)
2425 goto fail1;
2426 if (img_convert(dst, dst_pix_fmt,
2427 tmp, int_pix_fmt, dst_width, dst_height) < 0)
2428 goto fail1;
2429 ret = 0;
2430 fail1:
2431 avpicture_free(tmp);
2432 return ret;
85c242d8 2433}
790c9ca7 2434#endif
85c242d8 2435
0469baf1 2436/* NOTE: we scan all the pixels to have an exact information */
da64ecc3 2437static int get_alpha_info_pal8(const AVPicture *src, int width, int height)
0469baf1
FB		 2438{
2439 const unsigned char *p;
2440 int src_wrap, ret, x, y;
2441 unsigned int a;
2442 uint32_t *palette = (uint32_t *)src->data[1];
115329f1 2443
0469baf1
FB		 2444 p = src->data[0];
2445 src_wrap = src->linesize[0] - width;
2446 ret = 0;
2447 for(y=0;y<height;y++) {
2448 for(x=0;x<width;x++) {
2449 a = palette[p[0]] >> 24;
2450 if (a == 0x00) {
2451 ret |= FF_ALPHA_TRANSP;
2452 } else if (a != 0xff) {
2453 ret |= FF_ALPHA_SEMI_TRANSP;
2454 }
2455 p++;
2456 }
2457 p += src_wrap;
2458 }
2459 return ret;
2460}
2461
2462/**
2463 * Tell if an image really has transparent alpha values.
2464 * @return ored mask of FF_ALPHA_xxx constants
2465 */
da64ecc3 2466int img_get_alpha_info(const AVPicture *src,
bb270c08 2467 int pix_fmt, int width, int height)
0469baf1 2468{
62a05b5b 2469 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
0469baf1
FB		 2470 int ret;
2471
2472 pf = &pix_fmt_info[pix_fmt];
2473 /* no alpha can be represented in format */
2474 if (!pf->is_alpha)
2475 return 0;
2476 switch(pix_fmt) {
2477 case PIX_FMT_RGBA32:
2478 ret = get_alpha_info_rgba32(src, width, height);
2479 break;
2480 case PIX_FMT_RGB555:
2481 ret = get_alpha_info_rgb555(src, width, height);
2482 break;
2483 case PIX_FMT_PAL8:
2484 ret = get_alpha_info_pal8(src, width, height);
2485 break;
2486 default:
2487 /* we do not know, so everything is indicated */
2488 ret = FF_ALPHA_TRANSP | FF_ALPHA_SEMI_TRANSP;
2489 break;
2490 }
2491 return ret;
2492}
5981f4e6
F		 2493
2494#ifdef HAVE_MMX
2495#define DEINT_INPLACE_LINE_LUM \
2496 movd_m2r(lum_m4[0],mm0);\
2497 movd_m2r(lum_m3[0],mm1);\
2498 movd_m2r(lum_m2[0],mm2);\
2499 movd_m2r(lum_m1[0],mm3);\
2500 movd_m2r(lum[0],mm4);\
2501 punpcklbw_r2r(mm7,mm0);\
2502 movd_r2m(mm2,lum_m4[0]);\
2503 punpcklbw_r2r(mm7,mm1);\
2504 punpcklbw_r2r(mm7,mm2);\
2505 punpcklbw_r2r(mm7,mm3);\
2506 punpcklbw_r2r(mm7,mm4);\
2507 paddw_r2r(mm3,mm1);\
2508 psllw_i2r(1,mm2);\
2509 paddw_r2r(mm4,mm0);\
2510 psllw_i2r(2,mm1);\
2511 paddw_r2r(mm6,mm2);\
2512 paddw_r2r(mm2,mm1);\
2513 psubusw_r2r(mm0,mm1);\
2514 psrlw_i2r(3,mm1);\
2515 packuswb_r2r(mm7,mm1);\
2516 movd_r2m(mm1,lum_m2[0]);
2517
2518#define DEINT_LINE_LUM \
2519 movd_m2r(lum_m4[0],mm0);\
2520 movd_m2r(lum_m3[0],mm1);\
2521 movd_m2r(lum_m2[0],mm2);\
2522 movd_m2r(lum_m1[0],mm3);\
2523 movd_m2r(lum[0],mm4);\
2524 punpcklbw_r2r(mm7,mm0);\
2525 punpcklbw_r2r(mm7,mm1);\
2526 punpcklbw_r2r(mm7,mm2);\
2527 punpcklbw_r2r(mm7,mm3);\
2528 punpcklbw_r2r(mm7,mm4);\
2529 paddw_r2r(mm3,mm1);\
2530 psllw_i2r(1,mm2);\
2531 paddw_r2r(mm4,mm0);\
2532 psllw_i2r(2,mm1);\
2533 paddw_r2r(mm6,mm2);\
2534 paddw_r2r(mm2,mm1);\
2535 psubusw_r2r(mm0,mm1);\
2536 psrlw_i2r(3,mm1);\
2537 packuswb_r2r(mm7,mm1);\
2538 movd_r2m(mm1,dst[0]);
2539#endif
2540
85c242d8 2541/* filter parameters: [-1 4 2 4 -1] // 8 */
115329f1 2542static void deinterlace_line(uint8_t *dst,
bb270c08
DB		 2543 const uint8_t *lum_m4, const uint8_t *lum_m3,
2544 const uint8_t *lum_m2, const uint8_t *lum_m1,
2545 const uint8_t *lum,
2546 int size)
85c242d8 2547{
5981f4e6 2548#ifndef HAVE_MMX
0c1a9eda 2549 uint8_t *cm = cropTbl + MAX_NEG_CROP;
85c242d8 2550 int sum;
85c242d8
FB		 2551
2552 for(;size > 0;size--) {
5981f4e6
F		 2553 sum = -lum_m4[0];
2554 sum += lum_m3[0] << 2;
2555 sum += lum_m2[0] << 1;
2556 sum += lum_m1[0] << 2;
2557 sum += -lum[0];
85c242d8 2558 dst[0] = cm[(sum + 4) >> 3];
5981f4e6
F		 2559 lum_m4++;
2560 lum_m3++;
2561 lum_m2++;
2562 lum_m1++;
2563 lum++;
85c242d8 2564 dst++;
85c242d8 2565 }
5981f4e6
F		 2566#else
2567
782c5984
MN		 2568 {
2569 mmx_t rounder;
2570 rounder.uw[0]=4;
2571 rounder.uw[1]=4;
2572 rounder.uw[2]=4;
2573 rounder.uw[3]=4;
2574 pxor_r2r(mm7,mm7);
2575 movq_m2r(rounder,mm6);
2576 }
5981f4e6
F		 2577 for (;size > 3; size-=4) {
2578 DEINT_LINE_LUM
2579 lum_m4+=4;
2580 lum_m3+=4;
2581 lum_m2+=4;
2582 lum_m1+=4;
2583 lum+=4;
2584 dst+=4;
2585 }
2586#endif
2587}
0c1a9eda 2588static void deinterlace_line_inplace(uint8_t *lum_m4, uint8_t *lum_m3, uint8_t *lum_m2, uint8_t *lum_m1, uint8_t *lum,
5981f4e6
F		 2589 int size)
2590{
2591#ifndef HAVE_MMX
0c1a9eda 2592 uint8_t *cm = cropTbl + MAX_NEG_CROP;
5981f4e6
F		 2593 int sum;
2594
2595 for(;size > 0;size--) {
2596 sum = -lum_m4[0];
2597 sum += lum_m3[0] << 2;
2598 sum += lum_m2[0] << 1;
2599 lum_m4[0]=lum_m2[0];
2600 sum += lum_m1[0] << 2;
2601 sum += -lum[0];
2602 lum_m2[0] = cm[(sum + 4) >> 3];
2603 lum_m4++;
2604 lum_m3++;
2605 lum_m2++;
2606 lum_m1++;
2607 lum++;
2608 }
2609#else
2610
782c5984
MN		 2611 {
2612 mmx_t rounder;
2613 rounder.uw[0]=4;
2614 rounder.uw[1]=4;
2615 rounder.uw[2]=4;
2616 rounder.uw[3]=4;
2617 pxor_r2r(mm7,mm7);
2618 movq_m2r(rounder,mm6);
2619 }
5981f4e6
F		 2620 for (;size > 3; size-=4) {
2621 DEINT_INPLACE_LINE_LUM
2622 lum_m4+=4;
2623 lum_m3+=4;
2624 lum_m2+=4;
2625 lum_m1+=4;
2626 lum+=4;
2627 }
2628#endif
85c242d8
FB		 2629}
2630
2631/* deinterlacing : 2 temporal taps, 3 spatial taps linear filter. The
2632 top field is copied as is, but the bottom field is deinterlaced
2633 against the top field. */
0c1a9eda 2634static void deinterlace_bottom_field(uint8_t *dst, int dst_wrap,
da64ecc3 2635 const uint8_t *src1, int src_wrap,
5981f4e6 2636 int width, int height)
85c242d8 2637{
da64ecc3 2638 const uint8_t *src_m2, *src_m1, *src_0, *src_p1, *src_p2;
5981f4e6
F		 2639 int y;
2640
2641 src_m2 = src1;
2642 src_m1 = src1;
2643 src_0=&src_m1[src_wrap];
2644 src_p1=&src_0[src_wrap];
2645 src_p2=&src_p1[src_wrap];
2646 for(y=0;y<(height-2);y+=2) {
2647 memcpy(dst,src_m1,width);
85c242d8 2648 dst += dst_wrap;
5981f4e6
F		 2649 deinterlace_line(dst,src_m2,src_m1,src_0,src_p1,src_p2,width);
2650 src_m2 = src_0;
2651 src_m1 = src_p1;
2652 src_0 = src_p2;
2653 src_p1 += 2*src_wrap;
2654 src_p2 += 2*src_wrap;
85c242d8 2655 dst += dst_wrap;
85c242d8 2656 }
5981f4e6
F		 2657 memcpy(dst,src_m1,width);
2658 dst += dst_wrap;
2659 /* do last line */
2660 deinterlace_line(dst,src_m2,src_m1,src_0,src_0,src_0,width);
2661}
2662
0c1a9eda 2663static void deinterlace_bottom_field_inplace(uint8_t *src1, int src_wrap,
bb270c08 2664 int width, int height)
5981f4e6 2665{
0c1a9eda 2666 uint8_t *src_m1, *src_0, *src_p1, *src_p2;
5981f4e6 2667 int y;
0c1a9eda
ZK		 2668 uint8_t *buf;
2669 buf = (uint8_t*)av_malloc(width);
5981f4e6
F		 2670
2671 src_m1 = src1;
2672 memcpy(buf,src_m1,width);
2673 src_0=&src_m1[src_wrap];
2674 src_p1=&src_0[src_wrap];
2675 src_p2=&src_p1[src_wrap];
2676 for(y=0;y<(height-2);y+=2) {
2677 deinterlace_line_inplace(buf,src_m1,src_0,src_p1,src_p2,width);
2678 src_m1 = src_p1;
2679 src_0 = src_p2;
2680 src_p1 += 2*src_wrap;
2681 src_p2 += 2*src_wrap;
2682 }
2683 /* do last line */
2684 deinterlace_line_inplace(buf,src_m1,src_0,src_0,src_0,width);
6000abfa 2685 av_free(buf);
85c242d8
FB		 2686}
2687
2688
5981f4e6 2689/* deinterlace - if not supported return -1 */
da64ecc3 2690int avpicture_deinterlace(AVPicture *dst, const AVPicture *src,
de6d9b64
FB		 2691 int pix_fmt, int width, int height)
2692{
85c242d8
FB		 2693 int i;
2694
2695 if (pix_fmt != PIX_FMT_YUV420P &&
2696 pix_fmt != PIX_FMT_YUV422P &&
47017dd8 2697 pix_fmt != PIX_FMT_YUV444P &&
bb270c08 2698 pix_fmt != PIX_FMT_YUV411P)
85c242d8 2699 return -1;
5981f4e6 2700 if ((width & 3) != 0 || (height & 3) != 0)
85c242d8 2701 return -1;
5981f4e6 2702
85c242d8
FB		 2703 for(i=0;i<3;i++) {
2704 if (i == 1) {
2705 switch(pix_fmt) {
2706 case PIX_FMT_YUV420P:
2707 width >>= 1;
2708 height >>= 1;
2709 break;
2710 case PIX_FMT_YUV422P:
2711 width >>= 1;
2712 break;
47017dd8
RS		 2713 case PIX_FMT_YUV411P:
2714 width >>= 2;
2715 break;
85c242d8
FB		 2716 default:
2717 break;
2718 }
2719 }
5981f4e6 2720 if (src == dst) {
da64ecc3 2721 deinterlace_bottom_field_inplace(dst->data[i], dst->linesize[i],
85c242d8 2722 width, height);
5981f4e6
F		 2723 } else {
2724 deinterlace_bottom_field(dst->data[i],dst->linesize[i],
2725 src->data[i], src->linesize[i],
2726 width, height);
2727 }
de6d9b64 2728 }
5981f4e6
F		 2729#ifdef HAVE_MMX
2730 emms();
2731#endif
85c242d8 2732 return 0;
de6d9b64 2733}
cd4af68a
ZK		 2734
2735#undef FIX
Libav master