/*

 * Misc image convertion routines

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

 *

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

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

 * License as published by the Free Software Foundation; either

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

 *

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

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

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

 * Lesser General Public License for more details.

 *

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

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

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

 */

/**

 * @file imgconvert.c

 * Misc image convertion routines.

 */

/* TODO:

 * - write 'ffimg' program to test all the image related stuff

 * - move all api to slice based system

 * - integrate deinterlacing, postprocessing and scaling in the conversion process

 */

#include "avcodec.h"

#include "dsputil.h"

#include "gstffmpegcodecmap.h"

#include <string.h>

#include <stdlib.h>

#define xglue(x, y) x ## y

#define glue(x, y) xglue(x, y)

/* this table gives more information about formats */

/* FIXME, this table is also in ffmpegcodecmap */

static PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {

  /* YUV formats */

  /* [PIX_FMT_YUV420P] = */ {

        /* .format         = */ PIX_FMT_YUV420P,

        /* .name           = */ "yuv420p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 1,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YVU420P] = */ {

        /* .format         = */ PIX_FMT_YVU420P,

        /* .name           = */ "yvu420p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 1,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUV422P] = */ {

        /* .format         = */ PIX_FMT_YUV422P,

        /* .name           = */ "yuv422p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUV444P] = */ {

        /* .format         = */ PIX_FMT_YUV444P,

        /* .name           = */ "yuv444p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUV422] = */ {

        /* .format         = */ PIX_FMT_YUV422,

        /* .name           = */ "yuv422",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_UYVY422] = */ {

        /* .format         = */ PIX_FMT_UYVY422,

        /* .name           = */ "uyvy422",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUV410P] = */ {

        /* .format         = */ PIX_FMT_YUV410P,

        /* .name           = */ "yuv410p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 2,

        /* .y_chroma_shift = */ 2,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YVU410P] = */ {

        /* .format         = */ PIX_FMT_YVU410P,

        /* .name           = */ "yvu410p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 2,

        /* .y_chroma_shift = */ 2,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUV411P] = */ {

        /* .format         = */ PIX_FMT_YUV411P,

        /* .name           = */ "yuv411p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 2,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* JPEG YUV */

  /* [PIX_FMT_YUVJ420P] = */ {

        /* .format         = */ PIX_FMT_YUVJ420P,

        /* .name           = */ "yuvj420p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV_JPEG,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 1,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUVJ422P] = */ {

        /* .format         = */ PIX_FMT_YUVJ422P,

        /* .name           = */ "yuvj422p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV_JPEG,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 1,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_YUVJ444P] = */ {

        /* .format         = */ PIX_FMT_YUVJ444P,

        /* .name           = */ "yuvj444p",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_YUV_JPEG,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* RGB formats */

  /* [PIX_FMT_RGB24] = */ {

        /* .format         = */ PIX_FMT_RGB24,

        /* .name           = */ "rgb24",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_BGR24] = */ {

        /* .format         = */ PIX_FMT_BGR24,

        /* .name           = */ "bgr24",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_RGB32] = */ {

        /* .format         = */ PIX_FMT_RGB32,

        /* .name           = */ "rgb32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_BGR32] = */ {

        /* .format         = */ PIX_FMT_BGR32,

        /* .name           = */ "bgr32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_RGB32] = */ {

        /* .format         = */ PIX_FMT_xRGB32,

        /* .name           = */ "xrgb32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_BGR32] = */ {

        /* .format         = */ PIX_FMT_BGRx32,

        /* .name           = */ "bgrx32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_RGBA32] = */ {

        /* .format         = */ PIX_FMT_RGBA32,

        /* .name           = */ "rgba32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_BGRA32] = */ {

        /* .format         = */ PIX_FMT_BGRA32,

        /* .name           = */ "bgra32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_ARGB32] = */ {

        /* .format         = */ PIX_FMT_ARGB32,

        /* .name           = */ "argb32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_ABGR32] = */ {

        /* .format         = */ PIX_FMT_ABGR32,

        /* .name           = */ "abgr32",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_RGB565] = */ {

        /* .format         = */ PIX_FMT_RGB565,

        /* .name           = */ "rgb565",

        /* .nb_channels    = */ 3,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 5,

      },

  /* [PIX_FMT_RGB555] = */ {

        /* .format         = */ PIX_FMT_RGB555,

        /* .name           = */ "rgb555",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 5,

      },

  /* gray / mono formats */

  /* [PIX_FMT_GRAY8] = */ {

        /* .format         = */ PIX_FMT_GRAY8,

        /* .name           = */ "gray",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_GRAY,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_MONOWHITE] = */ {

        /* .format         = */ PIX_FMT_MONOWHITE,

        /* .name           = */ "monow",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_GRAY,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 1,

      },

  /* [PIX_FMT_MONOBLACK] = */ {

        /* .format         = */ PIX_FMT_MONOBLACK,

        /* .name           = */ "monob",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_GRAY,

        /* .pixel_type     = */ FF_PIXEL_PLANAR,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 1,

      },

  /* paletted formats */

  /* [PIX_FMT_PAL8] = */ {

        /* .format         = */ PIX_FMT_PAL8,

        /* .name           = */ "pal8",

        /* .nb_channels    = */ 4,

        /* .color_type     = */ FF_COLOR_RGB,

        /* .pixel_type     = */ FF_PIXEL_PALETTE,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_XVMC_MPEG2_MC] = */ {

        /* .format         = */ PIX_FMT_XVMC_MPEG2_MC,

        /* .name           = */ "xvmcmc",

      },

  /* [PIX_FMT_XVMC_MPEG2_IDCT] = */ {

        /* .format         = */ PIX_FMT_XVMC_MPEG2_IDCT,

        /* .name           = */ "xvmcidct",

      },

  /* [PIX_FMT_UYVY411] = */ {

        /* .format         = */ PIX_FMT_UYVY411,

        /* .name           = */ "uyvy411",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 0,

        /* .x_chroma_shift = */ 2,

        /* .y_chroma_shift = */ 0,

        /* .depth          = */ 8,

      },

  /* [PIX_FMT_AYUV4444] = */ {

        /* .format         = */ PIX_FMT_AYUV4444,

        /* .name           = */ "ayuv4444",

        /* .nb_channels    = */ 1,

        /* .color_type     = */ FF_COLOR_YUV,

        /* .pixel_type     = */ FF_PIXEL_PACKED,

        /* .is_alpha       = */ 1,

        /* .x_chroma_shift = */ 0,

        /* .y_chroma_shift = */ 0,

        /*.depth = */ 8

      }

};

/* returns NULL if not found */

/* undid static since this is also used in gstffmpegcodecmap.c */

#ifdef __SYMBIAN32__

EXPORT_C

#endif

PixFmtInfo *

get_pix_fmt_info (enum PixelFormat format)

{

  int i;

  for (i = 0; i < sizeof (pix_fmt_info) / sizeof (pix_fmt_info[0]); i++) {

    if (pix_fmt_info[i].format == format) {

      return pix_fmt_info + i;

    }

  }

  /* since this doesn't get checked *anywhere*, we might as well warn

     here if we return NULL so you have *some* idea what's going on */

  g_warning

      ("Could not find info for pixel format %d out of %d known pixel formats. One segfault coming up",

      format, PIX_FMT_NB);

  return NULL;

}

#ifdef __SYMBIAN32__

EXPORT_C

#endif

void

avcodec_get_chroma_sub_sample (int pix_fmt, int *h_shift, int *v_shift)

{

  *h_shift = get_pix_fmt_info (pix_fmt)->x_chroma_shift;

  *v_shift = get_pix_fmt_info (pix_fmt)->y_chroma_shift;

}

#ifdef __SYMBIAN32__

EXPORT_C

#endif

const char *

avcodec_get_pix_fmt_name (int pix_fmt)

{

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

    return "???";

  else

    return get_pix_fmt_info (pix_fmt)->name;

}

#ifdef __SYMBIAN32__

EXPORT_C

#endif

enum PixelFormat

avcodec_get_pix_fmt (const char *name)

{

  int i;

  for (i = 0; i < PIX_FMT_NB; i++)

    if (!strcmp (pix_fmt_info[i].name, name))

      break;

  return pix_fmt_info[i].format;

}

#if 0

static int

avpicture_layout (const AVPicture * src, int pix_fmt, int width, int height,

    unsigned char *dest, int dest_size)

{

  PixFmtInfo *pf = get_pix_fmt_info (pix_fmt);

  int i, j, w, h, data_planes;

  const unsigned char *s;

  int size = avpicture_get_size (pix_fmt, width, height);

  if (size > dest_size)

    return -1;

  if (pf->pixel_type == FF_PIXEL_PACKED || pf->pixel_type == FF_PIXEL_PALETTE) {

    if (pix_fmt == PIX_FMT_YUV422 ||

        pix_fmt == PIX_FMT_UYVY422 ||

        pix_fmt == PIX_FMT_RGB565 || pix_fmt == PIX_FMT_RGB555)

      w = width * 2;

    else if (pix_fmt == PIX_FMT_UYVY411)

      w = width + width / 2;

    else if (pix_fmt == PIX_FMT_PAL8)

      w = width;

    else

      w = width * (pf->depth * pf->nb_channels / 8);

    data_planes = 1;

    h = height;

  } else {

    data_planes = pf->nb_channels;

    w = (width * pf->depth + 7) / 8;

    h = height;

  }

  for (i = 0; i < data_planes; i++) {

    if (i == 1) {

      w = width >> pf->x_chroma_shift;

      h = height >> pf->y_chroma_shift;

    }

    s = src->data[i];

    for (j = 0; j < h; j++) {

      memcpy (dest, s, w);

      dest += w;

      s += src->linesize[i];

    }

  }

  if (pf->pixel_type == FF_PIXEL_PALETTE)

    memcpy ((unsigned char *) (((size_t) dest + 3) & ~3), src->data[1],

        256 * 4);

  return size;

}

#endif

#ifdef __SYMBIAN32__

EXPORT_C

#endif

int

avpicture_get_size (int pix_fmt, int width, int height)

{

  AVPicture dummy_pict;

  return gst_ffmpegcsp_avpicture_fill (&dummy_pict, NULL, pix_fmt, width,

      height);

}

/**

 * compute the loss when converting from a pixel format to another 

 */

#ifdef __SYMBIAN32__

EXPORT_C