/* GStreamer

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

 * Copyright (C) 2005 David Schleef <ds@schleef.org>

 *

 * 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.

 */

/**

 * SECTION:element-videoscale

 * @see_also: videorate, ffmpegcolorspace

 *

 * This element resizes video frames. By default the element will try to

 * negotiate to the same size on the source and sinkpad so that no scaling

 * is needed. It is therefore safe to insert this element in a pipeline to

 * get more robust behaviour without any cost if no scaling is needed.

 * This element supports a wide range of color spaces including various YUV and

 * RGB formats and is therefore generally able to operate anywhere in a

 * pipeline.

 * <title>Example pipelines</title>

 * gst-launch -v filesrc location=videotestsrc.ogg ! oggdemux ! theoradec ! ffmpegcolorspace ! videoscale ! ximagesink

 * ximagesink cannot perform scaling, the video scaling will be performed by

 * videoscale when you resize the video window.

 * To create the test Ogg/Theora file refer to the documentation of theoraenc.

 * gst-launch -v filesrc location=videotestsrc.ogg ! oggdemux ! theoradec ! videoscale ! video/x-raw-yuv, width=50 ! xvimagesink

 * </refsect2>

 *

 * Last reviewed on 2006-03-02 (0.10.4)

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <string.h>

#include <gst/video/video.h>

#include "gstvideoscale.h"

#include "vs_image.h"

#include "vs_4tap.h"

/* debug variable definition */

GST_DEBUG_CATEGORY (video_scale_debug);

/* elementfactory information */

static const GstElementDetails video_scale_details =

GST_ELEMENT_DETAILS ("Video scaler",

    "Filter/Effect/Video",

    "Resizes video",

    "Wim Taymans <wim.taymans@chello.be>");

enum

{

  PROP_0,

  PROP_METHOD

      /* FILL ME */

};

static GstStaticCaps gst_video_scale_format_caps[] = {

  GST_STATIC_CAPS (GST_VIDEO_CAPS_RGBx),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_xRGB),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_BGRx),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_xBGR),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_RGBA),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_ARGB),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_BGRA),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_ABGR),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_RGB),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_BGR),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("AYUV")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("YUY2")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("YVYU")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("UYVY")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("Y800")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("YV12")),

  GST_STATIC_CAPS (GST_VIDEO_CAPS_RGB_16),

};

enum

{

  GST_VIDEO_SCALE_RGBx = 0,

  GST_VIDEO_SCALE_xRGB,

  GST_VIDEO_SCALE_BGRx,

  GST_VIDEO_SCALE_xBGR,

  GST_VIDEO_SCALE_RGBA,

  GST_VIDEO_SCALE_ARGB,

  GST_VIDEO_SCALE_BGRA,

  GST_VIDEO_SCALE_ABGR,

  GST_VIDEO_SCALE_RGB,

  GST_VIDEO_SCALE_BGR,

  GST_VIDEO_SCALE_AYUV,

  GST_VIDEO_SCALE_YUY2,

  GST_VIDEO_SCALE_YVYU,

  GST_VIDEO_SCALE_UYVY,

  GST_VIDEO_SCALE_Y,

  GST_VIDEO_SCALE_I420,

  GST_VIDEO_SCALE_YV12,

  GST_VIDEO_SCALE_RGB565,

};

#define GST_TYPE_VIDEO_SCALE_METHOD (gst_video_scale_method_get_type())

static GType

gst_video_scale_method_get_type (void)

{

  static GType video_scale_method_type = 0;

  static const GEnumValue video_scale_methods[] = {

    {GST_VIDEO_SCALE_NEAREST, "Nearest Neighbour", "nearest-neighbour"},

    {GST_VIDEO_SCALE_BILINEAR, "Bilinear", "bilinear"},

    {GST_VIDEO_SCALE_4TAP, "4-tap", "4-tap"},

    {0, NULL, NULL},

  };

  if (!video_scale_method_type) {

    video_scale_method_type =

        g_enum_register_static ("GstVideoScaleMethod", video_scale_methods);

  }

  return video_scale_method_type;

}

static GstCaps *

gst_video_scale_get_capslist (void)

{

  static GstCaps *caps;

  if (caps == NULL) {

    int i;

    caps = gst_caps_new_empty ();

    for (i = 0; i < G_N_ELEMENTS (gst_video_scale_format_caps); i++)

      gst_caps_append (caps,

          gst_caps_make_writable

          (gst_static_caps_get (&gst_video_scale_format_caps[i])));

  }

  return caps;

}

static GstPadTemplate *

gst_video_scale_src_template_factory (void)

{

  return gst_pad_template_new ("src", GST_PAD_SRC, GST_PAD_ALWAYS,

      gst_caps_ref (gst_video_scale_get_capslist ()));

}

static GstPadTemplate *

gst_video_scale_sink_template_factory (void)

{

  return gst_pad_template_new ("sink", GST_PAD_SINK, GST_PAD_ALWAYS,

      gst_caps_ref (gst_video_scale_get_capslist ()));

}

static void gst_video_scale_base_init (gpointer g_class);

static void gst_video_scale_class_init (GstVideoScaleClass * klass);

static void gst_video_scale_init (GstVideoScale * videoscale);

static void gst_video_scale_finalize (GstVideoScale * videoscale);

static gboolean gst_video_scale_src_event (GstBaseTransform * trans,

    GstEvent * event);

/* base transform vmethods */

static GstCaps *gst_video_scale_transform_caps (GstBaseTransform * trans,

    GstPadDirection direction, GstCaps * caps);

static gboolean gst_video_scale_set_caps (GstBaseTransform * trans,

    GstCaps * in, GstCaps * out);

static gboolean gst_video_scale_get_unit_size (GstBaseTransform * trans,

    GstCaps * caps, guint * size);

static GstFlowReturn gst_video_scale_transform (GstBaseTransform * trans,

    GstBuffer * in, GstBuffer * out);

static void gst_video_scale_fixate_caps (GstBaseTransform * base,

    GstPadDirection direction, GstCaps * caps, GstCaps * othercaps);

static void gst_video_scale_set_property (GObject * object, guint prop_id,

    const GValue * value, GParamSpec * pspec);

static void gst_video_scale_get_property (GObject * object, guint prop_id,

    GValue * value, GParamSpec * pspec);

static GstElementClass *parent_class = NULL;

#ifdef __SYMBIAN32__

EXPORT_C

#endif

GType

gst_video_scale_get_type (void)

{

  static GType video_scale_type = 0;

  if (!video_scale_type) {

    static const GTypeInfo video_scale_info = {

      sizeof (GstVideoScaleClass),

      gst_video_scale_base_init,

      NULL,

      (GClassInitFunc) gst_video_scale_class_init,

      NULL,

      NULL,

      sizeof (GstVideoScale),

      0,

      (GInstanceInitFunc) gst_video_scale_init,

    };

    video_scale_type =

        g_type_register_static (GST_TYPE_BASE_TRANSFORM, "GstVideoScale",

        &video_scale_info, 0);

  }

  return video_scale_type;

}

static void

gst_video_scale_base_init (gpointer g_class)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_set_details (element_class, &video_scale_details);

  gst_element_class_add_pad_template (element_class,

      gst_video_scale_sink_template_factory ());

  gst_element_class_add_pad_template (element_class,

      gst_video_scale_src_template_factory ());

}

static void

gst_video_scale_class_init (GstVideoScaleClass * klass)

{

  GObjectClass *gobject_class;

  GstBaseTransformClass *trans_class;

  gobject_class = (GObjectClass *) klass;

  trans_class = (GstBaseTransformClass *) klass;

  gobject_class->finalize = (GObjectFinalizeFunc) gst_video_scale_finalize;

  gobject_class->set_property = gst_video_scale_set_property;

  gobject_class->get_property = gst_video_scale_get_property;

  g_object_class_install_property (gobject_class, PROP_METHOD,

      g_param_spec_enum ("method", "method", "method",

  trans_class->transform_caps =

      GST_DEBUG_FUNCPTR (gst_video_scale_transform_caps);

  trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_video_scale_set_caps);

  trans_class->get_unit_size =

      GST_DEBUG_FUNCPTR (gst_video_scale_get_unit_size);

  trans_class->transform = GST_DEBUG_FUNCPTR (gst_video_scale_transform);

  trans_class->fixate_caps = GST_DEBUG_FUNCPTR (gst_video_scale_fixate_caps);

  trans_class->src_event = GST_DEBUG_FUNCPTR (gst_video_scale_src_event);

  trans_class->passthrough_on_same_caps = TRUE;

  parent_class = g_type_class_peek_parent (klass);

}

static void

gst_video_scale_init (GstVideoScale * videoscale)

{

  gst_base_transform_set_qos_enabled (GST_BASE_TRANSFORM (videoscale), TRUE);

  videoscale->tmp_buf = NULL;

  videoscale->method = DEFAULT_PROP_METHOD;

}

static void

gst_video_scale_finalize (GstVideoScale * videoscale)

{

  if (videoscale->tmp_buf)

    g_free (videoscale->tmp_buf);

  G_OBJECT_CLASS (parent_class)->finalize (G_OBJECT (videoscale));

}

static void

gst_video_scale_set_property (GObject * object, guint prop_id,

    const GValue * value, GParamSpec * pspec)

{

  GstVideoScale *vscale = GST_VIDEO_SCALE (object);

  switch (prop_id) {

    case PROP_METHOD:

      GST_OBJECT_LOCK (vscale);

      vscale->method = g_value_get_enum (value);

      GST_OBJECT_UNLOCK (vscale);

      break;

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

      break;

  }

}

static void

gst_video_scale_get_property (GObject * object, guint prop_id, GValue * value,

    GParamSpec * pspec)

{

  GstVideoScale *vscale = GST_VIDEO_SCALE (object);

  switch (prop_id) {

    case PROP_METHOD:

      GST_OBJECT_LOCK (vscale);

      g_value_set_enum (value, vscale->method);

      GST_OBJECT_UNLOCK (vscale);

      break;

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

      break;

  }

}

static GstCaps *

gst_video_scale_transform_caps (GstBaseTransform * trans,

    GstPadDirection direction, GstCaps * caps)

{

  GstVideoScale *videoscale;

  GstCaps *ret;

  GstStructure *structure;

  const GValue *par;

  gint method;

  /* this function is always called with a simple caps */

  g_return_val_if_fail (GST_CAPS_IS_SIMPLE (caps), NULL);

  videoscale = GST_VIDEO_SCALE (trans);

  GST_OBJECT_LOCK (videoscale);

  method = videoscale->method;

  GST_OBJECT_UNLOCK (videoscale);

  structure = gst_caps_get_structure (caps, 0);

  ret = gst_caps_copy (caps);

  gst_structure_set (structure,

      "width", GST_TYPE_INT_RANGE, 1, G_MAXINT,

      "height", GST_TYPE_INT_RANGE, 1, G_MAXINT, NULL);

  /* if pixel aspect ratio, make a range of it */

  if ((par = gst_structure_get_value (structure, "pixel-aspect-ratio"))) {

        "pixel-aspect-ratio", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);

  }

  GST_DEBUG_OBJECT (trans, "returning caps: %" GST_PTR_FORMAT, ret);

  return ret;

}

static int

gst_video_scale_get_format (GstCaps * caps)

{

  int i;

  GstCaps *icaps, *scaps;

  for (i = 0; i < G_N_ELEMENTS (gst_video_scale_format_caps); i++) {

    scaps = gst_static_caps_get (&gst_video_scale_format_caps[i]);

    icaps = gst_caps_intersect (caps, scaps);

    if (!gst_caps_is_empty (icaps)) {

      gst_caps_unref (icaps);

      return i;

    }

    gst_caps_unref (icaps);

  }

  return -1;

}

/* calculate the size of a buffer */

static gboolean

gst_video_scale_prepare_size (GstVideoScale * videoscale, gint format,

    VSImage * img, gint width, gint height, guint * size)

{

  gboolean res = TRUE;

  img->width = width;

  img->height = height;

  switch (format) {

    case GST_VIDEO_SCALE_RGBx:

    case GST_VIDEO_SCALE_xRGB:

    case GST_VIDEO_SCALE_BGRx:

    case GST_VIDEO_SCALE_xBGR:

    case GST_VIDEO_SCALE_RGBA:

    case GST_VIDEO_SCALE_ARGB:

    case GST_VIDEO_SCALE_BGRA:

    case GST_VIDEO_SCALE_ABGR:

    case GST_VIDEO_SCALE_AYUV:

      img->stride = img->width * 4;

      *size = img->stride * img->height;

      break;

    case GST_VIDEO_SCALE_RGB:

    case GST_VIDEO_SCALE_BGR:

      img->stride = GST_ROUND_UP_4 (img->width * 3);

      *size = img->stride * img->height;

      break;

    case GST_VIDEO_SCALE_YUY2:

    case GST_VIDEO_SCALE_YVYU:

    case GST_VIDEO_SCALE_UYVY:

      img->stride = GST_ROUND_UP_4 (img->width * 2);

      *size = img->stride * img->height;

      break;

    case GST_VIDEO_SCALE_Y:

      img->stride = GST_ROUND_UP_4 (img->width);

      *size = img->stride * img->height;

      break;

    case GST_VIDEO_SCALE_I420:

    case GST_VIDEO_SCALE_YV12:

    {

      gulong img_u_stride, img_u_height;

      img->stride = GST_ROUND_UP_4 (img->width);

      img_u_height = GST_ROUND_UP_2 (img->height) / 2;

      img_u_stride = GST_ROUND_UP_4 (img->stride / 2);

      *size = img->stride * GST_ROUND_UP_2 (img->height) +

          2 * img_u_stride * img_u_height;

      break;

    }

    case GST_VIDEO_SCALE_RGB565:

      img->stride = GST_ROUND_UP_4 (img->width * 2);

      *size = img->stride * img->height;

      break;

    case GST_VIDEO_SCALE_RGB555:

      img->stride = GST_ROUND_UP_4 (img->width * 2);

      *size = img->stride * img->height;

      break;

    default:

      goto unknown_format;

  }

  return res;

  /* ERRORS */

unknown_format:

  {

    GST_ELEMENT_ERROR (videoscale, STREAM, NOT_IMPLEMENTED, (NULL),

        ("Unsupported format %d", videoscale->format));

    return FALSE;

  }

}

static gboolean

parse_caps (GstCaps * caps, gint * format, gint * width, gint * height)

{

  gboolean ret;

  GstStructure *structure;

  structure = gst_caps_get_structure (caps, 0);

  ret = gst_structure_get_int (structure, "width", width);

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

  if (format)

    *format = gst_video_scale_get_format (caps);

  return ret;

}

static gboolean

gst_video_scale_set_caps (GstBaseTransform * trans, GstCaps * in, GstCaps * out)