db97f331819ec63c048838b3ebf660e2bc802617
[platform/upstream/gstreamer.git] / gst / rtp / gstrtph263ppay.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 <string.h>
25 #include <stdlib.h>
26 #include <stdio.h>
27
28 #include <gst/rtp/gstrtpbuffer.h>
29
30 #include "gstrtph263ppay.h"
31
32 #define DEFAULT_FRAGMENTATION_MODE   GST_FRAGMENTATION_MODE_NORMAL
33
34 enum
35 {
36   PROP_0,
37   PROP_FRAGMENTATION_MODE
38 };
39
40 #define GST_TYPE_FRAGMENTATION_MODE (gst_fragmentation_mode_get_type())
41 static GType
42 gst_fragmentation_mode_get_type (void)
43 {
44   static GType fragmentation_mode_type = 0;
45   static const GEnumValue fragmentation_mode[] = {
46     {GST_FRAGMENTATION_MODE_NORMAL, "Normal", "normal"},
47     {GST_FRAGMENTATION_MODE_SYNC, "Fragment at sync points", "sync"},
48     {0, NULL, NULL},
49   };
50
51   if (!fragmentation_mode_type) {
52     fragmentation_mode_type =
53         g_enum_register_static ("GstFragmentationMode", fragmentation_mode);
54   }
55   return fragmentation_mode_type;
56 }
57
58
59 GST_DEBUG_CATEGORY_STATIC (rtph263ppay_debug);
60 #define GST_CAT_DEFAULT rtph263ppay_debug
61
62 static GstStaticPadTemplate gst_rtp_h263p_pay_sink_template =
63 GST_STATIC_PAD_TEMPLATE ("sink",
64     GST_PAD_SINK,
65     GST_PAD_ALWAYS,
66     GST_STATIC_CAPS ("video/x-h263, " "variant = (string) \"itu\"")
67     );
68
69 /*
70  * We also return these in getcaps() as required by the SDP caps
71  *
72  * width = (int) [16, 4096]
73  * height = (int) [16, 4096]
74  * "annex-f = (boolean) {true, false},"
75  * "annex-i = (boolean) {true, false},"
76  * "annex-j = (boolean) {true, false},"
77  * "annex-l = (boolean) {true, false},"
78  * "annex-t = (boolean) {true, false},"
79  * "annex-v = (boolean) {true, false}")
80  */
81
82
83 static GstStaticPadTemplate gst_rtp_h263p_pay_src_template =
84     GST_STATIC_PAD_TEMPLATE ("src",
85     GST_PAD_SRC,
86     GST_PAD_ALWAYS,
87     GST_STATIC_CAPS ("application/x-rtp, "
88         "media = (string) \"video\", "
89         "payload = (int) " GST_RTP_PAYLOAD_DYNAMIC_STRING ", "
90         "clock-rate = (int) 90000, " "encoding-name = (string) \"H263-1998\"; "
91         "application/x-rtp, "
92         "media = (string) \"video\", "
93         "payload = (int) " GST_RTP_PAYLOAD_DYNAMIC_STRING ", "
94         "clock-rate = (int) 90000, " "encoding-name = (string) \"H263-2000\"")
95     );
96
97 static void gst_rtp_h263p_pay_finalize (GObject * object);
98
99 static void gst_rtp_h263p_pay_set_property (GObject * object, guint prop_id,
100     const GValue * value, GParamSpec * pspec);
101 static void gst_rtp_h263p_pay_get_property (GObject * object, guint prop_id,
102     GValue * value, GParamSpec * pspec);
103
104 static gboolean gst_rtp_h263p_pay_setcaps (GstBaseRTPPayload * payload,
105     GstCaps * caps);
106 static GstCaps *gst_rtp_h263p_pay_sink_getcaps (GstBaseRTPPayload * payload,
107     GstPad * pad, GstCaps * filter);
108 static GstFlowReturn gst_rtp_h263p_pay_handle_buffer (GstBaseRTPPayload *
109     payload, GstBuffer * buffer);
110
111 #define gst_rtp_h263p_pay_parent_class parent_class
112 G_DEFINE_TYPE (GstRtpH263PPay, gst_rtp_h263p_pay, GST_TYPE_BASE_RTP_PAYLOAD);
113
114 static void
115 gst_rtp_h263p_pay_class_init (GstRtpH263PPayClass * klass)
116 {
117   GObjectClass *gobject_class;
118   GstElementClass *gstelement_class;
119   GstBaseRTPPayloadClass *gstbasertppayload_class;
120
121   gobject_class = (GObjectClass *) klass;
122   gstelement_class = (GstElementClass *) klass;
123   gstbasertppayload_class = (GstBaseRTPPayloadClass *) klass;
124
125   gobject_class->finalize = gst_rtp_h263p_pay_finalize;
126   gobject_class->set_property = gst_rtp_h263p_pay_set_property;
127   gobject_class->get_property = gst_rtp_h263p_pay_get_property;
128
129   gstbasertppayload_class->set_caps = gst_rtp_h263p_pay_setcaps;
130   gstbasertppayload_class->get_caps = gst_rtp_h263p_pay_sink_getcaps;
131   gstbasertppayload_class->handle_buffer = gst_rtp_h263p_pay_handle_buffer;
132
133   g_object_class_install_property (G_OBJECT_CLASS (klass),
134       PROP_FRAGMENTATION_MODE, g_param_spec_enum ("fragmentation-mode",
135           "Fragmentation Mode",
136           "Packet Fragmentation Mode", GST_TYPE_FRAGMENTATION_MODE,
137           DEFAULT_FRAGMENTATION_MODE,
138           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
139
140   gst_element_class_add_pad_template (gstelement_class,
141       gst_static_pad_template_get (&gst_rtp_h263p_pay_src_template));
142   gst_element_class_add_pad_template (gstelement_class,
143       gst_static_pad_template_get (&gst_rtp_h263p_pay_sink_template));
144
145   gst_element_class_set_details_simple (gstelement_class, "RTP H263 payloader",
146       "Codec/Payloader/Network/RTP",
147       "Payload-encodes H263/+/++ video in RTP packets (RFC 4629)",
148       "Wim Taymans <wim.taymans@gmail.com>");
149
150   GST_DEBUG_CATEGORY_INIT (rtph263ppay_debug, "rtph263ppay",
151       0, "rtph263ppay (RFC 4629)");
152 }
153
154 static void
155 gst_rtp_h263p_pay_init (GstRtpH263PPay * rtph263ppay)
156 {
157   rtph263ppay->adapter = gst_adapter_new ();
158
159   rtph263ppay->fragmentation_mode = DEFAULT_FRAGMENTATION_MODE;
160 }
161
162 static void
163 gst_rtp_h263p_pay_finalize (GObject * object)
164 {
165   GstRtpH263PPay *rtph263ppay;
166
167   rtph263ppay = GST_RTP_H263P_PAY (object);
168
169   g_object_unref (rtph263ppay->adapter);
170   rtph263ppay->adapter = NULL;
171
172   G_OBJECT_CLASS (parent_class)->finalize (object);
173 }
174
175 static gboolean
176 gst_rtp_h263p_pay_setcaps (GstBaseRTPPayload * payload, GstCaps * caps)
177 {
178   gboolean res;
179   GstCaps *peercaps;
180   gchar *encoding_name = NULL;
181
182   g_return_val_if_fail (gst_caps_is_fixed (caps), FALSE);
183
184   peercaps =
185       gst_pad_peer_get_caps (GST_BASE_RTP_PAYLOAD_SRCPAD (payload), NULL);
186   if (peercaps) {
187     GstCaps *intersect = gst_caps_intersect (peercaps,
188         gst_pad_get_pad_template_caps (GST_BASE_RTP_PAYLOAD_SRCPAD (payload)));
189
190     gst_caps_unref (peercaps);
191     if (!gst_caps_is_empty (intersect)) {
192       GstStructure *s = gst_caps_get_structure (intersect, 0);
193       encoding_name = g_strdup (gst_structure_get_string (s, "encoding-name"));
194     }
195     gst_caps_unref (intersect);
196   }
197
198   if (!encoding_name)
199     encoding_name = g_strdup ("H263-1998");
200
201   gst_base_rtp_payload_set_options (payload, "video", TRUE,
202       (gchar *) encoding_name, 90000);
203   res = gst_base_rtp_payload_set_outcaps (payload, NULL);
204   g_free (encoding_name);
205
206   return res;
207 }
208
209 static void
210 caps_append (GstCaps * caps, GstStructure * in_s, guint x, guint y, guint mpi)
211 {
212   GstStructure *s;
213
214   if (!in_s)
215     return;
216
217   if (mpi < 1 || mpi > 32)
218     return;
219
220   s = gst_structure_copy (in_s);
221
222   gst_structure_set (s,
223       "width", GST_TYPE_INT_RANGE, 1, x,
224       "height", GST_TYPE_INT_RANGE, 1, y,
225       "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 30000, 1001 * mpi, NULL);
226
227   gst_caps_merge_structure (caps, s);
228 }
229
230
231 static GstCaps *
232 gst_rtp_h263p_pay_sink_getcaps (GstBaseRTPPayload * payload, GstPad * pad,
233     GstCaps * filter)
234 {
235   GstRtpH263PPay *rtph263ppay;
236   GstCaps *caps = gst_caps_new_empty ();
237   GstCaps *peercaps = NULL;
238   GstCaps *intersect = NULL;
239   guint i;
240
241   rtph263ppay = GST_RTP_H263P_PAY (payload);
242
243   peercaps =
244       gst_pad_peer_get_caps (GST_BASE_RTP_PAYLOAD_SRCPAD (payload), filter);
245   if (!peercaps)
246     return
247         gst_caps_copy (gst_pad_get_pad_template_caps
248         (GST_BASE_RTP_PAYLOAD_SINKPAD (payload)));
249
250   intersect = gst_caps_intersect (peercaps,
251       gst_pad_get_pad_template_caps (GST_BASE_RTP_PAYLOAD_SRCPAD (payload)));
252   gst_caps_unref (peercaps);
253
254   if (gst_caps_is_empty (intersect))
255     return intersect;
256
257   for (i = 0; i < gst_caps_get_size (intersect); i++) {
258     GstStructure *s = gst_caps_get_structure (intersect, i);
259     const gchar *encoding_name = gst_structure_get_string (s, "encoding-name");
260
261     if (!strcmp (encoding_name, "H263-2000")) {
262       const gchar *profile_str = gst_structure_get_string (s, "profile");
263       const gchar *level_str = gst_structure_get_string (s, "level");
264       int profile = 0;
265       int level = 0;
266
267       if (profile_str && level_str) {
268         gboolean i = FALSE, j = FALSE, l = FALSE, t = FALSE, f = FALSE,
269             v = FALSE;
270         GstStructure *new_s = gst_structure_new ("video/x-h263",
271             "variant", G_TYPE_STRING, "itu",
272             NULL);
273
274         profile = atoi (profile_str);
275         level = atoi (level_str);
276
277         /* These profiles are defined in the H.263 Annex X */
278         switch (profile) {
279           case 0:
280             /* The Baseline Profile (Profile 0) */
281             break;
282           case 1:
283             /* H.320 Coding Efficiency Version 2 Backward-Compatibility Profile
284              * (Profile 1)
285              * Baseline + Annexes I, J, L.4 and T
286              */
287             i = j = l = t = TRUE;
288             break;
289           case 2:
290             /* Version 1 Backward-Compatibility Profile (Profile 2)
291              * Baseline + Annex F
292              */
293             i = j = l = t = f = TRUE;
294             break;
295           case 3:
296             /* Version 2 Interactive and Streaming Wireless Profile
297              * Baseline + Annexes I, J, T
298              */
299             i = j = t = TRUE;
300             break;
301           case 4:
302             /* Version 3 Interactive and Streaming Wireless Profile (Profile 4)
303              * Baseline + Annexes I, J, T, V, W.6.3.8,
304              */
305             /* Missing W.6.3.8 */
306             i = j = t = v = TRUE;
307             break;
308           case 5:
309             /* Conversational High Compression Profile (Profile 5)
310              * Baseline + Annexes F, I, J, L.4, T, D, U
311              */
312             /* Missing D, U */
313             f = i = j = l = t = TRUE;
314             break;
315           case 6:
316             /* Conversational Internet Profile (Profile 6)
317              * Baseline + Annexes F, I, J, L.4, T, D, U and
318              * K with arbitratry slice ordering
319              */
320             /* Missing D, U, K with arbitratry slice ordering */
321             f = i = j = l = t = TRUE;
322             break;
323           case 7:
324             /* Conversational Interlace Profile (Profile 7)
325              * Baseline + Annexes F, I, J, L.4, T, D, U,  W.6.3.11
326              */
327             /* Missing D, U, W.6.3.11 */
328             f = i = j = l = t = TRUE;
329             break;
330           case 8:
331             /* High Latency Profile (Profile 8)
332              * Baseline + Annexes F, I, J, L.4, T, D, U, P.5, O.1.1 and
333              * K with arbitratry slice ordering
334              */
335             /* Missing D, U, P.5, O.1.1 */
336             f = i = j = l = t = TRUE;
337             break;
338         }
339
340
341         if (f || i || j || t || l || v) {
342           GValue list = { 0 };
343           GValue vstr = { 0 };
344
345           g_value_init (&list, GST_TYPE_LIST);
346           g_value_init (&vstr, G_TYPE_STRING);
347
348           g_value_set_static_string (&vstr, "h263");
349           gst_value_list_append_value (&list, &vstr);
350           g_value_set_static_string (&vstr, "h263p");
351           gst_value_list_append_value (&list, &vstr);
352
353           if (l || v) {
354             g_value_set_static_string (&vstr, "h263pp");
355             gst_value_list_append_value (&list, &vstr);
356           }
357           g_value_unset (&vstr);
358
359           gst_structure_set_value (new_s, "h263version", &list);
360           g_value_unset (&list);
361         } else {
362           gst_structure_set (new_s, "h263version", G_TYPE_STRING, "h263", NULL);
363         }
364
365
366         if (!f)
367           gst_structure_set (new_s, "annex-f", G_TYPE_BOOLEAN, FALSE, NULL);
368         if (!i)
369           gst_structure_set (new_s, "annex-i", G_TYPE_BOOLEAN, FALSE, NULL);
370         if (!j)
371           gst_structure_set (new_s, "annex-j", G_TYPE_BOOLEAN, FALSE, NULL);
372         if (!t)
373           gst_structure_set (new_s, "annex-t", G_TYPE_BOOLEAN, FALSE, NULL);
374         if (!l)
375           gst_structure_set (new_s, "annex-l", G_TYPE_BOOLEAN, FALSE, NULL);
376         if (!v)
377           gst_structure_set (new_s, "annex-v", G_TYPE_BOOLEAN, FALSE, NULL);
378
379
380         if (level <= 10 || level == 45) {
381           gst_structure_set (new_s,
382               "width", GST_TYPE_INT_RANGE, 1, 176,
383               "height", GST_TYPE_INT_RANGE, 1, 144,
384               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 30000, 2002, NULL);
385           gst_caps_merge_structure (caps, new_s);
386         } else if (level <= 20) {
387           GstStructure *s_copy = gst_structure_copy (new_s);
388
389           gst_structure_set (new_s,
390               "width", GST_TYPE_INT_RANGE, 1, 352,
391               "height", GST_TYPE_INT_RANGE, 1, 288,
392               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 30000, 2002, NULL);
393           gst_caps_merge_structure (caps, new_s);
394
395           gst_structure_set (s_copy,
396               "width", GST_TYPE_INT_RANGE, 1, 176,
397               "height", GST_TYPE_INT_RANGE, 1, 144,
398               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 30000, 1001, NULL);
399           gst_caps_merge_structure (caps, s_copy);
400         } else if (level <= 40) {
401
402           gst_structure_set (new_s,
403               "width", GST_TYPE_INT_RANGE, 1, 352,
404               "height", GST_TYPE_INT_RANGE, 1, 288,
405               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 30000, 1001, NULL);
406           gst_caps_merge_structure (caps, new_s);
407         } else if (level <= 50) {
408           GstStructure *s_copy = gst_structure_copy (new_s);
409
410           gst_structure_set (new_s,
411               "width", GST_TYPE_INT_RANGE, 1, 352,
412               "height", GST_TYPE_INT_RANGE, 1, 288,
413               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 50, 1, NULL);
414           gst_caps_merge_structure (caps, new_s);
415
416           gst_structure_set (s_copy,
417               "width", GST_TYPE_INT_RANGE, 1, 352,
418               "height", GST_TYPE_INT_RANGE, 1, 240,
419               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 60000, 1001, NULL);
420           gst_caps_merge_structure (caps, s_copy);
421         } else if (level <= 60) {
422           GstStructure *s_copy = gst_structure_copy (new_s);
423
424           gst_structure_set (new_s,
425               "width", GST_TYPE_INT_RANGE, 1, 720,
426               "height", GST_TYPE_INT_RANGE, 1, 288,
427               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 50, 1, NULL);
428           gst_caps_merge_structure (caps, new_s);
429
430           gst_structure_set (s_copy,
431               "width", GST_TYPE_INT_RANGE, 1, 720,
432               "height", GST_TYPE_INT_RANGE, 1, 240,
433               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 60000, 1001, NULL);
434           gst_caps_merge_structure (caps, s_copy);
435         } else if (level <= 70) {
436           GstStructure *s_copy = gst_structure_copy (new_s);
437
438           gst_structure_set (new_s,
439               "width", GST_TYPE_INT_RANGE, 1, 720,
440               "height", GST_TYPE_INT_RANGE, 1, 576,
441               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 50, 1, NULL);
442           gst_caps_merge_structure (caps, new_s);
443
444           gst_structure_set (s_copy,
445               "width", GST_TYPE_INT_RANGE, 1, 720,
446               "height", GST_TYPE_INT_RANGE, 1, 480,
447               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, 60000, 1001, NULL);
448           gst_caps_merge_structure (caps, s_copy);
449         } else {
450           gst_caps_merge_structure (caps, new_s);
451         }
452
453       } else {
454         GstStructure *new_s = gst_structure_new ("video/x-h263",
455             "variant", G_TYPE_STRING, "itu",
456             "h263version", G_TYPE_STRING, "h263",
457             NULL);
458
459         GST_DEBUG_OBJECT (rtph263ppay, "No profile or level specified"
460             " for H263-2000, defaulting to baseline H263");
461
462         gst_caps_merge_structure (caps, new_s);
463       }
464     } else {
465       gboolean f = FALSE, i = FALSE, j = FALSE, t = FALSE;
466       /* FIXME: ffmpeg support the Appendix K too, how do we express it ?
467        *   guint k;
468        */
469       const gchar *str;
470       GstStructure *new_s = gst_structure_new ("video/x-h263",
471           "variant", G_TYPE_STRING, "itu",
472           NULL);
473       gboolean added = FALSE;
474
475       str = gst_structure_get_string (s, "f");
476       if (str && !strcmp (str, "1"))
477         f = TRUE;
478
479       str = gst_structure_get_string (s, "i");
480       if (str && !strcmp (str, "1"))
481         i = TRUE;
482
483       str = gst_structure_get_string (s, "j");
484       if (str && !strcmp (str, "1"))
485         j = TRUE;
486
487       str = gst_structure_get_string (s, "t");
488       if (str && !strcmp (str, "1"))
489         t = TRUE;
490
491       if (f || i || j || t) {
492         GValue list = { 0 };
493         GValue vstr = { 0 };
494
495         g_value_init (&list, GST_TYPE_LIST);
496         g_value_init (&vstr, G_TYPE_STRING);
497
498         g_value_set_static_string (&vstr, "h263");
499         gst_value_list_append_value (&list, &vstr);
500         g_value_set_static_string (&vstr, "h263p");
501         gst_value_list_append_value (&list, &vstr);
502         g_value_unset (&vstr);
503
504         gst_structure_set_value (new_s, "h263version", &list);
505         g_value_unset (&list);
506       } else {
507         gst_structure_set (new_s, "h263version", G_TYPE_STRING, "h263", NULL);
508       }
509
510       if (!f)
511         gst_structure_set (new_s, "annex-f", G_TYPE_BOOLEAN, FALSE, NULL);
512       if (!i)
513         gst_structure_set (new_s, "annex-i", G_TYPE_BOOLEAN, FALSE, NULL);
514       if (!j)
515         gst_structure_set (new_s, "annex-j", G_TYPE_BOOLEAN, FALSE, NULL);
516       if (!t)
517         gst_structure_set (new_s, "annex-t", G_TYPE_BOOLEAN, FALSE, NULL);
518
519
520       str = gst_structure_get_string (s, "custom");
521       if (str) {
522         unsigned int xmax, ymax, mpi;
523         if (sscanf (str, "%u,%u,%u", &xmax, &ymax, &mpi) == 3) {
524           if (xmax % 4 && ymax % 4 && mpi >= 1 && mpi <= 32) {
525             caps_append (caps, new_s, xmax, ymax, mpi);
526             added = TRUE;
527           } else {
528             GST_WARNING_OBJECT (rtph263ppay, "Invalid custom framesize/MPI"
529                 " %u x %u at %u, ignoring", xmax, ymax, mpi);
530           }
531         } else {
532           GST_WARNING_OBJECT (rtph263ppay, "Invalid custom framesize/MPI: %s,"
533               " ignoring", str);
534         }
535       }
536
537       str = gst_structure_get_string (s, "16cif");
538       if (str) {
539         int mpi = atoi (str);
540         caps_append (caps, new_s, 1408, 1152, mpi);
541         added = TRUE;
542       }
543
544       str = gst_structure_get_string (s, "4cif");
545       if (str) {
546         int mpi = atoi (str);
547         caps_append (caps, new_s, 704, 576, mpi);
548         added = TRUE;
549       }
550
551       str = gst_structure_get_string (s, "cif");
552       if (str) {
553         int mpi = atoi (str);
554         caps_append (caps, new_s, 352, 288, mpi);
555         added = TRUE;
556       }
557
558       str = gst_structure_get_string (s, "qcif");
559       if (str) {
560         int mpi = atoi (str);
561         caps_append (caps, new_s, 176, 144, mpi);
562         added = TRUE;
563       }
564
565       str = gst_structure_get_string (s, "sqcif");
566       if (str) {
567         int mpi = atoi (str);
568         caps_append (caps, new_s, 128, 96, mpi);
569         added = TRUE;
570       }
571
572       if (added)
573         gst_structure_free (new_s);
574       else
575         gst_caps_merge_structure (caps, new_s);
576     }
577   }
578
579   gst_caps_unref (intersect);
580
581   return caps;
582 }
583
584
585 static void
586 gst_rtp_h263p_pay_set_property (GObject * object, guint prop_id,
587     const GValue * value, GParamSpec * pspec)
588 {
589   GstRtpH263PPay *rtph263ppay;
590
591   rtph263ppay = GST_RTP_H263P_PAY (object);
592
593   switch (prop_id) {
594     case PROP_FRAGMENTATION_MODE:
595       rtph263ppay->fragmentation_mode = g_value_get_enum (value);
596       break;
597     default:
598       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
599       break;
600   }
601 }
602
603 static void
604 gst_rtp_h263p_pay_get_property (GObject * object, guint prop_id,
605     GValue * value, GParamSpec * pspec)
606 {
607   GstRtpH263PPay *rtph263ppay;
608
609   rtph263ppay = GST_RTP_H263P_PAY (object);
610
611   switch (prop_id) {
612     case PROP_FRAGMENTATION_MODE:
613       g_value_set_enum (value, rtph263ppay->fragmentation_mode);
614       break;
615     default:
616       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
617       break;
618   }
619 }
620
621 static GstFlowReturn
622 gst_rtp_h263p_pay_flush (GstRtpH263PPay * rtph263ppay)
623 {
624   guint avail;
625   GstBuffer *outbuf;
626   GstFlowReturn ret;
627   gboolean fragmented;
628
629   avail = gst_adapter_available (rtph263ppay->adapter);
630   if (avail == 0)
631     return GST_FLOW_OK;
632
633   fragmented = FALSE;
634   /* This algorithm assumes the H263/+/++ encoder sends complete frames in each
635    * buffer */
636   /* With Fragmentation Mode at GST_FRAGMENTATION_MODE_NORMAL:
637    *  This algorithm implements the Follow-on packets method for packetization.
638    *  This assumes low packet loss network. 
639    * With Fragmentation Mode at GST_FRAGMENTATION_MODE_SYNC:
640    *  This algorithm separates large frames at synchronisation points (Segments)
641    *  (See RFC 4629 section 6). It would be interesting to have a property such as network
642    *  quality to select between both packetization methods */
643   /* TODO Add VRC supprt (See RFC 4629 section 5.2) */
644
645   while (avail > 0) {
646     guint towrite;
647     guint8 *payload;
648     guint payload_len;
649     gint header_len;
650     guint next_gop = 0;
651     gboolean found_gob = FALSE;
652     GstRTPBuffer rtp = { NULL };
653
654     if (rtph263ppay->fragmentation_mode == GST_FRAGMENTATION_MODE_SYNC) {
655       /* start after 1st gop possible */
656       guint parsed_len = 3;
657       const guint8 *parse_data = NULL;
658
659       parse_data = gst_adapter_map (rtph263ppay->adapter, avail);
660
661       /* Check if we have a gob or eos , eossbs */
662       /* FIXME EOS and EOSSBS packets should never contain any gobs and vice-versa */
663       if (avail >= 3 && *parse_data == 0 && *(parse_data + 1) == 0
664           && *(parse_data + 2) >= 0x80) {
665         GST_DEBUG_OBJECT (rtph263ppay, " Found GOB header");
666         found_gob = TRUE;
667       }
668       /* Find next and cut the packet accordingly */
669       /* TODO we should get as many gobs as possible until MTU is reached, this
670        * code seems to just get one GOB per packet */
671       while (parsed_len + 2 < avail) {
672         if (parse_data[parsed_len] == 0 && parse_data[parsed_len + 1] == 0
673             && parse_data[parsed_len + 2] >= 0x80) {
674           next_gop = parsed_len;
675           GST_DEBUG_OBJECT (rtph263ppay, " Next GOB Detected at :  %d",
676               next_gop);
677           break;
678         }
679         parsed_len++;
680       }
681       gst_adapter_unmap (rtph263ppay->adapter);
682     }
683
684     /* for picture start frames (non-fragmented), we need to remove the first
685      * two 0x00 bytes and set P=1 */
686     header_len = (fragmented && !found_gob) ? 2 : 0;
687
688     towrite = MIN (avail, gst_rtp_buffer_calc_payload_len
689         (GST_BASE_RTP_PAYLOAD_MTU (rtph263ppay) - header_len, 0, 0));
690
691     if (next_gop > 0)
692       towrite = MIN (next_gop, towrite);
693
694     payload_len = header_len + towrite;
695
696     outbuf = gst_rtp_buffer_new_allocate (payload_len, 0, 0);
697
698     gst_rtp_buffer_map (outbuf, GST_MAP_WRITE, &rtp);
699     /* last fragment gets the marker bit set */
700     gst_rtp_buffer_set_marker (&rtp, avail > towrite ? 0 : 1);
701
702     payload = gst_rtp_buffer_get_payload (&rtp);
703
704     gst_adapter_copy (rtph263ppay->adapter, &payload[header_len], 0, towrite);
705
706     /*  0                   1
707      *  0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5
708      * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
709      * |   RR    |P|V|   PLEN    |PEBIT|
710      * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
711      */
712     /* if fragmented or gop header , write p bit =1 */
713     payload[0] = (fragmented && !found_gob) ? 0x00 : 0x04;
714     payload[1] = 0;
715
716     GST_BUFFER_TIMESTAMP (outbuf) = rtph263ppay->first_timestamp;
717     GST_BUFFER_DURATION (outbuf) = rtph263ppay->first_duration;
718     gst_rtp_buffer_unmap (&rtp);
719
720     gst_adapter_flush (rtph263ppay->adapter, towrite);
721
722     ret =
723         gst_base_rtp_payload_push (GST_BASE_RTP_PAYLOAD (rtph263ppay), outbuf);
724
725     avail -= towrite;
726     fragmented = TRUE;
727   }
728
729   return ret;
730 }
731
732 static GstFlowReturn
733 gst_rtp_h263p_pay_handle_buffer (GstBaseRTPPayload * payload,
734     GstBuffer * buffer)
735 {
736   GstRtpH263PPay *rtph263ppay;
737   GstFlowReturn ret;
738
739   rtph263ppay = GST_RTP_H263P_PAY (payload);
740
741   rtph263ppay->first_timestamp = GST_BUFFER_TIMESTAMP (buffer);
742   rtph263ppay->first_duration = GST_BUFFER_DURATION (buffer);
743
744   /* we always encode and flush a full picture */
745   gst_adapter_push (rtph263ppay->adapter, buffer);
746   ret = gst_rtp_h263p_pay_flush (rtph263ppay);
747
748   return ret;
749 }
750
751 gboolean
752 gst_rtp_h263p_pay_plugin_init (GstPlugin * plugin)
753 {
754   return gst_element_register (plugin, "rtph263ppay",
755       GST_RANK_SECONDARY, GST_TYPE_RTP_H263P_PAY);
756 }