/*

 * GStreamer

 * Copyright (C) 2015 Matthew Waters <matthew@centricular.com>

 *

 * 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., 51 Franklin St, Fifth Floor,

 * Boston, MA 02110-1301, USA.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <stdio.h>

#include <gst/gl/gl.h>

#include "gstgluploadelement.h"

GST_DEBUG_CATEGORY_STATIC (gst_gl_upload_element_debug);

#define GST_CAT_DEFAULT gst_gl_upload_element_debug

#define gst_gl_upload_element_parent_class parent_class

G_DEFINE_TYPE_WITH_CODE (GstGLUploadElement, gst_gl_upload_element,

    GST_TYPE_GL_BASE_FILTER,

    GST_DEBUG_CATEGORY_INIT (gst_gl_upload_element_debug, "gluploadelement", 0,

        "glupload Element"););

static gboolean gst_gl_upload_element_get_unit_size (GstBaseTransform * trans,

    GstCaps * caps, gsize * size);

static GstCaps *_gst_gl_upload_element_transform_caps (GstBaseTransform * bt,

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

static gboolean _gst_gl_upload_element_set_caps (GstBaseTransform * bt,

    GstCaps * in_caps, GstCaps * out_caps);

static gboolean gst_gl_upload_element_filter_meta (GstBaseTransform * trans,

    GstQuery * query, GType api, const GstStructure * params);

static gboolean _gst_gl_upload_element_propose_allocation (GstBaseTransform *

    bt, GstQuery * decide_query, GstQuery * query);

static gboolean _gst_gl_upload_element_decide_allocation (GstBaseTransform *

    trans, GstQuery * query);

static GstFlowReturn

gst_gl_upload_element_prepare_output_buffer (GstBaseTransform * bt,

    GstBuffer * buffer, GstBuffer ** outbuf);

static GstFlowReturn gst_gl_upload_element_transform (GstBaseTransform * bt,

    GstBuffer * buffer, GstBuffer * outbuf);

static gboolean gst_gl_upload_element_stop (GstBaseTransform * bt);

static GstStaticPadTemplate gst_gl_upload_element_src_pad_template =

GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("video/x-raw(ANY)"));

static void

gst_gl_upload_element_finalize (GObject * object)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (object);

  if (upload->upload)

    gst_object_unref (upload->upload);

  upload->upload = NULL;

  G_OBJECT_CLASS (parent_class)->finalize (object);

}

static void

gst_gl_upload_element_class_init (GstGLUploadElementClass * klass)

{

  GstBaseTransformClass *bt_class = GST_BASE_TRANSFORM_CLASS (klass);

  GstElementClass *element_class = GST_ELEMENT_CLASS (klass);

  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

  GstCaps *upload_caps;

  bt_class->transform_caps = _gst_gl_upload_element_transform_caps;

  bt_class->set_caps = _gst_gl_upload_element_set_caps;

  bt_class->filter_meta = gst_gl_upload_element_filter_meta;

  bt_class->propose_allocation = _gst_gl_upload_element_propose_allocation;

  bt_class->decide_allocation = _gst_gl_upload_element_decide_allocation;

  bt_class->get_unit_size = gst_gl_upload_element_get_unit_size;

  bt_class->prepare_output_buffer = gst_gl_upload_element_prepare_output_buffer;

  bt_class->transform = gst_gl_upload_element_transform;

  bt_class->stop = gst_gl_upload_element_stop;

  bt_class->passthrough_on_same_caps = TRUE;

  gst_element_class_add_static_pad_template (element_class,

      &gst_gl_upload_element_src_pad_template);

  upload_caps = gst_gl_upload_get_input_template_caps ();

  gst_element_class_add_pad_template (element_class,

      gst_pad_template_new ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, upload_caps));

  gst_caps_unref (upload_caps);

  gst_element_class_set_metadata (element_class,

      "OpenGL uploader", "Filter/Video",

      "Uploads data into OpenGL", "Matthew Waters <matthew@centricular.com>");

  gobject_class->finalize = gst_gl_upload_element_finalize;

}

static void

gst_gl_upload_element_init (GstGLUploadElement * upload)

{

  gst_base_transform_set_prefer_passthrough (GST_BASE_TRANSFORM (upload), TRUE);

}

static gboolean

gst_gl_upload_element_stop (GstBaseTransform * bt)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);

  if (upload->upload) {

    gst_object_unref (upload->upload);

    upload->upload = NULL;

  }

  gst_caps_replace (&upload->in_caps, NULL);

  gst_caps_replace (&upload->out_caps, NULL);

  return GST_BASE_TRANSFORM_CLASS (parent_class)->stop (bt);

}

static gboolean

gst_gl_upload_element_get_unit_size (GstBaseTransform * trans, GstCaps * caps,

    gsize * size)

{

  gboolean ret = FALSE;

  GstVideoInfo info;

  ret = gst_video_info_from_caps (&info, caps);

  if (ret)

    *size = GST_VIDEO_INFO_SIZE (&info;;

  return TRUE;

}

static GstCaps *

_gst_gl_upload_element_transform_caps (GstBaseTransform * bt,

    GstPadDirection direction, GstCaps * caps, GstCaps * filter)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);

  GstGLContext *context = GST_GL_BASE_FILTER (bt)->context;

  if (upload->upload == NULL)

    upload->upload = gst_gl_upload_new (NULL);

  return gst_gl_upload_transform_caps (upload->upload, context, direction, caps,

      filter);

}

static gboolean

gst_gl_upload_element_filter_meta (GstBaseTransform * trans, GstQuery * query,

    GType api, const GstStructure * params)

{

  /* propose all metadata upstream */

  return TRUE;

}

static gboolean

_gst_gl_upload_element_propose_allocation (GstBaseTransform * bt,

    GstQuery * decide_query, GstQuery * query)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);

  gboolean ret;

  if (!upload->upload)

    return FALSE;

  ret = GST_BASE_TRANSFORM_CLASS (parent_class)->propose_allocation (bt,

      decide_query, query);

  gst_gl_upload_propose_allocation (upload->upload, decide_query, query);

  return ret;

}

static gboolean

_gst_gl_upload_element_decide_allocation (GstBaseTransform * trans,

    GstQuery * query)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (trans);

  GstGLContext *context;

  gboolean ret;

  ret =

      GST_BASE_TRANSFORM_CLASS

      (gst_gl_upload_element_parent_class)->decide_allocation (trans, query);

  if (!ret)

    return FALSE;

  /* GstGLBaseFilter populates ->context in ::decide_allocation so now it's the

   * time to set the ->upload context */

  context = GST_GL_BASE_FILTER (trans)->context;

  gst_gl_upload_set_context (upload->upload, context);

  return gst_gl_upload_set_caps (upload->upload, upload->in_caps,

      upload->out_caps);

}

static gboolean

_gst_gl_upload_element_set_caps (GstBaseTransform * bt, GstCaps * in_caps,

    GstCaps * out_caps)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);

  gst_caps_replace (&upload->in_caps, in_caps);

  gst_caps_replace (&upload->out_caps, out_caps);

  if (upload->upload)

    return gst_gl_upload_set_caps (upload->upload, in_caps, out_caps);

  return TRUE;

}

GstFlowReturn

gst_gl_upload_element_prepare_output_buffer (GstBaseTransform * bt,

    GstBuffer * buffer, GstBuffer ** outbuf)

{

  GstGLUploadElement *upload = GST_GL_UPLOAD_ELEMENT (bt);

  GstGLUploadReturn ret;

  GstBaseTransformClass *bclass;

  bclass = GST_BASE_TRANSFORM_GET_CLASS (bt);

  if (gst_base_transform_is_passthrough (bt)) {

    *outbuf = buffer;

    return GST_FLOW_OK;

  }

  if (!upload->upload)

    return GST_FLOW_NOT_NEGOTIATED;

  ret = gst_gl_upload_perform_with_buffer (upload->upload, buffer, outbuf);

  if (ret == GST_GL_UPLOAD_RECONFIGURE) {

    gst_base_transform_reconfigure_src (bt);

    return GST_FLOW_OK;

  }

  if (ret != GST_GL_UPLOAD_DONE || *outbuf == NULL) {

    GST_ELEMENT_ERROR (bt, RESOURCE, NOT_FOUND, ("%s",

            "Failed to upload buffer"), (NULL));

    if (*outbuf)

      gst_buffer_unref (*outbuf);

    return GST_FLOW_ERROR;

  }

  /* basetransform doesn't unref if they're the same */

  if (buffer == *outbuf)

    gst_buffer_unref (*outbuf);

  else

    bclass->copy_metadata (bt, buffer, *outbuf);

  return GST_FLOW_OK;

}

static GstFlowReturn

gst_gl_upload_element_transform (GstBaseTransform * bt, GstBuffer * buffer,

    GstBuffer * outbuf)

{

  return GST_FLOW_OK;

}