src/flac/encode.c

   1 /* flac - Command-line FLAC encoder/decoder
   2  * Copyright (C) 2000,2001,2002,2003,2004  Josh Coalson
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License
   6  * as published by the Free Software Foundation; either version 2
   7  * of the License, or (at your option) any later version.
   8  *
   9  * This program 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
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  17  */
  18
  19 #if defined _WIN32 && !defined __CYGWIN__
  20 /* where MSVC puts unlink() */
  21 # include <io.h>
  22 #else
  23 # include <unistd.h>
  24 #endif
  25 #include <limits.h> /* for LONG_MAX */
  26 #include <math.h> /* for floor() */
  27 #include <stdarg.h>
  28 #include <stdio.h> /* for FILE etc. */
  29 #include <stdlib.h> /* for malloc */
  30 #include <string.h> /* for strcmp() */
  31 #include "FLAC/all.h"
  32 #include "share/grabbag.h"
  33 #include "encode.h"
  34
  35 #ifdef HAVE_CONFIG_H
  36 #include <config.h>
  37 #endif
  38
  39 #ifdef FLAC__HAS_OGG
  40 #include "OggFLAC/stream_encoder.h"
  41 #include "OggFLAC/file_encoder.h"
  42 #endif
  43
  44 #ifdef min
  45 #undef min
  46 #endif
  47 #define min(x,y) ((x)<(y)?(x):(y))
  48 #ifdef max
  49 #undef max
  50 #endif
  51 #define max(x,y) ((x)>(y)?(x):(y))
  52
  53 /* this MUST be >= 588 so that sector aligning can take place with one read */
  54 #define CHUNK_OF_SAMPLES 2048
  55
  56 typedef struct {
  57 #ifdef FLAC__HAS_OGG
  58         FLAC__bool use_ogg;
  59 #endif
  60         FLAC__bool verify;
  61         FLAC__bool verbose;
  62         FLAC__bool is_stdout;
  63         const char *inbasefilename;
  64         const char *outfilename;
  65
  66         FLAC__uint64 skip;
  67         FLAC__uint64 until; /* a value of 0 mean end-of-stream (i.e. --until=-0) */
  68         FLAC__bool replay_gain;
  69         unsigned channels;
  70         unsigned bits_per_sample;
  71         unsigned sample_rate;
  72         FLAC__uint64 unencoded_size;
  73         FLAC__uint64 total_samples_to_encode;
  74         FLAC__uint64 bytes_written;
  75         FLAC__uint64 samples_written;
  76         unsigned blocksize;
  77         unsigned stats_mask;
  78
  79         /*
  80          * We use *.stream for encoding to stdout
  81          * We use *.file for encoding to a regular file
  82          */
  83         union {
  84                 union {
  85                         FLAC__StreamEncoder *stream;
  86                         FLAC__FileEncoder *file;
  87                 } flac;
  88 #ifdef FLAC__HAS_OGG
  89                 union {
  90                         OggFLAC__StreamEncoder *stream;
  91                         OggFLAC__FileEncoder *file;
  92                 } ogg;
  93 #endif
  94         } encoder;
  95
  96         FILE *fin;
  97         FILE *fout;
  98         FLAC__StreamMetadata *seek_table_template;
  99 } EncoderSession;
 100
 101
 102 static FLAC__bool is_big_endian_host_;
 103
 104 static unsigned char ucbuffer_[CHUNK_OF_SAMPLES*FLAC__MAX_CHANNELS*((FLAC__REFERENCE_CODEC_MAX_BITS_PER_SAMPLE+7)/8)];
 105 static signed char *scbuffer_ = (signed char *)ucbuffer_;
 106 static FLAC__uint16 *usbuffer_ = (FLAC__uint16 *)ucbuffer_;
 107 static FLAC__int16 *ssbuffer_ = (FLAC__int16 *)ucbuffer_;
 108
 109 static FLAC__int32 in_[FLAC__MAX_CHANNELS][CHUNK_OF_SAMPLES];
 110 static FLAC__int32 *input_[FLAC__MAX_CHANNELS];
 111
 112
 113 /*
 114  * unpublished debug routines from the FLAC libs
 115  */
 116 extern FLAC__bool FLAC__stream_encoder_disable_constant_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value);
 117 extern FLAC__bool FLAC__stream_encoder_disable_fixed_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value);
 118 extern FLAC__bool FLAC__stream_encoder_disable_verbatim_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value);
 119 extern FLAC__bool FLAC__file_encoder_disable_constant_subframes(FLAC__FileEncoder *encoder, FLAC__bool value);
 120 extern FLAC__bool FLAC__file_encoder_disable_fixed_subframes(FLAC__FileEncoder *encoder, FLAC__bool value);
 121 extern FLAC__bool FLAC__file_encoder_disable_verbatim_subframes(FLAC__FileEncoder *encoder, FLAC__bool value);
 122 #ifdef FLAC__HAS_OGG
 123 extern FLAC__bool OggFLAC__stream_encoder_disable_constant_subframes(OggFLAC__StreamEncoder *encoder, FLAC__bool value);
 124 extern FLAC__bool OggFLAC__stream_encoder_disable_fixed_subframes(OggFLAC__StreamEncoder *encoder, FLAC__bool value);
 125 extern FLAC__bool OggFLAC__stream_encoder_disable_verbatim_subframes(OggFLAC__StreamEncoder *encoder, FLAC__bool value);
 126 extern FLAC__bool OggFLAC__file_encoder_disable_constant_subframes(OggFLAC__FileEncoder *encoder, FLAC__bool value);
 127 extern FLAC__bool OggFLAC__file_encoder_disable_fixed_subframes(OggFLAC__FileEncoder *encoder, FLAC__bool value);
 128 extern FLAC__bool OggFLAC__file_encoder_disable_verbatim_subframes(OggFLAC__FileEncoder *encoder, FLAC__bool value);
 129 #endif
 130
 131 /*
 132  * local routines
 133  */
 134 static FLAC__bool EncoderSession_construct(EncoderSession *e, FLAC__bool use_ogg, FLAC__bool verify, FLAC__bool verbose, FILE *infile, const char *infilename, const char *outfilename);
 135 static void EncoderSession_destroy(EncoderSession *e);
 136 static int EncoderSession_finish_ok(EncoderSession *e, int info_align_carry, int info_align_zero);
 137 static int EncoderSession_finish_error(EncoderSession *e);
 138 static FLAC__bool EncoderSession_init_encoder(EncoderSession *e, encode_options_t options, unsigned channels, unsigned bps, unsigned sample_rate);
 139 static FLAC__bool EncoderSession_process(EncoderSession *e, const FLAC__int32 * const buffer[], unsigned samples);
 140 static FLAC__bool convert_to_seek_table_template(const char *requested_seek_points, int num_requested_seek_points, FLAC__StreamMetadata *cuesheet, EncoderSession *e);
 141 static FLAC__bool canonicalize_until_specification(utils__SkipUntilSpecification *spec, const char *inbasefilename, unsigned sample_rate, FLAC__uint64 skip, FLAC__uint64 total_samples_in_input);
 142 static void format_input(FLAC__int32 *dest[], unsigned wide_samples, FLAC__bool is_big_endian, FLAC__bool is_unsigned_samples, unsigned channels, unsigned bps);
 143 #ifdef FLAC__HAS_OGG
 144 static FLAC__StreamEncoderWriteStatus ogg_stream_encoder_write_callback(const OggFLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data);
 145 static void ogg_stream_encoder_metadata_callback(const OggFLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data);
 146 static void ogg_file_encoder_progress_callback(const OggFLAC__FileEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data);
 147 #endif
 148 static FLAC__StreamEncoderWriteStatus flac_stream_encoder_write_callback(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data);
 149 static void flac_stream_encoder_metadata_callback(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data);
 150 static void flac_file_encoder_progress_callback(const FLAC__FileEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data);
 151 static FLAC__bool parse_cuesheet_(FLAC__StreamMetadata **cuesheet, const char *cuesheet_filename, const char *inbasefilename, FLAC__uint64 lead_out_offset);
 152 static void print_stats(const EncoderSession *encoder_session);
 153 static void print_error_with_state(const EncoderSession *e, const char *message);
 154 static void print_verify_error(EncoderSession *e);
 155 static FLAC__bool read_little_endian_uint16(FILE *f, FLAC__uint16 *val, FLAC__bool eof_ok, const char *fn);
 156 static FLAC__bool read_little_endian_uint32(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn);
 157 static FLAC__bool read_big_endian_uint16(FILE *f, FLAC__uint16 *val, FLAC__bool eof_ok, const char *fn);
 158 static FLAC__bool read_big_endian_uint32(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn);
 159 static FLAC__bool read_sane_extended(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn);
 160
 161 /*
 162  * public routines
 163  */
 164 int
 165 flac__encode_aif(FILE *infile, long infilesize, const char *infilename, const char *outfilename,
 166         const FLAC__byte *lookahead, unsigned lookahead_length, wav_encode_options_t options)
 167 {
 168         EncoderSession encoder_session;
 169         FLAC__uint16 x;
 170         FLAC__uint32 xx;
 171         unsigned int channels= 0U, bps= 0U, sample_rate= 0U, sample_frames= 0U;
 172         FLAC__bool got_comm_chunk= false, got_ssnd_chunk= false;
 173         int info_align_carry= -1, info_align_zero= -1;
 174
 175         (void)infilesize; /* silence compiler warning about unused parameter */
 176         (void)lookahead; /* silence compiler warning about unused parameter */
 177         (void)lookahead_length; /* silence compiler warning about unused parameter */
 178
 179         if(!
 180                 EncoderSession_construct(
 181                         &encoder_session,
 182 #ifdef FLAC__HAS_OGG
 183                         options.common.use_ogg,
 184 #else
 185                         /*use_ogg=*/false,
 186 #endif
 187                         options.common.verify,
 188                         options.common.verbose,
 189                         infile,
 190                         infilename,
 191                         outfilename
 192                 )
 193         )
 194                 return 1;
 195
 196         /* lookahead[] already has "FORMxxxxAIFF", do sub-chunks */
 197
 198         while(1) {
 199                 size_t c= 0U;
 200                 char chunk_id[4];
 201
 202                 /* chunk identifier; really conservative about behavior of fread() and feof() */
 203                 if(feof(infile) || ((c= fread(chunk_id, 1U, 4U, infile)), c==0U && feof(infile)))
 204                         break;
 205                 else if(c<4U || feof(infile)) {
 206                         fprintf(stderr, "%s: ERROR: incomplete chunk identifier\n", encoder_session.inbasefilename);
 207                         return EncoderSession_finish_error(&encoder_session);
 208                 }
 209
 210                 if(got_comm_chunk==false && !strncmp(chunk_id, "COMM", 4)) { /* common chunk */
 211                         unsigned long skip;
 212
 213                         /* COMM chunk size */
 214                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 215                                 return EncoderSession_finish_error(&encoder_session);
 216                         else if(xx<18U) {
 217                                 fprintf(stderr, "%s: ERROR: non-standard 'COMM' chunk has length = %u\n", encoder_session.inbasefilename, (unsigned int)xx);
 218                                 return EncoderSession_finish_error(&encoder_session);
 219                         }
 220                         else if(xx!=18U)
 221                                 fprintf(stderr, "%s: WARNING: non-standard 'COMM' chunk has length = %u\n", encoder_session.inbasefilename, (unsigned int)xx);
 222                         skip= (xx-18U)+(xx & 1U);
 223
 224                         /* number of channels */
 225                         if(!read_big_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 226                                 return EncoderSession_finish_error(&encoder_session);
 227                         else if(x==0U || x>FLAC__MAX_CHANNELS) {
 228                                 fprintf(stderr, "%s: ERROR: unsupported number channels %u\n", encoder_session.inbasefilename, (unsigned int)x);
 229                                 return EncoderSession_finish_error(&encoder_session);
 230                         }
 231                         else if(options.common.sector_align && x!=2U) {
 232                                 fprintf(stderr, "%s: ERROR: file has %u channels, must be 2 for --sector-align\n", encoder_session.inbasefilename, (unsigned int)x);
 233                                 return EncoderSession_finish_error(&encoder_session);
 234                         }
 235                         channels= x;
 236
 237                         /* number of sample frames */
 238                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 239                                 return EncoderSession_finish_error(&encoder_session);
 240                         sample_frames= xx;
 241
 242                         /* bits per sample */
 243                         if(!read_big_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 244                                 return EncoderSession_finish_error(&encoder_session);
 245                         else if(x!=8U && x!=16U && x!=24U) {
 246                                 fprintf(stderr, "%s: ERROR: unsupported bits per sample %u\n", encoder_session.inbasefilename, (unsigned int)x);
 247                                 return EncoderSession_finish_error(&encoder_session);
 248                         }
 249                         else if(options.common.sector_align && x!=16U) {
 250                                 fprintf(stderr, "%s: ERROR: file has %u bits per sample, must be 16 for --sector-align\n", encoder_session.inbasefilename, (unsigned int)x);
 251                                 return EncoderSession_finish_error(&encoder_session);
 252                         }
 253                         bps= x;
 254
 255                         /* sample rate */
 256                         if(!read_sane_extended(infile, &xx, false, encoder_session.inbasefilename))
 257                                 return EncoderSession_finish_error(&encoder_session);
 258                         else if(!FLAC__format_sample_rate_is_valid(xx)) {
 259                                 fprintf(stderr, "%s: ERROR: unsupported sample rate %u\n", encoder_session.inbasefilename, (unsigned int)xx);
 260                                 return EncoderSession_finish_error(&encoder_session);
 261                         }
 262                         else if(options.common.sector_align && xx!=44100U) {
 263                                 fprintf(stderr, "%s: ERROR: file's sample rate is %u, must be 44100 for --sector-align\n", encoder_session.inbasefilename, (unsigned int)xx);
 264                                 return EncoderSession_finish_error(&encoder_session);
 265                         }
 266                         sample_rate= xx;
 267
 268                         /* skip any extra data in the COMM chunk */
 269                         FLAC__ASSERT(skip<=LONG_MAX);
 270                         while(skip>0U && fseek(infile, skip, SEEK_CUR)<0) {
 271                                 unsigned int need= min(skip, sizeof ucbuffer_);
 272                                 if(fread(ucbuffer_, 1U, need, infile)<need) {
 273                                         fprintf(stderr, "%s: ERROR during read while skipping extra COMM data\n", encoder_session.inbasefilename);
 274                                         return EncoderSession_finish_error(&encoder_session);
 275                                 }
 276                                 skip-= need;
 277                         }
 278
 279                         /*
 280                          * now that we know the sample rate, canonicalize the
 281                          * --skip string to a number of samples:
 282                          */
 283                         flac__utils_canonicalize_skip_until_specification(&options.common.skip_specification, sample_rate);
 284                         FLAC__ASSERT(options.common.skip_specification.value.samples >= 0);
 285                         encoder_session.skip = (FLAC__uint64)options.common.skip_specification.value.samples;
 286                         FLAC__ASSERT(!options.common.sector_align || encoder_session.skip == 0);
 287
 288                         got_comm_chunk= true;
 289                 }
 290                 else if(got_ssnd_chunk==false && !strncmp(chunk_id, "SSND", 4)) { /* sound data chunk */
 291                         unsigned int offset= 0U, block_size= 0U, align_remainder= 0U, data_bytes;
 292                         size_t bytes_per_frame= channels*(bps>>3);
 293                         FLAC__uint64 total_samples_in_input, trim = 0;
 294                         FLAC__bool pad= false;
 295
 296                         if(got_comm_chunk==false) {
 297                                 fprintf(stderr, "%s: ERROR: got 'SSND' chunk before 'COMM' chunk\n", encoder_session.inbasefilename);
 298                                 return EncoderSession_finish_error(&encoder_session);
 299                         }
 300
 301                         /* SSND chunk size */
 302                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 303                                 return EncoderSession_finish_error(&encoder_session);
 304                         else if(xx!=(sample_frames*bytes_per_frame + 8U)) {
 305                                 fprintf(stderr, "%s: ERROR: SSND chunk size inconsistent with sample frame count\n", encoder_session.inbasefilename);
 306                                 return EncoderSession_finish_error(&encoder_session);
 307                         }
 308                         data_bytes= xx;
 309                         pad= (data_bytes & 1U) ? true : false;
 310                         data_bytes-= 8U; /* discount the offset and block size fields */
 311
 312                         /* offset */
 313                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 314                                 return EncoderSession_finish_error(&encoder_session);
 315                         else if(xx!=0U) {
 316                                 fprintf(stderr, "%s: ERROR: offset is %u; must be 0\n", encoder_session.inbasefilename, (unsigned int)xx);
 317                                 return EncoderSession_finish_error(&encoder_session);
 318                         }
 319                         offset= xx;
 320
 321                         /* block size */
 322                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 323                                 return EncoderSession_finish_error(&encoder_session);
 324                         else if(xx!=0U) {
 325                                 fprintf(stderr, "%s: ERROR: block size is %u; must be 0\n", encoder_session.inbasefilename, (unsigned int)xx);
 326                                 return EncoderSession_finish_error(&encoder_session);
 327                         }
 328                         block_size= xx;
 329
 330                         /* *options.common.align_reservoir_samples will be 0 unless --sector-align is used */
 331                         FLAC__ASSERT(options.common.sector_align || *options.common.align_reservoir_samples == 0);
 332                         total_samples_in_input = data_bytes / bytes_per_frame + *options.common.align_reservoir_samples;
 333
 334                         /*
 335                          * now that we know the input size, canonicalize the
 336                          * --until string to an absolute sample number:
 337                          */
 338                         if(!canonicalize_until_specification(&options.common.until_specification, encoder_session.inbasefilename, sample_rate, encoder_session.skip, total_samples_in_input))
 339                                 return EncoderSession_finish_error(&encoder_session);
 340                         encoder_session.until = (FLAC__uint64)options.common.until_specification.value.samples;
 341                         FLAC__ASSERT(!options.common.sector_align || encoder_session.until == 0);
 342
 343                         if(encoder_session.skip>0U) {
 344                                 FLAC__uint64 remaining= encoder_session.skip*bytes_per_frame;
 345
 346                                 /* do 1<<30 bytes at a time, since 1<<30 is a nice round number, and */
 347                                 /* is guaranteed to be less than LONG_MAX */
 348                                 while(remaining>0U)
 349                                 {
 350                                         unsigned long skip= (unsigned long)(remaining % (1U<<30));
 351
 352                                         FLAC__ASSERT(skip<=LONG_MAX);
 353                                         while(skip>0 && fseek(infile, skip, SEEK_CUR)<0) {
 354                                                 unsigned int need= min(skip, sizeof ucbuffer_);
 355                                                 if(fread(ucbuffer_, 1U, need, infile)<need) {
 356                                                         fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 357                                                         return EncoderSession_finish_error(&encoder_session);
 358                                                 }
 359                                                 skip-= need;
 360                                         }
 361
 362                                         remaining-= skip;
 363                                 }
 364                         }
 365
 366                         data_bytes-= (unsigned int)encoder_session.skip*bytes_per_frame; /*@@@ WATCHOUT: 4GB limit */
 367                         encoder_session.total_samples_to_encode= total_samples_in_input - encoder_session.skip;
 368                         if(encoder_session.until > 0) {
 369                                 trim = total_samples_in_input - encoder_session.until;
 370                                 FLAC__ASSERT(total_samples_in_input > 0);
 371                                 FLAC__ASSERT(!options.common.sector_align);
 372                                 data_bytes-= (unsigned int)trim*bytes_per_frame;
 373                                 encoder_session.total_samples_to_encode-= trim;
 374                         }
 375                         if(options.common.sector_align) {
 376                                 align_remainder= (unsigned int)(encoder_session.total_samples_to_encode % 588U);
 377                                 if(options.common.is_last_file)
 378                                         encoder_session.total_samples_to_encode+= (588U-align_remainder); /* will pad with zeroes */
 379                                 else
 380                                         encoder_session.total_samples_to_encode-= align_remainder; /* will stop short and carry over to next file */
 381                         }
 382
 383                         /* +54 for the size of the AIFF headers; this is just an estimate for the progress indicator and doesn't need to be exact */
 384                         encoder_session.unencoded_size= encoder_session.total_samples_to_encode*bytes_per_frame+54;
 385
 386                         if(!EncoderSession_init_encoder(&encoder_session, options.common, channels, bps, sample_rate))
 387                                 return EncoderSession_finish_error(&encoder_session);
 388
 389                         /* first do any samples in the reservoir */
 390                         if(options.common.sector_align && *options.common.align_reservoir_samples>0U) {
 391
 392                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 *const *)options.common.align_reservoir, *options.common.align_reservoir_samples)) {
 393                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 394                                         return EncoderSession_finish_error(&encoder_session);
 395                                 }
 396                         }
 397
 398                         /* decrement the data_bytes counter if we need to align the file */
 399                         if(options.common.sector_align) {
 400                                 if(options.common.is_last_file)
 401                                         *options.common.align_reservoir_samples= 0U;
 402                                 else {
 403                                         *options.common.align_reservoir_samples= align_remainder;
 404                                         data_bytes-= (*options.common.align_reservoir_samples)*bytes_per_frame;
 405                                 }
 406                         }
 407
 408                         /* now do from the file */
 409                         while(data_bytes>0) {
 410                                 size_t bytes_read= fread(ucbuffer_, 1U, min(data_bytes, CHUNK_OF_SAMPLES*bytes_per_frame), infile);
 411
 412                                 if(bytes_read==0U) {
 413                                         if(ferror(infile)) {
 414                                                 fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
 415                                                 return EncoderSession_finish_error(&encoder_session);
 416                                         }
 417                                         else if(feof(infile)) {
 418                                                 fprintf(stderr, "%s: WARNING: unexpected EOF; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned int)encoder_session.total_samples_to_encode, (unsigned int)encoder_session.samples_written);
 419                                                 data_bytes= 0;
 420                                         }
 421                                 }
 422                                 else {
 423                                         if(bytes_read % bytes_per_frame != 0U) {
 424                                                 fprintf(stderr, "%s: ERROR: got partial sample\n", encoder_session.inbasefilename);
 425                                                 return EncoderSession_finish_error(&encoder_session);
 426                                         }
 427                                         else {
 428                                                 unsigned int frames= bytes_read/bytes_per_frame;
 429                                                 format_input(input_, frames, true, false, channels, bps);
 430
 431                                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 *const *)input_, frames)) {
 432                                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 433                                                         return EncoderSession_finish_error(&encoder_session);
 434                                                 }
 435                                                 else
 436                                                         data_bytes-= bytes_read;
 437                                         }
 438                                 }
 439                         }
 440
 441                         if(trim>0) {
 442                                 FLAC__uint64 remaining= (unsigned int)trim*bytes_per_frame;
 443
 444                                 FLAC__ASSERT(!options.common.sector_align);
 445
 446                                 /* do 1<<30 bytes at a time, since 1<<30 is a nice round number, and */
 447                                 /* is guaranteed to be less than LONG_MAX */
 448                                 while(remaining>0U)
 449                                 {
 450                                         unsigned long skip= (unsigned long)(remaining % (1U<<30));
 451
 452                                         FLAC__ASSERT(skip<=LONG_MAX);
 453                                         while(skip>0 && fseek(infile, skip, SEEK_CUR)<0) {
 454                                                 unsigned int need= min(skip, sizeof ucbuffer_);
 455                                                 if(fread(ucbuffer_, 1U, need, infile)<need) {
 456                                                         fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 457                                                         return EncoderSession_finish_error(&encoder_session);
 458                                                 }
 459                                                 skip-= need;
 460                                         }
 461
 462                                         remaining-= skip;
 463                                 }
 464                         }
 465
 466                         /* now read unaligned samples into reservoir or pad with zeroes if necessary */
 467                         if(options.common.sector_align) {
 468                                 if(options.common.is_last_file) {
 469                                         unsigned int pad_frames= 588U-align_remainder;
 470
 471                                         if(pad_frames<588U) {
 472                                                 unsigned int i;
 473
 474                                                 info_align_zero= pad_frames;
 475                                                 for(i= 0U; i<channels; ++i)
 476                                                         memset(input_[i], 0, pad_frames*(bps>>3));
 477
 478                                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 *const *)input_, pad_frames)) {
 479                                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 480                                                         return EncoderSession_finish_error(&encoder_session);
 481                                                 }
 482                                         }
 483                                 }
 484                                 else {
 485                                         if(*options.common.align_reservoir_samples > 0) {
 486                                                 size_t bytes_read= fread(ucbuffer_, 1U, (*options.common.align_reservoir_samples)*bytes_per_frame, infile);
 487
 488                                                 FLAC__ASSERT(CHUNK_OF_SAMPLES>=588U);
 489                                                 if(bytes_read==0U && ferror(infile)) {
 490                                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
 491                                                         return EncoderSession_finish_error(&encoder_session);
 492                                                 }
 493                                                 else if(bytes_read != (*options.common.align_reservoir_samples) * bytes_per_frame)
 494                                                         fprintf(stderr, "%s: WARNING: unexpected EOF; read %u bytes; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned int)bytes_read, (unsigned int)encoder_session.total_samples_to_encode, (unsigned int)encoder_session.samples_written);
 495                                                 else {
 496                                                         info_align_carry= *options.common.align_reservoir_samples;
 497                                                         format_input(options.common.align_reservoir, *options.common.align_reservoir_samples, true, false, channels, bps);
 498                                                 }
 499                                         }
 500                                 }
 501                         }
 502
 503                         if(pad==true) {
 504                                 unsigned char tmp;
 505
 506                                 if(fread(&tmp, 1U, 1U, infile)<1U) {
 507                                         fprintf(stderr, "%s: ERROR during read of SSND pad byte\n", encoder_session.inbasefilename);
 508                                         return EncoderSession_finish_error(&encoder_session);
 509                                 }
 510                         }
 511
 512                         got_ssnd_chunk= true;
 513                 }
 514                 else { /* other chunk */
 515                         if(!strncmp(chunk_id, "COMM", 4))
 516                                 fprintf(stderr, "%s: WARNING: skipping extra 'COMM' chunk\n", encoder_session.inbasefilename);
 517                         else if(!strncmp(chunk_id, "SSND", 4))
 518                                 fprintf(stderr, "%s: WARNING: skipping extra 'SSND' chunk\n", encoder_session.inbasefilename);
 519                         else
 520                                 fprintf(stderr, "%s: WARNING: skipping unknown chunk '%s'\n", encoder_session.inbasefilename, chunk_id);
 521
 522                         /* chunk size */
 523                         if(!read_big_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 524                                 return EncoderSession_finish_error(&encoder_session);
 525                         else {
 526                                 unsigned long skip= xx+(xx & 1U);
 527
 528                                 FLAC__ASSERT(skip<=LONG_MAX);
 529                                 while(skip>0U && fseek(infile, skip, SEEK_CUR)<0) {
 530                                         unsigned int need= min(skip, sizeof ucbuffer_);
 531                                         if(fread(ucbuffer_, 1U, need, infile)<need) {
 532                                                 fprintf(stderr, "%s: ERROR during read while skipping unknown chunk\n", encoder_session.inbasefilename);
 533                                                 return EncoderSession_finish_error(&encoder_session);
 534                                         }
 535                                         skip-= need;
 536                                 }
 537                         }
 538                 }
 539         }
 540
 541         if(got_ssnd_chunk==false && sample_frames!=0U) {
 542                 fprintf(stderr, "%s: ERROR: missing SSND chunk\n", encoder_session.inbasefilename);
 543                 return EncoderSession_finish_error(&encoder_session);
 544         }
 545
 546         return EncoderSession_finish_ok(&encoder_session, info_align_carry, info_align_zero);
 547 }
 548
 549 int flac__encode_wav(FILE *infile, long infilesize, const char *infilename, const char *outfilename, const FLAC__byte *lookahead, unsigned lookahead_length, wav_encode_options_t options)
 550 {
 551         EncoderSession encoder_session;
 552         FLAC__bool is_unsigned_samples = false;
 553         unsigned channels = 0, bps = 0, sample_rate = 0, data_bytes;
 554         size_t bytes_per_wide_sample, bytes_read;
 555         FLAC__uint16 x;
 556         FLAC__uint32 xx;
 557         FLAC__bool got_fmt_chunk = false, got_data_chunk = false;
 558         unsigned align_remainder = 0;
 559         int info_align_carry = -1, info_align_zero = -1;
 560
 561         (void)infilesize;
 562         (void)lookahead;
 563         (void)lookahead_length;
 564
 565         if(!
 566                 EncoderSession_construct(
 567                         &encoder_session,
 568 #ifdef FLAC__HAS_OGG
 569                         options.common.use_ogg,
 570 #else
 571                         /*use_ogg=*/false,
 572 #endif
 573                         options.common.verify,
 574                         options.common.verbose,
 575                         infile,
 576                         infilename,
 577                         outfilename
 578                 )
 579         )
 580                 return 1;
 581
 582         /*
 583          * lookahead[] already has "RIFFxxxxWAVE", do sub-chunks
 584          */
 585         while(!feof(infile)) {
 586                 if(!read_little_endian_uint32(infile, &xx, true, encoder_session.inbasefilename))
 587                         return EncoderSession_finish_error(&encoder_session);
 588                 if(feof(infile))
 589                         break;
 590                 if(xx == 0x20746d66 && !got_fmt_chunk) { /* "fmt " */
 591                         unsigned block_align;
 592
 593                         /* fmt sub-chunk size */
 594                         if(!read_little_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 595                                 return EncoderSession_finish_error(&encoder_session);
 596                         if(xx < 16) {
 597                                 fprintf(stderr, "%s: ERROR: found non-standard 'fmt ' sub-chunk which has length = %u\n", encoder_session.inbasefilename, (unsigned)xx);
 598                                 return EncoderSession_finish_error(&encoder_session);
 599                         }
 600                         else if(xx != 16 && xx != 18) {
 601                                 fprintf(stderr, "%s: WARNING: found non-standard 'fmt ' sub-chunk which has length = %u\n", encoder_session.inbasefilename, (unsigned)xx);
 602                         }
 603                         data_bytes = xx;
 604                         /* compression code */
 605                         if(!read_little_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 606                                 return EncoderSession_finish_error(&encoder_session);
 607                         if(x != 1) {
 608                                 fprintf(stderr, "%s: ERROR: unsupported compression type %u\n", encoder_session.inbasefilename, (unsigned)x);
 609                                 return EncoderSession_finish_error(&encoder_session);
 610                         }
 611                         /* number of channels */
 612                         if(!read_little_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 613                                 return EncoderSession_finish_error(&encoder_session);
 614                         if(x == 0 || x > FLAC__MAX_CHANNELS) {
 615                                 fprintf(stderr, "%s: ERROR: unsupported number channels %u\n", encoder_session.inbasefilename, (unsigned)x);
 616                                 return EncoderSession_finish_error(&encoder_session);
 617                         }
 618                         else if(options.common.sector_align && x != 2) {
 619                                 fprintf(stderr, "%s: ERROR: file has %u channels, must be 2 for --sector-align\n", encoder_session.inbasefilename, (unsigned)x);
 620                                 return EncoderSession_finish_error(&encoder_session);
 621                         }
 622                         channels = x;
 623                         /* sample rate */
 624                         if(!read_little_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 625                                 return EncoderSession_finish_error(&encoder_session);
 626                         if(!FLAC__format_sample_rate_is_valid(xx)) {
 627                                 fprintf(stderr, "%s: ERROR: unsupported sample rate %u\n", encoder_session.inbasefilename, (unsigned)xx);
 628                                 return EncoderSession_finish_error(&encoder_session);
 629                         }
 630                         else if(options.common.sector_align && xx != 44100) {
 631                                 fprintf(stderr, "%s: ERROR: file's sample rate is %u, must be 44100 for --sector-align\n", encoder_session.inbasefilename, (unsigned)xx);
 632                                 return EncoderSession_finish_error(&encoder_session);
 633                         }
 634                         sample_rate = xx;
 635                         /* avg bytes per second (ignored) */
 636                         if(!read_little_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 637                                 return EncoderSession_finish_error(&encoder_session);
 638                         /* block align */
 639                         if(!read_little_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 640                                 return EncoderSession_finish_error(&encoder_session);
 641                         block_align = x;
 642                         /* bits per sample */
 643                         if(!read_little_endian_uint16(infile, &x, false, encoder_session.inbasefilename))
 644                                 return EncoderSession_finish_error(&encoder_session);
 645                         if(x != 8 && x != 16 && x != 24) {
 646                                 fprintf(stderr, "%s: ERROR: unsupported bits-per-sample %u\n", encoder_session.inbasefilename, (unsigned)x);
 647                                 return EncoderSession_finish_error(&encoder_session);
 648                         }
 649                         else if(options.common.sector_align && x != 16) {
 650                                 fprintf(stderr, "%s: ERROR: file has %u bits per sample, must be 16 for --sector-align\n", encoder_session.inbasefilename, (unsigned)x);
 651                                 return EncoderSession_finish_error(&encoder_session);
 652                         }
 653                         bps = x;
 654                         if(bps * channels != block_align * 8) {
 655                                 fprintf(stderr, "%s: ERROR: unsupported block alignment (%u), for bits-per-sample=%u, channels=%u\n", encoder_session.inbasefilename, block_align, bps, channels);
 656                                 return EncoderSession_finish_error(&encoder_session);
 657                         }
 658                         is_unsigned_samples = (x == 8);
 659
 660                         /* skip any extra data in the fmt sub-chunk */
 661                         data_bytes -= 16;
 662                         if(data_bytes > 0) {
 663                                 unsigned left, need;
 664                                 for(left = data_bytes; left > 0; ) {
 665                                         need = min(left, CHUNK_OF_SAMPLES);
 666                                         if(fread(ucbuffer_, 1U, need, infile) < need) {
 667                                                 fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 668                                                 return EncoderSession_finish_error(&encoder_session);
 669                                         }
 670                                         left -= need;
 671                                 }
 672                         }
 673
 674                         /*
 675                          * now that we know the sample rate, canonicalize the
 676                          * --skip string to a number of samples:
 677                          */
 678                         flac__utils_canonicalize_skip_until_specification(&options.common.skip_specification, sample_rate);
 679                         FLAC__ASSERT(options.common.skip_specification.value.samples >= 0);
 680                         encoder_session.skip = (FLAC__uint64)options.common.skip_specification.value.samples;
 681                         FLAC__ASSERT(!options.common.sector_align || encoder_session.skip == 0);
 682
 683                         got_fmt_chunk = true;
 684                 }
 685                 else if(xx == 0x61746164 && !got_data_chunk && got_fmt_chunk) { /* "data" */
 686                         FLAC__uint64 total_samples_in_input, trim = 0;
 687
 688                         /* data size */
 689                         if(!read_little_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 690                                 return EncoderSession_finish_error(&encoder_session);
 691                         data_bytes = xx;
 692
 693                         bytes_per_wide_sample = channels * (bps >> 3);
 694
 695                         /* *options.common.align_reservoir_samples will be 0 unless --sector-align is used */
 696                         FLAC__ASSERT(options.common.sector_align || *options.common.align_reservoir_samples == 0);
 697                         total_samples_in_input = data_bytes / bytes_per_wide_sample + *options.common.align_reservoir_samples;
 698
 699                         /*
 700                          * now that we know the input size, canonicalize the
 701                          * --until string to an absolute sample number:
 702                          */
 703                         if(!canonicalize_until_specification(&options.common.until_specification, encoder_session.inbasefilename, sample_rate, encoder_session.skip, total_samples_in_input))
 704                                 return EncoderSession_finish_error(&encoder_session);
 705                         encoder_session.until = (FLAC__uint64)options.common.until_specification.value.samples;
 706                         FLAC__ASSERT(!options.common.sector_align || encoder_session.until == 0);
 707
 708                         if(encoder_session.skip > 0) {
 709                                 if(fseek(infile, bytes_per_wide_sample * (unsigned)encoder_session.skip, SEEK_CUR) < 0) {
 710                                         /* can't seek input, read ahead manually... */
 711                                         unsigned left, need;
 712                                         for(left = (unsigned)encoder_session.skip; left > 0; ) { /*@@@ WATCHOUT: 4GB limit */
 713                                                 need = min(left, CHUNK_OF_SAMPLES);
 714                                                 if(fread(ucbuffer_, bytes_per_wide_sample, need, infile) < need) {
 715                                                         fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 716                                                         return EncoderSession_finish_error(&encoder_session);
 717                                                 }
 718                                                 left -= need;
 719                                         }
 720                                 }
 721                         }
 722
 723                         data_bytes -= (unsigned)encoder_session.skip * bytes_per_wide_sample; /*@@@ WATCHOUT: 4GB limit */
 724                         encoder_session.total_samples_to_encode = total_samples_in_input - encoder_session.skip;
 725                         if(encoder_session.until > 0) {
 726                                 trim = total_samples_in_input - encoder_session.until;
 727                                 FLAC__ASSERT(total_samples_in_input > 0);
 728                                 FLAC__ASSERT(!options.common.sector_align);
 729                                 data_bytes -= (unsigned int)trim * bytes_per_wide_sample;
 730                                 encoder_session.total_samples_to_encode -= trim;
 731                         }
 732                         if(options.common.sector_align) {
 733                                 align_remainder = (unsigned)(encoder_session.total_samples_to_encode % 588);
 734                                 if(options.common.is_last_file)
 735                                         encoder_session.total_samples_to_encode += (588-align_remainder); /* will pad with zeroes */
 736                                 else
 737                                         encoder_session.total_samples_to_encode -= align_remainder; /* will stop short and carry over to next file */
 738                         }
 739
 740                         /* +44 for the size of the WAV headers; this is just an estimate for the progress indicator and doesn't need to be exact */
 741                         encoder_session.unencoded_size = encoder_session.total_samples_to_encode * bytes_per_wide_sample + 44;
 742
 743                         if(!EncoderSession_init_encoder(&encoder_session, options.common, channels, bps, sample_rate))
 744                                 return EncoderSession_finish_error(&encoder_session);
 745
 746                         /*
 747                          * first do any samples in the reservoir
 748                          */
 749                         if(options.common.sector_align && *options.common.align_reservoir_samples > 0) {
 750                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)options.common.align_reservoir, *options.common.align_reservoir_samples)) {
 751                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 752                                         return EncoderSession_finish_error(&encoder_session);
 753                                 }
 754                         }
 755
 756                         /*
 757                          * decrement the data_bytes counter if we need to align the file
 758                          */
 759                         if(options.common.sector_align) {
 760                                 if(options.common.is_last_file) {
 761                                         *options.common.align_reservoir_samples = 0;
 762                                 }
 763                                 else {
 764                                         *options.common.align_reservoir_samples = align_remainder;
 765                                         data_bytes -= (*options.common.align_reservoir_samples) * bytes_per_wide_sample;
 766                                 }
 767                         }
 768
 769                         /*
 770                          * now do from the file
 771                          */
 772                         while(data_bytes > 0) {
 773                                 bytes_read = fread(ucbuffer_, sizeof(unsigned char), min(data_bytes, CHUNK_OF_SAMPLES * bytes_per_wide_sample), infile);
 774                                 if(bytes_read == 0) {
 775                                         if(ferror(infile)) {
 776                                                 fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
 777                                                 return EncoderSession_finish_error(&encoder_session);
 778                                         }
 779                                         else if(feof(infile)) {
 780                                                 fprintf(stderr, "%s: WARNING: unexpected EOF; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned)encoder_session.total_samples_to_encode, (unsigned)encoder_session.samples_written);
 781                                                 data_bytes = 0;
 782                                         }
 783                                 }
 784                                 else {
 785                                         if(bytes_read % bytes_per_wide_sample != 0) {
 786                                                 fprintf(stderr, "%s: ERROR: got partial sample\n", encoder_session.inbasefilename);
 787                                                 return EncoderSession_finish_error(&encoder_session);
 788                                         }
 789                                         else {
 790                                                 unsigned wide_samples = bytes_read / bytes_per_wide_sample;
 791                                                 format_input(input_, wide_samples, false, is_unsigned_samples, channels, bps);
 792
 793                                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)input_, wide_samples)) {
 794                                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 795                                                         return EncoderSession_finish_error(&encoder_session);
 796                                                 }
 797                                                 data_bytes -= bytes_read;
 798                                         }
 799                                 }
 800                         }
 801
 802                         if(trim > 0) {
 803                                 if(fseek(infile, bytes_per_wide_sample * (unsigned)trim, SEEK_CUR) < 0) {
 804                                         /* can't seek input, read ahead manually... */
 805                                         unsigned left, need;
 806                                         for(left = (unsigned)trim; left > 0; ) { /*@@@ WATCHOUT: 4GB limit */
 807                                                 need = min(left, CHUNK_OF_SAMPLES);
 808                                                 if(fread(ucbuffer_, bytes_per_wide_sample, need, infile) < need) {
 809                                                         fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 810                                                         return EncoderSession_finish_error(&encoder_session);
 811                                                 }
 812                                                 left -= need;
 813                                         }
 814                                 }
 815                         }
 816
 817                         /*
 818                          * now read unaligned samples into reservoir or pad with zeroes if necessary
 819                          */
 820                         if(options.common.sector_align) {
 821                                 if(options.common.is_last_file) {
 822                                         unsigned wide_samples = 588 - align_remainder;
 823                                         if(wide_samples < 588) {
 824                                                 unsigned channel;
 825
 826                                                 info_align_zero = wide_samples;
 827                                                 data_bytes = wide_samples * (bps >> 3);
 828                                                 for(channel = 0; channel < channels; channel++)
 829                                                         memset(input_[channel], 0, data_bytes);
 830
 831                                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)input_, wide_samples)) {
 832                                                         print_error_with_state(&encoder_session, "ERROR during encoding");
 833                                                         return EncoderSession_finish_error(&encoder_session);
 834                                                 }
 835                                         }
 836                                 }
 837                                 else {
 838                                         if(*options.common.align_reservoir_samples > 0) {
 839                                                 FLAC__ASSERT(CHUNK_OF_SAMPLES >= 588);
 840                                                 bytes_read = fread(ucbuffer_, sizeof(unsigned char), (*options.common.align_reservoir_samples) * bytes_per_wide_sample, infile);
 841                                                 if(bytes_read == 0 && ferror(infile)) {
 842                                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
 843                                                         return EncoderSession_finish_error(&encoder_session);
 844                                                 }
 845                                                 else if(bytes_read != (*options.common.align_reservoir_samples) * bytes_per_wide_sample) {
 846                                                         fprintf(stderr, "%s: WARNING: unexpected EOF; read %u bytes; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned)bytes_read, (unsigned)encoder_session.total_samples_to_encode, (unsigned)encoder_session.samples_written);
 847                                                         data_bytes = 0;
 848                                                 }
 849                                                 else {
 850                                                         info_align_carry = *options.common.align_reservoir_samples;
 851                                                         format_input(options.common.align_reservoir, *options.common.align_reservoir_samples, false, is_unsigned_samples, channels, bps);
 852                                                 }
 853                                         }
 854                                 }
 855                         }
 856
 857                         got_data_chunk = true;
 858                 }
 859                 else {
 860                         if(xx == 0x20746d66 && got_fmt_chunk) { /* "fmt " */
 861                                 fprintf(stderr, "%s: WARNING: skipping extra 'fmt ' sub-chunk\n", encoder_session.inbasefilename);
 862                         }
 863                         else if(xx == 0x61746164) { /* "data" */
 864                                 if(got_data_chunk) {
 865                                         fprintf(stderr, "%s: WARNING: skipping extra 'data' sub-chunk\n", encoder_session.inbasefilename);
 866                                 }
 867                                 else if(!got_fmt_chunk) {
 868                                         fprintf(stderr, "%s: ERROR: got 'data' sub-chunk before 'fmt' sub-chunk\n", encoder_session.inbasefilename);
 869                                         return EncoderSession_finish_error(&encoder_session);
 870                                 }
 871                                 else {
 872                                         FLAC__ASSERT(0);
 873                                 }
 874                         }
 875                         else {
 876                                 fprintf(stderr, "%s: WARNING: skipping unknown sub-chunk '%c%c%c%c'\n", encoder_session.inbasefilename, (char)(xx&255), (char)((xx>>8)&255), (char)((xx>>16)&255), (char)(xx>>24));
 877                         }
 878                         /* sub-chunk size */
 879                         if(!read_little_endian_uint32(infile, &xx, false, encoder_session.inbasefilename))
 880                                 return EncoderSession_finish_error(&encoder_session);
 881                         if(fseek(infile, xx, SEEK_CUR) < 0) {
 882                                 /* can't seek input, read ahead manually... */
 883                                 unsigned left, need;
 884                                 const unsigned chunk = sizeof(ucbuffer_);
 885                                 for(left = xx; left > 0; ) {
 886                                         need = min(left, chunk);
 887                                         if(fread(ucbuffer_, 1, need, infile) < need) {
 888                                                 fprintf(stderr, "%s: ERROR during read while skipping unsupported sub-chunk\n", encoder_session.inbasefilename);
 889                                                 return EncoderSession_finish_error(&encoder_session);
 890                                         }
 891                                         left -= need;
 892                                 }
 893                         }
 894                 }
 895         }
 896
 897         return EncoderSession_finish_ok(&encoder_session, info_align_carry, info_align_zero);
 898 }
 899
 900 int flac__encode_raw(FILE *infile, long infilesize, const char *infilename, const char *outfilename, const FLAC__byte *lookahead, unsigned lookahead_length, raw_encode_options_t options)
 901 {
 902         EncoderSession encoder_session;
 903         size_t bytes_read;
 904         const size_t bytes_per_wide_sample = options.channels * (options.bps >> 3);
 905         unsigned align_remainder = 0;
 906         int info_align_carry = -1, info_align_zero = -1;
 907         FLAC__uint64 total_samples_in_input = 0;;
 908
 909         FLAC__ASSERT(!options.common.sector_align || options.channels == 2);
 910         FLAC__ASSERT(!options.common.sector_align || options.bps == 16);
 911         FLAC__ASSERT(!options.common.sector_align || options.sample_rate == 44100);
 912         FLAC__ASSERT(!options.common.sector_align || infilesize >= 0);
 913         FLAC__ASSERT(!options.common.replay_gain || options.channels <= 2);
 914         FLAC__ASSERT(!options.common.replay_gain || grabbag__replaygain_is_valid_sample_frequency(options.sample_rate));
 915
 916         if(!
 917                 EncoderSession_construct(
 918                         &encoder_session,
 919 #ifdef FLAC__HAS_OGG
 920                         options.common.use_ogg,
 921 #else
 922                         /*use_ogg=*/false,
 923 #endif
 924                         options.common.verify,
 925                         options.common.verbose,
 926                         infile,
 927                         infilename,
 928                         outfilename
 929                 )
 930         )
 931                 return 1;
 932
 933         /*
 934          * now that we know the sample rate, canonicalize the
 935          * --skip string to a number of samples:
 936          */
 937         flac__utils_canonicalize_skip_until_specification(&options.common.skip_specification, options.sample_rate);
 938         FLAC__ASSERT(options.common.skip_specification.value.samples >= 0);
 939         encoder_session.skip = (FLAC__uint64)options.common.skip_specification.value.samples;
 940         FLAC__ASSERT(!options.common.sector_align || encoder_session.skip == 0);
 941
 942         if(infilesize < 0)
 943                 total_samples_in_input = 0;
 944         else {
 945                 /* *options.common.align_reservoir_samples will be 0 unless --sector-align is used */
 946                 FLAC__ASSERT(options.common.sector_align || *options.common.align_reservoir_samples == 0);
 947                 total_samples_in_input = (unsigned)infilesize / bytes_per_wide_sample + *options.common.align_reservoir_samples;
 948         }
 949
 950         /*
 951          * now that we know the input size, canonicalize the
 952          * --until strings to a number of samples:
 953          */
 954         if(!canonicalize_until_specification(&options.common.until_specification, encoder_session.inbasefilename, options.sample_rate, encoder_session.skip, total_samples_in_input))
 955                 return EncoderSession_finish_error(&encoder_session);
 956         encoder_session.until = (FLAC__uint64)options.common.until_specification.value.samples;
 957         FLAC__ASSERT(!options.common.sector_align || encoder_session.until == 0);
 958
 959         encoder_session.total_samples_to_encode = total_samples_in_input - encoder_session.skip;
 960         if(encoder_session.until > 0) {
 961                 const FLAC__uint64 trim = total_samples_in_input - encoder_session.until;
 962                 FLAC__ASSERT(total_samples_in_input > 0);
 963                 FLAC__ASSERT(!options.common.sector_align);
 964                 encoder_session.total_samples_to_encode -= trim;
 965         }
 966         if(infilesize >= 0 && options.common.sector_align) {
 967                 FLAC__ASSERT(encoder_session.skip == 0);
 968                 align_remainder = (unsigned)(encoder_session.total_samples_to_encode % 588);
 969                 if(options.common.is_last_file)
 970                         encoder_session.total_samples_to_encode += (588-align_remainder); /* will pad with zeroes */
 971                 else
 972                         encoder_session.total_samples_to_encode -= align_remainder; /* will stop short and carry over to next file */
 973         }
 974         encoder_session.unencoded_size = encoder_session.total_samples_to_encode * bytes_per_wide_sample;
 975
 976         if(encoder_session.verbose && encoder_session.total_samples_to_encode <= 0)
 977                 fprintf(stderr, "(No runtime statistics possible; please wait for encoding to finish...)\n");
 978
 979         if(encoder_session.skip > 0) {
 980                 unsigned skip_bytes = bytes_per_wide_sample * (unsigned)encoder_session.skip;
 981                 if(skip_bytes > lookahead_length) {
 982                         skip_bytes -= lookahead_length;
 983                         lookahead_length = 0;
 984                         if(fseek(infile, (long)skip_bytes, SEEK_CUR) < 0) {
 985                                 /* can't seek input, read ahead manually... */
 986                                 unsigned left, need;
 987                                 const unsigned chunk = sizeof(ucbuffer_);
 988                                 for(left = skip_bytes; left > 0; ) {
 989                                         need = min(left, chunk);
 990                                         if(fread(ucbuffer_, 1, need, infile) < need) {
 991                                                 fprintf(stderr, "%s: ERROR during read while skipping samples\n", encoder_session.inbasefilename);
 992                                                 return EncoderSession_finish_error(&encoder_session);
 993                                         }
 994                                         left -= need;
 995                                 }
 996                         }
 997                 }
 998                 else {
 999                         lookahead += skip_bytes;
1000                         lookahead_length -= skip_bytes;
1001                 }
1002         }
1003
1004         if(!EncoderSession_init_encoder(&encoder_session, options.common, options.channels, options.bps, options.sample_rate))
1005                 return EncoderSession_finish_error(&encoder_session);
1006
1007         /*
1008          * first do any samples in the reservoir
1009          */
1010         if(options.common.sector_align && *options.common.align_reservoir_samples > 0) {
1011                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)options.common.align_reservoir, *options.common.align_reservoir_samples)) {
1012                         print_error_with_state(&encoder_session, "ERROR during encoding");
1013                         return EncoderSession_finish_error(&encoder_session);
1014                 }
1015         }
1016
1017         /*
1018          * decrement infilesize if we need to align the file
1019          */
1020         if(options.common.sector_align) {
1021                 FLAC__ASSERT(infilesize >= 0);
1022                 if(options.common.is_last_file) {
1023                         *options.common.align_reservoir_samples = 0;
1024                 }
1025                 else {
1026                         *options.common.align_reservoir_samples = align_remainder;
1027                         infilesize -= (long)((*options.common.align_reservoir_samples) * bytes_per_wide_sample);
1028                         FLAC__ASSERT(infilesize >= 0);
1029                 }
1030         }
1031
1032         /*
1033          * now do from the file
1034          */
1035         if(infilesize < 0) {
1036                 while(!feof(infile)) {
1037                         if(lookahead_length > 0) {
1038                                 FLAC__ASSERT(lookahead_length < CHUNK_OF_SAMPLES * bytes_per_wide_sample);
1039                                 memcpy(ucbuffer_, lookahead, lookahead_length);
1040                                 bytes_read = fread(ucbuffer_+lookahead_length, sizeof(unsigned char), CHUNK_OF_SAMPLES * bytes_per_wide_sample - lookahead_length, infile) + lookahead_length;
1041                                 if(ferror(infile)) {
1042                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
1043                                         return EncoderSession_finish_error(&encoder_session);
1044                                 }
1045                                 lookahead_length = 0;
1046                         }
1047                         else
1048                                 bytes_read = fread(ucbuffer_, sizeof(unsigned char), CHUNK_OF_SAMPLES * bytes_per_wide_sample, infile);
1049
1050                         if(bytes_read == 0) {
1051                                 if(ferror(infile)) {
1052                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
1053                                         return EncoderSession_finish_error(&encoder_session);
1054                                 }
1055                         }
1056                         else if(bytes_read % bytes_per_wide_sample != 0) {
1057                                 fprintf(stderr, "%s: ERROR: got partial sample\n", encoder_session.inbasefilename);
1058                                 return EncoderSession_finish_error(&encoder_session);
1059                         }
1060                         else {
1061                                 unsigned wide_samples = bytes_read / bytes_per_wide_sample;
1062                                 format_input(input_, wide_samples, options.is_big_endian, options.is_unsigned_samples, options.channels, options.bps);
1063
1064                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)input_, wide_samples)) {
1065                                         print_error_with_state(&encoder_session, "ERROR during encoding");
1066                                         return EncoderSession_finish_error(&encoder_session);
1067                                 }
1068                         }
1069                 }
1070         }
1071         else {
1072                 const FLAC__uint64 max_input_bytes = encoder_session.total_samples_to_encode * bytes_per_wide_sample;
1073                 FLAC__uint64 total_input_bytes_read = 0;
1074                 while(total_input_bytes_read < max_input_bytes) {
1075                         {
1076                                 size_t wanted = (CHUNK_OF_SAMPLES * bytes_per_wide_sample);
1077                                 wanted = min(wanted, (size_t)(max_input_bytes - total_input_bytes_read));
1078
1079                                 if(lookahead_length > 0) {
1080                                         FLAC__ASSERT(lookahead_length <= wanted);
1081                                         memcpy(ucbuffer_, lookahead, lookahead_length);
1082                                         wanted -= lookahead_length;
1083                                         bytes_read = lookahead_length;
1084                                         if(wanted > 0) {
1085                                                 bytes_read += fread(ucbuffer_+lookahead_length, sizeof(unsigned char), wanted, infile);
1086                                                 if(ferror(infile)) {
1087                                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
1088                                                         return EncoderSession_finish_error(&encoder_session);
1089                                                 }
1090                                         }
1091                                         lookahead_length = 0;
1092                                 }
1093                                 else
1094                                         bytes_read = fread(ucbuffer_, sizeof(unsigned char), wanted, infile);
1095                         }
1096
1097                         if(bytes_read == 0) {
1098                                 if(ferror(infile)) {
1099                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
1100                                         return EncoderSession_finish_error(&encoder_session);
1101                                 }
1102                                 else if(feof(infile)) {
1103                                         fprintf(stderr, "%s: WARNING: unexpected EOF; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned)encoder_session.total_samples_to_encode, (unsigned)encoder_session.samples_written);
1104                                         total_input_bytes_read = max_input_bytes;
1105                                 }
1106                         }
1107                         else {
1108                                 if(bytes_read % bytes_per_wide_sample != 0) {
1109                                         fprintf(stderr, "%s: ERROR: got partial sample\n", encoder_session.inbasefilename);
1110                                         return EncoderSession_finish_error(&encoder_session);
1111                                 }
1112                                 else {
1113                                         unsigned wide_samples = bytes_read / bytes_per_wide_sample;
1114                                         format_input(input_, wide_samples, options.is_big_endian, options.is_unsigned_samples, options.channels, options.bps);
1115
1116                                         if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)input_, wide_samples)) {
1117                                                 print_error_with_state(&encoder_session, "ERROR during encoding");
1118                                                 return EncoderSession_finish_error(&encoder_session);
1119                                         }
1120                                         total_input_bytes_read += bytes_read;
1121                                 }
1122                         }
1123                 }
1124         }
1125
1126         /*
1127          * now read unaligned samples into reservoir or pad with zeroes if necessary
1128          */
1129         if(options.common.sector_align) {
1130                 if(options.common.is_last_file) {
1131                         unsigned wide_samples = 588 - align_remainder;
1132                         if(wide_samples < 588) {
1133                                 unsigned channel, data_bytes;
1134
1135                                 info_align_zero = wide_samples;
1136                                 data_bytes = wide_samples * (options.bps >> 3);
1137                                 for(channel = 0; channel < options.channels; channel++)
1138                                         memset(input_[channel], 0, data_bytes);
1139
1140                                 if(!EncoderSession_process(&encoder_session, (const FLAC__int32 * const *)input_, wide_samples)) {
1141                                         print_error_with_state(&encoder_session, "ERROR during encoding");
1142                                         return EncoderSession_finish_error(&encoder_session);
1143                                 }
1144                         }
1145                 }
1146                 else {
1147                         if(*options.common.align_reservoir_samples > 0) {
1148                                 FLAC__ASSERT(CHUNK_OF_SAMPLES >= 588);
1149                                 bytes_read = fread(ucbuffer_, sizeof(unsigned char), (*options.common.align_reservoir_samples) * bytes_per_wide_sample, infile);
1150                                 if(bytes_read == 0 && ferror(infile)) {
1151                                         fprintf(stderr, "%s: ERROR during read\n", encoder_session.inbasefilename);
1152                                         return EncoderSession_finish_error(&encoder_session);
1153                                 }
1154                                 else if(bytes_read != (*options.common.align_reservoir_samples) * bytes_per_wide_sample) {
1155                                         fprintf(stderr, "%s: WARNING: unexpected EOF; read %u bytes; expected %u samples, got %u samples\n", encoder_session.inbasefilename, (unsigned)bytes_read, (unsigned)encoder_session.total_samples_to_encode, (unsigned)encoder_session.samples_written);
1156                                 }
1157                                 else {
1158                                         info_align_carry = *options.common.align_reservoir_samples;
1159                                         format_input(options.common.align_reservoir, *options.common.align_reservoir_samples, false, options.is_unsigned_samples, options.channels, options.bps);
1160                                 }
1161                         }
1162                 }
1163         }
1164
1165         return EncoderSession_finish_ok(&encoder_session, info_align_carry, info_align_zero);
1166 }
1167
1168 FLAC__bool EncoderSession_construct(EncoderSession *e, FLAC__bool use_ogg, FLAC__bool verify, FLAC__bool verbose, FILE *infile, const char *infilename, const char *outfilename)
1169 {
1170         unsigned i;
1171         FLAC__uint32 test = 1;
1172
1173         /*
1174          * initialize globals
1175          */
1176
1177         is_big_endian_host_ = (*((FLAC__byte*)(&test)))? false : true;
1178
1179         for(i = 0; i < FLAC__MAX_CHANNELS; i++)
1180                 input_[i] = &(in_[i][0]);
1181
1182
1183         /*
1184          * initialize instance
1185          */
1186
1187 #ifdef FLAC__HAS_OGG
1188         e->use_ogg = use_ogg;
1189 #else
1190         (void)use_ogg;
1191 #endif
1192         e->verify = verify;
1193         e->verbose = verbose;
1194
1195         e->is_stdout = (0 == strcmp(outfilename, "-"));
1196
1197         e->inbasefilename = grabbag__file_get_basename(infilename);
1198         e->outfilename = outfilename;
1199
1200         e->skip = 0; /* filled in later after the sample_rate is known */
1201         e->unencoded_size = 0;
1202         e->total_samples_to_encode = 0;
1203         e->bytes_written = 0;
1204         e->samples_written = 0;
1205         e->blocksize = 0;
1206         e->stats_mask = 0;
1207
1208         e->encoder.flac.stream = 0;
1209         e->encoder.flac.file = 0;
1210 #ifdef FLAC__HAS_OGG
1211         e->encoder.ogg.stream = 0;
1212         e->encoder.ogg.file = 0;
1213 #endif
1214
1215         e->fin = infile;
1216         e->fout = 0;
1217         e->seek_table_template = 0;
1218
1219         if(e->is_stdout) {
1220                 e->fout = grabbag__file_get_binary_stdout();
1221         }
1222
1223         if(0 == (e->seek_table_template = FLAC__metadata_object_new(FLAC__METADATA_TYPE_SEEKTABLE))) {
1224                 fprintf(stderr, "%s: ERROR allocating memory for seek table\n", e->inbasefilename);
1225                 return false;
1226         }
1227
1228 #ifdef FLAC__HAS_OGG
1229         if(e->use_ogg) {
1230                 if(e->is_stdout) {
1231                         e->encoder.ogg.stream = OggFLAC__stream_encoder_new();
1232                         if(0 == e->encoder.ogg.stream) {
1233                                 fprintf(stderr, "%s: ERROR creating the encoder instance\n", e->inbasefilename);
1234                                 EncoderSession_destroy(e);
1235                                 return false;
1236                         }
1237                 }
1238                 else {
1239                         e->encoder.ogg.file = OggFLAC__file_encoder_new();
1240                         if(0 == e->encoder.ogg.file) {
1241                                 fprintf(stderr, "%s: ERROR creating the encoder instance\n", e->inbasefilename);
1242                                 EncoderSession_destroy(e);
1243                                 return false;
1244                         }
1245                 }
1246         }
1247         else
1248 #endif
1249         if(e->is_stdout) {
1250                 e->encoder.flac.stream = FLAC__stream_encoder_new();
1251                 if(0 == e->encoder.flac.stream) {
1252                         fprintf(stderr, "%s: ERROR creating the encoder instance\n", e->inbasefilename);
1253                         EncoderSession_destroy(e);
1254                         return false;
1255                 }
1256         }
1257         else {
1258                 e->encoder.flac.file = FLAC__file_encoder_new();
1259                 if(0 == e->encoder.flac.file) {
1260                         fprintf(stderr, "%s: ERROR creating the encoder instance\n", e->inbasefilename);
1261                         EncoderSession_destroy(e);
1262                         return false;
1263                 }
1264         }
1265
1266         return true;
1267 }
1268
1269 void EncoderSession_destroy(EncoderSession *e)
1270 {
1271         if(e->fin != stdin)
1272                 fclose(e->fin);
1273         if(0 != e->fout && e->fout != stdout)
1274                 fclose(e->fout);
1275
1276 #ifdef FLAC__HAS_OGG
1277         if(e->use_ogg) {
1278                 if(e->is_stdout) {
1279                         if(0 != e->encoder.ogg.stream) {
1280                                 OggFLAC__stream_encoder_delete(e->encoder.ogg.stream);
1281                                 e->encoder.ogg.stream = 0;
1282                         }
1283                 }
1284                 else {
1285                         if(0 != e->encoder.ogg.file) {
1286                                 OggFLAC__file_encoder_delete(e->encoder.ogg.file);
1287                                 e->encoder.ogg.file = 0;
1288                         }
1289                 }
1290         }
1291         else
1292 #endif
1293         if(e->is_stdout) {
1294                 if(0 != e->encoder.flac.stream) {
1295                         FLAC__stream_encoder_delete(e->encoder.flac.stream);
1296                         e->encoder.flac.stream = 0;
1297                 }
1298         }
1299         else {
1300                 if(0 != e->encoder.flac.file) {
1301                         FLAC__file_encoder_delete(e->encoder.flac.file);
1302                         e->encoder.flac.file = 0;
1303                 }
1304         }
1305
1306         if(0 != e->seek_table_template) {
1307                 FLAC__metadata_object_delete(e->seek_table_template);
1308                 e->seek_table_template = 0;
1309         }
1310 }
1311
1312 int EncoderSession_finish_ok(EncoderSession *e, int info_align_carry, int info_align_zero)
1313 {
1314         FLAC__StreamEncoderState fse_state = FLAC__STREAM_ENCODER_OK;
1315         int ret = 0;
1316
1317 #ifdef FLAC__HAS_OGG
1318         if(e->use_ogg) {
1319                 if(e->is_stdout) {
1320                         if(e->encoder.ogg.stream) {
1321                                 fse_state = OggFLAC__stream_encoder_get_FLAC_stream_encoder_state(e->encoder.ogg.stream);
1322                                 OggFLAC__stream_encoder_finish(e->encoder.ogg.stream);
1323                         }
1324                 }
1325                 else {
1326                         if(e->encoder.ogg.file) {
1327                                 fse_state = OggFLAC__file_encoder_get_FLAC_stream_encoder_state(e->encoder.ogg.file);
1328                                 OggFLAC__file_encoder_finish(e->encoder.ogg.file);
1329                         }
1330                 }
1331         }
1332         else
1333 #endif
1334         if(e->is_stdout) {
1335                 if(e->encoder.flac.stream) {
1336                         fse_state = FLAC__stream_encoder_get_state(e->encoder.flac.stream);
1337                         FLAC__stream_encoder_finish(e->encoder.flac.stream);
1338                 }
1339         }
1340         else {
1341                 if(e->encoder.flac.file) {
1342                         fse_state = FLAC__file_encoder_get_stream_encoder_state(e->encoder.flac.file);
1343                         FLAC__file_encoder_finish(e->encoder.flac.file);
1344                 }
1345         }
1346
1347         if(e->verbose && e->total_samples_to_encode > 0) {
1348                 print_stats(e);
1349                 fprintf(stderr, "\n");
1350         }
1351
1352         if(fse_state == FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA) {
1353                 print_verify_error(e);
1354                 ret = 1;
1355         }
1356         else {
1357                 if(info_align_carry >= 0)
1358                         fprintf(stderr, "%s: INFO: sector alignment causing %d samples to be carried over\n", e->inbasefilename, info_align_carry);
1359                 if(info_align_zero >= 0)
1360                         fprintf(stderr, "%s: INFO: sector alignment causing %d zero samples to be appended\n", e->inbasefilename, info_align_zero);
1361         }
1362
1363         EncoderSession_destroy(e);
1364
1365         return ret;
1366 }
1367
1368 int EncoderSession_finish_error(EncoderSession *e)
1369 {
1370         FLAC__StreamEncoderState fse_state;
1371
1372         if(e->verbose && e->total_samples_to_encode > 0)
1373                 fprintf(stderr, "\n");
1374
1375 #ifdef FLAC__HAS_OGG
1376         if(e->use_ogg) {
1377                 if(e->is_stdout) {
1378                         fse_state = OggFLAC__stream_encoder_get_FLAC_stream_encoder_state(e->encoder.ogg.stream);
1379                 }
1380                 else {
1381                         fse_state = OggFLAC__file_encoder_get_FLAC_stream_encoder_state(e->encoder.ogg.file);
1382                 }
1383         }
1384         else
1385 #endif
1386         if(e->is_stdout) {
1387                 fse_state = FLAC__stream_encoder_get_state(e->encoder.flac.stream);
1388         }
1389         else {
1390                 fse_state = FLAC__file_encoder_get_stream_encoder_state(e->encoder.flac.file);
1391         }
1392
1393         if(fse_state == FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA)
1394                 print_verify_error(e);
1395         else
1396                 unlink(e->outfilename);
1397
1398         EncoderSession_destroy(e);
1399
1400         return 1;
1401 }
1402
1403 FLAC__bool EncoderSession_init_encoder(EncoderSession *e, encode_options_t options, unsigned channels, unsigned bps, unsigned sample_rate)
1404 {
1405         unsigned num_metadata;
1406         FLAC__StreamMetadata padding, *cuesheet = 0;
1407         FLAC__StreamMetadata *metadata[4];
1408
1409         e->replay_gain = options.replay_gain;
1410         e->channels = channels;
1411         e->bits_per_sample = bps;
1412         e->sample_rate = sample_rate;
1413
1414         if(e->replay_gain) {
1415                 if(channels != 1 && channels != 2) {
1416                         fprintf(stderr, "%s: ERROR, number of channels (%u) must be 1 or 2 for --replay-gain\n", e->inbasefilename, channels);
1417                         return false;
1418                 }
1419                 if(!grabbag__replaygain_is_valid_sample_frequency(sample_rate)) {
1420                         fprintf(stderr, "%s: ERROR, invalid sample rate (%u) for --replay-gain\n", e->inbasefilename, sample_rate);
1421                         return false;
1422                 }
1423                 if(options.is_first_file) {
1424                         if(!grabbag__replaygain_init(sample_rate)) {
1425                                 fprintf(stderr, "%s: ERROR initializing ReplayGain stage\n", e->inbasefilename);
1426                                 return false;
1427                         }
1428                 }
1429         }
1430
1431         if(channels != 2)
1432                 options.do_mid_side = options.loose_mid_side = false;
1433
1434         if(!parse_cuesheet_(&cuesheet, options.cuesheet_filename, e->inbasefilename, e->total_samples_to_encode))
1435                 return false;
1436
1437         if(!convert_to_seek_table_template(options.requested_seek_points, options.num_requested_seek_points, options.cued_seekpoints? cuesheet : 0, e)) {
1438                 fprintf(stderr, "%s: ERROR allocating memory for seek table\n", e->inbasefilename);
1439                 if(0 != cuesheet)
1440                         free(cuesheet);
1441                 return false;
1442         }
1443
1444         num_metadata = 0;
1445         if(e->seek_table_template->data.seek_table.num_points > 0) {
1446                 e->seek_table_template->is_last = false; /* the encoder will set this for us */
1447                 metadata[num_metadata++] = e->seek_table_template;
1448         }
1449         if(0 != cuesheet)
1450                 metadata[num_metadata++] = cuesheet;
1451         metadata[num_metadata++] = options.vorbis_comment;
1452         if(options.padding > 0) {
1453                 padding.is_last = false; /* the encoder will set this for us */
1454                 padding.type = FLAC__METADATA_TYPE_PADDING;
1455                 padding.length = (unsigned)options.padding;
1456                 metadata[num_metadata++] = &padding;
1457         }
1458
1459         e->blocksize = options.blocksize;
1460         e->stats_mask = (options.do_exhaustive_model_search || options.do_qlp_coeff_prec_search)? 0x0f : 0x3f;
1461
1462 #ifdef FLAC__HAS_OGG
1463         if(e->use_ogg) {
1464                 if(e->is_stdout) {
1465                         if(options.has_serial_number)
1466                                 OggFLAC__stream_encoder_set_serial_number(e->encoder.ogg.stream, options.serial_number);
1467                         OggFLAC__stream_encoder_set_verify(e->encoder.ogg.stream, options.verify);
1468                         OggFLAC__stream_encoder_set_streamable_subset(e->encoder.ogg.stream, !options.lax);
1469                         OggFLAC__stream_encoder_set_do_mid_side_stereo(e->encoder.ogg.stream, options.do_mid_side);
1470                         OggFLAC__stream_encoder_set_loose_mid_side_stereo(e->encoder.ogg.stream, options.loose_mid_side);
1471                         OggFLAC__stream_encoder_set_channels(e->encoder.ogg.stream, channels);
1472                         OggFLAC__stream_encoder_set_bits_per_sample(e->encoder.ogg.stream, bps);
1473                         OggFLAC__stream_encoder_set_sample_rate(e->encoder.ogg.stream, sample_rate);
1474                         OggFLAC__stream_encoder_set_blocksize(e->encoder.ogg.stream, options.blocksize);
1475                         OggFLAC__stream_encoder_set_max_lpc_order(e->encoder.ogg.stream, options.max_lpc_order);
1476                         OggFLAC__stream_encoder_set_qlp_coeff_precision(e->encoder.ogg.stream, options.qlp_coeff_precision);
1477                         OggFLAC__stream_encoder_set_do_qlp_coeff_prec_search(e->encoder.ogg.stream, options.do_qlp_coeff_prec_search);
1478                         OggFLAC__stream_encoder_set_do_escape_coding(e->encoder.ogg.stream, options.do_escape_coding);
1479                         OggFLAC__stream_encoder_set_do_exhaustive_model_search(e->encoder.ogg.stream, options.do_exhaustive_model_search);
1480                         OggFLAC__stream_encoder_set_min_residual_partition_order(e->encoder.ogg.stream, options.min_residual_partition_order);
1481                         OggFLAC__stream_encoder_set_max_residual_partition_order(e->encoder.ogg.stream, options.max_residual_partition_order);
1482                         OggFLAC__stream_encoder_set_rice_parameter_search_dist(e->encoder.ogg.stream, options.rice_parameter_search_dist);
1483                         OggFLAC__stream_encoder_set_total_samples_estimate(e->encoder.ogg.stream, e->total_samples_to_encode);
1484                         OggFLAC__stream_encoder_set_metadata(e->encoder.ogg.stream, (num_metadata > 0)? metadata : 0, num_metadata);
1485                         OggFLAC__stream_encoder_set_write_callback(e->encoder.ogg.stream, ogg_stream_encoder_write_callback);
1486                         OggFLAC__stream_encoder_set_metadata_callback(e->encoder.ogg.stream, ogg_stream_encoder_metadata_callback);
1487                         OggFLAC__stream_encoder_set_client_data(e->encoder.ogg.stream, e);
1488
1489                         OggFLAC__stream_encoder_disable_constant_subframes(e->encoder.ogg.stream, options.debug.disable_constant_subframes);
1490                         OggFLAC__stream_encoder_disable_fixed_subframes(e->encoder.ogg.stream, options.debug.disable_fixed_subframes);
1491                         OggFLAC__stream_encoder_disable_verbatim_subframes(e->encoder.ogg.stream, options.debug.disable_verbatim_subframes);
1492
1493                         if(OggFLAC__stream_encoder_init(e->encoder.ogg.stream) != FLAC__STREAM_ENCODER_OK) {
1494                                 print_error_with_state(e, "ERROR initializing encoder");
1495                                 if(0 != cuesheet)
1496                                         free(cuesheet);
1497                                 return false;
1498                         }
1499                 }
1500                 else {
1501                         if(options.has_serial_number)
1502                                 OggFLAC__file_encoder_set_serial_number(e->encoder.ogg.file, options.serial_number);
1503                         OggFLAC__file_encoder_set_filename(e->encoder.ogg.file, e->outfilename);
1504                         OggFLAC__file_encoder_set_verify(e->encoder.ogg.file, options.verify);
1505                         OggFLAC__file_encoder_set_streamable_subset(e->encoder.ogg.file, !options.lax);
1506                         OggFLAC__file_encoder_set_do_mid_side_stereo(e->encoder.ogg.file, options.do_mid_side);
1507                         OggFLAC__file_encoder_set_loose_mid_side_stereo(e->encoder.ogg.file, options.loose_mid_side);
1508                         OggFLAC__file_encoder_set_channels(e->encoder.ogg.file, channels);
1509                         OggFLAC__file_encoder_set_bits_per_sample(e->encoder.ogg.file, bps);
1510                         OggFLAC__file_encoder_set_sample_rate(e->encoder.ogg.file, sample_rate);
1511                         OggFLAC__file_encoder_set_blocksize(e->encoder.ogg.file, options.blocksize);
1512                         OggFLAC__file_encoder_set_max_lpc_order(e->encoder.ogg.file, options.max_lpc_order);
1513                         OggFLAC__file_encoder_set_qlp_coeff_precision(e->encoder.ogg.file, options.qlp_coeff_precision);
1514                         OggFLAC__file_encoder_set_do_qlp_coeff_prec_search(e->encoder.ogg.file, options.do_qlp_coeff_prec_search);
1515                         OggFLAC__file_encoder_set_do_escape_coding(e->encoder.ogg.file, options.do_escape_coding);
1516                         OggFLAC__file_encoder_set_do_exhaustive_model_search(e->encoder.ogg.file, options.do_exhaustive_model_search);
1517                         OggFLAC__file_encoder_set_min_residual_partition_order(e->encoder.ogg.file, options.min_residual_partition_order);
1518                         OggFLAC__file_encoder_set_max_residual_partition_order(e->encoder.ogg.file, options.max_residual_partition_order);
1519                         OggFLAC__file_encoder_set_rice_parameter_search_dist(e->encoder.ogg.file, options.rice_parameter_search_dist);
1520                         OggFLAC__file_encoder_set_total_samples_estimate(e->encoder.ogg.file, e->total_samples_to_encode);
1521                         OggFLAC__file_encoder_set_metadata(e->encoder.ogg.file, (num_metadata > 0)? metadata : 0, num_metadata);
1522                         OggFLAC__file_encoder_set_progress_callback(e->encoder.ogg.file, ogg_file_encoder_progress_callback);
1523                         OggFLAC__file_encoder_set_client_data(e->encoder.ogg.file, e);
1524
1525                         OggFLAC__file_encoder_disable_constant_subframes(e->encoder.ogg.file, options.debug.disable_constant_subframes);
1526                         OggFLAC__file_encoder_disable_fixed_subframes(e->encoder.ogg.file, options.debug.disable_fixed_subframes);
1527                         OggFLAC__file_encoder_disable_verbatim_subframes(e->encoder.ogg.file, options.debug.disable_verbatim_subframes);
1528
1529                         if(OggFLAC__file_encoder_init(e->encoder.ogg.file) != OggFLAC__FILE_ENCODER_OK) {
1530                                 print_error_with_state(e, "ERROR initializing encoder");
1531                                 if(0 != cuesheet)
1532                                         free(cuesheet);
1533                                 return false;
1534                         }
1535                 }
1536         }
1537         else
1538 #endif
1539         if(e->is_stdout) {
1540                 FLAC__stream_encoder_set_verify(e->encoder.flac.stream, options.verify);
1541                 FLAC__stream_encoder_set_streamable_subset(e->encoder.flac.stream, !options.lax);
1542                 FLAC__stream_encoder_set_do_mid_side_stereo(e->encoder.flac.stream, options.do_mid_side);
1543                 FLAC__stream_encoder_set_loose_mid_side_stereo(e->encoder.flac.stream, options.loose_mid_side);
1544                 FLAC__stream_encoder_set_channels(e->encoder.flac.stream, channels);
1545                 FLAC__stream_encoder_set_bits_per_sample(e->encoder.flac.stream, bps);
1546                 FLAC__stream_encoder_set_sample_rate(e->encoder.flac.stream, sample_rate);
1547                 FLAC__stream_encoder_set_blocksize(e->encoder.flac.stream, options.blocksize);
1548                 FLAC__stream_encoder_set_max_lpc_order(e->encoder.flac.stream, options.max_lpc_order);
1549                 FLAC__stream_encoder_set_qlp_coeff_precision(e->encoder.flac.stream, options.qlp_coeff_precision);
1550                 FLAC__stream_encoder_set_do_qlp_coeff_prec_search(e->encoder.flac.stream, options.do_qlp_coeff_prec_search);
1551                 FLAC__stream_encoder_set_do_escape_coding(e->encoder.flac.stream, options.do_escape_coding);
1552                 FLAC__stream_encoder_set_do_exhaustive_model_search(e->encoder.flac.stream, options.do_exhaustive_model_search);
1553                 FLAC__stream_encoder_set_min_residual_partition_order(e->encoder.flac.stream, options.min_residual_partition_order);
1554                 FLAC__stream_encoder_set_max_residual_partition_order(e->encoder.flac.stream, options.max_residual_partition_order);
1555                 FLAC__stream_encoder_set_rice_parameter_search_dist(e->encoder.flac.stream, options.rice_parameter_search_dist);
1556                 FLAC__stream_encoder_set_total_samples_estimate(e->encoder.flac.stream, e->total_samples_to_encode);
1557                 FLAC__stream_encoder_set_metadata(e->encoder.flac.stream, (num_metadata > 0)? metadata : 0, num_metadata);
1558                 FLAC__stream_encoder_set_write_callback(e->encoder.flac.stream, flac_stream_encoder_write_callback);
1559                 FLAC__stream_encoder_set_metadata_callback(e->encoder.flac.stream, flac_stream_encoder_metadata_callback);
1560                 FLAC__stream_encoder_set_client_data(e->encoder.flac.stream, e);
1561
1562                 FLAC__stream_encoder_disable_constant_subframes(e->encoder.flac.stream, options.debug.disable_constant_subframes);
1563                 FLAC__stream_encoder_disable_fixed_subframes(e->encoder.flac.stream, options.debug.disable_fixed_subframes);
1564                 FLAC__stream_encoder_disable_verbatim_subframes(e->encoder.flac.stream, options.debug.disable_verbatim_subframes);
1565
1566                 if(FLAC__stream_encoder_init(e->encoder.flac.stream) != FLAC__STREAM_ENCODER_OK) {
1567                         print_error_with_state(e, "ERROR initializing encoder");
1568                         if(0 != cuesheet)
1569                                 free(cuesheet);
1570                         return false;
1571                 }
1572         }
1573         else {
1574                 FLAC__file_encoder_set_filename(e->encoder.flac.file, e->outfilename);
1575                 FLAC__file_encoder_set_verify(e->encoder.flac.file, options.verify);
1576                 FLAC__file_encoder_set_streamable_subset(e->encoder.flac.file, !options.lax);
1577                 FLAC__file_encoder_set_do_mid_side_stereo(e->encoder.flac.file, options.do_mid_side);
1578                 FLAC__file_encoder_set_loose_mid_side_stereo(e->encoder.flac.file, options.loose_mid_side);
1579                 FLAC__file_encoder_set_channels(e->encoder.flac.file, channels);
1580                 FLAC__file_encoder_set_bits_per_sample(e->encoder.flac.file, bps);
1581                 FLAC__file_encoder_set_sample_rate(e->encoder.flac.file, sample_rate);
1582                 FLAC__file_encoder_set_blocksize(e->encoder.flac.file, options.blocksize);
1583                 FLAC__file_encoder_set_max_lpc_order(e->encoder.flac.file, options.max_lpc_order);
1584                 FLAC__file_encoder_set_qlp_coeff_precision(e->encoder.flac.file, options.qlp_coeff_precision);
1585                 FLAC__file_encoder_set_do_qlp_coeff_prec_search(e->encoder.flac.file, options.do_qlp_coeff_prec_search);
1586                 FLAC__file_encoder_set_do_escape_coding(e->encoder.flac.file, options.do_escape_coding);
1587                 FLAC__file_encoder_set_do_exhaustive_model_search(e->encoder.flac.file, options.do_exhaustive_model_search);
1588                 FLAC__file_encoder_set_min_residual_partition_order(e->encoder.flac.file, options.min_residual_partition_order);
1589                 FLAC__file_encoder_set_max_residual_partition_order(e->encoder.flac.file, options.max_residual_partition_order);
1590                 FLAC__file_encoder_set_rice_parameter_search_dist(e->encoder.flac.file, options.rice_parameter_search_dist);
1591                 FLAC__file_encoder_set_total_samples_estimate(e->encoder.flac.file, e->total_samples_to_encode);
1592                 FLAC__file_encoder_set_metadata(e->encoder.flac.file, (num_metadata > 0)? metadata : 0, num_metadata);
1593                 FLAC__file_encoder_set_progress_callback(e->encoder.flac.file, flac_file_encoder_progress_callback);
1594                 FLAC__file_encoder_set_client_data(e->encoder.flac.file, e);
1595
1596                 FLAC__file_encoder_disable_constant_subframes(e->encoder.flac.file, options.debug.disable_constant_subframes);
1597                 FLAC__file_encoder_disable_fixed_subframes(e->encoder.flac.file, options.debug.disable_fixed_subframes);
1598                 FLAC__file_encoder_disable_verbatim_subframes(e->encoder.flac.file, options.debug.disable_verbatim_subframes);
1599
1600                 if(FLAC__file_encoder_init(e->encoder.flac.file) != FLAC__FILE_ENCODER_OK) {
1601                         print_error_with_state(e, "ERROR initializing encoder");
1602                         if(0 != cuesheet)
1603                                 free(cuesheet);
1604                         return false;
1605                 }
1606         }
1607
1608         if(0 != cuesheet)
1609                 free(cuesheet);
1610
1611         return true;
1612 }
1613
1614 FLAC__bool EncoderSession_process(EncoderSession *e, const FLAC__int32 * const buffer[], unsigned samples)
1615 {
1616         if(e->replay_gain) {
1617                 if(!grabbag__replaygain_analyze(buffer, e->channels==2, e->bits_per_sample, samples))
1618                         fprintf(stderr, "%s: WARNING, error while calculating ReplayGain\n", e->inbasefilename);
1619         }
1620
1621 #ifdef FLAC__HAS_OGG
1622         if(e->use_ogg) {
1623                 if(e->is_stdout) {
1624                         return OggFLAC__stream_encoder_process(e->encoder.ogg.stream, buffer, samples);
1625                 }
1626                 else {
1627                         return OggFLAC__file_encoder_process(e->encoder.ogg.file, buffer, samples);
1628                 }
1629         }
1630         else
1631 #endif
1632         if(e->is_stdout) {
1633                 return FLAC__stream_encoder_process(e->encoder.flac.stream, buffer, samples);
1634         }
1635         else {
1636                 return FLAC__file_encoder_process(e->encoder.flac.file, buffer, samples);
1637         }
1638 }
1639
1640 FLAC__bool convert_to_seek_table_template(const char *requested_seek_points, int num_requested_seek_points, FLAC__StreamMetadata *cuesheet, EncoderSession *e)
1641 {
1642         const FLAC__bool only_placeholders = e->is_stdout;
1643         FLAC__bool has_real_points;
1644
1645         if(num_requested_seek_points == 0 && 0 == cuesheet)
1646                 return true;
1647
1648         if(num_requested_seek_points < 0) {
1649                 requested_seek_points = "10s;";
1650                 num_requested_seek_points = 1;
1651         }
1652
1653         if(num_requested_seek_points > 0) {
1654                 if(!grabbag__seektable_convert_specification_to_template(requested_seek_points, only_placeholders, e->total_samples_to_encode, e->sample_rate, e->seek_table_template, &has_real_points))
1655                         return false;
1656         }
1657
1658         if(0 != cuesheet) {
1659                 unsigned i, j;
1660                 const FLAC__StreamMetadata_CueSheet *cs = &cuesheet->data.cue_sheet;
1661                 for(i = 0; i < cs->num_tracks; i++) {
1662                         const FLAC__StreamMetadata_CueSheet_Track *tr = cs->tracks+i;
1663                         for(j = 0; j < tr->num_indices; j++) {
1664                                 if(!FLAC__metadata_object_seektable_template_append_point(e->seek_table_template, tr->offset + tr->indices[j].offset))
1665                                         return false;
1666                                 has_real_points = true;
1667                         }
1668                 }
1669                 if(has_real_points)
1670                         if(!FLAC__metadata_object_seektable_template_sort(e->seek_table_template, /*compact=*/true))
1671                                 return false;
1672         }
1673
1674         if(has_real_points) {
1675                 if(e->is_stdout)
1676                         fprintf(stderr, "%s: WARNING, cannot write back seekpoints when encoding to stdout\n", e->inbasefilename);
1677         }
1678
1679         return true;
1680 }
1681
1682 FLAC__bool canonicalize_until_specification(utils__SkipUntilSpecification *spec, const char *inbasefilename, unsigned sample_rate, FLAC__uint64 skip, FLAC__uint64 total_samples_in_input)
1683 {
1684         /* convert from mm:ss.sss to sample number if necessary */
1685         flac__utils_canonicalize_skip_until_specification(spec, sample_rate);
1686
1687         /* special case: if "--until=-0", use the special value '0' to mean "end-of-stream" */
1688         if(spec->is_relative && spec->value.samples == 0) {
1689                 spec->is_relative = false;
1690                 return true;
1691         }
1692
1693         /* in any other case the total samples in the input must be known */
1694         if(total_samples_in_input == 0) {
1695                 fprintf(stderr, "%s: ERROR, cannot use --until when input length is unknown\n", inbasefilename);
1696                 return false;
1697         }
1698
1699         FLAC__ASSERT(spec->value_is_samples);
1700
1701         /* convert relative specifications to absolute */
1702         if(spec->is_relative) {
1703                 if(spec->value.samples <= 0)
1704                         spec->value.samples += (FLAC__int64)total_samples_in_input;
1705                 else
1706                         spec->value.samples += skip;
1707                 spec->is_relative = false;
1708         }
1709
1710         /* error check */
1711         if(spec->value.samples < 0) {
1712                 fprintf(stderr, "%s: ERROR, --until value is before beginning of input\n", inbasefilename);
1713                 return false;
1714         }
1715         if((FLAC__uint64)spec->value.samples <= skip) {
1716                 fprintf(stderr, "%s: ERROR, --until value is before --skip point\n", inbasefilename);
1717                 return false;
1718         }
1719         if((FLAC__uint64)spec->value.samples > total_samples_in_input) {
1720                 fprintf(stderr, "%s: ERROR, --until value is after end of input\n", inbasefilename);
1721                 return false;
1722         }
1723
1724         return true;
1725 }
1726
1727 void format_input(FLAC__int32 *dest[], unsigned wide_samples, FLAC__bool is_big_endian, FLAC__bool is_unsigned_samples, unsigned channels, unsigned bps)
1728 {
1729         unsigned wide_sample, sample, channel, byte;
1730
1731         if(bps == 8) {
1732                 if(is_unsigned_samples) {
1733                         for(sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1734                                 for(channel = 0; channel < channels; channel++, sample++)
1735                                         dest[channel][wide_sample] = (FLAC__int32)ucbuffer_[sample] - 0x80;
1736                 }
1737                 else {
1738                         for(sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1739                                 for(channel = 0; channel < channels; channel++, sample++)
1740                                         dest[channel][wide_sample] = (FLAC__int32)scbuffer_[sample];
1741                 }
1742         }
1743         else if(bps == 16) {
1744                 if(is_big_endian != is_big_endian_host_) {
1745                         unsigned char tmp;
1746                         const unsigned bytes = wide_samples * channels * (bps >> 3);
1747                         for(byte = 0; byte < bytes; byte += 2) {
1748                                 tmp = ucbuffer_[byte];
1749                                 ucbuffer_[byte] = ucbuffer_[byte+1];
1750                                 ucbuffer_[byte+1] = tmp;
1751                         }
1752                 }
1753                 if(is_unsigned_samples) {
1754                         for(sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1755                                 for(channel = 0; channel < channels; channel++, sample++)
1756                                         dest[channel][wide_sample] = (FLAC__int32)usbuffer_[sample] - 0x8000;
1757                 }
1758                 else {
1759                         for(sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1760                                 for(channel = 0; channel < channels; channel++, sample++)
1761                                         dest[channel][wide_sample] = (FLAC__int32)ssbuffer_[sample];
1762                 }
1763         }
1764         else if(bps == 24) {
1765                 if(!is_big_endian) {
1766                         unsigned char tmp;
1767                         const unsigned bytes = wide_samples * channels * (bps >> 3);
1768                         for(byte = 0; byte < bytes; byte += 3) {
1769                                 tmp = ucbuffer_[byte];
1770                                 ucbuffer_[byte] = ucbuffer_[byte+2];
1771                                 ucbuffer_[byte+2] = tmp;
1772                         }
1773                 }
1774                 if(is_unsigned_samples) {
1775                         for(byte = sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1776                                 for(channel = 0; channel < channels; channel++, sample++) {
1777                                         dest[channel][wide_sample]  = ucbuffer_[byte++]; dest[channel][wide_sample] <<= 8;
1778                                         dest[channel][wide_sample] |= ucbuffer_[byte++]; dest[channel][wide_sample] <<= 8;
1779                                         dest[channel][wide_sample] |= ucbuffer_[byte++];
1780                                         dest[channel][wide_sample] -= 0x800000;
1781                                 }
1782                 }
1783                 else {
1784                         for(byte = sample = wide_sample = 0; wide_sample < wide_samples; wide_sample++)
1785                                 for(channel = 0; channel < channels; channel++, sample++) {
1786                                         dest[channel][wide_sample]  = scbuffer_[byte++]; dest[channel][wide_sample] <<= 8;
1787                                         dest[channel][wide_sample] |= ucbuffer_[byte++]; dest[channel][wide_sample] <<= 8;
1788                                         dest[channel][wide_sample] |= ucbuffer_[byte++];
1789                                 }
1790                 }
1791         }
1792         else {
1793                 FLAC__ASSERT(0);
1794         }
1795 }
1796
1797 #ifdef FLAC__HAS_OGG
1798 FLAC__StreamEncoderWriteStatus ogg_stream_encoder_write_callback(const OggFLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
1799 {
1800         EncoderSession *encoder_session = (EncoderSession*)client_data;
1801
1802         (void)encoder;
1803
1804         encoder_session->bytes_written += bytes;
1805         /*
1806          * With Ogg FLAC we don't get one write callback per frame and
1807          * we don't have a good number for 'samples', so we estimate based
1808          * on the frame number and the knowledge that all blocks (except
1809          * the last) are the same size.
1810          */
1811         (void)samples;
1812         encoder_session->samples_written = (current_frame+1) * encoder_session->blocksize;
1813
1814         if(encoder_session->verbose && encoder_session->total_samples_to_encode > 0 && !(current_frame & encoder_session->stats_mask))
1815                 print_stats(encoder_session);
1816
1817         if(flac__utils_fwrite(buffer, sizeof(FLAC__byte), bytes, encoder_session->fout) == bytes)
1818                 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
1819         else
1820                 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
1821 }
1822
1823 void ogg_stream_encoder_metadata_callback(const OggFLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
1824 {
1825         // do nothing, for compatibilty.  soon we will be using the ogg file encoder anyway.
1826         (void)encoder, (void)metadata, (void)client_data;
1827 }
1828
1829 void ogg_file_encoder_progress_callback(const OggFLAC__FileEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data)
1830 {
1831         EncoderSession *encoder_session = (EncoderSession*)client_data;
1832
1833         (void)encoder;
1834
1835         /*
1836          * With Ogg FLAC we don't get a value for 'samples_written', so we
1837          * estimate based on the frames written and the knowledge that all
1838          * blocks (except the last) are the same size.
1839          */
1840         samples_written = frames_written * encoder_session->blocksize;
1841         flac_file_encoder_progress_callback(0, bytes_written, samples_written, frames_written, total_frames_estimate, client_data);
1842 }
1843
1844 #endif
1845
1846 FLAC__StreamEncoderWriteStatus flac_stream_encoder_write_callback(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
1847 {
1848         EncoderSession *encoder_session = (EncoderSession*)client_data;
1849
1850         (void)encoder;
1851
1852         encoder_session->bytes_written += bytes;
1853         encoder_session->samples_written += samples;
1854
1855         if(samples && encoder_session->verbose && encoder_session->total_samples_to_encode > 0 && !(current_frame & encoder_session->stats_mask))
1856                 print_stats(encoder_session);
1857
1858         if(flac__utils_fwrite(buffer, sizeof(FLAC__byte), bytes, encoder_session->fout) == bytes)
1859                 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
1860         else
1861                 return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
1862 }
1863
1864 void flac_stream_encoder_metadata_callback(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
1865 {
1866         /*
1867          * Nothing to do; if we get here, we're decoding to stdout, in
1868          * which case we can't seek backwards to write new metadata.
1869          */
1870         (void)encoder, (void)metadata, (void)client_data;
1871 }
1872
1873 void flac_file_encoder_progress_callback(const FLAC__FileEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data)
1874 {
1875         EncoderSession *encoder_session = (EncoderSession*)client_data;
1876
1877         (void)encoder, (void)total_frames_estimate;
1878
1879         encoder_session->bytes_written = bytes_written;
1880         encoder_session->samples_written = samples_written;
1881
1882         if(encoder_session->verbose && encoder_session->total_samples_to_encode > 0 && !((frames_written-1) & encoder_session->stats_mask))
1883                 print_stats(encoder_session);
1884 }
1885
1886 FLAC__bool parse_cuesheet_(FLAC__StreamMetadata **cuesheet, const char *cuesheet_filename, const char *inbasefilename, FLAC__uint64 lead_out_offset)
1887 {
1888         FILE *f;
1889         unsigned last_line_read;
1890         const char *error_message;
1891
1892         if(0 == cuesheet_filename)
1893                 return true;
1894
1895         if(lead_out_offset == 0) {
1896                 fprintf(stderr, "%s: ERROR cannot import cuesheet when the number of input samples to encode is unknown\n", inbasefilename);
1897                 return false;
1898         }
1899
1900         if(0 == (f = fopen(cuesheet_filename, "r"))) {
1901                 fprintf(stderr, "%s: ERROR opening cuesheet \"%s\" for reading\n", inbasefilename, cuesheet_filename);
1902                 return false;
1903         }
1904
1905         *cuesheet = grabbag__cuesheet_parse(f, &error_message, &last_line_read, /*@@@is_cdda=*/true, lead_out_offset);
1906
1907         fclose(f);
1908
1909         if(0 == *cuesheet) {
1910                 fprintf(stderr, "%s: ERROR parsing cuesheet \"%s\" on line %u: %s\n", inbasefilename, cuesheet_filename, last_line_read, error_message);
1911                 return false;
1912         }
1913
1914         return true;
1915 }
1916
1917 void print_stats(const EncoderSession *encoder_session)
1918 {
1919         const FLAC__uint64 samples_written = min(encoder_session->total_samples_to_encode, encoder_session->samples_written);
1920 #if defined _MSC_VER || defined __MINGW32__
1921         /* with VC++ you have to spoon feed it the casting */
1922         const double progress = (double)(FLAC__int64)samples_written / (double)(FLAC__int64)encoder_session->total_samples_to_encode;
1923         const double ratio = (double)(FLAC__int64)encoder_session->bytes_written / ((double)(FLAC__int64)encoder_session->unencoded_size * progress);
1924 #else
1925         const double progress = (double)samples_written / (double)encoder_session->total_samples_to_encode;
1926         const double ratio = (double)encoder_session->bytes_written / ((double)encoder_session->unencoded_size * max(1.0, progress));
1927 #endif
1928
1929         if(samples_written == encoder_session->total_samples_to_encode) {
1930                 fprintf(stderr, "\r%s:%s wrote %u bytes, ratio=%0.3f",
1931                         encoder_session->inbasefilename,
1932                         encoder_session->verify? " Verify OK," : "",
1933                         (unsigned)encoder_session->bytes_written,
1934                         ratio
1935                 );
1936         }
1937         else {
1938                 fprintf(stderr, "\r%s: %u%% complete, ratio=%0.3f", encoder_session->inbasefilename, (unsigned)floor(progress * 100.0 + 0.5), ratio);
1939         }
1940 }
1941
1942 void print_error_with_state(const EncoderSession *e, const char *message)
1943 {
1944         const int ilen = strlen(e->inbasefilename) + 1;
1945         const char *state_string;
1946
1947         fprintf(stderr, "\n%s: %s\n", e->inbasefilename, message);
1948
1949 #ifdef FLAC__HAS_OGG
1950         if(e->use_ogg) {
1951                 if(e->is_stdout) {
1952                         state_string = OggFLAC__stream_encoder_get_resolved_state_string(e->encoder.ogg.stream);
1953                 }
1954                 else {
1955                         state_string = OggFLAC__file_encoder_get_resolved_state_string(e->encoder.ogg.file);
1956                 }
1957         }
1958         else
1959 #endif
1960         if(e->is_stdout) {
1961                 state_string = FLAC__stream_encoder_get_resolved_state_string(e->encoder.flac.stream);
1962         }
1963         else {
1964                 state_string = FLAC__file_encoder_get_resolved_state_string(e->encoder.flac.file);
1965         }
1966
1967         fprintf(stderr, "%*s state = %s\n", ilen, "", state_string);
1968 }
1969
1970 void print_verify_error(EncoderSession *e)
1971 {
1972         FLAC__uint64 absolute_sample;
1973         unsigned frame_number;
1974         unsigned channel;
1975         unsigned sample;
1976         FLAC__int32 expected;
1977         FLAC__int32 got;
1978
1979 #ifdef FLAC__HAS_OGG
1980         if(e->use_ogg) {
1981                 if(e->is_stdout) {
1982                         OggFLAC__stream_encoder_get_verify_decoder_error_stats(e->encoder.ogg.stream, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
1983                 }
1984                 else {
1985                         OggFLAC__file_encoder_get_verify_decoder_error_stats(e->encoder.ogg.file, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
1986                 }
1987         }
1988         else
1989 #endif
1990         if(e->is_stdout) {
1991                 FLAC__stream_encoder_get_verify_decoder_error_stats(e->encoder.flac.stream, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
1992         }
1993         else {
1994                 FLAC__file_encoder_get_verify_decoder_error_stats(e->encoder.flac.file, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
1995         }
1996
1997         fprintf(stderr, "%s: ERROR: mismatch in decoded data, verify FAILED!\n", e->inbasefilename);
1998         fprintf(stderr, "       Absolute sample=%u, frame=%u, channel=%u, sample=%u, expected %d, got %d\n", (unsigned)absolute_sample, frame_number, channel, sample, expected, got);
1999         fprintf(stderr, "       Please submit a bug report to\n");
2000         fprintf(stderr, "           http://sourceforge.net/bugs/?func=addbug&group_id=13478\n");
2001         fprintf(stderr, "       Make sure to include an email contact in the comment and/or use the\n");
2002         fprintf(stderr, "       \"Monitor\" feature to monitor the bug status.\n");
2003         fprintf(stderr, "Verify FAILED!  Do not trust %s\n", e->outfilename);
2004 }
2005
2006 FLAC__bool read_little_endian_uint16(FILE *f, FLAC__uint16 *val, FLAC__bool eof_ok, const char *fn)
2007 {
2008         size_t bytes_read = fread(val, 1, 2, f);
2009
2010         if(bytes_read == 0) {
2011                 if(!eof_ok) {
2012                         fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2013                         return false;
2014                 }
2015                 else
2016                         return true;
2017         }
2018         else if(bytes_read < 2) {
2019                 fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2020                 return false;
2021         }
2022         else {
2023                 if(is_big_endian_host_) {
2024                         FLAC__byte tmp, *b = (FLAC__byte*)val;
2025                         tmp = b[1]; b[1] = b[0]; b[0] = tmp;
2026                 }
2027                 return true;
2028         }
2029 }
2030
2031 FLAC__bool read_little_endian_uint32(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn)
2032 {
2033         size_t bytes_read = fread(val, 1, 4, f);
2034
2035         if(bytes_read == 0) {
2036                 if(!eof_ok) {
2037                         fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2038                         return false;
2039                 }
2040                 else
2041                         return true;
2042         }
2043         else if(bytes_read < 4) {
2044                 fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2045                 return false;
2046         }
2047         else {
2048                 if(is_big_endian_host_) {
2049                         FLAC__byte tmp, *b = (FLAC__byte*)val;
2050                         tmp = b[3]; b[3] = b[0]; b[0] = tmp;
2051                         tmp = b[2]; b[2] = b[1]; b[1] = tmp;
2052                 }
2053                 return true;
2054         }
2055 }
2056
2057 FLAC__bool
2058 read_big_endian_uint16(FILE *f, FLAC__uint16 *val, FLAC__bool eof_ok, const char *fn)
2059 {
2060         unsigned char buf[4];
2061         size_t bytes_read= fread(buf, 1, 2, f);
2062
2063         if(bytes_read==0U && eof_ok)
2064                 return true;
2065         else if(bytes_read<2U) {
2066                 fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2067                 return false;
2068         }
2069
2070         /* this is independent of host endianness */
2071         *val= (FLAC__uint16)(buf[0])<<8 | buf[1];
2072
2073         return true;
2074 }
2075
2076 FLAC__bool
2077 read_big_endian_uint32(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn)
2078 {
2079         unsigned char buf[4];
2080         size_t bytes_read= fread(buf, 1, 4, f);
2081
2082         if(bytes_read==0U && eof_ok)
2083                 return true;
2084         else if(bytes_read<4U) {
2085                 fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2086                 return false;
2087         }
2088
2089         /* this is independent of host endianness */
2090         *val= (FLAC__uint32)(buf[0])<<24 | (FLAC__uint32)(buf[1])<<16 |
2091                 (FLAC__uint32)(buf[2])<<8 | buf[3];
2092
2093         return true;
2094 }
2095
2096 FLAC__bool
2097 read_sane_extended(FILE *f, FLAC__uint32 *val, FLAC__bool eof_ok, const char *fn)
2098         /* Read an IEEE 754 80-bit (aka SANE) extended floating point value from 'f',
2099          * convert it into an integral value and store in 'val'.  Return false if only
2100          * between 1 and 9 bytes remain in 'f', if 0 bytes remain in 'f' and 'eof_ok' is
2101          * false, or if the value is negative, between zero and one, or too large to be
2102          * represented by 'val'; return true otherwise.
2103          */
2104 {
2105         unsigned int i;
2106         unsigned char buf[10];
2107         size_t bytes_read= fread(buf, 1U, 10U, f);
2108         FLAC__int16 e= ((FLAC__uint16)(buf[0])<<8 | (FLAC__uint16)(buf[1]))-0x3FFF;
2109         FLAC__int16 shift= 63-e;
2110         FLAC__uint64 p= 0U;
2111
2112         if(bytes_read==0U && eof_ok)
2113                 return true;
2114         else if(bytes_read<10U) {
2115                 fprintf(stderr, "%s: ERROR: unexpected EOF\n", fn);
2116                 return false;
2117         }
2118         else if((buf[0]>>7)==1U || e<0 || e>63) {
2119                 fprintf(stderr, "%s: ERROR: invalid floating-point value\n", fn);
2120                 return false;
2121         }
2122
2123         for(i= 0U; i<8U; ++i)
2124                 p|= (FLAC__uint64)(buf[i+2])<<(56U-i*8);
2125         *val= (FLAC__uint32)((p>>shift)+(p>>(shift-1) & 0x1));
2126
2127         return true;
2128 }