Remove old scaler.
[libav.git] / libavcodec / imgconvert.c
1 /*
2 * Misc image conversion routines
3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /**
23 * @file libavcodec/imgconvert.c
24 * misc image conversion routines
25 */
26
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 */
32
33 #include "avcodec.h"
34 #include "dsputil.h"
35 #include "colorspace.h"
36
37 #if HAVE_MMX
38 #include "x86/mmx.h"
39 #include "x86/dsputil_mmx.h"
40 #endif
41
42 #define xglue(x, y) x ## y
43 #define glue(x, y) xglue(x, y)
44
45 #define FF_COLOR_RGB 0 /**< RGB color space */
46 #define FF_COLOR_GRAY 1 /**< gray color space */
47 #define FF_COLOR_YUV 2 /**< YUV color space. 16 <= Y <= 235, 16 <= U, V <= 240 */
48 #define FF_COLOR_YUV_JPEG 3 /**< YUV color space. 0 <= Y <= 255, 0 <= U, V <= 255 */
49
50 #define FF_PIXEL_PLANAR 0 /**< each channel has one component in AVPicture */
51 #define FF_PIXEL_PACKED 1 /**< only one components containing all the channels */
52 #define FF_PIXEL_PALETTE 2 /**< one components containing indexes for a palette */
53
54 typedef struct PixFmtInfo {
55 const char *name;
56 uint8_t nb_channels; /**< number of channels (including alpha) */
57 uint8_t color_type; /**< color type (see FF_COLOR_xxx constants) */
58 uint8_t pixel_type; /**< pixel storage type (see FF_PIXEL_xxx constants) */
59 uint8_t is_alpha : 1; /**< true if alpha can be specified */
60 uint8_t is_hwaccel : 1; /**< true if this is an HW accelerated format */
61 uint8_t x_chroma_shift; /**< X chroma subsampling factor is 2 ^ shift */
62 uint8_t y_chroma_shift; /**< Y chroma subsampling factor is 2 ^ shift */
63 uint8_t depth; /**< bit depth of the color components */
64 } PixFmtInfo;
65
66 /* this table gives more information about formats */
67 static const PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {
68 /* YUV formats */
69 [PIX_FMT_YUV420P] = {
70 .name = "yuv420p",
71 .nb_channels = 3,
72 .color_type = FF_COLOR_YUV,
73 .pixel_type = FF_PIXEL_PLANAR,
74 .depth = 8,
75 .x_chroma_shift = 1, .y_chroma_shift = 1,
76 },
77 [PIX_FMT_YUV422P] = {
78 .name = "yuv422p",
79 .nb_channels = 3,
80 .color_type = FF_COLOR_YUV,
81 .pixel_type = FF_PIXEL_PLANAR,
82 .depth = 8,
83 .x_chroma_shift = 1, .y_chroma_shift = 0,
84 },
85 [PIX_FMT_YUV444P] = {
86 .name = "yuv444p",
87 .nb_channels = 3,
88 .color_type = FF_COLOR_YUV,
89 .pixel_type = FF_PIXEL_PLANAR,
90 .depth = 8,
91 .x_chroma_shift = 0, .y_chroma_shift = 0,
92 },
93 [PIX_FMT_YUYV422] = {
94 .name = "yuyv422",
95 .nb_channels = 1,
96 .color_type = FF_COLOR_YUV,
97 .pixel_type = FF_PIXEL_PACKED,
98 .depth = 8,
99 .x_chroma_shift = 1, .y_chroma_shift = 0,
100 },
101 [PIX_FMT_UYVY422] = {
102 .name = "uyvy422",
103 .nb_channels = 1,
104 .color_type = FF_COLOR_YUV,
105 .pixel_type = FF_PIXEL_PACKED,
106 .depth = 8,
107 .x_chroma_shift = 1, .y_chroma_shift = 0,
108 },
109 [PIX_FMT_YUV410P] = {
110 .name = "yuv410p",
111 .nb_channels = 3,
112 .color_type = FF_COLOR_YUV,
113 .pixel_type = FF_PIXEL_PLANAR,
114 .depth = 8,
115 .x_chroma_shift = 2, .y_chroma_shift = 2,
116 },
117 [PIX_FMT_YUV411P] = {
118 .name = "yuv411p",
119 .nb_channels = 3,
120 .color_type = FF_COLOR_YUV,
121 .pixel_type = FF_PIXEL_PLANAR,
122 .depth = 8,
123 .x_chroma_shift = 2, .y_chroma_shift = 0,
124 },
125 [PIX_FMT_YUV440P] = {
126 .name = "yuv440p",
127 .nb_channels = 3,
128 .color_type = FF_COLOR_YUV,
129 .pixel_type = FF_PIXEL_PLANAR,
130 .depth = 8,
131 .x_chroma_shift = 0, .y_chroma_shift = 1,
132 },
133
134 /* YUV formats with alpha plane */
135 [PIX_FMT_YUVA420P] = {
136 .name = "yuva420p",
137 .nb_channels = 4,
138 .color_type = FF_COLOR_YUV,
139 .pixel_type = FF_PIXEL_PLANAR,
140 .depth = 8,
141 .x_chroma_shift = 1, .y_chroma_shift = 1,
142 },
143
144 /* JPEG YUV */
145 [PIX_FMT_YUVJ420P] = {
146 .name = "yuvj420p",
147 .nb_channels = 3,
148 .color_type = FF_COLOR_YUV_JPEG,
149 .pixel_type = FF_PIXEL_PLANAR,
150 .depth = 8,
151 .x_chroma_shift = 1, .y_chroma_shift = 1,
152 },
153 [PIX_FMT_YUVJ422P] = {
154 .name = "yuvj422p",
155 .nb_channels = 3,
156 .color_type = FF_COLOR_YUV_JPEG,
157 .pixel_type = FF_PIXEL_PLANAR,
158 .depth = 8,
159 .x_chroma_shift = 1, .y_chroma_shift = 0,
160 },
161 [PIX_FMT_YUVJ444P] = {
162 .name = "yuvj444p",
163 .nb_channels = 3,
164 .color_type = FF_COLOR_YUV_JPEG,
165 .pixel_type = FF_PIXEL_PLANAR,
166 .depth = 8,
167 .x_chroma_shift = 0, .y_chroma_shift = 0,
168 },
169 [PIX_FMT_YUVJ440P] = {
170 .name = "yuvj440p",
171 .nb_channels = 3,
172 .color_type = FF_COLOR_YUV_JPEG,
173 .pixel_type = FF_PIXEL_PLANAR,
174 .depth = 8,
175 .x_chroma_shift = 0, .y_chroma_shift = 1,
176 },
177
178 /* RGB formats */
179 [PIX_FMT_RGB24] = {
180 .name = "rgb24",
181 .nb_channels = 3,
182 .color_type = FF_COLOR_RGB,
183 .pixel_type = FF_PIXEL_PACKED,
184 .depth = 8,
185 .x_chroma_shift = 0, .y_chroma_shift = 0,
186 },
187 [PIX_FMT_BGR24] = {
188 .name = "bgr24",
189 .nb_channels = 3,
190 .color_type = FF_COLOR_RGB,
191 .pixel_type = FF_PIXEL_PACKED,
192 .depth = 8,
193 .x_chroma_shift = 0, .y_chroma_shift = 0,
194 },
195 [PIX_FMT_RGB32] = {
196 .name = "rgb32",
197 .nb_channels = 4, .is_alpha = 1,
198 .color_type = FF_COLOR_RGB,
199 .pixel_type = FF_PIXEL_PACKED,
200 .depth = 8,
201 .x_chroma_shift = 0, .y_chroma_shift = 0,
202 },
203 [PIX_FMT_RGB48BE] = {
204 .name = "rgb48be",
205 .nb_channels = 3,
206 .color_type = FF_COLOR_RGB,
207 .pixel_type = FF_PIXEL_PACKED,
208 .depth = 16,
209 .x_chroma_shift = 0, .y_chroma_shift = 0,
210 },
211 [PIX_FMT_RGB48LE] = {
212 .name = "rgb48le",
213 .nb_channels = 3,
214 .color_type = FF_COLOR_RGB,
215 .pixel_type = FF_PIXEL_PACKED,
216 .depth = 16,
217 .x_chroma_shift = 0, .y_chroma_shift = 0,
218 },
219 [PIX_FMT_RGB565] = {
220 .name = "rgb565",
221 .nb_channels = 3,
222 .color_type = FF_COLOR_RGB,
223 .pixel_type = FF_PIXEL_PACKED,
224 .depth = 5,
225 .x_chroma_shift = 0, .y_chroma_shift = 0,
226 },
227 [PIX_FMT_RGB555] = {
228 .name = "rgb555",
229 .nb_channels = 3,
230 .color_type = FF_COLOR_RGB,
231 .pixel_type = FF_PIXEL_PACKED,
232 .depth = 5,
233 .x_chroma_shift = 0, .y_chroma_shift = 0,
234 },
235
236 /* gray / mono formats */
237 [PIX_FMT_GRAY16BE] = {
238 .name = "gray16be",
239 .nb_channels = 1,
240 .color_type = FF_COLOR_GRAY,
241 .pixel_type = FF_PIXEL_PLANAR,
242 .depth = 16,
243 },
244 [PIX_FMT_GRAY16LE] = {
245 .name = "gray16le",
246 .nb_channels = 1,
247 .color_type = FF_COLOR_GRAY,
248 .pixel_type = FF_PIXEL_PLANAR,
249 .depth = 16,
250 },
251 [PIX_FMT_GRAY8] = {
252 .name = "gray",
253 .nb_channels = 1,
254 .color_type = FF_COLOR_GRAY,
255 .pixel_type = FF_PIXEL_PLANAR,
256 .depth = 8,
257 },
258 [PIX_FMT_MONOWHITE] = {
259 .name = "monow",
260 .nb_channels = 1,
261 .color_type = FF_COLOR_GRAY,
262 .pixel_type = FF_PIXEL_PLANAR,
263 .depth = 1,
264 },
265 [PIX_FMT_MONOBLACK] = {
266 .name = "monob",
267 .nb_channels = 1,
268 .color_type = FF_COLOR_GRAY,
269 .pixel_type = FF_PIXEL_PLANAR,
270 .depth = 1,
271 },
272
273 /* paletted formats */
274 [PIX_FMT_PAL8] = {
275 .name = "pal8",
276 .nb_channels = 4, .is_alpha = 1,
277 .color_type = FF_COLOR_RGB,
278 .pixel_type = FF_PIXEL_PALETTE,
279 .depth = 8,
280 },
281 [PIX_FMT_XVMC_MPEG2_MC] = {
282 .name = "xvmcmc",
283 .is_hwaccel = 1,
284 },
285 [PIX_FMT_XVMC_MPEG2_IDCT] = {
286 .name = "xvmcidct",
287 .is_hwaccel = 1,
288 },
289 [PIX_FMT_VDPAU_MPEG1] = {
290 .name = "vdpau_mpeg1",
291 .is_hwaccel = 1,
292 .x_chroma_shift = 1, .y_chroma_shift = 1,
293 },
294 [PIX_FMT_VDPAU_MPEG2] = {
295 .name = "vdpau_mpeg2",
296 .is_hwaccel = 1,
297 .x_chroma_shift = 1, .y_chroma_shift = 1,
298 },
299 [PIX_FMT_VDPAU_H264] = {
300 .name = "vdpau_h264",
301 .is_hwaccel = 1,
302 .x_chroma_shift = 1, .y_chroma_shift = 1,
303 },
304 [PIX_FMT_VDPAU_WMV3] = {
305 .name = "vdpau_wmv3",
306 .is_hwaccel = 1,
307 .x_chroma_shift = 1, .y_chroma_shift = 1,
308 },
309 [PIX_FMT_VDPAU_VC1] = {
310 .name = "vdpau_vc1",
311 .is_hwaccel = 1,
312 .x_chroma_shift = 1, .y_chroma_shift = 1,
313 },
314 [PIX_FMT_UYYVYY411] = {
315 .name = "uyyvyy411",
316 .nb_channels = 1,
317 .color_type = FF_COLOR_YUV,
318 .pixel_type = FF_PIXEL_PACKED,
319 .depth = 8,
320 .x_chroma_shift = 2, .y_chroma_shift = 0,
321 },
322 [PIX_FMT_BGR32] = {
323 .name = "bgr32",
324 .nb_channels = 4, .is_alpha = 1,
325 .color_type = FF_COLOR_RGB,
326 .pixel_type = FF_PIXEL_PACKED,
327 .depth = 8,
328 .x_chroma_shift = 0, .y_chroma_shift = 0,
329 },
330 [PIX_FMT_BGR565] = {
331 .name = "bgr565",
332 .nb_channels = 3,
333 .color_type = FF_COLOR_RGB,
334 .pixel_type = FF_PIXEL_PACKED,
335 .depth = 5,
336 .x_chroma_shift = 0, .y_chroma_shift = 0,
337 },
338 [PIX_FMT_BGR555] = {
339 .name = "bgr555",
340 .nb_channels = 3,
341 .color_type = FF_COLOR_RGB,
342 .pixel_type = FF_PIXEL_PACKED,
343 .depth = 5,
344 .x_chroma_shift = 0, .y_chroma_shift = 0,
345 },
346 [PIX_FMT_RGB8] = {
347 .name = "rgb8",
348 .nb_channels = 1,
349 .color_type = FF_COLOR_RGB,
350 .pixel_type = FF_PIXEL_PACKED,
351 .depth = 8,
352 .x_chroma_shift = 0, .y_chroma_shift = 0,
353 },
354 [PIX_FMT_RGB4] = {
355 .name = "rgb4",
356 .nb_channels = 1,
357 .color_type = FF_COLOR_RGB,
358 .pixel_type = FF_PIXEL_PACKED,
359 .depth = 4,
360 .x_chroma_shift = 0, .y_chroma_shift = 0,
361 },
362 [PIX_FMT_RGB4_BYTE] = {
363 .name = "rgb4_byte",
364 .nb_channels = 1,
365 .color_type = FF_COLOR_RGB,
366 .pixel_type = FF_PIXEL_PACKED,
367 .depth = 8,
368 .x_chroma_shift = 0, .y_chroma_shift = 0,
369 },
370 [PIX_FMT_BGR8] = {
371 .name = "bgr8",
372 .nb_channels = 1,
373 .color_type = FF_COLOR_RGB,
374 .pixel_type = FF_PIXEL_PACKED,
375 .depth = 8,
376 .x_chroma_shift = 0, .y_chroma_shift = 0,
377 },
378 [PIX_FMT_BGR4] = {
379 .name = "bgr4",
380 .nb_channels = 1,
381 .color_type = FF_COLOR_RGB,
382 .pixel_type = FF_PIXEL_PACKED,
383 .depth = 4,
384 .x_chroma_shift = 0, .y_chroma_shift = 0,
385 },
386 [PIX_FMT_BGR4_BYTE] = {
387 .name = "bgr4_byte",
388 .nb_channels = 1,
389 .color_type = FF_COLOR_RGB,
390 .pixel_type = FF_PIXEL_PACKED,
391 .depth = 8,
392 .x_chroma_shift = 0, .y_chroma_shift = 0,
393 },
394 [PIX_FMT_NV12] = {
395 .name = "nv12",
396 .nb_channels = 2,
397 .color_type = FF_COLOR_YUV,
398 .pixel_type = FF_PIXEL_PLANAR,
399 .depth = 8,
400 .x_chroma_shift = 1, .y_chroma_shift = 1,
401 },
402 [PIX_FMT_NV21] = {
403 .name = "nv12",
404 .nb_channels = 2,
405 .color_type = FF_COLOR_YUV,
406 .pixel_type = FF_PIXEL_PLANAR,
407 .depth = 8,
408 .x_chroma_shift = 1, .y_chroma_shift = 1,
409 },
410
411 [PIX_FMT_BGR32_1] = {
412 .name = "bgr32_1",
413 .nb_channels = 4, .is_alpha = 1,
414 .color_type = FF_COLOR_RGB,
415 .pixel_type = FF_PIXEL_PACKED,
416 .depth = 8,
417 .x_chroma_shift = 0, .y_chroma_shift = 0,
418 },
419 [PIX_FMT_RGB32_1] = {
420 .name = "rgb32_1",
421 .nb_channels = 4, .is_alpha = 1,
422 .color_type = FF_COLOR_RGB,
423 .pixel_type = FF_PIXEL_PACKED,
424 .depth = 8,
425 .x_chroma_shift = 0, .y_chroma_shift = 0,
426 },
427
428 /* VA API formats */
429 [PIX_FMT_VAAPI_MOCO] = {
430 .name = "vaapi_moco",
431 .is_hwaccel = 1,
432 .x_chroma_shift = 1, .y_chroma_shift = 1,
433 },
434 [PIX_FMT_VAAPI_IDCT] = {
435 .name = "vaapi_idct",
436 .is_hwaccel = 1,
437 .x_chroma_shift = 1, .y_chroma_shift = 1,
438 },
439 [PIX_FMT_VAAPI_VLD] = {
440 .name = "vaapi_vld",
441 .is_hwaccel = 1,
442 .x_chroma_shift = 1, .y_chroma_shift = 1,
443 },
444 };
445
446 void avcodec_get_chroma_sub_sample(int pix_fmt, int *h_shift, int *v_shift)
447 {
448 *h_shift = pix_fmt_info[pix_fmt].x_chroma_shift;
449 *v_shift = pix_fmt_info[pix_fmt].y_chroma_shift;
450 }
451
452 const char *avcodec_get_pix_fmt_name(int pix_fmt)
453 {
454 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB)
455 return NULL;
456 else
457 return pix_fmt_info[pix_fmt].name;
458 }
459
460 enum PixelFormat avcodec_get_pix_fmt(const char* name)
461 {
462 int i;
463
464 for (i=0; i < PIX_FMT_NB; i++)
465 if (!strcmp(pix_fmt_info[i].name, name))
466 return i;
467 return PIX_FMT_NONE;
468 }
469
470 void avcodec_pix_fmt_string (char *buf, int buf_size, int pix_fmt)
471 {
472 /* print header */
473 if (pix_fmt < 0)
474 snprintf (buf, buf_size,
475 "name " " nb_channels" " depth" " is_alpha"
476 );
477 else{
478 PixFmtInfo info= pix_fmt_info[pix_fmt];
479
480 char is_alpha_char= info.is_alpha ? 'y' : 'n';
481
482 snprintf (buf, buf_size,
483 "%-10s" " %1d " " %2d " " %c ",
484 info.name,
485 info.nb_channels,
486 info.depth,
487 is_alpha_char
488 );
489 }
490 }
491
492 int ff_is_hwaccel_pix_fmt(enum PixelFormat pix_fmt)
493 {
494 return pix_fmt_info[pix_fmt].is_hwaccel;
495 }
496
497 int ff_set_systematic_pal(uint32_t pal[256], enum PixelFormat pix_fmt){
498 int i;
499
500 for(i=0; i<256; i++){
501 int r,g,b;
502
503 switch(pix_fmt) {
504 case PIX_FMT_RGB8:
505 r= (i>>5 )*36;
506 g= ((i>>2)&7)*36;
507 b= (i&3 )*85;
508 break;
509 case PIX_FMT_BGR8:
510 b= (i>>6 )*85;
511 g= ((i>>3)&7)*36;
512 r= (i&7 )*36;
513 break;
514 case PIX_FMT_RGB4_BYTE:
515 r= (i>>3 )*255;
516 g= ((i>>1)&3)*85;
517 b= (i&1 )*255;
518 break;
519 case PIX_FMT_BGR4_BYTE:
520 b= (i>>3 )*255;
521 g= ((i>>1)&3)*85;
522 r= (i&1 )*255;
523 break;
524 case PIX_FMT_GRAY8:
525 r=b=g= i;
526 break;
527 default:
528 return -1;
529 }
530 pal[i] = b + (g<<8) + (r<<16);
531 }
532
533 return 0;
534 }
535
536 int ff_fill_linesize(AVPicture *picture, int pix_fmt, int width)
537 {
538 int w2;
539 const PixFmtInfo *pinfo;
540
541 memset(picture->linesize, 0, sizeof(picture->linesize));
542
543 pinfo = &pix_fmt_info[pix_fmt];
544 switch(pix_fmt) {
545 case PIX_FMT_YUV420P:
546 case PIX_FMT_YUV422P:
547 case PIX_FMT_YUV444P:
548 case PIX_FMT_YUV410P:
549 case PIX_FMT_YUV411P:
550 case PIX_FMT_YUV440P:
551 case PIX_FMT_YUVJ420P:
552 case PIX_FMT_YUVJ422P:
553 case PIX_FMT_YUVJ444P:
554 case PIX_FMT_YUVJ440P:
555 w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;
556 picture->linesize[0] = width;
557 picture->linesize[1] = w2;
558 picture->linesize[2] = w2;
559 break;
560 case PIX_FMT_YUVA420P:
561 w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;
562 picture->linesize[0] = width;
563 picture->linesize[1] = w2;
564 picture->linesize[2] = w2;
565 picture->linesize[3] = width;
566 break;
567 case PIX_FMT_NV12:
568 case PIX_FMT_NV21:
569 w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;
570 picture->linesize[0] = width;
571 picture->linesize[1] = w2;
572 break;
573 case PIX_FMT_RGB24:
574 case PIX_FMT_BGR24:
575 picture->linesize[0] = width * 3;
576 break;
577 case PIX_FMT_RGB32:
578 case PIX_FMT_BGR32:
579 case PIX_FMT_RGB32_1:
580 case PIX_FMT_BGR32_1:
581 picture->linesize[0] = width * 4;
582 break;
583 case PIX_FMT_RGB48BE:
584 case PIX_FMT_RGB48LE:
585 picture->linesize[0] = width * 6;
586 break;
587 case PIX_FMT_GRAY16BE:
588 case PIX_FMT_GRAY16LE:
589 case PIX_FMT_BGR555:
590 case PIX_FMT_BGR565:
591 case PIX_FMT_RGB555:
592 case PIX_FMT_RGB565:
593 case PIX_FMT_YUYV422:
594 picture->linesize[0] = width * 2;
595 break;
596 case PIX_FMT_UYVY422:
597 picture->linesize[0] = width * 2;
598 break;
599 case PIX_FMT_UYYVYY411:
600 picture->linesize[0] = width + width/2;
601 break;
602 case PIX_FMT_RGB4:
603 case PIX_FMT_BGR4:
604 picture->linesize[0] = width / 2;
605 break;
606 case PIX_FMT_MONOWHITE:
607 case PIX_FMT_MONOBLACK:
608 picture->linesize[0] = (width + 7) >> 3;
609 break;
610 case PIX_FMT_PAL8:
611 case PIX_FMT_RGB8:
612 case PIX_FMT_BGR8:
613 case PIX_FMT_RGB4_BYTE:
614 case PIX_FMT_BGR4_BYTE:
615 case PIX_FMT_GRAY8:
616 picture->linesize[0] = width;
617 picture->linesize[1] = 4;
618 break;
619 default:
620 return -1;
621 }
622 return 0;
623 }
624
625 int ff_fill_pointer(AVPicture *picture, uint8_t *ptr, int pix_fmt,
626 int height)
627 {
628 int size, h2, size2;
629 const PixFmtInfo *pinfo;
630
631 pinfo = &pix_fmt_info[pix_fmt];
632 size = picture->linesize[0] * height;
633 switch(pix_fmt) {
634 case PIX_FMT_YUV420P:
635 case PIX_FMT_YUV422P:
636 case PIX_FMT_YUV444P:
637 case PIX_FMT_YUV410P:
638 case PIX_FMT_YUV411P:
639 case PIX_FMT_YUV440P:
640 case PIX_FMT_YUVJ420P:
641 case PIX_FMT_YUVJ422P:
642 case PIX_FMT_YUVJ444P:
643 case PIX_FMT_YUVJ440P:
644 h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;
645 size2 = picture->linesize[1] * h2;
646 picture->data[0] = ptr;
647 picture->data[1] = picture->data[0] + size;
648 picture->data[2] = picture->data[1] + size2;
649 picture->data[3] = NULL;
650 return size + 2 * size2;
651 case PIX_FMT_YUVA420P:
652 h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;
653 size2 = picture->linesize[1] * h2;
654 picture->data[0] = ptr;
655 picture->data[1] = picture->data[0] + size;
656 picture->data[2] = picture->data[1] + size2;
657 picture->data[3] = picture->data[1] + size2 + size2;
658 return 2 * size + 2 * size2;
659 case PIX_FMT_NV12:
660 case PIX_FMT_NV21:
661 h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;
662 size2 = picture->linesize[1] * h2 * 2;
663 picture->data[0] = ptr;
664 picture->data[1] = picture->data[0] + size;
665 picture->data[2] = NULL;
666 picture->data[3] = NULL;
667 return size + 2 * size2;
668 case PIX_FMT_RGB24:
669 case PIX_FMT_BGR24:
670 case PIX_FMT_RGB32:
671 case PIX_FMT_BGR32:
672 case PIX_FMT_RGB32_1:
673 case PIX_FMT_BGR32_1:
674 case PIX_FMT_RGB48BE:
675 case PIX_FMT_RGB48LE:
676 case PIX_FMT_GRAY16BE:
677 case PIX_FMT_GRAY16LE:
678 case PIX_FMT_BGR555:
679 case PIX_FMT_BGR565:
680 case PIX_FMT_RGB555:
681 case PIX_FMT_RGB565:
682 case PIX_FMT_YUYV422:
683 case PIX_FMT_UYVY422:
684 case PIX_FMT_UYYVYY411:
685 case PIX_FMT_RGB4:
686 case PIX_FMT_BGR4:
687 case PIX_FMT_MONOWHITE:
688 case PIX_FMT_MONOBLACK:
689 picture->data[0] = ptr;
690 picture->data[1] = NULL;
691 picture->data[2] = NULL;
692 picture->data[3] = NULL;
693 return size;
694 case PIX_FMT_PAL8:
695 case PIX_FMT_RGB8:
696 case PIX_FMT_BGR8:
697 case PIX_FMT_RGB4_BYTE:
698 case PIX_FMT_BGR4_BYTE:
699 case PIX_FMT_GRAY8:
700 size2 = (size + 3) & ~3;
701 picture->data[0] = ptr;
702 picture->data[1] = ptr + size2; /* palette is stored here as 256 32 bit words */
703 picture->data[2] = NULL;
704 picture->data[3] = NULL;
705 return size2 + 256 * 4;
706 default:
707 picture->data[0] = NULL;
708 picture->data[1] = NULL;
709 picture->data[2] = NULL;
710 picture->data[3] = NULL;
711 return -1;
712 }
713 }
714
715 int avpicture_fill(AVPicture *picture, uint8_t *ptr,
716 int pix_fmt, int width, int height)
717 {
718
719 if(avcodec_check_dimensions(NULL, width, height))
720 return -1;
721
722 if (ff_fill_linesize(picture, pix_fmt, width))
723 return -1;
724
725 return ff_fill_pointer(picture, ptr, pix_fmt, height);
726 }
727
728 int avpicture_layout(const AVPicture* src, int pix_fmt, int width, int height,
729 unsigned char *dest, int dest_size)
730 {
731 const PixFmtInfo* pf = &pix_fmt_info[pix_fmt];
732 int i, j, w, ow, h, oh, data_planes;
733 const unsigned char* s;
734 int size = avpicture_get_size(pix_fmt, width, height);
735
736 if (size > dest_size || size < 0)
737 return -1;
738
739 if (pf->pixel_type == FF_PIXEL_PACKED || pf->pixel_type == FF_PIXEL_PALETTE) {
740 if (pix_fmt == PIX_FMT_YUYV422 ||
741 pix_fmt == PIX_FMT_UYVY422 ||
742 pix_fmt == PIX_FMT_BGR565 ||
743 pix_fmt == PIX_FMT_BGR555 ||
744 pix_fmt == PIX_FMT_RGB565 ||
745 pix_fmt == PIX_FMT_RGB555)
746 w = width * 2;
747 else if (pix_fmt == PIX_FMT_UYYVYY411)
748 w = width + width/2;
749 else if (pix_fmt == PIX_FMT_PAL8)
750 w = width;
751 else
752 w = width * (pf->depth * pf->nb_channels / 8);
753
754 data_planes = 1;
755 h = height;
756 } else {
757 data_planes = pf->nb_channels;
758 w = (width*pf->depth + 7)/8;
759 h = height;
760 }
761
762 ow = w;
763 oh = h;
764
765 for (i=0; i<data_planes; i++) {
766 if (i == 1) {
767 w = width >> pf->x_chroma_shift;
768 h = height >> pf->y_chroma_shift;
769 } else if (i == 3) {
770 w = ow;
771 h = oh;
772 }
773 s = src->data[i];
774 for(j=0; j<h; j++) {
775 memcpy(dest, s, w);
776 dest += w;
777 s += src->linesize[i];
778 }
779 }
780
781 if (pf->pixel_type == FF_PIXEL_PALETTE)
782 memcpy((unsigned char *)(((size_t)dest + 3) & ~3), src->data[1], 256 * 4);
783
784 return size;
785 }
786
787 int avpicture_get_size(int pix_fmt, int width, int height)
788 {
789 AVPicture dummy_pict;
790 return avpicture_fill(&dummy_pict, NULL, pix_fmt, width, height);
791 }
792
793 int avcodec_get_pix_fmt_loss(int dst_pix_fmt, int src_pix_fmt,
794 int has_alpha)
795 {
796 const PixFmtInfo *pf, *ps;
797 int loss;
798
799 ps = &pix_fmt_info[src_pix_fmt];
800 pf = &pix_fmt_info[dst_pix_fmt];
801
802 /* compute loss */
803 loss = 0;
804 pf = &pix_fmt_info[dst_pix_fmt];
805 if (pf->depth < ps->depth ||
806 (dst_pix_fmt == PIX_FMT_RGB555 && src_pix_fmt == PIX_FMT_RGB565))
807 loss |= FF_LOSS_DEPTH;
808 if (pf->x_chroma_shift > ps->x_chroma_shift ||
809 pf->y_chroma_shift > ps->y_chroma_shift)
810 loss |= FF_LOSS_RESOLUTION;
811 switch(pf->color_type) {
812 case FF_COLOR_RGB:
813 if (ps->color_type != FF_COLOR_RGB &&
814 ps->color_type != FF_COLOR_GRAY)
815 loss |= FF_LOSS_COLORSPACE;
816 break;
817 case FF_COLOR_GRAY:
818 if (ps->color_type != FF_COLOR_GRAY)
819 loss |= FF_LOSS_COLORSPACE;
820 break;
821 case FF_COLOR_YUV:
822 if (ps->color_type != FF_COLOR_YUV)
823 loss |= FF_LOSS_COLORSPACE;
824 break;
825 case FF_COLOR_YUV_JPEG:
826 if (ps->color_type != FF_COLOR_YUV_JPEG &&
827 ps->color_type != FF_COLOR_YUV &&
828 ps->color_type != FF_COLOR_GRAY)
829 loss |= FF_LOSS_COLORSPACE;
830 break;
831 default:
832 /* fail safe test */
833 if (ps->color_type != pf->color_type)
834 loss |= FF_LOSS_COLORSPACE;
835 break;
836 }
837 if (pf->color_type == FF_COLOR_GRAY &&
838 ps->color_type != FF_COLOR_GRAY)
839 loss |= FF_LOSS_CHROMA;
840 if (!pf->is_alpha && (ps->is_alpha && has_alpha))
841 loss |= FF_LOSS_ALPHA;
842 if (pf->pixel_type == FF_PIXEL_PALETTE &&
843 (ps->pixel_type != FF_PIXEL_PALETTE && ps->color_type != FF_COLOR_GRAY))
844 loss |= FF_LOSS_COLORQUANT;
845 return loss;
846 }
847
848 static int avg_bits_per_pixel(int pix_fmt)
849 {
850 int bits;
851 const PixFmtInfo *pf;
852
853 pf = &pix_fmt_info[pix_fmt];
854 switch(pf->pixel_type) {
855 case FF_PIXEL_PACKED:
856 switch(pix_fmt) {
857 case PIX_FMT_YUYV422:
858 case PIX_FMT_UYVY422:
859 case PIX_FMT_RGB565:
860 case PIX_FMT_RGB555:
861 case PIX_FMT_BGR565:
862 case PIX_FMT_BGR555:
863 bits = 16;
864 break;
865 case PIX_FMT_UYYVYY411:
866 bits = 12;
867 break;
868 default:
869 bits = pf->depth * pf->nb_channels;
870 break;
871 }
872 break;
873 case FF_PIXEL_PLANAR:
874 if (pf->x_chroma_shift == 0 && pf->y_chroma_shift == 0) {
875 bits = pf->depth * pf->nb_channels;
876 } else {
877 bits = pf->depth + ((2 * pf->depth) >>
878 (pf->x_chroma_shift + pf->y_chroma_shift));
879 }
880 break;
881 case FF_PIXEL_PALETTE:
882 bits = 8;
883 break;
884 default:
885 bits = -1;
886 break;
887 }
888 return bits;
889 }
890
891 static int avcodec_find_best_pix_fmt1(int64_t pix_fmt_mask,
892 int src_pix_fmt,
893 int has_alpha,
894 int loss_mask)
895 {
896 int dist, i, loss, min_dist, dst_pix_fmt;
897
898 /* find exact color match with smallest size */
899 dst_pix_fmt = -1;
900 min_dist = 0x7fffffff;
901 for(i = 0;i < PIX_FMT_NB; i++) {
902 if (pix_fmt_mask & (1ULL << i)) {
903 loss = avcodec_get_pix_fmt_loss(i, src_pix_fmt, has_alpha) & loss_mask;
904 if (loss == 0) {
905 dist = avg_bits_per_pixel(i);
906 if (dist < min_dist) {
907 min_dist = dist;
908 dst_pix_fmt = i;
909 }
910 }
911 }
912 }
913 return dst_pix_fmt;
914 }
915
916 int avcodec_find_best_pix_fmt(int64_t pix_fmt_mask, int src_pix_fmt,
917 int has_alpha, int *loss_ptr)
918 {
919 int dst_pix_fmt, loss_mask, i;
920 static const int loss_mask_order[] = {
921 ~0, /* no loss first */
922 ~FF_LOSS_ALPHA,
923 ~FF_LOSS_RESOLUTION,
924 ~(FF_LOSS_COLORSPACE | FF_LOSS_RESOLUTION),
925 ~FF_LOSS_COLORQUANT,
926 ~FF_LOSS_DEPTH,
927 0,
928 };
929
930 /* try with successive loss */
931 i = 0;
932 for(;;) {
933 loss_mask = loss_mask_order[i++];
934 dst_pix_fmt = avcodec_find_best_pix_fmt1(pix_fmt_mask, src_pix_fmt,
935 has_alpha, loss_mask);
936 if (dst_pix_fmt >= 0)
937 goto found;
938 if (loss_mask == 0)
939 break;
940 }
941 return -1;
942 found:
943 if (loss_ptr)
944 *loss_ptr = avcodec_get_pix_fmt_loss(dst_pix_fmt, src_pix_fmt, has_alpha);
945 return dst_pix_fmt;
946 }
947
948 void ff_img_copy_plane(uint8_t *dst, int dst_wrap,
949 const uint8_t *src, int src_wrap,
950 int width, int height)
951 {
952 if((!dst) || (!src))
953 return;
954 for(;height > 0; height--) {
955 memcpy(dst, src, width);
956 dst += dst_wrap;
957 src += src_wrap;
958 }
959 }
960
961 int ff_get_plane_bytewidth(enum PixelFormat pix_fmt, int width, int plane)
962 {
963 int bits;
964 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
965
966 pf = &pix_fmt_info[pix_fmt];
967 switch(pf->pixel_type) {
968 case FF_PIXEL_PACKED:
969 switch(pix_fmt) {
970 case PIX_FMT_YUYV422:
971 case PIX_FMT_UYVY422:
972 case PIX_FMT_RGB565:
973 case PIX_FMT_RGB555:
974 case PIX_FMT_BGR565:
975 case PIX_FMT_BGR555:
976 bits = 16;
977 break;
978 case PIX_FMT_UYYVYY411:
979 bits = 12;
980 break;
981 default:
982 bits = pf->depth * pf->nb_channels;
983 break;
984 }
985 return (width * bits + 7) >> 3;
986 break;
987 case FF_PIXEL_PLANAR:
988 if (plane == 1 || plane == 2)
989 width= -((-width)>>pf->x_chroma_shift);
990
991 return (width * pf->depth + 7) >> 3;
992 break;
993 case FF_PIXEL_PALETTE:
994 if (plane == 0)
995 return width;
996 break;
997 }
998
999 return -1;
1000 }
1001
1002 void av_picture_copy(AVPicture *dst, const AVPicture *src,
1003 int pix_fmt, int width, int height)
1004 {
1005 int i;
1006 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
1007
1008 pf = &pix_fmt_info[pix_fmt];
1009 switch(pf->pixel_type) {
1010 case FF_PIXEL_PACKED:
1011 case FF_PIXEL_PLANAR:
1012 for(i = 0; i < pf->nb_channels; i++) {
1013 int h;
1014 int bwidth = ff_get_plane_bytewidth(pix_fmt, width, i);
1015 h = height;
1016 if (i == 1 || i == 2) {
1017 h= -((-height)>>pf->y_chroma_shift);
1018 }
1019 ff_img_copy_plane(dst->data[i], dst->linesize[i],
1020 src->data[i], src->linesize[i],
1021 bwidth, h);
1022 }
1023 break;
1024 case FF_PIXEL_PALETTE:
1025 ff_img_copy_plane(dst->data[0], dst->linesize[0],
1026 src->data[0], src->linesize[0],
1027 width, height);
1028 /* copy the palette */
1029 ff_img_copy_plane(dst->data[1], dst->linesize[1],
1030 src->data[1], src->linesize[1],
1031 4, 256);
1032 break;
1033 }
1034 }
1035
1036 /* 2x2 -> 1x1 */
1037 void ff_shrink22(uint8_t *dst, int dst_wrap,
1038 const uint8_t *src, int src_wrap,
1039 int width, int height)
1040 {
1041 int w;
1042 const uint8_t *s1, *s2;
1043 uint8_t *d;
1044
1045 for(;height > 0; height--) {
1046 s1 = src;
1047 s2 = s1 + src_wrap;
1048 d = dst;
1049 for(w = width;w >= 4; w-=4) {
1050 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
1051 d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
1052 d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
1053 d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
1054 s1 += 8;
1055 s2 += 8;
1056 d += 4;
1057 }
1058 for(;w > 0; w--) {
1059 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
1060 s1 += 2;
1061 s2 += 2;
1062 d++;
1063 }
1064 src += 2 * src_wrap;
1065 dst += dst_wrap;
1066 }
1067 }
1068
1069 /* 4x4 -> 1x1 */
1070 void ff_shrink44(uint8_t *dst, int dst_wrap,
1071 const uint8_t *src, int src_wrap,
1072 int width, int height)
1073 {
1074 int w;
1075 const uint8_t *s1, *s2, *s3, *s4;
1076 uint8_t *d;
1077
1078 for(;height > 0; height--) {
1079 s1 = src;
1080 s2 = s1 + src_wrap;
1081 s3 = s2 + src_wrap;
1082 s4 = s3 + src_wrap;
1083 d = dst;
1084 for(w = width;w > 0; w--) {
1085 d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
1086 s2[0] + s2[1] + s2[2] + s2[3] +
1087 s3[0] + s3[1] + s3[2] + s3[3] +
1088 s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
1089 s1 += 4;
1090 s2 += 4;
1091 s3 += 4;
1092 s4 += 4;
1093 d++;
1094 }
1095 src += 4 * src_wrap;
1096 dst += dst_wrap;
1097 }
1098 }
1099
1100 /* 8x8 -> 1x1 */
1101 void ff_shrink88(uint8_t *dst, int dst_wrap,
1102 const uint8_t *src, int src_wrap,
1103 int width, int height)
1104 {
1105 int w, i;
1106
1107 for(;height > 0; height--) {
1108 for(w = width;w > 0; w--) {
1109 int tmp=0;
1110 for(i=0; i<8; i++){
1111 tmp += src[0] + src[1] + src[2] + src[3] + src[4] + src[5] + src[6] + src[7];
1112 src += src_wrap;
1113 }
1114 *(dst++) = (tmp + 32)>>6;
1115 src += 8 - 8*src_wrap;
1116 }
1117 src += 8*src_wrap - 8*width;
1118 dst += dst_wrap - width;
1119 }
1120 }
1121
1122
1123 int avpicture_alloc(AVPicture *picture,
1124 int pix_fmt, int width, int height)
1125 {
1126 int size;
1127 void *ptr;
1128
1129 size = avpicture_get_size(pix_fmt, width, height);
1130 if(size<0)
1131 goto fail;
1132 ptr = av_malloc(size);
1133 if (!ptr)
1134 goto fail;
1135 avpicture_fill(picture, ptr, pix_fmt, width, height);
1136 if(picture->data[1] && !picture->data[2])
1137 ff_set_systematic_pal((uint32_t*)picture->data[1], pix_fmt);
1138
1139 return 0;
1140 fail:
1141 memset(picture, 0, sizeof(AVPicture));
1142 return -1;
1143 }
1144
1145 void avpicture_free(AVPicture *picture)
1146 {
1147 av_free(picture->data[0]);
1148 }
1149
1150 /* return true if yuv planar */
1151 static inline int is_yuv_planar(const PixFmtInfo *ps)
1152 {
1153 return (ps->color_type == FF_COLOR_YUV ||
1154 ps->color_type == FF_COLOR_YUV_JPEG) &&
1155 ps->pixel_type == FF_PIXEL_PLANAR;
1156 }
1157
1158 int av_picture_crop(AVPicture *dst, const AVPicture *src,
1159 int pix_fmt, int top_band, int left_band)
1160 {
1161 int y_shift;
1162 int x_shift;
1163
1164 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB || !is_yuv_planar(&pix_fmt_info[pix_fmt]))
1165 return -1;
1166
1167 y_shift = pix_fmt_info[pix_fmt].y_chroma_shift;
1168 x_shift = pix_fmt_info[pix_fmt].x_chroma_shift;
1169
1170 dst->data[0] = src->data[0] + (top_band * src->linesize[0]) + left_band;
1171 dst->data[1] = src->data[1] + ((top_band >> y_shift) * src->linesize[1]) + (left_band >> x_shift);
1172 dst->data[2] = src->data[2] + ((top_band >> y_shift) * src->linesize[2]) + (left_band >> x_shift);
1173
1174 dst->linesize[0] = src->linesize[0];
1175 dst->linesize[1] = src->linesize[1];
1176 dst->linesize[2] = src->linesize[2];
1177 return 0;
1178 }
1179
1180 int av_picture_pad(AVPicture *dst, const AVPicture *src, int height, int width,
1181 int pix_fmt, int padtop, int padbottom, int padleft, int padright,
1182 int *color)
1183 {
1184 uint8_t *optr;
1185 int y_shift;
1186 int x_shift;
1187 int yheight;
1188 int i, y;
1189
1190 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB ||
1191 !is_yuv_planar(&pix_fmt_info[pix_fmt])) return -1;
1192
1193 for (i = 0; i < 3; i++) {
1194 x_shift = i ? pix_fmt_info[pix_fmt].x_chroma_shift : 0;
1195 y_shift = i ? pix_fmt_info[pix_fmt].y_chroma_shift : 0;
1196
1197 if (padtop || padleft) {
1198 memset(dst->data[i], color[i],
1199 dst->linesize[i] * (padtop >> y_shift) + (padleft >> x_shift));
1200 }
1201
1202 if (padleft || padright) {
1203 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1204 (dst->linesize[i] - (padright >> x_shift));
1205 yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
1206 for (y = 0; y < yheight; y++) {
1207 memset(optr, color[i], (padleft + padright) >> x_shift);
1208 optr += dst->linesize[i];
1209 }
1210 }
1211
1212 if (src) { /* first line */
1213 uint8_t *iptr = src->data[i];
1214 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1215 (padleft >> x_shift);
1216 memcpy(optr, iptr, (width - padleft - padright) >> x_shift);
1217 iptr += src->linesize[i];
1218 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1219 (dst->linesize[i] - (padright >> x_shift));
1220 yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
1221 for (y = 0; y < yheight; y++) {
1222 memset(optr, color[i], (padleft + padright) >> x_shift);
1223 memcpy(optr + ((padleft + padright) >> x_shift), iptr,
1224 (width - padleft - padright) >> x_shift);
1225 iptr += src->linesize[i];
1226 optr += dst->linesize[i];
1227 }
1228 }
1229
1230 if (padbottom || padright) {
1231 optr = dst->data[i] + dst->linesize[i] *
1232 ((height - padbottom) >> y_shift) - (padright >> x_shift);
1233 memset(optr, color[i],dst->linesize[i] *
1234 (padbottom >> y_shift) + (padright >> x_shift));
1235 }
1236 }
1237 return 0;
1238 }
1239
1240 /* NOTE: we scan all the pixels to have an exact information */
1241 static int get_alpha_info_pal8(const AVPicture *src, int width, int height)
1242 {
1243 const unsigned char *p;
1244 int src_wrap, ret, x, y;
1245 unsigned int a;
1246 uint32_t *palette = (uint32_t *)src->data[1];
1247
1248 p = src->data[0];
1249 src_wrap = src->linesize[0] - width;
1250 ret = 0;
1251 for(y=0;y<height;y++) {
1252 for(x=0;x<width;x++) {
1253 a = palette[p[0]] >> 24;
1254 if (a == 0x00) {
1255 ret |= FF_ALPHA_TRANSP;
1256 } else if (a != 0xff) {
1257 ret |= FF_ALPHA_SEMI_TRANSP;
1258 }
1259 p++;
1260 }
1261 p += src_wrap;
1262 }
1263 return ret;
1264 }
1265
1266 int img_get_alpha_info(const AVPicture *src,
1267 int pix_fmt, int width, int height)
1268 {
1269 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
1270 int ret;
1271
1272 pf = &pix_fmt_info[pix_fmt];
1273 /* no alpha can be represented in format */
1274 if (!pf->is_alpha)
1275 return 0;
1276 switch(pix_fmt) {
1277 case PIX_FMT_PAL8:
1278 ret = get_alpha_info_pal8(src, width, height);
1279 break;
1280 default:
1281 /* we do not know, so everything is indicated */
1282 ret = FF_ALPHA_TRANSP | FF_ALPHA_SEMI_TRANSP;
1283 break;
1284 }
1285 return ret;
1286 }
1287
1288 #if HAVE_MMX
1289 #define DEINT_INPLACE_LINE_LUM \
1290 movd_m2r(lum_m4[0],mm0);\
1291 movd_m2r(lum_m3[0],mm1);\
1292 movd_m2r(lum_m2[0],mm2);\
1293 movd_m2r(lum_m1[0],mm3);\
1294 movd_m2r(lum[0],mm4);\
1295 punpcklbw_r2r(mm7,mm0);\
1296 movd_r2m(mm2,lum_m4[0]);\
1297 punpcklbw_r2r(mm7,mm1);\
1298 punpcklbw_r2r(mm7,mm2);\
1299 punpcklbw_r2r(mm7,mm3);\
1300 punpcklbw_r2r(mm7,mm4);\
1301 paddw_r2r(mm3,mm1);\
1302 psllw_i2r(1,mm2);\
1303 paddw_r2r(mm4,mm0);\
1304 psllw_i2r(2,mm1);\
1305 paddw_r2r(mm6,mm2);\
1306 paddw_r2r(mm2,mm1);\
1307 psubusw_r2r(mm0,mm1);\
1308 psrlw_i2r(3,mm1);\
1309 packuswb_r2r(mm7,mm1);\
1310 movd_r2m(mm1,lum_m2[0]);
1311
1312 #define DEINT_LINE_LUM \
1313 movd_m2r(lum_m4[0],mm0);\
1314 movd_m2r(lum_m3[0],mm1);\
1315 movd_m2r(lum_m2[0],mm2);\
1316 movd_m2r(lum_m1[0],mm3);\
1317 movd_m2r(lum[0],mm4);\
1318 punpcklbw_r2r(mm7,mm0);\
1319 punpcklbw_r2r(mm7,mm1);\
1320 punpcklbw_r2r(mm7,mm2);\
1321 punpcklbw_r2r(mm7,mm3);\
1322 punpcklbw_r2r(mm7,mm4);\
1323 paddw_r2r(mm3,mm1);\
1324 psllw_i2r(1,mm2);\
1325 paddw_r2r(mm4,mm0);\
1326 psllw_i2r(2,mm1);\
1327 paddw_r2r(mm6,mm2);\
1328 paddw_r2r(mm2,mm1);\
1329 psubusw_r2r(mm0,mm1);\
1330 psrlw_i2r(3,mm1);\
1331 packuswb_r2r(mm7,mm1);\
1332 movd_r2m(mm1,dst[0]);
1333 #endif
1334
1335 /* filter parameters: [-1 4 2 4 -1] // 8 */
1336 static void deinterlace_line(uint8_t *dst,
1337 const uint8_t *lum_m4, const uint8_t *lum_m3,
1338 const uint8_t *lum_m2, const uint8_t *lum_m1,
1339 const uint8_t *lum,
1340 int size)
1341 {
1342 #if !HAVE_MMX
1343 uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
1344 int sum;
1345
1346 for(;size > 0;size--) {
1347 sum = -lum_m4[0];
1348 sum += lum_m3[0] << 2;
1349 sum += lum_m2[0] << 1;
1350 sum += lum_m1[0] << 2;
1351 sum += -lum[0];
1352 dst[0] = cm[(sum + 4) >> 3];
1353 lum_m4++;
1354 lum_m3++;
1355 lum_m2++;
1356 lum_m1++;
1357 lum++;
1358 dst++;
1359 }
1360 #else
1361
1362 {
1363 pxor_r2r(mm7,mm7);
1364 movq_m2r(ff_pw_4,mm6);
1365 }
1366 for (;size > 3; size-=4) {
1367 DEINT_LINE_LUM
1368 lum_m4+=4;
1369 lum_m3+=4;
1370 lum_m2+=4;
1371 lum_m1+=4;
1372 lum+=4;
1373 dst+=4;
1374 }
1375 #endif
1376 }
1377 static void deinterlace_line_inplace(uint8_t *lum_m4, uint8_t *lum_m3, uint8_t *lum_m2, uint8_t *lum_m1, uint8_t *lum,
1378 int size)
1379 {
1380 #if !HAVE_MMX
1381 uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
1382 int sum;
1383
1384 for(;size > 0;size--) {
1385 sum = -lum_m4[0];
1386 sum += lum_m3[0] << 2;
1387 sum += lum_m2[0] << 1;
1388 lum_m4[0]=lum_m2[0];
1389 sum += lum_m1[0] << 2;
1390 sum += -lum[0];
1391 lum_m2[0] = cm[(sum + 4) >> 3];
1392 lum_m4++;
1393 lum_m3++;
1394 lum_m2++;
1395 lum_m1++;
1396 lum++;
1397 }
1398 #else
1399
1400 {
1401 pxor_r2r(mm7,mm7);
1402 movq_m2r(ff_pw_4,mm6);
1403 }
1404 for (;size > 3; size-=4) {
1405 DEINT_INPLACE_LINE_LUM
1406 lum_m4+=4;
1407 lum_m3+=4;
1408 lum_m2+=4;
1409 lum_m1+=4;
1410 lum+=4;
1411 }
1412 #endif
1413 }
1414
1415 /* deinterlacing : 2 temporal taps, 3 spatial taps linear filter. The
1416 top field is copied as is, but the bottom field is deinterlaced
1417 against the top field. */
1418 static void deinterlace_bottom_field(uint8_t *dst, int dst_wrap,
1419 const uint8_t *src1, int src_wrap,
1420 int width, int height)
1421 {
1422 const uint8_t *src_m2, *src_m1, *src_0, *src_p1, *src_p2;
1423 int y;
1424
1425 src_m2 = src1;
1426 src_m1 = src1;
1427 src_0=&src_m1[src_wrap];
1428 src_p1=&src_0[src_wrap];
1429 src_p2=&src_p1[src_wrap];
1430 for(y=0;y<(height-2);y+=2) {
1431 memcpy(dst,src_m1,width);
1432 dst += dst_wrap;
1433 deinterlace_line(dst,src_m2,src_m1,src_0,src_p1,src_p2,width);
1434 src_m2 = src_0;
1435 src_m1 = src_p1;
1436 src_0 = src_p2;
1437 src_p1 += 2*src_wrap;
1438 src_p2 += 2*src_wrap;
1439 dst += dst_wrap;
1440 }
1441 memcpy(dst,src_m1,width);
1442 dst += dst_wrap;
1443 /* do last line */
1444 deinterlace_line(dst,src_m2,src_m1,src_0,src_0,src_0,width);
1445 }
1446
1447 static void deinterlace_bottom_field_inplace(uint8_t *src1, int src_wrap,
1448 int width, int height)
1449 {
1450 uint8_t *src_m1, *src_0, *src_p1, *src_p2;
1451 int y;
1452 uint8_t *buf;
1453 buf = (uint8_t*)av_malloc(width);
1454
1455 src_m1 = src1;
1456 memcpy(buf,src_m1,width);
1457 src_0=&src_m1[src_wrap];
1458 src_p1=&src_0[src_wrap];
1459 src_p2=&src_p1[src_wrap];
1460 for(y=0;y<(height-2);y+=2) {
1461 deinterlace_line_inplace(buf,src_m1,src_0,src_p1,src_p2,width);
1462 src_m1 = src_p1;
1463 src_0 = src_p2;
1464 src_p1 += 2*src_wrap;
1465 src_p2 += 2*src_wrap;
1466 }
1467 /* do last line */
1468 deinterlace_line_inplace(buf,src_m1,src_0,src_0,src_0,width);
1469 av_free(buf);
1470 }
1471
1472 int avpicture_deinterlace(AVPicture *dst, const AVPicture *src,
1473 int pix_fmt, int width, int height)
1474 {
1475 int i;
1476
1477 if (pix_fmt != PIX_FMT_YUV420P &&
1478 pix_fmt != PIX_FMT_YUV422P &&
1479 pix_fmt != PIX_FMT_YUV444P &&
1480 pix_fmt != PIX_FMT_YUV411P &&
1481 pix_fmt != PIX_FMT_GRAY8)
1482 return -1;
1483 if ((width & 3) != 0 || (height & 3) != 0)
1484 return -1;
1485
1486 for(i=0;i<3;i++) {
1487 if (i == 1) {
1488 switch(pix_fmt) {
1489 case PIX_FMT_YUV420P:
1490 width >>= 1;
1491 height >>= 1;
1492 break;
1493 case PIX_FMT_YUV422P:
1494 width >>= 1;
1495 break;
1496 case PIX_FMT_YUV411P:
1497 width >>= 2;
1498 break;
1499 default:
1500 break;
1501 }
1502 if (pix_fmt == PIX_FMT_GRAY8) {
1503 break;
1504 }
1505 }
1506 if (src == dst) {
1507 deinterlace_bottom_field_inplace(dst->data[i], dst->linesize[i],
1508 width, height);
1509 } else {
1510 deinterlace_bottom_field(dst->data[i],dst->linesize[i],
1511 src->data[i], src->linesize[i],
1512 width, height);
1513 }
1514 }
1515 emms_c();
1516 return 0;
1517 }
1518