src/libFLAC++/seekable_stream_encoder.cpp

   1 /* libFLAC++ - Free Lossless Audio Codec library
   2  * Copyright (C) 2002,2003,2004,2005,2006  Josh Coalson
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * - Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *
  11  * - Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the distribution.
  14  *
  15  * - Neither the name of the Xiph.org Foundation nor the names of its
  16  * contributors may be used to endorse or promote products derived from
  17  * this software without specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
  23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  24  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  25  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  26  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  27  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  28  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  29  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include "FLAC++/encoder.h"
  33 #include "FLAC/assert.h"
  34
  35 #ifdef _MSC_VER
  36 // warning C4800: 'int' : forcing to bool 'true' or 'false' (performance warning)
  37 #pragma warning ( disable : 4800 )
  38 #endif
  39
  40 namespace FLAC {
  41         namespace Encoder {
  42
  43                 SeekableStream::SeekableStream():
  44                 encoder_(::FLAC__seekable_stream_encoder_new())
  45                 { }
  46
  47                 SeekableStream::~SeekableStream()
  48                 {
  49                         if(0 != encoder_) {
  50                                 ::FLAC__seekable_stream_encoder_finish(encoder_);
  51                                 ::FLAC__seekable_stream_encoder_delete(encoder_);
  52                         }
  53                 }
  54
  55                 bool SeekableStream::is_valid() const
  56                 {
  57                         return 0 != encoder_;
  58                 }
  59
  60                 bool SeekableStream::set_verify(bool value)
  61                 {
  62                         FLAC__ASSERT(is_valid());
  63                         return (bool)::FLAC__seekable_stream_encoder_set_verify(encoder_, value);
  64                 }
  65
  66                 bool SeekableStream::set_streamable_subset(bool value)
  67                 {
  68                         FLAC__ASSERT(is_valid());
  69                         return (bool)::FLAC__seekable_stream_encoder_set_streamable_subset(encoder_, value);
  70                 }
  71
  72                 bool SeekableStream::set_do_mid_side_stereo(bool value)
  73                 {
  74                         FLAC__ASSERT(is_valid());
  75                         return (bool)::FLAC__seekable_stream_encoder_set_do_mid_side_stereo(encoder_, value);
  76                 }
  77
  78                 bool SeekableStream::set_loose_mid_side_stereo(bool value)
  79                 {
  80                         FLAC__ASSERT(is_valid());
  81                         return (bool)::FLAC__seekable_stream_encoder_set_loose_mid_side_stereo(encoder_, value);
  82                 }
  83
  84                 bool SeekableStream::set_channels(unsigned value)
  85                 {
  86                         FLAC__ASSERT(is_valid());
  87                         return (bool)::FLAC__seekable_stream_encoder_set_channels(encoder_, value);
  88                 }
  89
  90                 bool SeekableStream::set_bits_per_sample(unsigned value)
  91                 {
  92                         FLAC__ASSERT(is_valid());
  93                         return (bool)::FLAC__seekable_stream_encoder_set_bits_per_sample(encoder_, value);
  94                 }
  95
  96                 bool SeekableStream::set_sample_rate(unsigned value)
  97                 {
  98                         FLAC__ASSERT(is_valid());
  99                         return (bool)::FLAC__seekable_stream_encoder_set_sample_rate(encoder_, value);
 100                 }
 101
 102                 bool SeekableStream::set_blocksize(unsigned value)
 103                 {
 104                         FLAC__ASSERT(is_valid());
 105                         return (bool)::FLAC__seekable_stream_encoder_set_blocksize(encoder_, value);
 106                 }
 107
 108                 bool SeekableStream::set_apodization(const char *specification)
 109                 {
 110                         FLAC__ASSERT(is_valid());
 111                         return (bool)::FLAC__seekable_stream_encoder_set_apodization(encoder_, specification);
 112                 }
 113
 114                 bool SeekableStream::set_max_lpc_order(unsigned value)
 115                 {
 116                         FLAC__ASSERT(is_valid());
 117                         return (bool)::FLAC__seekable_stream_encoder_set_max_lpc_order(encoder_, value);
 118                 }
 119
 120                 bool SeekableStream::set_qlp_coeff_precision(unsigned value)
 121                 {
 122                         FLAC__ASSERT(is_valid());
 123                         return (bool)::FLAC__seekable_stream_encoder_set_qlp_coeff_precision(encoder_, value);
 124                 }
 125
 126                 bool SeekableStream::set_do_qlp_coeff_prec_search(bool value)
 127                 {
 128                         FLAC__ASSERT(is_valid());
 129                         return (bool)::FLAC__seekable_stream_encoder_set_do_qlp_coeff_prec_search(encoder_, value);
 130                 }
 131
 132                 bool SeekableStream::set_do_escape_coding(bool value)
 133                 {
 134                         FLAC__ASSERT(is_valid());
 135                         return (bool)::FLAC__seekable_stream_encoder_set_do_escape_coding(encoder_, value);
 136                 }
 137
 138                 bool SeekableStream::set_do_exhaustive_model_search(bool value)
 139                 {
 140                         FLAC__ASSERT(is_valid());
 141                         return (bool)::FLAC__seekable_stream_encoder_set_do_exhaustive_model_search(encoder_, value);
 142                 }
 143
 144                 bool SeekableStream::set_min_residual_partition_order(unsigned value)
 145                 {
 146                         FLAC__ASSERT(is_valid());
 147                         return (bool)::FLAC__seekable_stream_encoder_set_min_residual_partition_order(encoder_, value);
 148                 }
 149
 150                 bool SeekableStream::set_max_residual_partition_order(unsigned value)
 151                 {
 152                         FLAC__ASSERT(is_valid());
 153                         return (bool)::FLAC__seekable_stream_encoder_set_max_residual_partition_order(encoder_, value);
 154                 }
 155
 156                 bool SeekableStream::set_rice_parameter_search_dist(unsigned value)
 157                 {
 158                         FLAC__ASSERT(is_valid());
 159                         return (bool)::FLAC__seekable_stream_encoder_set_rice_parameter_search_dist(encoder_, value);
 160                 }
 161
 162                 bool SeekableStream::set_total_samples_estimate(FLAC__uint64 value)
 163                 {
 164                         FLAC__ASSERT(is_valid());
 165                         return (bool)::FLAC__seekable_stream_encoder_set_total_samples_estimate(encoder_, value);
 166                 }
 167
 168                 bool SeekableStream::set_metadata(::FLAC__StreamMetadata **metadata, unsigned num_blocks)
 169                 {
 170                         FLAC__ASSERT(is_valid());
 171                         return (bool)::FLAC__seekable_stream_encoder_set_metadata(encoder_, metadata, num_blocks);
 172                 }
 173
 174                 bool SeekableStream::set_metadata(FLAC::Metadata::Prototype **metadata, unsigned num_blocks)
 175                 {
 176                         FLAC__ASSERT(is_valid());
 177 #if (defined _MSC_VER) || (defined __SUNPRO_CC)
 178                         // MSVC++ can't handle:
 179                         // ::FLAC__StreamMetadata *m[num_blocks];
 180                         // so we do this ugly workaround
 181                         ::FLAC__StreamMetadata **m = new ::FLAC__StreamMetadata*[num_blocks];
 182 #else
 183                         ::FLAC__StreamMetadata *m[num_blocks];
 184 #endif
 185                         for(unsigned i = 0; i < num_blocks; i++) {
 186                                 // we can get away with this since we know the encoder will only correct the is_last flags
 187                                 m[i] = const_cast< ::FLAC__StreamMetadata*>((::FLAC__StreamMetadata*)metadata[i]);
 188                         }
 189 #if (defined _MSC_VER) || (defined __SUNPRO_CC)
 190                         // complete the hack
 191                         const bool ok = (bool)::FLAC__seekable_stream_encoder_set_metadata(encoder_, m, num_blocks);
 192                         delete [] m;
 193                         return ok;
 194 #else
 195                         return (bool)::FLAC__seekable_stream_encoder_set_metadata(encoder_, m, num_blocks);
 196 #endif
 197                 }
 198
 199                 SeekableStream::State SeekableStream::get_state() const
 200                 {
 201                         FLAC__ASSERT(is_valid());
 202                         return State(::FLAC__seekable_stream_encoder_get_state(encoder_));
 203                 }
 204
 205                 Stream::State SeekableStream::get_stream_encoder_state() const
 206                 {
 207                         FLAC__ASSERT(is_valid());
 208                         return Stream::State(::FLAC__seekable_stream_encoder_get_stream_encoder_state(encoder_));
 209                 }
 210
 211                 Decoder::Stream::State SeekableStream::get_verify_decoder_state() const
 212                 {
 213                         FLAC__ASSERT(is_valid());
 214                         return Decoder::Stream::State(::FLAC__seekable_stream_encoder_get_verify_decoder_state(encoder_));
 215                 }
 216
 217                 void SeekableStream::get_verify_decoder_error_stats(FLAC__uint64 *absolute_sample, unsigned *frame_number, unsigned *channel, unsigned *sample, FLAC__int32 *expected, FLAC__int32 *got)
 218                 {
 219                         FLAC__ASSERT(is_valid());
 220                         ::FLAC__seekable_stream_encoder_get_verify_decoder_error_stats(encoder_, absolute_sample, frame_number, channel, sample, expected, got);
 221                 }
 222
 223                 bool SeekableStream::get_verify() const
 224                 {
 225                         FLAC__ASSERT(is_valid());
 226                         return (bool)::FLAC__seekable_stream_encoder_get_verify(encoder_);
 227                 }
 228
 229                 bool SeekableStream::get_streamable_subset() const
 230                 {
 231                         FLAC__ASSERT(is_valid());
 232                         return (bool)::FLAC__seekable_stream_encoder_get_streamable_subset(encoder_);
 233                 }
 234
 235                 bool SeekableStream::get_do_mid_side_stereo() const
 236                 {
 237                         FLAC__ASSERT(is_valid());
 238                         return (bool)::FLAC__seekable_stream_encoder_get_do_mid_side_stereo(encoder_);
 239                 }
 240
 241                 bool SeekableStream::get_loose_mid_side_stereo() const
 242                 {
 243                         FLAC__ASSERT(is_valid());
 244                         return (bool)::FLAC__seekable_stream_encoder_get_loose_mid_side_stereo(encoder_);
 245                 }
 246
 247                 unsigned SeekableStream::get_channels() const
 248                 {
 249                         FLAC__ASSERT(is_valid());
 250                         return ::FLAC__seekable_stream_encoder_get_channels(encoder_);
 251                 }
 252
 253                 unsigned SeekableStream::get_bits_per_sample() const
 254                 {
 255                         FLAC__ASSERT(is_valid());
 256                         return ::FLAC__seekable_stream_encoder_get_bits_per_sample(encoder_);
 257                 }
 258
 259                 unsigned SeekableStream::get_sample_rate() const
 260                 {
 261                         FLAC__ASSERT(is_valid());
 262                         return ::FLAC__seekable_stream_encoder_get_sample_rate(encoder_);
 263                 }
 264
 265                 unsigned SeekableStream::get_blocksize() const
 266                 {
 267                         FLAC__ASSERT(is_valid());
 268                         return ::FLAC__seekable_stream_encoder_get_blocksize(encoder_);
 269                 }
 270
 271                 unsigned SeekableStream::get_max_lpc_order() const
 272                 {
 273                         FLAC__ASSERT(is_valid());
 274                         return ::FLAC__seekable_stream_encoder_get_max_lpc_order(encoder_);
 275                 }
 276
 277                 unsigned SeekableStream::get_qlp_coeff_precision() const
 278                 {
 279                         FLAC__ASSERT(is_valid());
 280                         return ::FLAC__seekable_stream_encoder_get_qlp_coeff_precision(encoder_);
 281                 }
 282
 283                 bool SeekableStream::get_do_qlp_coeff_prec_search() const
 284                 {
 285                         FLAC__ASSERT(is_valid());
 286                         return (bool)::FLAC__seekable_stream_encoder_get_do_qlp_coeff_prec_search(encoder_);
 287                 }
 288
 289                 bool SeekableStream::get_do_escape_coding() const
 290                 {
 291                         FLAC__ASSERT(is_valid());
 292                         return (bool)::FLAC__seekable_stream_encoder_get_do_escape_coding(encoder_);
 293                 }
 294
 295                 bool SeekableStream::get_do_exhaustive_model_search() const
 296                 {
 297                         FLAC__ASSERT(is_valid());
 298                         return (bool)::FLAC__seekable_stream_encoder_get_do_exhaustive_model_search(encoder_);
 299                 }
 300
 301                 unsigned SeekableStream::get_min_residual_partition_order() const
 302                 {
 303                         FLAC__ASSERT(is_valid());
 304                         return ::FLAC__seekable_stream_encoder_get_min_residual_partition_order(encoder_);
 305                 }
 306
 307                 unsigned SeekableStream::get_max_residual_partition_order() const
 308                 {
 309                         FLAC__ASSERT(is_valid());
 310                         return ::FLAC__seekable_stream_encoder_get_max_residual_partition_order(encoder_);
 311                 }
 312
 313                 unsigned SeekableStream::get_rice_parameter_search_dist() const
 314                 {
 315                         FLAC__ASSERT(is_valid());
 316                         return ::FLAC__seekable_stream_encoder_get_rice_parameter_search_dist(encoder_);
 317                 }
 318
 319                 FLAC__uint64 SeekableStream::get_total_samples_estimate() const
 320                 {
 321                         FLAC__ASSERT(is_valid());
 322                         return ::FLAC__seekable_stream_encoder_get_total_samples_estimate(encoder_);
 323                 }
 324
 325                 SeekableStream::State SeekableStream::init()
 326                 {
 327                         FLAC__ASSERT(is_valid());
 328                         ::FLAC__seekable_stream_encoder_set_seek_callback(encoder_, seek_callback_);
 329                         ::FLAC__seekable_stream_encoder_set_tell_callback(encoder_, tell_callback_);
 330                         ::FLAC__seekable_stream_encoder_set_write_callback(encoder_, write_callback_);
 331                         ::FLAC__seekable_stream_encoder_set_client_data(encoder_, (void*)this);
 332                         return State(::FLAC__seekable_stream_encoder_init(encoder_));
 333                 }
 334
 335                 void SeekableStream::finish()
 336                 {
 337                         FLAC__ASSERT(is_valid());
 338                         ::FLAC__seekable_stream_encoder_finish(encoder_);
 339                 }
 340
 341                 bool SeekableStream::process(const FLAC__int32 * const buffer[], unsigned samples)
 342                 {
 343                         FLAC__ASSERT(is_valid());
 344                         return (bool)::FLAC__seekable_stream_encoder_process(encoder_, buffer, samples);
 345                 }
 346
 347                 bool SeekableStream::process_interleaved(const FLAC__int32 buffer[], unsigned samples)
 348                 {
 349                         FLAC__ASSERT(is_valid());
 350                         return (bool)::FLAC__seekable_stream_encoder_process_interleaved(encoder_, buffer, samples);
 351                 }
 352
 353                 ::FLAC__SeekableStreamEncoderSeekStatus SeekableStream::seek_callback_(const ::FLAC__SeekableStreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data)
 354                 {
 355                         (void)encoder;
 356                         FLAC__ASSERT(0 != client_data);
 357                         SeekableStream *instance = reinterpret_cast<SeekableStream *>(client_data);
 358                         FLAC__ASSERT(0 != instance);
 359                         return instance->seek_callback(absolute_byte_offset);
 360                 }
 361
 362                 ::FLAC__SeekableStreamEncoderTellStatus SeekableStream::tell_callback_(const ::FLAC__SeekableStreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data)
 363                 {
 364                         (void)encoder;
 365                         FLAC__ASSERT(0 != client_data);
 366                         SeekableStream *instance = reinterpret_cast<SeekableStream *>(client_data);
 367                         FLAC__ASSERT(0 != instance);
 368                         return instance->tell_callback(absolute_byte_offset);
 369                 }
 370
 371                 ::FLAC__StreamEncoderWriteStatus SeekableStream::write_callback_(const ::FLAC__SeekableStreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
 372                 {
 373                         (void)encoder;
 374                         FLAC__ASSERT(0 != client_data);
 375                         SeekableStream *instance = reinterpret_cast<SeekableStream *>(client_data);
 376                         FLAC__ASSERT(0 != instance);
 377                         return instance->write_callback(buffer, bytes, samples, current_frame);
 378                 }
 379
 380         }
 381 }