gst/rtp/gstrtpmp4vdepay.c

   1 /* GStreamer
   2  * Copyright (C) <2005> Wim Taymans <wim@fluendo.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Library General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Library General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Library General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #  include "config.h"
  22 #endif
  23
  24 #include <gst/rtp/gstrtpbuffer.h>
  25
  26 #include <string.h>
  27 #include "gstrtpmp4vdepay.h"
  28
  29 GST_DEBUG_CATEGORY_STATIC (rtpmp4vdepay_debug);
  30 #define GST_CAT_DEFAULT (rtpmp4vdepay_debug)
  31
  32 /* elementfactory information */
  33 static const GstElementDetails gst_rtp_mp4vdepay_details =
  34 GST_ELEMENT_DETAILS ("RTP packet depayloader",
  35     "Codec/Depayloader/Network",
  36     "Extracts MPEG4 video from RTP packets (RFC 3016)",
  37     "Wim Taymans <wim@fluendo.com>");
  38
  39 static GstStaticPadTemplate gst_rtp_mp4v_depay_src_template =
  40 GST_STATIC_PAD_TEMPLATE ("src",
  41     GST_PAD_SRC,
  42     GST_PAD_ALWAYS,
  43     GST_STATIC_CAPS ("video/mpeg,"
  44         "mpegversion=(int) 4," "systemstream=(boolean)false")
  45     );
  46
  47 static GstStaticPadTemplate gst_rtp_mp4v_depay_sink_template =
  48 GST_STATIC_PAD_TEMPLATE ("sink",
  49     GST_PAD_SINK,
  50     GST_PAD_ALWAYS,
  51     GST_STATIC_CAPS ("application/x-rtp, "
  52         "media = (string) \"video\", "
  53         "payload = (int) " GST_RTP_PAYLOAD_DYNAMIC_STRING ", "
  54         "clock-rate = (int) [1, MAX ], " "encoding-name = (string) \"MP4V-ES\""
  55         /* All optional parameters
  56          *
  57          * "profile-level-id=[1,MAX]"
  58          * "config="
  59          */
  60     )
  61     );
  62
  63 GST_BOILERPLATE (GstRtpMP4VDepay, gst_rtp_mp4v_depay, GstBaseRTPDepayload,
  64     GST_TYPE_BASE_RTP_DEPAYLOAD);
  65
  66 static void gst_rtp_mp4v_depay_finalize (GObject * object);
  67
  68 static gboolean gst_rtp_mp4v_depay_setcaps (GstBaseRTPDepayload * depayload,
  69     GstCaps * caps);
  70 static GstBuffer *gst_rtp_mp4v_depay_process (GstBaseRTPDepayload * depayload,
  71     GstBuffer * buf);
  72
  73 static GstStateChangeReturn gst_rtp_mp4v_depay_change_state (GstElement *
  74     element, GstStateChange transition);
  75
  76
  77 static void
  78 gst_rtp_mp4v_depay_base_init (gpointer klass)
  79 {
  80   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
  81
  82   gst_element_class_add_pad_template (element_class,
  83       gst_static_pad_template_get (&gst_rtp_mp4v_depay_src_template));
  84   gst_element_class_add_pad_template (element_class,
  85       gst_static_pad_template_get (&gst_rtp_mp4v_depay_sink_template));
  86
  87   gst_element_class_set_details (element_class, &gst_rtp_mp4vdepay_details);
  88 }
  89
  90 static void
  91 gst_rtp_mp4v_depay_class_init (GstRtpMP4VDepayClass * klass)
  92 {
  93   GObjectClass *gobject_class;
  94   GstElementClass *gstelement_class;
  95   GstBaseRTPDepayloadClass *gstbasertpdepayload_class;
  96
  97   gobject_class = (GObjectClass *) klass;
  98   gstelement_class = (GstElementClass *) klass;
  99   gstbasertpdepayload_class = (GstBaseRTPDepayloadClass *) klass;
 100
 101   gobject_class->finalize = gst_rtp_mp4v_depay_finalize;
 102
 103   gstelement_class->change_state = gst_rtp_mp4v_depay_change_state;
 104
 105   gstbasertpdepayload_class->process = gst_rtp_mp4v_depay_process;
 106   gstbasertpdepayload_class->set_caps = gst_rtp_mp4v_depay_setcaps;
 107
 108   GST_DEBUG_CATEGORY_INIT (rtpmp4vdepay_debug, "rtpmp4vdepay", 0,
 109       "MPEG4 video RTP Depayloader");
 110 }
 111
 112 static void
 113 gst_rtp_mp4v_depay_init (GstRtpMP4VDepay * rtpmp4vdepay,
 114     GstRtpMP4VDepayClass * klass)
 115 {
 116   rtpmp4vdepay->adapter = gst_adapter_new ();
 117 }
 118
 119 static void
 120 gst_rtp_mp4v_depay_finalize (GObject * object)
 121 {
 122   GstRtpMP4VDepay *rtpmp4vdepay;
 123
 124   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (object);
 125
 126   g_object_unref (rtpmp4vdepay->adapter);
 127   rtpmp4vdepay->adapter = NULL;
 128
 129   G_OBJECT_CLASS (parent_class)->finalize (object);
 130 }
 131
 132 static gboolean
 133 gst_rtp_mp4v_depay_setcaps (GstBaseRTPDepayload * depayload, GstCaps * caps)
 134 {
 135   GstStructure *structure;
 136   GstRtpMP4VDepay *rtpmp4vdepay;
 137   GstCaps *srccaps;
 138   const gchar *str;
 139   gint clock_rate = 90000;      /* default */
 140
 141   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (depayload);
 142
 143   structure = gst_caps_get_structure (caps, 0);
 144
 145   if (gst_structure_has_field (structure, "clock-rate"))
 146     gst_structure_get_int (structure, "clock-rate", &clock_rate);
 147   depayload->clock_rate = clock_rate;
 148
 149   srccaps = gst_caps_new_simple ("video/mpeg",
 150       "mpegversion", G_TYPE_INT, 4,
 151       "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 152
 153   if ((str = gst_structure_get_string (structure, "config"))) {
 154     GValue v = { 0 };
 155
 156     g_value_init (&v, GST_TYPE_BUFFER);
 157     if (gst_value_deserialize (&v, str)) {
 158       GstBuffer *buffer;
 159
 160       buffer = gst_value_get_buffer (&v);
 161       gst_buffer_ref (buffer);
 162       g_value_unset (&v);
 163
 164       gst_caps_set_simple (srccaps,
 165           "codec_data", GST_TYPE_BUFFER, buffer, NULL);
 166     } else {
 167       g_warning ("cannot convert config to buffer");
 168     }
 169   }
 170   gst_pad_set_caps (depayload->srcpad, srccaps);
 171   gst_caps_unref (srccaps);
 172
 173   return TRUE;
 174 }
 175
 176 static GstBuffer *
 177 gst_rtp_mp4v_depay_process (GstBaseRTPDepayload * depayload, GstBuffer * buf)
 178 {
 179   GstRtpMP4VDepay *rtpmp4vdepay;
 180   GstBuffer *outbuf;
 181
 182   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (depayload);
 183
 184   if (!gst_rtp_buffer_validate (buf))
 185     goto bad_packet;
 186
 187   /* flush remaining data on discont */
 188   if (GST_BUFFER_IS_DISCONT (buf))
 189     gst_adapter_clear (rtpmp4vdepay->adapter);
 190
 191   outbuf = gst_rtp_buffer_get_payload_buffer (buf);
 192   gst_adapter_push (rtpmp4vdepay->adapter, outbuf);
 193
 194   /* if this was the last packet of the VOP, create and push a buffer */
 195   if (gst_rtp_buffer_get_marker (buf)) {
 196     guint avail;
 197
 198     avail = gst_adapter_available (rtpmp4vdepay->adapter);
 199
 200     outbuf = gst_adapter_take_buffer (rtpmp4vdepay->adapter, avail);
 201     gst_buffer_set_caps (outbuf, GST_PAD_CAPS (depayload->srcpad));
 202
 203     GST_DEBUG ("gst_rtp_mp4v_depay_chain: pushing buffer of size %d",
 204         GST_BUFFER_SIZE (outbuf));
 205
 206     return outbuf;
 207   }
 208   return NULL;
 209
 210 bad_packet:
 211   {
 212     GST_ELEMENT_WARNING (rtpmp4vdepay, STREAM, DECODE,
 213         ("Packet did not validate"), (NULL));
 214     return NULL;
 215   }
 216 }
 217
 218 static GstStateChangeReturn
 219 gst_rtp_mp4v_depay_change_state (GstElement * element,
 220     GstStateChange transition)
 221 {
 222   GstRtpMP4VDepay *rtpmp4vdepay;
 223   GstStateChangeReturn ret;
 224
 225   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (element);
 226
 227   switch (transition) {
 228     case GST_STATE_CHANGE_READY_TO_PAUSED:
 229       gst_adapter_clear (rtpmp4vdepay->adapter);
 230       break;
 231     default:
 232       break;
 233   }
 234
 235   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 236
 237   switch (transition) {
 238     default:
 239       break;
 240   }
 241   return ret;
 242 }
 243
 244 gboolean
 245 gst_rtp_mp4v_depay_plugin_init (GstPlugin * plugin)
 246 {
 247   return gst_element_register (plugin, "rtpmp4vdepay",
 248       GST_RANK_MARGINAL, GST_TYPE_RTP_MP4V_DEPAY);
 249 }