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/tag/tag.h>
  29 #include <gst/gsttaginterface.h>
  30 #include "flac_compat.h"
  31
  32 static GstPadTemplate *src_template, *sink_template;
  33
  34 /* elementfactory information */
  35 GstElementDetails flacenc_details = {
  36   "FLAC encoder",
  37   "Codec/Encoder/Audio",
  38   "Encodes audio with the FLAC lossless audio encoder",
  39   "Wim Taymans <wim.taymans@chello.be>",
  40 };
  41
  42 /* FlacEnc signals and args */
  43 enum
  44 {
  45   /* FILL ME */
  46   LAST_SIGNAL
  47 };
  48
  49 enum
  50 {
  51   ARG_0,
  52   ARG_QUALITY,
  53   ARG_STREAMABLE_SUBSET,
  54   ARG_MID_SIDE_STEREO,
  55   ARG_LOOSE_MID_SIDE_STEREO,
  56   ARG_BLOCKSIZE,
  57   ARG_MAX_LPC_ORDER,
  58   ARG_QLP_COEFF_PRECISION,
  59   ARG_QLP_COEFF_PREC_SEARCH,
  60   ARG_ESCAPE_CODING,
  61   ARG_EXHAUSTIVE_MODEL_SEARCH,
  62   ARG_MIN_RESIDUAL_PARTITION_ORDER,
  63   ARG_MAX_RESIDUAL_PARTITION_ORDER,
  64   ARG_RICE_PARAMETER_SEARCH_DIST
  65 };
  66
  67 static void gst_flacenc_base_init (gpointer g_class);
  68 static void gst_flacenc_init (FlacEnc * flacenc);
  69 static void gst_flacenc_class_init (FlacEncClass * klass);
  70 static void gst_flacenc_finalize (GObject * object);
  71
  72 static GstPadLinkReturn
  73 gst_flacenc_sinkconnect (GstPad * pad, const GstCaps * caps);
  74 static void gst_flacenc_chain (GstPad * pad, GstData * _data);
  75
  76 static gboolean gst_flacenc_update_quality (FlacEnc * flacenc, gint quality);
  77 static void gst_flacenc_set_property (GObject * object, guint prop_id,
  78     const GValue * value, GParamSpec * pspec);
  79 static void gst_flacenc_get_property (GObject * object, guint prop_id,
  80     GValue * value, GParamSpec * pspec);
  81 static GstElementStateReturn gst_flacenc_change_state (GstElement * element);
  82
  83 static FLAC__StreamEncoderWriteStatus
  84 gst_flacenc_write_callback (const FLAC__SeekableStreamEncoder * encoder,
  85     const FLAC__byte buffer[], unsigned bytes,
  86     unsigned samples, unsigned current_frame, void *client_data);
  87 static FLAC__SeekableStreamEncoderSeekStatus
  88 gst_flacenc_seek_callback (const FLAC__SeekableStreamEncoder * encoder,
  89     FLAC__uint64 absolute_byte_offset, void *client_data);
  90
  91 static GstElementClass *parent_class = NULL;
  92
  93 /*static guint gst_flacenc_signals[LAST_SIGNAL] = { 0 }; */
  94
  95 GType
  96 flacenc_get_type (void)
  97 {
  98   static GType flacenc_type = 0;
  99
 100   if (!flacenc_type) {
 101     static const GTypeInfo flacenc_info = {
 102       sizeof (FlacEncClass),
 103       gst_flacenc_base_init,
 104       NULL,
 105       (GClassInitFunc) gst_flacenc_class_init,
 106       NULL,
 107       NULL,
 108       sizeof (FlacEnc),
 109       0,
 110       (GInstanceInitFunc) gst_flacenc_init,
 111     };
 112
 113     static const GInterfaceInfo tag_setter_info = {
 114       NULL,
 115       NULL,
 116       NULL
 117     };
 118
 119     flacenc_type =
 120         g_type_register_static (GST_TYPE_ELEMENT, "FlacEnc", &flacenc_info, 0);
 121     g_type_add_interface_static (flacenc_type, GST_TYPE_TAG_SETTER,
 122         &tag_setter_info);
 123   }
 124   return flacenc_type;
 125 }
 126
 127 typedef struct
 128 {
 129   gboolean exhaustive_model_search;
 130   gboolean escape_coding;
 131   gboolean mid_side;
 132   gboolean loose_mid_side;
 133   guint qlp_coeff_precision;
 134   gboolean qlp_coeff_prec_search;
 135   guint min_residual_partition_order;
 136   guint max_residual_partition_order;
 137   guint rice_parameter_search_dist;
 138   guint max_lpc_order;
 139   guint blocksize;
 140 }
 141 FlacEncParams;
 142
 143 static const FlacEncParams flacenc_params[] = {
 144   {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 2, 2, 0, 0, 1152},
 145   {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 2, 2, 0, 0, 1152},
 146   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 3, 0, 0, 1152},
 147   {FALSE, FALSE, FALSE, FALSE, 0, FALSE, 3, 3, 0, 6, 4608},
 148   {FALSE, FALSE, TRUE, TRUE, 0, FALSE, 3, 3, 0, 8, 4608},
 149   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 3, 3, 0, 8, 4608},
 150   {FALSE, FALSE, TRUE, FALSE, 0, FALSE, 0, 4, 0, 8, 4608},
 151   {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 8, 4608},
 152   {TRUE, FALSE, TRUE, FALSE, 0, FALSE, 0, 6, 0, 12, 4608},
 153   {TRUE, TRUE, TRUE, FALSE, 0, FALSE, 0, 16, 0, 32, 4608},
 154 };
 155
 156 #define DEFAULT_QUALITY 5
 157
 158 #define GST_TYPE_FLACENC_QUALITY (gst_flacenc_quality_get_type ())
 159 GType
 160 gst_flacenc_quality_get_type (void)
 161 {
 162   static GType qtype = 0;
 163
 164   if (qtype == 0) {
 165     static const GEnumValue values[] = {
 166       {0, "0", "0 - Fastest compression"},
 167       {1, "1", "1"},
 168       {2, "2", "2"},
 169       {3, "3", "3"},
 170       {4, "4", "4"},
 171       {5, "5", "5 - Default"},
 172       {6, "6", "6"},
 173       {7, "7", "7"},
 174       {8, "8", "8 - Highest compression "},
 175       {9, "9", "9 - Insane"},
 176       {0, NULL, NULL}
 177     };
 178
 179     qtype = g_enum_register_static ("FlacEncQuality", values);
 180   }
 181   return qtype;
 182 }
 183
 184 static GstCaps *
 185 flac_caps_factory (void)
 186 {
 187   return gst_caps_new_simple ("audio/x-flac", NULL);
 188   /* "rate",            GST_PROPS_INT_RANGE (11025, 48000),
 189    * "channels",        GST_PROPS_INT_RANGE (1, 2), */
 190 }
 191
 192 static GstCaps *
 193 raw_caps_factory (void)
 194 {
 195   return gst_caps_new_simple ("audio/x-raw-int",
 196       "endianness", G_TYPE_INT, G_BYTE_ORDER,
 197       "signed", G_TYPE_BOOLEAN, TRUE,
 198       "width", G_TYPE_INT, 16,
 199       "depth", G_TYPE_INT, 16,
 200       "rate", GST_TYPE_INT_RANGE, 11025, 48000,
 201       "channels", GST_TYPE_INT_RANGE, 1, 2, NULL);
 202 }
 203
 204 static void
 205 gst_flacenc_base_init (gpointer g_class)
 206 {
 207   GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 208   GstCaps *raw_caps, *flac_caps;
 209
 210   raw_caps = raw_caps_factory ();
 211   flac_caps = flac_caps_factory ();
 212
 213   sink_template = gst_pad_template_new ("sink", GST_PAD_SINK,
 214       GST_PAD_ALWAYS, raw_caps);
 215   src_template = gst_pad_template_new ("src", GST_PAD_SRC,
 216       GST_PAD_ALWAYS, flac_caps);
 217   gst_element_class_add_pad_template (element_class, sink_template);
 218   gst_element_class_add_pad_template (element_class, src_template);
 219   gst_element_class_set_details (element_class, &flacenc_details);
 220 }
 221
 222 static void
 223 gst_flacenc_class_init (FlacEncClass * klass)
 224 {
 225   GObjectClass *gobject_class;
 226   GstElementClass *gstelement_class;
 227
 228   gobject_class = (GObjectClass *) klass;
 229   gstelement_class = (GstElementClass *) klass;
 230
 231   parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
 232
 233   /* we have no properties atm so this is a bit silly */
 234   gobject_class->set_property = gst_flacenc_set_property;
 235   gobject_class->get_property = gst_flacenc_get_property;
 236   gobject_class->finalize = gst_flacenc_finalize;
 237
 238   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_QUALITY,
 239       g_param_spec_enum ("quality",
 240           "Quality",
 241           "Speed versus compression tradeoff",
 242           GST_TYPE_FLACENC_QUALITY, DEFAULT_QUALITY, G_PARAM_READWRITE));
 243   g_object_class_install_property (G_OBJECT_CLASS (klass),
 244       ARG_STREAMABLE_SUBSET, g_param_spec_boolean ("streamable_subset",
 245           "Streamable subset",
 246           "true to limit encoder to generating a Subset stream, else false",
 247           TRUE, G_PARAM_READWRITE));
 248   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_MID_SIDE_STEREO,
 249       g_param_spec_boolean ("mid_side_stereo", "Do mid side stereo",
 250           "Do mid side stereo (only for stereo input)",
 251           flacenc_params[DEFAULT_QUALITY].mid_side, G_PARAM_READWRITE));
 252   g_object_class_install_property (G_OBJECT_CLASS (klass),
 253       ARG_LOOSE_MID_SIDE_STEREO, g_param_spec_boolean ("loose_mid_side_stereo",
 254           "Loose mid side stereo", "Loose mid side stereo",
 255           flacenc_params[DEFAULT_QUALITY].loose_mid_side, G_PARAM_READWRITE));
 256   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BLOCKSIZE,
 257       g_param_spec_uint ("blocksize", "Blocksize", "Blocksize in samples",
 258           FLAC__MIN_BLOCK_SIZE, FLAC__MAX_BLOCK_SIZE,
 259           flacenc_params[DEFAULT_QUALITY].blocksize, G_PARAM_READWRITE));
 260   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_MAX_LPC_ORDER,
 261       g_param_spec_uint ("max_lpc_order", "Max LPC order",
 262           "Max LPC order; 0 => use only fixed predictors", 0,
 263           FLAC__MAX_LPC_ORDER, flacenc_params[DEFAULT_QUALITY].max_lpc_order,
 264           G_PARAM_READWRITE));
 265   g_object_class_install_property (G_OBJECT_CLASS (klass),
 266       ARG_QLP_COEFF_PRECISION, g_param_spec_uint ("qlp_coeff_precision",
 267           "QLP coefficients precision",
 268           "Precision in bits of quantized linear-predictor coefficients; 0 = automatic",
 269           0, 32, flacenc_params[DEFAULT_QUALITY].qlp_coeff_precision,
 270           G_PARAM_READWRITE));
 271   g_object_class_install_property (G_OBJECT_CLASS (klass),
 272       ARG_QLP_COEFF_PREC_SEARCH, g_param_spec_boolean ("qlp_coeff_prec_search",
 273           "Do QLP coefficients precision search",
 274           "false = use qlp_coeff_precision, "
 275           "true = search around qlp_coeff_precision, take best",
 276           flacenc_params[DEFAULT_QUALITY].qlp_coeff_prec_search,
 277           G_PARAM_READWRITE));
 278   g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_ESCAPE_CODING,
 279       g_param_spec_boolean ("escape_coding", "Do Escape coding",
 280           "search for escape codes in the entropy coding stage "
 281           "for slightly better compression",
 282           flacenc_params[DEFAULT_QUALITY].escape_coding, G_PARAM_READWRITE));
 283   g_object_class_install_property (G_OBJECT_CLASS (klass),
 284       ARG_EXHAUSTIVE_MODEL_SEARCH,
 285       g_param_spec_boolean ("exhaustive_model_search",
 286           "Do exhaustive model search",
 287           "do exhaustive search of LP coefficient quantization (expensive!)",
 288           flacenc_params[DEFAULT_QUALITY].exhaustive_model_search,
 289           G_PARAM_READWRITE));
 290   g_object_class_install_property (G_OBJECT_CLASS (klass),
 291       ARG_MIN_RESIDUAL_PARTITION_ORDER,
 292       g_param_spec_uint ("min_residual_partition_order",
 293           "Min residual partition order",
 294           "Min residual partition order (above 4 doesn't usually help much)", 0,
 295           16, flacenc_params[DEFAULT_QUALITY].min_residual_partition_order,
 296           G_PARAM_READWRITE));
 297   g_object_class_install_property (G_OBJECT_CLASS (klass),
 298       ARG_MAX_RESIDUAL_PARTITION_ORDER,
 299       g_param_spec_uint ("max_residual_partition_order",
 300           "Max residual partition order",
 301           "Max residual partition order (above 4 doesn't usually help much)", 0,
 302           16, flacenc_params[DEFAULT_QUALITY].max_residual_partition_order,
 303           G_PARAM_READWRITE));
 304   g_object_class_install_property (G_OBJECT_CLASS (klass),
 305       ARG_RICE_PARAMETER_SEARCH_DIST,
 306       g_param_spec_uint ("rice_parameter_search_dist",
 307           "rice_parameter_search_dist",
 308           "0 = try only calc'd parameter k; else try all [k-dist..k+dist] "
 309           "parameters, use best", 0, FLAC__MAX_RICE_PARTITION_ORDER,
 310           flacenc_params[DEFAULT_QUALITY].rice_parameter_search_dist,
 311           G_PARAM_READWRITE));
 312
 313   gstelement_class->change_state = gst_flacenc_change_state;
 314 }
 315
 316 static void
 317 gst_flacenc_init (FlacEnc * flacenc)
 318 {
 319   flacenc->sinkpad = gst_pad_new_from_template (sink_template, "sink");
 320   gst_element_add_pad (GST_ELEMENT (flacenc), flacenc->sinkpad);
 321   gst_pad_set_chain_function (flacenc->sinkpad, gst_flacenc_chain);
 322   gst_pad_set_link_function (flacenc->sinkpad, gst_flacenc_sinkconnect);
 323
 324   flacenc->srcpad = gst_pad_new_from_template (src_template, "src");
 325   gst_element_add_pad (GST_ELEMENT (flacenc), flacenc->srcpad);
 326
 327   GST_FLAG_SET (flacenc, GST_ELEMENT_EVENT_AWARE);
 328
 329   flacenc->encoder = FLAC__seekable_stream_encoder_new ();
 330
 331   flacenc->negotiated = FALSE;
 332   flacenc->offset = 0;
 333   flacenc->first = TRUE;
 334   flacenc->first_buf = NULL;
 335   flacenc->data = NULL;
 336   gst_flacenc_update_quality (flacenc, DEFAULT_QUALITY);
 337   flacenc->tags = gst_tag_list_new ();
 338 }
 339
 340 static void
 341 gst_flacenc_finalize (GObject * object)
 342 {
 343   FlacEnc *flacenc = GST_FLACENC (object);
 344
 345   FLAC__seekable_stream_encoder_delete (flacenc->encoder);
 346
 347   G_OBJECT_CLASS (parent_class)->finalize (object);
 348 }
 349
 350 static GstPadLinkReturn
 351 gst_flacenc_sinkconnect (GstPad * pad, const GstCaps * caps)
 352 {
 353   GstPadLinkReturn ret;
 354   FlacEnc *flacenc;
 355   GstStructure *structure;
 356
 357   flacenc = GST_FLACENC (gst_pad_get_parent (pad));
 358
 359   structure = gst_caps_get_structure (caps, 0);
 360
 361   gst_structure_get_int (structure, "channels", &flacenc->channels);
 362   gst_structure_get_int (structure, "depth", &flacenc->depth);
 363   gst_structure_get_int (structure, "rate", &flacenc->sample_rate);
 364
 365   caps = gst_caps_new_simple ("audio/x-flac",
 366       "channels", G_TYPE_INT, flacenc->channels,
 367       "rate", G_TYPE_INT, flacenc->sample_rate, NULL);
 368
 369   ret = gst_pad_try_set_caps (flacenc->srcpad, caps);
 370   if (ret <= 0) {
 371     return ret;
 372   }
 373
 374   FLAC__seekable_stream_encoder_set_bits_per_sample (flacenc->encoder,
 375       flacenc->depth);
 376   FLAC__seekable_stream_encoder_set_sample_rate (flacenc->encoder,
 377       flacenc->sample_rate);
 378   FLAC__seekable_stream_encoder_set_channels (flacenc->encoder,
 379       flacenc->channels);
 380
 381   flacenc->negotiated = TRUE;
 382
 383   return ret;
 384 }
 385
 386 static gboolean
 387 gst_flacenc_update_quality (FlacEnc * flacenc, gint quality)
 388 {
 389   flacenc->quality = quality;
 390
 391 #define DO_UPDATE(name, val, str)                               \
 392   G_STMT_START {                                                \
 393     if (FLAC__seekable_stream_encoder_get_##name (flacenc->encoder) !=  \
 394         flacenc_params[quality].val) {                          \
 395       FLAC__seekable_stream_encoder_set_##name (flacenc->encoder,       \
 396           flacenc_params[quality].val);                         \
 397       g_object_notify (G_OBJECT (flacenc), str);                \
 398     }                                                           \
 399   } G_STMT_END
 400
 401   g_object_freeze_notify (G_OBJECT (flacenc));
 402
 403   if (flacenc->channels == 2) {
 404     DO_UPDATE (do_mid_side_stereo, mid_side, "mid_side_stereo");
 405     DO_UPDATE (loose_mid_side_stereo, loose_mid_side, "loose_mid_side");
 406   }
 407
 408   DO_UPDATE (blocksize, blocksize, "blocksize");
 409   DO_UPDATE (max_lpc_order, max_lpc_order, "max_lpc_order");
 410   DO_UPDATE (qlp_coeff_precision, qlp_coeff_precision, "qlp_coeff_precision");
 411   DO_UPDATE (do_qlp_coeff_prec_search, qlp_coeff_prec_search,
 412       "qlp_coeff_prec_search");
 413   DO_UPDATE (do_escape_coding, escape_coding, "escape_coding");
 414   DO_UPDATE (do_exhaustive_model_search, exhaustive_model_search,
 415       "exhaustive_model_search");
 416   DO_UPDATE (min_residual_partition_order, min_residual_partition_order,
 417       "min_residual_partition_order");
 418   DO_UPDATE (max_residual_partition_order, max_residual_partition_order,
 419       "max_residual_partition_order");
 420   DO_UPDATE (rice_parameter_search_dist, rice_parameter_search_dist,
 421       "rice_parameter_search_dist");
 422
 423 #undef DO_UPDATE
 424
 425   g_object_thaw_notify (G_OBJECT (flacenc));
 426
 427   return TRUE;
 428 }
 429
 430 static FLAC__SeekableStreamEncoderSeekStatus
 431 gst_flacenc_seek_callback (const FLAC__SeekableStreamEncoder * encoder,
 432     FLAC__uint64 absolute_byte_offset, void *client_data)
 433 {
 434   FlacEnc *flacenc;
 435   GstEvent *event;
 436
 437   flacenc = GST_FLACENC (client_data);
 438
 439   if (flacenc->stopped)
 440     return FLAC__STREAM_ENCODER_OK;
 441
 442   event =
 443       gst_event_new_seek ((GstSeekType) (int) (GST_FORMAT_BYTES |
 444           GST_SEEK_METHOD_SET), absolute_byte_offset);
 445
 446   if (event) {
 447     gst_pad_push (flacenc->srcpad, GST_DATA (event));
 448     flacenc->offset = absolute_byte_offset;
 449   }
 450
 451   return FLAC__STREAM_ENCODER_OK;
 452 }
 453
 454 static FLAC__StreamEncoderWriteStatus
 455 gst_flacenc_write_callback (const FLAC__SeekableStreamEncoder * encoder,
 456     const FLAC__byte buffer[], unsigned bytes,
 457     unsigned samples, unsigned current_frame, void *client_data)
 458 {
 459   FlacEnc *flacenc;
 460   GstBuffer *outbuf;
 461
 462   flacenc = GST_FLACENC (client_data);
 463
 464   if (flacenc->stopped)
 465     return FLAC__STREAM_ENCODER_OK;
 466
 467   outbuf = gst_buffer_new_and_alloc (bytes);
 468
 469   memcpy (GST_BUFFER_DATA (outbuf), buffer, bytes);
 470
 471   if (flacenc->first) {
 472     flacenc->first_buf = outbuf;
 473     gst_buffer_ref (outbuf);
 474     flacenc->first = FALSE;
 475   }
 476
 477   gst_pad_push (flacenc->srcpad, GST_DATA (outbuf));
 478   flacenc->offset += bytes;
 479
 480   return FLAC__STREAM_ENCODER_OK;
 481 }
 482
 483 static FLAC__SeekableStreamEncoderTellStatus
 484 gst_flacenc_tell_callback (const FLAC__SeekableStreamEncoder * encoder,
 485     FLAC__uint64 * absolute_byte_offset, void *client_data)
 486 {
 487   FlacEnc *flacenc = GST_FLACENC (client_data);
 488
 489   *absolute_byte_offset = flacenc->offset;
 490
 491   return FLAC__STREAM_ENCODER_OK;
 492 }
 493
 494 static void
 495 add_one_tag (const GstTagList * list, const gchar * tag, gpointer user_data)
 496 {
 497   GList *comments;
 498   GList *it;
 499   FlacEnc *flacenc = GST_FLACENC (user_data);
 500
 501   comments = gst_tag_to_vorbis_comments (list, tag);
 502   for (it = comments; it != NULL; it = it->next) {
 503     FLAC__StreamMetadata_VorbisComment_Entry commment_entry;
 504
 505     commment_entry.length = strlen (it->data);
 506     commment_entry.entry = it->data;
 507     FLAC__metadata_object_vorbiscomment_insert_comment (flacenc->meta[0],
 508         flacenc->meta[0]->data.vorbis_comment.num_comments,
 509         commment_entry, TRUE);
 510     g_free (it->data);
 511   }
 512   g_list_free (comments);
 513 }
 514
 515 static void
 516 gst_flacenc_set_metadata (FlacEnc * flacenc)
 517 {
 518   const GstTagList *user_tags;
 519   GstTagList *copy;
 520
 521   g_return_if_fail (flacenc != NULL);
 522   user_tags = gst_tag_setter_get_list (GST_TAG_SETTER (flacenc));
 523   if ((flacenc->tags == NULL) && (user_tags == NULL)) {
 524     return;
 525   }
 526   copy = gst_tag_list_merge (user_tags, flacenc->tags,
 527       gst_tag_setter_get_merge_mode (GST_TAG_SETTER (flacenc)));
 528   flacenc->meta = g_malloc (sizeof (FLAC__StreamMetadata **));
 529
 530   flacenc->meta[0] =
 531       FLAC__metadata_object_new (FLAC__METADATA_TYPE_VORBIS_COMMENT);
 532   gst_tag_list_foreach (copy, add_one_tag, flacenc);
 533
 534   if (FLAC__seekable_stream_encoder_set_metadata (flacenc->encoder,
 535           flacenc->meta, 1) != true)
 536     g_warning ("Dude, i'm already initialized!");
 537   gst_tag_list_free (copy);
 538 }
 539
 540
 541 static void
 542 gst_flacenc_chain (GstPad * pad, GstData * _data)
 543 {
 544   GstBuffer *buf = GST_BUFFER (_data);
 545   FlacEnc *flacenc;
 546   FLAC__int32 *data;
 547   gulong insize;
 548   gint samples, depth;
 549   gulong i;
 550   FLAC__bool res;
 551
 552   g_return_if_fail (buf != NULL);
 553
 554   flacenc = GST_FLACENC (gst_pad_get_parent (pad));
 555
 556   if (GST_IS_EVENT (buf)) {
 557     GstEvent *event = GST_EVENT (buf);
 558
 559     switch (GST_EVENT_TYPE (event)) {
 560       case GST_EVENT_EOS:
 561         FLAC__seekable_stream_encoder_finish (flacenc->encoder);
 562         break;
 563       case GST_EVENT_TAG:
 564         if (flacenc->tags) {
 565           gst_tag_list_insert (flacenc->tags, gst_event_tag_get_list (event),
 566               GST_TAG_MERGE_REPLACE);
 567         } else {
 568           g_assert_not_reached ();
 569         }
 570         break;
 571       default:
 572         break;
 573     }
 574     gst_pad_event_default (pad, event);
 575     return;
 576   }
 577
 578   if (!flacenc->negotiated) {
 579     GST_ELEMENT_ERROR (flacenc, CORE, NEGOTIATION, (NULL),
 580         ("format wasn't negotiated before chain function"));
 581     return;
 582   }
 583
 584   depth = flacenc->depth;
 585
 586   insize = GST_BUFFER_SIZE (buf);
 587   samples = insize / ((depth + 7) >> 3);
 588
 589   if (FLAC__seekable_stream_encoder_get_state (flacenc->encoder) ==
 590       FLAC__SEEKABLE_STREAM_ENCODER_UNINITIALIZED) {
 591     FLAC__SeekableStreamEncoderState state;
 592
 593     FLAC__seekable_stream_encoder_set_write_callback (flacenc->encoder,
 594         gst_flacenc_write_callback);
 595     FLAC__seekable_stream_encoder_set_seek_callback (flacenc->encoder,
 596         gst_flacenc_seek_callback);
 597     FLAC__seekable_stream_encoder_set_tell_callback (flacenc->encoder,
 598         gst_flacenc_tell_callback);
 599
 600     FLAC__seekable_stream_encoder_set_client_data (flacenc->encoder, flacenc);
 601
 602     gst_flacenc_set_metadata (flacenc);
 603     state = FLAC__seekable_stream_encoder_init (flacenc->encoder);
 604     if (state != FLAC__STREAM_ENCODER_OK) {
 605       GST_ELEMENT_ERROR (flacenc, LIBRARY, INIT, (NULL),
 606           ("could not initialize encoder (wrong parameters?)"));
 607       return;
 608     }
 609   }
 610
 611   /* we keep a pointer in the flacenc struct because we are freeing the data
 612    * after a push opreration that might never return */
 613   data = flacenc->data = g_malloc (samples * sizeof (FLAC__int32));
 614
 615   if (depth == 8) {
 616     gint8 *indata = (gint8 *) GST_BUFFER_DATA (buf);
 617
 618
 619     for (i = 0; i < samples; i++) {
 620       data[i] = (FLAC__int32) * indata++;
 621     }
 622   } else if (depth == 16) {
 623     gint16 *indata = (gint16 *) GST_BUFFER_DATA (buf);
 624
 625     for (i = 0; i < samples; i++) {
 626       data[i] = (FLAC__int32) * indata++;
 627     }
 628   }
 629
 630   gst_buffer_unref (buf);
 631
 632   res = FLAC__seekable_stream_encoder_process_interleaved (flacenc->encoder,
 633       (const FLAC__int32 *) data, samples / flacenc->channels);
 634
 635   g_free (flacenc->data);
 636   flacenc->data = NULL;
 637
 638   if (!res) {
 639     GST_ELEMENT_ERROR (flacenc, STREAM, ENCODE, (NULL), (NULL));
 640   }
 641 }
 642
 643 static void
 644 gst_flacenc_set_property (GObject * object, guint prop_id,
 645     const GValue * value, GParamSpec * pspec)
 646 {
 647   FlacEnc *this;
 648
 649   this = (FlacEnc *) object;
 650   switch (prop_id) {
 651     case ARG_QUALITY:
 652       gst_flacenc_update_quality (this, g_value_get_enum (value));
 653       break;
 654     case ARG_STREAMABLE_SUBSET:
 655       FLAC__seekable_stream_encoder_set_streamable_subset (this->encoder,
 656           g_value_get_boolean (value));
 657       break;
 658     case ARG_MID_SIDE_STEREO:
 659       FLAC__seekable_stream_encoder_set_do_mid_side_stereo (this->encoder,
 660           g_value_get_boolean (value));
 661       break;
 662     case ARG_LOOSE_MID_SIDE_STEREO:
 663       FLAC__seekable_stream_encoder_set_loose_mid_side_stereo (this->encoder,
 664           g_value_get_boolean (value));
 665       break;
 666     case ARG_BLOCKSIZE:
 667       FLAC__seekable_stream_encoder_set_blocksize (this->encoder,
 668           g_value_get_uint (value));
 669       break;
 670     case ARG_MAX_LPC_ORDER:
 671       FLAC__seekable_stream_encoder_set_max_lpc_order (this->encoder,
 672           g_value_get_uint (value));
 673       break;
 674     case ARG_QLP_COEFF_PRECISION:
 675       FLAC__seekable_stream_encoder_set_qlp_coeff_precision (this->encoder,
 676           g_value_get_uint (value));
 677       break;
 678     case ARG_QLP_COEFF_PREC_SEARCH:
 679       FLAC__seekable_stream_encoder_set_do_qlp_coeff_prec_search (this->encoder,
 680           g_value_get_boolean (value));
 681       break;
 682     case ARG_ESCAPE_CODING:
 683       FLAC__seekable_stream_encoder_set_do_escape_coding (this->encoder,
 684           g_value_get_boolean (value));
 685       break;
 686     case ARG_EXHAUSTIVE_MODEL_SEARCH:
 687       FLAC__seekable_stream_encoder_set_do_exhaustive_model_search (this->
 688           encoder, g_value_get_boolean (value));
 689       break;
 690     case ARG_MIN_RESIDUAL_PARTITION_ORDER:
 691       FLAC__seekable_stream_encoder_set_min_residual_partition_order (this->
 692           encoder, g_value_get_uint (value));
 693       break;
 694     case ARG_MAX_RESIDUAL_PARTITION_ORDER:
 695       FLAC__seekable_stream_encoder_set_max_residual_partition_order (this->
 696           encoder, g_value_get_uint (value));
 697       break;
 698     case ARG_RICE_PARAMETER_SEARCH_DIST:
 699       FLAC__seekable_stream_encoder_set_rice_parameter_search_dist (this->
 700           encoder, g_value_get_uint (value));
 701       break;
 702     default:
 703       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 704       return;
 705   }
 706 }
 707
 708 static void
 709 gst_flacenc_get_property (GObject * object, guint prop_id,
 710     GValue * value, GParamSpec * pspec)
 711 {
 712   FlacEnc *this;
 713
 714   this = (FlacEnc *) object;
 715
 716   switch (prop_id) {
 717     case ARG_QUALITY:
 718       g_value_set_enum (value, this->quality);
 719       break;
 720     case ARG_STREAMABLE_SUBSET:
 721       g_value_set_boolean (value,
 722           FLAC__seekable_stream_encoder_get_streamable_subset (this->encoder));
 723       break;
 724     case ARG_MID_SIDE_STEREO:
 725       g_value_set_boolean (value,
 726           FLAC__seekable_stream_encoder_get_do_mid_side_stereo (this->encoder));
 727       break;
 728     case ARG_LOOSE_MID_SIDE_STEREO:
 729       g_value_set_boolean (value,
 730           FLAC__seekable_stream_encoder_get_loose_mid_side_stereo (this->
 731               encoder));
 732       break;
 733     case ARG_BLOCKSIZE:
 734       g_value_set_uint (value,
 735           FLAC__seekable_stream_encoder_get_blocksize (this->encoder));
 736       break;
 737     case ARG_MAX_LPC_ORDER:
 738       g_value_set_uint (value,
 739           FLAC__seekable_stream_encoder_get_max_lpc_order (this->encoder));
 740       break;
 741     case ARG_QLP_COEFF_PRECISION:
 742       g_value_set_uint (value,
 743           FLAC__seekable_stream_encoder_get_qlp_coeff_precision (this->
 744               encoder));
 745       break;
 746     case ARG_QLP_COEFF_PREC_SEARCH:
 747       g_value_set_boolean (value,
 748           FLAC__seekable_stream_encoder_get_do_qlp_coeff_prec_search (this->
 749               encoder));
 750       break;
 751     case ARG_ESCAPE_CODING:
 752       g_value_set_boolean (value,
 753           FLAC__seekable_stream_encoder_get_do_escape_coding (this->encoder));
 754       break;
 755     case ARG_EXHAUSTIVE_MODEL_SEARCH:
 756       g_value_set_boolean (value,
 757           FLAC__seekable_stream_encoder_get_do_exhaustive_model_search (this->
 758               encoder));
 759       break;
 760     case ARG_MIN_RESIDUAL_PARTITION_ORDER:
 761       g_value_set_uint (value,
 762           FLAC__seekable_stream_encoder_get_min_residual_partition_order (this->
 763               encoder));
 764       break;
 765     case ARG_MAX_RESIDUAL_PARTITION_ORDER:
 766       g_value_set_uint (value,
 767           FLAC__seekable_stream_encoder_get_max_residual_partition_order (this->
 768               encoder));
 769       break;
 770     case ARG_RICE_PARAMETER_SEARCH_DIST:
 771       g_value_set_uint (value,
 772           FLAC__seekable_stream_encoder_get_rice_parameter_search_dist (this->
 773               encoder));
 774       break;
 775     default:
 776       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 777       break;
 778   }
 779 }
 780
 781 static GstElementStateReturn
 782 gst_flacenc_change_state (GstElement * element)
 783 {
 784   FlacEnc *flacenc = GST_FLACENC (element);
 785
 786   switch (GST_STATE_TRANSITION (element)) {
 787     case GST_STATE_NULL_TO_READY:
 788     case GST_STATE_READY_TO_PAUSED:
 789       flacenc->first = TRUE;
 790       flacenc->stopped = FALSE;
 791       break;
 792     case GST_STATE_PAUSED_TO_PLAYING:
 793     case GST_STATE_PLAYING_TO_PAUSED:
 794       break;
 795     case GST_STATE_PAUSED_TO_READY:
 796       if (FLAC__seekable_stream_encoder_get_state (flacenc->encoder) !=
 797           FLAC__STREAM_ENCODER_UNINITIALIZED) {
 798         flacenc->stopped = TRUE;
 799         FLAC__seekable_stream_encoder_finish (flacenc->encoder);
 800       }
 801       flacenc->negotiated = FALSE;
 802       flacenc->offset = 0;
 803       if (flacenc->first_buf)
 804         gst_buffer_unref (flacenc->first_buf);
 805       flacenc->first_buf = NULL;
 806       g_free (flacenc->data);
 807       flacenc->data = NULL;
 808       if (flacenc->meta) {
 809         FLAC__metadata_object_delete (flacenc->meta[0]);
 810         g_free (flacenc->meta);
 811         flacenc->meta = NULL;
 812       }
 813       break;
 814     case GST_STATE_READY_TO_NULL:
 815     default:
 816       break;
 817   }
 818
 819   if (GST_ELEMENT_CLASS (parent_class)->change_state)
 820     return GST_ELEMENT_CLASS (parent_class)->change_state (element);
 821
 822   return GST_STATE_SUCCESS;
 823 }