/* GStreamer
 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
 * Copyright (C) <2003> David Schleef <ds@schleef.org>
 * Copyright (C) 2003 Arwed v. Merkatz <v.merkatz@gmx.net>
 * Copyright (C) 2006 Mark Nauwelaerts <manauw@skynet.be>
 * Copyright (C) 2018 United States Government, Joshua M. Doe <oss@nvl.army.mil>
 *
 * 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-misbirunpack
*
* Unpack MISB IR packed video to GRAY16.
*
* <refsect2>
* <title>Example launch line</title>
* |[
* gst-launch videotestsrc ! misbirpack ! misbirunpack ! ffmpegcolorspace ! autovideosink
* ]|
* </refsect2>
*/

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "gstmisbirunpack.h"

#include <gst/video/video.h>

//#include "gstextractcolororc-dist.h"

/* GstMisbIrUnpack signals and args */
enum
{
  /* FILL ME */
  LAST_SIGNAL
};

enum
{
  PROP_0,
  PROP_OFFSET,
  PROP_SHIFT,
  PROP_SWAP,
  PROP_LUMA_MASK,
  PROP_CHROMA_MASK,
  PROP_LAST
};

#define DEFAULT_PROP_OFFSET -64
#define DEFAULT_PROP_SHIFT 8
#define DEFAULT_PROP_SWAP FALSE
#define DEFAULT_PROP_LUMA_MASK 0xff
#define DEFAULT_PROP_CHROMA_MASK 0xff

/* the capabilities of the inputs and outputs */
static GstStaticPadTemplate gst_misb_ir_unpack_sink_template =
GST_STATIC_PAD_TEMPLATE ("sink",
    GST_PAD_SINK,
    GST_PAD_ALWAYS,
    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("{ v210, UYVY }"))
    );

static GstStaticPadTemplate gst_misb_ir_unpack_src_template =
GST_STATIC_PAD_TEMPLATE ("src",
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE ("GRAY16_LE"))
    );


/* GObject vmethod declarations */
static void gst_misb_ir_unpack_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec);
static void gst_misb_ir_unpack_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec);
static void gst_misb_ir_unpack_dispose (GObject * object);

/* GstBaseTransform vmethod declarations */
static GstCaps *gst_misb_ir_unpack_transform_caps (GstBaseTransform * trans,
    GstPadDirection direction, GstCaps * caps, GstCaps * filter_caps);

/* GstVideoFilter vmethod declarations */
static gboolean gst_misb_ir_unpack_set_info (GstVideoFilter * filter,
    GstCaps * incaps, GstVideoInfo * in_info, GstCaps * outcaps,
    GstVideoInfo * out_info);
static GstFlowReturn gst_misb_ir_unpack_transform_frame (GstVideoFilter *
    filter, GstVideoFrame * in_frame, GstVideoFrame * out_frame);

/* GstMisbIrUnpack method declarations */
static void gst_misb_ir_unpack_reset (GstMisbIrUnpack * filter);

/* setup debug */
GST_DEBUG_CATEGORY_STATIC (misb_ir_unpack_debug);
#define GST_CAT_DEFAULT misb_ir_unpack_debug

G_DEFINE_TYPE (GstMisbIrUnpack, gst_misb_ir_unpack, GST_TYPE_VIDEO_FILTER);

/************************************************************************/
/* GObject vmethod implementations                                      */
/************************************************************************/

/**
 * gst_misb_ir_unpack_dispose:
 * @object: #GObject.
 *
 */
static void
gst_misb_ir_unpack_dispose (GObject * object)
{
  GstMisbIrUnpack *misb_ir_unpack = GST_MISB_IR_UNPACK (object);

  GST_DEBUG ("dispose");

  gst_misb_ir_unpack_reset (misb_ir_unpack);

  /* chain up to the parent class */
  G_OBJECT_CLASS (gst_misb_ir_unpack_parent_class)->dispose (object);
}

/**
 * gst_misb_ir_unpack_class_init:
 * @object: #GstMisbIrUnpackClass.
 *
 */
static void
gst_misb_ir_unpack_class_init (GstMisbIrUnpackClass * klass)
{
  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
  GstElementClass *gstelement_class = GST_ELEMENT_CLASS (klass);
  GstBaseTransformClass *gstbasetransform_class =
      GST_BASE_TRANSFORM_CLASS (klass);
  GstVideoFilterClass *gstvideofilter_class = GST_VIDEO_FILTER_CLASS (klass);

  GST_DEBUG ("class init");

  /* Register GObject vmethods */
  gobject_class->dispose = GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_dispose);
  gobject_class->set_property =
      GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_set_property);
  gobject_class->get_property =
      GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_get_property);

  /* Install GObject properties */
  g_object_class_install_property (G_OBJECT_CLASS (klass),
      PROP_OFFSET, g_param_spec_int ("offset",
          "Offset value",
          "Offset value to apply during unpacking", -0xffff, 0xffff,
          DEFAULT_PROP_OFFSET, G_PARAM_STATIC_STRINGS | G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass),
      PROP_SHIFT, g_param_spec_uint ("shift",
          "Shift value",
          "Bits to left shift luminance component during unpacking", 0, 15,
          DEFAULT_PROP_SHIFT, G_PARAM_STATIC_STRINGS | G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass),
      PROP_SWAP, g_param_spec_boolean ("swap", "Swap luma and chroma",
          "Whether to swap luminance and chrominance components",
          DEFAULT_PROP_SWAP, G_PARAM_STATIC_STRINGS | G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass),
      PROP_LUMA_MASK, g_param_spec_uint ("luma-mask",
          "Luma mask",
          "Mask to bitwise AND with luma after applying offset", 0, 0xffff,
          DEFAULT_PROP_LUMA_MASK, G_PARAM_STATIC_STRINGS | G_PARAM_READWRITE));
  g_object_class_install_property (G_OBJECT_CLASS (klass),
      PROP_CHROMA_MASK, g_param_spec_uint ("chroma-mask",
          "Chroma mask",
          "Mask to bitwise AND with chroma after applying offset", 0, 0xffff,
          DEFAULT_PROP_LUMA_MASK, G_PARAM_STATIC_STRINGS | G_PARAM_READWRITE));

  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&gst_misb_ir_unpack_sink_template));
  gst_element_class_add_pad_template (gstelement_class,
      gst_static_pad_template_get (&gst_misb_ir_unpack_src_template));

  gst_element_class_set_static_metadata (gstelement_class,
      "Unpack MISB IR video", "Filter/Effect/Video",
      "Unpack MISB IR video according to ST 0402.2 Method 2",
      "Joshua M. Doe <oss@nvl.army.mil>");

  /* Register GstBaseTransform vmethods */
  gstbasetransform_class->transform_caps =
      GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_transform_caps);

  gstvideofilter_class->set_info =
      GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_set_info);
  gstvideofilter_class->transform_frame =
      GST_DEBUG_FUNCPTR (gst_misb_ir_unpack_transform_frame);
}

static void
gst_misb_ir_unpack_init (GstMisbIrUnpack * filt)
{
  GST_DEBUG_OBJECT (filt, "init class instance");

  filt->offset_value = DEFAULT_PROP_OFFSET;
  filt->shift_value = DEFAULT_PROP_SHIFT;
  filt->swap = DEFAULT_PROP_SWAP;
  filt->luma_mask = DEFAULT_PROP_LUMA_MASK;
  filt->chroma_mask = DEFAULT_PROP_CHROMA_MASK;

  gst_base_transform_set_in_place (GST_BASE_TRANSFORM (filt), FALSE);

  gst_misb_ir_unpack_reset (filt);
}

static void
gst_misb_ir_unpack_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec)
{
  GstMisbIrUnpack *filt = GST_MISB_IR_UNPACK (object);

  GST_DEBUG_OBJECT (filt, "setting property %s", pspec->name);

  switch (prop_id) {
    case PROP_OFFSET:
      filt->offset_value = g_value_get_int (value);
      break;
    case PROP_SHIFT:
      filt->shift_value = g_value_get_uint (value);
      break;
    case PROP_SWAP:
      filt->swap = g_value_get_boolean (value);
      break;
    case PROP_LUMA_MASK:
      filt->luma_mask = g_value_get_uint (value);
      break;
    case PROP_CHROMA_MASK:
      filt->chroma_mask = g_value_get_uint (value);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

static void
gst_misb_ir_unpack_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec)
{
  GstMisbIrUnpack *filt = GST_MISB_IR_UNPACK (object);

  GST_DEBUG_OBJECT (filt, "getting property %s", pspec->name);

  switch (prop_id) {
    case PROP_OFFSET:
      g_value_set_int (value, filt->offset_value);
      break;
    case PROP_SHIFT:
      g_value_set_uint (value, filt->shift_value);
      break;
    case PROP_SWAP:
      g_value_set_boolean (value, filt->swap);
      break;
    case PROP_LUMA_MASK:
      g_value_set_uint (value, filt->luma_mask);
      break;
    case PROP_CHROMA_MASK:
      g_value_set_uint (value, filt->chroma_mask);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}

GstCaps *
gst_misb_ir_unpack_transform_caps (GstBaseTransform * trans,
    GstPadDirection direction, GstCaps * caps, GstCaps * filter_caps)
{
  GstMisbIrUnpack *filt = GST_MISB_IR_UNPACK (trans);
  GstStructure *structure, *newstruct;
  GstCaps *newcaps;
  guint i, n;

  GST_LOG_OBJECT (filt, "transforming caps from %" GST_PTR_FORMAT, caps);

  newcaps = gst_caps_new_empty ();
  n = gst_caps_get_size (caps);
  for (i = 0; i < n; ++i) {
    structure = gst_caps_get_structure (caps, i);
    if (direction == GST_PAD_SINK) {
      newstruct =
          gst_structure_new_from_string ("video/x-raw,format=GRAY16_LE");
    } else {
      newstruct =
          gst_structure_new_from_string ("video/x-raw,format={v210,UYVY}");
    }

    gst_structure_set_value (newstruct, "width",
        gst_structure_get_value (structure, "width"));
    gst_structure_set_value (newstruct, "height",
        gst_structure_get_value (structure, "height"));
    gst_structure_set_value (newstruct, "framerate",
        gst_structure_get_value (structure, "framerate"));

    gst_caps_append_structure (newcaps, newstruct);
  }


  if (!gst_caps_is_empty (newcaps) && filter_caps) {
    GstCaps *tmp = gst_caps_intersect_full (filter_caps, newcaps,
        GST_CAPS_INTERSECT_FIRST);
    gst_caps_replace (&newcaps, tmp);
    gst_caps_unref (tmp);
  }

  GST_LOG_OBJECT (filt, "transformed caps to %" GST_PTR_FORMAT, newcaps);

  return newcaps;
}

static gboolean
gst_misb_ir_unpack_set_info (GstVideoFilter * filter, GstCaps * incaps,
    GstVideoInfo * in_info, GstCaps * outcaps, GstVideoInfo * out_info)
{
  GstMisbIrUnpack *filt = GST_MISB_IR_UNPACK (filter);
  gboolean res = TRUE;

  GST_DEBUG_OBJECT (filt,
      "set_caps: in %" GST_PTR_FORMAT " out %" GST_PTR_FORMAT, incaps, outcaps);

  memcpy (&filt->info_in, in_info, sizeof (GstVideoInfo));
  memcpy (&filt->info_out, out_info, sizeof (GstVideoInfo));

  return res;
}

static GstFlowReturn
gst_misb_ir_unpack_transform_frame (GstVideoFilter * filter,
    GstVideoFrame * in_frame, GstVideoFrame * out_frame)
{
  GstMisbIrUnpack *filt = GST_MISB_IR_UNPACK (filter);
  GTimer *timer = NULL;
  gint16 offset = filt->offset_value;
  guint shift = filt->shift_value;
  gint x, y;
  guint16 *dst;

  GST_LOG_OBJECT (filt, "Performing non-inplace transform");

#if 0
  timer = g_timer_new ();
#endif

  if (filt->info_in.finfo->format == GST_VIDEO_FORMAT_v210) {
    for (y = 0; y < GST_VIDEO_FRAME_COMP_HEIGHT (in_frame, 0); y++) {
      guint32 *src = (guint32 *) (GST_VIDEO_FRAME_COMP_DATA (in_frame, 0) +
          y * GST_VIDEO_FRAME_COMP_STRIDE (in_frame, 0));
      dst = (guint16 *) (GST_VIDEO_FRAME_COMP_DATA (out_frame, 0) +
          y * GST_VIDEO_FRAME_COMP_STRIDE (out_frame, 0));
      for (x = 0; x < GST_VIDEO_FRAME_COMP_WIDTH (in_frame, 0);) {
        guint32 word0 = *src++;
        guint32 word1 = *src++;
        guint16 luma, chroma, temp;

        chroma = word0 & 0x3ff;
        luma = (word0 & 0xffc00) >> 10;
        if (filt->swap) {
          temp = chroma;
          chroma = luma;
          luma = temp;
        }
        dst[x++] =
            ((chroma + offset) & filt->chroma_mask) | (((luma +
                    offset) & filt->luma_mask) << shift);

        chroma = (word0 & 0x3ff00000) >> 20;
        luma = word1 & 0x3ff;
        if (filt->swap) {
          temp = chroma;
          chroma = luma;
          luma = temp;
        }
        dst[x++] =
            (chroma + offset) & filt->chroma_mask | ((luma +
                offset) & filt->luma_mask) << shift;

        chroma = (word1 & 0xffc00) >> 10;
        luma = (word1 & 0x3ff00000) >> 20;
        if (filt->swap) {
          temp = chroma;
          chroma = luma;
          luma = temp;
        }
        dst[x++] =
            (chroma + offset) & filt->chroma_mask | ((luma +
                offset) & filt->luma_mask) << shift;
      }
    }
  } else if (filt->info_in.finfo->format == GST_VIDEO_FORMAT_UYVY) {
    for (y = 0; y < GST_VIDEO_FRAME_COMP_HEIGHT (in_frame, 0); y++) {
      guint8 *src = (guint8 *) (GST_VIDEO_FRAME_COMP_DATA (in_frame, 0) +
          y * GST_VIDEO_FRAME_COMP_STRIDE (in_frame, 0));
      dst = (guint16 *) (GST_VIDEO_FRAME_COMP_DATA (out_frame, 0) +
          y * GST_VIDEO_FRAME_COMP_STRIDE (out_frame, 0));
      for (x = 0; x < GST_VIDEO_FRAME_COMP_WIDTH (in_frame, 0);) {
        guint8 chroma = *src++;
        guint8 luma = *src++;
        guint8 temp;

        if (filt->swap) {
          temp = chroma;
          chroma = luma;
          luma = temp;
        }
        dst[x++] =
            ((chroma + offset) & filt->chroma_mask) | (((luma +
                    offset) & filt->luma_mask) << shift);
      }
    }
  }
#if 0
  GST_LOG_OBJECT (filt, "Processing took %.3f ms", g_timer_elapsed (timer,
          NULL) * 1000);
  g_timer_destroy (timer);
#endif

  return GST_FLOW_OK;
}


static void
gst_misb_ir_unpack_reset (GstMisbIrUnpack * misb_ir_unpack)
{
  gst_video_info_init (&misb_ir_unpack->info_in);
  gst_video_info_init (&misb_ir_unpack->info_out);
}