falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1756 Commits
32 Branches
353MB
Tree: d90cc6af20
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'd90cc6af20'
${ noResults }
FFmpeg/libavcodec/imgconvert.c

1516 lines
52KB
Raw Blame History 

  1. /*

  2.  * Misc image convertion routines

  3.  * Copyright (c) 2001, 2002, 2003 Fabrice Bellard.

  4.  *

  5.  * This library is free software; you can redistribute it and/or

  6.  * modify it under the terms of the GNU Lesser General Public

  7.  * License as published by the Free Software Foundation; either

  8.  * version 2 of the License, or (at your option) any later version.

  9.  *

  10.  * This library is distributed in the hope that it will be useful,

  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  13.  * Lesser General Public License for more details.

  14.  *

  15.  * You should have received a copy of the GNU Lesser General Public

  16.  * License along with this library; if not, write to the Free Software

  17.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

  18.  */

  19. #include "avcodec.h"

  20. #include "dsputil.h"

  21. 

  22. #ifdef USE_FASTMEMCPY

  23. #include "fastmemcpy.h"

  24. #endif

  25. 

  26. #ifdef HAVE_MMX

  27. #include "i386/mmx.h"

  28. #endif

  29. 

  30. typedef struct PixFmtInfo {

  31.     const char *name;

  32.     UINT8 nb_components;     /* number of components in AVPicture array  */

  33.     UINT8 is_yuv : 1;    /* true if YUV instead of RGB color space */

  34.     UINT8 is_packed : 1; /* true if multiple components in same word */

  35.     UINT8 is_paletted : 1; /* true if paletted */

  36.     UINT8 is_alpha : 1;    /* true if alpha can be specified */

  37.     UINT8 is_gray : 1;     /* true if gray or monochrome format */

  38.     UINT8 x_chroma_shift; /* X chroma subsampling factor is 2 ^ shift */

  39.     UINT8 y_chroma_shift; /* Y chroma subsampling factor is 2 ^ shift */

  40. } PixFmtInfo;

  41. 

  42. /* this table gives more information about formats */

  43. static PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {

  44.     /* YUV formats */

  45.     [PIX_FMT_YUV420P] = {

  46.         .name = "yuv420p",

  47.         .nb_components = 3, .is_yuv = 1,

  48.         .x_chroma_shift = 1, .y_chroma_shift = 1, 

  49.     },

  50.     [PIX_FMT_YUV422P] = {

  51.         .name = "yuv422p",

  52.         .nb_components = 3, .is_yuv = 1,

  53.         .x_chroma_shift = 1, .y_chroma_shift = 0, 

  54.     },

  55.     [PIX_FMT_YUV444P] = {

  56.         .name = "yuv444p",

  57.         .nb_components = 3, .is_yuv = 1,

  58.         .x_chroma_shift = 0, .y_chroma_shift = 0, 

  59.     },

  60.     [PIX_FMT_YUV422] = {

  61.         .name = "yuv422",

  62.         .nb_components = 1, .is_yuv = 1, .is_packed = 1,

  63.         .x_chroma_shift = 1, .y_chroma_shift = 0,

  64.     },

  65.     [PIX_FMT_YUV410P] = {

  66.         .name = "yuv410p",

  67.         .nb_components = 3, .is_yuv = 1,

  68.         .x_chroma_shift = 2, .y_chroma_shift = 2,

  69.     },

  70.     [PIX_FMT_YUV411P] = {

  71.         .name = "yuv411p",

  72.         .nb_components = 3, .is_yuv = 1,

  73.         .x_chroma_shift = 2, .y_chroma_shift = 0,

  74.     },

  75. 

  76.     /* RGB formats */

  77.     [PIX_FMT_RGB24] = {

  78.         .name = "rgb24",

  79.         .nb_components = 1, .is_packed = 1,

  80.     },

  81.     [PIX_FMT_BGR24] = {

  82.         .name = "bgr24",

  83.         .nb_components = 1, .is_packed = 1,

  84.     },

  85.     [PIX_FMT_RGBA32] = {

  86.         .name = "rgba32",

  87.         .nb_components = 1, .is_packed = 1, .is_alpha = 1,

  88.     },

  89.     [PIX_FMT_RGB565] = {

  90.         .name = "rgb565",

  91.         .nb_components = 1, .is_packed = 1,

  92.     },

  93.     [PIX_FMT_RGB555] = {

  94.         .name = "rgb555",

  95.         .nb_components = 1, .is_packed = 1, .is_alpha = 1,

  96.     },

  97. 

  98.     /* gray / mono formats */

  99.     [PIX_FMT_GRAY8] = {

  100.         .name = "gray",

  101.         .nb_components = 1, .is_gray = 1,

  102.     },

  103.     [PIX_FMT_MONOWHITE] = {

  104.         .name = "monow",

  105.         .nb_components = 1, .is_packed = 1, .is_gray = 1,

  106.     },

  107.     [PIX_FMT_MONOBLACK] = {

  108.         .name = "monob",

  109.         .nb_components = 1, .is_packed = 1, .is_gray = 1,

  110.     },

  111. };

  112. 

  113. void avcodec_get_chroma_sub_sample(int pix_fmt, int *h_shift, int *v_shift)

  114. {

  115.     if (pix_fmt_info[pix_fmt].is_yuv) {

  116.         *h_shift = pix_fmt_info[pix_fmt].x_chroma_shift;

  117.         *v_shift = pix_fmt_info[pix_fmt].y_chroma_shift;

  118.     } else {

  119.         *h_shift=0;

  120.         *v_shift=0;

  121.     }

  122. }

  123. 

  124. const char *avcodec_get_pix_fmt_name(int pix_fmt)

  125. {

  126.     if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB)

  127.         return "???";

  128.     else

  129.         return pix_fmt_info[pix_fmt].name;

  130. }

  131. 

  132. /* Picture field are filled with 'ptr' addresses. Also return size */

  133. int avpicture_fill(AVPicture *picture, UINT8 *ptr,

  134.                    int pix_fmt, int width, int height)

  135. {

  136.     int size, w2, h2, size2;

  137.     PixFmtInfo *pinfo;

  138.     

  139.     pinfo = &pix_fmt_info[pix_fmt];

  140.     size = width * height;

  141.     switch(pix_fmt) {

  142.     case PIX_FMT_YUV420P:

  143.     case PIX_FMT_YUV422P:

  144.     case PIX_FMT_YUV444P:

  145.     case PIX_FMT_YUV410P:

  146.     case PIX_FMT_YUV411P:

  147.         w2 = (width + (1 << pinfo->x_chroma_shift) - 1) >> pinfo->x_chroma_shift;

  148.         h2 = (height + (1 << pinfo->y_chroma_shift) - 1) >> pinfo->y_chroma_shift;

  149.         size2 = w2 * h2;

  150.         picture->data[0] = ptr;

  151.         picture->data[1] = picture->data[0] + size;

  152.         picture->data[2] = picture->data[1] + size2;

  153.         picture->linesize[0] = width;

  154.         picture->linesize[1] = w2;

  155.         picture->linesize[2] = w2;

  156.         return size + 2 * size2;

  157.     case PIX_FMT_RGB24:

  158.     case PIX_FMT_BGR24:

  159.         picture->data[0] = ptr;

  160.         picture->data[1] = NULL;

  161.         picture->data[2] = NULL;

  162.         picture->linesize[0] = width * 3;

  163.         return size * 3;

  164.     case PIX_FMT_RGBA32:

  165.         picture->data[0] = ptr;

  166.         picture->data[1] = NULL;

  167.         picture->data[2] = NULL;

  168.         picture->linesize[0] = width * 4;

  169.         return size * 4;

  170.     case PIX_FMT_RGB555:

  171.     case PIX_FMT_RGB565:

  172.     case PIX_FMT_YUV422:

  173.         picture->data[0] = ptr;

  174.         picture->data[1] = NULL;

  175.         picture->data[2] = NULL;

  176.         picture->linesize[0] = width * 2;

  177.         return size * 2;

  178.     case PIX_FMT_GRAY8:

  179.         picture->data[0] = ptr;

  180.         picture->data[1] = NULL;

  181.         picture->data[2] = NULL;

  182.         picture->linesize[0] = width;

  183.         return size;

  184.     case PIX_FMT_MONOWHITE:

  185.     case PIX_FMT_MONOBLACK:

  186.         picture->data[0] = ptr;

  187.         picture->data[1] = NULL;

  188.         picture->data[2] = NULL;

  189.         picture->linesize[0] = (width + 7) >> 3;

  190.         return picture->linesize[0] * height;

  191.     default:

  192.         picture->data[0] = NULL;

  193.         picture->data[1] = NULL;

  194.         picture->data[2] = NULL;

  195.         return -1;

  196.     }

  197. }

  198. 

  199. int avpicture_get_size(int pix_fmt, int width, int height)

  200. {

  201.     AVPicture dummy_pict;

  202.     return avpicture_fill(&dummy_pict, NULL, pix_fmt, width, height);

  203. }

  204. 

  205. 

  206. /* XXX: totally non optimized */

  207. 

  208. static void yuv422_to_yuv420p(AVPicture *dst, AVPicture *src,

  209.                               int width, int height)

  210. {

  211.     UINT8 *lum, *cb, *cr;

  212.     int x, y;

  213.     const UINT8 *p;

  214.  

  215.     lum = dst->data[0];

  216.     cb = dst->data[1];

  217.     cr = dst->data[2];

  218.     p = src->data[0];

  219.    

  220.     for(y=0;y<height;y+=2) {

  221.         for(x=0;x<width;x+=2) {

  222.             lum[0] = p[0];

  223.             cb[0] = p[1];

  224.             lum[1] = p[2];

  225.             cr[0] = p[3];

  226.             p += 4;

  227.             lum += 2;

  228.             cb++;

  229.             cr++;

  230.         }

  231.         for(x=0;x<width;x+=2) {

  232.             lum[0] = p[0];

  233.             lum[1] = p[2];

  234.             p += 4;

  235.             lum += 2;

  236.         }

  237.     }

  238. }

  239. 

  240. #define SCALEBITS 8

  241. #define ONE_HALF  (1 << (SCALEBITS - 1))

  242. #define FIX(x)		((int) ((x) * (1L<<SCALEBITS) + 0.5))

  243. 

  244. /* XXX: use generic filter ? */

  245. /* 1x2 -> 1x1 */

  246. static void shrink2(UINT8 *dst, int dst_wrap, 

  247.                     UINT8 *src, int src_wrap,

  248.                     int width, int height)

  249. {

  250.     int w;

  251.     UINT8 *s1, *s2, *d;

  252. 

  253.     for(;height > 0; height--) {

  254.         s1 = src;

  255.         s2 = s1 + src_wrap;

  256.         d = dst;

  257.         for(w = width;w >= 4; w-=4) {

  258.             d[0] = (s1[0] + s2[0]) >> 1;

  259.             d[1] = (s1[1] + s2[1]) >> 1;

  260.             d[2] = (s1[2] + s2[2]) >> 1;

  261.             d[3] = (s1[3] + s2[3]) >> 1;

  262.             s1 += 4;

  263.             s2 += 4;

  264.             d += 4;

  265.         }

  266.         for(;w > 0; w--) {

  267.             d[0] = (s1[0] + s2[0]) >> 1;

  268.             s1++;

  269.             s2++;

  270.             d++;

  271.         }

  272.         src += 2 * src_wrap;

  273.         dst += dst_wrap;

  274.     }

  275. }

  276. 

  277. /* 2x2 -> 1x1 */

  278. static void shrink22(UINT8 *dst, int dst_wrap, 

  279.                      UINT8 *src, int src_wrap,

  280.                      int width, int height)

  281. {

  282.     int w;

  283.     UINT8 *s1, *s2, *d;

  284. 

  285.     for(;height > 0; height--) {

  286.         s1 = src;

  287.         s2 = s1 + src_wrap;

  288.         d = dst;

  289.         for(w = width;w >= 4; w-=4) {

  290.             d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 1;

  291.             d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 1;

  292.             d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 1;

  293.             d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 1;

  294.             s1 += 8;

  295.             s2 += 8;

  296.             d += 4;

  297.         }

  298.         for(;w > 0; w--) {

  299.             d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 1;

  300.             s1 += 2;

  301.             s2 += 2;

  302.             d++;

  303.         }

  304.         src += 2 * src_wrap;

  305.         dst += dst_wrap;

  306.     }

  307. }

  308. 

  309. /* 1x1 -> 2x2 */

  310. static void grow22(UINT8 *dst, int dst_wrap,

  311.                      UINT8 *src, int src_wrap,

  312.                      int width, int height)

  313. {

  314.     int w;

  315.     UINT8 *s1, *d;

  316. 

  317.     for(;height > 0; height--) {

  318.         s1 = src;

  319.         d = dst;

  320.         for(w = width;w >= 4; w-=4) {

  321.             d[1] = d[0] = s1[0];

  322.             d[3] = d[2] = s1[1];

  323.             s1 += 2;

  324.             d += 4;

  325.         }

  326.         for(;w > 0; w--) {

  327.             d[0] = s1[0];

  328.             s1 ++;

  329.             d++;

  330.         }

  331.         if (height%2)

  332.             src += src_wrap;

  333.         dst += dst_wrap;

  334.     }

  335. }

  336. 

  337. /* 1x2 -> 2x1 */

  338. static void conv411(UINT8 *dst, int dst_wrap, 

  339.                     UINT8 *src, int src_wrap,

  340.                     int width, int height)

  341. {

  342.     int w, c;

  343.     UINT8 *s1, *s2, *d;

  344. 

  345.     for(;height > 0; height--) {

  346.         s1 = src;

  347.         s2 = src + src_wrap;

  348.         d = dst;

  349.         for(w = width;w > 0; w--) {

  350.             c = (s1[0] + s2[0]) >> 1;

  351.             d[0] = c;

  352.             d[1] = c;

  353.             s1++;

  354.             s2++;

  355.             d += 2;

  356.         }

  357.         src += src_wrap * 2;

  358.         dst += dst_wrap;

  359.     }

  360. }

  361. 

  362. static void img_copy(UINT8 *dst, int dst_wrap, 

  363.                      UINT8 *src, int src_wrap,

  364.                      int width, int height)

  365. {

  366.     for(;height > 0; height--) {

  367.         memcpy(dst, src, width);

  368.         dst += dst_wrap;

  369.         src += src_wrap;

  370.     }

  371. }

  372. 

  373. #define SCALE_BITS 10

  374. 

  375. #define C_Y  (76309 >> (16 - SCALE_BITS))

  376. #define C_RV (117504 >> (16 - SCALE_BITS))

  377. #define C_BU (138453 >> (16 - SCALE_BITS))

  378. #define C_GU (13954 >> (16 - SCALE_BITS))

  379. #define C_GV (34903 >> (16 - SCALE_BITS))

  380. 

  381. #define YUV_TO_RGB2(r, g, b, y1)\

  382. {\

  383.     y = (y1 - 16) * C_Y;\

  384.     r = cm[(y + r_add) >> SCALE_BITS];\

  385.     g = cm[(y + g_add) >> SCALE_BITS];\

  386.     b = cm[(y + b_add) >> SCALE_BITS];\

  387. }

  388. 

  389. /* XXX: no chroma interpolating is done */

  390. #define RGB_FUNCTIONS(rgb_name)                                         \

  391.                                                                         \

  392. static void yuv420p_to_ ## rgb_name (AVPicture *dst, AVPicture *src,    \

  393.                                      int width, int height)             \

  394. {                                                                       \

  395.     UINT8 *y1_ptr, *y2_ptr, *cb_ptr, *cr_ptr, *d, *d1, *d2;             \

  396.     int w, y, cb, cr, r_add, g_add, b_add, width2;                      \

  397.     UINT8 *cm = cropTbl + MAX_NEG_CROP;                                 \

  398.     unsigned int r, g, b;                                               \

  399.                                                                         \

  400.     d = dst->data[0];                                                   \

  401.     y1_ptr = src->data[0];                                              \

  402.     cb_ptr = src->data[1];                                              \

  403.     cr_ptr = src->data[2];                                              \

  404.     width2 = (width + 1) >> 1;                                          \

  405.     for(;height >= 2; height -= 2) {                                    \

  406.         d1 = d;                                                         \

  407.         d2 = d + dst->linesize[0];                                      \

  408.         y2_ptr = y1_ptr + src->linesize[0];                             \

  409.         for(w = width; w >= 2; w -= 2) {                                \

  410.             cb = cb_ptr[0] - 128;                                       \

  411.             cr = cr_ptr[0] - 128;                                       \

  412.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  413.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  414.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  415.                                                                         \

  416.             /* output 4 pixels */                                       \

  417.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  418.             RGB_OUT(d1, r, g, b);                                       \

  419.                                                                         \

  420.             YUV_TO_RGB2(r, g, b, y1_ptr[1]);                            \

  421.             RGB_OUT(d1 + BPP, r, g, b);                                 \

  422.                                                                         \

  423.             YUV_TO_RGB2(r, g, b, y2_ptr[0]);                            \

  424.             RGB_OUT(d2, r, g, b);                                       \

  425.                                                                         \

  426.             YUV_TO_RGB2(r, g, b, y2_ptr[1]);                            \

  427.             RGB_OUT(d2 + BPP, r, g, b);                                 \

  428.                                                                         \

  429.             d1 += 2 * BPP;                                              \

  430.             d2 += 2 * BPP;                                              \

  431.                                                                         \

  432.             y1_ptr += 2;                                                \

  433.             y2_ptr += 2;                                                \

  434.             cb_ptr++;                                                   \

  435.             cr_ptr++;                                                   \

  436.         }                                                               \

  437.         /* handle odd width */                                          \

  438.         if (w) {                                                        \

  439.             cb = cb_ptr[0] - 128;                                       \

  440.             cr = cr_ptr[0] - 128;                                       \

  441.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  442.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  443.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  444.                                                                         \

  445.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  446.             RGB_OUT(d1, r, g, b);                                       \

  447.                                                                         \

  448.             YUV_TO_RGB2(r, g, b, y2_ptr[0]);                            \

  449.             RGB_OUT(d2, r, g, b);                                       \

  450.             d1 += BPP;                                                  \

  451.             d2 += BPP;                                                  \

  452.             y1_ptr++;                                                   \

  453.             y2_ptr++;                                                   \

  454.             cb_ptr++;                                                   \

  455.             cr_ptr++;                                                   \

  456.         }                                                               \

  457.         d += 2 * dst->linesize[0];                                      \

  458.         y1_ptr += 2 * src->linesize[0] - width;                         \

  459.         cb_ptr += src->linesize[1] - width2;                            \

  460.         cr_ptr += src->linesize[2] - width2;                            \

  461.     }                                                                   \

  462.     /* handle odd height */                                             \

  463.     if (height) {                                                       \

  464.         d1 = d;                                                         \

  465.         for(w = width; w >= 2; w -= 2) {                                \

  466.             cb = cb_ptr[0] - 128;                                       \

  467.             cr = cr_ptr[0] - 128;                                       \

  468.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  469.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  470.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  471.                                                                         \

  472.             /* output 2 pixels */                                       \

  473.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  474.             RGB_OUT(d1, r, g, b);                                       \

  475.                                                                         \

  476.             YUV_TO_RGB2(r, g, b, y1_ptr[1]);                            \

  477.             RGB_OUT(d1 + BPP, r, g, b);                                 \

  478.                                                                         \

  479.             d1 += 2 * BPP;                                              \

  480.                                                                         \

  481.             y1_ptr += 2;                                                \

  482.             cb_ptr++;                                                   \

  483.             cr_ptr++;                                                   \

  484.         }                                                               \

  485.         /* handle width */                                              \

  486.         if (w) {                                                        \

  487.             cb = cb_ptr[0] - 128;                                       \

  488.             cr = cr_ptr[0] - 128;                                       \

  489.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  490.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  491.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  492.                                                                         \

  493.             /* output 2 pixels */                                       \

  494.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  495.             RGB_OUT(d1, r, g, b);                                       \

  496.             d1 += BPP;                                                  \

  497.                                                                         \

  498.             y1_ptr++;                                                   \

  499.             cb_ptr++;                                                   \

  500.             cr_ptr++;                                                   \

  501.         }                                                               \

  502.     }                                                                   \

  503. }                                                                       \

  504.                                                                         \

  505. /* XXX: no chroma interpolating is done */                              \

  506. static void yuv422p_to_ ## rgb_name (AVPicture *dst, AVPicture *src,    \

  507.                                     int width, int height)              \

  508. {                                                                       \

  509.     UINT8 *y1_ptr, *cb_ptr, *cr_ptr, *d, *d1;                           \

  510.     int w, y, cb, cr, r_add, g_add, b_add, width2;                      \

  511.     UINT8 *cm = cropTbl + MAX_NEG_CROP;                                 \

  512.     unsigned int r, g, b;                                               \

  513.                                                                         \

  514.     d = dst->data[0];                                                   \

  515.     y1_ptr = src->data[0];                                              \

  516.     cb_ptr = src->data[1];                                              \

  517.     cr_ptr = src->data[2];                                              \

  518.     width2 = (width + 1) >> 1;                                          \

  519.     for(;height > 0; height --) {                                       \

  520.         d1 = d;                                                         \

  521.         for(w = width; w >= 2; w -= 2) {                                \

  522.             cb = cb_ptr[0] - 128;                                       \

  523.             cr = cr_ptr[0] - 128;                                       \

  524.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  525.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  526.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  527.                                                                         \

  528.             /* output 2 pixels */                                       \

  529.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  530.             RGB_OUT(d1, r, g, b);                                       \

  531.                                                                         \

  532.             YUV_TO_RGB2(r, g, b, y1_ptr[1]);                            \

  533.             RGB_OUT(d1 + BPP, r, g, b);                                 \

  534.                                                                         \

  535.             d1 += 2 * BPP;                                              \

  536.                                                                         \

  537.             y1_ptr += 2;                                                \

  538.             cb_ptr++;                                                   \

  539.             cr_ptr++;                                                   \

  540.         }                                                               \

  541.         /* handle width */                                              \

  542.         if (w) {                                                        \

  543.             cb = cb_ptr[0] - 128;                                       \

  544.             cr = cr_ptr[0] - 128;                                       \

  545.             r_add = C_RV * cr + (1 << (SCALE_BITS - 1));                \

  546.             g_add = - C_GU * cb - C_GV * cr + (1 << (SCALE_BITS - 1));  \

  547.             b_add = C_BU * cb + (1 << (SCALE_BITS - 1));                \

  548.                                                                         \

  549.             /* output 2 pixels */                                       \

  550.             YUV_TO_RGB2(r, g, b, y1_ptr[0]);                            \

  551.             RGB_OUT(d1, r, g, b);                                       \

  552.             d1 += BPP;                                                  \

  553.                                                                         \

  554.             y1_ptr++;                                                   \

  555.             cb_ptr++;                                                   \

  556.             cr_ptr++;                                                   \

  557.         }                                                               \

  558.         d += dst->linesize[0];                                          \

  559.         y1_ptr += src->linesize[0] - width;                             \

  560.         cb_ptr += src->linesize[1] - width2;                            \

  561.         cr_ptr += src->linesize[2] - width2;                            \

  562.     }                                                                   \

  563. }                                                                       \

  564.                                                                         \

  565. static void rgb_name ## _to_yuv420p(AVPicture *dst, AVPicture *src,     \

  566.                                     int width, int height)              \

  567. {                                                                       \

  568.     int wrap, wrap3, x, y;                                              \

  569.     int r, g, b, r1, g1, b1;                                            \

  570.     UINT8 *lum, *cb, *cr;                                               \

  571.     const UINT8 *p;                                                     \

  572.                                                                         \

  573.     lum = dst->data[0];                                                 \

  574.     cb = dst->data[1];                                                  \

  575.     cr = dst->data[2];                                                  \

  576.                                                                         \

  577.     wrap = width;                                                       \

  578.     wrap3 = width * BPP;                                                \

  579.     p = src->data[0];                                                   \

  580.     for(y=0;y<height;y+=2) {                                            \

  581.         for(x=0;x<width;x+=2) {                                         \

  582.             RGB_IN(r, g, b, p);                                         \

  583.             r1 = r;                                                     \

  584.             g1 = g;                                                     \

  585.             b1 = b;                                                     \

  586.             lum[0] = (FIX(0.29900) * r + FIX(0.58700) * g +             \

  587.                       FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;        \

  588.             RGB_IN(r, g, b, p + BPP);                                   \

  589.             r1 += r;                                                    \

  590.             g1 += g;                                                    \

  591.             b1 += b;                                                    \

  592.             lum[1] = (FIX(0.29900) * r + FIX(0.58700) * g +             \

  593.                       FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;        \

  594.             p += wrap3;                                                 \

  595.             lum += wrap;                                                \

  596.                                                                         \

  597.             RGB_IN(r, g, b, p);                                         \

  598.             r1 += r;                                                    \

  599.             g1 += g;                                                    \

  600.             b1 += b;                                                    \

  601.             lum[0] = (FIX(0.29900) * r + FIX(0.58700) * g +             \

  602.                       FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;        \

  603.                                                                         \

  604.             RGB_IN(r, g, b, p + BPP);                                   \

  605.             r1 += r;                                                    \

  606.             g1 += g;                                                    \

  607.             b1 += b;                                                    \

  608.             lum[1] = (FIX(0.29900) * r + FIX(0.58700) * g +             \

  609.                       FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;        \

  610.                                                                         \

  611.             cb[0] = ((- FIX(0.16874) * r1 - FIX(0.33126) * g1 +         \

  612.                       FIX(0.50000) * b1 + 4 * ONE_HALF - 1) >>          \

  613.                      (SCALEBITS + 2)) + 128;                            \

  614.             cr[0] = ((FIX(0.50000) * r1 - FIX(0.41869) * g1 -           \

  615.                      FIX(0.08131) * b1 + 4 * ONE_HALF - 1) >>           \

  616.                      (SCALEBITS + 2)) + 128;                            \

  617.                                                                         \

  618.             cb++;                                                       \

  619.             cr++;                                                       \

  620.             p += -wrap3 + 2 * BPP;                                      \

  621.             lum += -wrap + 2;                                           \

  622.         }                                                               \

  623.         p += wrap3;                                                     \

  624.         lum += wrap;                                                    \

  625.     }                                                                   \

  626. }                                                                       \

  627.                                                                         \

  628. static void rgb_name ## _to_gray(AVPicture *dst, AVPicture *src,        \

  629.                                  int width, int height)                 \

  630. {                                                                       \

  631.     const unsigned char *p;                                             \

  632.     unsigned char *q;                                                   \

  633.     int r, g, b, dst_wrap, src_wrap;                                    \

  634.     int x, y;                                                           \

  635.                                                                         \

  636.     p = src->data[0];                                                   \

  637.     src_wrap = src->linesize[0] - BPP * width;                          \

  638.                                                                         \

  639.     q = dst->data[0];                                                   \

  640.     dst_wrap = dst->linesize[0] - width;                                \

  641.                                                                         \

  642.     for(y=0;y<height;y++) {                                             \

  643.         for(x=0;x<width;x++) {                                          \

  644.             RGB_IN(r, g, b, p);                                         \

  645.             q[0] = (FIX(0.29900) * r + FIX(0.58700) * g +               \

  646.                     FIX(0.11400) * b + ONE_HALF) >> SCALEBITS;          \

  647.             q++;                                                        \

  648.             p += BPP;                                                   \

  649.         }                                                               \

  650.         p += src_wrap;                                                  \

  651.         q += dst_wrap;                                                  \

  652.     }                                                                   \

  653. }                                                                       \

  654.                                                                         \

  655. static void gray_to_ ## rgb_name(AVPicture *dst, AVPicture *src,        \

  656.                                  int width, int height)                 \

  657. {                                                                       \

  658.     const unsigned char *p;                                             \

  659.     unsigned char *q;                                                   \

  660.     int r, dst_wrap, src_wrap;                                          \

  661.     int x, y;                                                           \

  662.                                                                         \

  663.     p = src->data[0];                                                   \

  664.     src_wrap = src->linesize[0] - width;                                \

  665.                                                                         \

  666.     q = dst->data[0];                                                   \

  667.     dst_wrap = dst->linesize[0] - BPP * width;                          \

  668.                                                                         \

  669.     for(y=0;y<height;y++) {                                             \

  670.         for(x=0;x<width;x++) {                                          \

  671.             r = p[0];                                                   \

  672.             RGB_OUT(q, r, r, r);                                        \

  673.             q += BPP;                                                   \

  674.             p ++;                                                       \

  675.         }                                                               \

  676.         p += src_wrap;                                                  \

  677.         q += dst_wrap;                                                  \

  678.     }                                                                   \

  679. }

  680. 

  681. /* copy bit n to bits 0 ... n - 1 */

  682. static inline unsigned int bitcopy_n(unsigned int a, int n)

  683. {

  684.     int mask;

  685.     mask = (1 << n) - 1;

  686.     return (a & (0xff & ~mask)) | ((-((a >> n) & 1)) & mask);

  687. }

  688. 

  689. /* rgb555 handling */

  690. 

  691. #define RGB_IN(r, g, b, s)\

  692. {\

  693.     unsigned int v = ((UINT16 *)(s))[0];\

  694.     r = bitcopy_n(v >> (10 - 3), 3);\

  695.     g = bitcopy_n(v >> (5 - 3), 3);\

  696.     b = bitcopy_n(v << 3, 3);\

  697. }

  698. 

  699. #define RGB_OUT(d, r, g, b)\

  700. {\

  701.     ((UINT16 *)(d))[0] = ((r >> 3) << 10) | ((g >> 3) << 5) | (b >> 3) | 0x8000;\

  702. }

  703. 

  704. #define BPP 2

  705. 

  706. RGB_FUNCTIONS(rgb555)

  707. 

  708. #undef RGB_IN

  709. #undef RGB_OUT

  710. #undef BPP

  711. 

  712. /* rgb565 handling */

  713. 

  714. #define RGB_IN(r, g, b, s)\

  715. {\

  716.     unsigned int v = ((UINT16 *)(s))[0];\

  717.     r = bitcopy_n(v >> (11 - 3), 3);\

  718.     g = bitcopy_n(v >> (5 - 2), 2);\

  719.     b = bitcopy_n(v << 3, 3);\

  720. }

  721. 

  722. #define RGB_OUT(d, r, g, b)\

  723. {\

  724.     ((UINT16 *)(d))[0] = ((r >> 3) << 11) | ((g >> 2) << 5) | (b >> 3);\

  725. }

  726. 

  727. #define BPP 2

  728. 

  729. RGB_FUNCTIONS(rgb565)

  730. 

  731. #undef RGB_IN

  732. #undef RGB_OUT

  733. #undef BPP

  734. 

  735. /* bgr24 handling */

  736. 

  737. #define RGB_IN(r, g, b, s)\

  738. {\

  739.     b = (s)[0];\

  740.     g = (s)[1];\

  741.     r = (s)[2];\

  742. }

  743. 

  744. #define RGB_OUT(d, r, g, b)\

  745. {\

  746.     (d)[0] = b;\

  747.     (d)[1] = g;\

  748.     (d)[2] = r;\

  749. }

  750. 

  751. #define BPP 3

  752. 

  753. RGB_FUNCTIONS(bgr24)

  754. 

  755. #undef RGB_IN

  756. #undef RGB_OUT

  757. #undef BPP

  758. 

  759. /* rgb24 handling */

  760. 

  761. #define RGB_IN(r, g, b, s)\

  762. {\

  763.     r = (s)[0];\

  764.     g = (s)[1];\

  765.     b = (s)[2];\

  766. }

  767. 

  768. #define RGB_OUT(d, r, g, b)\

  769. {\

  770.     (d)[0] = r;\

  771.     (d)[1] = g;\

  772.     (d)[2] = b;\

  773. }

  774. 

  775. #define BPP 3

  776. 

  777. RGB_FUNCTIONS(rgb24)

  778. 

  779. #undef RGB_IN

  780. #undef RGB_OUT

  781. #undef BPP

  782. 

  783. /* rgba32 handling */

  784. 

  785. #define RGB_IN(r, g, b, s)\

  786. {\

  787.     unsigned int v = ((UINT32 *)(s))[0];\

  788.     r = (v >> 16) & 0xff;\

  789.     g = (v >> 8) & 0xff;\

  790.     b = v & 0xff;\

  791. }

  792. 

  793. #define RGB_OUT(d, r, g, b)\

  794. {\

  795.     ((UINT32 *)(d))[0] = (0xff << 24) | (r << 16) | (g << 8) | b;\

  796. }

  797. 

  798. #define BPP 4

  799. 

  800. RGB_FUNCTIONS(rgba32)

  801. 

  802. #undef RGB_IN

  803. #undef RGB_OUT

  804. #undef BPP

  805. 

  806. 

  807. static void rgb24_to_rgb565(AVPicture *dst, AVPicture *src,

  808.                             int width, int height)

  809. {

  810.     const unsigned char *p;

  811.     unsigned char *q;

  812.     int r, g, b, dst_wrap, src_wrap;

  813.     int x, y;

  814. 

  815.     p = src->data[0];

  816.     src_wrap = src->linesize[0] - 3 * width;

  817. 

  818.     q = dst->data[0];

  819.     dst_wrap = dst->linesize[0] - 2 * width;

  820. 

  821.     for(y=0;y<height;y++) {

  822.         for(x=0;x<width;x++) {

  823.             r = p[0];

  824.             g = p[1];

  825.             b = p[2];

  826. 

  827.             ((unsigned short *)q)[0] = 

  828.                 ((r >> 3) << 11) | ((g >> 2) << 5) | (b >> 3);

  829.             q += 2;

  830.             p += 3;

  831.         }

  832.         p += src_wrap;

  833.         q += dst_wrap;

  834.     }

  835. }

  836. 

  837. /* NOTE: we also add a dummy alpha bit */

  838. static void rgb24_to_rgb555(AVPicture *dst, AVPicture *src,

  839.                             int width, int height)

  840. {

  841.     const unsigned char *p;

  842.     unsigned char *q;

  843.     int r, g, b, dst_wrap, src_wrap;

  844.     int x, y;

  845. 

  846.     p = src->data[0];

  847.     src_wrap = src->linesize[0] - 3 * width;

  848. 

  849.     q = dst->data[0];

  850.     dst_wrap = dst->linesize[0] - 2 * width;

  851. 

  852.     for(y=0;y<height;y++) {

  853.         for(x=0;x<width;x++) {

  854.             r = p[0];

  855.             g = p[1];

  856.             b = p[2];

  857. 

  858.             ((unsigned short *)q)[0] = 

  859.                 ((r >> 3) << 10) | ((g >> 3) << 5) | (b >> 3) | 0x8000;

  860.             q += 2;

  861.             p += 3;

  862.         }

  863.         p += src_wrap;

  864.         q += dst_wrap;

  865.     }

  866. }

  867. 

  868. static void mono_to_gray(AVPicture *dst, AVPicture *src,

  869.                          int width, int height, int xor_mask)

  870. {

  871.     const unsigned char *p;

  872.     unsigned char *q;

  873.     int v, dst_wrap, src_wrap;

  874.     int y, w;

  875. 

  876.     p = src->data[0];

  877.     src_wrap = src->linesize[0] - ((width + 7) >> 3);

  878. 

  879.     q = dst->data[0];

  880.     dst_wrap = dst->linesize[0] - width;

  881.     for(y=0;y<height;y++) {

  882.         w = width; 

  883.         while (w >= 8) {

  884.             v = *p++ ^ xor_mask;

  885.             q[0] = -(v >> 7);

  886.             q[1] = -((v >> 6) & 1);

  887.             q[2] = -((v >> 5) & 1);

  888.             q[3] = -((v >> 4) & 1);

  889.             q[4] = -((v >> 3) & 1);

  890.             q[5] = -((v >> 2) & 1);

  891.             q[6] = -((v >> 1) & 1);

  892.             q[7] = -((v >> 0) & 1);

  893.             w -= 8;

  894.             q += 8;

  895.         }

  896.         if (w > 0) {

  897.             v = *p++ ^ xor_mask;

  898.             do {

  899.                 q[0] = -((v >> 7) & 1);

  900.                 q++;

  901.                 v <<= 1;

  902.             } while (--w);

  903.         }

  904.         p += src_wrap;

  905.         q += dst_wrap;

  906.     }

  907. }

  908. 

  909. static void monowhite_to_gray(AVPicture *dst, AVPicture *src,

  910.                                int width, int height)

  911. {

  912.     mono_to_gray(dst, src, width, height, 0xff);

  913. }

  914. 

  915. static void monoblack_to_gray(AVPicture *dst, AVPicture *src,

  916.                                int width, int height)

  917. {

  918.     mono_to_gray(dst, src, width, height, 0x00);

  919. }

  920. 

  921. static void gray_to_mono(AVPicture *dst, AVPicture *src,

  922.                          int width, int height, int xor_mask)

  923. {

  924.     int n;

  925.     const UINT8 *s;

  926.     UINT8 *d;

  927.     int j, b, v, n1, src_wrap, dst_wrap, y;

  928. 

  929.     s = src->data[0];

  930.     src_wrap = src->linesize[0] - width;

  931. 

  932.     d = dst->data[0];

  933.     dst_wrap = dst->linesize[0] - ((width + 7) >> 3);

  934.     printf("%d %d\n", width, height);

  935. 

  936.     for(y=0;y<height;y++) {

  937.         n = width;

  938.         while (n >= 8) {

  939.             v = 0;

  940.             for(j=0;j<8;j++) {

  941.                 b = s[0];

  942.                 s++;

  943.                 v = (v << 1) | (b >> 7);

  944.             }

  945.             d[0] = v ^ xor_mask;

  946.             d++;

  947.             n -= 8;

  948.         }

  949.         if (n > 0) {

  950.             n1 = n;

  951.             v = 0;

  952.             while (n > 0) {

  953.                 b = s[0];

  954.                 s++;

  955.                 v = (v << 1) | (b >> 7);

  956.                 n--;

  957.             }

  958.             d[0] = (v << (8 - (n1 & 7))) ^ xor_mask;

  959.             d++;

  960.         }

  961.         s += src_wrap;

  962.         d += dst_wrap;

  963.     }

  964. }

  965. 

  966. static void gray_to_monowhite(AVPicture *dst, AVPicture *src,

  967.                               int width, int height)

  968. {

  969.     gray_to_mono(dst, src, width, height, 0xff);

  970. }

  971. 

  972. static void gray_to_monoblack(AVPicture *dst, AVPicture *src,

  973.                               int width, int height)

  974. {

  975.     gray_to_mono(dst, src, width, height, 0x00);

  976. }

  977. 

  978. typedef struct ConvertEntry {

  979.     void (*convert)(AVPicture *dst, AVPicture *src, int width, int height);

  980. } ConvertEntry;

  981. 

  982. /* add each new convertion function in this table */

  983. /* constraints;

  984.    - all non YUV modes must convert at least to and from PIX_FMT_RGB24

  985. */

  986. static ConvertEntry convert_table[PIX_FMT_NB][PIX_FMT_NB] = {

  987.     [PIX_FMT_YUV420P] = {

  988.         [PIX_FMT_RGB555] = { 

  989.             .convert = yuv420p_to_rgb555

  990.         },

  991.         [PIX_FMT_RGB565] = { 

  992.             .convert = yuv420p_to_rgb565

  993.         },

  994.         [PIX_FMT_BGR24] = { 

  995.             .convert = yuv420p_to_bgr24

  996.         },

  997.         [PIX_FMT_RGB24] = { 

  998.             .convert = yuv420p_to_rgb24

  999.         },

  1000.         [PIX_FMT_RGBA32] = { 

  1001.             .convert = yuv420p_to_rgba32

  1002.         },

  1003.     },

  1004.     [PIX_FMT_YUV422P] = {

  1005.         [PIX_FMT_RGB555] = { 

  1006.             .convert = yuv422p_to_rgb555

  1007.         },

  1008.         [PIX_FMT_RGB565] = { 

  1009.             .convert = yuv422p_to_rgb565

  1010.         },

  1011.         [PIX_FMT_BGR24] = { 

  1012.             .convert = yuv422p_to_bgr24

  1013.         },

  1014.         [PIX_FMT_RGB24] = { 

  1015.             .convert = yuv422p_to_rgb24

  1016.         },

  1017.         [PIX_FMT_RGBA32] = { 

  1018.             .convert = yuv422p_to_rgba32

  1019.         },

  1020.     },

  1021.     [PIX_FMT_YUV422] = { 

  1022.         [PIX_FMT_YUV420P] = { 

  1023.             .convert = yuv422_to_yuv420p,

  1024.         },

  1025.     },

  1026. 

  1027.     [PIX_FMT_RGB24] = {

  1028.         [PIX_FMT_YUV420P] = { 

  1029.             .convert = rgb24_to_yuv420p

  1030.         },

  1031.         [PIX_FMT_RGB565] = { 

  1032.             .convert = rgb24_to_rgb565

  1033.         },

  1034.         [PIX_FMT_RGB555] = { 

  1035.             .convert = rgb24_to_rgb555

  1036.         },

  1037.         [PIX_FMT_GRAY8] = { 

  1038.             .convert = rgb24_to_gray

  1039.         },

  1040.     },

  1041.     [PIX_FMT_RGBA32] = {

  1042.         [PIX_FMT_YUV420P] = { 

  1043.             .convert = rgba32_to_yuv420p

  1044.         },

  1045.         [PIX_FMT_GRAY8] = { 

  1046.             .convert = rgba32_to_gray

  1047.         },

  1048.     },

  1049.     [PIX_FMT_BGR24] = {

  1050.         [PIX_FMT_YUV420P] = { 

  1051.             .convert = bgr24_to_yuv420p

  1052.         },

  1053.         [PIX_FMT_GRAY8] = { 

  1054.             .convert = bgr24_to_gray

  1055.         },

  1056.     },

  1057.     [PIX_FMT_RGB555] = {

  1058.         [PIX_FMT_YUV420P] = { 

  1059.             .convert = rgb555_to_yuv420p

  1060.         },

  1061.         [PIX_FMT_GRAY8] = { 

  1062.             .convert = rgb555_to_gray

  1063.         },

  1064.     },

  1065.     [PIX_FMT_RGB565] = {

  1066.         [PIX_FMT_YUV420P] = { 

  1067.             .convert = rgb565_to_yuv420p

  1068.         },

  1069.         [PIX_FMT_GRAY8] = { 

  1070.             .convert = rgb565_to_gray

  1071.         },

  1072.     },

  1073.     [PIX_FMT_GRAY8] = {

  1074.         [PIX_FMT_RGB555] = { 

  1075.             .convert = gray_to_rgb555

  1076.         },

  1077.         [PIX_FMT_RGB565] = { 

  1078.             .convert = gray_to_rgb565

  1079.         },

  1080.         [PIX_FMT_RGB24] = { 

  1081.             .convert = gray_to_rgb24

  1082.         },

  1083.         [PIX_FMT_BGR24] = { 

  1084.             .convert = gray_to_bgr24

  1085.         },

  1086.         [PIX_FMT_RGBA32] = { 

  1087.             .convert = gray_to_rgba32

  1088.         },

  1089.         [PIX_FMT_MONOWHITE] = { 

  1090.             .convert = gray_to_monowhite

  1091.         },

  1092.         [PIX_FMT_MONOBLACK] = { 

  1093.             .convert = gray_to_monoblack

  1094.         },

  1095.     },

  1096.     [PIX_FMT_MONOWHITE] = {

  1097.         [PIX_FMT_GRAY8] = { 

  1098.             .convert = monowhite_to_gray

  1099.         },

  1100.     },

  1101.     [PIX_FMT_MONOBLACK] = {

  1102.         [PIX_FMT_GRAY8] = { 

  1103.             .convert = monoblack_to_gray

  1104.         },

  1105.     },

  1106. };

  1107. 

  1108. static int avpicture_alloc(AVPicture *picture,

  1109.                            int pix_fmt, int width, int height)

  1110. {

  1111.     int size;

  1112.     void *ptr;

  1113. 

  1114.     size = avpicture_get_size(pix_fmt, width, height);

  1115.     if (size < 0)

  1116.         goto fail;

  1117.     ptr = av_malloc(size);

  1118.     if (!ptr)

  1119.         goto fail;

  1120.     avpicture_fill(picture, ptr, pix_fmt, width, height);

  1121.     return 0;

  1122.  fail:

  1123.     memset(picture, 0, sizeof(AVPicture));

  1124.     return -1;

  1125. }

  1126. 

  1127. static void avpicture_free(AVPicture *picture)

  1128. {

  1129.     av_free(picture->data[0]);

  1130. }

  1131. 

  1132. /* XXX: always use linesize. Return -1 if not supported */

  1133. int img_convert(AVPicture *dst, int dst_pix_fmt,

  1134.                 AVPicture *src, int src_pix_fmt, 

  1135.                 int src_width, int src_height)

  1136. {

  1137.     int i, ret, dst_width, dst_height, int_pix_fmt;

  1138.     PixFmtInfo *src_pix, *dst_pix;

  1139.     ConvertEntry *ce;

  1140.     AVPicture tmp1, *tmp = &tmp1;

  1141. 

  1142.     if (src_pix_fmt < 0 || src_pix_fmt >= PIX_FMT_NB ||

  1143.         dst_pix_fmt < 0 || dst_pix_fmt >= PIX_FMT_NB)

  1144.         return -1;

  1145.     if (src_width <= 0 || src_height <= 0)

  1146.         return 0;

  1147. 

  1148.     dst_width = src_width;

  1149.     dst_height = src_height;

  1150. 

  1151.     dst_pix = &pix_fmt_info[dst_pix_fmt];

  1152.     src_pix = &pix_fmt_info[src_pix_fmt];

  1153.     if (src_pix_fmt == dst_pix_fmt) {

  1154.         /* XXX: incorrect */

  1155.         /* same format: just copy */

  1156.         for(i = 0; i < dst_pix->nb_components; i++) {

  1157.             int w, h;

  1158.             w = dst_width;

  1159.             h = dst_height;

  1160.             if (dst_pix->is_yuv && (i == 1 || i == 2)) {

  1161.                 w >>= dst_pix->x_chroma_shift;

  1162.                 h >>= dst_pix->y_chroma_shift;

  1163.             }

  1164.             img_copy(dst->data[i], dst->linesize[i],

  1165.                      src->data[i], src->linesize[i],

  1166.                      w, h);

  1167.         }

  1168.         return 0;

  1169.     }

  1170. 

  1171.     ce = &convert_table[src_pix_fmt][dst_pix_fmt];

  1172.     if (ce->convert) {

  1173.         /* specific convertion routine */

  1174.         ce->convert(dst, src, dst_width, dst_height);

  1175.         return 0;

  1176.     }

  1177. 

  1178.     /* gray to YUV */

  1179.     if (dst_pix->is_yuv && src_pix_fmt == PIX_FMT_GRAY8) {

  1180.         int w, h, y;

  1181.         uint8_t *d;

  1182. 

  1183.         img_copy(dst->data[0], dst->linesize[0],

  1184.                  src->data[0], src->linesize[0],

  1185.                  dst_width, dst_height);

  1186.         /* fill U and V with 128 */

  1187.         w = dst_width;

  1188.         h = dst_height;

  1189.         w >>= dst_pix->x_chroma_shift;

  1190.         h >>= dst_pix->y_chroma_shift;

  1191.         for(i = 1; i <= 2; i++) {

  1192.             d = dst->data[i];

  1193.             for(y = 0; y< h; y++) {

  1194.                 memset(d, 128, w);

  1195.                 d += dst->linesize[i];

  1196.             }

  1197.         }

  1198.         return 0;

  1199.     }

  1200. 

  1201.     /* YUV to gray */

  1202.     if (src_pix->is_yuv && dst_pix_fmt == PIX_FMT_GRAY8) {

  1203.         img_copy(dst->data[0], dst->linesize[0],

  1204.                  src->data[0], src->linesize[0],

  1205.                  dst_width, dst_height);

  1206.         return 0;

  1207.     }

  1208. 

  1209.     /* YUV to YUV */

  1210.     if (dst_pix->is_yuv && src_pix->is_yuv) {

  1211.         int x_shift, y_shift, w, h;

  1212.         void (*resize_func)(UINT8 *dst, int dst_wrap, 

  1213.                             UINT8 *src, int src_wrap,

  1214.                             int width, int height);

  1215. 

  1216.         /* compute chroma size of the smallest dimensions */

  1217.         w = dst_width;

  1218.         h = dst_height;

  1219.         if (dst_pix->x_chroma_shift >= src_pix->x_chroma_shift)

  1220.             w >>= dst_pix->x_chroma_shift;

  1221.         else

  1222.             w >>= src_pix->x_chroma_shift;

  1223.         if (dst_pix->y_chroma_shift >= src_pix->y_chroma_shift)

  1224.             h >>= dst_pix->y_chroma_shift;

  1225.         else

  1226.             h >>= src_pix->y_chroma_shift;

  1227. 

  1228.         x_shift = (dst_pix->x_chroma_shift - src_pix->x_chroma_shift);

  1229.         y_shift = (dst_pix->y_chroma_shift - src_pix->y_chroma_shift);

  1230.         if (x_shift == 0 && y_shift == 0) {

  1231.             resize_func = img_copy; /* should never happen */

  1232.         } else if (x_shift == 0 && y_shift == 1) {

  1233.             resize_func = shrink2;

  1234.         } else if (x_shift == 1 && y_shift == 1) {

  1235.             resize_func = shrink22;

  1236.         } else if (x_shift == -1 && y_shift == -1) {

  1237.             resize_func = grow22;

  1238.         } else if (x_shift == -1 && y_shift == 1) {

  1239.             resize_func = conv411;

  1240.         } else {

  1241.             /* currently not handled */

  1242.             return -1;

  1243.         }

  1244. 

  1245.         img_copy(dst->data[0], dst->linesize[0],

  1246.                  src->data[0], src->linesize[0],

  1247.                  dst_width, dst_height);

  1248. 

  1249.         for(i = 1;i <= 2; i++)

  1250.             resize_func(dst->data[i], dst->linesize[i],

  1251.                         src->data[i], src->linesize[i],

  1252.                         w, h);

  1253.        return 0;

  1254.     }

  1255. 

  1256.     /* try to use an intermediate format */

  1257.     if (src_pix_fmt == PIX_FMT_MONOWHITE ||

  1258.         src_pix_fmt == PIX_FMT_MONOBLACK ||

  1259.         dst_pix_fmt == PIX_FMT_MONOWHITE ||

  1260.         dst_pix_fmt == PIX_FMT_MONOBLACK) {

  1261.         int_pix_fmt = PIX_FMT_GRAY8;

  1262.     } else {

  1263.         int_pix_fmt = PIX_FMT_RGB24;

  1264.     }

  1265.     if (avpicture_alloc(tmp, int_pix_fmt, dst_width, dst_height) < 0)

  1266.         return -1;

  1267.     ret = -1;

  1268.     if (img_convert(tmp, int_pix_fmt,

  1269.                     src, src_pix_fmt, src_width, src_height) < 0)

  1270.         goto fail1;

  1271.     if (img_convert(dst, dst_pix_fmt,

  1272.                     tmp, int_pix_fmt, dst_width, dst_height) < 0)

  1273.         goto fail1;

  1274.     ret = 0;

  1275.  fail1:

  1276.     avpicture_free(tmp);

  1277.     return ret;

  1278. }

  1279. 

  1280. 

  1281. #ifdef HAVE_MMX

  1282. #define DEINT_INPLACE_LINE_LUM \

  1283.                     movd_m2r(lum_m4[0],mm0);\

  1284.                     movd_m2r(lum_m3[0],mm1);\

  1285.                     movd_m2r(lum_m2[0],mm2);\

  1286.                     movd_m2r(lum_m1[0],mm3);\

  1287.                     movd_m2r(lum[0],mm4);\

  1288.                     punpcklbw_r2r(mm7,mm0);\

  1289.                     movd_r2m(mm2,lum_m4[0]);\

  1290.                     punpcklbw_r2r(mm7,mm1);\

  1291.                     punpcklbw_r2r(mm7,mm2);\

  1292.                     punpcklbw_r2r(mm7,mm3);\

  1293.                     punpcklbw_r2r(mm7,mm4);\

  1294.                     paddw_r2r(mm3,mm1);\

  1295.                     psllw_i2r(1,mm2);\

  1296.                     paddw_r2r(mm4,mm0);\

  1297.                     psllw_i2r(2,mm1);\

  1298.                     paddw_r2r(mm6,mm2);\

  1299.                     paddw_r2r(mm2,mm1);\

  1300.                     psubusw_r2r(mm0,mm1);\

  1301.                     psrlw_i2r(3,mm1);\

  1302.                     packuswb_r2r(mm7,mm1);\

  1303.                     movd_r2m(mm1,lum_m2[0]);

  1304. 

  1305. #define DEINT_LINE_LUM \

  1306.                     movd_m2r(lum_m4[0],mm0);\

  1307.                     movd_m2r(lum_m3[0],mm1);\

  1308.                     movd_m2r(lum_m2[0],mm2);\

  1309.                     movd_m2r(lum_m1[0],mm3);\

  1310.                     movd_m2r(lum[0],mm4);\

  1311.                     punpcklbw_r2r(mm7,mm0);\

  1312.                     punpcklbw_r2r(mm7,mm1);\

  1313.                     punpcklbw_r2r(mm7,mm2);\

  1314.                     punpcklbw_r2r(mm7,mm3);\

  1315.                     punpcklbw_r2r(mm7,mm4);\

  1316.                     paddw_r2r(mm3,mm1);\

  1317.                     psllw_i2r(1,mm2);\

  1318.                     paddw_r2r(mm4,mm0);\

  1319.                     psllw_i2r(2,mm1);\

  1320.                     paddw_r2r(mm6,mm2);\

  1321.                     paddw_r2r(mm2,mm1);\

  1322.                     psubusw_r2r(mm0,mm1);\

  1323.                     psrlw_i2r(3,mm1);\

  1324.                     packuswb_r2r(mm7,mm1);\

  1325.                     movd_r2m(mm1,dst[0]);

  1326. #endif

  1327. 

  1328. /* filter parameters: [-1 4 2 4 -1] // 8 */

  1329. static void deinterlace_line(UINT8 *dst, UINT8 *lum_m4, UINT8 *lum_m3, UINT8 *lum_m2, UINT8 *lum_m1, UINT8 *lum,

  1330.                                 int size)

  1331. {

  1332. #ifndef HAVE_MMX

  1333.     UINT8 *cm = cropTbl + MAX_NEG_CROP;

  1334.     int sum;

  1335. 

  1336.     for(;size > 0;size--) {

  1337.         sum = -lum_m4[0];

  1338.         sum += lum_m3[0] << 2;

  1339.         sum += lum_m2[0] << 1;

  1340.         sum += lum_m1[0] << 2;

  1341.         sum += -lum[0];

  1342.         dst[0] = cm[(sum + 4) >> 3];

  1343.         lum_m4++;

  1344.         lum_m3++;

  1345.         lum_m2++;

  1346.         lum_m1++;

  1347.         lum++;

  1348.         dst++;

  1349.     }

  1350. #else

  1351. 

  1352.     {

  1353.         mmx_t rounder;

  1354.         rounder.uw[0]=4;

  1355.         rounder.uw[1]=4;

  1356.         rounder.uw[2]=4;

  1357.         rounder.uw[3]=4;

  1358.         pxor_r2r(mm7,mm7);

  1359.         movq_m2r(rounder,mm6);

  1360.     }

  1361.     for (;size > 3; size-=4) {

  1362.         DEINT_LINE_LUM

  1363.         lum_m4+=4;

  1364.         lum_m3+=4;

  1365.         lum_m2+=4;

  1366.         lum_m1+=4;

  1367.         lum+=4;

  1368.         dst+=4;

  1369.     }

  1370. #endif

  1371. }

  1372. static void deinterlace_line_inplace(UINT8 *lum_m4, UINT8 *lum_m3, UINT8 *lum_m2, UINT8 *lum_m1, UINT8 *lum,

  1373.                              int size)

  1374. {

  1375. #ifndef HAVE_MMX

  1376.     UINT8 *cm = cropTbl + MAX_NEG_CROP;

  1377.     int sum;

  1378. 

  1379.     for(;size > 0;size--) {

  1380.         sum = -lum_m4[0];

  1381.         sum += lum_m3[0] << 2;

  1382.         sum += lum_m2[0] << 1;

  1383.         lum_m4[0]=lum_m2[0];

  1384.         sum += lum_m1[0] << 2;

  1385.         sum += -lum[0];

  1386.         lum_m2[0] = cm[(sum + 4) >> 3];

  1387.         lum_m4++;

  1388.         lum_m3++;

  1389.         lum_m2++;

  1390.         lum_m1++;

  1391.         lum++;

  1392.     }

  1393. #else

  1394. 

  1395.     {

  1396.         mmx_t rounder;

  1397.         rounder.uw[0]=4;

  1398.         rounder.uw[1]=4;

  1399.         rounder.uw[2]=4;

  1400.         rounder.uw[3]=4;

  1401.         pxor_r2r(mm7,mm7);

  1402.         movq_m2r(rounder,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 *dst, int dst_wrap,

  1419.                                     UINT8 *src1, int src_wrap,

  1420.                                     int width, int height)

  1421. {

  1422.     UINT8 *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 *src1, int src_wrap,

  1448.                                      int width, int height)

  1449. {

  1450.     UINT8 *src_m1, *src_0, *src_p1, *src_p2;

  1451.     int y;

  1452.     UINT8 *buf;

  1453.     buf = (UINT8*)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. 

  1473. /* deinterlace - if not supported return -1 */

  1474. int avpicture_deinterlace(AVPicture *dst, AVPicture *src,

  1475.                           int pix_fmt, int width, int height)

  1476. {

  1477.     int i;

  1478. 

  1479.     if (pix_fmt != PIX_FMT_YUV420P &&

  1480.         pix_fmt != PIX_FMT_YUV422P &&

  1481.         pix_fmt != PIX_FMT_YUV444P)

  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.             default:

  1497.                 break;

  1498.             }

  1499.         }

  1500.         if (src == dst) {

  1501.             deinterlace_bottom_field_inplace(src->data[i], src->linesize[i],

  1502.                                  width, height);

  1503.         } else {

  1504.             deinterlace_bottom_field(dst->data[i],dst->linesize[i],

  1505.                                         src->data[i], src->linesize[i],

  1506.                                         width, height);

  1507.         }

  1508.     }

  1509. #ifdef HAVE_MMX

  1510.     emms();

  1511. #endif

  1512.     return 0;

  1513. }

  1514. 

  1515. #undef FIX