gst/rtp/gstrtpmp4vdepay.c

   1 /* GStreamer
   2  * Copyright (C) <2005> Wim Taymans <wim.taymans@gmail.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 static GstStaticPadTemplate gst_rtp_mp4v_depay_src_template =
  33 GST_STATIC_PAD_TEMPLATE ("src",
  34     GST_PAD_SRC,
  35     GST_PAD_ALWAYS,
  36     GST_STATIC_CAPS ("video/mpeg,"
  37         "mpegversion=(int) 4," "systemstream=(boolean)false")
  38     );
  39
  40 static GstStaticPadTemplate gst_rtp_mp4v_depay_sink_template =
  41 GST_STATIC_PAD_TEMPLATE ("sink",
  42     GST_PAD_SINK,
  43     GST_PAD_ALWAYS,
  44     GST_STATIC_CAPS ("application/x-rtp, "
  45         "media = (string) \"video\", "
  46         "payload = (int) " GST_RTP_PAYLOAD_DYNAMIC_STRING ", "
  47         "clock-rate = (int) [1, MAX ], " "encoding-name = (string) \"MP4V-ES\""
  48         /* All optional parameters
  49          *
  50          * "profile-level-id=[1,MAX]"
  51          * "config="
  52          */
  53     )
  54     );
  55
  56 #define gst_rtp_mp4v_depay_parent_class parent_class
  57 G_DEFINE_TYPE (GstRtpMP4VDepay, gst_rtp_mp4v_depay,
  58     GST_TYPE_RTP_BASE_DEPAYLOAD);
  59
  60 static void gst_rtp_mp4v_depay_finalize (GObject * object);
  61
  62 static gboolean gst_rtp_mp4v_depay_setcaps (GstRTPBaseDepayload * depayload,
  63     GstCaps * caps);
  64 static GstBuffer *gst_rtp_mp4v_depay_process (GstRTPBaseDepayload * depayload,
  65     GstBuffer * buf);
  66
  67 static GstStateChangeReturn gst_rtp_mp4v_depay_change_state (GstElement *
  68     element, GstStateChange transition);
  69
  70 static void
  71 gst_rtp_mp4v_depay_class_init (GstRtpMP4VDepayClass * klass)
  72 {
  73   GObjectClass *gobject_class;
  74   GstElementClass *gstelement_class;
  75   GstRTPBaseDepayloadClass *gstrtpbasedepayload_class;
  76
  77   gobject_class = (GObjectClass *) klass;
  78   gstelement_class = (GstElementClass *) klass;
  79   gstrtpbasedepayload_class = (GstRTPBaseDepayloadClass *) klass;
  80
  81   gobject_class->finalize = gst_rtp_mp4v_depay_finalize;
  82
  83   gstelement_class->change_state = gst_rtp_mp4v_depay_change_state;
  84
  85   gstrtpbasedepayload_class->process = gst_rtp_mp4v_depay_process;
  86   gstrtpbasedepayload_class->set_caps = gst_rtp_mp4v_depay_setcaps;
  87
  88   gst_element_class_add_pad_template (gstelement_class,
  89       gst_static_pad_template_get (&gst_rtp_mp4v_depay_src_template));
  90   gst_element_class_add_pad_template (gstelement_class,
  91       gst_static_pad_template_get (&gst_rtp_mp4v_depay_sink_template));
  92
  93   gst_element_class_set_static_metadata (gstelement_class,
  94       "RTP MPEG4 video depayloader", "Codec/Depayloader/Network/RTP",
  95       "Extracts MPEG4 video from RTP packets (RFC 3016)",
  96       "Wim Taymans <wim.taymans@gmail.com>");
  97
  98   GST_DEBUG_CATEGORY_INIT (rtpmp4vdepay_debug, "rtpmp4vdepay", 0,
  99       "MPEG4 video RTP Depayloader");
 100 }
 101
 102 static void
 103 gst_rtp_mp4v_depay_init (GstRtpMP4VDepay * rtpmp4vdepay)
 104 {
 105   rtpmp4vdepay->adapter = gst_adapter_new ();
 106 }
 107
 108 static void
 109 gst_rtp_mp4v_depay_finalize (GObject * object)
 110 {
 111   GstRtpMP4VDepay *rtpmp4vdepay;
 112
 113   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (object);
 114
 115   g_object_unref (rtpmp4vdepay->adapter);
 116   rtpmp4vdepay->adapter = NULL;
 117
 118   G_OBJECT_CLASS (parent_class)->finalize (object);
 119 }
 120
 121 static gboolean
 122 gst_rtp_mp4v_depay_setcaps (GstRTPBaseDepayload * depayload, GstCaps * caps)
 123 {
 124   GstStructure *structure;
 125   GstCaps *srccaps;
 126   const gchar *str;
 127   gint clock_rate;
 128   gboolean res;
 129
 130   structure = gst_caps_get_structure (caps, 0);
 131
 132   if (!gst_structure_get_int (structure, "clock-rate", &clock_rate))
 133     clock_rate = 90000;         /* default */
 134   depayload->clock_rate = clock_rate;
 135
 136   srccaps = gst_caps_new_simple ("video/mpeg",
 137       "mpegversion", G_TYPE_INT, 4,
 138       "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 139
 140   if ((str = gst_structure_get_string (structure, "config"))) {
 141     GValue v = { 0 };
 142
 143     g_value_init (&v, GST_TYPE_BUFFER);
 144     if (gst_value_deserialize (&v, str)) {
 145       GstBuffer *buffer;
 146
 147       buffer = gst_value_get_buffer (&v);
 148       gst_caps_set_simple (srccaps,
 149           "codec_data", GST_TYPE_BUFFER, buffer, NULL);
 150       /* caps takes ref */
 151       g_value_unset (&v);
 152     } else {
 153       g_warning ("cannot convert config to buffer");
 154     }
 155   }
 156   res = gst_pad_set_caps (depayload->srcpad, srccaps);
 157   gst_caps_unref (srccaps);
 158
 159   return res;
 160 }
 161
 162 static GstBuffer *
 163 gst_rtp_mp4v_depay_process (GstRTPBaseDepayload * depayload, GstBuffer * buf)
 164 {
 165   GstRtpMP4VDepay *rtpmp4vdepay;
 166   GstBuffer *pbuf, *outbuf = NULL;
 167   GstRTPBuffer rtp = { NULL };
 168   gboolean marker;
 169
 170   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (depayload);
 171
 172   /* flush remaining data on discont */
 173   if (GST_BUFFER_IS_DISCONT (buf))
 174     gst_adapter_clear (rtpmp4vdepay->adapter);
 175
 176   gst_rtp_buffer_map (buf, GST_MAP_READ, &rtp);
 177   pbuf = gst_rtp_buffer_get_payload_buffer (&rtp);
 178   marker = gst_rtp_buffer_get_marker (&rtp);
 179   gst_rtp_buffer_unmap (&rtp);
 180
 181   gst_adapter_push (rtpmp4vdepay->adapter, pbuf);
 182
 183   /* if this was the last packet of the VOP, create and push a buffer */
 184   if (marker) {
 185     guint avail;
 186
 187     avail = gst_adapter_available (rtpmp4vdepay->adapter);
 188     outbuf = gst_adapter_take_buffer (rtpmp4vdepay->adapter, avail);
 189
 190     GST_DEBUG ("gst_rtp_mp4v_depay_chain: pushing buffer of size %"
 191         G_GSIZE_FORMAT, gst_buffer_get_size (outbuf));
 192   }
 193   return outbuf;
 194 }
 195
 196 static GstStateChangeReturn
 197 gst_rtp_mp4v_depay_change_state (GstElement * element,
 198     GstStateChange transition)
 199 {
 200   GstRtpMP4VDepay *rtpmp4vdepay;
 201   GstStateChangeReturn ret;
 202
 203   rtpmp4vdepay = GST_RTP_MP4V_DEPAY (element);
 204
 205   switch (transition) {
 206     case GST_STATE_CHANGE_READY_TO_PAUSED:
 207       gst_adapter_clear (rtpmp4vdepay->adapter);
 208       break;
 209     default:
 210       break;
 211   }
 212
 213   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 214
 215   switch (transition) {
 216     default:
 217       break;
 218   }
 219   return ret;
 220 }
 221
 222 gboolean
 223 gst_rtp_mp4v_depay_plugin_init (GstPlugin * plugin)
 224 {
 225   return gst_element_register (plugin, "rtpmp4vdepay",
 226       GST_RANK_SECONDARY, GST_TYPE_RTP_MP4V_DEPAY);
 227 }