gst/rtp/gstrtph263pdepay.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
  13  */
  14
  15 #ifdef HAVE_CONFIG_H
  16 #  include "config.h"
  17 #endif
  18
  19 #include <string.h>
  20
  21 #include <gst/rtp/gstrtpbuffer.h>
  22 #include "gstrtph263pdec.h"
  23
  24 /* elementfactory information */
  25 static GstElementDetails gst_rtp_h263pdec_details = {
  26   "RTP packet parser",
  27   "Codec/Parser/Network",
  28   "Extracts H263+ video from RTP packets (RFC 2429)",
  29   "Wim Taymans <wim@fluendo.com>"
  30 };
  31
  32 /* RtpH263PDec signals and args */
  33 enum
  34 {
  35   /* FILL ME */
  36   LAST_SIGNAL
  37 };
  38
  39 enum
  40 {
  41   ARG_0,
  42   ARG_FREQUENCY
  43 };
  44
  45 static GstStaticPadTemplate gst_rtph263pdec_src_template =
  46 GST_STATIC_PAD_TEMPLATE ("src",
  47     GST_PAD_SRC,
  48     GST_PAD_ALWAYS,
  49     GST_STATIC_CAPS ("video/x-h263")
  50     );
  51
  52 static GstStaticPadTemplate gst_rtph263pdec_sink_template =
  53 GST_STATIC_PAD_TEMPLATE ("sink",
  54     GST_PAD_SINK,
  55     GST_PAD_ALWAYS,
  56     GST_STATIC_CAPS ("application/x-rtp, "
  57         "media = (string) \"video\", "
  58         "payload = (int) [ 96, 255 ], "
  59         "clock_rate = (int) 90000, " "encoding_name = (string) \"H263-1998\"")
  60     );
  61
  62
  63 static void gst_rtph263pdec_class_init (GstRtpH263PDecClass * klass);
  64 static void gst_rtph263pdec_base_init (GstRtpH263PDecClass * klass);
  65 static void gst_rtph263pdec_init (GstRtpH263PDec * rtph263pdec);
  66 static void gst_rtph263pdec_finalize (GObject * object);
  67
  68 static GstFlowReturn gst_rtph263pdec_chain (GstPad * pad, GstBuffer * buffer);
  69
  70 static void gst_rtph263pdec_set_property (GObject * object, guint prop_id,
  71     const GValue * value, GParamSpec * pspec);
  72 static void gst_rtph263pdec_get_property (GObject * object, guint prop_id,
  73     GValue * value, GParamSpec * pspec);
  74
  75 static GstStateChangeReturn gst_rtph263pdec_change_state (GstElement *
  76     element, GstStateChange transition);
  77
  78 static GstElementClass *parent_class = NULL;
  79
  80 static GType
  81 gst_rtph263pdec_get_type (void)
  82 {
  83   static GType rtph263pdec_type = 0;
  84
  85   if (!rtph263pdec_type) {
  86     static const GTypeInfo rtph263pdec_info = {
  87       sizeof (GstRtpH263PDecClass),
  88       (GBaseInitFunc) gst_rtph263pdec_base_init,
  89       NULL,
  90       (GClassInitFunc) gst_rtph263pdec_class_init,
  91       NULL,
  92       NULL,
  93       sizeof (GstRtpH263PDec),
  94       0,
  95       (GInstanceInitFunc) gst_rtph263pdec_init,
  96     };
  97
  98     rtph263pdec_type =
  99         g_type_register_static (GST_TYPE_ELEMENT, "GstRtpH263PDec",
 100         &rtph263pdec_info, 0);
 101   }
 102   return rtph263pdec_type;
 103 }
 104
 105 static void
 106 gst_rtph263pdec_base_init (GstRtpH263PDecClass * klass)
 107 {
 108   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
 109
 110   gst_element_class_add_pad_template (element_class,
 111       gst_static_pad_template_get (&gst_rtph263pdec_src_template));
 112   gst_element_class_add_pad_template (element_class,
 113       gst_static_pad_template_get (&gst_rtph263pdec_sink_template));
 114
 115   gst_element_class_set_details (element_class, &gst_rtp_h263pdec_details);
 116 }
 117
 118 static void
 119 gst_rtph263pdec_class_init (GstRtpH263PDecClass * klass)
 120 {
 121   GObjectClass *gobject_class;
 122   GstElementClass *gstelement_class;
 123
 124   gobject_class = (GObjectClass *) klass;
 125   gstelement_class = (GstElementClass *) klass;
 126
 127   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
 128
 129   gobject_class->finalize = gst_rtph263pdec_finalize;
 130
 131   gobject_class->set_property = gst_rtph263pdec_set_property;
 132   gobject_class->get_property = gst_rtph263pdec_get_property;
 133
 134   gstelement_class->change_state = gst_rtph263pdec_change_state;
 135 }
 136
 137 static void
 138 gst_rtph263pdec_init (GstRtpH263PDec * rtph263pdec)
 139 {
 140   rtph263pdec->srcpad =
 141       gst_pad_new_from_template (gst_static_pad_template_get
 142       (&gst_rtph263pdec_src_template), "src");
 143   gst_element_add_pad (GST_ELEMENT (rtph263pdec), rtph263pdec->srcpad);
 144
 145   rtph263pdec->sinkpad =
 146       gst_pad_new_from_template (gst_static_pad_template_get
 147       (&gst_rtph263pdec_sink_template), "sink");
 148   gst_pad_set_chain_function (rtph263pdec->sinkpad, gst_rtph263pdec_chain);
 149   gst_element_add_pad (GST_ELEMENT (rtph263pdec), rtph263pdec->sinkpad);
 150
 151   rtph263pdec->adapter = gst_adapter_new ();
 152 }
 153
 154 static void
 155 gst_rtph263pdec_finalize (GObject * object)
 156 {
 157   GstRtpH263PDec *rtph263pdec;
 158
 159   rtph263pdec = GST_RTP_H263P_DEC (object);
 160
 161   g_object_unref (rtph263pdec->adapter);
 162   rtph263pdec->adapter = NULL;
 163
 164   G_OBJECT_CLASS (parent_class)->finalize (object);
 165 }
 166
 167 static GstFlowReturn
 168 gst_rtph263pdec_chain (GstPad * pad, GstBuffer * buf)
 169 {
 170   GstRtpH263PDec *rtph263pdec;
 171   GstBuffer *outbuf;
 172   GstFlowReturn ret;
 173
 174   /* GstRTPPayload pt; */
 175
 176   rtph263pdec = GST_RTP_H263P_DEC (GST_OBJECT_PARENT (pad));
 177
 178   if (!gst_rtpbuffer_validate (buf))
 179     goto bad_packet;
 180
 181   /*
 182      if ((pt = gst_rtpbuffer_get_payload_type (buf)) != 0)
 183      goto bad_payload;
 184    */
 185
 186   {
 187     gint payload_len;
 188     guint8 *payload;
 189     gboolean P, V, M;
 190     guint32 timestamp;
 191     guint header_len;
 192     guint8 PLEN;
 193
 194     payload_len = gst_rtpbuffer_get_payload_len (buf);
 195     payload = gst_rtpbuffer_get_payload (buf);
 196
 197     header_len = 2;
 198
 199     M = gst_rtpbuffer_get_marker (buf);
 200     P = (payload[0] & 0x04) == 0x04;
 201     V = (payload[0] & 0x02) == 0x02;
 202     PLEN = ((payload[0] & 0x1) << 5) | (payload[1] >> 3);
 203
 204     if (V) {
 205       header_len++;
 206     }
 207     if (PLEN) {
 208       header_len += PLEN;
 209     }
 210
 211     if (P) {
 212       header_len -= 2;
 213       payload[header_len] = 0;
 214       payload[header_len + 1] = 0;
 215     }
 216
 217     /* strip off header */
 218     payload += header_len;
 219     payload_len -= header_len;
 220
 221     timestamp = gst_rtpbuffer_get_timestamp (buf);
 222
 223     if (M) {
 224       /* frame is completed: append to previous, push it out */
 225       guint avail;
 226       guint8 *data;
 227
 228       avail = gst_adapter_available (rtph263pdec->adapter);
 229
 230       outbuf = gst_buffer_new_and_alloc (avail + payload_len);
 231
 232       /* prepend previous data */
 233       if (avail > 0) {
 234         data = (guint8 *) gst_adapter_peek (rtph263pdec->adapter, avail);
 235         memcpy (GST_BUFFER_DATA (outbuf), data, avail);
 236         gst_adapter_flush (rtph263pdec->adapter, avail);
 237       }
 238       memcpy (GST_BUFFER_DATA (outbuf) + avail, payload, payload_len);
 239
 240       GST_BUFFER_TIMESTAMP (outbuf) = timestamp * GST_SECOND / 90000;
 241       gst_buffer_set_caps (outbuf,
 242           (GstCaps *) gst_pad_get_pad_template_caps (rtph263pdec->srcpad));
 243
 244       ret = gst_pad_push (rtph263pdec->srcpad, outbuf);
 245     } else {
 246       /* frame not completed: store in adapter */
 247       outbuf = gst_buffer_new_and_alloc (payload_len);
 248
 249       memcpy (GST_BUFFER_DATA (outbuf), payload, payload_len);
 250
 251       gst_adapter_push (rtph263pdec->adapter, outbuf);
 252
 253       ret = GST_FLOW_OK;
 254     }
 255
 256     gst_buffer_unref (buf);
 257   }
 258
 259   return ret;
 260
 261 bad_packet:
 262   {
 263     GST_DEBUG ("Packet does not validate");
 264     gst_buffer_unref (buf);
 265     return GST_FLOW_ERROR;
 266   }
 267   /*
 268      bad_payload:
 269      {
 270      GST_DEBUG ("Unexpected payload type %u", pt);
 271
 272      gst_buffer_unref (buf);
 273      return GST_FLOW_ERROR;
 274      }
 275    */
 276 }
 277
 278 static void
 279 gst_rtph263pdec_set_property (GObject * object, guint prop_id,
 280     const GValue * value, GParamSpec * pspec)
 281 {
 282   GstRtpH263PDec *rtph263pdec;
 283
 284   rtph263pdec = GST_RTP_H263P_DEC (object);
 285
 286   switch (prop_id) {
 287     default:
 288       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 289       break;
 290   }
 291 }
 292
 293 static void
 294 gst_rtph263pdec_get_property (GObject * object, guint prop_id, GValue * value,
 295     GParamSpec * pspec)
 296 {
 297   GstRtpH263PDec *rtph263pdec;
 298
 299   rtph263pdec = GST_RTP_H263P_DEC (object);
 300
 301   switch (prop_id) {
 302     default:
 303       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 304       break;
 305   }
 306 }
 307
 308 static GstStateChangeReturn
 309 gst_rtph263pdec_change_state (GstElement * element, GstStateChange transition)
 310 {
 311   GstRtpH263PDec *rtph263pdec;
 312   GstStateChangeReturn ret;
 313
 314   rtph263pdec = GST_RTP_H263P_DEC (element);
 315
 316   switch (transition) {
 317     case GST_STATE_CHANGE_NULL_TO_READY:
 318       break;
 319     case GST_STATE_CHANGE_READY_TO_PAUSED:
 320       gst_adapter_clear (rtph263pdec->adapter);
 321       break;
 322     default:
 323       break;
 324   }
 325
 326   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 327
 328   switch (transition) {
 329     case GST_STATE_CHANGE_READY_TO_NULL:
 330       break;
 331     default:
 332       break;
 333   }
 334   return ret;
 335 }
 336
 337 gboolean
 338 gst_rtph263pdec_plugin_init (GstPlugin * plugin)
 339 {
 340   return gst_element_register (plugin, "rtph263pdec",
 341       GST_RANK_NONE, GST_TYPE_RTP_H263P_DEC);
 342 }