Blame gst/encoding/gstsmartencoder.c

Packit 971217
/* GStreamer Smart Video Encoder element
Packit 971217
 * Copyright (C) <2010> Edward Hervey <bilboed@gmail.com>
Packit 971217
 *
Packit 971217
 * This library is free software; you can redistribute it and/or
Packit 971217
 * modify it under the terms of the GNU Library General Public
Packit 971217
 * License as published by the Free Software Foundation; either
Packit 971217
 * version 2 of the License, or (at your option) any later version.
Packit 971217
 *
Packit 971217
 * This library is distributed in the hope that it will be useful,
Packit 971217
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 971217
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Packit 971217
 * Library General Public License for more details.
Packit 971217
 *
Packit 971217
 * You should have received a copy of the GNU Library General Public
Packit 971217
 * License along with this library; if not, write to the
Packit 971217
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
Packit 971217
 * Boston, MA 02110-1301, USA.
Packit 971217
 */
Packit 971217
Packit 971217
/* TODO:
Packit 971217
 * * Implement get_caps/set_caps (store/forward caps)
Packit 971217
 * * Adjust template caps to the formats we can support
Packit 971217
 **/
Packit 971217
Packit 971217
#ifdef HAVE_CONFIG_H
Packit 971217
#include "config.h"
Packit 971217
#endif
Packit 971217
Packit 971217
#include <string.h>
Packit 971217
#include "gstsmartencoder.h"
Packit 971217
Packit 971217
GST_DEBUG_CATEGORY_STATIC (smart_encoder_debug);
Packit 971217
#define GST_CAT_DEFAULT smart_encoder_debug
Packit 971217
Packit 971217
/* FIXME : Update this with new caps */
Packit 971217
/* WARNING : We can only allow formats with closed-GOP */
Packit 971217
#define ALLOWED_CAPS "video/x-h263;video/x-intel-h263;"\
Packit 971217
  "video/mpeg,mpegversion=(int)1,systemstream=(boolean)false;"\
Packit 971217
  "video/mpeg,mpegversion=(int)2,systemstream=(boolean)false;"
Packit 971217
Packit 971217
static GstStaticPadTemplate src_template =
Packit 971217
GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC,
Packit 971217
    GST_PAD_ALWAYS,
Packit 971217
    GST_STATIC_CAPS (ALLOWED_CAPS)
Packit 971217
    );
Packit 971217
Packit 971217
static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink",
Packit 971217
    GST_PAD_SINK,
Packit 971217
    GST_PAD_ALWAYS,
Packit 971217
    GST_STATIC_CAPS (ALLOWED_CAPS)
Packit 971217
    );
Packit 971217
Packit 971217
static GQuark INTERNAL_ELEMENT;
Packit 971217
Packit 971217
/* GstSmartEncoder signals and args */
Packit 971217
enum
Packit 971217
{
Packit 971217
  /* FILL ME */
Packit 971217
  LAST_SIGNAL
Packit 971217
};
Packit 971217
Packit 971217
enum
Packit 971217
{
Packit 971217
  PROP_0
Packit 971217
      /* FILL ME */
Packit 971217
};
Packit 971217
Packit 971217
static void
Packit 971217
_do_init (void)
Packit 971217
{
Packit 971217
  INTERNAL_ELEMENT = g_quark_from_static_string ("internal-element");
Packit 971217
};
Packit 971217
Packit 971217
G_DEFINE_TYPE_EXTENDED (GstSmartEncoder, gst_smart_encoder, GST_TYPE_ELEMENT, 0,
Packit 971217
    _do_init ());
Packit 971217
Packit 971217
static void gst_smart_encoder_dispose (GObject * object);
Packit 971217
Packit 971217
static gboolean setup_recoder_pipeline (GstSmartEncoder * smart_encoder);
Packit 971217
Packit 971217
static GstFlowReturn gst_smart_encoder_chain (GstPad * pad, GstObject * parent,
Packit 971217
    GstBuffer * buf);
Packit 971217
static gboolean smart_encoder_sink_event (GstPad * pad, GstObject * parent,
Packit 971217
    GstEvent * event);
Packit 971217
static gboolean smart_encoder_sink_query (GstPad * pad, GstObject * parent,
Packit 971217
    GstQuery * query);
Packit 971217
static GstCaps *smart_encoder_sink_getcaps (GstPad * pad, GstCaps * filter);
Packit 971217
static GstStateChangeReturn
Packit 971217
gst_smart_encoder_change_state (GstElement * element,
Packit 971217
    GstStateChange transition);
Packit 971217
Packit 971217
static void
Packit 971217
gst_smart_encoder_class_init (GstSmartEncoderClass * klass)
Packit 971217
{
Packit 971217
  GObjectClass *gobject_class;
Packit 971217
  GstElementClass *element_class;
Packit 971217
Packit 971217
  element_class = (GstElementClass *) klass;
Packit 971217
  gobject_class = G_OBJECT_CLASS (klass);
Packit 971217
Packit 971217
  gst_smart_encoder_parent_class = g_type_class_peek_parent (klass);
Packit 971217
Packit 971217
  gst_element_class_add_static_pad_template (element_class, &src_template);
Packit 971217
  gst_element_class_add_static_pad_template (element_class, &sink_template);
Packit 971217
Packit 971217
  gst_element_class_set_static_metadata (element_class, "Smart Video Encoder",
Packit 971217
      "Codec/Recoder/Video",
Packit 971217
      "Re-encodes portions of Video that lay on segment boundaries",
Packit 971217
      "Edward Hervey <bilboed@gmail.com>");
Packit 971217
Packit 971217
  gobject_class->dispose = (GObjectFinalizeFunc) (gst_smart_encoder_dispose);
Packit 971217
  element_class->change_state = gst_smart_encoder_change_state;
Packit 971217
Packit 971217
  GST_DEBUG_CATEGORY_INIT (smart_encoder_debug, "smartencoder", 0,
Packit 971217
      "Smart Encoder");
Packit 971217
}
Packit 971217
Packit 971217
static void
Packit 971217
smart_encoder_reset (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  gst_segment_init (smart_encoder->segment, GST_FORMAT_UNDEFINED);
Packit 971217
Packit 971217
  if (smart_encoder->encoder) {
Packit 971217
    /* Clean up/remove elements */
Packit 971217
    gst_element_set_state (smart_encoder->encoder, GST_STATE_NULL);
Packit 971217
    gst_element_set_state (smart_encoder->decoder, GST_STATE_NULL);
Packit 971217
    gst_element_set_bus (smart_encoder->encoder, NULL);
Packit 971217
    gst_element_set_bus (smart_encoder->decoder, NULL);
Packit 971217
    gst_pad_set_active (smart_encoder->internal_srcpad, FALSE);
Packit 971217
    gst_pad_set_active (smart_encoder->internal_sinkpad, FALSE);
Packit 971217
    gst_object_unref (smart_encoder->encoder);
Packit 971217
    gst_object_unref (smart_encoder->decoder);
Packit 971217
    gst_object_unref (smart_encoder->internal_srcpad);
Packit 971217
    gst_object_unref (smart_encoder->internal_sinkpad);
Packit 971217
Packit 971217
    smart_encoder->encoder = NULL;
Packit 971217
    smart_encoder->decoder = NULL;
Packit 971217
    smart_encoder->internal_sinkpad = NULL;
Packit 971217
    smart_encoder->internal_srcpad = NULL;
Packit 971217
  }
Packit 971217
Packit 971217
  if (smart_encoder->newsegment) {
Packit 971217
    gst_event_unref (smart_encoder->newsegment);
Packit 971217
    smart_encoder->newsegment = NULL;
Packit 971217
  }
Packit 971217
}
Packit 971217
Packit 971217
Packit 971217
static void
Packit 971217
gst_smart_encoder_init (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  smart_encoder->sinkpad =
Packit 971217
      gst_pad_new_from_static_template (&sink_template, "sink");
Packit 971217
  gst_pad_set_chain_function (smart_encoder->sinkpad, gst_smart_encoder_chain);
Packit 971217
  gst_pad_set_event_function (smart_encoder->sinkpad, smart_encoder_sink_event);
Packit 971217
  gst_pad_set_query_function (smart_encoder->sinkpad, smart_encoder_sink_query);
Packit 971217
  gst_element_add_pad (GST_ELEMENT (smart_encoder), smart_encoder->sinkpad);
Packit 971217
Packit 971217
  smart_encoder->srcpad =
Packit 971217
      gst_pad_new_from_static_template (&src_template, "src");
Packit 971217
  gst_pad_use_fixed_caps (smart_encoder->srcpad);
Packit 971217
  gst_element_add_pad (GST_ELEMENT (smart_encoder), smart_encoder->srcpad);
Packit 971217
Packit 971217
  smart_encoder->segment = gst_segment_new ();
Packit 971217
Packit 971217
  smart_encoder_reset (smart_encoder);
Packit 971217
}
Packit 971217
Packit 971217
void
Packit 971217
gst_smart_encoder_dispose (GObject * object)
Packit 971217
{
Packit 971217
  GstSmartEncoder *smart_encoder = (GstSmartEncoder *) object;
Packit 971217
Packit 971217
  if (smart_encoder->segment)
Packit 971217
    gst_segment_free (smart_encoder->segment);
Packit 971217
  smart_encoder->segment = NULL;
Packit 971217
  if (smart_encoder->available_caps)
Packit 971217
    gst_caps_unref (smart_encoder->available_caps);
Packit 971217
  smart_encoder->available_caps = NULL;
Packit 971217
  G_OBJECT_CLASS (gst_smart_encoder_parent_class)->dispose (object);
Packit 971217
}
Packit 971217
Packit 971217
static GstFlowReturn
Packit 971217
gst_smart_encoder_reencode_gop (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  GstFlowReturn res = GST_FLOW_OK;
Packit 971217
  GList *tmp;
Packit 971217
Packit 971217
  if (smart_encoder->encoder == NULL) {
Packit 971217
    if (!setup_recoder_pipeline (smart_encoder))
Packit 971217
      return GST_FLOW_ERROR;
Packit 971217
  }
Packit 971217
Packit 971217
  /* Activate elements */
Packit 971217
  /* Set elements to PAUSED */
Packit 971217
  gst_element_set_state (smart_encoder->encoder, GST_STATE_PAUSED);
Packit 971217
  gst_element_set_state (smart_encoder->decoder, GST_STATE_PAUSED);
Packit 971217
Packit 971217
  GST_INFO ("Pushing Flush start/stop to clean decoder/encoder");
Packit 971217
  gst_pad_push_event (smart_encoder->internal_srcpad,
Packit 971217
      gst_event_new_flush_start ());
Packit 971217
  gst_pad_push_event (smart_encoder->internal_srcpad,
Packit 971217
      gst_event_new_flush_stop (TRUE));
Packit 971217
Packit 971217
  /* push newsegment */
Packit 971217
  GST_INFO ("Pushing newsegment %" GST_PTR_FORMAT, smart_encoder->newsegment);
Packit 971217
  gst_pad_push_event (smart_encoder->internal_srcpad,
Packit 971217
      gst_event_ref (smart_encoder->newsegment));
Packit 971217
Packit 971217
  /* Push buffers through our pads */
Packit 971217
  GST_DEBUG ("Pushing pending buffers");
Packit 971217
Packit 971217
  for (tmp = smart_encoder->pending_gop; tmp; tmp = tmp->next) {
Packit 971217
    GstBuffer *buf = (GstBuffer *) tmp->data;
Packit 971217
Packit 971217
    res = gst_pad_push (smart_encoder->internal_srcpad, buf);
Packit 971217
    if (G_UNLIKELY (res != GST_FLOW_OK))
Packit 971217
      break;
Packit 971217
  }
Packit 971217
Packit 971217
  if (G_UNLIKELY (res != GST_FLOW_OK)) {
Packit 971217
    GST_WARNING ("Error pushing pending buffers : %s", gst_flow_get_name (res));
Packit 971217
    /* Remove pending bfufers */
Packit 971217
    for (tmp = smart_encoder->pending_gop; tmp; tmp = tmp->next) {
Packit 971217
      gst_buffer_unref ((GstBuffer *) tmp->data);
Packit 971217
    }
Packit 971217
  } else {
Packit 971217
    GST_INFO ("Pushing out EOS to flush out decoder/encoder");
Packit 971217
    gst_pad_push_event (smart_encoder->internal_srcpad, gst_event_new_eos ());
Packit 971217
  }
Packit 971217
Packit 971217
  /* Activate elements */
Packit 971217
  /* Set elements to PAUSED */
Packit 971217
  gst_element_set_state (smart_encoder->encoder, GST_STATE_NULL);
Packit 971217
  gst_element_set_state (smart_encoder->decoder, GST_STATE_NULL);
Packit 971217
Packit 971217
  g_list_free (smart_encoder->pending_gop);
Packit 971217
  smart_encoder->pending_gop = NULL;
Packit 971217
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static GstFlowReturn
Packit 971217
gst_smart_encoder_push_pending_gop (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  guint64 cstart, cstop;
Packit 971217
  GList *tmp;
Packit 971217
  GstFlowReturn res = GST_FLOW_OK;
Packit 971217
Packit 971217
  GST_DEBUG ("Pushing pending GOP (%" GST_TIME_FORMAT " -- %" GST_TIME_FORMAT
Packit 971217
      ")", GST_TIME_ARGS (smart_encoder->gop_start),
Packit 971217
      GST_TIME_ARGS (smart_encoder->gop_stop));
Packit 971217
Packit 971217
  /* If GOP is entirely within segment, just push downstream */
Packit 971217
  if (gst_segment_clip (smart_encoder->segment, GST_FORMAT_TIME,
Packit 971217
          smart_encoder->gop_start, smart_encoder->gop_stop, &cstart, &cstop)) {
Packit 971217
    if ((cstart != smart_encoder->gop_start)
Packit 971217
        || (cstop != smart_encoder->gop_stop)) {
Packit 971217
      GST_DEBUG ("GOP needs to be re-encoded from %" GST_TIME_FORMAT " to %"
Packit 971217
          GST_TIME_FORMAT, GST_TIME_ARGS (cstart), GST_TIME_ARGS (cstop));
Packit 971217
      res = gst_smart_encoder_reencode_gop (smart_encoder);
Packit 971217
    } else {
Packit 971217
      /* The whole GOP is within the segment, push all pending buffers downstream */
Packit 971217
      GST_DEBUG ("GOP doesn't need to be modified, pushing downstream");
Packit 971217
      for (tmp = smart_encoder->pending_gop; tmp; tmp = tmp->next) {
Packit 971217
        GstBuffer *buf = (GstBuffer *) tmp->data;
Packit 971217
        res = gst_pad_push (smart_encoder->srcpad, buf);
Packit 971217
        if (G_UNLIKELY (res != GST_FLOW_OK))
Packit 971217
          break;
Packit 971217
      }
Packit 971217
    }
Packit 971217
  } else {
Packit 971217
    /* The whole GOP is outside the segment, there's most likely
Packit 971217
     * a bug somewhere. */
Packit 971217
    GST_WARNING
Packit 971217
        ("GOP is entirely outside of the segment, upstream gave us too much data");
Packit 971217
    for (tmp = smart_encoder->pending_gop; tmp; tmp = tmp->next) {
Packit 971217
      gst_buffer_unref ((GstBuffer *) tmp->data);
Packit 971217
    }
Packit 971217
  }
Packit 971217
Packit 971217
  if (smart_encoder->pending_gop) {
Packit 971217
    g_list_free (smart_encoder->pending_gop);
Packit 971217
    smart_encoder->pending_gop = NULL;
Packit 971217
  }
Packit 971217
  smart_encoder->gop_start = GST_CLOCK_TIME_NONE;
Packit 971217
  smart_encoder->gop_stop = GST_CLOCK_TIME_NONE;
Packit 971217
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static GstFlowReturn
Packit 971217
gst_smart_encoder_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
Packit 971217
{
Packit 971217
  GstSmartEncoder *smart_encoder;
Packit 971217
  GstFlowReturn res = GST_FLOW_OK;
Packit 971217
  gboolean discont, keyframe;
Packit 971217
Packit 971217
  smart_encoder = GST_SMART_ENCODER (parent);
Packit 971217
Packit 971217
  discont = GST_BUFFER_IS_DISCONT (buf);
Packit 971217
  keyframe = !GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT);
Packit 971217
Packit 971217
  GST_DEBUG ("New buffer %s %s %" GST_TIME_FORMAT,
Packit 971217
      discont ? "discont" : "",
Packit 971217
      keyframe ? "keyframe" : "", GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)));
Packit 971217
Packit 971217
  if (keyframe) {
Packit 971217
    GST_DEBUG ("Got a keyframe");
Packit 971217
Packit 971217
    /* If there's a pending GOP, flush it out */
Packit 971217
    if (smart_encoder->pending_gop) {
Packit 971217
      /* Mark gop_stop */
Packit 971217
      smart_encoder->gop_stop = GST_BUFFER_TIMESTAMP (buf);
Packit 971217
Packit 971217
      /* flush pending */
Packit 971217
      res = gst_smart_encoder_push_pending_gop (smart_encoder);
Packit 971217
      if (G_UNLIKELY (res != GST_FLOW_OK))
Packit 971217
        goto beach;
Packit 971217
    }
Packit 971217
Packit 971217
    /* Mark gop_start for new gop */
Packit 971217
    smart_encoder->gop_start = GST_BUFFER_TIMESTAMP (buf);
Packit 971217
  }
Packit 971217
Packit 971217
  /* Store buffer */
Packit 971217
  smart_encoder->pending_gop = g_list_append (smart_encoder->pending_gop, buf);
Packit 971217
  /* Update GOP stop position */
Packit 971217
  if (GST_BUFFER_TIMESTAMP_IS_VALID (buf)) {
Packit 971217
    smart_encoder->gop_stop = GST_BUFFER_TIMESTAMP (buf);
Packit 971217
    if (GST_BUFFER_DURATION_IS_VALID (buf))
Packit 971217
      smart_encoder->gop_stop += GST_BUFFER_DURATION (buf);
Packit 971217
  }
Packit 971217
Packit 971217
  GST_DEBUG ("Buffer stored , Current GOP : %" GST_TIME_FORMAT " -- %"
Packit 971217
      GST_TIME_FORMAT, GST_TIME_ARGS (smart_encoder->gop_start),
Packit 971217
      GST_TIME_ARGS (smart_encoder->gop_stop));
Packit 971217
Packit 971217
beach:
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static gboolean
Packit 971217
smart_encoder_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
Packit 971217
{
Packit 971217
  gboolean res = TRUE;
Packit 971217
  GstSmartEncoder *smart_encoder = GST_SMART_ENCODER (parent);
Packit 971217
Packit 971217
  switch (GST_EVENT_TYPE (event)) {
Packit 971217
    case GST_EVENT_FLUSH_STOP:
Packit 971217
      smart_encoder_reset (smart_encoder);
Packit 971217
      break;
Packit 971217
    case GST_EVENT_SEGMENT:
Packit 971217
    {
Packit 971217
      gst_event_copy_segment (event, smart_encoder->segment);
Packit 971217
Packit 971217
      GST_DEBUG_OBJECT (smart_encoder, "segment: %" GST_SEGMENT_FORMAT,
Packit 971217
          smart_encoder->segment);
Packit 971217
      if (smart_encoder->segment->format != GST_FORMAT_TIME) {
Packit 971217
        GST_ERROR
Packit 971217
            ("smart_encoder can not handle streams not specified in GST_FORMAT_TIME");
Packit 971217
        gst_event_unref (event);
Packit 971217
        return FALSE;
Packit 971217
      }
Packit 971217
Packit 971217
      /* And keep a copy for further usage */
Packit 971217
      if (smart_encoder->newsegment)
Packit 971217
        gst_event_unref (smart_encoder->newsegment);
Packit 971217
      smart_encoder->newsegment = gst_event_ref (event);
Packit 971217
    }
Packit 971217
      break;
Packit 971217
    case GST_EVENT_EOS:
Packit 971217
      GST_DEBUG ("Eos, flushing remaining data");
Packit 971217
      if (smart_encoder->segment->format == GST_FORMAT_TIME)
Packit 971217
        gst_smart_encoder_push_pending_gop (smart_encoder);
Packit 971217
      break;
Packit 971217
    default:
Packit 971217
      break;
Packit 971217
  }
Packit 971217
Packit 971217
  res = gst_pad_push_event (smart_encoder->srcpad, event);
Packit 971217
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static GstCaps *
Packit 971217
smart_encoder_sink_getcaps (GstPad * pad, GstCaps * filter)
Packit 971217
{
Packit 971217
  GstCaps *peer, *tmpl, *res;
Packit 971217
  GstSmartEncoder *smart_encoder = GST_SMART_ENCODER (gst_pad_get_parent (pad));
Packit 971217
Packit 971217
  /* Use computed caps */
Packit 971217
  if (smart_encoder->available_caps)
Packit 971217
    tmpl = gst_caps_ref (smart_encoder->available_caps);
Packit 971217
  else
Packit 971217
    tmpl = gst_static_pad_template_get_caps (&src_template);
Packit 971217
Packit 971217
  /* Try getting it from downstream */
Packit 971217
  peer = gst_pad_peer_query_caps (smart_encoder->srcpad, tmpl);
Packit 971217
Packit 971217
  if (peer == NULL) {
Packit 971217
    res = tmpl;
Packit 971217
  } else {
Packit 971217
    res = peer;
Packit 971217
    gst_caps_unref (tmpl);
Packit 971217
  }
Packit 971217
Packit 971217
  gst_object_unref (smart_encoder);
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static gboolean
Packit 971217
smart_encoder_sink_query (GstPad * pad, GstObject * parent, GstQuery * query)
Packit 971217
{
Packit 971217
  gboolean res;
Packit 971217
Packit 971217
  switch (GST_QUERY_TYPE (query)) {
Packit 971217
    case GST_QUERY_CAPS:
Packit 971217
    {
Packit 971217
      GstCaps *filter, *caps;
Packit 971217
Packit 971217
      gst_query_parse_caps (query, &filter);
Packit 971217
      caps = smart_encoder_sink_getcaps (pad, filter);
Packit 971217
      gst_query_set_caps_result (query, caps);
Packit 971217
      gst_caps_unref (caps);
Packit 971217
      res = TRUE;
Packit 971217
      break;
Packit 971217
    }
Packit 971217
    default:
Packit 971217
      res = gst_pad_query_default (pad, parent, query);
Packit 971217
      break;
Packit 971217
  }
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
/*****************************************
Packit 971217
 *    Internal encoder/decoder pipeline  *
Packit 971217
 ******************************************/
Packit 971217
Packit 971217
static GstElementFactory *
Packit 971217
get_decoder_factory (GstCaps * caps)
Packit 971217
{
Packit 971217
  GstElementFactory *fact = NULL;
Packit 971217
  GList *decoders, *tmp;
Packit 971217
Packit 971217
  tmp =
Packit 971217
      gst_element_factory_list_get_elements (GST_ELEMENT_FACTORY_TYPE_DECODER,
Packit 971217
      GST_RANK_MARGINAL);
Packit 971217
  decoders = gst_element_factory_list_filter (tmp, caps, GST_PAD_SINK, FALSE);
Packit 971217
  gst_plugin_feature_list_free (tmp);
Packit 971217
Packit 971217
  for (tmp = decoders; tmp; tmp = tmp->next) {
Packit 971217
    /* We just pick the first one */
Packit 971217
    fact = (GstElementFactory *) tmp->data;
Packit 971217
    gst_object_ref (fact);
Packit 971217
    break;
Packit 971217
  }
Packit 971217
Packit 971217
  gst_plugin_feature_list_free (decoders);
Packit 971217
Packit 971217
  return fact;
Packit 971217
}
Packit 971217
Packit 971217
static GstElementFactory *
Packit 971217
get_encoder_factory (GstCaps * caps)
Packit 971217
{
Packit 971217
  GstElementFactory *fact = NULL;
Packit 971217
  GList *encoders, *tmp;
Packit 971217
Packit 971217
  tmp =
Packit 971217
      gst_element_factory_list_get_elements (GST_ELEMENT_FACTORY_TYPE_ENCODER,
Packit 971217
      GST_RANK_MARGINAL);
Packit 971217
  encoders = gst_element_factory_list_filter (tmp, caps, GST_PAD_SRC, FALSE);
Packit 971217
  gst_plugin_feature_list_free (tmp);
Packit 971217
Packit 971217
  for (tmp = encoders; tmp; tmp = tmp->next) {
Packit 971217
    /* We just pick the first one */
Packit 971217
    fact = (GstElementFactory *) tmp->data;
Packit 971217
    gst_object_ref (fact);
Packit 971217
    break;
Packit 971217
  }
Packit 971217
Packit 971217
  gst_plugin_feature_list_free (encoders);
Packit 971217
Packit 971217
  return fact;
Packit 971217
}
Packit 971217
Packit 971217
static GstElement *
Packit 971217
get_decoder (GstCaps * caps)
Packit 971217
{
Packit 971217
  GstElementFactory *fact = get_decoder_factory (caps);
Packit 971217
  GstElement *res = NULL;
Packit 971217
Packit 971217
  if (fact) {
Packit 971217
    res = gst_element_factory_create (fact, "internal-decoder");
Packit 971217
    gst_object_unref (fact);
Packit 971217
  }
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static GstElement *
Packit 971217
get_encoder (GstCaps * caps)
Packit 971217
{
Packit 971217
  GstElementFactory *fact = get_encoder_factory (caps);
Packit 971217
  GstElement *res = NULL;
Packit 971217
Packit 971217
  if (fact) {
Packit 971217
    res = gst_element_factory_create (fact, "internal-encoder");
Packit 971217
    gst_object_unref (fact);
Packit 971217
  }
Packit 971217
  return res;
Packit 971217
}
Packit 971217
Packit 971217
static GstFlowReturn
Packit 971217
internal_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
Packit 971217
{
Packit 971217
  GstSmartEncoder *smart_encoder =
Packit 971217
      g_object_get_qdata ((GObject *) pad, INTERNAL_ELEMENT);
Packit 971217
Packit 971217
  return gst_pad_push (smart_encoder->srcpad, buf);
Packit 971217
}
Packit 971217
Packit 971217
static gboolean
Packit 971217
setup_recoder_pipeline (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  GstPad *tmppad;
Packit 971217
  GstCaps *caps;
Packit 971217
Packit 971217
  /* Fast path */
Packit 971217
  if (G_UNLIKELY (smart_encoder->encoder))
Packit 971217
    return TRUE;
Packit 971217
Packit 971217
  GST_DEBUG ("Creating internal decoder and encoder");
Packit 971217
Packit 971217
  /* Create decoder/encoder */
Packit 971217
  caps = gst_pad_get_current_caps (smart_encoder->sinkpad);
Packit 971217
  smart_encoder->decoder = get_decoder (caps);
Packit 971217
  if (G_UNLIKELY (smart_encoder->decoder == NULL))
Packit 971217
    goto no_decoder;
Packit 971217
  gst_caps_unref (caps);
Packit 971217
  gst_element_set_bus (smart_encoder->decoder, GST_ELEMENT_BUS (smart_encoder));
Packit 971217
Packit 971217
  caps = gst_pad_get_current_caps (smart_encoder->sinkpad);
Packit 971217
  smart_encoder->encoder = get_encoder (caps);
Packit 971217
  if (G_UNLIKELY (smart_encoder->encoder == NULL))
Packit 971217
    goto no_encoder;
Packit 971217
  gst_caps_unref (caps);
Packit 971217
  gst_element_set_bus (smart_encoder->encoder, GST_ELEMENT_BUS (smart_encoder));
Packit 971217
Packit 971217
  GST_DEBUG ("Creating internal pads");
Packit 971217
Packit 971217
  /* Create internal pads */
Packit 971217
Packit 971217
  /* Source pad which we'll use to feed data to decoders */
Packit 971217
  smart_encoder->internal_srcpad = gst_pad_new ("internal_src", GST_PAD_SRC);
Packit 971217
  g_object_set_qdata ((GObject *) smart_encoder->internal_srcpad,
Packit 971217
      INTERNAL_ELEMENT, smart_encoder);
Packit 971217
  gst_pad_set_active (smart_encoder->internal_srcpad, TRUE);
Packit 971217
Packit 971217
  /* Sink pad which will get the buffers from the encoder.
Packit 971217
   * Note: We don't need an event function since we'll be discarding all
Packit 971217
   * of them. */
Packit 971217
  smart_encoder->internal_sinkpad = gst_pad_new ("internal_sink", GST_PAD_SINK);
Packit 971217
  g_object_set_qdata ((GObject *) smart_encoder->internal_sinkpad,
Packit 971217
      INTERNAL_ELEMENT, smart_encoder);
Packit 971217
  gst_pad_set_chain_function (smart_encoder->internal_sinkpad, internal_chain);
Packit 971217
  gst_pad_set_active (smart_encoder->internal_sinkpad, TRUE);
Packit 971217
Packit 971217
  GST_DEBUG ("Linking pads to elements");
Packit 971217
Packit 971217
  /* Link everything */
Packit 971217
  tmppad = gst_element_get_static_pad (smart_encoder->encoder, "src");
Packit 971217
  if (GST_PAD_LINK_FAILED (gst_pad_link (tmppad,
Packit 971217
              smart_encoder->internal_sinkpad)))
Packit 971217
    goto sinkpad_link_fail;
Packit 971217
  gst_object_unref (tmppad);
Packit 971217
Packit 971217
  if (!gst_element_link (smart_encoder->decoder, smart_encoder->encoder))
Packit 971217
    goto encoder_decoder_link_fail;
Packit 971217
Packit 971217
  tmppad = gst_element_get_static_pad (smart_encoder->decoder, "sink");
Packit 971217
  if (GST_PAD_LINK_FAILED (gst_pad_link (smart_encoder->internal_srcpad,
Packit 971217
              tmppad)))
Packit 971217
    goto srcpad_link_fail;
Packit 971217
  gst_object_unref (tmppad);
Packit 971217
Packit 971217
  GST_DEBUG ("Done creating internal elements/pads");
Packit 971217
Packit 971217
  return TRUE;
Packit 971217
Packit 971217
no_decoder:
Packit 971217
  {
Packit 971217
    GST_WARNING ("Couldn't find a decoder for %" GST_PTR_FORMAT, caps);
Packit 971217
    gst_caps_unref (caps);
Packit 971217
    return FALSE;
Packit 971217
  }
Packit 971217
Packit 971217
no_encoder:
Packit 971217
  {
Packit 971217
    GST_WARNING ("Couldn't find an encoder for %" GST_PTR_FORMAT, caps);
Packit 971217
    gst_caps_unref (caps);
Packit 971217
    return FALSE;
Packit 971217
  }
Packit 971217
Packit 971217
srcpad_link_fail:
Packit 971217
  {
Packit 971217
    gst_object_unref (tmppad);
Packit 971217
    GST_WARNING ("Couldn't link internal srcpad to decoder");
Packit 971217
    return FALSE;
Packit 971217
  }
Packit 971217
Packit 971217
sinkpad_link_fail:
Packit 971217
  {
Packit 971217
    gst_object_unref (tmppad);
Packit 971217
    GST_WARNING ("Couldn't link encoder to internal sinkpad");
Packit 971217
    return FALSE;
Packit 971217
  }
Packit 971217
Packit 971217
encoder_decoder_link_fail:
Packit 971217
  {
Packit 971217
    GST_WARNING ("Couldn't link decoder to encoder");
Packit 971217
    return FALSE;
Packit 971217
  }
Packit 971217
}
Packit 971217
Packit 971217
static GstStateChangeReturn
Packit 971217
gst_smart_encoder_find_elements (GstSmartEncoder * smart_encoder)
Packit 971217
{
Packit 971217
  guint i, n;
Packit 971217
  GstCaps *tmpl, *st, *res;
Packit 971217
  GstElementFactory *dec, *enc;
Packit 971217
  GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
Packit 971217
Packit 971217
  if (G_UNLIKELY (smart_encoder->available_caps))
Packit 971217
    goto beach;
Packit 971217
Packit 971217
  /* Iterate over all pad template caps and see if we have both an
Packit 971217
   * encoder and a decoder for those media types */
Packit 971217
  tmpl = gst_static_pad_template_get_caps (&src_template);
Packit 971217
  res = gst_caps_new_empty ();
Packit 971217
  n = gst_caps_get_size (tmpl);
Packit 971217
Packit 971217
  for (i = 0; i < n; i++) {
Packit 971217
    st = gst_caps_copy_nth (tmpl, i);
Packit 971217
    GST_DEBUG_OBJECT (smart_encoder,
Packit 971217
        "Checking for available decoder and encoder for %" GST_PTR_FORMAT, st);
Packit 971217
    if (!(dec = get_decoder_factory (st))) {
Packit 971217
      gst_caps_unref (st);
Packit 971217
      continue;
Packit 971217
    }
Packit 971217
    gst_object_unref (dec);
Packit 971217
    if (!(enc = get_encoder_factory (st))) {
Packit 971217
      gst_caps_unref (st);
Packit 971217
      continue;
Packit 971217
    }
Packit 971217
    gst_object_unref (enc);
Packit 971217
    GST_DEBUG_OBJECT (smart_encoder, "OK");
Packit 971217
    gst_caps_append (res, st);
Packit 971217
  }
Packit 971217
Packit 971217
  gst_caps_unref (tmpl);
Packit 971217
Packit 971217
  if (gst_caps_is_empty (res)) {
Packit 971217
    gst_caps_unref (res);
Packit 971217
    ret = GST_STATE_CHANGE_FAILURE;
Packit 971217
  } else
Packit 971217
    smart_encoder->available_caps = res;
Packit 971217
Packit 971217
  GST_DEBUG_OBJECT (smart_encoder, "Done, available_caps:%" GST_PTR_FORMAT,
Packit 971217
      smart_encoder->available_caps);
Packit 971217
Packit 971217
beach:
Packit 971217
  return ret;
Packit 971217
}
Packit 971217
Packit 971217
/******************************************
Packit 971217
 *    GstElement vmethod implementations  *
Packit 971217
 ******************************************/
Packit 971217
Packit 971217
static GstStateChangeReturn
Packit 971217
gst_smart_encoder_change_state (GstElement * element, GstStateChange transition)
Packit 971217
{
Packit 971217
  GstSmartEncoder *smart_encoder;
Packit 971217
  GstStateChangeReturn ret;
Packit 971217
Packit 971217
  g_return_val_if_fail (GST_IS_SMART_ENCODER (element),
Packit 971217
      GST_STATE_CHANGE_FAILURE);
Packit 971217
Packit 971217
  smart_encoder = GST_SMART_ENCODER (element);
Packit 971217
Packit 971217
  switch (transition) {
Packit 971217
    case GST_STATE_CHANGE_NULL_TO_READY:
Packit 971217
      /* Figure out which elements are available  */
Packit 971217
      if ((ret =
Packit 971217
              gst_smart_encoder_find_elements (smart_encoder)) ==
Packit 971217
          GST_STATE_CHANGE_FAILURE)
Packit 971217
        goto beach;
Packit 971217
      break;
Packit 971217
    default:
Packit 971217
      break;
Packit 971217
  }
Packit 971217
Packit 971217
  ret =
Packit 971217
      GST_ELEMENT_CLASS (gst_smart_encoder_parent_class)->change_state (element,
Packit 971217
      transition);
Packit 971217
Packit 971217
  switch (transition) {
Packit 971217
    case GST_STATE_CHANGE_PAUSED_TO_READY:
Packit 971217
      smart_encoder_reset (smart_encoder);
Packit 971217
      break;
Packit 971217
    default:
Packit 971217
      break;
Packit 971217
  }
Packit 971217
Packit 971217
beach:
Packit 971217
  return ret;
Packit 971217
}