2 * Copyright (C) <2007> Thijs Vermeir <thijsvermeir@gmail.com>
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.
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.
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.
26 #include <gst/rtp/gstrtpbuffer.h>
28 #include "gstrtpmpvpay.h"
30 GST_DEBUG_CATEGORY_STATIC (rtpmpvpay_debug);
31 #define GST_CAT_DEFAULT (rtpmpvpay_debug)
33 static GstStaticPadTemplate gst_rtp_mpv_pay_sink_template =
34 GST_STATIC_PAD_TEMPLATE ("sink",
37 GST_STATIC_CAPS ("video/mpeg, "
38 "mpegversion = (int) 2, systemstream = (boolean) FALSE")
41 static GstStaticPadTemplate gst_rtp_mpv_pay_src_template =
42 GST_STATIC_PAD_TEMPLATE ("src",
45 GST_STATIC_CAPS ("application/x-rtp, "
46 "media = (string) \"video\", "
47 "payload = (int) " GST_RTP_PAYLOAD_MPV_STRING ", "
48 "clock-rate = (int) 90000, " "encoding-name = (string) \"MPV\"")
51 static GstStateChangeReturn gst_rtp_mpv_pay_change_state (GstElement * element,
52 GstStateChange transition);
54 static void gst_rtp_mpv_pay_finalize (GObject * object);
56 static GstFlowReturn gst_rtp_mpv_pay_flush (GstRTPMPVPay * rtpmpvpay);
57 static gboolean gst_rtp_mpv_pay_setcaps (GstRTPBasePayload * payload,
59 static GstFlowReturn gst_rtp_mpv_pay_handle_buffer (GstRTPBasePayload *
60 payload, GstBuffer * buffer);
61 static gboolean gst_rtp_mpv_pay_handle_event (GstRTPBasePayload * payload,
64 #define gst_rtp_mpv_pay_parent_class parent_class
65 G_DEFINE_TYPE (GstRTPMPVPay, gst_rtp_mpv_pay, GST_TYPE_RTP_BASE_PAYLOAD);
68 gst_rtp_mpv_pay_class_init (GstRTPMPVPayClass * klass)
70 GObjectClass *gobject_class;
71 GstElementClass *gstelement_class;
72 GstRTPBasePayloadClass *gstrtpbasepayload_class;
74 gobject_class = (GObjectClass *) klass;
75 gstelement_class = (GstElementClass *) klass;
76 gstrtpbasepayload_class = (GstRTPBasePayloadClass *) klass;
78 gobject_class->finalize = gst_rtp_mpv_pay_finalize;
80 gstelement_class->change_state = gst_rtp_mpv_pay_change_state;
82 gst_element_class_add_pad_template (gstelement_class,
83 gst_static_pad_template_get (&gst_rtp_mpv_pay_sink_template));
84 gst_element_class_add_pad_template (gstelement_class,
85 gst_static_pad_template_get (&gst_rtp_mpv_pay_src_template));
87 gst_element_class_set_details_simple (gstelement_class,
88 "RTP MPEG2 ES video payloader", "Codec/Payloader/Network/RTP",
89 "Payload-encodes MPEG2 ES into RTP packets (RFC 2250)",
90 "Thijs Vermeir <thijsvermeir@gmail.com>");
92 gstrtpbasepayload_class->set_caps = gst_rtp_mpv_pay_setcaps;
93 gstrtpbasepayload_class->handle_buffer = gst_rtp_mpv_pay_handle_buffer;
94 gstrtpbasepayload_class->handle_event = gst_rtp_mpv_pay_handle_event;
96 GST_DEBUG_CATEGORY_INIT (rtpmpvpay_debug, "rtpmpvpay", 0,
97 "MPEG2 ES Video RTP Payloader");
101 gst_rtp_mpv_pay_init (GstRTPMPVPay * rtpmpvpay)
103 GST_RTP_BASE_PAYLOAD (rtpmpvpay)->clock_rate = 90000;
104 GST_RTP_BASE_PAYLOAD_PT (rtpmpvpay) = GST_RTP_PAYLOAD_MPV;
106 rtpmpvpay->adapter = gst_adapter_new ();
110 gst_rtp_mpv_pay_finalize (GObject * object)
112 GstRTPMPVPay *rtpmpvpay;
114 rtpmpvpay = GST_RTP_MPV_PAY (object);
116 g_object_unref (rtpmpvpay->adapter);
117 rtpmpvpay->adapter = NULL;
119 G_OBJECT_CLASS (parent_class)->finalize (object);
123 gst_rtp_mpv_pay_reset (GstRTPMPVPay * pay)
127 gst_adapter_clear (pay->adapter);
128 GST_DEBUG_OBJECT (pay, "reset depayloader");
132 gst_rtp_mpv_pay_setcaps (GstRTPBasePayload * payload, GstCaps * caps)
134 gst_rtp_base_payload_set_options (payload, "video", FALSE, "MPV", 90000);
135 return gst_rtp_base_payload_set_outcaps (payload, NULL);
139 gst_rtp_mpv_pay_handle_event (GstRTPBasePayload * payload, GstEvent * event)
142 GstRTPMPVPay *rtpmpvpay;
144 rtpmpvpay = GST_RTP_MPV_PAY (payload);
146 switch (GST_EVENT_TYPE (event)) {
148 /* make sure we push the last packets in the adapter on EOS */
149 gst_rtp_mpv_pay_flush (rtpmpvpay);
151 case GST_EVENT_FLUSH_STOP:
152 gst_rtp_mpv_pay_reset (rtpmpvpay);
159 GST_RTP_BASE_PAYLOAD_CLASS (parent_class)->handle_event (payload, event);
165 gst_rtp_mpv_pay_flush (GstRTPMPVPay * rtpmpvpay)
173 avail = gst_adapter_available (rtpmpvpay->adapter);
181 GstRTPBuffer rtp = { NULL };
183 packet_len = gst_rtp_buffer_calc_packet_len (avail, 4, 0);
185 towrite = MIN (packet_len, GST_RTP_BASE_PAYLOAD_MTU (rtpmpvpay));
187 payload_len = gst_rtp_buffer_calc_payload_len (towrite, 4, 0);
189 outbuf = gst_rtp_buffer_new_allocate (payload_len, 4, 0);
191 gst_rtp_buffer_map (outbuf, GST_MAP_WRITE, &rtp);
193 payload = gst_rtp_buffer_get_payload (&rtp);
194 /* enable MPEG Video-specific header
197 * 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
198 * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
199 * | MBZ |T| TR | |N|S|B|E| P | | BFC | | FFC |
200 * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
204 /* fill in the MPEG Video-specific header
205 * data is set to 0x0 here
207 memset (payload, 0x0, 4);
209 gst_adapter_copy (rtpmpvpay->adapter, payload + 4, 0, payload_len);
210 gst_adapter_flush (rtpmpvpay->adapter, payload_len);
212 avail -= payload_len;
214 gst_rtp_buffer_set_marker (&rtp, avail == 0);
215 gst_rtp_buffer_unmap (&rtp);
217 GST_BUFFER_TIMESTAMP (outbuf) = rtpmpvpay->first_ts;
219 ret = gst_rtp_base_payload_push (GST_RTP_BASE_PAYLOAD (rtpmpvpay), outbuf);
226 gst_rtp_mpv_pay_handle_buffer (GstRTPBasePayload * basepayload,
229 GstRTPMPVPay *rtpmpvpay;
230 guint avail, packet_len;
231 GstClockTime timestamp, duration;
232 GstFlowReturn ret = GST_FLOW_OK;
234 rtpmpvpay = GST_RTP_MPV_PAY (basepayload);
236 timestamp = GST_BUFFER_TIMESTAMP (buffer);
237 duration = GST_BUFFER_DURATION (buffer);
239 if (GST_BUFFER_IS_DISCONT (buffer)) {
240 GST_DEBUG_OBJECT (rtpmpvpay, "DISCONT");
241 gst_rtp_mpv_pay_reset (rtpmpvpay);
244 avail = gst_adapter_available (rtpmpvpay->adapter);
249 if (rtpmpvpay->first_ts == GST_CLOCK_TIME_NONE || avail == 0)
250 rtpmpvpay->first_ts = timestamp;
253 rtpmpvpay->duration = duration;
255 rtpmpvpay->duration += duration;
258 gst_adapter_push (rtpmpvpay->adapter, buffer);
259 avail = gst_adapter_available (rtpmpvpay->adapter);
261 /* get packet length of previous data and this new data,
262 * payload length includes a 4 byte MPEG video-specific header */
263 packet_len = gst_rtp_buffer_calc_packet_len (avail, 4, 0);
264 GST_LOG_OBJECT (rtpmpvpay, "available %d, rtp packet length %d", avail,
267 if (gst_rtp_base_payload_is_filled (basepayload,
268 packet_len, rtpmpvpay->duration)) {
269 ret = gst_rtp_mpv_pay_flush (rtpmpvpay);
271 rtpmpvpay->first_ts = timestamp;
277 static GstStateChangeReturn
278 gst_rtp_mpv_pay_change_state (GstElement * element, GstStateChange transition)
280 GstRTPMPVPay *rtpmpvpay;
281 GstStateChangeReturn ret;
283 rtpmpvpay = GST_RTP_MPV_PAY (element);
285 switch (transition) {
286 case GST_STATE_CHANGE_READY_TO_PAUSED:
287 gst_rtp_mpv_pay_reset (rtpmpvpay);
293 ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
295 switch (transition) {
296 case GST_STATE_CHANGE_PAUSED_TO_READY:
297 gst_rtp_mpv_pay_reset (rtpmpvpay);
307 gst_rtp_mpv_pay_plugin_init (GstPlugin * plugin)
309 return gst_element_register (plugin, "rtpmpvpay",
310 GST_RANK_SECONDARY, GST_TYPE_RTP_MPV_PAY);