Merge remote-tracking branch 'origin/master' into 0.11
[platform/upstream/gst-plugins-good.git] / gst / rtp / gstrtpmpvpay.c
1 /* GStreamer
2  * Copyright (C) <2007> Thijs Vermeir <thijsvermeir@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 <stdlib.h>
25 #include <string.h>
26 #include <gst/rtp/gstrtpbuffer.h>
27
28 #include "gstrtpmpvpay.h"
29
30 GST_DEBUG_CATEGORY_STATIC (rtpmpvpay_debug);
31 #define GST_CAT_DEFAULT (rtpmpvpay_debug)
32
33 static GstStaticPadTemplate gst_rtp_mpv_pay_sink_template =
34 GST_STATIC_PAD_TEMPLATE ("sink",
35     GST_PAD_SINK,
36     GST_PAD_ALWAYS,
37     GST_STATIC_CAPS ("video/mpeg, "
38         "mpegversion = (int) 2, systemstream = (boolean) FALSE")
39     );
40
41 static GstStaticPadTemplate gst_rtp_mpv_pay_src_template =
42 GST_STATIC_PAD_TEMPLATE ("src",
43     GST_PAD_SRC,
44     GST_PAD_ALWAYS,
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\"")
49     );
50
51 static GstStateChangeReturn gst_rtp_mpv_pay_change_state (GstElement * element,
52     GstStateChange transition);
53
54 static void gst_rtp_mpv_pay_finalize (GObject * object);
55
56 static GstFlowReturn gst_rtp_mpv_pay_flush (GstRTPMPVPay * rtpmpvpay);
57 static gboolean gst_rtp_mpv_pay_setcaps (GstRTPBasePayload * payload,
58     GstCaps * caps);
59 static GstFlowReturn gst_rtp_mpv_pay_handle_buffer (GstRTPBasePayload *
60     payload, GstBuffer * buffer);
61 static gboolean gst_rtp_mpv_pay_sink_event (GstRTPBasePayload * payload,
62     GstEvent * event);
63
64 #define gst_rtp_mpv_pay_parent_class parent_class
65 G_DEFINE_TYPE (GstRTPMPVPay, gst_rtp_mpv_pay, GST_TYPE_RTP_BASE_PAYLOAD);
66
67 static void
68 gst_rtp_mpv_pay_class_init (GstRTPMPVPayClass * klass)
69 {
70   GObjectClass *gobject_class;
71   GstElementClass *gstelement_class;
72   GstRTPBasePayloadClass *gstrtpbasepayload_class;
73
74   gobject_class = (GObjectClass *) klass;
75   gstelement_class = (GstElementClass *) klass;
76   gstrtpbasepayload_class = (GstRTPBasePayloadClass *) klass;
77
78   gobject_class->finalize = gst_rtp_mpv_pay_finalize;
79
80   gstelement_class->change_state = gst_rtp_mpv_pay_change_state;
81
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));
86
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>");
91
92   gstrtpbasepayload_class->set_caps = gst_rtp_mpv_pay_setcaps;
93   gstrtpbasepayload_class->handle_buffer = gst_rtp_mpv_pay_handle_buffer;
94   gstrtpbasepayload_class->sink_event = gst_rtp_mpv_pay_sink_event;
95
96   GST_DEBUG_CATEGORY_INIT (rtpmpvpay_debug, "rtpmpvpay", 0,
97       "MPEG2 ES Video RTP Payloader");
98 }
99
100 static void
101 gst_rtp_mpv_pay_init (GstRTPMPVPay * rtpmpvpay)
102 {
103   GST_RTP_BASE_PAYLOAD (rtpmpvpay)->clock_rate = 90000;
104   GST_RTP_BASE_PAYLOAD_PT (rtpmpvpay) = GST_RTP_PAYLOAD_MPV;
105
106   rtpmpvpay->adapter = gst_adapter_new ();
107 }
108
109 static void
110 gst_rtp_mpv_pay_finalize (GObject * object)
111 {
112   GstRTPMPVPay *rtpmpvpay;
113
114   rtpmpvpay = GST_RTP_MPV_PAY (object);
115
116   g_object_unref (rtpmpvpay->adapter);
117   rtpmpvpay->adapter = NULL;
118
119   G_OBJECT_CLASS (parent_class)->finalize (object);
120 }
121
122 static void
123 gst_rtp_mpv_pay_reset (GstRTPMPVPay * pay)
124 {
125   pay->first_ts = -1;
126   pay->duration = 0;
127   gst_adapter_clear (pay->adapter);
128   GST_DEBUG_OBJECT (pay, "reset depayloader");
129 }
130
131 static gboolean
132 gst_rtp_mpv_pay_setcaps (GstRTPBasePayload * payload, GstCaps * caps)
133 {
134   gst_rtp_base_payload_set_options (payload, "video", FALSE, "MPV", 90000);
135   return gst_rtp_base_payload_set_outcaps (payload, NULL);
136 }
137
138 static gboolean
139 gst_rtp_mpv_pay_sink_event (GstRTPBasePayload * payload, GstEvent * event)
140 {
141   gboolean ret;
142   GstRTPMPVPay *rtpmpvpay;
143
144   rtpmpvpay = GST_RTP_MPV_PAY (payload);
145
146   switch (GST_EVENT_TYPE (event)) {
147     case GST_EVENT_EOS:
148       /* make sure we push the last packets in the adapter on EOS */
149       gst_rtp_mpv_pay_flush (rtpmpvpay);
150       break;
151     case GST_EVENT_FLUSH_STOP:
152       gst_rtp_mpv_pay_reset (rtpmpvpay);
153       break;
154     default:
155       break;
156   }
157
158   ret = GST_RTP_BASE_PAYLOAD_CLASS (parent_class)->sink_event (payload, event);
159
160   return ret;
161 }
162
163 static GstFlowReturn
164 gst_rtp_mpv_pay_flush (GstRTPMPVPay * rtpmpvpay)
165 {
166   GstBuffer *outbuf;
167   GstFlowReturn ret;
168   guint avail;
169
170   guint8 *payload;
171
172   avail = gst_adapter_available (rtpmpvpay->adapter);
173
174   ret = GST_FLOW_OK;
175
176   while (avail > 0) {
177     guint towrite;
178     guint packet_len;
179     guint payload_len;
180     GstRTPBuffer rtp = { NULL };
181
182     packet_len = gst_rtp_buffer_calc_packet_len (avail, 4, 0);
183
184     towrite = MIN (packet_len, GST_RTP_BASE_PAYLOAD_MTU (rtpmpvpay));
185
186     payload_len = gst_rtp_buffer_calc_payload_len (towrite, 4, 0);
187
188     outbuf = gst_rtp_buffer_new_allocate (payload_len, 4, 0);
189
190     gst_rtp_buffer_map (outbuf, GST_MAP_WRITE, &rtp);
191
192     payload = gst_rtp_buffer_get_payload (&rtp);
193     /* enable MPEG Video-specific header
194      *
195      *  0                   1                   2                   3
196      *  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
197      * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
198      * |    MBZ  |T|         TR        | |N|S|B|E|  P  | | BFC | | FFC |
199      * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
200      *                                  AN              FBV     FFV
201      */
202
203     /* fill in the MPEG Video-specific header 
204      * data is set to 0x0 here
205      */
206     memset (payload, 0x0, 4);
207
208     gst_adapter_copy (rtpmpvpay->adapter, payload + 4, 0, payload_len);
209     gst_adapter_flush (rtpmpvpay->adapter, payload_len);
210
211     avail -= payload_len;
212
213     gst_rtp_buffer_set_marker (&rtp, avail == 0);
214     gst_rtp_buffer_unmap (&rtp);
215
216     GST_BUFFER_TIMESTAMP (outbuf) = rtpmpvpay->first_ts;
217
218     ret = gst_rtp_base_payload_push (GST_RTP_BASE_PAYLOAD (rtpmpvpay), outbuf);
219   }
220
221   return ret;
222 }
223
224 static GstFlowReturn
225 gst_rtp_mpv_pay_handle_buffer (GstRTPBasePayload * basepayload,
226     GstBuffer * buffer)
227 {
228   GstRTPMPVPay *rtpmpvpay;
229   guint avail, packet_len;
230   GstClockTime timestamp, duration;
231   GstFlowReturn ret = GST_FLOW_OK;
232
233   rtpmpvpay = GST_RTP_MPV_PAY (basepayload);
234
235   timestamp = GST_BUFFER_TIMESTAMP (buffer);
236   duration = GST_BUFFER_DURATION (buffer);
237
238   if (GST_BUFFER_IS_DISCONT (buffer)) {
239     GST_DEBUG_OBJECT (rtpmpvpay, "DISCONT");
240     gst_rtp_mpv_pay_reset (rtpmpvpay);
241   }
242
243   avail = gst_adapter_available (rtpmpvpay->adapter);
244
245   if (duration == -1)
246     duration = 0;
247
248   if (rtpmpvpay->first_ts == GST_CLOCK_TIME_NONE || avail == 0)
249     rtpmpvpay->first_ts = timestamp;
250
251   if (avail == 0) {
252     rtpmpvpay->duration = duration;
253   } else {
254     rtpmpvpay->duration += duration;
255   }
256
257   gst_adapter_push (rtpmpvpay->adapter, buffer);
258   avail = gst_adapter_available (rtpmpvpay->adapter);
259
260   /* get packet length of previous data and this new data,
261    * payload length includes a 4 byte MPEG video-specific header */
262   packet_len = gst_rtp_buffer_calc_packet_len (avail, 4, 0);
263   GST_LOG_OBJECT (rtpmpvpay, "available %d, rtp packet length %d", avail,
264       packet_len);
265
266   if (gst_rtp_base_payload_is_filled (basepayload,
267           packet_len, rtpmpvpay->duration)) {
268     ret = gst_rtp_mpv_pay_flush (rtpmpvpay);
269   } else {
270     rtpmpvpay->first_ts = timestamp;
271   }
272
273   return ret;
274 }
275
276 static GstStateChangeReturn
277 gst_rtp_mpv_pay_change_state (GstElement * element, GstStateChange transition)
278 {
279   GstRTPMPVPay *rtpmpvpay;
280   GstStateChangeReturn ret;
281
282   rtpmpvpay = GST_RTP_MPV_PAY (element);
283
284   switch (transition) {
285     case GST_STATE_CHANGE_READY_TO_PAUSED:
286       gst_rtp_mpv_pay_reset (rtpmpvpay);
287       break;
288     default:
289       break;
290   }
291
292   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
293
294   switch (transition) {
295     case GST_STATE_CHANGE_PAUSED_TO_READY:
296       gst_rtp_mpv_pay_reset (rtpmpvpay);
297       break;
298     default:
299       break;
300   }
301   return ret;
302 }
303
304
305 gboolean
306 gst_rtp_mpv_pay_plugin_init (GstPlugin * plugin)
307 {
308   return gst_element_register (plugin, "rtpmpvpay",
309       GST_RANK_SECONDARY, GST_TYPE_RTP_MPV_PAY);
310 }