vorbisenc.c 10.2 KB
Newer Older
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
/* Gnome-Streamer
 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */


#include <stdlib.h>
#include <string.h>

#include <vorbis/vorbisenc.h>

#include "vorbisenc.h"



extern GstPadTemplate *enc_src_template, *enc_sink_template;

/* elementfactory information */
GstElementDetails vorbisenc_details = {
  "Ogg Vorbis encoder",
  "Filter/Audio/Encoder",
  "Encodes audio in OGG Vorbis format",
  VERSION,
Wim Taymans's avatar
Wim Taymans committed
38
  "Monty <monty@xiph.org>, " 
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
39
40
41
42
43
  "Wim Taymans <wim.taymans@chello.be>",
  "(C) 2000",
};

/* VorbisEnc signals and args */
Wim Taymans's avatar
Wim Taymans committed
44
45
enum
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
46
47
48
49
  /* FILL ME */
  LAST_SIGNAL
};

Wim Taymans's avatar
Wim Taymans committed
50
51
enum
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
52
53
54
55
  ARG_0,
  ARG_BITRATE,
};

Wim Taymans's avatar
Wim Taymans committed
56
57
static void 	gst_vorbisenc_class_init 	(VorbisEncClass * klass);
static void 	gst_vorbisenc_init 		(VorbisEnc * vorbisenc);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
58

Wim Taymans's avatar
Wim Taymans committed
59
60
static void 	gst_vorbisenc_chain 		(GstPad * pad, GstBuffer * buf);
static void 	gst_vorbisenc_setup 		(VorbisEnc * vorbisenc);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
61

Wim Taymans's avatar
Wim Taymans committed
62
63
64
65
static void 	gst_vorbisenc_get_property 	(GObject * object, guint prop_id, GValue * value,
						 GParamSpec * pspec);
static void 	gst_vorbisenc_set_property 	(GObject * object, guint prop_id, const GValue * value,
						 GParamSpec * pspec);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
66
67
68
69
70
71
72
73
74
75
76

static GstElementClass *parent_class = NULL;
/*static guint gst_vorbisenc_signals[LAST_SIGNAL] = { 0 }; */

GType
vorbisenc_get_type (void)
{
  static GType vorbisenc_type = 0;

  if (!vorbisenc_type) {
    static const GTypeInfo vorbisenc_info = {
Wim Taymans's avatar
Wim Taymans committed
77
78
      sizeof (VorbisEncClass), 
      NULL,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
79
      NULL,
Wim Taymans's avatar
Wim Taymans committed
80
      (GClassInitFunc) gst_vorbisenc_class_init,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
81
82
      NULL,
      NULL,
Wim Taymans's avatar
Wim Taymans committed
83
      sizeof (VorbisEnc),
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
84
      0,
Wim Taymans's avatar
Wim Taymans committed
85
      (GInstanceInitFunc) gst_vorbisenc_init,
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
86
    };
Wim Taymans's avatar
Wim Taymans committed
87
88

    vorbisenc_type = g_type_register_static (GST_TYPE_ELEMENT, "VorbisEnc", &vorbisenc_info, 0);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
89
90
91
92
93
  }
  return vorbisenc_type;
}

static void
Wim Taymans's avatar
Wim Taymans committed
94
gst_vorbisenc_class_init (VorbisEncClass * klass)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
95
96
97
98
{
  GObjectClass *gobject_class;
  GstElementClass *gstelement_class;

Wim Taymans's avatar
Wim Taymans committed
99
100
  gobject_class = (GObjectClass *) klass;
  gstelement_class = (GstElementClass *) klass;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
101

Wim Taymans's avatar
Wim Taymans committed
102
103
104
  g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BITRATE, 
    g_param_spec_int ("bitrate", "bitrate", "bitrate", 
	    G_MININT, G_MAXINT, 0, G_PARAM_READWRITE));
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
105

Wim Taymans's avatar
Wim Taymans committed
106
  parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
107
108
109
110
111

  gobject_class->set_property = gst_vorbisenc_set_property;
  gobject_class->get_property = gst_vorbisenc_get_property;
}

112
113
static GstPadConnectReturn
gst_vorbisenc_sinkconnect (GstPad * pad, GstCaps * caps)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
114
115
116
117
118
{
  VorbisEnc *vorbisenc;

  vorbisenc = GST_VORBISENC (gst_pad_get_parent (pad));

119
120
121
  if (!GST_CAPS_IS_FIXED (caps))
    return GST_PAD_CONNECT_DELAYED;

Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
122
123
124
125
  vorbisenc->channels = gst_caps_get_int (caps, "channels");
  vorbisenc->frequency = gst_caps_get_int (caps, "rate");

  gst_vorbisenc_setup (vorbisenc);
126
127
128
129
130

  if (vorbisenc->setup)
    return GST_PAD_CONNECT_OK;

  return GST_PAD_CONNECT_REFUSED;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
131
132
133
}

static void
Wim Taymans's avatar
Wim Taymans committed
134
gst_vorbisenc_init (VorbisEnc * vorbisenc)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
135
136
{
  vorbisenc->sinkpad = gst_pad_new_from_template (enc_sink_template, "sink");
Wim Taymans's avatar
Wim Taymans committed
137
138
  gst_element_add_pad (GST_ELEMENT (vorbisenc), vorbisenc->sinkpad);
  gst_pad_set_chain_function (vorbisenc->sinkpad, gst_vorbisenc_chain);
139
  gst_pad_set_connect_function (vorbisenc->sinkpad, gst_vorbisenc_sinkconnect);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
140
141

  vorbisenc->srcpad = gst_pad_new_from_template (enc_src_template, "src");
Wim Taymans's avatar
Wim Taymans committed
142
  gst_element_add_pad (GST_ELEMENT (vorbisenc), vorbisenc->srcpad);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
143
144
145
146

  vorbisenc->channels = 2;
  vorbisenc->frequency = 44100;
  vorbisenc->bitrate = 128000;
Wim Taymans's avatar
Wim Taymans committed
147
148
149
150
  vorbisenc->setup = FALSE;

  /* we're chained and we can deal with events */
  GST_FLAG_SET (vorbisenc, GST_ELEMENT_EVENT_AWARE);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
151
152
153
}

static void
Wim Taymans's avatar
Wim Taymans committed
154
gst_vorbisenc_setup (VorbisEnc * vorbisenc)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
155
{
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
156
  static const gchar *comment = "Track encoded with GStreamer";
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
157
158
159
160
  /********** Encode setup ************/

  /* choose an encoding mode */
  /* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
Wim Taymans's avatar
Wim Taymans committed
161
162
163
  vorbis_info_init (&vorbisenc->vi);
  vorbis_encode_init (&vorbisenc->vi, vorbisenc->channels, vorbisenc->frequency,
		      -1, vorbisenc->bitrate, -1);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
164
165

  /* add a comment */
Wim Taymans's avatar
Wim Taymans committed
166
  vorbis_comment_init (&vorbisenc->vc);
Wim Taymans's avatar
Wim Taymans committed
167
  vorbis_comment_add (&vorbisenc->vc, (gchar *)comment);
Wim Taymans's avatar
Wim Taymans committed
168
169
  gst_element_send_event (GST_ELEMENT (vorbisenc),
             gst_event_new_info ("comment", GST_PROPS_STRING (comment), NULL));
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
170
171

  /* set up the analysis state and auxiliary encoding storage */
Wim Taymans's avatar
Wim Taymans committed
172
173
  vorbis_analysis_init (&vorbisenc->vd, &vorbisenc->vi);
  vorbis_block_init (&vorbisenc->vd, &vorbisenc->vb);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
174
175
176
177

  /* set up our packet->stream encoder */
  /* pick a random serial number; that way we can more likely build
     chained streams just by concatenation */
Wim Taymans's avatar
Wim Taymans committed
178
179
  srand (time (NULL));
  ogg_stream_init (&vorbisenc->os, rand ());
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
180
181
182
183
184
185
186
187
188
189
190
191
192

  /* Vorbis streams begin with three headers; the initial header (with
     most of the codec setup parameters) which is mandated by the Ogg
     bitstream spec.  The second header holds any comment fields.  The
     third header holds the bitstream codebook.  We merely need to
     make the headers, then pass them to libvorbis one at a time;
     libvorbis handles the additional Ogg bitstream constraints */

  {
    ogg_packet header;
    ogg_packet header_comm;
    ogg_packet header_code;

Wim Taymans's avatar
Wim Taymans committed
193
194
195
196
197
    vorbis_analysis_headerout (&vorbisenc->vd, &vorbisenc->vc, &header, &header_comm, &header_code);
    ogg_stream_packetin (&vorbisenc->os, &header);	/* automatically placed in its own
							   page */
    ogg_stream_packetin (&vorbisenc->os, &header_comm);
    ogg_stream_packetin (&vorbisenc->os, &header_code);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
198
199
200

    /* no need to write out here.  We'll get to that in the main loop */
  }
Wim Taymans's avatar
Wim Taymans committed
201
202

  vorbisenc->setup = TRUE;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
203
204
205
}

static void
Wim Taymans's avatar
Wim Taymans committed
206
gst_vorbisenc_chain (GstPad * pad, GstBuffer * buf)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
207
208
209
{
  VorbisEnc *vorbisenc;

Wim Taymans's avatar
Wim Taymans committed
210
211
212
  g_return_if_fail (pad != NULL);
  g_return_if_fail (GST_IS_PAD (pad));
  g_return_if_fail (buf != NULL);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
213
214
215

  vorbisenc = GST_VORBISENC (gst_pad_get_parent (pad));

Wim Taymans's avatar
Wim Taymans committed
216
  if (!vorbisenc->setup) {
Wim Taymans's avatar
Wim Taymans committed
217
    gst_element_error (GST_ELEMENT (vorbisenc), "encoder not initialized (input is not audio?)");
Wim Taymans's avatar
Wim Taymans committed
218
219
220
221
222
223
    if (GST_IS_BUFFER (buf))
      gst_buffer_unref (buf);
    else
      gst_pad_event_default (pad, GST_EVENT (buf));
    return;
  }
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
224

Wim Taymans's avatar
Wim Taymans committed
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
  if (GST_IS_EVENT (buf)) {
    switch (GST_EVENT_TYPE (buf)) {
      case GST_EVENT_EOS:
        /* end of file.  this can be done implicitly in the mainline,
           but it's easier to see here in non-clever fashion.
           Tell the library we're at end of stream so that it can handle
           the last frame and mark end of stream in the output properly */
        vorbis_analysis_wrote (&vorbisenc->vd, 0);
      default:
	gst_pad_event_default (pad, GST_EVENT (buf));
	break;
    }
  }
  else {
    gint16 *data;
    gulong size;
    gulong i, j;
    float **buffer;
  
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
244
    /* data to encode */
Wim Taymans's avatar
Wim Taymans committed
245
246
    data = (gint16 *) GST_BUFFER_DATA (buf);
    size = GST_BUFFER_SIZE (buf) / 2;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
247
248

    /* expose the buffer to submit data */
Wim Taymans's avatar
Wim Taymans committed
249
    buffer = vorbis_analysis_buffer (&vorbisenc->vd, size / vorbisenc->channels);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
250
251

    /* uninterleave samples */
Wim Taymans's avatar
Wim Taymans committed
252
253
254
    for (i = 0; i < size / vorbisenc->channels; i++) {
      for (j = 0; j < vorbisenc->channels; j++)
	buffer[j][i] = data[i * vorbisenc->channels + j] / 32768.f;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
255
256
257
    }

    /* tell the library how much we actually submitted */
Wim Taymans's avatar
Wim Taymans committed
258
259
260
    vorbis_analysis_wrote (&vorbisenc->vd, size / vorbisenc->channels);

    gst_buffer_unref (buf);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
261
262
263
264
265
  }

  /* vorbis does some data preanalysis, then divvies up blocks for
     more involved (potentially parallel) processing.  Get a single
     block for encoding now */
Wim Taymans's avatar
Wim Taymans committed
266
  while (vorbis_analysis_blockout (&vorbisenc->vd, &vorbisenc->vb) == 1) {
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
267
268

    /* analysis */
Wim Taymans's avatar
Wim Taymans committed
269
    vorbis_analysis (&vorbisenc->vb, &vorbisenc->op);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
270
271

    /* weld the packet into the bitstream */
Wim Taymans's avatar
Wim Taymans committed
272
    ogg_stream_packetin (&vorbisenc->os, &vorbisenc->op);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
273
274

    /* write out pages (if any) */
Wim Taymans's avatar
Wim Taymans committed
275
276
277
    while (!vorbisenc->eos) {
      int result = ogg_stream_pageout (&vorbisenc->os, &vorbisenc->og);
      GstBuffer *outbuf;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
278

Wim Taymans's avatar
Wim Taymans committed
279
280
      if (result == 0)
	break;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
281

Wim Taymans's avatar
Wim Taymans committed
282
283
284
      outbuf = gst_buffer_new ();
      GST_BUFFER_DATA (outbuf) = g_malloc (vorbisenc->og.header_len + vorbisenc->og.body_len);
      GST_BUFFER_SIZE (outbuf) = vorbisenc->og.header_len + vorbisenc->og.body_len;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
285

Wim Taymans's avatar
Wim Taymans committed
286
287
288
      memcpy (GST_BUFFER_DATA (outbuf), vorbisenc->og.header, vorbisenc->og.header_len);
      memcpy (GST_BUFFER_DATA (outbuf) + vorbisenc->og.header_len, vorbisenc->og.body,
	      vorbisenc->og.body_len);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
289

Wim Taymans's avatar
Wim Taymans committed
290
291
292
      GST_DEBUG (0, "vorbisenc: encoded buffer of %d bytes\n", GST_BUFFER_SIZE (outbuf));

      gst_pad_push (vorbisenc->srcpad, outbuf);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
293
294
295

      /* this could be set above, but for illustrative purposes, I do
         it here (to show that vorbis does know where the stream ends) */
Wim Taymans's avatar
Wim Taymans committed
296
297
      if (ogg_page_eos (&vorbisenc->og)) {
	vorbisenc->eos = 1;
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
298
299
300
301
302
303
304
      }
    }
  }

  if (vorbisenc->eos) {
    /* clean up and exit.  vorbis_info_clear() must be called last */

Wim Taymans's avatar
Wim Taymans committed
305
306
307
308
    ogg_stream_clear (&vorbisenc->os);
    vorbis_block_clear (&vorbisenc->vb);
    vorbis_dsp_clear (&vorbisenc->vd);
    vorbis_info_clear (&vorbisenc->vi);
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
309
310
311
312
313
  }

}

static void
Wim Taymans's avatar
Wim Taymans committed
314
gst_vorbisenc_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
{
  VorbisEnc *vorbisenc;

  /* it's not null if we got it, but it might not be ours */
  g_return_if_fail (GST_IS_VORBISENC (object));

  vorbisenc = GST_VORBISENC (object);

  switch (prop_id) {
    case ARG_BITRATE:
      g_value_set_int (value, vorbisenc->bitrate);
      break;
    default:
      break;
  }
}

static void
Wim Taymans's avatar
Wim Taymans committed
333
334
gst_vorbisenc_set_property (GObject * object, guint prop_id, const GValue * value,
			    GParamSpec * pspec)
Thomas Vander Stichele's avatar
Thomas Vander Stichele committed
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
{
  VorbisEnc *vorbisenc;

  /* it's not null if we got it, but it might not be ours */
  g_return_if_fail (GST_IS_VORBISENC (object));

  vorbisenc = GST_VORBISENC (object);

  switch (prop_id) {
    case ARG_BITRATE:
      vorbisenc->bitrate = g_value_get_int (value);
      break;
    default:
      break;
  }
}