/* GStreamer

 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>

 *

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

 * modify it under the terms of the GNU Library 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

 * Library General Public License for more details.

 *

 * You should have received a copy of the GNU Library 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.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

/* non-GST-specific stuff */

#include "gstvideotestsrc.h"

#include "videotestsrc.h"

#include <gst/liboil.h>

#include <string.h>

#include <stdlib.h>

#include <math.h>

#ifndef M_PI

#define M_PI  3.14159265358979323846

#endif

static unsigned char

random_char (void)

{

  static unsigned int state;

  state *= 1103515245;

  state += 12345;

  return (state >> 16) & 0xff;

}

#if 0

static void

random_chars (unsigned char *dest, int nbytes)

{

  int i;

  static unsigned int state;

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

    state *= 1103515245;

    state += 12345;

    dest[i] = (state >> 16);

  }

}

#endif

#if 0

static void

paint_rect_random (unsigned char *dest, int stride, int x, int y, int w, int h)

{

  unsigned char *d = dest + stride * y + x;

  int i;

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

    random_chars (d, w);

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char color)

{

  unsigned char *d = dest + stride * y + x;

  int i;

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

    oil_splat_u8_ns (d, &color, w);

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect_s2 (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char col)

{

  unsigned char *d = dest + stride * y + x * 2;

  unsigned char *dp;

  int i, j;

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

    dp = d;

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

      *dp = col;

      dp += 2;

    }

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect2 (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char *col)

{

  unsigned char *d = dest + stride * y + x * 2;

  unsigned char *dp;

  int i, j;

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

    dp = d;

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

      *dp++ = col[0];

      *dp++ = col[1];

    }

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect3 (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char *col)

{

  unsigned char *d = dest + stride * y + x * 3;

  unsigned char *dp;

  int i, j;

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

    dp = d;

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

      *dp++ = col[0];

      *dp++ = col[1];

      *dp++ = col[2];

    }

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect4 (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char *col)

{

  unsigned char *d = dest + stride * y + x * 4;

  unsigned char *dp;

  int i, j;

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

    dp = d;

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

      *dp++ = col[0];

      *dp++ = col[1];

      *dp++ = col[2];

      *dp++ = col[3];

    }

    d += stride;

  }

}

#endif

#if 0

static void

paint_rect_s4 (unsigned char *dest, int stride, int x, int y, int w, int h,

    unsigned char col)

{

  unsigned char *d = dest + stride * y + x * 4;

  unsigned char *dp;

  int i, j;

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

    dp = d;

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

      *dp = col;

      dp += 4;

    }

    d += stride;

  }

}

#endif

enum

{

  COLOR_WHITE = 0,

  COLOR_YELLOW,

  COLOR_CYAN,

  COLOR_GREEN,

  COLOR_MAGENTA,

  COLOR_RED,

  COLOR_BLUE,

  COLOR_BLACK,

  COLOR_NEG_I,

  COLOR_POS_Q,

  COLOR_SUPER_BLACK,

  COLOR_DARK_GREY

};

static const struct vts_color_struct vts_colors[] = {

  /* 100% white */

  {255, 128, 128, 255, 255, 255, 255},

  /* yellow */

  {226, 0, 155, 255, 255, 0, 255},

  /* cyan */

  {179, 170, 0, 0, 255, 255, 255},

  /* green */

  {150, 46, 21, 0, 255, 0, 255},

  /* magenta */

  {105, 212, 235, 255, 0, 255, 255},

  /* red */

  {76, 85, 255, 255, 0, 0, 255},

  /* blue */

  {29, 255, 107, 0, 0, 255, 255},

  /* black */

  {16, 128, 128, 0, 0, 0, 255},

  /* -I */

  {16, 198, 21, 0, 0, 128, 255},

  /* +Q */

  {16, 235, 198, 0, 128, 255, 255},

  /* superblack */

  {0, 128, 128, 0, 0, 0, 255},

  /* 5% grey */

  {32, 128, 128, 32, 32, 32, 255},

};

#if 0

/*                        wht  yel  cya  grn  mag  red  blu  blk   -I    Q, superblack, dark grey */

static int y_colors[] = { 255, 226, 179, 150, 105, 76, 29, 16, 16, 16, 0, 32 };

static int u_colors[] =

    { 128, 0, 170, 46, 212, 85, 255, 128, 198, 235, 128, 128 };

static int v_colors[] =

    { 128, 155, 0, 21, 235, 255, 107, 128, 21, 198, 128, 128 };

/*                        wht  yel  cya  grn  mag  red  blu  blk   -I    Q  superblack, dark grey */

static int r_colors[] = { 255, 255, 0, 0, 255, 255, 0, 0, 0, 0, 0, 32 };

static int g_colors[] = { 255, 255, 255, 255, 0, 0, 0, 0, 0, 128, 0, 32 };

static int b_colors[] = { 255, 0, 255, 0, 255, 0, 255, 0, 128, 255, 0, 32 };

#endif

static void paint_setup_I420 (paintinfo * p, unsigned char *dest);

static void paint_setup_YV12 (paintinfo * p, unsigned char *dest);

static void paint_setup_YUY2 (paintinfo * p, unsigned char *dest);

static void paint_setup_UYVY (paintinfo * p, unsigned char *dest);

static void paint_setup_YVYU (paintinfo * p, unsigned char *dest);

static void paint_setup_IYU2 (paintinfo * p, unsigned char *dest);

static void paint_setup_Y41B (paintinfo * p, unsigned char *dest);

static void paint_setup_Y42B (paintinfo * p, unsigned char *dest);

static void paint_setup_Y800 (paintinfo * p, unsigned char *dest);

static void paint_setup_AYUV (paintinfo * p, unsigned char *dest);

#if 0

static void paint_setup_IMC1 (paintinfo * p, unsigned char *dest);

static void paint_setup_IMC2 (paintinfo * p, unsigned char *dest);

static void paint_setup_IMC3 (paintinfo * p, unsigned char *dest);

static void paint_setup_IMC4 (paintinfo * p, unsigned char *dest);

#endif

static void paint_setup_YUV9 (paintinfo * p, unsigned char *dest);

static void paint_setup_YVU9 (paintinfo * p, unsigned char *dest);

static void paint_setup_ARGB8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_ABGR8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_RGBA8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_BGRA8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_xRGB8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_xBGR8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_RGBx8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_BGRx8888 (paintinfo * p, unsigned char *dest);

static void paint_setup_RGB888 (paintinfo * p, unsigned char *dest);

static void paint_setup_BGR888 (paintinfo * p, unsigned char *dest);

static void paint_setup_RGB565 (paintinfo * p, unsigned char *dest);

static void paint_setup_xRGB1555 (paintinfo * p, unsigned char *dest);

static void paint_setup_bayer (paintinfo * p, unsigned char *dest);

static void paint_hline_I420 (paintinfo * p, int x, int y, int w);

static void paint_hline_YUY2 (paintinfo * p, int x, int y, int w);

static void paint_hline_IYU2 (paintinfo * p, int x, int y, int w);

static void paint_hline_Y41B (paintinfo * p, int x, int y, int w);

static void paint_hline_Y42B (paintinfo * p, int x, int y, int w);

static void paint_hline_Y800 (paintinfo * p, int x, int y, int w);

static void paint_hline_AYUV (paintinfo * p, int x, int y, int w);

#if 0

static void paint_hline_IMC1 (paintinfo * p, int x, int y, int w);

#endif

static void paint_hline_YUV9 (paintinfo * p, int x, int y, int w);

static void paint_hline_str4 (paintinfo * p, int x, int y, int w);

static void paint_hline_str3 (paintinfo * p, int x, int y, int w);

static void paint_hline_RGB565 (paintinfo * p, int x, int y, int w);

static void paint_hline_xRGB1555 (paintinfo * p, int x, int y, int w);

static void paint_hline_bayer (paintinfo * p, int x, int y, int w);

struct fourcc_list_struct fourcc_list[] = {

/* packed */

  {VTS_YUV, "YUY2", "YUY2", 16, paint_setup_YUY2, paint_hline_YUY2},

  {VTS_YUV, "UYVY", "UYVY", 16, paint_setup_UYVY, paint_hline_YUY2},

  {VTS_YUV, "Y422", "Y422", 16, paint_setup_UYVY, paint_hline_YUY2},

  {VTS_YUV, "UYNV", "UYNV", 16, paint_setup_UYVY, paint_hline_YUY2},    /* FIXME: UYNV? */

  {VTS_YUV, "YVYU", "YVYU", 16, paint_setup_YVYU, paint_hline_YUY2},

  {VTS_YUV, "AYUV", "AYUV", 32, paint_setup_AYUV, paint_hline_AYUV},

  /* interlaced */

  /*{ VTS_YUV,  "IUYV", "IUY2", 16, paint_setup_YVYU, paint_hline_YUY2 }, */

  /* inverted */

  /*{ VTS_YUV,  "cyuv", "cyuv", 16, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "Y41P", "Y41P", 12, paint_setup_YVYU, paint_hline_YUY2 }, */

  /* interlaced */

  /*{ VTS_YUV,  "IY41", "IY41", 12, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "Y211", "Y211", 8, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "Y41T", "Y41T", 12, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "Y42P", "Y42P", 16, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "CLJR", "CLJR", 8, paint_setup_YVYU, paint_hline_YUY2 }, */

  /*{ VTS_YUV,  "IYU1", "IYU1", 12, paint_setup_YVYU, paint_hline_YUY2 }, */

  {VTS_YUV, "IYU2", "IYU2", 24, paint_setup_IYU2, paint_hline_IYU2},

/* planar */

  /* YVU9 */

  {VTS_YUV, "YVU9", "YVU9", 9, paint_setup_YVU9, paint_hline_YUV9},

  /* YUV9 */

  {VTS_YUV, "YUV9", "YUV9", 9, paint_setup_YUV9, paint_hline_YUV9},

  /* IF09 */

  /* YV12 */

  {VTS_YUV, "YV12", "YV12", 12, paint_setup_YV12, paint_hline_I420},

  /* I420 */

  {VTS_YUV, "I420", "I420", 12, paint_setup_I420, paint_hline_I420},

  /* NV12 */

  /* NV21 */

#if 0

  /* IMC1 */

  {VTS_YUV, "IMC1", "IMC1", 16, paint_setup_IMC1, paint_hline_IMC1},

  /* IMC2 */

  {VTS_YUV, "IMC2", "IMC2", 12, paint_setup_IMC2, paint_hline_IMC1},

  /* IMC3 */

  {VTS_YUV, "IMC3", "IMC3", 16, paint_setup_IMC3, paint_hline_IMC1},

  /* IMC4 */

  {VTS_YUV, "IMC4", "IMC4", 12, paint_setup_IMC4, paint_hline_IMC1},

#endif

  /* CLPL */

  /* Y41B */

  {VTS_YUV, "Y41B", "Y41B", 12, paint_setup_Y41B, paint_hline_Y41B},

  /* Y42B */

  {VTS_YUV, "Y42B", "Y42B", 16, paint_setup_Y42B, paint_hline_Y42B},

  /* Y800 grayscale */

  {VTS_YUV, "Y800", "Y800", 8, paint_setup_Y800, paint_hline_Y800},

  {VTS_RGB, "RGB ", "xRGB8888", 32, paint_setup_xRGB8888, paint_hline_str4, 24,

      0x00ff0000, 0x0000ff00, 0x000000ff},

  {VTS_RGB, "RGB ", "xBGR8888", 32, paint_setup_xBGR8888, paint_hline_str4, 24,

      0x000000ff, 0x0000ff00, 0x00ff0000},

  {VTS_RGB, "RGB ", "RGBx8888", 32, paint_setup_RGBx8888, paint_hline_str4, 24,

      0xff000000, 0x00ff0000, 0x0000ff00},

  {VTS_RGB, "RGB ", "BGRx8888", 32, paint_setup_BGRx8888, paint_hline_str4, 24,

      0x0000ff00, 0x00ff0000, 0xff000000},

  {VTS_RGB, "RGB ", "ARGB8888", 32, paint_setup_ARGB8888, paint_hline_str4, 32,

      0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000},

  {VTS_RGB, "RGB ", "ABGR8888", 32, paint_setup_ABGR8888, paint_hline_str4, 32,

      0x000000ff, 0x0000ff00, 0x00ff0000, 0xff000000},

  {VTS_RGB, "RGB ", "RGBA8888", 32, paint_setup_RGBA8888, paint_hline_str4, 32,

      0xff000000, 0x00ff0000, 0x0000ff00, 0x000000ff},

  {VTS_RGB, "RGB ", "BGRA8888", 32, paint_setup_BGRA8888, paint_hline_str4, 32,

      0x0000ff00, 0x00ff0000, 0xff000000, 0x000000ff},

  {VTS_RGB, "RGB ", "RGB888", 24, paint_setup_RGB888, paint_hline_str3, 24,

      0x00ff0000, 0x0000ff00, 0x000000ff},

  {VTS_RGB, "RGB ", "BGR888", 24, paint_setup_BGR888, paint_hline_str3, 24,

      0x000000ff, 0x0000ff00, 0x00ff0000},

  {VTS_RGB, "RGB ", "RGB565", 16, paint_setup_RGB565, paint_hline_RGB565, 16,

      0x0000f800, 0x000007e0, 0x0000001f},

  {VTS_RGB, "RGB ", "xRGB1555", 16, paint_setup_xRGB1555, paint_hline_xRGB1555,

        15,

      0x00007c00, 0x000003e0, 0x0000001f},

  {VTS_BAYER, "BAY8", "Bayer", 8, paint_setup_bayer, paint_hline_bayer}

};

int n_fourccs = G_N_ELEMENTS (fourcc_list);

struct fourcc_list_struct *

paintinfo_find_by_structure (const GstStructure * structure)

{

  int i;

  const char *media_type = gst_structure_get_name (structure);

  int ret;

  g_return_val_if_fail (structure, NULL);

  if (strcmp (media_type, "video/x-raw-yuv") == 0) {

    char *s;

    int fourcc;

    guint32 format;

    ret = gst_structure_get_fourcc (structure, "format", &format);

    if (!ret)

      return NULL;

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

      s = fourcc_list[i].fourcc;

      /* g_print("testing %" GST_FOURCC_FORMAT " and %s\n", GST_FOURCC_ARGS(format), s); */

      fourcc = GST_MAKE_FOURCC (s[0], s[1], s[2], s[3]);

      if (fourcc_list[i].type == VTS_YUV && fourcc == format) {

        return fourcc_list + i;

      }

    }

  } else if (strcmp (media_type, "video/x-raw-rgb") == 0) {

    int red_mask;

    int green_mask;

    int blue_mask;

    int alpha_mask;

    int depth;

    int bpp;

    ret = gst_structure_get_int (structure, "red_mask", &red_mask);

    ret &= gst_structure_get_int (structure, "green_mask", &green_mask);

    ret &= gst_structure_get_int (structure, "blue_mask", &blue_mask);

    ret &= gst_structure_get_int (structure, "depth", &depth);

    ret &= gst_structure_get_int (structure, "bpp", &bpp);

    if (depth == 32) {

      ret &= gst_structure_get_int (structure, "alpha_mask", &alpha_mask);

      ret &= (alpha_mask != 0);

    } else {

      alpha_mask = 0;

    }

    if (!ret) {

      GST_WARNING ("incomplete caps structure: %" GST_PTR_FORMAT, structure);

      return NULL;

    }

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

      if (fourcc_list[i].type == VTS_RGB &&

          fourcc_list[i].red_mask == red_mask &&

          fourcc_list[i].green_mask == green_mask &&

          fourcc_list[i].blue_mask == blue_mask &&

          (alpha_mask == 0 || fourcc_list[i].alpha_mask == alpha_mask) &&

          fourcc_list[i].depth == depth && fourcc_list[i].bitspp == bpp) {

        return fourcc_list + i;

      }

    }

    return NULL;

  } else if (strcmp (media_type, "video/x-raw-bayer") == 0) {

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

      if (fourcc_list[i].type == VTS_BAYER) {

        return fourcc_list + i;

      }

    }

    return NULL;

  }

  g_critical ("format not found for media type %s", media_type);

  return NULL;

}

struct fourcc_list_struct *

paintrect_find_fourcc (int find_fourcc)

{

  int i;

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

    char *s;

    int fourcc;

    s = fourcc_list[i].fourcc;

    fourcc = GST_MAKE_FOURCC (s[0], s[1], s[2], s[3]);

    if (find_fourcc == fourcc) {

      /* If YUV format, it's good */

      if (!fourcc_list[i].type == VTS_YUV) {

        return fourcc_list + i;

      }

      return fourcc_list + i;

    }

  }

  return NULL;

}

struct fourcc_list_struct *

paintrect_find_name (const char *name)

{

  int i;

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

    if (strcmp (name, fourcc_list[i].name) == 0) {

      return fourcc_list + i;

    }

  }

  return NULL;

}

GstStructure *

paint_get_structure (struct fourcc_list_struct * format)

{

  GstStructure *structure = NULL;

  unsigned int fourcc;

  int endianness;

  g_return_val_if_fail (format, NULL);

  fourcc =

      GST_MAKE_FOURCC (format->fourcc[0], format->fourcc[1], format->fourcc[2],

      format->fourcc[3]);

  switch (format->type) {

    case VTS_RGB:

      if (format->bitspp == 16) {

        endianness = G_BYTE_ORDER;

      } else {

        endianness = G_BIG_ENDIAN;