2 * Copyright (C) <2009> Wim Taymans <wim.taymans@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., 51 Franklin St, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
26 #include <gst/rtp/gstrtpbuffer.h>
27 #include <gst/audio/audio.h>
29 #include "gstrtpceltdepay.h"
30 #include "gstrtputils.h"
32 GST_DEBUG_CATEGORY_STATIC (rtpceltdepay_debug);
33 #define GST_CAT_DEFAULT (rtpceltdepay_debug)
35 /* RtpCELTDepay signals and args */
37 #define DEFAULT_FRAMESIZE 480
38 #define DEFAULT_CHANNELS 1
39 #define DEFAULT_CLOCKRATE 32000
52 static GstStaticPadTemplate gst_rtp_celt_depay_sink_template =
53 GST_STATIC_PAD_TEMPLATE ("sink",
56 GST_STATIC_CAPS ("application/x-rtp, "
57 "media = (string) \"audio\", "
58 "clock-rate = (int) [32000, 48000], "
59 "encoding-name = (string) \"CELT\"")
62 static GstStaticPadTemplate gst_rtp_celt_depay_src_template =
63 GST_STATIC_PAD_TEMPLATE ("src",
66 GST_STATIC_CAPS ("audio/x-celt")
69 static GstBuffer *gst_rtp_celt_depay_process (GstRTPBaseDepayload * depayload,
71 static gboolean gst_rtp_celt_depay_setcaps (GstRTPBaseDepayload * depayload,
74 #define gst_rtp_celt_depay_parent_class parent_class
75 G_DEFINE_TYPE (GstRtpCELTDepay, gst_rtp_celt_depay,
76 GST_TYPE_RTP_BASE_DEPAYLOAD);
79 gst_rtp_celt_depay_class_init (GstRtpCELTDepayClass * klass)
81 GstElementClass *gstelement_class;
82 GstRTPBaseDepayloadClass *gstrtpbasedepayload_class;
84 GST_DEBUG_CATEGORY_INIT (rtpceltdepay_debug, "rtpceltdepay", 0,
85 "CELT RTP Depayloader");
87 gstelement_class = (GstElementClass *) klass;
88 gstrtpbasedepayload_class = (GstRTPBaseDepayloadClass *) klass;
90 gst_element_class_add_static_pad_template (gstelement_class,
91 &gst_rtp_celt_depay_src_template);
92 gst_element_class_add_static_pad_template (gstelement_class,
93 &gst_rtp_celt_depay_sink_template);
95 gst_element_class_set_static_metadata (gstelement_class,
96 "RTP CELT depayloader", "Codec/Depayloader/Network/RTP",
97 "Extracts CELT audio from RTP packets",
98 "Wim Taymans <wim.taymans@gmail.com>");
100 gstrtpbasedepayload_class->process_rtp_packet = gst_rtp_celt_depay_process;
101 gstrtpbasedepayload_class->set_caps = gst_rtp_celt_depay_setcaps;
105 gst_rtp_celt_depay_init (GstRtpCELTDepay * rtpceltdepay)
110 * vendor string (len bytes),
111 * user_len 4 (0) bytes LE
113 static const gchar gst_rtp_celt_comment[] =
114 "\045\0\0\0Depayloaded with GStreamer celtdepay\0\0\0\0";
117 gst_rtp_celt_depay_setcaps (GstRTPBaseDepayload * depayload, GstCaps * caps)
119 GstStructure *structure;
120 GstRtpCELTDepay *rtpceltdepay;
121 gint clock_rate, nb_channels = 0, frame_size = 0;
129 rtpceltdepay = GST_RTP_CELT_DEPAY (depayload);
131 structure = gst_caps_get_structure (caps, 0);
133 if (!gst_structure_get_int (structure, "clock-rate", &clock_rate))
135 depayload->clock_rate = clock_rate;
137 if ((params = gst_structure_get_string (structure, "encoding-params")))
138 nb_channels = atoi (params);
140 nb_channels = DEFAULT_CHANNELS;
142 if ((params = gst_structure_get_string (structure, "frame-size")))
143 frame_size = atoi (params);
145 frame_size = DEFAULT_FRAMESIZE;
146 rtpceltdepay->frame_size = frame_size;
148 GST_DEBUG_OBJECT (depayload, "clock-rate=%d channels=%d frame-size=%d",
149 clock_rate, nb_channels, frame_size);
151 /* construct minimal header and comment packet for the decoder */
152 buf = gst_buffer_new_and_alloc (60);
153 gst_buffer_map (buf, &map, GST_MAP_WRITE);
155 memcpy (ptr, "CELT ", 8);
157 memcpy (ptr, "1.1.12", 7);
159 GST_WRITE_UINT32_LE (ptr, 0x80000006); /* version */
161 GST_WRITE_UINT32_LE (ptr, 56); /* header_size */
163 GST_WRITE_UINT32_LE (ptr, clock_rate); /* rate */
165 GST_WRITE_UINT32_LE (ptr, nb_channels); /* channels */
167 GST_WRITE_UINT32_LE (ptr, frame_size); /* frame-size */
169 GST_WRITE_UINT32_LE (ptr, -1); /* overlap */
171 GST_WRITE_UINT32_LE (ptr, -1); /* bytes_per_packet */
173 GST_WRITE_UINT32_LE (ptr, 0); /* extra headers */
174 gst_buffer_unmap (buf, &map);
176 srccaps = gst_caps_new_empty_simple ("audio/x-celt");
177 res = gst_pad_set_caps (depayload->srcpad, srccaps);
178 gst_caps_unref (srccaps);
180 gst_rtp_base_depayload_push (GST_RTP_BASE_DEPAYLOAD (rtpceltdepay), buf);
182 buf = gst_buffer_new_and_alloc (sizeof (gst_rtp_celt_comment));
183 gst_buffer_fill (buf, 0, gst_rtp_celt_comment, sizeof (gst_rtp_celt_comment));
185 gst_rtp_base_depayload_push (GST_RTP_BASE_DEPAYLOAD (rtpceltdepay), buf);
192 GST_ERROR_OBJECT (depayload, "no clock-rate specified");
198 gst_rtp_celt_depay_process (GstRTPBaseDepayload * depayload, GstRTPBuffer * rtp)
200 GstBuffer *outbuf = NULL;
202 guint offset, pos, payload_len, total_size, size;
204 gint clock_rate = 0, frame_size = 0;
205 GstClockTime framesize_ns = 0, timestamp;
207 GstRtpCELTDepay *rtpceltdepay;
209 rtpceltdepay = GST_RTP_CELT_DEPAY (depayload);
210 clock_rate = depayload->clock_rate;
211 frame_size = rtpceltdepay->frame_size;
212 framesize_ns = gst_util_uint64_scale_int (frame_size, GST_SECOND, clock_rate);
214 timestamp = GST_BUFFER_PTS (rtp->buffer);
216 GST_LOG_OBJECT (depayload,
217 "got %" G_GSIZE_FORMAT " bytes, mark %d ts %u seqn %d",
218 gst_buffer_get_size (rtp->buffer), gst_rtp_buffer_get_marker (rtp),
219 gst_rtp_buffer_get_timestamp (rtp), gst_rtp_buffer_get_seq (rtp));
221 GST_LOG_OBJECT (depayload, "got clock-rate=%d, frame_size=%d, "
222 "_ns=%" GST_TIME_FORMAT ", timestamp=%" GST_TIME_FORMAT, clock_rate,
223 frame_size, GST_TIME_ARGS (framesize_ns), GST_TIME_ARGS (timestamp));
225 payload = gst_rtp_buffer_get_payload (rtp);
226 payload_len = gst_rtp_buffer_get_payload_len (rtp);
228 /* first count how many bytes are consumed by the size headers and make offset
229 * point to the first data byte */
232 while (total_size < payload_len) {
234 s = payload[offset++];
239 /* offset is now pointing to the payload */
242 while (total_size < payload_len) {
251 outbuf = gst_rtp_buffer_get_payload_subbuffer (rtp, offset, size);
254 if (frame_size != -1 && clock_rate != -1) {
255 GST_BUFFER_PTS (outbuf) = timestamp + framesize_ns * n;
256 GST_BUFFER_DURATION (outbuf) = framesize_ns;
258 GST_LOG_OBJECT (depayload, "push timestamp=%"
259 GST_TIME_FORMAT ", duration=%" GST_TIME_FORMAT,
260 GST_TIME_ARGS (GST_BUFFER_PTS (outbuf)),
261 GST_TIME_ARGS (GST_BUFFER_DURATION (outbuf)));
263 gst_rtp_drop_non_audio_meta (depayload, outbuf);
265 gst_rtp_base_depayload_push (depayload, outbuf);
272 gst_rtp_celt_depay_plugin_init (GstPlugin * plugin)
274 return gst_element_register (plugin, "rtpceltdepay",
275 GST_RANK_SECONDARY, GST_TYPE_RTP_CELT_DEPAY);