2 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
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.
20 * SECTION:element-flacenc
21 * @see_also: #GstFlacDec
23 * flacenc encodes FLAC streams.
24 * <ulink url="http://flac.sourceforge.net/">FLAC</ulink>
25 * is a Free Lossless Audio Codec. FLAC audio can directly be written into
26 * a file, or embedded into containers such as oggmux or matroskamux.
29 * <title>Example launch line</title>
31 * gst-launch-1.0 audiotestsrc num-buffers=100 ! flacenc ! filesink location=beep.flac
32 * ]| Encode a short sine wave into FLAC
34 * gst-launch-1.0 cdparanoiasrc mode=continuous ! queue ! audioconvert ! flacenc ! filesink location=cd.flac
35 * ]| Rip a whole audio CD into a single FLAC file, with the track table saved as a CUE sheet inside the FLAC file
37 * gst-launch-1.0 cdparanoiasrc track=5 ! queue ! audioconvert ! flacenc ! filesink location=track5.flac
38 * ]| Rip track 5 of an audio CD and encode it losslessly to a FLAC file
42 /* TODO: - We currently don't handle discontinuities in the stream in a useful
43 * way and instead rely on the developer plugging in audiorate if
44 * the stream contains discontinuities.
53 #include <gstflacenc.h>
54 #include <gst/audio/audio.h>
55 #include <gst/tag/tag.h>
56 #include <gst/gsttagsetter.h>
58 /* Taken from http://flac.sourceforge.net/format.html#frame_header */
59 static const GstAudioChannelPosition channel_positions[8][8] = {
60 {GST_AUDIO_CHANNEL_POSITION_MONO},
61 {GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
62 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT}, {
63 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
64 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
65 GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER}, {
66 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
67 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
68 GST_AUDIO_CHANNEL_POSITION_REAR_LEFT,
69 GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT}, {
70 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
71 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
72 GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER,
73 GST_AUDIO_CHANNEL_POSITION_REAR_LEFT,
74 GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT}, {
75 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
76 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
77 GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER,
78 GST_AUDIO_CHANNEL_POSITION_LFE1,
79 GST_AUDIO_CHANNEL_POSITION_REAR_LEFT,
80 GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT},
81 /* FIXME: 7/8 channel layouts are not defined in the FLAC specs */
83 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
84 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
85 GST_AUDIO_CHANNEL_POSITION_REAR_LEFT,
86 GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT,
87 GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER,
88 GST_AUDIO_CHANNEL_POSITION_LFE1,
89 GST_AUDIO_CHANNEL_POSITION_REAR_CENTER}, {
90 GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT,
91 GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT,
92 GST_AUDIO_CHANNEL_POSITION_REAR_LEFT,
93 GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT,
94 GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER,
95 GST_AUDIO_CHANNEL_POSITION_LFE1,
96 GST_AUDIO_CHANNEL_POSITION_SIDE_LEFT,
97 GST_AUDIO_CHANNEL_POSITION_SIDE_RIGHT}
100 static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
103 GST_STATIC_CAPS ("audio/x-flac")
110 PROP_STREAMABLE_SUBSET,
111 PROP_MID_SIDE_STEREO,
112 PROP_LOOSE_MID_SIDE_STEREO,
115 PROP_QLP_COEFF_PRECISION,
116 PROP_QLP_COEFF_PREC_SEARCH,
118 PROP_EXHAUSTIVE_MODEL_SEARCH,
119 PROP_MIN_RESIDUAL_PARTITION_ORDER,
120 PROP_MAX_RESIDUAL_PARTITION_ORDER,
121 PROP_RICE_PARAMETER_SEARCH_DIST,
126 GST_DEBUG_CATEGORY_STATIC (flacenc_debug);
127 #define GST_CAT_DEFAULT flacenc_debug
129 #define gst_flac_enc_parent_class parent_class
130 G_DEFINE_TYPE_WITH_CODE (GstFlacEnc, gst_flac_enc, GST_TYPE_AUDIO_ENCODER,
131 G_IMPLEMENT_INTERFACE (GST_TYPE_TAG_SETTER, NULL)
132 G_IMPLEMENT_INTERFACE (GST_TYPE_TOC_SETTER, NULL)
135 static gboolean gst_flac_enc_start (GstAudioEncoder * enc);
136 static gboolean gst_flac_enc_stop (GstAudioEncoder * enc);
137 static gboolean gst_flac_enc_set_format (GstAudioEncoder * enc,
138 GstAudioInfo * info);
139 static GstFlowReturn gst_flac_enc_handle_frame (GstAudioEncoder * enc,
141 static GstCaps *gst_flac_enc_getcaps (GstAudioEncoder * enc, GstCaps * filter);
142 static gboolean gst_flac_enc_sink_event (GstAudioEncoder * enc,
144 static gboolean gst_flac_enc_sink_query (GstAudioEncoder * enc,
147 static void gst_flac_enc_finalize (GObject * object);
149 static GstCaps *gst_flac_enc_generate_sink_caps (void);
151 static gboolean gst_flac_enc_update_quality (GstFlacEnc * flacenc,
153 static void gst_flac_enc_set_property (GObject * object, guint prop_id,
154 const GValue * value, GParamSpec * pspec);
155 static void gst_flac_enc_get_property (GObject * object, guint prop_id,
156 GValue * value, GParamSpec * pspec);
158 static FLAC__StreamEncoderWriteStatus
159 gst_flac_enc_write_callback (const FLAC__StreamEncoder * encoder,
160 const FLAC__byte buffer[], size_t bytes,
161 unsigned samples, unsigned current_frame, void *client_data);
162 static FLAC__StreamEncoderSeekStatus
163 gst_flac_enc_seek_callback (const FLAC__StreamEncoder * encoder,
164 FLAC__uint64 absolute_byte_offset, void *client_data);
165 static FLAC__StreamEncoderTellStatus
166 gst_flac_enc_tell_callback (const FLAC__StreamEncoder * encoder,
167 FLAC__uint64 * absolute_byte_offset, void *client_data);
171 gboolean exhaustive_model_search;
172 gboolean escape_coding;
174 gboolean loose_mid_side;
175 guint qlp_coeff_precision;
176 gboolean qlp_coeff_prec_search;
177 guint min_residual_partition_order;
178 guint max_residual_partition_order;
179 guint rice_parameter_search_dist;
185 static const GstFlacEncParams flacenc_params[] = {
186 {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 2, 2, 0, 0, 1152},
187 {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 2, 2, 0, 0, 1152},
188 {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 3, 0, 0, 1152},
189 {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 3, 3, 0, 6, 4608},
190 {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 3, 3, 0, 8, 4608},
191 {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 3, 3, 0, 8, 4608},
192 {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 4, 0, 8, 4608},
193 {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 8, 4608},
194 {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 12, 4608},
195 {TRUE, TRUE, TRUE, FALSE, 0, FALSE, 0, 16, 0, 32, 4608},
198 #define DEFAULT_QUALITY 5
199 #define DEFAULT_PADDING 0
200 #define DEFAULT_SEEKPOINTS -10
202 #define GST_TYPE_FLAC_ENC_QUALITY (gst_flac_enc_quality_get_type ())
204 gst_flac_enc_quality_get_type (void)
206 static GType qtype = 0;
209 static const GEnumValue values[] = {
210 {0, "0 - Fastest compression", "0"},
215 {5, "5 - Default", "5"},
218 {8, "8 - Highest compression", "8"},
219 {9, "9 - Insane", "9"},
223 qtype = g_enum_register_static ("GstFlacEncQuality", values);
229 gst_flac_enc_class_init (GstFlacEncClass * klass)
231 GObjectClass *gobject_class;
232 GstElementClass *gstelement_class;
233 GstAudioEncoderClass *base_class;
235 GstPadTemplate *sink_templ;
237 gobject_class = (GObjectClass *) klass;
238 gstelement_class = (GstElementClass *) klass;
239 base_class = (GstAudioEncoderClass *) (klass);
241 GST_DEBUG_CATEGORY_INIT (flacenc_debug, "flacenc", 0,
242 "Flac encoding element");
244 gobject_class->set_property = gst_flac_enc_set_property;
245 gobject_class->get_property = gst_flac_enc_get_property;
246 gobject_class->finalize = gst_flac_enc_finalize;
248 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_QUALITY,
249 g_param_spec_enum ("quality",
251 "Speed versus compression tradeoff",
252 GST_TYPE_FLAC_ENC_QUALITY, DEFAULT_QUALITY,
253 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
254 g_object_class_install_property (G_OBJECT_CLASS (klass),
255 PROP_STREAMABLE_SUBSET, g_param_spec_boolean ("streamable-subset",
257 "true to limit encoder to generating a Subset stream, else false",
259 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
260 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MID_SIDE_STEREO,
261 g_param_spec_boolean ("mid-side-stereo", "Do mid side stereo",
262 "Do mid side stereo (only for stereo input)",
263 flacenc_params[DEFAULT_QUALITY].mid_side,
264 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
265 g_object_class_install_property (G_OBJECT_CLASS (klass),
266 PROP_LOOSE_MID_SIDE_STEREO, g_param_spec_boolean ("loose-mid-side-stereo",
267 "Loose mid side stereo", "Loose mid side stereo",
268 flacenc_params[DEFAULT_QUALITY].loose_mid_side,
269 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
270 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BLOCKSIZE,
271 g_param_spec_uint ("blocksize", "Blocksize", "Blocksize in samples",
272 FLAC__MIN_BLOCK_SIZE, FLAC__MAX_BLOCK_SIZE,
273 flacenc_params[DEFAULT_QUALITY].blocksize,
274 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
275 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MAX_LPC_ORDER,
276 g_param_spec_uint ("max-lpc-order", "Max LPC order",
277 "Max LPC order; 0 => use only fixed predictors", 0,
278 FLAC__MAX_LPC_ORDER, flacenc_params[DEFAULT_QUALITY].max_lpc_order,
279 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
280 g_object_class_install_property (G_OBJECT_CLASS (klass),
281 PROP_QLP_COEFF_PRECISION, g_param_spec_uint ("qlp-coeff-precision",
282 "QLP coefficients precision",
283 "Precision in bits of quantized linear-predictor coefficients; 0 = automatic",
284 0, 32, flacenc_params[DEFAULT_QUALITY].qlp_coeff_precision,
285 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
286 g_object_class_install_property (G_OBJECT_CLASS (klass),
287 PROP_QLP_COEFF_PREC_SEARCH, g_param_spec_boolean ("qlp-coeff-prec-search",
288 "Do QLP coefficients precision search",
289 "false = use qlp_coeff_precision, "
290 "true = search around qlp_coeff_precision, take best",
291 flacenc_params[DEFAULT_QUALITY].qlp_coeff_prec_search,
292 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
293 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ESCAPE_CODING,
294 g_param_spec_boolean ("escape-coding", "Do Escape coding",
295 "search for escape codes in the entropy coding stage "
296 "for slightly better compression",
297 flacenc_params[DEFAULT_QUALITY].escape_coding,
298 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
299 g_object_class_install_property (G_OBJECT_CLASS (klass),
300 PROP_EXHAUSTIVE_MODEL_SEARCH,
301 g_param_spec_boolean ("exhaustive-model-search",
302 "Do exhaustive model search",
303 "do exhaustive search of LP coefficient quantization (expensive!)",
304 flacenc_params[DEFAULT_QUALITY].exhaustive_model_search,
305 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
306 g_object_class_install_property (G_OBJECT_CLASS (klass),
307 PROP_MIN_RESIDUAL_PARTITION_ORDER,
308 g_param_spec_uint ("min-residual-partition-order",
309 "Min residual partition order",
310 "Min residual partition order (above 4 doesn't usually help much)", 0,
311 16, flacenc_params[DEFAULT_QUALITY].min_residual_partition_order,
312 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
313 g_object_class_install_property (G_OBJECT_CLASS (klass),
314 PROP_MAX_RESIDUAL_PARTITION_ORDER,
315 g_param_spec_uint ("max-residual-partition-order",
316 "Max residual partition order",
317 "Max residual partition order (above 4 doesn't usually help much)", 0,
318 16, flacenc_params[DEFAULT_QUALITY].max_residual_partition_order,
319 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
320 g_object_class_install_property (G_OBJECT_CLASS (klass),
321 PROP_RICE_PARAMETER_SEARCH_DIST,
322 g_param_spec_uint ("rice-parameter-search-dist",
323 "rice_parameter_search_dist",
324 "0 = try only calc'd parameter k; else try all [k-dist..k+dist] "
325 "parameters, use best", 0, FLAC__MAX_RICE_PARTITION_ORDER,
326 flacenc_params[DEFAULT_QUALITY].rice_parameter_search_dist,
327 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
328 g_object_class_install_property (G_OBJECT_CLASS (klass),
330 g_param_spec_uint ("padding",
332 "Write a PADDING block with this length in bytes", 0, G_MAXUINT,
334 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
335 g_object_class_install_property (G_OBJECT_CLASS (klass),
337 g_param_spec_int ("seekpoints",
339 "Add SEEKTABLE metadata (if > 0, number of entries, if < 0, interval in sec)",
342 G_PARAM_READWRITE | G_PARAM_CONSTRUCT | G_PARAM_STATIC_STRINGS));
344 gst_element_class_add_static_pad_template (gstelement_class, &src_factory);
346 sink_caps = gst_flac_enc_generate_sink_caps ();
347 sink_templ = gst_pad_template_new ("sink",
348 GST_PAD_SINK, GST_PAD_ALWAYS, sink_caps);
349 gst_element_class_add_pad_template (gstelement_class, sink_templ);
350 gst_caps_unref (sink_caps);
352 gst_element_class_set_static_metadata (gstelement_class, "FLAC audio encoder",
353 "Codec/Encoder/Audio",
354 "Encodes audio with the FLAC lossless audio encoder",
355 "Wim Taymans <wim.taymans@chello.be>");
357 base_class->start = GST_DEBUG_FUNCPTR (gst_flac_enc_start);
358 base_class->stop = GST_DEBUG_FUNCPTR (gst_flac_enc_stop);
359 base_class->set_format = GST_DEBUG_FUNCPTR (gst_flac_enc_set_format);
360 base_class->handle_frame = GST_DEBUG_FUNCPTR (gst_flac_enc_handle_frame);
361 base_class->getcaps = GST_DEBUG_FUNCPTR (gst_flac_enc_getcaps);
362 base_class->sink_event = GST_DEBUG_FUNCPTR (gst_flac_enc_sink_event);
363 base_class->sink_query = GST_DEBUG_FUNCPTR (gst_flac_enc_sink_query);
367 gst_flac_enc_init (GstFlacEnc * flacenc)
369 GstAudioEncoder *enc = GST_AUDIO_ENCODER (flacenc);
371 flacenc->encoder = FLAC__stream_encoder_new ();
372 gst_flac_enc_update_quality (flacenc, DEFAULT_QUALITY);
374 /* arrange granulepos marking (and required perfect ts) */
375 gst_audio_encoder_set_mark_granule (enc, TRUE);
376 gst_audio_encoder_set_perfect_timestamp (enc, TRUE);
380 gst_flac_enc_finalize (GObject * object)
382 GstFlacEnc *flacenc = GST_FLAC_ENC (object);
384 FLAC__stream_encoder_delete (flacenc->encoder);
386 G_OBJECT_CLASS (parent_class)->finalize (object);
390 gst_flac_enc_start (GstAudioEncoder * enc)
392 GstFlacEnc *flacenc = GST_FLAC_ENC (enc);
394 GST_DEBUG_OBJECT (enc, "start");
395 flacenc->stopped = TRUE;
396 flacenc->got_headers = FALSE;
397 flacenc->last_flow = GST_FLOW_OK;
399 flacenc->eos = FALSE;
400 flacenc->tags = gst_tag_list_new_empty ();
402 flacenc->samples_in = 0;
403 flacenc->samples_out = 0;
409 gst_flac_enc_stop (GstAudioEncoder * enc)
411 GstFlacEnc *flacenc = GST_FLAC_ENC (enc);
413 GST_DEBUG_OBJECT (enc, "stop");
414 gst_tag_list_unref (flacenc->tags);
415 flacenc->tags = NULL;
417 gst_toc_unref (flacenc->toc);
419 if (FLAC__stream_encoder_get_state (flacenc->encoder) !=
420 FLAC__STREAM_ENCODER_UNINITIALIZED) {
421 flacenc->stopped = TRUE;
422 FLAC__stream_encoder_finish (flacenc->encoder);
425 FLAC__metadata_object_delete (flacenc->meta[0]);
427 if (flacenc->meta[1])
428 FLAC__metadata_object_delete (flacenc->meta[1]);
430 if (flacenc->meta[2])
431 FLAC__metadata_object_delete (flacenc->meta[2]);
433 if (flacenc->meta[3])
434 FLAC__metadata_object_delete (flacenc->meta[3]);
436 g_free (flacenc->meta);
437 flacenc->meta = NULL;
439 g_list_foreach (flacenc->headers, (GFunc) gst_mini_object_unref, NULL);
440 g_list_free (flacenc->headers);
441 flacenc->headers = NULL;
443 gst_tag_setter_reset_tags (GST_TAG_SETTER (enc));
444 gst_toc_setter_reset (GST_TOC_SETTER (enc));
450 add_one_tag (const GstTagList * list, const gchar * tag, gpointer user_data)
454 GstFlacEnc *flacenc = GST_FLAC_ENC (user_data);
456 /* IMAGE and PREVIEW_IMAGE tags are already written
457 * differently, no need to store them inside the
458 * vorbiscomments too */
459 if (strcmp (tag, GST_TAG_IMAGE) == 0
460 || strcmp (tag, GST_TAG_PREVIEW_IMAGE) == 0)
463 comments = gst_tag_to_vorbis_comments (list, tag);
464 for (it = comments; it != NULL; it = it->next) {
465 FLAC__StreamMetadata_VorbisComment_Entry commment_entry;
467 commment_entry.length = strlen (it->data);
468 commment_entry.entry = it->data;
469 FLAC__metadata_object_vorbiscomment_insert_comment (flacenc->meta[0],
470 flacenc->meta[0]->data.vorbis_comment.num_comments,
471 commment_entry, TRUE);
474 g_list_free (comments);
478 add_cuesheet (const GstToc * toc, guint sample_rate,
479 FLAC__StreamMetadata * cuesheet)
484 const gchar *is_legal;
487 GstTocEntry *entry, *subentry = NULL;
488 FLAC__StreamMetadata_CueSheet *cs;
489 FLAC__StreamMetadata_CueSheet_Track *track;
491 cs = &cuesheet->data.cue_sheet;
495 /* check if the TOC entries is valid */
496 list = gst_toc_get_entries (toc);
498 if (gst_toc_entry_is_alternative (entry)) {
499 list = gst_toc_entry_get_sub_entries (entry);
501 subentry = list->data;
502 if (!gst_toc_entry_is_sequence (subentry))
504 list = g_list_next (list);
506 list = gst_toc_entry_get_sub_entries (entry);
508 if (gst_toc_entry_is_sequence (entry)) {
511 if (!gst_toc_entry_is_sequence (entry))
513 list = g_list_next (list);
515 list = gst_toc_get_entries (toc);
518 /* add tracks in cuesheet */
521 gst_toc_entry_get_start_stop_times (entry, &start, &stop);
522 tags = gst_toc_entry_get_tags (entry);
524 gst_tag_list_get_string (tags, GST_TAG_ISRC, &isrc);
525 track = FLAC__metadata_object_cuesheet_track_new ();
527 (FLAC__uint64) gst_util_uint64_scale_round (start, sample_rate,
529 track->number = (FLAC__byte) track_num + 1;
530 if (isrc != NULL && strlen (isrc) <= 12)
531 g_strlcpy (track->isrc, isrc, 13);
532 if (track->number <= 0)
534 if (!FLAC__metadata_object_cuesheet_insert_track (cuesheet, track_num,
537 if (!FLAC__metadata_object_cuesheet_track_insert_blank_index (cuesheet,
541 list = g_list_next (list);
544 if (cs->num_tracks <= 0)
547 /* add lead-out track in cuesheet */
548 track = FLAC__metadata_object_cuesheet_track_new ();
550 (FLAC__uint64) gst_util_uint64_scale_round (stop, sample_rate,
553 if (!FLAC__metadata_object_cuesheet_insert_track (cuesheet, cs->num_tracks,
557 /* check if the cuesheet is valid */
558 if (!FLAC__metadata_object_cuesheet_is_legal (cuesheet, FALSE, &is_legal)) {
559 g_warning ("%s\n", is_legal);
567 gst_flac_enc_set_metadata (GstFlacEnc * flacenc, GstAudioInfo * info,
568 guint64 total_samples)
570 const GstTagList *user_tags;
573 gint n_images, n_preview_images;
574 FLAC__StreamMetadata *cuesheet;
576 g_return_if_fail (flacenc != NULL);
578 user_tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (flacenc));
579 if ((flacenc->tags == NULL) && (user_tags == NULL)) {
582 copy = gst_tag_list_merge (user_tags, flacenc->tags,
583 gst_tag_setter_get_tag_merge_mode (GST_TAG_SETTER (flacenc)));
584 n_images = gst_tag_list_get_tag_size (copy, GST_TAG_IMAGE);
585 n_preview_images = gst_tag_list_get_tag_size (copy, GST_TAG_PREVIEW_IMAGE);
588 g_new0 (FLAC__StreamMetadata *, 4 + n_images + n_preview_images);
591 FLAC__metadata_object_new (FLAC__METADATA_TYPE_VORBIS_COMMENT);
592 gst_tag_list_foreach (copy, add_one_tag, flacenc);
595 flacenc->toc = gst_toc_setter_get_toc (GST_TOC_SETTER (flacenc));
598 cuesheet = FLAC__metadata_object_new (FLAC__METADATA_TYPE_CUESHEET);
599 if (add_cuesheet (flacenc->toc, GST_AUDIO_INFO_RATE (info), cuesheet)) {
600 flacenc->meta[entries] = cuesheet;
603 FLAC__metadata_object_delete (cuesheet);
604 flacenc->meta[entries] = NULL;
608 if (n_images + n_preview_images > 0) {
612 const GstStructure *structure;
613 GstTagImageType image_type = GST_TAG_IMAGE_TYPE_NONE;
614 gint i, width = 0, height = 0, png_icon_count = 0, other_icon_count = 0;
617 for (i = 0; i < n_images + n_preview_images; i++) {
618 gboolean is_preview_image = (i >= n_images);
621 if (!gst_tag_list_get_sample_index (copy, GST_TAG_IMAGE, i, &sample))
624 if (!gst_tag_list_get_sample_index (copy, GST_TAG_PREVIEW_IMAGE,
625 i - n_images, &sample))
629 structure = gst_sample_get_info (sample);
630 caps = gst_sample_get_caps (sample);
632 GST_FIXME_OBJECT (flacenc, "Image tag without caps");
633 gst_sample_unref (sample);
637 flacenc->meta[entries] =
638 FLAC__metadata_object_new (FLAC__METADATA_TYPE_PICTURE);
640 GST_LOG_OBJECT (flacenc, "image info: %" GST_PTR_FORMAT, structure);
643 gst_structure_get (structure, "image-type", GST_TYPE_TAG_IMAGE_TYPE,
646 image_type = GST_TAG_IMAGE_TYPE_NONE;
648 GST_LOG_OBJECT (flacenc, "image caps: %" GST_PTR_FORMAT, caps);
650 structure = gst_caps_get_structure (caps, 0);
651 gst_structure_get (structure, "width", G_TYPE_INT, &width,
652 "height", G_TYPE_INT, &height, NULL);
654 /* Convert to ID3v2 APIC image type */
655 if (image_type == GST_TAG_IMAGE_TYPE_NONE) {
656 if (is_preview_image) {
657 /* 1 - 32x32 pixels 'file icon' (PNG only)
658 * 2 - Other file icon
659 * There may only be one each of picture type 1 and 2 in a file. */
660 if (width == 32 && height == 32
661 && gst_structure_has_name (structure, "image/png")
662 && png_icon_count++ == 0) {
664 } else if (width <= 32 && height <= 32 && other_icon_count++ == 0) {
667 image_type = 0; /* Other */
670 image_type = 0; /* Other */
673 /* GStreamer enum is the same but without the two icon types 1+2 */
674 image_type = image_type + 2;
677 buffer = gst_sample_get_buffer (sample);
678 gst_buffer_map (buffer, &map, GST_MAP_READ);
679 FLAC__metadata_object_picture_set_data (flacenc->meta[entries],
680 map.data, map.size, TRUE);
681 gst_buffer_unmap (buffer, &map);
683 GST_LOG_OBJECT (flacenc, "Setting picture type %d", image_type);
684 flacenc->meta[entries]->data.picture.type = image_type;
686 if (width > 0 && height > 0) {
687 flacenc->meta[entries]->data.picture.width = width;
688 flacenc->meta[entries]->data.picture.height = height;
691 FLAC__metadata_object_picture_set_mime_type (flacenc->meta[entries],
692 (char *) gst_structure_get_name (structure), TRUE);
694 gst_sample_unref (sample);
699 if (flacenc->seekpoints && total_samples != GST_CLOCK_TIME_NONE) {
703 flacenc->meta[entries] =
704 FLAC__metadata_object_new (FLAC__METADATA_TYPE_SEEKTABLE);
705 if (flacenc->seekpoints > 0) {
707 FLAC__metadata_object_seektable_template_append_spaced_points
708 (flacenc->meta[entries], flacenc->seekpoints, total_samples);
710 samples = -flacenc->seekpoints * GST_AUDIO_INFO_RATE (info);
712 FLAC__metadata_object_seektable_template_append_spaced_points_by_samples
713 (flacenc->meta[entries], samples, total_samples);
716 GST_DEBUG_OBJECT (flacenc, "adding seekpoint template %d failed",
717 flacenc->seekpoints);
718 FLAC__metadata_object_delete (flacenc->meta[1]);
719 flacenc->meta[entries] = NULL;
723 } else if (flacenc->seekpoints && total_samples == GST_CLOCK_TIME_NONE) {
724 GST_WARNING_OBJECT (flacenc, "total time unknown; can not add seekpoints");
727 if (flacenc->padding > 0) {
728 flacenc->meta[entries] =
729 FLAC__metadata_object_new (FLAC__METADATA_TYPE_PADDING);
730 flacenc->meta[entries]->length = flacenc->padding;
734 if (FLAC__stream_encoder_set_metadata (flacenc->encoder,
735 flacenc->meta, entries) != true)
736 g_warning ("Dude, i'm already initialized!");
738 gst_tag_list_unref (copy);
742 gst_flac_enc_generate_sink_caps (void)
746 GValue v_list = { 0, };
748 GstStructure *s, *s2;
750 g_value_init (&v_list, GST_TYPE_LIST);
751 g_value_init (&v, G_TYPE_STRING);
753 /* Use system's endianness */
754 g_value_set_static_string (&v, "S8");
755 gst_value_list_append_value (&v_list, &v);
756 g_value_set_static_string (&v, GST_AUDIO_NE (S16));
757 gst_value_list_append_value (&v_list, &v);
758 g_value_set_static_string (&v, GST_AUDIO_NE (S24));
759 gst_value_list_append_value (&v_list, &v);
760 g_value_set_static_string (&v, GST_AUDIO_NE (S24_32));
761 gst_value_list_append_value (&v_list, &v);
764 s = gst_structure_new_empty ("audio/x-raw");
765 gst_structure_take_value (s, "format", &v_list);
767 gst_structure_set (s, "layout", G_TYPE_STRING, "interleaved",
768 "rate", GST_TYPE_INT_RANGE, 1, 655350, NULL);
770 ret = gst_caps_new_empty ();
771 s2 = gst_structure_copy (s);
772 gst_structure_set (s2, "channels", G_TYPE_INT, 1, NULL);
773 gst_caps_append_structure (ret, s2);
774 for (i = 2; i <= 8; i++) {
775 guint64 channel_mask;
777 s2 = gst_structure_copy (s);
778 gst_audio_channel_positions_to_mask (channel_positions[i - 1], i,
779 FALSE, &channel_mask);
780 gst_structure_set (s2, "channels", G_TYPE_INT, i, "channel-mask",
781 GST_TYPE_BITMASK, channel_mask, NULL);
783 gst_caps_append_structure (ret, s2);
785 gst_structure_free (s);
791 gst_flac_enc_getcaps (GstAudioEncoder * enc, GstCaps * filter)
793 GstCaps *ret = NULL, *caps = NULL;
796 pad = GST_AUDIO_ENCODER_SINK_PAD (enc);
798 ret = gst_pad_get_current_caps (pad);
800 ret = gst_pad_get_pad_template_caps (pad);
803 GST_DEBUG_OBJECT (pad, "Return caps %" GST_PTR_FORMAT, ret);
805 caps = gst_audio_encoder_proxy_getcaps (enc, ret, filter);
806 gst_caps_unref (ret);
812 gst_flac_enc_peer_query_total_samples (GstFlacEnc * flacenc, GstPad * pad)
816 gst_audio_encoder_get_audio_info (GST_AUDIO_ENCODER (flacenc));
818 GST_DEBUG_OBJECT (flacenc, "querying peer for DEFAULT format duration");
819 if (gst_pad_peer_query_duration (pad, GST_FORMAT_DEFAULT, &duration)
820 && duration != GST_CLOCK_TIME_NONE)
823 GST_DEBUG_OBJECT (flacenc, "querying peer for TIME format duration");
825 if (gst_pad_peer_query_duration (pad, GST_FORMAT_TIME, &duration)
826 && duration != GST_CLOCK_TIME_NONE) {
827 GST_DEBUG_OBJECT (flacenc, "peer reported duration %" GST_TIME_FORMAT,
828 GST_TIME_ARGS (duration));
829 duration = GST_CLOCK_TIME_TO_FRAMES (duration, GST_AUDIO_INFO_RATE (info));
834 GST_DEBUG_OBJECT (flacenc, "Upstream reported no total samples");
835 return GST_CLOCK_TIME_NONE;
838 GST_DEBUG_OBJECT (flacenc,
839 "Upstream reported %" G_GUINT64_FORMAT " total samples", duration);
845 gst_flac_enc_get_latency (GstFlacEnc * flacenc)
847 /* The FLAC specification states that the data is processed in blocks,
848 * regardless of the number of channels. Thus, The latency can be calculated
849 * using the blocksize and rate. For example a 1 second block sampled at
850 * 44.1KHz has a blocksize of 44100 */
852 /* Get the blocksize */
853 const guint blocksize = FLAC__stream_encoder_get_blocksize (flacenc->encoder);
855 /* Get the sample rate in KHz */
856 const guint rate = FLAC__stream_encoder_get_sample_rate (flacenc->encoder);
860 /* Calculate the latecy */
861 return (blocksize * GST_SECOND) / rate;
865 gst_flac_enc_set_format (GstAudioEncoder * enc, GstAudioInfo * info)
868 guint64 total_samples = GST_CLOCK_TIME_NONE;
869 FLAC__StreamEncoderInitStatus init_status;
871 flacenc = GST_FLAC_ENC (enc);
873 /* if configured again, means something changed, can't handle that */
874 if (FLAC__stream_encoder_get_state (flacenc->encoder) !=
875 FLAC__STREAM_ENCODER_UNINITIALIZED)
876 goto encoder_already_initialized;
878 /* delay setting output caps/format until we have all headers */
880 gst_audio_get_channel_reorder_map (GST_AUDIO_INFO_CHANNELS (info),
881 channel_positions[GST_AUDIO_INFO_CHANNELS (info) - 1], info->position,
882 flacenc->channel_reorder_map);
884 total_samples = gst_flac_enc_peer_query_total_samples (flacenc,
885 GST_AUDIO_ENCODER_SINK_PAD (enc));
887 FLAC__stream_encoder_set_bits_per_sample (flacenc->encoder,
888 GST_AUDIO_INFO_DEPTH (info));
889 FLAC__stream_encoder_set_sample_rate (flacenc->encoder,
890 GST_AUDIO_INFO_RATE (info));
891 FLAC__stream_encoder_set_channels (flacenc->encoder,
892 GST_AUDIO_INFO_CHANNELS (info));
894 if (total_samples != GST_CLOCK_TIME_NONE)
895 FLAC__stream_encoder_set_total_samples_estimate (flacenc->encoder,
896 MIN (total_samples, G_GUINT64_CONSTANT (0x0FFFFFFFFF)));
898 gst_flac_enc_set_metadata (flacenc, info, total_samples);
900 /* callbacks clear to go now;
901 * write callbacks receives headers during init */
902 flacenc->stopped = FALSE;
904 init_status = FLAC__stream_encoder_init_stream (flacenc->encoder,
905 gst_flac_enc_write_callback, gst_flac_enc_seek_callback,
906 gst_flac_enc_tell_callback, NULL, flacenc);
907 if (init_status != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
908 goto failed_to_initialize;
910 /* feedback to base class */
911 gst_audio_encoder_set_latency (enc,
912 gst_flac_enc_get_latency (flacenc), gst_flac_enc_get_latency (flacenc));
916 encoder_already_initialized:
918 g_warning ("flac already initialized -- fixme allow this");
921 failed_to_initialize:
923 GST_ELEMENT_ERROR (flacenc, LIBRARY, INIT, (NULL),
924 ("could not initialize encoder (wrong parameters?) %d", init_status));
930 gst_flac_enc_update_quality (GstFlacEnc * flacenc, gint quality)
933 gst_audio_encoder_get_audio_info (GST_AUDIO_ENCODER (flacenc));
935 flacenc->quality = quality;
937 #define DO_UPDATE(name, val, str) \
939 if (FLAC__stream_encoder_get_##name (flacenc->encoder) != \
940 flacenc_params[quality].val) { \
941 FLAC__stream_encoder_set_##name (flacenc->encoder, \
942 flacenc_params[quality].val); \
943 g_object_notify (G_OBJECT (flacenc), str); \
947 g_object_freeze_notify (G_OBJECT (flacenc));
949 if (GST_AUDIO_INFO_CHANNELS (info) == 2
950 || GST_AUDIO_INFO_CHANNELS (info) == 0) {
951 DO_UPDATE (do_mid_side_stereo, mid_side, "mid_side_stereo");
952 DO_UPDATE (loose_mid_side_stereo, loose_mid_side, "loose_mid_side");
955 DO_UPDATE (blocksize, blocksize, "blocksize");
956 DO_UPDATE (max_lpc_order, max_lpc_order, "max_lpc_order");
957 DO_UPDATE (qlp_coeff_precision, qlp_coeff_precision, "qlp_coeff_precision");
958 DO_UPDATE (do_qlp_coeff_prec_search, qlp_coeff_prec_search,
959 "qlp_coeff_prec_search");
960 DO_UPDATE (do_escape_coding, escape_coding, "escape_coding");
961 DO_UPDATE (do_exhaustive_model_search, exhaustive_model_search,
962 "exhaustive_model_search");
963 DO_UPDATE (min_residual_partition_order, min_residual_partition_order,
964 "min_residual_partition_order");
965 DO_UPDATE (max_residual_partition_order, max_residual_partition_order,
966 "max_residual_partition_order");
967 DO_UPDATE (rice_parameter_search_dist, rice_parameter_search_dist,
968 "rice_parameter_search_dist");
972 g_object_thaw_notify (G_OBJECT (flacenc));
977 static FLAC__StreamEncoderSeekStatus
978 gst_flac_enc_seek_callback (const FLAC__StreamEncoder * encoder,
979 FLAC__uint64 absolute_byte_offset, void *client_data)
985 flacenc = GST_FLAC_ENC (client_data);
987 if (flacenc->stopped)
988 return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
990 if ((peerpad = gst_pad_get_peer (GST_AUDIO_ENCODER_SRC_PAD (flacenc)))) {
994 gboolean seekable = FALSE;
996 /* try to seek to the beginning of the output */
997 query = gst_query_new_seeking (GST_FORMAT_BYTES);
998 if (gst_pad_query (peerpad, query)) {
1001 gst_query_parse_seeking (query, &format, &seekable, NULL, NULL);
1002 if (format != GST_FORMAT_BYTES)
1005 GST_LOG_OBJECT (flacenc, "SEEKING query not handled");
1007 gst_query_unref (query);
1010 GST_DEBUG_OBJECT (flacenc, "downstream not seekable; not rewriting");
1011 gst_object_unref (peerpad);
1012 return FLAC__STREAM_ENCODER_SEEK_STATUS_UNSUPPORTED;
1015 gst_segment_init (&seg, GST_FORMAT_BYTES);
1016 seg.start = absolute_byte_offset;
1017 seg.stop = GST_BUFFER_OFFSET_NONE;
1019 event = gst_event_new_segment (&seg);
1021 ret = gst_pad_send_event (peerpad, event);
1022 gst_object_unref (peerpad);
1025 GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " %s",
1026 (guint64) absolute_byte_offset, "succeeded");
1028 GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " %s",
1029 (guint64) absolute_byte_offset, "failed");
1030 return FLAC__STREAM_ENCODER_SEEK_STATUS_UNSUPPORTED;
1033 GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " failed (no peer pad)",
1034 (guint64) absolute_byte_offset);
1037 flacenc->offset = absolute_byte_offset;
1038 return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
1042 notgst_value_array_append_buffer (GValue * array_val, GstBuffer * buf)
1044 GValue value = { 0, };
1046 g_value_init (&value, GST_TYPE_BUFFER);
1047 /* copy buffer to avoid problems with circular refcounts */
1048 buf = gst_buffer_copy (buf);
1049 /* again, for good measure */
1050 GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_HEADER);
1051 gst_value_set_buffer (&value, buf);
1052 gst_buffer_unref (buf);
1053 gst_value_array_append_value (array_val, &value);
1054 g_value_unset (&value);
1057 #define HDR_TYPE_STREAMINFO 0
1058 #define HDR_TYPE_VORBISCOMMENT 4
1060 static GstFlowReturn
1061 gst_flac_enc_process_stream_headers (GstFlacEnc * enc)
1063 GstBuffer *vorbiscomment = NULL;
1064 GstBuffer *streaminfo = NULL;
1065 GstBuffer *marker = NULL;
1066 GValue array = { 0, };
1069 GstFlowReturn ret = GST_FLOW_OK;
1070 GstAudioInfo *info =
1071 gst_audio_encoder_get_audio_info (GST_AUDIO_ENCODER (enc));
1073 caps = gst_caps_new_simple ("audio/x-flac",
1074 "channels", G_TYPE_INT, GST_AUDIO_INFO_CHANNELS (info),
1075 "rate", G_TYPE_INT, GST_AUDIO_INFO_RATE (info), NULL);
1077 for (l = enc->headers; l != NULL; l = l->next) {
1083 /* mark buffers so oggmux will ignore them if it already muxed the
1084 * header buffers from the streamheaders field in the caps */
1085 l->data = gst_buffer_make_writable (GST_BUFFER_CAST (l->data));
1087 buf = GST_BUFFER_CAST (l->data);
1088 GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_HEADER);
1090 gst_buffer_map (buf, &map, GST_MAP_READ);
1094 /* find initial 4-byte marker which we need to skip later on */
1095 if (size == 4 && memcmp (data, "fLaC", 4) == 0) {
1097 } else if (size > 1 && (data[0] & 0x7f) == HDR_TYPE_STREAMINFO) {
1099 } else if (size > 1 && (data[0] & 0x7f) == HDR_TYPE_VORBISCOMMENT) {
1100 vorbiscomment = buf;
1103 gst_buffer_unmap (buf, &map);
1106 if (marker == NULL || streaminfo == NULL || vorbiscomment == NULL) {
1107 GST_WARNING_OBJECT (enc, "missing header %p %p %p, muxing into container "
1108 "formats may be broken", marker, streaminfo, vorbiscomment);
1112 g_value_init (&array, GST_TYPE_ARRAY);
1114 /* add marker including STREAMINFO header */
1122 /* minus one for the marker that is merged with streaminfo here */
1123 num = g_list_length (enc->headers) - 1;
1125 slen = gst_buffer_get_size (streaminfo);
1126 buf = gst_buffer_new_and_alloc (13 + slen);
1128 gst_buffer_map (buf, &map, GST_MAP_WRITE);
1131 memcpy (bdata + 1, "FLAC", 4);
1132 bdata[5] = 0x01; /* mapping version major */
1133 bdata[6] = 0x00; /* mapping version minor */
1134 bdata[7] = (num & 0xFF00) >> 8;
1135 bdata[8] = (num & 0x00FF) >> 0;
1136 memcpy (bdata + 9, "fLaC", 4);
1137 gst_buffer_extract (streaminfo, 0, bdata + 13, slen);
1138 gst_buffer_unmap (buf, &map);
1140 notgst_value_array_append_buffer (&array, buf);
1141 gst_buffer_unref (buf);
1144 /* add VORBISCOMMENT header */
1145 notgst_value_array_append_buffer (&array, vorbiscomment);
1147 /* add other headers, if there are any */
1148 for (l = enc->headers; l != NULL; l = l->next) {
1149 GstBuffer *buf = GST_BUFFER_CAST (l->data);
1151 if (buf != marker && buf != streaminfo && buf != vorbiscomment) {
1152 notgst_value_array_append_buffer (&array, buf);
1156 gst_structure_set_value (gst_caps_get_structure (caps, 0),
1157 "streamheader", &array);
1158 g_value_unset (&array);
1161 gst_audio_encoder_set_output_format (GST_AUDIO_ENCODER (enc), caps);
1163 gst_audio_encoder_set_headers (GST_AUDIO_ENCODER (enc), enc->headers);
1164 enc->headers = NULL;
1166 gst_caps_unref (caps);
1171 static FLAC__StreamEncoderWriteStatus
1172 gst_flac_enc_write_callback (const FLAC__StreamEncoder * encoder,
1173 const FLAC__byte buffer[], size_t bytes,
1174 unsigned samples, unsigned current_frame, void *client_data)
1176 GstFlowReturn ret = GST_FLOW_OK;
1177 GstFlacEnc *flacenc;
1179 GstSegment *segment;
1180 GstClockTime duration;
1182 flacenc = GST_FLAC_ENC (client_data);
1184 if (flacenc->stopped)
1185 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
1187 outbuf = gst_buffer_new_and_alloc (bytes);
1188 gst_buffer_fill (outbuf, 0, buffer, bytes);
1190 /* we assume libflac passes us stuff neatly framed */
1191 if (!flacenc->got_headers) {
1193 GST_DEBUG_OBJECT (flacenc, "Got header, queueing (%u bytes)",
1195 flacenc->headers = g_list_append (flacenc->headers, outbuf);
1196 /* note: it's important that we increase our byte offset */
1199 GST_INFO_OBJECT (flacenc, "Non-header packet, we have all headers now");
1200 ret = gst_flac_enc_process_stream_headers (flacenc);
1201 flacenc->got_headers = TRUE;
1205 if (flacenc->got_headers && samples == 0) {
1206 /* header fixup, push downstream directly */
1207 GST_DEBUG_OBJECT (flacenc, "Fixing up headers at pos=%" G_GUINT64_FORMAT
1208 ", size=%u", flacenc->offset, (guint) bytes);
1210 GST_MEMDUMP_OBJECT (flacenc, "Presumed header fragment",
1211 GST_BUFFER_DATA (outbuf), GST_BUFFER_SIZE (outbuf));
1213 ret = gst_pad_push (GST_AUDIO_ENCODER_SRC_PAD (flacenc), outbuf);
1215 /* regular frame data, pass to base class */
1216 if (flacenc->eos && flacenc->samples_in == flacenc->samples_out + samples) {
1217 /* If encoding part of a frame, and we have no set stop time on
1218 * the output segment, we update the segment stop time to reflect
1219 * the last sample. This will let oggmux set the last page's
1220 * granpos to tell a decoder the dummy samples should be clipped.
1222 segment = &GST_AUDIO_ENCODER_OUTPUT_SEGMENT (flacenc);
1223 if (!GST_CLOCK_TIME_IS_VALID (segment->stop)) {
1224 GST_DEBUG_OBJECT (flacenc,
1225 "No stop time and partial frame, updating segment");
1227 gst_util_uint64_scale (flacenc->samples_out + samples,
1229 FLAC__stream_encoder_get_sample_rate (flacenc->encoder));
1230 segment->stop = segment->start + duration;
1231 GST_DEBUG_OBJECT (flacenc, "new output segment %" GST_SEGMENT_FORMAT,
1233 gst_pad_push_event (GST_AUDIO_ENCODER_SRC_PAD (flacenc),
1234 gst_event_new_segment (segment));
1238 GST_LOG ("Pushing buffer: samples=%u, size=%u, pos=%" G_GUINT64_FORMAT,
1239 samples, (guint) bytes, flacenc->offset);
1240 ret = gst_audio_encoder_finish_frame (GST_AUDIO_ENCODER (flacenc),
1244 if (ret != GST_FLOW_OK)
1245 GST_DEBUG_OBJECT (flacenc, "flow: %s", gst_flow_get_name (ret));
1247 flacenc->last_flow = ret;
1250 flacenc->offset += bytes;
1252 if (ret != GST_FLOW_OK)
1253 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
1255 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
1258 static FLAC__StreamEncoderTellStatus
1259 gst_flac_enc_tell_callback (const FLAC__StreamEncoder * encoder,
1260 FLAC__uint64 * absolute_byte_offset, void *client_data)
1262 GstFlacEnc *flacenc = GST_FLAC_ENC (client_data);
1264 *absolute_byte_offset = flacenc->offset;
1266 return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
1270 gst_flac_enc_sink_event (GstAudioEncoder * enc, GstEvent * event)
1272 GstFlacEnc *flacenc;
1273 GstTagList *taglist;
1275 gboolean ret = FALSE;
1277 flacenc = GST_FLAC_ENC (enc);
1279 GST_DEBUG ("Received %s event on sinkpad, %" GST_PTR_FORMAT,
1280 GST_EVENT_TYPE_NAME (event), event);
1282 switch (GST_EVENT_TYPE (event)) {
1284 flacenc->eos = TRUE;
1285 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_event (enc, event);
1288 if (flacenc->tags) {
1289 gst_event_parse_tag (event, &taglist);
1290 gst_tag_list_insert (flacenc->tags, taglist,
1291 gst_tag_setter_get_tag_merge_mode (GST_TAG_SETTER (flacenc)));
1293 g_assert_not_reached ();
1295 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_event (enc, event);
1298 gst_event_parse_toc (event, &toc, NULL);
1300 if (flacenc->toc != toc) {
1302 gst_toc_unref (flacenc->toc);
1306 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_event (enc, event);
1308 case GST_EVENT_SEGMENT:
1309 flacenc->samples_in = 0;
1310 flacenc->samples_out = 0;
1311 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_event (enc, event);
1314 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_event (enc, event);
1322 gst_flac_enc_sink_query (GstAudioEncoder * enc, GstQuery * query)
1324 GstPad *pad = GST_AUDIO_ENCODER_SINK_PAD (enc);
1325 gboolean ret = FALSE;
1327 GST_DEBUG ("Received %s query on sinkpad, %" GST_PTR_FORMAT,
1328 GST_QUERY_TYPE_NAME (query), query);
1330 switch (GST_QUERY_TYPE (query)) {
1331 case GST_QUERY_ACCEPT_CAPS:{
1332 GstCaps *acceptable, *caps;
1334 acceptable = gst_pad_get_current_caps (pad);
1335 if (acceptable == NULL) {
1336 acceptable = gst_pad_get_pad_template_caps (pad);
1339 gst_query_parse_accept_caps (query, &caps);
1341 gst_query_set_accept_caps_result (query,
1342 gst_caps_is_subset (caps, acceptable));
1343 gst_caps_unref (acceptable);
1348 ret = GST_AUDIO_ENCODER_CLASS (parent_class)->sink_query (enc, query);
1355 #if G_BYTE_ORDER == G_LITTLE_ENDIAN
1356 #define READ_INT24 GST_READ_UINT24_LE
1358 #define READ_INT24 GST_READ_UINT24_BE
1361 static GstFlowReturn
1362 gst_flac_enc_handle_frame (GstAudioEncoder * enc, GstBuffer * buffer)
1364 GstFlacEnc *flacenc;
1366 gint samples, width, channels;
1371 GstAudioInfo *info =
1372 gst_audio_encoder_get_audio_info (GST_AUDIO_ENCODER (enc));
1375 flacenc = GST_FLAC_ENC (enc);
1377 /* base class ensures configuration */
1378 g_return_val_if_fail (GST_AUDIO_INFO_WIDTH (info) != 0,
1379 GST_FLOW_NOT_NEGOTIATED);
1381 width = GST_AUDIO_INFO_WIDTH (info);
1382 channels = GST_AUDIO_INFO_CHANNELS (info);
1383 reorder_map = flacenc->channel_reorder_map;
1385 if (G_UNLIKELY (!buffer)) {
1387 GST_DEBUG_OBJECT (flacenc, "finish encoding");
1388 FLAC__stream_encoder_finish (flacenc->encoder);
1390 /* can't handle intermittent draining/resyncing */
1391 GST_ELEMENT_WARNING (flacenc, STREAM, FORMAT, (NULL),
1392 ("Stream discontinuity detected. "
1393 "The output may have wrong timestamps, "
1394 "consider using audiorate to handle discontinuities"));
1396 return flacenc->last_flow;
1399 gst_buffer_map (buffer, &map, GST_MAP_READ);
1400 samples = map.size / (width >> 3);
1402 data = g_malloc (samples * sizeof (FLAC__int32));
1404 samples /= channels;
1405 GST_LOG_OBJECT (flacenc, "processing %d samples, %d channels", samples,
1408 gint8 *indata = (gint8 *) map.data;
1410 for (i = 0; i < samples; i++)
1411 for (j = 0; j < channels; j++)
1412 data[i * channels + reorder_map[j]] =
1413 (FLAC__int32) indata[i * channels + j];
1414 } else if (width == 16) {
1415 gint16 *indata = (gint16 *) map.data;
1417 for (i = 0; i < samples; i++)
1418 for (j = 0; j < channels; j++)
1419 data[i * channels + reorder_map[j]] =
1420 (FLAC__int32) indata[i * channels + j];
1421 } else if (width == 24) {
1422 guint8 *indata = (guint8 *) map.data;
1425 for (i = 0; i < samples; i++)
1426 for (j = 0; j < channels; j++) {
1427 val = READ_INT24 (&indata[3 * (i * channels + j)]);
1428 if (val & 0x00800000)
1430 data[i * channels + reorder_map[j]] = (FLAC__int32) val;
1432 } else if (width == 32) {
1433 gint32 *indata = (gint32 *) map.data;
1435 for (i = 0; i < samples; i++)
1436 for (j = 0; j < channels; j++)
1437 data[i * channels + reorder_map[j]] =
1438 (FLAC__int32) indata[i * channels + j];
1440 g_assert_not_reached ();
1442 gst_buffer_unmap (buffer, &map);
1444 res = FLAC__stream_encoder_process_interleaved (flacenc->encoder,
1445 (const FLAC__int32 *) data, samples);
1446 flacenc->samples_in += samples;
1451 if (flacenc->last_flow == GST_FLOW_OK)
1452 return GST_FLOW_ERROR;
1454 return flacenc->last_flow;
1461 gst_flac_enc_set_property (GObject * object, guint prop_id,
1462 const GValue * value, GParamSpec * pspec)
1464 GstFlacEnc *this = GST_FLAC_ENC (object);
1465 GstAudioEncoder *enc = GST_AUDIO_ENCODER (object);
1466 guint64 curr_latency = 0, old_latency = gst_flac_enc_get_latency (this);
1468 GST_OBJECT_LOCK (this);
1472 gst_flac_enc_update_quality (this, g_value_get_enum (value));
1474 case PROP_STREAMABLE_SUBSET:
1475 FLAC__stream_encoder_set_streamable_subset (this->encoder,
1476 g_value_get_boolean (value));
1478 case PROP_MID_SIDE_STEREO:
1479 FLAC__stream_encoder_set_do_mid_side_stereo (this->encoder,
1480 g_value_get_boolean (value));
1482 case PROP_LOOSE_MID_SIDE_STEREO:
1483 FLAC__stream_encoder_set_loose_mid_side_stereo (this->encoder,
1484 g_value_get_boolean (value));
1486 case PROP_BLOCKSIZE:
1487 FLAC__stream_encoder_set_blocksize (this->encoder,
1488 g_value_get_uint (value));
1490 case PROP_MAX_LPC_ORDER:
1491 FLAC__stream_encoder_set_max_lpc_order (this->encoder,
1492 g_value_get_uint (value));
1494 case PROP_QLP_COEFF_PRECISION:
1495 FLAC__stream_encoder_set_qlp_coeff_precision (this->encoder,
1496 g_value_get_uint (value));
1498 case PROP_QLP_COEFF_PREC_SEARCH:
1499 FLAC__stream_encoder_set_do_qlp_coeff_prec_search (this->encoder,
1500 g_value_get_boolean (value));
1502 case PROP_ESCAPE_CODING:
1503 FLAC__stream_encoder_set_do_escape_coding (this->encoder,
1504 g_value_get_boolean (value));
1506 case PROP_EXHAUSTIVE_MODEL_SEARCH:
1507 FLAC__stream_encoder_set_do_exhaustive_model_search (this->encoder,
1508 g_value_get_boolean (value));
1510 case PROP_MIN_RESIDUAL_PARTITION_ORDER:
1511 FLAC__stream_encoder_set_min_residual_partition_order (this->encoder,
1512 g_value_get_uint (value));
1514 case PROP_MAX_RESIDUAL_PARTITION_ORDER:
1515 FLAC__stream_encoder_set_max_residual_partition_order (this->encoder,
1516 g_value_get_uint (value));
1518 case PROP_RICE_PARAMETER_SEARCH_DIST:
1519 FLAC__stream_encoder_set_rice_parameter_search_dist (this->encoder,
1520 g_value_get_uint (value));
1523 this->padding = g_value_get_uint (value);
1525 case PROP_SEEKPOINTS:
1526 this->seekpoints = g_value_get_int (value);
1529 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1533 GST_OBJECT_UNLOCK (this);
1535 /* Update latency if it has changed */
1536 curr_latency = gst_flac_enc_get_latency (this);
1537 if (old_latency != curr_latency)
1538 gst_audio_encoder_set_latency (enc, curr_latency, curr_latency);
1542 gst_flac_enc_get_property (GObject * object, guint prop_id,
1543 GValue * value, GParamSpec * pspec)
1545 GstFlacEnc *this = GST_FLAC_ENC (object);
1547 GST_OBJECT_LOCK (this);
1551 g_value_set_enum (value, this->quality);
1553 case PROP_STREAMABLE_SUBSET:
1554 g_value_set_boolean (value,
1555 FLAC__stream_encoder_get_streamable_subset (this->encoder));
1557 case PROP_MID_SIDE_STEREO:
1558 g_value_set_boolean (value,
1559 FLAC__stream_encoder_get_do_mid_side_stereo (this->encoder));
1561 case PROP_LOOSE_MID_SIDE_STEREO:
1562 g_value_set_boolean (value,
1563 FLAC__stream_encoder_get_loose_mid_side_stereo (this->encoder));
1565 case PROP_BLOCKSIZE:
1566 g_value_set_uint (value,
1567 FLAC__stream_encoder_get_blocksize (this->encoder));
1569 case PROP_MAX_LPC_ORDER:
1570 g_value_set_uint (value,
1571 FLAC__stream_encoder_get_max_lpc_order (this->encoder));
1573 case PROP_QLP_COEFF_PRECISION:
1574 g_value_set_uint (value,
1575 FLAC__stream_encoder_get_qlp_coeff_precision (this->encoder));
1577 case PROP_QLP_COEFF_PREC_SEARCH:
1578 g_value_set_boolean (value,
1579 FLAC__stream_encoder_get_do_qlp_coeff_prec_search (this->encoder));
1581 case PROP_ESCAPE_CODING:
1582 g_value_set_boolean (value,
1583 FLAC__stream_encoder_get_do_escape_coding (this->encoder));
1585 case PROP_EXHAUSTIVE_MODEL_SEARCH:
1586 g_value_set_boolean (value,
1587 FLAC__stream_encoder_get_do_exhaustive_model_search (this->encoder));
1589 case PROP_MIN_RESIDUAL_PARTITION_ORDER:
1590 g_value_set_uint (value,
1591 FLAC__stream_encoder_get_min_residual_partition_order
1594 case PROP_MAX_RESIDUAL_PARTITION_ORDER:
1595 g_value_set_uint (value,
1596 FLAC__stream_encoder_get_max_residual_partition_order
1599 case PROP_RICE_PARAMETER_SEARCH_DIST:
1600 g_value_set_uint (value,
1601 FLAC__stream_encoder_get_rice_parameter_search_dist (this->encoder));
1604 g_value_set_uint (value, this->padding);
1606 case PROP_SEEKPOINTS:
1607 g_value_set_int (value, this->seekpoints);
1610 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1614 GST_OBJECT_UNLOCK (this);