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