2 * Misc image conversion routines
3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
5 * This file is part of FFmpeg.
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.
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.
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
24 * misc image conversion routines
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
36 #include "imgconvert.h"
37 #include "libavutil/colorspace.h"
38 #include "libavutil/pixdesc.h"
42 #include "x86/dsputil_mmx.h"
45 #define xglue(x, y) x ## y
46 #define glue(x, y) xglue(x, y)
48 #define FF_COLOR_RGB 0 /**< RGB color space */
49 #define FF_COLOR_GRAY 1 /**< gray color space */
50 #define FF_COLOR_YUV 2 /**< YUV color space. 16 <= Y <= 235, 16 <= U, V <= 240 */
51 #define FF_COLOR_YUV_JPEG 3 /**< YUV color space. 0 <= Y <= 255, 0 <= U, V <= 255 */
53 #define FF_PIXEL_PLANAR 0 /**< each channel has one component in AVPicture */
54 #define FF_PIXEL_PACKED 1 /**< only one components containing all the channels */
55 #define FF_PIXEL_PALETTE 2 /**< one components containing indexes for a palette */
57 typedef struct PixFmtInfo {
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) */
61 uint8_t is_alpha : 1; /**< true if alpha can be specified */
62 uint8_t depth; /**< bit depth of the color components */
65 /* this table gives more information about formats */
66 static const PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {
70 .color_type = FF_COLOR_YUV,
71 .pixel_type = FF_PIXEL_PLANAR,
76 .color_type = FF_COLOR_YUV,
77 .pixel_type = FF_PIXEL_PLANAR,
82 .color_type = FF_COLOR_YUV,
83 .pixel_type = FF_PIXEL_PLANAR,
88 .color_type = FF_COLOR_YUV,
89 .pixel_type = FF_PIXEL_PACKED,
94 .color_type = FF_COLOR_YUV,
95 .pixel_type = FF_PIXEL_PACKED,
100 .color_type = FF_COLOR_YUV,
101 .pixel_type = FF_PIXEL_PLANAR,
104 [PIX_FMT_YUV411P] = {
106 .color_type = FF_COLOR_YUV,
107 .pixel_type = FF_PIXEL_PLANAR,
110 [PIX_FMT_YUV440P] = {
112 .color_type = FF_COLOR_YUV,
113 .pixel_type = FF_PIXEL_PLANAR,
116 [PIX_FMT_YUV420P16LE] = {
118 .color_type = FF_COLOR_YUV,
119 .pixel_type = FF_PIXEL_PLANAR,
122 [PIX_FMT_YUV422P16LE] = {
124 .color_type = FF_COLOR_YUV,
125 .pixel_type = FF_PIXEL_PLANAR,
128 [PIX_FMT_YUV444P16LE] = {
130 .color_type = FF_COLOR_YUV,
131 .pixel_type = FF_PIXEL_PLANAR,
134 [PIX_FMT_YUV420P16BE] = {
136 .color_type = FF_COLOR_YUV,
137 .pixel_type = FF_PIXEL_PLANAR,
140 [PIX_FMT_YUV422P16BE] = {
142 .color_type = FF_COLOR_YUV,
143 .pixel_type = FF_PIXEL_PLANAR,
146 [PIX_FMT_YUV444P16BE] = {
148 .color_type = FF_COLOR_YUV,
149 .pixel_type = FF_PIXEL_PLANAR,
154 /* YUV formats with alpha plane */
155 [PIX_FMT_YUVA420P] = {
157 .color_type = FF_COLOR_YUV,
158 .pixel_type = FF_PIXEL_PLANAR,
163 [PIX_FMT_YUVJ420P] = {
165 .color_type = FF_COLOR_YUV_JPEG,
166 .pixel_type = FF_PIXEL_PLANAR,
169 [PIX_FMT_YUVJ422P] = {
171 .color_type = FF_COLOR_YUV_JPEG,
172 .pixel_type = FF_PIXEL_PLANAR,
175 [PIX_FMT_YUVJ444P] = {
177 .color_type = FF_COLOR_YUV_JPEG,
178 .pixel_type = FF_PIXEL_PLANAR,
181 [PIX_FMT_YUVJ440P] = {
183 .color_type = FF_COLOR_YUV_JPEG,
184 .pixel_type = FF_PIXEL_PLANAR,
191 .color_type = FF_COLOR_RGB,
192 .pixel_type = FF_PIXEL_PACKED,
197 .color_type = FF_COLOR_RGB,
198 .pixel_type = FF_PIXEL_PACKED,
202 .nb_channels = 4, .is_alpha = 1,
203 .color_type = FF_COLOR_RGB,
204 .pixel_type = FF_PIXEL_PACKED,
207 [PIX_FMT_RGB48BE] = {
209 .color_type = FF_COLOR_RGB,
210 .pixel_type = FF_PIXEL_PACKED,
213 [PIX_FMT_RGB48LE] = {
215 .color_type = FF_COLOR_RGB,
216 .pixel_type = FF_PIXEL_PACKED,
219 [PIX_FMT_RGB565BE] = {
221 .color_type = FF_COLOR_RGB,
222 .pixel_type = FF_PIXEL_PACKED,
225 [PIX_FMT_RGB565LE] = {
227 .color_type = FF_COLOR_RGB,
228 .pixel_type = FF_PIXEL_PACKED,
231 [PIX_FMT_RGB555BE] = {
233 .color_type = FF_COLOR_RGB,
234 .pixel_type = FF_PIXEL_PACKED,
237 [PIX_FMT_RGB555LE] = {
239 .color_type = FF_COLOR_RGB,
240 .pixel_type = FF_PIXEL_PACKED,
243 [PIX_FMT_RGB444BE] = {
245 .color_type = FF_COLOR_RGB,
246 .pixel_type = FF_PIXEL_PACKED,
249 [PIX_FMT_RGB444LE] = {
251 .color_type = FF_COLOR_RGB,
252 .pixel_type = FF_PIXEL_PACKED,
256 /* gray / mono formats */
257 [PIX_FMT_GRAY16BE] = {
259 .color_type = FF_COLOR_GRAY,
260 .pixel_type = FF_PIXEL_PLANAR,
263 [PIX_FMT_GRAY16LE] = {
265 .color_type = FF_COLOR_GRAY,
266 .pixel_type = FF_PIXEL_PLANAR,
271 .color_type = FF_COLOR_GRAY,
272 .pixel_type = FF_PIXEL_PLANAR,
275 [PIX_FMT_MONOWHITE] = {
277 .color_type = FF_COLOR_GRAY,
278 .pixel_type = FF_PIXEL_PLANAR,
281 [PIX_FMT_MONOBLACK] = {
283 .color_type = FF_COLOR_GRAY,
284 .pixel_type = FF_PIXEL_PLANAR,
288 /* paletted formats */
290 .nb_channels = 4, .is_alpha = 1,
291 .color_type = FF_COLOR_RGB,
292 .pixel_type = FF_PIXEL_PALETTE,
295 [PIX_FMT_UYYVYY411] = {
297 .color_type = FF_COLOR_YUV,
298 .pixel_type = FF_PIXEL_PACKED,
302 .nb_channels = 4, .is_alpha = 1,
303 .color_type = FF_COLOR_RGB,
304 .pixel_type = FF_PIXEL_PACKED,
307 [PIX_FMT_BGR565BE] = {
309 .color_type = FF_COLOR_RGB,
310 .pixel_type = FF_PIXEL_PACKED,
313 [PIX_FMT_BGR565LE] = {
315 .color_type = FF_COLOR_RGB,
316 .pixel_type = FF_PIXEL_PACKED,
319 [PIX_FMT_BGR555BE] = {
321 .color_type = FF_COLOR_RGB,
322 .pixel_type = FF_PIXEL_PACKED,
325 [PIX_FMT_BGR555LE] = {
327 .color_type = FF_COLOR_RGB,
328 .pixel_type = FF_PIXEL_PACKED,
331 [PIX_FMT_BGR444BE] = {
333 .color_type = FF_COLOR_RGB,
334 .pixel_type = FF_PIXEL_PACKED,
337 [PIX_FMT_BGR444LE] = {
339 .color_type = FF_COLOR_RGB,
340 .pixel_type = FF_PIXEL_PACKED,
345 .color_type = FF_COLOR_RGB,
346 .pixel_type = FF_PIXEL_PACKED,
351 .color_type = FF_COLOR_RGB,
352 .pixel_type = FF_PIXEL_PACKED,
355 [PIX_FMT_RGB4_BYTE] = {
357 .color_type = FF_COLOR_RGB,
358 .pixel_type = FF_PIXEL_PACKED,
363 .color_type = FF_COLOR_RGB,
364 .pixel_type = FF_PIXEL_PACKED,
369 .color_type = FF_COLOR_RGB,
370 .pixel_type = FF_PIXEL_PACKED,
373 [PIX_FMT_BGR4_BYTE] = {
375 .color_type = FF_COLOR_RGB,
376 .pixel_type = FF_PIXEL_PACKED,
381 .color_type = FF_COLOR_YUV,
382 .pixel_type = FF_PIXEL_PLANAR,
387 .color_type = FF_COLOR_YUV,
388 .pixel_type = FF_PIXEL_PLANAR,
393 .nb_channels = 4, .is_alpha = 1,
394 .color_type = FF_COLOR_RGB,
395 .pixel_type = FF_PIXEL_PACKED,
399 .nb_channels = 4, .is_alpha = 1,
400 .color_type = FF_COLOR_RGB,
401 .pixel_type = FF_PIXEL_PACKED,
406 void avcodec_get_chroma_sub_sample(enum PixelFormat pix_fmt, int *h_shift, int *v_shift)
408 *h_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_w;
409 *v_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_h;
412 const char *avcodec_get_pix_fmt_name(enum PixelFormat pix_fmt)
414 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB)
417 return av_pix_fmt_descriptors[pix_fmt].name;
420 #if LIBAVCODEC_VERSION_MAJOR < 53
421 enum PixelFormat avcodec_get_pix_fmt(const char *name)
423 return av_get_pix_fmt(name);
427 void avcodec_pix_fmt_string (char *buf, int buf_size, enum PixelFormat pix_fmt)
431 snprintf (buf, buf_size,
432 "name " " nb_channels" " depth" " is_alpha"
435 PixFmtInfo info= pix_fmt_info[pix_fmt];
437 char is_alpha_char= info.is_alpha ? 'y' : 'n';
439 snprintf (buf, buf_size,
441 av_pix_fmt_descriptors[pix_fmt].name,
449 int ff_is_hwaccel_pix_fmt(enum PixelFormat pix_fmt)
451 return av_pix_fmt_descriptors[pix_fmt].flags & PIX_FMT_HWACCEL;
454 int ff_set_systematic_pal(uint32_t pal[256], enum PixelFormat pix_fmt){
457 for(i=0; i<256; i++){
471 case PIX_FMT_RGB4_BYTE:
476 case PIX_FMT_BGR4_BYTE:
487 pal[i] = b + (g<<8) + (r<<16);
493 static int fill_image_linesize(int linesize[4], enum PixelFormat pix_fmt, int width)
496 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
497 int max_plane_step [4];
498 int max_plane_step_comp[4];
500 memset(linesize, 0, 4*sizeof(linesize[0]));
502 if (desc->flags & PIX_FMT_HWACCEL)
505 if (desc->flags & PIX_FMT_BITSTREAM) {
506 linesize[0] = (width * (desc->comp[0].step_minus1+1) + 7) >> 3;
510 memset(max_plane_step , 0, sizeof(max_plane_step ));
511 memset(max_plane_step_comp, 0, sizeof(max_plane_step_comp));
512 for (i = 0; i < 4; i++) {
513 const AVComponentDescriptor *comp = &(desc->comp[i]);
514 if ((comp->step_minus1+1) > max_plane_step[comp->plane]) {
515 max_plane_step [comp->plane] = comp->step_minus1+1;
516 max_plane_step_comp[comp->plane] = i;
520 for (i = 0; i < 4; i++) {
521 int s = (max_plane_step_comp[i] == 1 || max_plane_step_comp[i] == 2) ? desc->log2_chroma_w : 0;
522 linesize[i] = max_plane_step[i] * (((width + (1 << s) - 1)) >> s);
528 int ff_fill_linesize(AVPicture *picture, enum PixelFormat pix_fmt, int width)
530 return fill_image_linesize(picture->linesize, pix_fmt, width);
533 static int fill_image_data_ptr(uint8_t *data[4], uint8_t *ptr, enum PixelFormat pix_fmt,
534 int height, const int linesize[4])
537 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
539 size = linesize[0] * height;
541 case PIX_FMT_YUV420P:
542 case PIX_FMT_YUV422P:
543 case PIX_FMT_YUV444P:
544 case PIX_FMT_YUV410P:
545 case PIX_FMT_YUV411P:
546 case PIX_FMT_YUV440P:
547 case PIX_FMT_YUVJ420P:
548 case PIX_FMT_YUVJ422P:
549 case PIX_FMT_YUVJ444P:
550 case PIX_FMT_YUVJ440P:
551 case PIX_FMT_YUV420P16LE:
552 case PIX_FMT_YUV422P16LE:
553 case PIX_FMT_YUV444P16LE:
554 case PIX_FMT_YUV420P16BE:
555 case PIX_FMT_YUV422P16BE:
556 case PIX_FMT_YUV444P16BE:
557 h2 = (height + (1 << desc->log2_chroma_h) - 1) >> desc->log2_chroma_h;
558 size2 = linesize[1] * h2;
560 data[1] = data[0] + size;
561 data[2] = data[1] + size2;
563 return size + 2 * size2;
564 case PIX_FMT_YUVA420P:
565 h2 = (height + (1 << desc->log2_chroma_h) - 1) >> desc->log2_chroma_h;
566 size2 = linesize[1] * h2;
568 data[1] = data[0] + size;
569 data[2] = data[1] + size2;
570 data[3] = data[1] + size2 + size2;
571 return 2 * size + 2 * size2;
574 h2 = (height + (1 << desc->log2_chroma_h) - 1) >> desc->log2_chroma_h;
575 size2 = linesize[1] * h2;
577 data[1] = data[0] + size;
587 case PIX_FMT_RGB48BE:
588 case PIX_FMT_RGB48LE:
589 case PIX_FMT_GRAY16BE:
590 case PIX_FMT_GRAY16LE:
591 case PIX_FMT_BGR444BE:
592 case PIX_FMT_BGR444LE:
593 case PIX_FMT_BGR555BE:
594 case PIX_FMT_BGR555LE:
595 case PIX_FMT_BGR565BE:
596 case PIX_FMT_BGR565LE:
597 case PIX_FMT_RGB444BE:
598 case PIX_FMT_RGB444LE:
599 case PIX_FMT_RGB555BE:
600 case PIX_FMT_RGB555LE:
601 case PIX_FMT_RGB565BE:
602 case PIX_FMT_RGB565LE:
603 case PIX_FMT_YUYV422:
604 case PIX_FMT_UYVY422:
605 case PIX_FMT_UYYVYY411:
608 case PIX_FMT_MONOWHITE:
609 case PIX_FMT_MONOBLACK:
619 case PIX_FMT_RGB4_BYTE:
620 case PIX_FMT_BGR4_BYTE:
622 size2 = (size + 3) & ~3;
624 data[1] = ptr + size2; /* palette is stored here as 256 32 bit words */
627 return size2 + 256 * 4;
637 int ff_fill_pointer(AVPicture *picture, uint8_t *ptr, enum PixelFormat pix_fmt,
640 return fill_image_data_ptr(picture->data, ptr, pix_fmt, height, picture->linesize);
643 int avpicture_fill(AVPicture *picture, uint8_t *ptr,
644 enum PixelFormat pix_fmt, int width, int height)
647 if(avcodec_check_dimensions(NULL, width, height))
650 if (ff_fill_linesize(picture, pix_fmt, width))
653 return ff_fill_pointer(picture, ptr, pix_fmt, height);
656 int avpicture_layout(const AVPicture* src, enum PixelFormat pix_fmt, int width, int height,
657 unsigned char *dest, int dest_size)
659 const PixFmtInfo* pf = &pix_fmt_info[pix_fmt];
660 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
661 int i, j, w, ow, h, oh, data_planes;
662 const unsigned char* s;
663 int size = avpicture_get_size(pix_fmt, width, height);
665 if (size > dest_size || size < 0)
668 if (pf->pixel_type == FF_PIXEL_PACKED || pf->pixel_type == FF_PIXEL_PALETTE) {
669 if (pix_fmt == PIX_FMT_YUYV422 ||
670 pix_fmt == PIX_FMT_UYVY422 ||
671 pix_fmt == PIX_FMT_BGR565BE ||
672 pix_fmt == PIX_FMT_BGR565LE ||
673 pix_fmt == PIX_FMT_BGR555BE ||
674 pix_fmt == PIX_FMT_BGR555LE ||
675 pix_fmt == PIX_FMT_BGR444BE ||
676 pix_fmt == PIX_FMT_BGR444LE ||
677 pix_fmt == PIX_FMT_RGB565BE ||
678 pix_fmt == PIX_FMT_RGB565LE ||
679 pix_fmt == PIX_FMT_RGB555BE ||
680 pix_fmt == PIX_FMT_RGB555LE ||
681 pix_fmt == PIX_FMT_RGB444BE ||
682 pix_fmt == PIX_FMT_RGB444LE)
684 else if (pix_fmt == PIX_FMT_UYYVYY411)
686 else if (pix_fmt == PIX_FMT_PAL8)
689 w = width * (pf->depth * pf->nb_channels / 8);
694 data_planes = pf->nb_channels;
695 w = (width*pf->depth + 7)/8;
702 for (i=0; i<data_planes; i++) {
704 w = (- ((-width) >> desc->log2_chroma_w) * pf->depth + 7) / 8;
705 h = -((-height) >> desc->log2_chroma_h);
706 if (pix_fmt == PIX_FMT_NV12 || pix_fmt == PIX_FMT_NV21)
716 s += src->linesize[i];
720 if (pf->pixel_type == FF_PIXEL_PALETTE)
721 memcpy((unsigned char *)(((size_t)dest + 3) & ~3), src->data[1], 256 * 4);
726 int avpicture_get_size(enum PixelFormat pix_fmt, int width, int height)
728 AVPicture dummy_pict;
729 if(avcodec_check_dimensions(NULL, width, height))
734 case PIX_FMT_RGB4_BYTE:
735 case PIX_FMT_BGR4_BYTE:
737 // do not include palette for these pseudo-paletted formats
738 return width * height;
740 return avpicture_fill(&dummy_pict, NULL, pix_fmt, width, height);
743 int avcodec_get_pix_fmt_loss(enum PixelFormat dst_pix_fmt, enum PixelFormat src_pix_fmt,
746 const PixFmtInfo *pf, *ps;
747 const AVPixFmtDescriptor *src_desc = &av_pix_fmt_descriptors[src_pix_fmt];
748 const AVPixFmtDescriptor *dst_desc = &av_pix_fmt_descriptors[dst_pix_fmt];
751 ps = &pix_fmt_info[src_pix_fmt];
755 pf = &pix_fmt_info[dst_pix_fmt];
756 if (pf->depth < ps->depth ||
757 ((dst_pix_fmt == PIX_FMT_RGB555BE || dst_pix_fmt == PIX_FMT_RGB555LE ||
758 dst_pix_fmt == PIX_FMT_BGR555BE || dst_pix_fmt == PIX_FMT_BGR555LE) &&
759 (src_pix_fmt == PIX_FMT_RGB565BE || src_pix_fmt == PIX_FMT_RGB565LE ||
760 src_pix_fmt == PIX_FMT_BGR565BE || src_pix_fmt == PIX_FMT_BGR565LE)))
761 loss |= FF_LOSS_DEPTH;
762 if (dst_desc->log2_chroma_w > src_desc->log2_chroma_w ||
763 dst_desc->log2_chroma_h > src_desc->log2_chroma_h)
764 loss |= FF_LOSS_RESOLUTION;
765 switch(pf->color_type) {
767 if (ps->color_type != FF_COLOR_RGB &&
768 ps->color_type != FF_COLOR_GRAY)
769 loss |= FF_LOSS_COLORSPACE;
772 if (ps->color_type != FF_COLOR_GRAY)
773 loss |= FF_LOSS_COLORSPACE;
776 if (ps->color_type != FF_COLOR_YUV)
777 loss |= FF_LOSS_COLORSPACE;
779 case FF_COLOR_YUV_JPEG:
780 if (ps->color_type != FF_COLOR_YUV_JPEG &&
781 ps->color_type != FF_COLOR_YUV &&
782 ps->color_type != FF_COLOR_GRAY)
783 loss |= FF_LOSS_COLORSPACE;
787 if (ps->color_type != pf->color_type)
788 loss |= FF_LOSS_COLORSPACE;
791 if (pf->color_type == FF_COLOR_GRAY &&
792 ps->color_type != FF_COLOR_GRAY)
793 loss |= FF_LOSS_CHROMA;
794 if (!pf->is_alpha && (ps->is_alpha && has_alpha))
795 loss |= FF_LOSS_ALPHA;
796 if (pf->pixel_type == FF_PIXEL_PALETTE &&
797 (ps->pixel_type != FF_PIXEL_PALETTE && ps->color_type != FF_COLOR_GRAY))
798 loss |= FF_LOSS_COLORQUANT;
802 static int avg_bits_per_pixel(enum PixelFormat pix_fmt)
805 const PixFmtInfo *pf;
806 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
808 pf = &pix_fmt_info[pix_fmt];
809 switch(pf->pixel_type) {
810 case FF_PIXEL_PACKED:
812 case PIX_FMT_YUYV422:
813 case PIX_FMT_UYVY422:
814 case PIX_FMT_RGB565BE:
815 case PIX_FMT_RGB565LE:
816 case PIX_FMT_RGB555BE:
817 case PIX_FMT_RGB555LE:
818 case PIX_FMT_RGB444BE:
819 case PIX_FMT_RGB444LE:
820 case PIX_FMT_BGR565BE:
821 case PIX_FMT_BGR565LE:
822 case PIX_FMT_BGR555BE:
823 case PIX_FMT_BGR555LE:
824 case PIX_FMT_BGR444BE:
825 case PIX_FMT_BGR444LE:
828 case PIX_FMT_UYYVYY411:
832 bits = pf->depth * pf->nb_channels;
836 case FF_PIXEL_PLANAR:
837 if (desc->log2_chroma_w == 0 && desc->log2_chroma_h == 0) {
838 bits = pf->depth * pf->nb_channels;
840 bits = pf->depth + ((2 * pf->depth) >>
841 (desc->log2_chroma_w + desc->log2_chroma_h));
844 case FF_PIXEL_PALETTE:
854 static enum PixelFormat avcodec_find_best_pix_fmt1(int64_t pix_fmt_mask,
855 enum PixelFormat src_pix_fmt,
859 int dist, i, loss, min_dist;
860 enum PixelFormat dst_pix_fmt;
862 /* find exact color match with smallest size */
863 dst_pix_fmt = PIX_FMT_NONE;
864 min_dist = 0x7fffffff;
865 for(i = 0;i < PIX_FMT_NB; i++) {
866 if (pix_fmt_mask & (1ULL << i)) {
867 loss = avcodec_get_pix_fmt_loss(i, src_pix_fmt, has_alpha) & loss_mask;
869 dist = avg_bits_per_pixel(i);
870 if (dist < min_dist) {
880 enum PixelFormat avcodec_find_best_pix_fmt(int64_t pix_fmt_mask, enum PixelFormat src_pix_fmt,
881 int has_alpha, int *loss_ptr)
883 enum PixelFormat dst_pix_fmt;
885 static const int loss_mask_order[] = {
886 ~0, /* no loss first */
889 ~(FF_LOSS_COLORSPACE | FF_LOSS_RESOLUTION),
895 /* try with successive loss */
898 loss_mask = loss_mask_order[i++];
899 dst_pix_fmt = avcodec_find_best_pix_fmt1(pix_fmt_mask, src_pix_fmt,
900 has_alpha, loss_mask);
901 if (dst_pix_fmt >= 0)
909 *loss_ptr = avcodec_get_pix_fmt_loss(dst_pix_fmt, src_pix_fmt, has_alpha);
913 void ff_img_copy_plane(uint8_t *dst, int dst_wrap,
914 const uint8_t *src, int src_wrap,
915 int width, int height)
919 for(;height > 0; height--) {
920 memcpy(dst, src, width);
926 int ff_get_plane_bytewidth(enum PixelFormat pix_fmt, int width, int plane)
929 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
930 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
932 pf = &pix_fmt_info[pix_fmt];
933 switch(pf->pixel_type) {
934 case FF_PIXEL_PACKED:
936 case PIX_FMT_YUYV422:
937 case PIX_FMT_UYVY422:
938 case PIX_FMT_RGB565BE:
939 case PIX_FMT_RGB565LE:
940 case PIX_FMT_RGB555BE:
941 case PIX_FMT_RGB555LE:
942 case PIX_FMT_RGB444BE:
943 case PIX_FMT_RGB444LE:
944 case PIX_FMT_BGR565BE:
945 case PIX_FMT_BGR565LE:
946 case PIX_FMT_BGR555BE:
947 case PIX_FMT_BGR555LE:
948 case PIX_FMT_BGR444BE:
949 case PIX_FMT_BGR444LE:
952 case PIX_FMT_UYYVYY411:
956 bits = pf->depth * pf->nb_channels;
959 return (width * bits + 7) >> 3;
961 case FF_PIXEL_PLANAR:
962 if ((pix_fmt != PIX_FMT_NV12 && pix_fmt != PIX_FMT_NV21) &&
963 (plane == 1 || plane == 2))
964 width= -((-width)>>desc->log2_chroma_w);
966 return (width * pf->depth + 7) >> 3;
968 case FF_PIXEL_PALETTE:
977 void av_picture_copy(AVPicture *dst, const AVPicture *src,
978 enum PixelFormat pix_fmt, int width, int height)
981 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
982 const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
984 switch(pf->pixel_type) {
985 case FF_PIXEL_PACKED:
986 case FF_PIXEL_PLANAR:
987 for(i = 0; i < pf->nb_channels; i++) {
989 int bwidth = ff_get_plane_bytewidth(pix_fmt, width, i);
991 if (i == 1 || i == 2) {
992 h= -((-height)>>desc->log2_chroma_h);
994 ff_img_copy_plane(dst->data[i], dst->linesize[i],
995 src->data[i], src->linesize[i],
999 case FF_PIXEL_PALETTE:
1000 ff_img_copy_plane(dst->data[0], dst->linesize[0],
1001 src->data[0], src->linesize[0],
1003 /* copy the palette */
1004 memcpy(dst->data[1], src->data[1], 4*256);
1010 void ff_shrink22(uint8_t *dst, int dst_wrap,
1011 const uint8_t *src, int src_wrap,
1012 int width, int height)
1015 const uint8_t *s1, *s2;
1018 for(;height > 0; height--) {
1022 for(w = width;w >= 4; w-=4) {
1023 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
1024 d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
1025 d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
1026 d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
1032 d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
1037 src += 2 * src_wrap;
1043 void ff_shrink44(uint8_t *dst, int dst_wrap,
1044 const uint8_t *src, int src_wrap,
1045 int width, int height)
1048 const uint8_t *s1, *s2, *s3, *s4;
1051 for(;height > 0; height--) {
1057 for(w = width;w > 0; w--) {
1058 d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
1059 s2[0] + s2[1] + s2[2] + s2[3] +
1060 s3[0] + s3[1] + s3[2] + s3[3] +
1061 s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
1068 src += 4 * src_wrap;
1074 void ff_shrink88(uint8_t *dst, int dst_wrap,
1075 const uint8_t *src, int src_wrap,
1076 int width, int height)
1080 for(;height > 0; height--) {
1081 for(w = width;w > 0; w--) {
1084 tmp += src[0] + src[1] + src[2] + src[3] + src[4] + src[5] + src[6] + src[7];
1087 *(dst++) = (tmp + 32)>>6;
1088 src += 8 - 8*src_wrap;
1090 src += 8*src_wrap - 8*width;
1091 dst += dst_wrap - width;
1096 int avpicture_alloc(AVPicture *picture,
1097 enum PixelFormat pix_fmt, int width, int height)
1102 size = avpicture_fill(picture, NULL, pix_fmt, width, height);
1105 ptr = av_malloc(size);
1108 avpicture_fill(picture, ptr, pix_fmt, width, height);
1109 if(picture->data[1] && !picture->data[2])
1110 ff_set_systematic_pal((uint32_t*)picture->data[1], pix_fmt);
1114 memset(picture, 0, sizeof(AVPicture));
1118 void avpicture_free(AVPicture *picture)
1120 av_free(picture->data[0]);
1123 /* return true if yuv planar */
1124 static inline int is_yuv_planar(const PixFmtInfo *ps)
1126 return (ps->color_type == FF_COLOR_YUV ||
1127 ps->color_type == FF_COLOR_YUV_JPEG) &&
1128 ps->pixel_type == FF_PIXEL_PLANAR;
1131 int av_picture_crop(AVPicture *dst, const AVPicture *src,
1132 enum PixelFormat pix_fmt, int top_band, int left_band)
1137 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB || !is_yuv_planar(&pix_fmt_info[pix_fmt]))
1140 y_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_h;
1141 x_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_w;
1143 dst->data[0] = src->data[0] + (top_band * src->linesize[0]) + left_band;
1144 dst->data[1] = src->data[1] + ((top_band >> y_shift) * src->linesize[1]) + (left_band >> x_shift);
1145 dst->data[2] = src->data[2] + ((top_band >> y_shift) * src->linesize[2]) + (left_band >> x_shift);
1147 dst->linesize[0] = src->linesize[0];
1148 dst->linesize[1] = src->linesize[1];
1149 dst->linesize[2] = src->linesize[2];
1153 int av_picture_pad(AVPicture *dst, const AVPicture *src, int height, int width,
1154 enum PixelFormat pix_fmt, int padtop, int padbottom, int padleft, int padright,
1163 if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB ||
1164 !is_yuv_planar(&pix_fmt_info[pix_fmt])) return -1;
1166 for (i = 0; i < 3; i++) {
1167 x_shift = i ? av_pix_fmt_descriptors[pix_fmt].log2_chroma_w : 0;
1168 y_shift = i ? av_pix_fmt_descriptors[pix_fmt].log2_chroma_h : 0;
1170 if (padtop || padleft) {
1171 memset(dst->data[i], color[i],
1172 dst->linesize[i] * (padtop >> y_shift) + (padleft >> x_shift));
1175 if (padleft || padright) {
1176 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1177 (dst->linesize[i] - (padright >> x_shift));
1178 yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
1179 for (y = 0; y < yheight; y++) {
1180 memset(optr, color[i], (padleft + padright) >> x_shift);
1181 optr += dst->linesize[i];
1185 if (src) { /* first line */
1186 uint8_t *iptr = src->data[i];
1187 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1188 (padleft >> x_shift);
1189 memcpy(optr, iptr, (width - padleft - padright) >> x_shift);
1190 iptr += src->linesize[i];
1191 optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
1192 (dst->linesize[i] - (padright >> x_shift));
1193 yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
1194 for (y = 0; y < yheight; y++) {
1195 memset(optr, color[i], (padleft + padright) >> x_shift);
1196 memcpy(optr + ((padleft + padright) >> x_shift), iptr,
1197 (width - padleft - padright) >> x_shift);
1198 iptr += src->linesize[i];
1199 optr += dst->linesize[i];
1203 if (padbottom || padright) {
1204 optr = dst->data[i] + dst->linesize[i] *
1205 ((height - padbottom) >> y_shift) - (padright >> x_shift);
1206 memset(optr, color[i],dst->linesize[i] *
1207 (padbottom >> y_shift) + (padright >> x_shift));
1213 /* NOTE: we scan all the pixels to have an exact information */
1214 static int get_alpha_info_pal8(const AVPicture *src, int width, int height)
1216 const unsigned char *p;
1217 int src_wrap, ret, x, y;
1219 uint32_t *palette = (uint32_t *)src->data[1];
1222 src_wrap = src->linesize[0] - width;
1224 for(y=0;y<height;y++) {
1225 for(x=0;x<width;x++) {
1226 a = palette[p[0]] >> 24;
1228 ret |= FF_ALPHA_TRANSP;
1229 } else if (a != 0xff) {
1230 ret |= FF_ALPHA_SEMI_TRANSP;
1239 int img_get_alpha_info(const AVPicture *src,
1240 enum PixelFormat pix_fmt, int width, int height)
1242 const PixFmtInfo *pf = &pix_fmt_info[pix_fmt];
1245 /* no alpha can be represented in format */
1250 ret = get_alpha_info_pal8(src, width, height);
1253 /* we do not know, so everything is indicated */
1254 ret = FF_ALPHA_TRANSP | FF_ALPHA_SEMI_TRANSP;
1261 #define DEINT_INPLACE_LINE_LUM \
1262 movd_m2r(lum_m4[0],mm0);\
1263 movd_m2r(lum_m3[0],mm1);\
1264 movd_m2r(lum_m2[0],mm2);\
1265 movd_m2r(lum_m1[0],mm3);\
1266 movd_m2r(lum[0],mm4);\
1267 punpcklbw_r2r(mm7,mm0);\
1268 movd_r2m(mm2,lum_m4[0]);\
1269 punpcklbw_r2r(mm7,mm1);\
1270 punpcklbw_r2r(mm7,mm2);\
1271 punpcklbw_r2r(mm7,mm3);\
1272 punpcklbw_r2r(mm7,mm4);\
1273 paddw_r2r(mm3,mm1);\
1275 paddw_r2r(mm4,mm0);\
1277 paddw_r2r(mm6,mm2);\
1278 paddw_r2r(mm2,mm1);\
1279 psubusw_r2r(mm0,mm1);\
1281 packuswb_r2r(mm7,mm1);\
1282 movd_r2m(mm1,lum_m2[0]);
1284 #define DEINT_LINE_LUM \
1285 movd_m2r(lum_m4[0],mm0);\
1286 movd_m2r(lum_m3[0],mm1);\
1287 movd_m2r(lum_m2[0],mm2);\
1288 movd_m2r(lum_m1[0],mm3);\
1289 movd_m2r(lum[0],mm4);\
1290 punpcklbw_r2r(mm7,mm0);\
1291 punpcklbw_r2r(mm7,mm1);\
1292 punpcklbw_r2r(mm7,mm2);\
1293 punpcklbw_r2r(mm7,mm3);\
1294 punpcklbw_r2r(mm7,mm4);\
1295 paddw_r2r(mm3,mm1);\
1297 paddw_r2r(mm4,mm0);\
1299 paddw_r2r(mm6,mm2);\
1300 paddw_r2r(mm2,mm1);\
1301 psubusw_r2r(mm0,mm1);\
1303 packuswb_r2r(mm7,mm1);\
1304 movd_r2m(mm1,dst[0]);
1307 /* filter parameters: [-1 4 2 4 -1] // 8 */
1308 static void deinterlace_line(uint8_t *dst,
1309 const uint8_t *lum_m4, const uint8_t *lum_m3,
1310 const uint8_t *lum_m2, const uint8_t *lum_m1,
1315 uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
1318 for(;size > 0;size--) {
1320 sum += lum_m3[0] << 2;
1321 sum += lum_m2[0] << 1;
1322 sum += lum_m1[0] << 2;
1324 dst[0] = cm[(sum + 4) >> 3];
1336 movq_m2r(ff_pw_4,mm6);
1338 for (;size > 3; size-=4) {
1349 static void deinterlace_line_inplace(uint8_t *lum_m4, uint8_t *lum_m3, uint8_t *lum_m2, uint8_t *lum_m1, uint8_t *lum,
1353 uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
1356 for(;size > 0;size--) {
1358 sum += lum_m3[0] << 2;
1359 sum += lum_m2[0] << 1;
1360 lum_m4[0]=lum_m2[0];
1361 sum += lum_m1[0] << 2;
1363 lum_m2[0] = cm[(sum + 4) >> 3];
1374 movq_m2r(ff_pw_4,mm6);
1376 for (;size > 3; size-=4) {
1377 DEINT_INPLACE_LINE_LUM
1387 /* deinterlacing : 2 temporal taps, 3 spatial taps linear filter. The
1388 top field is copied as is, but the bottom field is deinterlaced
1389 against the top field. */
1390 static void deinterlace_bottom_field(uint8_t *dst, int dst_wrap,
1391 const uint8_t *src1, int src_wrap,
1392 int width, int height)
1394 const uint8_t *src_m2, *src_m1, *src_0, *src_p1, *src_p2;
1399 src_0=&src_m1[src_wrap];
1400 src_p1=&src_0[src_wrap];
1401 src_p2=&src_p1[src_wrap];
1402 for(y=0;y<(height-2);y+=2) {
1403 memcpy(dst,src_m1,width);
1405 deinterlace_line(dst,src_m2,src_m1,src_0,src_p1,src_p2,width);
1409 src_p1 += 2*src_wrap;
1410 src_p2 += 2*src_wrap;
1413 memcpy(dst,src_m1,width);
1416 deinterlace_line(dst,src_m2,src_m1,src_0,src_0,src_0,width);
1419 static void deinterlace_bottom_field_inplace(uint8_t *src1, int src_wrap,
1420 int width, int height)
1422 uint8_t *src_m1, *src_0, *src_p1, *src_p2;
1425 buf = (uint8_t*)av_malloc(width);
1428 memcpy(buf,src_m1,width);
1429 src_0=&src_m1[src_wrap];
1430 src_p1=&src_0[src_wrap];
1431 src_p2=&src_p1[src_wrap];
1432 for(y=0;y<(height-2);y+=2) {
1433 deinterlace_line_inplace(buf,src_m1,src_0,src_p1,src_p2,width);
1436 src_p1 += 2*src_wrap;
1437 src_p2 += 2*src_wrap;
1440 deinterlace_line_inplace(buf,src_m1,src_0,src_0,src_0,width);
1444 int avpicture_deinterlace(AVPicture *dst, const AVPicture *src,
1445 enum PixelFormat pix_fmt, int width, int height)
1449 if (pix_fmt != PIX_FMT_YUV420P &&
1450 pix_fmt != PIX_FMT_YUV422P &&
1451 pix_fmt != PIX_FMT_YUV444P &&
1452 pix_fmt != PIX_FMT_YUV411P &&
1453 pix_fmt != PIX_FMT_GRAY8)
1455 if ((width & 3) != 0 || (height & 3) != 0)
1461 case PIX_FMT_YUV420P:
1465 case PIX_FMT_YUV422P:
1468 case PIX_FMT_YUV411P:
1474 if (pix_fmt == PIX_FMT_GRAY8) {
1479 deinterlace_bottom_field_inplace(dst->data[i], dst->linesize[i],
1482 deinterlace_bottom_field(dst->data[i],dst->linesize[i],
1483 src->data[i], src->linesize[i],