ext/flac/gstflacenc.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   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
  21 #ifdef HAVE_CONFIG_H
  22 #include "config.h"
  23 #endif
  24 #include <stdlib.h>
  25 #include <string.h>
  26
  27 #include <gstflacenc.h>
  28 #include <gst/audio/audio.h>
  29 #include <gst/tag/tag.h>
  30 #include <gst/gsttagsetter.h>
  31 #include "flac_compat.h"
  32
  33
  34 GstElementDetails flacenc_details = {
  35   "FLAC encoder",
  36   "Codec/Encoder/Audio",
  37   "Encodes audio with the FLAC lossless audio encoder",
  38   "Wim Taymans <wim.taymans@chello.be>",
  39 };
  40
  41 #define FLAC_SINK_CAPS \
  42   "audio/x-raw-int, "               \
  43   "endianness = (int) BYTE_ORDER, " \
  44   "signed = (boolean) TRUE, "       \
  45   "width = (int) 16, "              \
  46   "depth = (int) 16, "              \
  47   "rate = (int) [ 11025, 48000 ], " \
  48   "channels = (int) [ 1, 2 ]"
  49
  50 static GstStaticPadTemplate src_factory = GST_STATIC_PAD_TEMPLATE ("src",
  51     GST_PAD_SRC,
  52     GST_PAD_ALWAYS,
  53     GST_STATIC_CAPS ("audio/x-flac")
  54     );
  55
  56 static GstStaticPadTemplate sink_factory = GST_STATIC_PAD_TEMPLATE ("sink",
  57     GST_PAD_SINK,
  58     GST_PAD_ALWAYS,
  59     GST_STATIC_CAPS (FLAC_SINK_CAPS)
  60     );
  61
  62 enum
  63 {
  64   PROP_0,
  65   PROP_QUALITY,
  66   PROP_STREAMABLE_SUBSET,
  67   PROP_MID_SIDE_STEREO,
  68   PROP_LOOSE_MID_SIDE_STEREO,
  69   PROP_BLOCKSIZE,
  70   PROP_MAX_LPC_ORDER,
  71   PROP_QLP_COEFF_PRECISION,
  72   PROP_QLP_COEFF_PREC_SEARCH,
  73   PROP_ESCAPE_CODING,
  74   PROP_EXHAUSTIVE_MODEL_SEARCH,
  75   PROP_MIN_RESIDUAL_PARTITION_ORDER,
  76   PROP_MAX_RESIDUAL_PARTITION_ORDER,
  77   PROP_RICE_PARAMETER_SEARCH_DIST
  78 };
  79
  80 GST_DEBUG_CATEGORY_STATIC (flacenc_debug);
  81 #define GST_CAT_DEFAULT flacenc_debug
  82
  83
  84 #define _do_init(type)                                                          \
  85   G_STMT_START{                                                                 \
  86     static const GInterfaceInfo tag_setter_info = {                             \
  87       NULL,                                                                     \
  88       NULL,                                                                     \
  89       NULL                                                                      \
  90     };                                                                          \
  91     g_type_add_interface_static (type, GST_TYPE_TAG_SETTER,                     \
  92                                  &tag_setter_info);                             \
  93   }G_STMT_END
  94
  95 GST_BOILERPLATE_FULL (GstFlacEnc, gst_flac_enc, GstElement, GST_TYPE_ELEMENT,
  96     _do_init);
  97
  98 static void gst_flac_enc_finalize (GObject * object);
  99
 100 static gboolean gst_flac_enc_sink_setcaps (GstPad * pad, GstCaps * caps);
 101 static gboolean gst_flac_enc_sink_event (GstPad * pad, GstEvent * event);
 102 static GstFlowReturn gst_flac_enc_chain (GstPad * pad, GstBuffer * buffer);
 103
 104 static gboolean gst_flac_enc_update_quality (GstFlacEnc * flacenc,
 105     gint quality);
 106 static void gst_flac_enc_set_property (GObject * object, guint prop_id,
 107     const GValue * value, GParamSpec * pspec);
 108 static void gst_flac_enc_get_property (GObject * object, guint prop_id,
 109     GValue * value, GParamSpec * pspec);
 110 static GstStateChangeReturn gst_flac_enc_change_state (GstElement * element,
 111     GstStateChange transition);
 112
 113 static FLAC__StreamEncoderWriteStatus
 114 gst_flac_enc_write_callback (const FLAC__SeekableStreamEncoder * encoder,
 115     const FLAC__byte buffer[], unsigned bytes,
 116     unsigned samples, unsigned current_frame, void *client_data);
 117 static FLAC__SeekableStreamEncoderSeekStatus
 118 gst_flac_enc_seek_callback (const FLAC__SeekableStreamEncoder * encoder,
 119     FLAC__uint64 absolute_byte_offset, void *client_data);
 120 static FLAC__SeekableStreamEncoderTellStatus
 121 gst_flac_enc_tell_callback (const FLAC__SeekableStreamEncoder * encoder,
 122     FLAC__uint64 * absolute_byte_offset, void *client_data);
 123
 124 typedef struct
 125 {
 126   gboolean exhaustive_model_search;
 127   gboolean escape_coding;
 128   gboolean mid_side;
 129   gboolean loose_mid_side;
 130   guint qlp_coeff_precision;
 131   gboolean qlp_coeff_prec_search;
 132   guint min_residual_partition_order;
 133   guint max_residual_partition_order;
 134   guint rice_parameter_search_dist;
 135   guint max_lpc_order;
 136   guint blocksize;
 137 }
 138 GstFlacEncParams;
 139
 140 static const GstFlacEncParams flacenc_params[] = {
 141   {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 2, 2, 0, 0, 1152},
 142   {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 2, 2, 0, 0, 1152},
 143   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 3, 0, 0, 1152},
 144   {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 3, 3, 0, 6, 4608},
 145   {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 3, 3, 0, 8, 4608},
 146   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 3, 3, 0, 8, 4608},
 147   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 4, 0, 8, 4608},
 148   {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 8, 4608},
 149   {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 12, 4608},
 150   {TRUE, TRUE, TRUE, FALSE, 0, FALSE, 0, 16, 0, 32, 4608},
 151 };
 152
 153 #define DEFAULT_QUALITY 5
 154
 155 #define GST_TYPE_FLAC_ENC_QUALITY (gst_flac_enc_quality_get_type ())
 156 GType
 157 gst_flac_enc_quality_get_type (void)
 158 {
 159   static GType qtype = 0;
 160
 161   if (qtype == 0) {
 162     static const GEnumValue values[] = {
 163       {0, "0 - Fastest compression", "0"},
 164       {1, "1", "1"},
 165       {2, "2", "2"},
 166       {3, "3", "3"},
 167       {4, "4", "4"},
 168       {5, "5 - Default", "5"},
 169       {6, "6", "6"},
 170       {7, "7", "7"},
 171       {8, "8 - Highest compression", "8"},
 172       {9, "9 - Insane", "9"},
 173       {0, NULL, NULL}
 174     };
 175
 176     qtype = g_enum_register_static ("GstFlacEncQuality", values);
 177   }
 178   return qtype;
 179 }
 180
 181 static void
 182 gst_flac_enc_base_init (gpointer g_class)
 183 {
 184   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 185
 186   gst_element_class_add_pad_template (element_class,
 187       gst_static_pad_template_get (&src_factory));
 188   gst_element_class_add_pad_template (element_class,
 189       gst_static_pad_template_get (&sink_factory));
 190
 191   gst_element_class_set_details (element_class, &flacenc_details);
 192
 193   GST_DEBUG_CATEGORY_INIT (flacenc_debug, "flacenc", 0,
 194       "Flac encoding element");
 195 }
 196
 197 static void
 198 gst_flac_enc_class_init (GstFlacEncClass * klass)
 199 {
 200   GObjectClass *gobject_class;
 201   GstElementClass *gstelement_class;
 202
 203   gobject_class = (GObjectClass *) klass;
 204   gstelement_class = (GstElementClass *) klass;
 205
 206   gobject_class->set_property = gst_flac_enc_set_property;
 207   gobject_class->get_property = gst_flac_enc_get_property;
 208   gobject_class->finalize = gst_flac_enc_finalize;
 209
 210   g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_QUALITY,
 211       g_param_spec_enum ("quality",
 212           "Quality",
 213           "Speed versus compression tradeoff",
 214           GST_TYPE_FLAC_ENC_QUALITY, DEFAULT_QUALITY, G_PARAM_READWRITE));
 215   g_object_class_install_property (G_OBJECT_CLASS (klass),
 216       PROP_STREAMABLE_SUBSET, g_param_spec_boolean ("streamable_subset",
 217           "Streamable subset",
 218           "true to limit encoder to generating a Subset stream, else false",
 219           TRUE, G_PARAM_READWRITE));
 220   g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MID_SIDE_STEREO,
 221       g_param_spec_boolean ("mid_side_stereo", "Do mid side stereo",
 222           "Do mid side stereo (only for stereo input)",
 223           flacenc_params[DEFAULT_QUALITY].mid_side, G_PARAM_READWRITE));
 224   g_object_class_install_property (G_OBJECT_CLASS (klass),
 225       PROP_LOOSE_MID_SIDE_STEREO, g_param_spec_boolean ("loose_mid_side_stereo",
 226           "Loose mid side stereo", "Loose mid side stereo",
 227           flacenc_params[DEFAULT_QUALITY].loose_mid_side, G_PARAM_READWRITE));
 228   g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_BLOCKSIZE,
 229       g_param_spec_uint ("blocksize", "Blocksize", "Blocksize in samples",
 230           FLAC__MIN_BLOCK_SIZE, FLAC__MAX_BLOCK_SIZE,
 231           flacenc_params[DEFAULT_QUALITY].blocksize, G_PARAM_READWRITE));
 232   g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_MAX_LPC_ORDER,
 233       g_param_spec_uint ("max_lpc_order", "Max LPC order",
 234           "Max LPC order; 0 => use only fixed predictors", 0,
 235           FLAC__MAX_LPC_ORDER, flacenc_params[DEFAULT_QUALITY].max_lpc_order,
 236           G_PARAM_READWRITE));
 237   g_object_class_install_property (G_OBJECT_CLASS (klass),
 238       PROP_QLP_COEFF_PRECISION, g_param_spec_uint ("qlp_coeff_precision",
 239           "QLP coefficients precision",
 240           "Precision in bits of quantized linear-predictor coefficients; 0 = automatic",
 241           0, 32, flacenc_params[DEFAULT_QUALITY].qlp_coeff_precision,
 242           G_PARAM_READWRITE));
 243   g_object_class_install_property (G_OBJECT_CLASS (klass),
 244       PROP_QLP_COEFF_PREC_SEARCH, g_param_spec_boolean ("qlp_coeff_prec_search",
 245           "Do QLP coefficients precision search",
 246           "false = use qlp_coeff_precision, "
 247           "true = search around qlp_coeff_precision, take best",
 248           flacenc_params[DEFAULT_QUALITY].qlp_coeff_prec_search,
 249           G_PARAM_READWRITE));
 250   g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ESCAPE_CODING,
 251       g_param_spec_boolean ("escape_coding", "Do Escape coding",
 252           "search for escape codes in the entropy coding stage "
 253           "for slightly better compression",
 254           flacenc_params[DEFAULT_QUALITY].escape_coding, G_PARAM_READWRITE));
 255   g_object_class_install_property (G_OBJECT_CLASS (klass),
 256       PROP_EXHAUSTIVE_MODEL_SEARCH,
 257       g_param_spec_boolean ("exhaustive_model_search",
 258           "Do exhaustive model search",
 259           "do exhaustive search of LP coefficient quantization (expensive!)",
 260           flacenc_params[DEFAULT_QUALITY].exhaustive_model_search,
 261           G_PARAM_READWRITE));
 262   g_object_class_install_property (G_OBJECT_CLASS (klass),
 263       PROP_MIN_RESIDUAL_PARTITION_ORDER,
 264       g_param_spec_uint ("min_residual_partition_order",
 265           "Min residual partition order",
 266           "Min residual partition order (above 4 doesn't usually help much)", 0,
 267           16, flacenc_params[DEFAULT_QUALITY].min_residual_partition_order,
 268           G_PARAM_READWRITE));
 269   g_object_class_install_property (G_OBJECT_CLASS (klass),
 270       PROP_MAX_RESIDUAL_PARTITION_ORDER,
 271       g_param_spec_uint ("max_residual_partition_order",
 272           "Max residual partition order",
 273           "Max residual partition order (above 4 doesn't usually help much)", 0,
 274           16, flacenc_params[DEFAULT_QUALITY].max_residual_partition_order,
 275           G_PARAM_READWRITE));
 276   g_object_class_install_property (G_OBJECT_CLASS (klass),
 277       PROP_RICE_PARAMETER_SEARCH_DIST,
 278       g_param_spec_uint ("rice_parameter_search_dist",
 279           "rice_parameter_search_dist",
 280           "0 = try only calc'd parameter k; else try all [k-dist..k+dist] "
 281           "parameters, use best", 0, FLAC__MAX_RICE_PARTITION_ORDER,
 282           flacenc_params[DEFAULT_QUALITY].rice_parameter_search_dist,
 283           G_PARAM_READWRITE));
 284
 285   gstelement_class->change_state = gst_flac_enc_change_state;
 286 }
 287
 288 static void
 289 gst_flac_enc_init (GstFlacEnc * flacenc, GstFlacEncClass * klass)
 290 {
 291   GstElementClass *eclass = GST_ELEMENT_CLASS (klass);
 292
 293   flacenc->sinkpad =
 294       gst_pad_new_from_template (gst_element_class_get_pad_template (eclass,
 295           "sink"), "sink");
 296   gst_element_add_pad (GST_ELEMENT (flacenc), flacenc->sinkpad);
 297   gst_pad_set_chain_function (flacenc->sinkpad, gst_flac_enc_chain);
 298   gst_pad_set_event_function (flacenc->sinkpad, gst_flac_enc_sink_event);
 299   gst_pad_set_setcaps_function (flacenc->sinkpad, gst_flac_enc_sink_setcaps);
 300
 301   flacenc->srcpad =
 302       gst_pad_new_from_template (gst_element_class_get_pad_template (eclass,
 303           "src"), "src");
 304   gst_pad_use_fixed_caps (flacenc->srcpad);
 305   gst_element_add_pad (GST_ELEMENT (flacenc), flacenc->srcpad);
 306
 307   flacenc->encoder = FLAC__seekable_stream_encoder_new ();
 308
 309   flacenc->offset = 0;
 310   flacenc->samples_written = 0;
 311   gst_flac_enc_update_quality (flacenc, DEFAULT_QUALITY);
 312   flacenc->tags = gst_tag_list_new ();
 313 }
 314
 315 static void
 316 gst_flac_enc_finalize (GObject * object)
 317 {
 318   GstFlacEnc *flacenc = GST_FLAC_ENC (object);
 319
 320   FLAC__seekable_stream_encoder_delete (flacenc->encoder);
 321
 322   G_OBJECT_CLASS (parent_class)->finalize (object);
 323 }
 324
 325 static void
 326 add_one_tag (const GstTagList * list, const gchar * tag, gpointer user_data)
 327 {
 328   GList *comments;
 329   GList *it;
 330   GstFlacEnc *flacenc = GST_FLAC_ENC (user_data);
 331
 332   comments = gst_tag_to_vorbis_comments (list, tag);
 333   for (it = comments; it != NULL; it = it->next) {
 334     FLAC__StreamMetadata_VorbisComment_Entry commment_entry;
 335
 336     commment_entry.length = strlen (it->data);
 337     commment_entry.entry = it->data;
 338     FLAC__metadata_object_vorbiscomment_insert_comment (flacenc->meta[0],
 339         flacenc->meta[0]->data.vorbis_comment.num_comments,
 340         commment_entry, TRUE);
 341     g_free (it->data);
 342   }
 343   g_list_free (comments);
 344 }
 345
 346 static void
 347 gst_flac_enc_set_metadata (GstFlacEnc * flacenc)
 348 {
 349   const GstTagList *user_tags;
 350   GstTagList *copy;
 351
 352   g_return_if_fail (flacenc != NULL);
 353   user_tags = gst_tag_setter_get_tag_list (GST_TAG_SETTER (flacenc));
 354   if ((flacenc->tags == NULL) && (user_tags == NULL)) {
 355     return;
 356   }
 357   copy = gst_tag_list_merge (user_tags, flacenc->tags,
 358       gst_tag_setter_get_tag_merge_mode (GST_TAG_SETTER (flacenc)));
 359   flacenc->meta = g_malloc (sizeof (FLAC__StreamMetadata **));
 360
 361   flacenc->meta[0] =
 362       FLAC__metadata_object_new (FLAC__METADATA_TYPE_VORBIS_COMMENT);
 363   gst_tag_list_foreach (copy, add_one_tag, flacenc);
 364
 365   if (FLAC__seekable_stream_encoder_set_metadata (flacenc->encoder,
 366           flacenc->meta, 1) != true)
 367     g_warning ("Dude, i'm already initialized!");
 368   gst_tag_list_free (copy);
 369 }
 370
 371 static gboolean
 372 gst_flac_enc_sink_setcaps (GstPad * pad, GstCaps * caps)
 373 {
 374   GstFlacEnc *flacenc;
 375   GstStructure *structure;
 376   FLAC__SeekableStreamEncoderState state;
 377
 378   /* takes a ref on flacenc */
 379   flacenc = GST_FLAC_ENC (gst_pad_get_parent (pad));
 380
 381   if (FLAC__seekable_stream_encoder_get_state (flacenc->encoder) !=
 382       FLAC__SEEKABLE_STREAM_ENCODER_UNINITIALIZED)
 383     goto encoder_already_initialized;
 384
 385   structure = gst_caps_get_structure (caps, 0);
 386
 387   if (!gst_structure_get_int (structure, "channels", &flacenc->channels)
 388       || !gst_structure_get_int (structure, "depth", &flacenc->depth)
 389       || !gst_structure_get_int (structure, "rate", &flacenc->sample_rate))
 390     /* we got caps incompatible with the template? */
 391     g_return_val_if_reached (FALSE);
 392
 393   caps = gst_caps_new_simple ("audio/x-flac",
 394       "channels", G_TYPE_INT, flacenc->channels,
 395       "rate", G_TYPE_INT, flacenc->sample_rate, NULL);
 396
 397   if (!gst_pad_set_caps (flacenc->srcpad, caps))
 398     goto setting_src_caps_failed;
 399
 400   gst_caps_unref (caps);
 401
 402   FLAC__seekable_stream_encoder_set_bits_per_sample (flacenc->encoder,
 403       flacenc->depth);
 404   FLAC__seekable_stream_encoder_set_sample_rate (flacenc->encoder,
 405       flacenc->sample_rate);
 406   FLAC__seekable_stream_encoder_set_channels (flacenc->encoder,
 407       flacenc->channels);
 408
 409   FLAC__seekable_stream_encoder_set_write_callback (flacenc->encoder,
 410       gst_flac_enc_write_callback);
 411   FLAC__seekable_stream_encoder_set_seek_callback (flacenc->encoder,
 412       gst_flac_enc_seek_callback);
 413   FLAC__seekable_stream_encoder_set_tell_callback (flacenc->encoder,
 414       gst_flac_enc_tell_callback);
 415
 416   FLAC__seekable_stream_encoder_set_client_data (flacenc->encoder, flacenc);
 417
 418   gst_flac_enc_set_metadata (flacenc);
 419
 420   state = FLAC__seekable_stream_encoder_init (flacenc->encoder);
 421   if (state != FLAC__STREAM_ENCODER_OK)
 422     goto failed_to_initialize;
 423
 424   gst_object_unref (flacenc);
 425
 426   return TRUE;
 427
 428 encoder_already_initialized:
 429   {
 430     g_warning ("flac already initialized -- fixme allow this");
 431     gst_object_unref (flacenc);
 432     return FALSE;
 433   }
 434 setting_src_caps_failed:
 435   {
 436     GST_DEBUG_OBJECT (flacenc,
 437         "Couldn't set caps on source pad: %" GST_PTR_FORMAT, caps);
 438     gst_caps_unref (caps);
 439     gst_object_unref (flacenc);
 440     return FALSE;
 441   }
 442 failed_to_initialize:
 443   {
 444     GST_ELEMENT_ERROR (flacenc, LIBRARY, INIT, (NULL),
 445         ("could not initialize encoder (wrong parameters?)"));
 446     gst_object_unref (flacenc);
 447     return FALSE;
 448   }
 449 }
 450
 451 static gboolean
 452 gst_flac_enc_update_quality (GstFlacEnc * flacenc, gint quality)
 453 {
 454   flacenc->quality = quality;
 455
 456 #define DO_UPDATE(name, val, str)                                               \
 457   G_STMT_START {                                                                \
 458     if (FLAC__seekable_stream_encoder_get_##name (flacenc->encoder) !=          \
 459         flacenc_params[quality].val) {                                          \
 460       FLAC__seekable_stream_encoder_set_##name (flacenc->encoder,               \
 461           flacenc_params[quality].val);                                         \
 462       g_object_notify (G_OBJECT (flacenc), str);                                \
 463     }                                                                           \
 464   } G_STMT_END
 465
 466   g_object_freeze_notify (G_OBJECT (flacenc));
 467
 468   if (flacenc->channels == 2) {
 469     DO_UPDATE (do_mid_side_stereo, mid_side, "mid_side_stereo");
 470     DO_UPDATE (loose_mid_side_stereo, loose_mid_side, "loose_mid_side");
 471   }
 472
 473   DO_UPDATE (blocksize, blocksize, "blocksize");
 474   DO_UPDATE (max_lpc_order, max_lpc_order, "max_lpc_order");
 475   DO_UPDATE (qlp_coeff_precision, qlp_coeff_precision, "qlp_coeff_precision");
 476   DO_UPDATE (do_qlp_coeff_prec_search, qlp_coeff_prec_search,
 477       "qlp_coeff_prec_search");
 478   DO_UPDATE (do_escape_coding, escape_coding, "escape_coding");
 479   DO_UPDATE (do_exhaustive_model_search, exhaustive_model_search,
 480       "exhaustive_model_search");
 481   DO_UPDATE (min_residual_partition_order, min_residual_partition_order,
 482       "min_residual_partition_order");
 483   DO_UPDATE (max_residual_partition_order, max_residual_partition_order,
 484       "max_residual_partition_order");
 485   DO_UPDATE (rice_parameter_search_dist, rice_parameter_search_dist,
 486       "rice_parameter_search_dist");
 487
 488 #undef DO_UPDATE
 489
 490   g_object_thaw_notify (G_OBJECT (flacenc));
 491
 492   return TRUE;
 493 }
 494
 495 static FLAC__SeekableStreamEncoderSeekStatus
 496 gst_flac_enc_seek_callback (const FLAC__SeekableStreamEncoder * encoder,
 497     FLAC__uint64 absolute_byte_offset, void *client_data)
 498 {
 499   GstFlacEnc *flacenc;
 500   GstEvent *event;
 501   GstPad *peerpad;
 502
 503   flacenc = GST_FLAC_ENC (client_data);
 504
 505   if (flacenc->stopped)
 506     return FLAC__STREAM_ENCODER_OK;
 507
 508   event = gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_BYTES,
 509       absolute_byte_offset, GST_BUFFER_OFFSET_NONE, 0);
 510
 511   if ((peerpad = gst_pad_get_peer (flacenc->srcpad))) {
 512     gboolean ret = gst_pad_send_event (peerpad, event);
 513
 514     gst_object_unref (peerpad);
 515
 516     if (ret) {
 517       GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " %s", absolute_byte_offset,
 518           "succeeded");
 519     } else {
 520       GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " %s", absolute_byte_offset,
 521           "failed");
 522     }
 523   } else {
 524     GST_DEBUG ("Seek to %" G_GUINT64_FORMAT " failed (no peer pad)",
 525         absolute_byte_offset);
 526   }
 527
 528   flacenc->offset = absolute_byte_offset;
 529
 530   return FLAC__STREAM_ENCODER_OK;
 531 }
 532
 533 static FLAC__StreamEncoderWriteStatus
 534 gst_flac_enc_write_callback (const FLAC__SeekableStreamEncoder * encoder,
 535     const FLAC__byte buffer[], unsigned bytes,
 536     unsigned samples, unsigned current_frame, void *client_data)
 537 {
 538   GstFlowReturn ret;
 539   GstFlacEnc *flacenc;
 540   GstBuffer *outbuf;
 541
 542   flacenc = GST_FLAC_ENC (client_data);
 543
 544   if (flacenc->stopped)
 545     return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
 546
 547   if (gst_pad_alloc_buffer_and_set_caps (flacenc->srcpad, flacenc->offset,
 548           bytes, GST_PAD_CAPS (flacenc->srcpad), &outbuf) != GST_FLOW_OK) {
 549     return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
 550   }
 551
 552   memcpy (GST_BUFFER_DATA (outbuf), buffer, bytes);
 553
 554   if (samples > 0 && flacenc->samples_written != (guint64) - 1) {
 555     GST_BUFFER_TIMESTAMP (outbuf) =
 556         GST_FRAMES_TO_CLOCK_TIME (flacenc->samples_written,
 557         flacenc->sample_rate);
 558     GST_BUFFER_DURATION (outbuf) =
 559         GST_FRAMES_TO_CLOCK_TIME (samples, flacenc->sample_rate);
 560     /* offset_end = granulepos for ogg muxer */
 561     GST_BUFFER_OFFSET_END (outbuf) = flacenc->samples_written + samples;
 562   } else {
 563     GST_BUFFER_TIMESTAMP (outbuf) = GST_CLOCK_TIME_NONE;
 564     GST_BUFFER_DURATION (outbuf) = GST_CLOCK_TIME_NONE;
 565   }
 566
 567   GST_DEBUG ("Pushing buffer: ts=%" GST_TIME_FORMAT ", samples=%u, size=%u, "
 568       "pos=%" G_GUINT64_FORMAT, GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (outbuf)),
 569       samples, bytes, flacenc->offset);
 570
 571   ret = gst_pad_push (flacenc->srcpad, outbuf);
 572
 573   flacenc->offset += bytes;
 574   flacenc->samples_written += samples;
 575
 576   if (ret != GST_FLOW_OK && GST_FLOW_IS_FATAL (ret))
 577     return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
 578
 579   return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
 580 }
 581
 582 static FLAC__SeekableStreamEncoderTellStatus
 583 gst_flac_enc_tell_callback (const FLAC__SeekableStreamEncoder * encoder,
 584     FLAC__uint64 * absolute_byte_offset, void *client_data)
 585 {
 586   GstFlacEnc *flacenc = GST_FLAC_ENC (client_data);
 587
 588   *absolute_byte_offset = flacenc->offset;
 589
 590   return FLAC__STREAM_ENCODER_OK;
 591 }
 592
 593 static gboolean
 594 gst_flac_enc_sink_event (GstPad * pad, GstEvent * event)
 595 {
 596   GstFlacEnc *flacenc;
 597   GstTagList *taglist;
 598   gboolean ret = TRUE;
 599
 600   flacenc = GST_FLAC_ENC (gst_pad_get_parent (pad));
 601
 602   GST_DEBUG ("Received %s event on sinkpad", GST_EVENT_TYPE_NAME (event));
 603
 604   switch (GST_EVENT_TYPE (event)) {
 605     case GST_EVENT_NEWSEGMENT:{
 606       GstFormat format;
 607       gint64 start, stream_time;
 608
 609       if (flacenc->offset == 0) {
 610         gst_event_parse_new_segment (event, NULL, NULL, &format, &start, NULL,
 611             &stream_time);
 612       } else {
 613         start = -1;
 614       }
 615       if (start != 0) {
 616         if (flacenc->offset > 0)
 617           GST_DEBUG ("Not handling mid-stream newsegment event");
 618         else
 619           GST_DEBUG ("Not handling newsegment event with non-zero start");
 620       } else {
 621         GstEvent *e = gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_BYTES,
 622             0, -1, 0);
 623
 624         ret = gst_pad_push_event (flacenc->srcpad, e);
 625       }
 626       if (stream_time != 0) {
 627         GST_DEBUG ("Not handling non-zero stream time");
 628       }
 629       gst_event_unref (event);
 630       /* don't push it downstream, we'll generate our own via seek to 0 */
 631       break;
 632     }
 633     case GST_EVENT_EOS:
 634       FLAC__seekable_stream_encoder_finish (flacenc->encoder);
 635       ret = gst_pad_event_default (pad, event);
 636       break;
 637     case GST_EVENT_TAG:
 638       if (flacenc->tags) {
 639         gst_event_parse_tag (event, &taglist);
 640         gst_tag_list_insert (flacenc->tags, taglist, GST_TAG_MERGE_REPLACE);
 641       } else {
 642         g_assert_not_reached ();
 643       }
 644       ret = gst_pad_event_default (pad, event);
 645       break;
 646     default:
 647       ret = gst_pad_event_default (pad, event);
 648       break;
 649   }
 650
 651   gst_object_unref (flacenc);
 652
 653   return ret;
 654 }
 655
 656 static GstFlowReturn
 657 gst_flac_enc_chain (GstPad * pad, GstBuffer * buffer)
 658 {
 659   GstFlacEnc *flacenc;
 660   FLAC__int32 *data;
 661   gulong insize;
 662   gint samples, depth;
 663   gulong i;
 664   FLAC__bool res;
 665
 666   flacenc = GST_FLAC_ENC (gst_pad_get_parent (pad));
 667
 668   depth = flacenc->depth;
 669
 670   insize = GST_BUFFER_SIZE (buffer);
 671   samples = insize / ((depth + 7) >> 3);
 672
 673   data = g_malloc (samples * sizeof (FLAC__int32));
 674
 675   if (depth == 8) {
 676     gint8 *indata = (gint8 *) GST_BUFFER_DATA (buffer);
 677
 678     for (i = 0; i < samples; i++)
 679       data[i] = (FLAC__int32) indata[i];
 680   } else if (depth == 16) {
 681     gint16 *indata = (gint16 *) GST_BUFFER_DATA (buffer);
 682
 683     for (i = 0; i < samples; i++)
 684       data[i] = (FLAC__int32) indata[i];
 685   } else {
 686     g_assert_not_reached ();
 687   }
 688
 689   gst_buffer_unref (buffer);
 690
 691   res = FLAC__seekable_stream_encoder_process_interleaved (flacenc->encoder,
 692       (const FLAC__int32 *) data, samples / flacenc->channels);
 693
 694   g_free (data);
 695
 696   gst_object_unref (flacenc);
 697
 698   if (res)
 699     return GST_FLOW_OK;
 700   else
 701     return GST_FLOW_ERROR;
 702 }
 703
 704 static void
 705 gst_flac_enc_set_property (GObject * object, guint prop_id,
 706     const GValue * value, GParamSpec * pspec)
 707 {
 708   GstFlacEnc *this = GST_FLAC_ENC (object);
 709
 710   GST_OBJECT_LOCK (this);
 711
 712   switch (prop_id) {
 713     case PROP_QUALITY:
 714       gst_flac_enc_update_quality (this, g_value_get_enum (value));
 715       break;
 716     case PROP_STREAMABLE_SUBSET:
 717       FLAC__seekable_stream_encoder_set_streamable_subset (this->encoder,
 718           g_value_get_boolean (value));
 719       break;
 720     case PROP_MID_SIDE_STEREO:
 721       FLAC__seekable_stream_encoder_set_do_mid_side_stereo (this->encoder,
 722           g_value_get_boolean (value));
 723       break;
 724     case PROP_LOOSE_MID_SIDE_STEREO:
 725       FLAC__seekable_stream_encoder_set_loose_mid_side_stereo (this->encoder,
 726           g_value_get_boolean (value));
 727       break;
 728     case PROP_BLOCKSIZE:
 729       FLAC__seekable_stream_encoder_set_blocksize (this->encoder,
 730           g_value_get_uint (value));
 731       break;
 732     case PROP_MAX_LPC_ORDER:
 733       FLAC__seekable_stream_encoder_set_max_lpc_order (this->encoder,
 734           g_value_get_uint (value));
 735       break;
 736     case PROP_QLP_COEFF_PRECISION:
 737       FLAC__seekable_stream_encoder_set_qlp_coeff_precision (this->encoder,
 738           g_value_get_uint (value));
 739       break;
 740     case PROP_QLP_COEFF_PREC_SEARCH:
 741       FLAC__seekable_stream_encoder_set_do_qlp_coeff_prec_search (this->encoder,
 742           g_value_get_boolean (value));
 743       break;
 744     case PROP_ESCAPE_CODING:
 745       FLAC__seekable_stream_encoder_set_do_escape_coding (this->encoder,
 746           g_value_get_boolean (value));
 747       break;
 748     case PROP_EXHAUSTIVE_MODEL_SEARCH:
 749       FLAC__seekable_stream_encoder_set_do_exhaustive_model_search (this->
 750           encoder, g_value_get_boolean (value));
 751       break;
 752     case PROP_MIN_RESIDUAL_PARTITION_ORDER:
 753       FLAC__seekable_stream_encoder_set_min_residual_partition_order (this->
 754           encoder, g_value_get_uint (value));
 755       break;
 756     case PROP_MAX_RESIDUAL_PARTITION_ORDER:
 757       FLAC__seekable_stream_encoder_set_max_residual_partition_order (this->
 758           encoder, g_value_get_uint (value));
 759       break;
 760     case PROP_RICE_PARAMETER_SEARCH_DIST:
 761       FLAC__seekable_stream_encoder_set_rice_parameter_search_dist (this->
 762           encoder, g_value_get_uint (value));
 763       break;
 764     default:
 765       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 766       break;
 767   }
 768
 769   GST_OBJECT_UNLOCK (this);
 770 }
 771
 772 static void
 773 gst_flac_enc_get_property (GObject * object, guint prop_id,
 774     GValue * value, GParamSpec * pspec)
 775 {
 776   GstFlacEnc *this = GST_FLAC_ENC (object);
 777
 778   GST_OBJECT_LOCK (this);
 779
 780   switch (prop_id) {
 781     case PROP_QUALITY:
 782       g_value_set_enum (value, this->quality);
 783       break;
 784     case PROP_STREAMABLE_SUBSET:
 785       g_value_set_boolean (value,
 786           FLAC__seekable_stream_encoder_get_streamable_subset (this->encoder));
 787       break;
 788     case PROP_MID_SIDE_STEREO:
 789       g_value_set_boolean (value,
 790           FLAC__seekable_stream_encoder_get_do_mid_side_stereo (this->encoder));
 791       break;
 792     case PROP_LOOSE_MID_SIDE_STEREO:
 793       g_value_set_boolean (value,
 794           FLAC__seekable_stream_encoder_get_loose_mid_side_stereo (this->
 795               encoder));
 796       break;
 797     case PROP_BLOCKSIZE:
 798       g_value_set_uint (value,
 799           FLAC__seekable_stream_encoder_get_blocksize (this->encoder));
 800       break;
 801     case PROP_MAX_LPC_ORDER:
 802       g_value_set_uint (value,
 803           FLAC__seekable_stream_encoder_get_max_lpc_order (this->encoder));
 804       break;
 805     case PROP_QLP_COEFF_PRECISION:
 806       g_value_set_uint (value,
 807           FLAC__seekable_stream_encoder_get_qlp_coeff_precision (this->
 808               encoder));
 809       break;
 810     case PROP_QLP_COEFF_PREC_SEARCH:
 811       g_value_set_boolean (value,
 812           FLAC__seekable_stream_encoder_get_do_qlp_coeff_prec_search (this->
 813               encoder));
 814       break;
 815     case PROP_ESCAPE_CODING:
 816       g_value_set_boolean (value,
 817           FLAC__seekable_stream_encoder_get_do_escape_coding (this->encoder));
 818       break;
 819     case PROP_EXHAUSTIVE_MODEL_SEARCH:
 820       g_value_set_boolean (value,
 821           FLAC__seekable_stream_encoder_get_do_exhaustive_model_search (this->
 822               encoder));
 823       break;
 824     case PROP_MIN_RESIDUAL_PARTITION_ORDER:
 825       g_value_set_uint (value,
 826           FLAC__seekable_stream_encoder_get_min_residual_partition_order (this->
 827               encoder));
 828       break;
 829     case PROP_MAX_RESIDUAL_PARTITION_ORDER:
 830       g_value_set_uint (value,
 831           FLAC__seekable_stream_encoder_get_max_residual_partition_order (this->
 832               encoder));
 833       break;
 834     case PROP_RICE_PARAMETER_SEARCH_DIST:
 835       g_value_set_uint (value,
 836           FLAC__seekable_stream_encoder_get_rice_parameter_search_dist (this->
 837               encoder));
 838       break;
 839     default:
 840       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 841       break;
 842   }
 843
 844   GST_OBJECT_UNLOCK (this);
 845 }
 846
 847 static GstStateChangeReturn
 848 gst_flac_enc_change_state (GstElement * element, GstStateChange transition)
 849 {
 850   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
 851   GstFlacEnc *flacenc = GST_FLAC_ENC (element);
 852
 853   switch (transition) {
 854     case GST_STATE_CHANGE_NULL_TO_READY:
 855     case GST_STATE_CHANGE_READY_TO_PAUSED:
 856       flacenc->stopped = FALSE;
 857       break;
 858     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
 859     default:
 860       break;
 861   }
 862
 863   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 864
 865   switch (transition) {
 866     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
 867       break;
 868     case GST_STATE_CHANGE_PAUSED_TO_READY:
 869       if (FLAC__seekable_stream_encoder_get_state (flacenc->encoder) !=
 870           FLAC__STREAM_ENCODER_UNINITIALIZED) {
 871         flacenc->stopped = TRUE;
 872         FLAC__seekable_stream_encoder_finish (flacenc->encoder);
 873       }
 874       flacenc->offset = 0;
 875       flacenc->samples_written = 0;
 876       if (flacenc->meta) {
 877         FLAC__metadata_object_delete (flacenc->meta[0]);
 878         g_free (flacenc->meta);
 879         flacenc->meta = NULL;
 880       }
 881       break;
 882     case GST_STATE_CHANGE_READY_TO_NULL:
 883     default:
 884       break;
 885   }
 886
 887   return ret;
 888 }