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