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