fix ANSI C violation w.r.t. const arrays
[flac.git] / src / libFLAC / stream_encoder.c
1 /* libFLAC - Free Lossless Audio Codec library
2  * Copyright (C) 2000,2001,2002,2003,2004,2005  Josh Coalson
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * - Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *
11  * - Redistributions in binary form must reproduce the above copyright
12  * notice, this list of conditions and the following disclaimer in the
13  * documentation and/or other materials provided with the distribution.
14  *
15  * - Neither the name of the Xiph.org Foundation nor the names of its
16  * contributors may be used to endorse or promote products derived from
17  * this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30  */
31
32 #include <limits.h>
33 #include <stdio.h>
34 #include <stdlib.h> /* for malloc() */
35 #include <string.h> /* for memcpy() */
36 #include "FLAC/assert.h"
37 #include "FLAC/stream_decoder.h"
38 #include "protected/stream_encoder.h"
39 #include "private/bitbuffer.h"
40 #include "private/bitmath.h"
41 #include "private/crc.h"
42 #include "private/cpu.h"
43 #include "private/fixed.h"
44 #include "private/format.h"
45 #include "private/lpc.h"
46 #include "private/md5.h"
47 #include "private/memory.h"
48 #include "private/stream_encoder_framing.h"
49
50 #ifdef HAVE_CONFIG_H
51 #include <config.h>
52 #endif
53
54 #ifdef min
55 #undef min
56 #endif
57 #define min(x,y) ((x)<(y)?(x):(y))
58
59 #ifdef max
60 #undef max
61 #endif
62 #define max(x,y) ((x)>(y)?(x):(y))
63
64 typedef struct {
65         FLAC__int32 *data[FLAC__MAX_CHANNELS];
66         unsigned size; /* of each data[] in samples */
67         unsigned tail;
68 } verify_input_fifo;
69
70 typedef struct {
71         const FLAC__byte *data;
72         unsigned capacity;
73         unsigned bytes;
74 } verify_output;
75
76 typedef enum {
77         ENCODER_IN_MAGIC = 0,
78         ENCODER_IN_METADATA = 1,
79         ENCODER_IN_AUDIO = 2
80 } EncoderStateHint;
81
82 /***********************************************************************
83  *
84  * Private class method prototypes
85  *
86  ***********************************************************************/
87
88 static void set_defaults_(FLAC__StreamEncoder *encoder);
89 static void free_(FLAC__StreamEncoder *encoder);
90 static FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size);
91 static FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples);
92 static FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame);
93 static FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame);
94
95 static FLAC__bool process_subframe_(
96         FLAC__StreamEncoder *encoder,
97         unsigned min_partition_order,
98         unsigned max_partition_order,
99         FLAC__bool precompute_partition_sums,
100         const FLAC__FrameHeader *frame_header,
101         unsigned subframe_bps,
102         const FLAC__int32 integer_signal[],
103 #ifndef FLAC__INTEGER_ONLY_LIBRARY
104         const FLAC__real real_signal[],
105 #endif
106         FLAC__Subframe *subframe[2],
107         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
108         FLAC__int32 *residual[2],
109         unsigned *best_subframe,
110         unsigned *best_bits
111 );
112
113 static FLAC__bool add_subframe_(
114         FLAC__StreamEncoder *encoder,
115         const FLAC__FrameHeader *frame_header,
116         unsigned subframe_bps,
117         const FLAC__Subframe *subframe,
118         FLAC__BitBuffer *frame
119 );
120
121 static unsigned evaluate_constant_subframe_(
122         const FLAC__int32 signal,
123         unsigned subframe_bps,
124         FLAC__Subframe *subframe
125 );
126
127 static unsigned evaluate_fixed_subframe_(
128         FLAC__StreamEncoder *encoder,
129         const FLAC__int32 signal[],
130         FLAC__int32 residual[],
131         FLAC__uint32 abs_residual[],
132         FLAC__uint64 abs_residual_partition_sums[],
133         unsigned raw_bits_per_partition[],
134         unsigned blocksize,
135         unsigned subframe_bps,
136         unsigned order,
137         unsigned rice_parameter,
138         unsigned min_partition_order,
139         unsigned max_partition_order,
140         FLAC__bool precompute_partition_sums,
141         FLAC__bool do_escape_coding,
142         unsigned rice_parameter_search_dist,
143         FLAC__Subframe *subframe,
144         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
145 );
146
147 #ifndef FLAC__INTEGER_ONLY_LIBRARY
148 static unsigned evaluate_lpc_subframe_(
149         FLAC__StreamEncoder *encoder,
150         const FLAC__int32 signal[],
151         FLAC__int32 residual[],
152         FLAC__uint32 abs_residual[],
153         FLAC__uint64 abs_residual_partition_sums[],
154         unsigned raw_bits_per_partition[],
155         const FLAC__real lp_coeff[],
156         unsigned blocksize,
157         unsigned subframe_bps,
158         unsigned order,
159         unsigned qlp_coeff_precision,
160         unsigned rice_parameter,
161         unsigned min_partition_order,
162         unsigned max_partition_order,
163         FLAC__bool precompute_partition_sums,
164         FLAC__bool do_escape_coding,
165         unsigned rice_parameter_search_dist,
166         FLAC__Subframe *subframe,
167         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
168 );
169 #endif
170
171 static unsigned evaluate_verbatim_subframe_(
172         const FLAC__int32 signal[],
173         unsigned blocksize,
174         unsigned subframe_bps,
175         FLAC__Subframe *subframe
176 );
177
178 static unsigned find_best_partition_order_(
179         struct FLAC__StreamEncoderPrivate *private_,
180         const FLAC__int32 residual[],
181         FLAC__uint32 abs_residual[],
182         FLAC__uint64 abs_residual_partition_sums[],
183         unsigned raw_bits_per_partition[],
184         unsigned residual_samples,
185         unsigned predictor_order,
186         unsigned rice_parameter,
187         unsigned min_partition_order,
188         unsigned max_partition_order,
189         FLAC__bool precompute_partition_sums,
190         FLAC__bool do_escape_coding,
191         unsigned rice_parameter_search_dist,
192         FLAC__EntropyCodingMethod_PartitionedRice *best_partitioned_rice
193 );
194
195 static void precompute_partition_info_sums_(
196         const FLAC__uint32 abs_residual[],
197         FLAC__uint64 abs_residual_partition_sums[],
198         unsigned residual_samples,
199         unsigned predictor_order,
200         unsigned min_partition_order,
201         unsigned max_partition_order
202 );
203
204 static void precompute_partition_info_escapes_(
205         const FLAC__int32 residual[],
206         unsigned raw_bits_per_partition[],
207         unsigned residual_samples,
208         unsigned predictor_order,
209         unsigned min_partition_order,
210         unsigned max_partition_order
211 );
212
213 #ifdef DONT_ESTIMATE_RICE_BITS
214 static FLAC__bool set_partitioned_rice_(
215         const FLAC__uint32 abs_residual[],
216         const FLAC__int32 residual[],
217         const unsigned residual_samples,
218         const unsigned predictor_order,
219         const unsigned suggested_rice_parameter,
220         const unsigned rice_parameter_search_dist,
221         const unsigned partition_order,
222         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
223         unsigned *bits
224 );
225
226 static FLAC__bool set_partitioned_rice_with_precompute_(
227         const FLAC__int32 residual[],
228         const FLAC__uint64 abs_residual_partition_sums[],
229         const unsigned raw_bits_per_partition[],
230         const unsigned residual_samples,
231         const unsigned predictor_order,
232         const unsigned suggested_rice_parameter,
233         const unsigned rice_parameter_search_dist,
234         const unsigned partition_order,
235         const FLAC__bool search_for_escapes,
236         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
237         unsigned *bits
238 );
239 #else
240 static FLAC__bool set_partitioned_rice_(
241         const FLAC__uint32 abs_residual[],
242         const unsigned residual_samples,
243         const unsigned predictor_order,
244         const unsigned suggested_rice_parameter,
245         const unsigned rice_parameter_search_dist,
246         const unsigned partition_order,
247         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
248         unsigned *bits
249 );
250
251 static FLAC__bool set_partitioned_rice_with_precompute_(
252         const FLAC__uint32 abs_residual[],
253         const FLAC__uint64 abs_residual_partition_sums[],
254         const unsigned raw_bits_per_partition[],
255         const unsigned residual_samples,
256         const unsigned predictor_order,
257         const unsigned suggested_rice_parameter,
258         const unsigned rice_parameter_search_dist,
259         const unsigned partition_order,
260         const FLAC__bool search_for_escapes,
261         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
262         unsigned *bits
263 );
264 #endif
265
266 static unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples);
267
268 /* verify-related routines: */
269 static void append_to_verify_fifo_(
270         verify_input_fifo *fifo,
271         const FLAC__int32 * const input[],
272         unsigned input_offset,
273         unsigned channels,
274         unsigned wide_samples
275 );
276
277 static void append_to_verify_fifo_interleaved_(
278         verify_input_fifo *fifo,
279         const FLAC__int32 input[],
280         unsigned input_offset,
281         unsigned channels,
282         unsigned wide_samples
283 );
284
285 static FLAC__StreamDecoderReadStatus verify_read_callback_(
286         const FLAC__StreamDecoder *decoder,
287         FLAC__byte buffer[],
288         unsigned *bytes,
289         void *client_data
290 );
291
292 static FLAC__StreamDecoderWriteStatus verify_write_callback_(
293         const FLAC__StreamDecoder *decoder,
294         const FLAC__Frame *frame,
295         const FLAC__int32 * const buffer[],
296         void *client_data
297 );
298
299 static void verify_metadata_callback_(
300         const FLAC__StreamDecoder *decoder,
301         const FLAC__StreamMetadata *metadata,
302         void *client_data
303 );
304
305 static void verify_error_callback_(
306         const FLAC__StreamDecoder *decoder,
307         FLAC__StreamDecoderErrorStatus status,
308         void *client_data
309 );
310
311
312 /***********************************************************************
313  *
314  * Private class data
315  *
316  ***********************************************************************/
317
318 typedef struct FLAC__StreamEncoderPrivate {
319         unsigned input_capacity;                          /* current size (in samples) of the signal and residual buffers */
320         FLAC__int32 *integer_signal[FLAC__MAX_CHANNELS];  /* the integer version of the input signal */
321         FLAC__int32 *integer_signal_mid_side[2];          /* the integer version of the mid-side input signal (stereo only) */
322 #ifndef FLAC__INTEGER_ONLY_LIBRARY
323         FLAC__real *real_signal[FLAC__MAX_CHANNELS];      /* the floating-point version of the input signal */
324         FLAC__real *real_signal_mid_side[2];              /* the floating-point version of the mid-side input signal (stereo only) */
325 #endif
326         unsigned subframe_bps[FLAC__MAX_CHANNELS];        /* the effective bits per sample of the input signal (stream bps - wasted bits) */
327         unsigned subframe_bps_mid_side[2];                /* the effective bits per sample of the mid-side input signal (stream bps - wasted bits + 0/1) */
328         FLAC__int32 *residual_workspace[FLAC__MAX_CHANNELS][2]; /* each channel has a candidate and best workspace where the subframe residual signals will be stored */
329         FLAC__int32 *residual_workspace_mid_side[2][2];
330         FLAC__Subframe subframe_workspace[FLAC__MAX_CHANNELS][2];
331         FLAC__Subframe subframe_workspace_mid_side[2][2];
332         FLAC__Subframe *subframe_workspace_ptr[FLAC__MAX_CHANNELS][2];
333         FLAC__Subframe *subframe_workspace_ptr_mid_side[2][2];
334         FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents_workspace[FLAC__MAX_CHANNELS][2];
335         FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents_workspace_mid_side[FLAC__MAX_CHANNELS][2];
336         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents_workspace_ptr[FLAC__MAX_CHANNELS][2];
337         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents_workspace_ptr_mid_side[FLAC__MAX_CHANNELS][2];
338         unsigned best_subframe[FLAC__MAX_CHANNELS];       /* index into the above workspaces */
339         unsigned best_subframe_mid_side[2];
340         unsigned best_subframe_bits[FLAC__MAX_CHANNELS];  /* size in bits of the best subframe for each channel */
341         unsigned best_subframe_bits_mid_side[2];
342         FLAC__uint32 *abs_residual;                       /* workspace where abs(candidate residual) is stored */
343         FLAC__uint64 *abs_residual_partition_sums;        /* workspace where the sum of abs(candidate residual) for each partition is stored */
344         unsigned *raw_bits_per_partition;                 /* workspace where the sum of silog2(candidate residual) for each partition is stored */
345         FLAC__BitBuffer *frame;                           /* the current frame being worked on */
346         unsigned loose_mid_side_stereo_frames;            /* rounded number of frames the encoder will use before trying both independent and mid/side frames again */
347         unsigned loose_mid_side_stereo_frame_count;       /* number of frames using the current channel assignment */
348         FLAC__ChannelAssignment last_channel_assignment;
349         FLAC__StreamMetadata metadata;
350         unsigned current_sample_number;
351         unsigned current_frame_number;
352         struct FLAC__MD5Context md5context;
353         FLAC__CPUInfo cpuinfo;
354 #ifndef FLAC__INTEGER_ONLY_LIBRARY
355         unsigned (*local_fixed_compute_best_predictor)(const FLAC__int32 data[], unsigned data_len, FLAC__float residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1]);
356 #else
357         unsigned (*local_fixed_compute_best_predictor)(const FLAC__int32 data[], unsigned data_len, FLAC__fixedpoint residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1]);
358 #endif
359 #ifndef FLAC__INTEGER_ONLY_LIBRARY
360         void (*local_lpc_compute_autocorrelation)(const FLAC__real data[], unsigned data_len, unsigned lag, FLAC__real autoc[]);
361         void (*local_lpc_compute_residual_from_qlp_coefficients)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
362         void (*local_lpc_compute_residual_from_qlp_coefficients_64bit)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
363         void (*local_lpc_compute_residual_from_qlp_coefficients_16bit)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
364 #endif
365         FLAC__bool use_wide_by_block;          /* use slow 64-bit versions of some functions because of the block size */
366         FLAC__bool use_wide_by_partition;      /* use slow 64-bit versions of some functions because of the min partition order and blocksize */
367         FLAC__bool use_wide_by_order;          /* use slow 64-bit versions of some functions because of the lpc order */
368         FLAC__bool precompute_partition_sums;  /* our initial guess as to whether precomputing the partitions sums will be a speed improvement */
369         FLAC__bool disable_constant_subframes;
370         FLAC__bool disable_fixed_subframes;
371         FLAC__bool disable_verbatim_subframes;
372         FLAC__StreamEncoderWriteCallback write_callback;
373         FLAC__StreamEncoderMetadataCallback metadata_callback;
374         void *client_data;
375         /* unaligned (original) pointers to allocated data */
376         FLAC__int32 *integer_signal_unaligned[FLAC__MAX_CHANNELS];
377         FLAC__int32 *integer_signal_mid_side_unaligned[2];
378 #ifndef FLAC__INTEGER_ONLY_LIBRARY
379         FLAC__real *real_signal_unaligned[FLAC__MAX_CHANNELS];
380         FLAC__real *real_signal_mid_side_unaligned[2];
381 #endif
382         FLAC__int32 *residual_workspace_unaligned[FLAC__MAX_CHANNELS][2];
383         FLAC__int32 *residual_workspace_mid_side_unaligned[2][2];
384         FLAC__uint32 *abs_residual_unaligned;
385         FLAC__uint64 *abs_residual_partition_sums_unaligned;
386         unsigned *raw_bits_per_partition_unaligned;
387         /*
388          * These fields have been moved here from private function local
389          * declarations merely to save stack space during encoding.
390          */
391 #ifndef FLAC__INTEGER_ONLY_LIBRARY
392         FLAC__real lp_coeff[FLAC__MAX_LPC_ORDER][FLAC__MAX_LPC_ORDER]; /* from process_subframe_() */
393 #endif
394         FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents_extra[2]; /* from find_best_partition_order_() */
395         /*
396          * The data for the verify section
397          */
398         struct {
399                 FLAC__StreamDecoder *decoder;
400                 EncoderStateHint state_hint;
401                 FLAC__bool needs_magic_hack;
402                 verify_input_fifo input_fifo;
403                 verify_output output;
404                 struct {
405                         FLAC__uint64 absolute_sample;
406                         unsigned frame_number;
407                         unsigned channel;
408                         unsigned sample;
409                         FLAC__int32 expected;
410                         FLAC__int32 got;
411                 } error_stats;
412         } verify;
413         FLAC__bool is_being_deleted; /* if true, call to ..._finish() from ..._delete() will not call the callbacks */
414 } FLAC__StreamEncoderPrivate;
415
416 /***********************************************************************
417  *
418  * Public static class data
419  *
420  ***********************************************************************/
421
422 FLAC_API const char * const FLAC__StreamEncoderStateString[] = {
423         "FLAC__STREAM_ENCODER_OK",
424         "FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR",
425         "FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA",
426         "FLAC__STREAM_ENCODER_INVALID_CALLBACK",
427         "FLAC__STREAM_ENCODER_INVALID_NUMBER_OF_CHANNELS",
428         "FLAC__STREAM_ENCODER_INVALID_BITS_PER_SAMPLE",
429         "FLAC__STREAM_ENCODER_INVALID_SAMPLE_RATE",
430         "FLAC__STREAM_ENCODER_INVALID_BLOCK_SIZE",
431         "FLAC__STREAM_ENCODER_INVALID_MAX_LPC_ORDER",
432         "FLAC__STREAM_ENCODER_INVALID_QLP_COEFF_PRECISION",
433         "FLAC__STREAM_ENCODER_MID_SIDE_CHANNELS_MISMATCH",
434         "FLAC__STREAM_ENCODER_MID_SIDE_SAMPLE_SIZE_MISMATCH",
435         "FLAC__STREAM_ENCODER_ILLEGAL_MID_SIDE_FORCE",
436         "FLAC__STREAM_ENCODER_BLOCK_SIZE_TOO_SMALL_FOR_LPC_ORDER",
437         "FLAC__STREAM_ENCODER_NOT_STREAMABLE",
438         "FLAC__STREAM_ENCODER_FRAMING_ERROR",
439         "FLAC__STREAM_ENCODER_INVALID_METADATA",
440         "FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING",
441         "FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_WRITING",
442         "FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR",
443         "FLAC__STREAM_ENCODER_ALREADY_INITIALIZED",
444         "FLAC__STREAM_ENCODER_UNINITIALIZED"
445 };
446
447 FLAC_API const char * const FLAC__StreamEncoderWriteStatusString[] = {
448         "FLAC__STREAM_ENCODER_WRITE_STATUS_OK",
449         "FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR"
450 };
451
452 /***********************************************************************
453  *
454  * Class constructor/destructor
455  *
456  */
457 FLAC_API FLAC__StreamEncoder *FLAC__stream_encoder_new()
458 {
459         FLAC__StreamEncoder *encoder;
460         unsigned i;
461
462         FLAC__ASSERT(sizeof(int) >= 4); /* we want to die right away if this is not true */
463
464         encoder = (FLAC__StreamEncoder*)calloc(1, sizeof(FLAC__StreamEncoder));
465         if(encoder == 0) {
466                 return 0;
467         }
468
469         encoder->protected_ = (FLAC__StreamEncoderProtected*)calloc(1, sizeof(FLAC__StreamEncoderProtected));
470         if(encoder->protected_ == 0) {
471                 free(encoder);
472                 return 0;
473         }
474
475         encoder->private_ = (FLAC__StreamEncoderPrivate*)calloc(1, sizeof(FLAC__StreamEncoderPrivate));
476         if(encoder->private_ == 0) {
477                 free(encoder->protected_);
478                 free(encoder);
479                 return 0;
480         }
481
482         encoder->private_->frame = FLAC__bitbuffer_new();
483         if(encoder->private_->frame == 0) {
484                 free(encoder->private_);
485                 free(encoder->protected_);
486                 free(encoder);
487                 return 0;
488         }
489
490         set_defaults_(encoder);
491
492         encoder->private_->is_being_deleted = false;
493
494         for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
495                 encoder->private_->subframe_workspace_ptr[i][0] = &encoder->private_->subframe_workspace[i][0];
496                 encoder->private_->subframe_workspace_ptr[i][1] = &encoder->private_->subframe_workspace[i][1];
497         }
498         for(i = 0; i < 2; i++) {
499                 encoder->private_->subframe_workspace_ptr_mid_side[i][0] = &encoder->private_->subframe_workspace_mid_side[i][0];
500                 encoder->private_->subframe_workspace_ptr_mid_side[i][1] = &encoder->private_->subframe_workspace_mid_side[i][1];
501         }
502         for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
503                 encoder->private_->partitioned_rice_contents_workspace_ptr[i][0] = &encoder->private_->partitioned_rice_contents_workspace[i][0];
504                 encoder->private_->partitioned_rice_contents_workspace_ptr[i][1] = &encoder->private_->partitioned_rice_contents_workspace[i][1];
505         }
506         for(i = 0; i < 2; i++) {
507                 encoder->private_->partitioned_rice_contents_workspace_ptr_mid_side[i][0] = &encoder->private_->partitioned_rice_contents_workspace_mid_side[i][0];
508                 encoder->private_->partitioned_rice_contents_workspace_ptr_mid_side[i][1] = &encoder->private_->partitioned_rice_contents_workspace_mid_side[i][1];
509         }
510
511         for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
512                 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_workspace[i][0]);
513                 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_workspace[i][1]);
514         }
515         for(i = 0; i < 2; i++) {
516                 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_workspace_mid_side[i][0]);
517                 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_workspace_mid_side[i][1]);
518         }
519         for(i = 0; i < 2; i++)
520                 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_extra[i]);
521
522         encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
523
524         return encoder;
525 }
526
527 FLAC_API void FLAC__stream_encoder_delete(FLAC__StreamEncoder *encoder)
528 {
529         unsigned i;
530
531         FLAC__ASSERT(0 != encoder);
532         FLAC__ASSERT(0 != encoder->protected_);
533         FLAC__ASSERT(0 != encoder->private_);
534         FLAC__ASSERT(0 != encoder->private_->frame);
535
536         encoder->private_->is_being_deleted = true;
537
538         FLAC__stream_encoder_finish(encoder);
539
540         if(0 != encoder->private_->verify.decoder)
541                 FLAC__stream_decoder_delete(encoder->private_->verify.decoder);
542
543         for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
544                 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_workspace[i][0]);
545                 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_workspace[i][1]);
546         }
547         for(i = 0; i < 2; i++) {
548                 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_workspace_mid_side[i][0]);
549                 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_workspace_mid_side[i][1]);
550         }
551         for(i = 0; i < 2; i++)
552                 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_extra[i]);
553
554         FLAC__bitbuffer_delete(encoder->private_->frame);
555         free(encoder->private_);
556         free(encoder->protected_);
557         free(encoder);
558 }
559
560 /***********************************************************************
561  *
562  * Public class methods
563  *
564  ***********************************************************************/
565
566 FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_init(FLAC__StreamEncoder *encoder)
567 {
568         unsigned i;
569         FLAC__bool metadata_has_seektable, metadata_has_vorbis_comment;
570
571         FLAC__ASSERT(0 != encoder);
572
573         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
574                 return encoder->protected_->state = FLAC__STREAM_ENCODER_ALREADY_INITIALIZED;
575
576         encoder->protected_->state = FLAC__STREAM_ENCODER_OK;
577
578         if(0 == encoder->private_->write_callback || 0 == encoder->private_->metadata_callback)
579                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_CALLBACK;
580
581         if(encoder->protected_->channels == 0 || encoder->protected_->channels > FLAC__MAX_CHANNELS)
582                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_NUMBER_OF_CHANNELS;
583
584         if(encoder->protected_->do_mid_side_stereo && encoder->protected_->channels != 2)
585                 return encoder->protected_->state = FLAC__STREAM_ENCODER_MID_SIDE_CHANNELS_MISMATCH;
586
587         if(encoder->protected_->loose_mid_side_stereo && !encoder->protected_->do_mid_side_stereo)
588                 return encoder->protected_->state = FLAC__STREAM_ENCODER_ILLEGAL_MID_SIDE_FORCE;
589
590         if(encoder->protected_->bits_per_sample >= 32)
591                 encoder->protected_->do_mid_side_stereo = false; /* since we do 32-bit math, the side channel would have 33 bps and overflow */
592
593         if(encoder->protected_->bits_per_sample < FLAC__MIN_BITS_PER_SAMPLE || encoder->protected_->bits_per_sample > FLAC__REFERENCE_CODEC_MAX_BITS_PER_SAMPLE)
594                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_BITS_PER_SAMPLE;
595
596         if(!FLAC__format_sample_rate_is_valid(encoder->protected_->sample_rate))
597                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_SAMPLE_RATE;
598
599         if(encoder->protected_->blocksize < FLAC__MIN_BLOCK_SIZE || encoder->protected_->blocksize > FLAC__MAX_BLOCK_SIZE)
600                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_BLOCK_SIZE;
601
602         if(encoder->protected_->max_lpc_order > FLAC__MAX_LPC_ORDER)
603                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_MAX_LPC_ORDER;
604
605         if(encoder->protected_->blocksize < encoder->protected_->max_lpc_order)
606                 return encoder->protected_->state = FLAC__STREAM_ENCODER_BLOCK_SIZE_TOO_SMALL_FOR_LPC_ORDER;
607
608         if(encoder->protected_->qlp_coeff_precision == 0) {
609                 if(encoder->protected_->bits_per_sample < 16) {
610                         /* @@@ need some data about how to set this here w.r.t. blocksize and sample rate */
611                         /* @@@ until then we'll make a guess */
612                         encoder->protected_->qlp_coeff_precision = max(FLAC__MIN_QLP_COEFF_PRECISION, 2 + encoder->protected_->bits_per_sample / 2);
613                 }
614                 else if(encoder->protected_->bits_per_sample == 16) {
615                         if(encoder->protected_->blocksize <= 192)
616                                 encoder->protected_->qlp_coeff_precision = 7;
617                         else if(encoder->protected_->blocksize <= 384)
618                                 encoder->protected_->qlp_coeff_precision = 8;
619                         else if(encoder->protected_->blocksize <= 576)
620                                 encoder->protected_->qlp_coeff_precision = 9;
621                         else if(encoder->protected_->blocksize <= 1152)
622                                 encoder->protected_->qlp_coeff_precision = 10;
623                         else if(encoder->protected_->blocksize <= 2304)
624                                 encoder->protected_->qlp_coeff_precision = 11;
625                         else if(encoder->protected_->blocksize <= 4608)
626                                 encoder->protected_->qlp_coeff_precision = 12;
627                         else
628                                 encoder->protected_->qlp_coeff_precision = 13;
629                 }
630                 else {
631                         if(encoder->protected_->blocksize <= 384)
632                                 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION-2;
633                         else if(encoder->protected_->blocksize <= 1152)
634                                 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION-1;
635                         else
636                                 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
637                 }
638                 FLAC__ASSERT(encoder->protected_->qlp_coeff_precision <= FLAC__MAX_QLP_COEFF_PRECISION);
639         }
640         else if(encoder->protected_->qlp_coeff_precision < FLAC__MIN_QLP_COEFF_PRECISION || encoder->protected_->qlp_coeff_precision > FLAC__MAX_QLP_COEFF_PRECISION)
641                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_QLP_COEFF_PRECISION;
642
643         if(encoder->protected_->streamable_subset) {
644                 if(
645                         encoder->protected_->blocksize != 192 &&
646                         encoder->protected_->blocksize != 576 &&
647                         encoder->protected_->blocksize != 1152 &&
648                         encoder->protected_->blocksize != 2304 &&
649                         encoder->protected_->blocksize != 4608 &&
650                         encoder->protected_->blocksize != 256 &&
651                         encoder->protected_->blocksize != 512 &&
652                         encoder->protected_->blocksize != 1024 &&
653                         encoder->protected_->blocksize != 2048 &&
654                         encoder->protected_->blocksize != 4096 &&
655                         encoder->protected_->blocksize != 8192 &&
656                         encoder->protected_->blocksize != 16384
657                 )
658                         return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
659                 if(
660                         encoder->protected_->sample_rate != 8000 &&
661                         encoder->protected_->sample_rate != 16000 &&
662                         encoder->protected_->sample_rate != 22050 &&
663                         encoder->protected_->sample_rate != 24000 &&
664                         encoder->protected_->sample_rate != 32000 &&
665                         encoder->protected_->sample_rate != 44100 &&
666                         encoder->protected_->sample_rate != 48000 &&
667                         encoder->protected_->sample_rate != 96000
668                 )
669                         return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
670                 if(
671                         encoder->protected_->bits_per_sample != 8 &&
672                         encoder->protected_->bits_per_sample != 12 &&
673                         encoder->protected_->bits_per_sample != 16 &&
674                         encoder->protected_->bits_per_sample != 20 &&
675                         encoder->protected_->bits_per_sample != 24
676                 )
677                         return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
678                 if(encoder->protected_->max_residual_partition_order > FLAC__SUBSET_MAX_RICE_PARTITION_ORDER)
679                         return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
680         }
681
682         if(encoder->protected_->max_residual_partition_order >= (1u << FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN))
683                 encoder->protected_->max_residual_partition_order = (1u << FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN) - 1;
684         if(encoder->protected_->min_residual_partition_order >= encoder->protected_->max_residual_partition_order)
685                 encoder->protected_->min_residual_partition_order = encoder->protected_->max_residual_partition_order;
686
687         /* validate metadata */
688         if(0 == encoder->protected_->metadata && encoder->protected_->num_metadata_blocks > 0)
689                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
690         metadata_has_seektable = false;
691         metadata_has_vorbis_comment = false;
692         for(i = 0; i < encoder->protected_->num_metadata_blocks; i++) {
693                 if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_STREAMINFO)
694                         return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
695                 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_SEEKTABLE) {
696                         if(metadata_has_seektable) /* only one is allowed */
697                                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
698                         metadata_has_seektable = true;
699                         if(!FLAC__format_seektable_is_legal(&encoder->protected_->metadata[i]->data.seek_table))
700                                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
701                 }
702                 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_VORBIS_COMMENT) {
703                         if(metadata_has_vorbis_comment) /* only one is allowed */
704                                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
705                         metadata_has_vorbis_comment = true;
706                 }
707                 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_CUESHEET) {
708                         if(!FLAC__format_cuesheet_is_legal(&encoder->protected_->metadata[i]->data.cue_sheet, encoder->protected_->metadata[i]->data.cue_sheet.is_cd, /*violation=*/0))
709                                 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
710                 }
711         }
712
713         encoder->private_->input_capacity = 0;
714         for(i = 0; i < encoder->protected_->channels; i++) {
715                 encoder->private_->integer_signal_unaligned[i] = encoder->private_->integer_signal[i] = 0;
716 #ifndef FLAC__INTEGER_ONLY_LIBRARY
717                 encoder->private_->real_signal_unaligned[i] = encoder->private_->real_signal[i] = 0;
718 #endif
719         }
720         for(i = 0; i < 2; i++) {
721                 encoder->private_->integer_signal_mid_side_unaligned[i] = encoder->private_->integer_signal_mid_side[i] = 0;
722 #ifndef FLAC__INTEGER_ONLY_LIBRARY
723                 encoder->private_->real_signal_mid_side_unaligned[i] = encoder->private_->real_signal_mid_side[i] = 0;
724 #endif
725         }
726         for(i = 0; i < encoder->protected_->channels; i++) {
727                 encoder->private_->residual_workspace_unaligned[i][0] = encoder->private_->residual_workspace[i][0] = 0;
728                 encoder->private_->residual_workspace_unaligned[i][1] = encoder->private_->residual_workspace[i][1] = 0;
729                 encoder->private_->best_subframe[i] = 0;
730         }
731         for(i = 0; i < 2; i++) {
732                 encoder->private_->residual_workspace_mid_side_unaligned[i][0] = encoder->private_->residual_workspace_mid_side[i][0] = 0;
733                 encoder->private_->residual_workspace_mid_side_unaligned[i][1] = encoder->private_->residual_workspace_mid_side[i][1] = 0;
734                 encoder->private_->best_subframe_mid_side[i] = 0;
735         }
736         encoder->private_->abs_residual_unaligned = encoder->private_->abs_residual = 0;
737         encoder->private_->abs_residual_partition_sums_unaligned = encoder->private_->abs_residual_partition_sums = 0;
738         encoder->private_->raw_bits_per_partition_unaligned = encoder->private_->raw_bits_per_partition = 0;
739 #ifndef FLAC__INTEGER_ONLY_LIBRARY
740         encoder->private_->loose_mid_side_stereo_frames = (unsigned)((FLAC__double)encoder->protected_->sample_rate * 0.4 / (FLAC__double)encoder->protected_->blocksize + 0.5);
741 #else
742         /* 26214 is the approximate fixed-point equivalent to 0.4 (0.4 * 2^16) */
743         /* sample rate can be up to 655350 Hz, and thus use 20 bits, so we do the multiply&divide by hand */
744         FLAC__ASSERT(FLAC__MAX_SAMPLE_RATE <= 655350);
745         FLAC__ASSERT(FLAC__MAX_BLOCK_SIZE <= 65535);
746         FLAC__ASSERT(encoder->protected_->sample_rate <= 655350);
747         FLAC__ASSERT(encoder->protected_->blocksize <= 65535);
748         encoder->private_->loose_mid_side_stereo_frames = (unsigned)FLAC__fixedpoint_trunc((((FLAC__uint64)(encoder->protected_->sample_rate) * (FLAC__uint64)(26214)) << 16) / (encoder->protected_->blocksize<<16) + FLAC__FP_ONE_HALF);
749 #endif
750         if(encoder->private_->loose_mid_side_stereo_frames == 0)
751                 encoder->private_->loose_mid_side_stereo_frames = 1;
752         encoder->private_->loose_mid_side_stereo_frame_count = 0;
753         encoder->private_->current_sample_number = 0;
754         encoder->private_->current_frame_number = 0;
755
756         encoder->private_->use_wide_by_block = (encoder->protected_->bits_per_sample + FLAC__bitmath_ilog2(encoder->protected_->blocksize)+1 > 30);
757         encoder->private_->use_wide_by_order = (encoder->protected_->bits_per_sample + FLAC__bitmath_ilog2(max(encoder->protected_->max_lpc_order, FLAC__MAX_FIXED_ORDER))+1 > 30); /*@@@ need to use this? */
758         encoder->private_->use_wide_by_partition = (false); /*@@@ need to set this */
759
760         /*
761          * get the CPU info and set the function pointers
762          */
763         FLAC__cpu_info(&encoder->private_->cpuinfo);
764         /* first default to the non-asm routines */
765 #ifndef FLAC__INTEGER_ONLY_LIBRARY
766         encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation;
767 #endif
768         encoder->private_->local_fixed_compute_best_predictor = FLAC__fixed_compute_best_predictor;
769 #ifndef FLAC__INTEGER_ONLY_LIBRARY
770         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients = FLAC__lpc_compute_residual_from_qlp_coefficients;
771         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_64bit = FLAC__lpc_compute_residual_from_qlp_coefficients_wide;
772         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit = FLAC__lpc_compute_residual_from_qlp_coefficients;
773 #endif
774         /* now override with asm where appropriate */
775 #ifndef FLAC__INTEGER_ONLY_LIBRARY
776 # ifndef FLAC__NO_ASM
777         if(encoder->private_->cpuinfo.use_asm) {
778 #  ifdef FLAC__CPU_IA32
779                 FLAC__ASSERT(encoder->private_->cpuinfo.type == FLAC__CPUINFO_TYPE_IA32);
780 #   ifdef FLAC__HAS_NASM
781 #    ifdef FLAC__SSE_OS
782                 if(encoder->private_->cpuinfo.data.ia32.sse) {
783                         if(encoder->protected_->max_lpc_order < 4)
784                                 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32_sse_lag_4;
785                         else if(encoder->protected_->max_lpc_order < 8)
786                                 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32_sse_lag_8;
787                         else if(encoder->protected_->max_lpc_order < 12)
788                                 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32_sse_lag_12;
789                         else
790                                 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32;
791                 }
792                 else
793 #    endif /* FLAC__SSE_OS */
794                 if(encoder->private_->cpuinfo.data.ia32._3dnow)
795                         encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32_3dnow;
796                 else
797                         encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32;
798                 if(encoder->private_->cpuinfo.data.ia32.mmx) {
799                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients = FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32;
800                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit = FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32_mmx;
801                 }
802                 else {
803                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients = FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32;
804                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit = FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32;
805                 }
806                 if(encoder->private_->cpuinfo.data.ia32.mmx && encoder->private_->cpuinfo.data.ia32.cmov)
807                         encoder->private_->local_fixed_compute_best_predictor = FLAC__fixed_compute_best_predictor_asm_ia32_mmx_cmov;
808 #   endif /* FLAC__HAS_NASM */
809 #  endif /* FLAC__CPU_IA32 */
810         }
811 # endif /* !FLAC__NO_ASM */
812 #endif /* !FLAC__INTEGER_ONLY_LIBRARY */
813         /* finally override based on wide-ness if necessary */
814         if(encoder->private_->use_wide_by_block) {
815                 encoder->private_->local_fixed_compute_best_predictor = FLAC__fixed_compute_best_predictor_wide;
816         }
817
818         /* we require precompute_partition_sums if do_escape_coding because of their intertwined nature */
819         encoder->private_->precompute_partition_sums = (encoder->protected_->max_residual_partition_order > encoder->protected_->min_residual_partition_order) || encoder->protected_->do_escape_coding;
820
821         if(!resize_buffers_(encoder, encoder->protected_->blocksize)) {
822                 /* the above function sets the state for us in case of an error */
823                 return encoder->protected_->state;
824         }
825
826         if(!FLAC__bitbuffer_init(encoder->private_->frame))
827                 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
828
829         /*
830          * Set up the verify stuff if necessary
831          */
832         if(encoder->protected_->verify) {
833                 /*
834                  * First, set up the fifo which will hold the
835                  * original signal to compare against
836                  */
837                 encoder->private_->verify.input_fifo.size = encoder->protected_->blocksize;
838                 for(i = 0; i < encoder->protected_->channels; i++) {
839                         if(0 == (encoder->private_->verify.input_fifo.data[i] = (FLAC__int32*)malloc(sizeof(FLAC__int32) * encoder->private_->verify.input_fifo.size)))
840                                 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
841                 }
842                 encoder->private_->verify.input_fifo.tail = 0;
843
844                 /*
845                  * Now set up a stream decoder for verification
846                  */
847                 encoder->private_->verify.decoder = FLAC__stream_decoder_new();
848                 if(0 == encoder->private_->verify.decoder)
849                         return encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
850
851                 FLAC__stream_decoder_set_read_callback(encoder->private_->verify.decoder, verify_read_callback_);
852                 FLAC__stream_decoder_set_write_callback(encoder->private_->verify.decoder, verify_write_callback_);
853                 FLAC__stream_decoder_set_metadata_callback(encoder->private_->verify.decoder, verify_metadata_callback_);
854                 FLAC__stream_decoder_set_error_callback(encoder->private_->verify.decoder, verify_error_callback_);
855                 FLAC__stream_decoder_set_client_data(encoder->private_->verify.decoder, encoder);
856                 if(FLAC__stream_decoder_init(encoder->private_->verify.decoder) != FLAC__STREAM_DECODER_SEARCH_FOR_METADATA)
857                         return encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
858         }
859         encoder->private_->verify.error_stats.absolute_sample = 0;
860         encoder->private_->verify.error_stats.frame_number = 0;
861         encoder->private_->verify.error_stats.channel = 0;
862         encoder->private_->verify.error_stats.sample = 0;
863         encoder->private_->verify.error_stats.expected = 0;
864         encoder->private_->verify.error_stats.got = 0;
865
866         /*
867          * write the stream header
868          */
869         if(encoder->protected_->verify)
870                 encoder->private_->verify.state_hint = ENCODER_IN_MAGIC;
871         if(!FLAC__bitbuffer_write_raw_uint32(encoder->private_->frame, FLAC__STREAM_SYNC, FLAC__STREAM_SYNC_LEN))
872                 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
873         if(!write_bitbuffer_(encoder, 0)) {
874                 /* the above function sets the state for us in case of an error */
875                 return encoder->protected_->state;
876         }
877
878         /*
879          * write the STREAMINFO metadata block
880          */
881         if(encoder->protected_->verify)
882                 encoder->private_->verify.state_hint = ENCODER_IN_METADATA;
883         encoder->private_->metadata.type = FLAC__METADATA_TYPE_STREAMINFO;
884         encoder->private_->metadata.is_last = false; /* we will have at a minimum a VORBIS_COMMENT afterwards */
885         encoder->private_->metadata.length = FLAC__STREAM_METADATA_STREAMINFO_LENGTH;
886         encoder->private_->metadata.data.stream_info.min_blocksize = encoder->protected_->blocksize; /* this encoder uses the same blocksize for the whole stream */
887         encoder->private_->metadata.data.stream_info.max_blocksize = encoder->protected_->blocksize;
888         encoder->private_->metadata.data.stream_info.min_framesize = 0; /* we don't know this yet; have to fill it in later */
889         encoder->private_->metadata.data.stream_info.max_framesize = 0; /* we don't know this yet; have to fill it in later */
890         encoder->private_->metadata.data.stream_info.sample_rate = encoder->protected_->sample_rate;
891         encoder->private_->metadata.data.stream_info.channels = encoder->protected_->channels;
892         encoder->private_->metadata.data.stream_info.bits_per_sample = encoder->protected_->bits_per_sample;
893         encoder->private_->metadata.data.stream_info.total_samples = encoder->protected_->total_samples_estimate; /* we will replace this later with the real total */
894         memset(encoder->private_->metadata.data.stream_info.md5sum, 0, 16); /* we don't know this yet; have to fill it in later */
895         FLAC__MD5Init(&encoder->private_->md5context);
896         if(!FLAC__bitbuffer_clear(encoder->private_->frame))
897                 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
898         if(!FLAC__add_metadata_block(&encoder->private_->metadata, encoder->private_->frame))
899                 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
900         if(!write_bitbuffer_(encoder, 0)) {
901                 /* the above function sets the state for us in case of an error */
902                 return encoder->protected_->state;
903         }
904
905         /*
906          * Now that the STREAMINFO block is written, we can init this to an
907          * absurdly-high value...
908          */
909         encoder->private_->metadata.data.stream_info.min_framesize = (1u << FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN) - 1;
910         /* ... and clear this to 0 */
911         encoder->private_->metadata.data.stream_info.total_samples = 0;
912
913         /*
914          * Check to see if the supplied metadata contains a VORBIS_COMMENT;
915          * if not, we will write an empty one (FLAC__add_metadata_block()
916          * automatically supplies the vendor string).
917          *
918          * WATCHOUT: libOggFLAC depends on us to write this block after the
919          * STREAMINFO since that's what the mapping requires.  (In the case
920          * that metadata_has_vorbis_comment is true it will have already
921          * insured that the metadata list is properly ordered.)
922          */
923         if(!metadata_has_vorbis_comment) {
924                 FLAC__StreamMetadata vorbis_comment;
925                 vorbis_comment.type = FLAC__METADATA_TYPE_VORBIS_COMMENT;
926                 vorbis_comment.is_last = (encoder->protected_->num_metadata_blocks == 0);
927                 vorbis_comment.length = 4 + 4; /* MAGIC NUMBER */
928                 vorbis_comment.data.vorbis_comment.vendor_string.length = 0;
929                 vorbis_comment.data.vorbis_comment.vendor_string.entry = 0;
930                 vorbis_comment.data.vorbis_comment.num_comments = 0;
931                 vorbis_comment.data.vorbis_comment.comments = 0;
932                 if(!FLAC__bitbuffer_clear(encoder->private_->frame))
933                         return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
934                 if(!FLAC__add_metadata_block(&vorbis_comment, encoder->private_->frame))
935                         return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
936                 if(!write_bitbuffer_(encoder, 0)) {
937                         /* the above function sets the state for us in case of an error */
938                         return encoder->protected_->state;
939                 }
940         }
941
942         /*
943          * write the user's metadata blocks
944          */
945         for(i = 0; i < encoder->protected_->num_metadata_blocks; i++) {
946                 encoder->protected_->metadata[i]->is_last = (i == encoder->protected_->num_metadata_blocks - 1);
947                 if(!FLAC__bitbuffer_clear(encoder->private_->frame))
948                         return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
949                 if(!FLAC__add_metadata_block(encoder->protected_->metadata[i], encoder->private_->frame))
950                         return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
951                 if(!write_bitbuffer_(encoder, 0)) {
952                         /* the above function sets the state for us in case of an error */
953                         return encoder->protected_->state;
954                 }
955         }
956
957         if(encoder->protected_->verify)
958                 encoder->private_->verify.state_hint = ENCODER_IN_AUDIO;
959
960         return encoder->protected_->state;
961 }
962
963 FLAC_API void FLAC__stream_encoder_finish(FLAC__StreamEncoder *encoder)
964 {
965         FLAC__ASSERT(0 != encoder);
966
967         if(encoder->protected_->state == FLAC__STREAM_ENCODER_UNINITIALIZED)
968                 return;
969
970         if(encoder->protected_->state == FLAC__STREAM_ENCODER_OK && !encoder->private_->is_being_deleted) {
971                 if(encoder->private_->current_sample_number != 0) {
972                         encoder->protected_->blocksize = encoder->private_->current_sample_number;
973                         process_frame_(encoder, true); /* true => is last frame */
974                 }
975         }
976
977         FLAC__MD5Final(encoder->private_->metadata.data.stream_info.md5sum, &encoder->private_->md5context);
978
979         if(encoder->protected_->state == FLAC__STREAM_ENCODER_OK && !encoder->private_->is_being_deleted) {
980                 encoder->private_->metadata_callback(encoder, &encoder->private_->metadata, encoder->private_->client_data);
981         }
982
983         if(encoder->protected_->verify && 0 != encoder->private_->verify.decoder)
984                 FLAC__stream_decoder_finish(encoder->private_->verify.decoder);
985
986         free_(encoder);
987         set_defaults_(encoder);
988
989         encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
990 }
991
992 FLAC_API FLAC__bool FLAC__stream_encoder_set_verify(FLAC__StreamEncoder *encoder, FLAC__bool value)
993 {
994         FLAC__ASSERT(0 != encoder);
995         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
996                 return false;
997         encoder->protected_->verify = value;
998         return true;
999 }
1000
1001 FLAC_API FLAC__bool FLAC__stream_encoder_set_streamable_subset(FLAC__StreamEncoder *encoder, FLAC__bool value)
1002 {
1003         FLAC__ASSERT(0 != encoder);
1004         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1005                 return false;
1006         encoder->protected_->streamable_subset = value;
1007         return true;
1008 }
1009
1010 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
1011 {
1012         FLAC__ASSERT(0 != encoder);
1013         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1014                 return false;
1015         encoder->protected_->do_mid_side_stereo = value;
1016         return true;
1017 }
1018
1019 FLAC_API FLAC__bool FLAC__stream_encoder_set_loose_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
1020 {
1021         FLAC__ASSERT(0 != encoder);
1022         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1023                 return false;
1024         encoder->protected_->loose_mid_side_stereo = value;
1025         return true;
1026 }
1027
1028 FLAC_API FLAC__bool FLAC__stream_encoder_set_channels(FLAC__StreamEncoder *encoder, unsigned value)
1029 {
1030         FLAC__ASSERT(0 != encoder);
1031         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1032                 return false;
1033         encoder->protected_->channels = value;
1034         return true;
1035 }
1036
1037 FLAC_API FLAC__bool FLAC__stream_encoder_set_bits_per_sample(FLAC__StreamEncoder *encoder, unsigned value)
1038 {
1039         FLAC__ASSERT(0 != encoder);
1040         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1041                 return false;
1042         encoder->protected_->bits_per_sample = value;
1043         return true;
1044 }
1045
1046 FLAC_API FLAC__bool FLAC__stream_encoder_set_sample_rate(FLAC__StreamEncoder *encoder, unsigned value)
1047 {
1048         FLAC__ASSERT(0 != encoder);
1049         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1050                 return false;
1051         encoder->protected_->sample_rate = value;
1052         return true;
1053 }
1054
1055 FLAC_API FLAC__bool FLAC__stream_encoder_set_blocksize(FLAC__StreamEncoder *encoder, unsigned value)
1056 {
1057         FLAC__ASSERT(0 != encoder);
1058         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1059                 return false;
1060         encoder->protected_->blocksize = value;
1061         return true;
1062 }
1063
1064 FLAC_API FLAC__bool FLAC__stream_encoder_set_max_lpc_order(FLAC__StreamEncoder *encoder, unsigned value)
1065 {
1066         FLAC__ASSERT(0 != encoder);
1067         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1068                 return false;
1069         encoder->protected_->max_lpc_order = value;
1070         return true;
1071 }
1072
1073 FLAC_API FLAC__bool FLAC__stream_encoder_set_qlp_coeff_precision(FLAC__StreamEncoder *encoder, unsigned value)
1074 {
1075         FLAC__ASSERT(0 != encoder);
1076         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1077                 return false;
1078         encoder->protected_->qlp_coeff_precision = value;
1079         return true;
1080 }
1081
1082 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_qlp_coeff_prec_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
1083 {
1084         FLAC__ASSERT(0 != encoder);
1085         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1086                 return false;
1087         encoder->protected_->do_qlp_coeff_prec_search = value;
1088         return true;
1089 }
1090
1091 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_escape_coding(FLAC__StreamEncoder *encoder, FLAC__bool value)
1092 {
1093         FLAC__ASSERT(0 != encoder);
1094         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1095                 return false;
1096 #if 0
1097         /*@@@ deprecated: */
1098         encoder->protected_->do_escape_coding = value;
1099 #else
1100         (void)value;
1101 #endif
1102         return true;
1103 }
1104
1105 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_exhaustive_model_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
1106 {
1107         FLAC__ASSERT(0 != encoder);
1108         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1109                 return false;
1110         encoder->protected_->do_exhaustive_model_search = value;
1111         return true;
1112 }
1113
1114 FLAC_API FLAC__bool FLAC__stream_encoder_set_min_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
1115 {
1116         FLAC__ASSERT(0 != encoder);
1117         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1118                 return false;
1119         encoder->protected_->min_residual_partition_order = value;
1120         return true;
1121 }
1122
1123 FLAC_API FLAC__bool FLAC__stream_encoder_set_max_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
1124 {
1125         FLAC__ASSERT(0 != encoder);
1126         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1127                 return false;
1128         encoder->protected_->max_residual_partition_order = value;
1129         return true;
1130 }
1131
1132 FLAC_API FLAC__bool FLAC__stream_encoder_set_rice_parameter_search_dist(FLAC__StreamEncoder *encoder, unsigned value)
1133 {
1134         FLAC__ASSERT(0 != encoder);
1135         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1136                 return false;
1137 #if 0
1138         /*@@@ deprecated: */
1139         encoder->protected_->rice_parameter_search_dist = value;
1140 #else
1141         (void)value;
1142 #endif
1143         return true;
1144 }
1145
1146 FLAC_API FLAC__bool FLAC__stream_encoder_set_total_samples_estimate(FLAC__StreamEncoder *encoder, FLAC__uint64 value)
1147 {
1148         FLAC__ASSERT(0 != encoder);
1149         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1150                 return false;
1151         encoder->protected_->total_samples_estimate = value;
1152         return true;
1153 }
1154
1155 FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata(FLAC__StreamEncoder *encoder, FLAC__StreamMetadata **metadata, unsigned num_blocks)
1156 {
1157         FLAC__ASSERT(0 != encoder);
1158         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1159                 return false;
1160         encoder->protected_->metadata = metadata;
1161         encoder->protected_->num_metadata_blocks = num_blocks;
1162         return true;
1163 }
1164
1165 FLAC_API FLAC__bool FLAC__stream_encoder_set_write_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderWriteCallback value)
1166 {
1167         FLAC__ASSERT(0 != encoder);
1168         FLAC__ASSERT(0 != value);
1169         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1170                 return false;
1171         encoder->private_->write_callback = value;
1172         return true;
1173 }
1174
1175 FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderMetadataCallback value)
1176 {
1177         FLAC__ASSERT(0 != encoder);
1178         FLAC__ASSERT(0 != value);
1179         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1180                 return false;
1181         encoder->private_->metadata_callback = value;
1182         return true;
1183 }
1184
1185 FLAC_API FLAC__bool FLAC__stream_encoder_set_client_data(FLAC__StreamEncoder *encoder, void *value)
1186 {
1187         FLAC__ASSERT(0 != encoder);
1188         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1189                 return false;
1190         encoder->private_->client_data = value;
1191         return true;
1192 }
1193
1194 /*
1195  * These three functions are not static, but not publically exposed in
1196  * include/FLAC/ either.  They are used by the test suite.
1197  */
1198 FLAC_API FLAC__bool FLAC__stream_encoder_disable_constant_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1199 {
1200         FLAC__ASSERT(0 != encoder);
1201         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1202                 return false;
1203         encoder->private_->disable_constant_subframes = value;
1204         return true;
1205 }
1206
1207 FLAC_API FLAC__bool FLAC__stream_encoder_disable_fixed_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1208 {
1209         FLAC__ASSERT(0 != encoder);
1210         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1211                 return false;
1212         encoder->private_->disable_fixed_subframes = value;
1213         return true;
1214 }
1215
1216 FLAC_API FLAC__bool FLAC__stream_encoder_disable_verbatim_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1217 {
1218         FLAC__ASSERT(0 != encoder);
1219         if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1220                 return false;
1221         encoder->private_->disable_verbatim_subframes = value;
1222         return true;
1223 }
1224
1225 FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_get_state(const FLAC__StreamEncoder *encoder)
1226 {
1227         FLAC__ASSERT(0 != encoder);
1228         return encoder->protected_->state;
1229 }
1230
1231 FLAC_API FLAC__StreamDecoderState FLAC__stream_encoder_get_verify_decoder_state(const FLAC__StreamEncoder *encoder)
1232 {
1233         FLAC__ASSERT(0 != encoder);
1234         if(encoder->protected_->verify)
1235                 return FLAC__stream_decoder_get_state(encoder->private_->verify.decoder);
1236         else
1237                 return FLAC__STREAM_DECODER_UNINITIALIZED;
1238 }
1239
1240 FLAC_API const char *FLAC__stream_encoder_get_resolved_state_string(const FLAC__StreamEncoder *encoder)
1241 {
1242         if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR)
1243                 return FLAC__StreamEncoderStateString[encoder->protected_->state];
1244         else
1245                 return FLAC__stream_decoder_get_resolved_state_string(encoder->private_->verify.decoder);
1246 }
1247
1248 FLAC_API void FLAC__stream_encoder_get_verify_decoder_error_stats(const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_sample, unsigned *frame_number, unsigned *channel, unsigned *sample, FLAC__int32 *expected, FLAC__int32 *got)
1249 {
1250         FLAC__ASSERT(0 != encoder);
1251         if(0 != absolute_sample)
1252                 *absolute_sample = encoder->private_->verify.error_stats.absolute_sample;
1253         if(0 != frame_number)
1254                 *frame_number = encoder->private_->verify.error_stats.frame_number;
1255         if(0 != channel)
1256                 *channel = encoder->private_->verify.error_stats.channel;
1257         if(0 != sample)
1258                 *sample = encoder->private_->verify.error_stats.sample;
1259         if(0 != expected)
1260                 *expected = encoder->private_->verify.error_stats.expected;
1261         if(0 != got)
1262                 *got = encoder->private_->verify.error_stats.got;
1263 }
1264
1265 FLAC_API FLAC__bool FLAC__stream_encoder_get_verify(const FLAC__StreamEncoder *encoder)
1266 {
1267         FLAC__ASSERT(0 != encoder);
1268         return encoder->protected_->verify;
1269 }
1270
1271 FLAC_API FLAC__bool FLAC__stream_encoder_get_streamable_subset(const FLAC__StreamEncoder *encoder)
1272 {
1273         FLAC__ASSERT(0 != encoder);
1274         return encoder->protected_->streamable_subset;
1275 }
1276
1277 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_mid_side_stereo(const FLAC__StreamEncoder *encoder)
1278 {
1279         FLAC__ASSERT(0 != encoder);
1280         return encoder->protected_->do_mid_side_stereo;
1281 }
1282
1283 FLAC_API FLAC__bool FLAC__stream_encoder_get_loose_mid_side_stereo(const FLAC__StreamEncoder *encoder)
1284 {
1285         FLAC__ASSERT(0 != encoder);
1286         return encoder->protected_->loose_mid_side_stereo;
1287 }
1288
1289 FLAC_API unsigned FLAC__stream_encoder_get_channels(const FLAC__StreamEncoder *encoder)
1290 {
1291         FLAC__ASSERT(0 != encoder);
1292         return encoder->protected_->channels;
1293 }
1294
1295 FLAC_API unsigned FLAC__stream_encoder_get_bits_per_sample(const FLAC__StreamEncoder *encoder)
1296 {
1297         FLAC__ASSERT(0 != encoder);
1298         return encoder->protected_->bits_per_sample;
1299 }
1300
1301 FLAC_API unsigned FLAC__stream_encoder_get_sample_rate(const FLAC__StreamEncoder *encoder)
1302 {
1303         FLAC__ASSERT(0 != encoder);
1304         return encoder->protected_->sample_rate;
1305 }
1306
1307 FLAC_API unsigned FLAC__stream_encoder_get_blocksize(const FLAC__StreamEncoder *encoder)
1308 {
1309         FLAC__ASSERT(0 != encoder);
1310         return encoder->protected_->blocksize;
1311 }
1312
1313 FLAC_API unsigned FLAC__stream_encoder_get_max_lpc_order(const FLAC__StreamEncoder *encoder)
1314 {
1315         FLAC__ASSERT(0 != encoder);
1316         return encoder->protected_->max_lpc_order;
1317 }
1318
1319 FLAC_API unsigned FLAC__stream_encoder_get_qlp_coeff_precision(const FLAC__StreamEncoder *encoder)
1320 {
1321         FLAC__ASSERT(0 != encoder);
1322         return encoder->protected_->qlp_coeff_precision;
1323 }
1324
1325 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_qlp_coeff_prec_search(const FLAC__StreamEncoder *encoder)
1326 {
1327         FLAC__ASSERT(0 != encoder);
1328         return encoder->protected_->do_qlp_coeff_prec_search;
1329 }
1330
1331 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_escape_coding(const FLAC__StreamEncoder *encoder)
1332 {
1333         FLAC__ASSERT(0 != encoder);
1334         return encoder->protected_->do_escape_coding;
1335 }
1336
1337 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_exhaustive_model_search(const FLAC__StreamEncoder *encoder)
1338 {
1339         FLAC__ASSERT(0 != encoder);
1340         return encoder->protected_->do_exhaustive_model_search;
1341 }
1342
1343 FLAC_API unsigned FLAC__stream_encoder_get_min_residual_partition_order(const FLAC__StreamEncoder *encoder)
1344 {
1345         FLAC__ASSERT(0 != encoder);
1346         return encoder->protected_->min_residual_partition_order;
1347 }
1348
1349 FLAC_API unsigned FLAC__stream_encoder_get_max_residual_partition_order(const FLAC__StreamEncoder *encoder)
1350 {
1351         FLAC__ASSERT(0 != encoder);
1352         return encoder->protected_->max_residual_partition_order;
1353 }
1354
1355 FLAC_API unsigned FLAC__stream_encoder_get_rice_parameter_search_dist(const FLAC__StreamEncoder *encoder)
1356 {
1357         FLAC__ASSERT(0 != encoder);
1358         return encoder->protected_->rice_parameter_search_dist;
1359 }
1360
1361 FLAC_API FLAC__uint64 FLAC__stream_encoder_get_total_samples_estimate(const FLAC__StreamEncoder *encoder)
1362 {
1363         FLAC__ASSERT(0 != encoder);
1364         return encoder->protected_->total_samples_estimate;
1365 }
1366
1367 FLAC_API FLAC__bool FLAC__stream_encoder_process(FLAC__StreamEncoder *encoder, const FLAC__int32 * const buffer[], unsigned samples)
1368 {
1369         unsigned i, j, channel;
1370         FLAC__int32 x, mid, side;
1371         const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
1372
1373         FLAC__ASSERT(0 != encoder);
1374         FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1375
1376         j = 0;
1377         /*
1378          * we have several flavors of the same basic loop, optimized for
1379          * different conditions:
1380          */
1381         if(encoder->protected_->max_lpc_order > 0) {
1382                 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1383                         /*
1384                          * stereo coding: unroll channel loop
1385                          * with LPC: calculate floating point version of signal
1386                          */
1387                         do {
1388                                 if(encoder->protected_->verify)
1389                                         append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1390
1391                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1392                                         x = mid = side = buffer[0][j];
1393                                         encoder->private_->integer_signal[0][i] = x;
1394 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1395                                         encoder->private_->real_signal[0][i] = (FLAC__real)x;
1396 #endif
1397                                         x = buffer[1][j];
1398                                         encoder->private_->integer_signal[1][i] = x;
1399 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1400                                         encoder->private_->real_signal[1][i] = (FLAC__real)x;
1401 #endif
1402                                         mid += x;
1403                                         side -= x;
1404                                         mid >>= 1; /* NOTE: not the same as 'mid = (buffer[0][j] + buffer[1][j]) / 2' ! */
1405                                         encoder->private_->integer_signal_mid_side[1][i] = side;
1406                                         encoder->private_->integer_signal_mid_side[0][i] = mid;
1407 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1408                                         encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1409                                         encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
1410 #endif
1411                                         encoder->private_->current_sample_number++;
1412                                 }
1413                                 if(i == blocksize) {
1414                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1415                                                 return false;
1416                                 }
1417                         } while(j < samples);
1418                 }
1419                 else {
1420                         /*
1421                          * independent channel coding: buffer each channel in inner loop
1422                          * with LPC: calculate floating point version of signal
1423                          */
1424                         do {
1425                                 if(encoder->protected_->verify)
1426                                         append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1427
1428                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1429                                         for(channel = 0; channel < channels; channel++) {
1430                                                 x = buffer[channel][j];
1431                                                 encoder->private_->integer_signal[channel][i] = x;
1432 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1433                                                 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
1434 #endif
1435                                         }
1436                                         encoder->private_->current_sample_number++;
1437                                 }
1438                                 if(i == blocksize) {
1439                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1440                                                 return false;
1441                                 }
1442                         } while(j < samples);
1443                 }
1444         }
1445         else {
1446                 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1447                         /*
1448                          * stereo coding: unroll channel loop
1449                          * without LPC: no need to calculate floating point version of signal
1450                          */
1451                         do {
1452                                 if(encoder->protected_->verify)
1453                                         append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1454
1455                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1456                                         encoder->private_->integer_signal[0][i] = mid = side = buffer[0][j];
1457                                         x = buffer[1][j];
1458                                         encoder->private_->integer_signal[1][i] = x;
1459                                         mid += x;
1460                                         side -= x;
1461                                         mid >>= 1; /* NOTE: not the same as 'mid = (buffer[0][j] + buffer[1][j]) / 2' ! */
1462                                         encoder->private_->integer_signal_mid_side[1][i] = side;
1463                                         encoder->private_->integer_signal_mid_side[0][i] = mid;
1464                                         encoder->private_->current_sample_number++;
1465                                 }
1466                                 if(i == blocksize) {
1467                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1468                                                 return false;
1469                                 }
1470                         } while(j < samples);
1471                 }
1472                 else {
1473                         /*
1474                          * independent channel coding: buffer each channel in inner loop
1475                          * without LPC: no need to calculate floating point version of signal
1476                          */
1477                         do {
1478                                 if(encoder->protected_->verify)
1479                                         append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1480
1481                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1482                                         for(channel = 0; channel < channels; channel++)
1483                                                 encoder->private_->integer_signal[channel][i] = buffer[channel][j];
1484                                         encoder->private_->current_sample_number++;
1485                                 }
1486                                 if(i == blocksize) {
1487                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1488                                                 return false;
1489                                 }
1490                         } while(j < samples);
1491                 }
1492         }
1493
1494         return true;
1495 }
1496
1497 FLAC_API FLAC__bool FLAC__stream_encoder_process_interleaved(FLAC__StreamEncoder *encoder, const FLAC__int32 buffer[], unsigned samples)
1498 {
1499         unsigned i, j, k, channel;
1500         FLAC__int32 x, mid, side;
1501         const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
1502
1503         FLAC__ASSERT(0 != encoder);
1504         FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1505
1506         j = k = 0;
1507         /*
1508          * we have several flavors of the same basic loop, optimized for
1509          * different conditions:
1510          */
1511         if(encoder->protected_->max_lpc_order > 0) {
1512                 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1513                         /*
1514                          * stereo coding: unroll channel loop
1515                          * with LPC: calculate floating point version of signal
1516                          */
1517                         do {
1518                                 if(encoder->protected_->verify)
1519                                         append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1520
1521                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1522                                         x = mid = side = buffer[k++];
1523                                         encoder->private_->integer_signal[0][i] = x;
1524 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1525                                         encoder->private_->real_signal[0][i] = (FLAC__real)x;
1526 #endif
1527                                         x = buffer[k++];
1528                                         encoder->private_->integer_signal[1][i] = x;
1529 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1530                                         encoder->private_->real_signal[1][i] = (FLAC__real)x;
1531 #endif
1532                                         mid += x;
1533                                         side -= x;
1534                                         mid >>= 1; /* NOTE: not the same as 'mid = (left + right) / 2' ! */
1535                                         encoder->private_->integer_signal_mid_side[1][i] = side;
1536                                         encoder->private_->integer_signal_mid_side[0][i] = mid;
1537 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1538                                         encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1539                                         encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
1540 #endif
1541                                         encoder->private_->current_sample_number++;
1542                                 }
1543                                 if(i == blocksize) {
1544                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1545                                                 return false;
1546                                 }
1547                         } while(j < samples);
1548                 }
1549                 else {
1550                         /*
1551                          * independent channel coding: buffer each channel in inner loop
1552                          * with LPC: calculate floating point version of signal
1553                          */
1554                         do {
1555                                 if(encoder->protected_->verify)
1556                                         append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1557
1558                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1559                                         for(channel = 0; channel < channels; channel++) {
1560                                                 x = buffer[k++];
1561                                                 encoder->private_->integer_signal[channel][i] = x;
1562 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1563                                                 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
1564 #endif
1565                                         }
1566                                         encoder->private_->current_sample_number++;
1567                                 }
1568                                 if(i == blocksize) {
1569                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1570                                                 return false;
1571                                 }
1572                         } while(j < samples);
1573                 }
1574         }
1575         else {
1576                 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1577                         /*
1578                          * stereo coding: unroll channel loop
1579                          * without LPC: no need to calculate floating point version of signal
1580                          */
1581                         do {
1582                                 if(encoder->protected_->verify)
1583                                         append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1584
1585                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1586                                         encoder->private_->integer_signal[0][i] = mid = side = buffer[k++];
1587                                         x = buffer[k++];
1588                                         encoder->private_->integer_signal[1][i] = x;
1589                                         mid += x;
1590                                         side -= x;
1591                                         mid >>= 1; /* NOTE: not the same as 'mid = (left + right) / 2' ! */
1592                                         encoder->private_->integer_signal_mid_side[1][i] = side;
1593                                         encoder->private_->integer_signal_mid_side[0][i] = mid;
1594                                         encoder->private_->current_sample_number++;
1595                                 }
1596                                 if(i == blocksize) {
1597                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1598                                                 return false;
1599                                 }
1600                         } while(j < samples);
1601                 }
1602                 else {
1603                         /*
1604                          * independent channel coding: buffer each channel in inner loop
1605                          * without LPC: no need to calculate floating point version of signal
1606                          */
1607                         do {
1608                                 if(encoder->protected_->verify)
1609                                         append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1610
1611                                 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1612                                         for(channel = 0; channel < channels; channel++)
1613                                                 encoder->private_->integer_signal[channel][i] = buffer[k++];
1614                                         encoder->private_->current_sample_number++;
1615                                 }
1616                                 if(i == blocksize) {
1617                                         if(!process_frame_(encoder, false)) /* false => not last frame */
1618                                                 return false;
1619                                 }
1620                         } while(j < samples);
1621                 }
1622         }
1623
1624         return true;
1625 }
1626
1627 /***********************************************************************
1628  *
1629  * Private class methods
1630  *
1631  ***********************************************************************/
1632
1633 void set_defaults_(FLAC__StreamEncoder *encoder)
1634 {
1635         FLAC__ASSERT(0 != encoder);
1636
1637         encoder->protected_->verify = false;
1638         encoder->protected_->streamable_subset = true;
1639         encoder->protected_->do_mid_side_stereo = false;
1640         encoder->protected_->loose_mid_side_stereo = false;
1641         encoder->protected_->channels = 2;
1642         encoder->protected_->bits_per_sample = 16;
1643         encoder->protected_->sample_rate = 44100;
1644         encoder->protected_->blocksize = 1152;
1645         encoder->protected_->max_lpc_order = 0;
1646         encoder->protected_->qlp_coeff_precision = 0;
1647         encoder->protected_->do_qlp_coeff_prec_search = false;
1648         encoder->protected_->do_exhaustive_model_search = false;
1649         encoder->protected_->do_escape_coding = false;
1650         encoder->protected_->min_residual_partition_order = 0;
1651         encoder->protected_->max_residual_partition_order = 0;
1652         encoder->protected_->rice_parameter_search_dist = 0;
1653         encoder->protected_->total_samples_estimate = 0;
1654         encoder->protected_->metadata = 0;
1655         encoder->protected_->num_metadata_blocks = 0;
1656
1657         encoder->private_->disable_constant_subframes = false;
1658         encoder->private_->disable_fixed_subframes = false;
1659         encoder->private_->disable_verbatim_subframes = false;
1660         encoder->private_->write_callback = 0;
1661         encoder->private_->metadata_callback = 0;
1662         encoder->private_->client_data = 0;
1663 }
1664
1665 void free_(FLAC__StreamEncoder *encoder)
1666 {
1667         unsigned i, channel;
1668
1669         FLAC__ASSERT(0 != encoder);
1670         for(i = 0; i < encoder->protected_->channels; i++) {
1671                 if(0 != encoder->private_->integer_signal_unaligned[i]) {
1672                         free(encoder->private_->integer_signal_unaligned[i]);
1673                         encoder->private_->integer_signal_unaligned[i] = 0;
1674                 }
1675 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1676                 if(0 != encoder->private_->real_signal_unaligned[i]) {
1677                         free(encoder->private_->real_signal_unaligned[i]);
1678                         encoder->private_->real_signal_unaligned[i] = 0;
1679                 }
1680 #endif
1681         }
1682         for(i = 0; i < 2; i++) {
1683                 if(0 != encoder->private_->integer_signal_mid_side_unaligned[i]) {
1684                         free(encoder->private_->integer_signal_mid_side_unaligned[i]);
1685                         encoder->private_->integer_signal_mid_side_unaligned[i] = 0;
1686                 }
1687 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1688                 if(0 != encoder->private_->real_signal_mid_side_unaligned[i]) {
1689                         free(encoder->private_->real_signal_mid_side_unaligned[i]);
1690                         encoder->private_->real_signal_mid_side_unaligned[i] = 0;
1691                 }
1692 #endif
1693         }
1694         for(channel = 0; channel < encoder->protected_->channels; channel++) {
1695                 for(i = 0; i < 2; i++) {
1696                         if(0 != encoder->private_->residual_workspace_unaligned[channel][i]) {
1697                                 free(encoder->private_->residual_workspace_unaligned[channel][i]);
1698                                 encoder->private_->residual_workspace_unaligned[channel][i] = 0;
1699                         }
1700                 }
1701         }
1702         for(channel = 0; channel < 2; channel++) {
1703                 for(i = 0; i < 2; i++) {
1704                         if(0 != encoder->private_->residual_workspace_mid_side_unaligned[channel][i]) {
1705                                 free(encoder->private_->residual_workspace_mid_side_unaligned[channel][i]);
1706                                 encoder->private_->residual_workspace_mid_side_unaligned[channel][i] = 0;
1707                         }
1708                 }
1709         }
1710         if(0 != encoder->private_->abs_residual_unaligned) {
1711                 free(encoder->private_->abs_residual_unaligned);
1712                 encoder->private_->abs_residual_unaligned = 0;
1713         }
1714         if(0 != encoder->private_->abs_residual_partition_sums_unaligned) {
1715                 free(encoder->private_->abs_residual_partition_sums_unaligned);
1716                 encoder->private_->abs_residual_partition_sums_unaligned = 0;
1717         }
1718         if(0 != encoder->private_->raw_bits_per_partition_unaligned) {
1719                 free(encoder->private_->raw_bits_per_partition_unaligned);
1720                 encoder->private_->raw_bits_per_partition_unaligned = 0;
1721         }
1722         if(encoder->protected_->verify) {
1723                 for(i = 0; i < encoder->protected_->channels; i++) {
1724                         if(0 != encoder->private_->verify.input_fifo.data[i]) {
1725                                 free(encoder->private_->verify.input_fifo.data[i]);
1726                                 encoder->private_->verify.input_fifo.data[i] = 0;
1727                         }
1728                 }
1729         }
1730         FLAC__bitbuffer_free(encoder->private_->frame);
1731 }
1732
1733 FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size)
1734 {
1735         FLAC__bool ok;
1736         unsigned i, channel;
1737
1738         FLAC__ASSERT(new_size > 0);
1739         FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1740         FLAC__ASSERT(encoder->private_->current_sample_number == 0);
1741
1742         /* To avoid excessive malloc'ing, we only grow the buffer; no shrinking. */
1743         if(new_size <= encoder->private_->input_capacity)
1744                 return true;
1745
1746         ok = true;
1747
1748         /* WATCHOUT: FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32_mmx()
1749          * requires that the input arrays (in our case the integer signals)
1750          * have a buffer of up to 3 zeroes in front (at negative indices) for
1751          * alignment purposes; we use 4 to keep the data well-aligned.
1752          */
1753
1754         for(i = 0; ok && i < encoder->protected_->channels; i++) {
1755                 ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size+4, &encoder->private_->integer_signal_unaligned[i], &encoder->private_->integer_signal[i]);
1756 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1757                 if(encoder->protected_->max_lpc_order > 0)
1758                         ok = ok && FLAC__memory_alloc_aligned_real_array(new_size, &encoder->private_->real_signal_unaligned[i], &encoder->private_->real_signal[i]);
1759 #endif
1760                 memset(encoder->private_->integer_signal[i], 0, sizeof(FLAC__int32)*4);
1761                 encoder->private_->integer_signal[i] += 4;
1762         }
1763         for(i = 0; ok && i < 2; i++) {
1764                 ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size+4, &encoder->private_->integer_signal_mid_side_unaligned[i], &encoder->private_->integer_signal_mid_side[i]);
1765 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1766                 if(encoder->protected_->max_lpc_order > 0)
1767                         ok = ok && FLAC__memory_alloc_aligned_real_array(new_size, &encoder->private_->real_signal_mid_side_unaligned[i], &encoder->private_->real_signal_mid_side[i]);
1768 #endif
1769                 memset(encoder->private_->integer_signal_mid_side[i], 0, sizeof(FLAC__int32)*4);
1770                 encoder->private_->integer_signal_mid_side[i] += 4;
1771         }
1772         for(channel = 0; ok && channel < encoder->protected_->channels; channel++) {
1773                 for(i = 0; ok && i < 2; i++) {
1774                         ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size, &encoder->private_->residual_workspace_unaligned[channel][i], &encoder->private_->residual_workspace[channel][i]);
1775                 }
1776         }
1777         for(channel = 0; ok && channel < 2; channel++) {
1778                 for(i = 0; ok && i < 2; i++) {
1779                         ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size, &encoder->private_->residual_workspace_mid_side_unaligned[channel][i], &encoder->private_->residual_workspace_mid_side[channel][i]);
1780                 }
1781         }
1782         ok = ok && FLAC__memory_alloc_aligned_uint32_array(new_size, &encoder->private_->abs_residual_unaligned, &encoder->private_->abs_residual);
1783         if(encoder->private_->precompute_partition_sums || encoder->protected_->do_escape_coding) /* we require precompute_partition_sums if do_escape_coding because of their intertwined nature */
1784                 ok = ok && FLAC__memory_alloc_aligned_uint64_array(new_size * 2, &encoder->private_->abs_residual_partition_sums_unaligned, &encoder->private_->abs_residual_partition_sums);
1785         if(encoder->protected_->do_escape_coding)
1786                 ok = ok && FLAC__memory_alloc_aligned_unsigned_array(new_size * 2, &encoder->private_->raw_bits_per_partition_unaligned, &encoder->private_->raw_bits_per_partition);
1787
1788         if(ok)
1789                 encoder->private_->input_capacity = new_size;
1790         else
1791                 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1792
1793         return ok;
1794 }
1795
1796 FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples)
1797 {
1798         const FLAC__byte *buffer;
1799         unsigned bytes;
1800
1801         FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1802
1803         FLAC__bitbuffer_get_buffer(encoder->private_->frame, &buffer, &bytes);
1804
1805         if(encoder->protected_->verify) {
1806                 encoder->private_->verify.output.data = buffer;
1807                 encoder->private_->verify.output.bytes = bytes;
1808                 if(encoder->private_->verify.state_hint == ENCODER_IN_MAGIC) {
1809                         encoder->private_->verify.needs_magic_hack = true;
1810                 }
1811                 else {
1812                         if(!FLAC__stream_decoder_process_single(encoder->private_->verify.decoder)) {
1813                                 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1814                                 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA)
1815                                         encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
1816                                 return false;
1817                         }
1818                 }
1819         }
1820
1821         if(encoder->private_->write_callback(encoder, buffer, bytes, samples, encoder->private_->current_frame_number, encoder->private_->client_data) != FLAC__STREAM_ENCODER_WRITE_STATUS_OK) {
1822                 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1823                 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_WRITING;
1824                 return false;
1825         }
1826
1827         FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1828
1829         if(samples > 0) {
1830                 encoder->private_->metadata.data.stream_info.min_framesize = min(bytes, encoder->private_->metadata.data.stream_info.min_framesize);
1831                 encoder->private_->metadata.data.stream_info.max_framesize = max(bytes, encoder->private_->metadata.data.stream_info.max_framesize);
1832         }
1833
1834         return true;
1835 }
1836
1837 FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
1838 {
1839         FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1840
1841         /*
1842          * Accumulate raw signal to the MD5 signature
1843          */
1844         if(!FLAC__MD5Accumulate(&encoder->private_->md5context, (const FLAC__int32 * const *)encoder->private_->integer_signal, encoder->protected_->channels, encoder->protected_->blocksize, (encoder->protected_->bits_per_sample+7) / 8)) {
1845                 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1846                 return false;
1847         }
1848
1849         /*
1850          * Process the frame header and subframes into the frame bitbuffer
1851          */
1852         if(!process_subframes_(encoder, is_last_frame)) {
1853                 /* the above function sets the state for us in case of an error */
1854                 return false;
1855         }
1856
1857         /*
1858          * Zero-pad the frame to a byte_boundary
1859          */
1860         if(!FLAC__bitbuffer_zero_pad_to_byte_boundary(encoder->private_->frame)) {
1861                 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1862                 return false;
1863         }
1864
1865         /*
1866          * CRC-16 the whole thing
1867          */
1868         FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1869         FLAC__bitbuffer_write_raw_uint32(encoder->private_->frame, FLAC__bitbuffer_get_write_crc16(encoder->private_->frame), FLAC__FRAME_FOOTER_CRC_LEN);
1870
1871         /*
1872          * Write it
1873          */
1874         if(!write_bitbuffer_(encoder, encoder->protected_->blocksize)) {
1875                 /* the above function sets the state for us in case of an error */
1876                 return false;
1877         }
1878
1879         /*
1880          * Get ready for the next frame
1881          */
1882         encoder->private_->current_sample_number = 0;
1883         encoder->private_->current_frame_number++;
1884         encoder->private_->metadata.data.stream_info.total_samples += (FLAC__uint64)encoder->protected_->blocksize;
1885
1886         return true;
1887 }
1888
1889 FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
1890 {
1891         FLAC__FrameHeader frame_header;
1892         unsigned channel, min_partition_order = encoder->protected_->min_residual_partition_order, max_partition_order;
1893         FLAC__bool do_independent, do_mid_side, precompute_partition_sums;
1894
1895         /*
1896          * Calculate the min,max Rice partition orders
1897          */
1898         if(is_last_frame) {
1899                 max_partition_order = 0;
1900         }
1901         else {
1902                 max_partition_order = FLAC__format_get_max_rice_partition_order_from_blocksize(encoder->protected_->blocksize);
1903                 max_partition_order = min(max_partition_order, encoder->protected_->max_residual_partition_order);
1904         }
1905         min_partition_order = min(min_partition_order, max_partition_order);
1906
1907         precompute_partition_sums = encoder->private_->precompute_partition_sums && ((max_partition_order > min_partition_order) || encoder->protected_->do_escape_coding);
1908
1909         /*
1910          * Setup the frame
1911          */
1912         if(!FLAC__bitbuffer_clear(encoder->private_->frame)) {
1913                 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1914                 return false;
1915         }
1916         frame_header.blocksize = encoder->protected_->blocksize;
1917         frame_header.sample_rate = encoder->protected_->sample_rate;
1918         frame_header.channels = encoder->protected_->channels;
1919         frame_header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT; /* the default unless the encoder determines otherwise */
1920         frame_header.bits_per_sample = encoder->protected_->bits_per_sample;
1921         frame_header.number_type = FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER;
1922         frame_header.number.frame_number = encoder->private_->current_frame_number;
1923
1924         /*
1925          * Figure out what channel assignments to try
1926          */
1927         if(encoder->protected_->do_mid_side_stereo) {
1928                 if(encoder->protected_->loose_mid_side_stereo) {
1929                         if(encoder->private_->loose_mid_side_stereo_frame_count == 0) {
1930                                 do_independent = true;
1931                                 do_mid_side = true;
1932                         }
1933                         else {
1934                                 do_independent = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT);
1935                                 do_mid_side = !do_independent;
1936                         }
1937                 }
1938                 else {
1939                         do_independent = true;
1940                         do_mid_side = true;
1941                 }
1942         }
1943         else {
1944                 do_independent = true;
1945                 do_mid_side = false;
1946         }
1947
1948         FLAC__ASSERT(do_independent || do_mid_side);
1949
1950         /*
1951          * Check for wasted bits; set effective bps for each subframe
1952          */
1953         if(do_independent) {
1954                 for(channel = 0; channel < encoder->protected_->channels; channel++) {
1955                         const unsigned w = get_wasted_bits_(encoder->private_->integer_signal[channel], encoder->protected_->blocksize);
1956                         encoder->private_->subframe_workspace[channel][0].wasted_bits = encoder->private_->subframe_workspace[channel][1].wasted_bits = w;
1957                         encoder->private_->subframe_bps[channel] = encoder->protected_->bits_per_sample - w;
1958                 }
1959         }
1960         if(do_mid_side) {
1961                 FLAC__ASSERT(encoder->protected_->channels == 2);
1962                 for(channel = 0; channel < 2; channel++) {
1963                         const unsigned w = get_wasted_bits_(encoder->private_->integer_signal_mid_side[channel], encoder->protected_->blocksize);
1964                         encoder->private_->subframe_workspace_mid_side[channel][0].wasted_bits = encoder->private_->subframe_workspace_mid_side[channel][1].wasted_bits = w;
1965                         encoder->private_->subframe_bps_mid_side[channel] = encoder->protected_->bits_per_sample - w + (channel==0? 0:1);
1966                 }
1967         }
1968
1969         /*
1970          * First do a normal encoding pass of each independent channel
1971          */
1972         if(do_independent) {
1973                 for(channel = 0; channel < encoder->protected_->channels; channel++) {
1974                         if(!
1975                                 process_subframe_(
1976                                         encoder,
1977                                         min_partition_order,
1978                                         max_partition_order,
1979                                         precompute_partition_sums,
1980                                         &frame_header,
1981                                         encoder->private_->subframe_bps[channel],
1982                                         encoder->private_->integer_signal[channel],
1983 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1984                                         encoder->private_->real_signal[channel],
1985 #endif
1986                                         encoder->private_->subframe_workspace_ptr[channel],
1987                                         encoder->private_->partitioned_rice_contents_workspace_ptr[channel],
1988                                         encoder->private_->residual_workspace[channel],
1989                                         encoder->private_->best_subframe+channel,
1990                                         encoder->private_->best_subframe_bits+channel
1991                                 )
1992                         )
1993                                 return false;
1994                 }
1995         }
1996
1997         /*
1998          * Now do mid and side channels if requested
1999          */
2000         if(do_mid_side) {
2001                 FLAC__ASSERT(encoder->protected_->channels == 2);
2002
2003                 for(channel = 0; channel < 2; channel++) {
2004                         if(!
2005                                 process_subframe_(
2006                                         encoder,
2007                                         min_partition_order,
2008                                         max_partition_order,
2009                                         precompute_partition_sums,
2010                                         &frame_header,
2011                                         encoder->private_->subframe_bps_mid_side[channel],
2012                                         encoder->private_->integer_signal_mid_side[channel],
2013 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2014                                         encoder->private_->real_signal_mid_side[channel],
2015 #endif
2016                                         encoder->private_->subframe_workspace_ptr_mid_side[channel],
2017                                         encoder->private_->partitioned_rice_contents_workspace_ptr_mid_side[channel],
2018                                         encoder->private_->residual_workspace_mid_side[channel],
2019                                         encoder->private_->best_subframe_mid_side+channel,
2020                                         encoder->private_->best_subframe_bits_mid_side+channel
2021                                 )
2022                         )
2023                                 return false;
2024                 }
2025         }
2026
2027         /*
2028          * Compose the frame bitbuffer
2029          */
2030         if(do_mid_side) {
2031                 unsigned left_bps = 0, right_bps = 0; /* initialized only to prevent superfluous compiler warning */
2032                 FLAC__Subframe *left_subframe = 0, *right_subframe = 0; /* initialized only to prevent superfluous compiler warning */
2033                 FLAC__ChannelAssignment channel_assignment;
2034
2035                 FLAC__ASSERT(encoder->protected_->channels == 2);
2036
2037                 if(encoder->protected_->loose_mid_side_stereo && encoder->private_->loose_mid_side_stereo_frame_count > 0) {
2038                         channel_assignment = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT? FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT : FLAC__CHANNEL_ASSIGNMENT_MID_SIDE);
2039                 }
2040                 else {
2041                         unsigned bits[4]; /* WATCHOUT - indexed by FLAC__ChannelAssignment */
2042                         unsigned min_bits;
2043                         FLAC__ChannelAssignment ca;
2044
2045                         FLAC__ASSERT(do_independent && do_mid_side);
2046
2047                         /* We have to figure out which channel assignent results in the smallest frame */
2048                         bits[FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT] = encoder->private_->best_subframe_bits         [0] + encoder->private_->best_subframe_bits         [1];
2049                         bits[FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE  ] = encoder->private_->best_subframe_bits         [0] + encoder->private_->best_subframe_bits_mid_side[1];
2050                         bits[FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE ] = encoder->private_->best_subframe_bits         [1] + encoder->private_->best_subframe_bits_mid_side[1];
2051                         bits[FLAC__CHANNEL_ASSIGNMENT_MID_SIDE   ] = encoder->private_->best_subframe_bits_mid_side[0] + encoder->private_->best_subframe_bits_mid_side[1];
2052
2053                         for(channel_assignment = (FLAC__ChannelAssignment)0, min_bits = bits[0], ca = (FLAC__ChannelAssignment)1; (int)ca <= 3; ca = (FLAC__ChannelAssignment)((int)ca + 1)) {
2054                                 if(bits[ca] < min_bits) {
2055                                         min_bits = bits[ca];
2056                                         channel_assignment = ca;
2057                                 }
2058                         }
2059                 }
2060
2061                 frame_header.channel_assignment = channel_assignment;
2062
2063                 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
2064                         encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
2065                         return false;
2066                 }
2067
2068                 switch(channel_assignment) {
2069                         case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
2070                                 left_subframe  = &encoder->private_->subframe_workspace         [0][encoder->private_->best_subframe         [0]];
2071                                 right_subframe = &encoder->private_->subframe_workspace         [1][encoder->private_->best_subframe         [1]];
2072                                 break;
2073                         case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
2074                                 left_subframe  = &encoder->private_->subframe_workspace         [0][encoder->private_->best_subframe         [0]];
2075                                 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
2076                                 break;
2077                         case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
2078                                 left_subframe  = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
2079                                 right_subframe = &encoder->private_->subframe_workspace         [1][encoder->private_->best_subframe         [1]];
2080                                 break;
2081                         case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
2082                                 left_subframe  = &encoder->private_->subframe_workspace_mid_side[0][encoder->private_->best_subframe_mid_side[0]];
2083                                 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
2084                                 break;
2085                         default:
2086                                 FLAC__ASSERT(0);
2087                 }
2088
2089                 switch(channel_assignment) {
2090                         case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
2091                                 left_bps  = encoder->private_->subframe_bps         [0];
2092                                 right_bps = encoder->private_->subframe_bps         [1];
2093                                 break;
2094                         case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
2095                                 left_bps  = encoder->private_->subframe_bps         [0];
2096                                 right_bps = encoder->private_->subframe_bps_mid_side[1];
2097                                 break;
2098                         case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
2099                                 left_bps  = encoder->private_->subframe_bps_mid_side[1];
2100                                 right_bps = encoder->private_->subframe_bps         [1];
2101                                 break;
2102                         case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
2103                                 left_bps  = encoder->private_->subframe_bps_mid_side[0];
2104                                 right_bps = encoder->private_->subframe_bps_mid_side[1];
2105                                 break;
2106                         default:
2107                                 FLAC__ASSERT(0);
2108                 }
2109
2110                 /* note that encoder_add_subframe_ sets the state for us in case of an error */
2111                 if(!add_subframe_(encoder, &frame_header, left_bps , left_subframe , encoder->private_->frame))
2112                         return false;
2113                 if(!add_subframe_(encoder, &frame_header, right_bps, right_subframe, encoder->private_->frame))
2114                         return false;
2115         }
2116         else {
2117                 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
2118                         encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
2119                         return false;
2120                 }
2121
2122                 for(channel = 0; channel < encoder->protected_->channels; channel++) {
2123                         if(!add_subframe_(encoder, &frame_header, encoder->private_->subframe_bps[channel], &encoder->private_->subframe_workspace[channel][encoder->private_->best_subframe[channel]], encoder->private_->frame)) {
2124                                 /* the above function sets the state for us in case of an error */
2125                                 return false;
2126                         }
2127                 }
2128         }
2129
2130         if(encoder->protected_->loose_mid_side_stereo) {
2131                 encoder->private_->loose_mid_side_stereo_frame_count++;
2132                 if(encoder->private_->loose_mid_side_stereo_frame_count >= encoder->private_->loose_mid_side_stereo_frames)
2133                         encoder->private_->loose_mid_side_stereo_frame_count = 0;
2134         }
2135
2136         encoder->private_->last_channel_assignment = frame_header.channel_assignment;
2137
2138         return true;
2139 }
2140
2141 FLAC__bool process_subframe_(
2142         FLAC__StreamEncoder *encoder,
2143         unsigned min_partition_order,
2144         unsigned max_partition_order,
2145         FLAC__bool precompute_partition_sums,
2146         const FLAC__FrameHeader *frame_header,
2147         unsigned subframe_bps,
2148         const FLAC__int32 integer_signal[],
2149 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2150         const FLAC__real real_signal[],
2151 #endif
2152         FLAC__Subframe *subframe[2],
2153         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
2154         FLAC__int32 *residual[2],
2155         unsigned *best_subframe,
2156         unsigned *best_bits
2157 )
2158 {
2159 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2160         FLAC__float fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
2161 #else
2162         FLAC__fixedpoint fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
2163 #endif
2164 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2165         FLAC__double lpc_residual_bits_per_sample;
2166         FLAC__real autoc[FLAC__MAX_LPC_ORDER+1]; /* WATCHOUT: the size is important even though encoder->protected_->max_lpc_order might be less; some asm routines need all the space */
2167         FLAC__double lpc_error[FLAC__MAX_LPC_ORDER];
2168         unsigned min_lpc_order, max_lpc_order, lpc_order;
2169         unsigned min_qlp_coeff_precision, max_qlp_coeff_precision, qlp_coeff_precision;
2170 #endif
2171         unsigned min_fixed_order, max_fixed_order, guess_fixed_order, fixed_order;
2172         unsigned rice_parameter;
2173         unsigned _candidate_bits, _best_bits;
2174         unsigned _best_subframe;
2175
2176         /* verbatim subframe is the baseline against which we measure other compressed subframes */
2177         _best_subframe = 0;
2178         if(encoder->private_->disable_verbatim_subframes && frame_header->blocksize >= FLAC__MAX_FIXED_ORDER)
2179                 _best_bits = UINT_MAX;
2180         else
2181                 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
2182
2183         if(frame_header->blocksize >= FLAC__MAX_FIXED_ORDER) {
2184                 unsigned signal_is_constant = false;
2185                 guess_fixed_order = encoder->private_->local_fixed_compute_best_predictor(integer_signal+FLAC__MAX_FIXED_ORDER, frame_header->blocksize-FLAC__MAX_FIXED_ORDER, fixed_residual_bits_per_sample);
2186                 /* check for constant subframe */
2187                 if(
2188                         !encoder->private_->disable_constant_subframes &&
2189 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2190                         fixed_residual_bits_per_sample[1] == 0.0
2191 #else
2192                         fixed_residual_bits_per_sample[1] == FLAC__FP_ZERO
2193 #endif
2194                 ) {
2195                         /* the above means it's possible all samples are the same value; now double-check it: */
2196                         unsigned i;
2197                         signal_is_constant = true;
2198                         for(i = 1; i < frame_header->blocksize; i++) {
2199                                 if(integer_signal[0] != integer_signal[i]) {
2200                                         signal_is_constant = false;
2201                                         break;
2202                                 }
2203                         }
2204                 }
2205                 if(signal_is_constant) {
2206                         _candidate_bits = evaluate_constant_subframe_(integer_signal[0], subframe_bps, subframe[!_best_subframe]);
2207                         if(_candidate_bits < _best_bits) {
2208                                 _best_subframe = !_best_subframe;
2209                                 _best_bits = _candidate_bits;
2210                         }
2211                 }
2212                 else {
2213                         if(!encoder->private_->disable_fixed_subframes || (encoder->protected_->max_lpc_order == 0 && _best_bits == UINT_MAX)) {
2214                                 /* encode fixed */
2215                                 if(encoder->protected_->do_exhaustive_model_search) {
2216                                         min_fixed_order = 0;
2217                                         max_fixed_order = FLAC__MAX_FIXED_ORDER;
2218                                 }
2219                                 else {
2220                                         min_fixed_order = max_fixed_order = guess_fixed_order;
2221                                 }
2222                                 for(fixed_order = min_fixed_order; fixed_order <= max_fixed_order; fixed_order++) {
2223 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2224                                         if(fixed_residual_bits_per_sample[fixed_order] >= (FLAC__float)subframe_bps)
2225                                                 continue; /* don't even try */
2226                                         rice_parameter = (fixed_residual_bits_per_sample[fixed_order] > 0.0)? (unsigned)(fixed_residual_bits_per_sample[fixed_order]+0.5) : 0; /* 0.5 is for rounding */
2227 #else
2228                                         if(FLAC__fixedpoint_trunc(fixed_residual_bits_per_sample[fixed_order]) >= (int)subframe_bps)
2229                                                 continue; /* don't even try */
2230                                         rice_parameter = (fixed_residual_bits_per_sample[fixed_order] > FLAC__FP_ZERO)? (unsigned)FLAC__fixedpoint_trunc(fixed_residual_bits_per_sample[fixed_order]+FLAC__FP_ONE_HALF) : 0; /* 0.5 is for rounding */
2231 #endif
2232 #ifndef FLAC__SYMMETRIC_RICE
2233                                         rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
2234 #endif
2235                                         if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2236 #ifdef DEBUG_VERBOSE
2237                                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @0\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2238 #endif
2239                                                 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2240                                         }
2241                                         _candidate_bits =
2242                                                 evaluate_fixed_subframe_(
2243                                                         encoder,
2244                                                         integer_signal,
2245                                                         residual[!_best_subframe],
2246                                                         encoder->private_->abs_residual,
2247                                                         encoder->private_->abs_residual_partition_sums,
2248                                                         encoder->private_->raw_bits_per_partition,
2249                                                         frame_header->blocksize,
2250                                                         subframe_bps,
2251                                                         fixed_order,
2252                                                         rice_parameter,
2253                                                         min_partition_order,
2254                                                         max_partition_order,
2255                                                         precompute_partition_sums,
2256                                                         encoder->protected_->do_escape_coding,
2257                                                         encoder->protected_->rice_parameter_search_dist,
2258                                                         subframe[!_best_subframe],
2259                                                         partitioned_rice_contents[!_best_subframe]
2260                                                 );
2261                                         if(_candidate_bits < _best_bits) {
2262                                                 _best_subframe = !_best_subframe;
2263                                                 _best_bits = _candidate_bits;
2264                                         }
2265                                 }
2266                         }
2267
2268 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2269                         /* encode lpc */
2270                         if(encoder->protected_->max_lpc_order > 0) {
2271                                 if(encoder->protected_->max_lpc_order >= frame_header->blocksize)
2272                                         max_lpc_order = frame_header->blocksize-1;
2273                                 else
2274                                         max_lpc_order = encoder->protected_->max_lpc_order;
2275                                 if(max_lpc_order > 0) {
2276                                         encoder->private_->local_lpc_compute_autocorrelation(real_signal, frame_header->blocksize, max_lpc_order+1, autoc);
2277                                         /* if autoc[0] == 0.0, the signal is constant and we usually won't get here, but it can happen */
2278                                         if(autoc[0] != 0.0) {
2279                                                 FLAC__lpc_compute_lp_coefficients(autoc, max_lpc_order, encoder->private_->lp_coeff, lpc_error);
2280                                                 if(encoder->protected_->do_exhaustive_model_search) {
2281                                                         min_lpc_order = 1;
2282                                                 }
2283                                                 else {
2284                                                         unsigned guess_lpc_order = FLAC__lpc_compute_best_order(lpc_error, max_lpc_order, frame_header->blocksize, subframe_bps);
2285                                                         min_lpc_order = max_lpc_order = guess_lpc_order;
2286                                                 }
2287                                                 for(lpc_order = min_lpc_order; lpc_order <= max_lpc_order; lpc_order++) {
2288                                                         lpc_residual_bits_per_sample = FLAC__lpc_compute_expected_bits_per_residual_sample(lpc_error[lpc_order-1], frame_header->blocksize-lpc_order);
2289                                                         if(lpc_residual_bits_per_sample >= (FLAC__double)subframe_bps)
2290                                                                 continue; /* don't even try */
2291                                                         rice_parameter = (lpc_residual_bits_per_sample > 0.0)? (unsigned)(lpc_residual_bits_per_sample+0.5) : 0; /* 0.5 is for rounding */
2292 #ifndef FLAC__SYMMETRIC_RICE
2293                                                         rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
2294 #endif
2295                                                         if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2296 #ifdef DEBUG_VERBOSE
2297                                                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @1\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2298 #endif
2299                                                                 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2300                                                         }
2301                                                         if(encoder->protected_->do_qlp_coeff_prec_search) {
2302                                                                 min_qlp_coeff_precision = FLAC__MIN_QLP_COEFF_PRECISION;
2303                                                                 /* ensure a 32-bit datapath throughout for 16bps or less */
2304                                                                 if(subframe_bps <= 16)
2305                                                                         max_qlp_coeff_precision = min(32 - subframe_bps - lpc_order, FLAC__MAX_QLP_COEFF_PRECISION);
2306                                                                 else
2307                                                                         max_qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
2308                                                         }
2309                                                         else {
2310                                                                 min_qlp_coeff_precision = max_qlp_coeff_precision = encoder->protected_->qlp_coeff_precision;
2311                                                         }
2312                                                         for(qlp_coeff_precision = min_qlp_coeff_precision; qlp_coeff_precision <= max_qlp_coeff_precision; qlp_coeff_precision++) {
2313                                                                 _candidate_bits =
2314                                                                         evaluate_lpc_subframe_(
2315                                                                                 encoder,
2316                                                                                 integer_signal,
2317                                                                                 residual[!_best_subframe],
2318                                                                                 encoder->private_->abs_residual,
2319                                                                                 encoder->private_->abs_residual_partition_sums,
2320                                                                                 encoder->private_->raw_bits_per_partition,
2321                                                                                 encoder->private_->lp_coeff[lpc_order-1],
2322                                                                                 frame_header->blocksize,
2323                                                                                 subframe_bps,
2324                                                                                 lpc_order,
2325                                                                                 qlp_coeff_precision,
2326                                                                                 rice_parameter,
2327                                                                                 min_partition_order,
2328                                                                                 max_partition_order,
2329                                                                                 precompute_partition_sums,
2330                                                                                 encoder->protected_->do_escape_coding,
2331                                                                                 encoder->protected_->rice_parameter_search_dist,
2332                                                                                 subframe[!_best_subframe],
2333                                                                                 partitioned_rice_contents[!_best_subframe]
2334                                                                         );
2335                                                                 if(_candidate_bits > 0) { /* if == 0, there was a problem quantizing the lpcoeffs */
2336                                                                         if(_candidate_bits < _best_bits) {
2337                                                                                 _best_subframe = !_best_subframe;
2338                                                                                 _best_bits = _candidate_bits;
2339                                                                         }
2340                                                                 }
2341                                                         }
2342                                                 }
2343                                         }
2344                                 }
2345                         }
2346 #endif /* !defined FLAC__INTEGER_ONLY_LIBRARY */
2347                 }
2348         }
2349
2350         /* under rare circumstances this can happen when all but lpc subframe types are disabled: */
2351         if(_best_bits == UINT_MAX) {
2352                 FLAC__ASSERT(_best_subframe == 0);
2353                 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
2354         }
2355
2356         *best_subframe = _best_subframe;
2357         *best_bits = _best_bits;
2358
2359         return true;
2360 }
2361
2362 FLAC__bool add_subframe_(
2363         FLAC__StreamEncoder *encoder,
2364         const FLAC__FrameHeader *frame_header,
2365         unsigned subframe_bps,
2366         const FLAC__Subframe *subframe,
2367         FLAC__BitBuffer *frame
2368 )
2369 {
2370         switch(subframe->type) {
2371                 case FLAC__SUBFRAME_TYPE_CONSTANT:
2372                         if(!FLAC__subframe_add_constant(&(subframe->data.constant), subframe_bps, subframe->wasted_bits, frame)) {
2373                                 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
2374                                 return false;
2375                         }
2376                         break;
2377                 case FLAC__SUBFRAME_TYPE_FIXED:
2378                         if(!FLAC__subframe_add_fixed(&(subframe->data.fixed), frame_header->blocksize - subframe->data.fixed.order, subframe_bps, subframe->wasted_bits, frame)) {
2379                                 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
2380                                 return false;
2381                         }
2382                         break;
2383                 case FLAC__SUBFRAME_TYPE_LPC:
2384                         if(!FLAC__subframe_add_lpc(&(subframe->data.lpc), frame_header->blocksize - subframe->data.lpc.order, subframe_bps, subframe->wasted_bits, frame)) {
2385                                 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
2386                                 return false;
2387                         }
2388                         break;
2389                 case FLAC__SUBFRAME_TYPE_VERBATIM:
2390                         if(!FLAC__subframe_add_verbatim(&(subframe->data.verbatim), frame_header->blocksize, subframe_bps, subframe->wasted_bits, frame)) {
2391                                 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
2392                                 return false;
2393                         }
2394                         break;
2395                 default:
2396                         FLAC__ASSERT(0);
2397         }
2398
2399         return true;
2400 }
2401
2402 unsigned evaluate_constant_subframe_(
2403         const FLAC__int32 signal,
2404         unsigned subframe_bps,
2405         FLAC__Subframe *subframe
2406 )
2407 {
2408         subframe->type = FLAC__SUBFRAME_TYPE_CONSTANT;
2409         subframe->data.constant.value = signal;
2410
2411         return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + subframe_bps;
2412 }
2413
2414 unsigned evaluate_fixed_subframe_(
2415         FLAC__StreamEncoder *encoder,
2416         const FLAC__int32 signal[],
2417         FLAC__int32 residual[],
2418         FLAC__uint32 abs_residual[],
2419         FLAC__uint64 abs_residual_partition_sums[],
2420         unsigned raw_bits_per_partition[],
2421         unsigned blocksize,
2422         unsigned subframe_bps,
2423         unsigned order,
2424         unsigned rice_parameter,
2425         unsigned min_partition_order,
2426         unsigned max_partition_order,
2427         FLAC__bool precompute_partition_sums,
2428         FLAC__bool do_escape_coding,
2429         unsigned rice_parameter_search_dist,
2430         FLAC__Subframe *subframe,
2431         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2432 )
2433 {
2434         unsigned i, residual_bits;
2435         const unsigned residual_samples = blocksize - order;
2436
2437         FLAC__fixed_compute_residual(signal+order, residual_samples, order, residual);
2438
2439         subframe->type = FLAC__SUBFRAME_TYPE_FIXED;
2440
2441         subframe->data.fixed.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
2442         subframe->data.fixed.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
2443         subframe->data.fixed.residual = residual;
2444
2445         residual_bits =
2446                 find_best_partition_order_(
2447                         encoder->private_,
2448                         residual,
2449                         abs_residual,
2450                         abs_residual_partition_sums,
2451                         raw_bits_per_partition,
2452                         residual_samples,
2453                         order,
2454                         rice_parameter,
2455                         min_partition_order,
2456                         max_partition_order,
2457                         precompute_partition_sums,
2458                         do_escape_coding,
2459                         rice_parameter_search_dist,
2460                         &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2461                 );
2462
2463         subframe->data.fixed.order = order;
2464         for(i = 0; i < order; i++)
2465                 subframe->data.fixed.warmup[i] = signal[i];
2466
2467         return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (order * subframe_bps) + residual_bits;
2468 }
2469
2470 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2471 unsigned evaluate_lpc_subframe_(
2472         FLAC__StreamEncoder *encoder,
2473         const FLAC__int32 signal[],
2474         FLAC__int32 residual[],
2475         FLAC__uint32 abs_residual[],
2476         FLAC__uint64 abs_residual_partition_sums[],
2477         unsigned raw_bits_per_partition[],
2478         const FLAC__real lp_coeff[],
2479         unsigned blocksize,
2480         unsigned subframe_bps,
2481         unsigned order,
2482         unsigned qlp_coeff_precision,
2483         unsigned rice_parameter,
2484         unsigned min_partition_order,
2485         unsigned max_partition_order,
2486         FLAC__bool precompute_partition_sums,
2487         FLAC__bool do_escape_coding,
2488         unsigned rice_parameter_search_dist,
2489         FLAC__Subframe *subframe,
2490         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2491 )
2492 {
2493         FLAC__int32 qlp_coeff[FLAC__MAX_LPC_ORDER];
2494         unsigned i, residual_bits;
2495         int quantization, ret;
2496         const unsigned residual_samples = blocksize - order;
2497
2498         /* try to keep qlp coeff precision such that only 32-bit math is required for decode of <=16bps streams */
2499         if(subframe_bps <= 16) {
2500                 FLAC__ASSERT(order > 0);
2501                 FLAC__ASSERT(order <= FLAC__MAX_LPC_ORDER);
2502                 qlp_coeff_precision = min(qlp_coeff_precision, 32 - subframe_bps - FLAC__bitmath_ilog2(order));
2503         }
2504
2505         ret = FLAC__lpc_quantize_coefficients(lp_coeff, order, qlp_coeff_precision, qlp_coeff, &quantization);
2506         if(ret != 0)
2507                 return 0; /* this is a hack to indicate to the caller that we can't do lp at this order on this subframe */
2508
2509         if(subframe_bps + qlp_coeff_precision + FLAC__bitmath_ilog2(order) <= 32)
2510                 if(subframe_bps <= 16 && qlp_coeff_precision <= 16)
2511                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2512                 else
2513                         encoder->private_->local_lpc_compute_residual_from_qlp_coefficients(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2514         else
2515                 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_64bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2516
2517         subframe->type = FLAC__SUBFRAME_TYPE_LPC;
2518
2519         subframe->data.lpc.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
2520         subframe->data.lpc.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
2521         subframe->data.lpc.residual = residual;
2522
2523         residual_bits =
2524                 find_best_partition_order_(
2525                         encoder->private_,
2526                         residual,
2527                         abs_residual,
2528                         abs_residual_partition_sums,
2529                         raw_bits_per_partition,
2530                         residual_samples,
2531                         order,
2532                         rice_parameter,
2533                         min_partition_order,
2534                         max_partition_order,
2535                         precompute_partition_sums,
2536                         do_escape_coding,
2537                         rice_parameter_search_dist,
2538                         &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2539                 );
2540
2541         subframe->data.lpc.order = order;
2542         subframe->data.lpc.qlp_coeff_precision = qlp_coeff_precision;
2543         subframe->data.lpc.quantization_level = quantization;
2544         memcpy(subframe->data.lpc.qlp_coeff, qlp_coeff, sizeof(FLAC__int32)*FLAC__MAX_LPC_ORDER);
2545         for(i = 0; i < order; i++)
2546                 subframe->data.lpc.warmup[i] = signal[i];
2547
2548         return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + FLAC__SUBFRAME_LPC_QLP_COEFF_PRECISION_LEN + FLAC__SUBFRAME_LPC_QLP_SHIFT_LEN + (order * (qlp_coeff_precision + subframe_bps)) + residual_bits;
2549 }
2550 #endif
2551
2552 unsigned evaluate_verbatim_subframe_(
2553         const FLAC__int32 signal[],
2554         unsigned blocksize,
2555         unsigned subframe_bps,
2556         FLAC__Subframe *subframe
2557 )
2558 {
2559         subframe->type = FLAC__SUBFRAME_TYPE_VERBATIM;
2560
2561         subframe->data.verbatim.data = signal;
2562
2563         return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (blocksize * subframe_bps);
2564 }
2565
2566 unsigned find_best_partition_order_(
2567         FLAC__StreamEncoderPrivate *private_,
2568         const FLAC__int32 residual[],
2569         FLAC__uint32 abs_residual[],
2570         FLAC__uint64 abs_residual_partition_sums[],
2571         unsigned raw_bits_per_partition[],
2572         unsigned residual_samples,
2573         unsigned predictor_order,
2574         unsigned rice_parameter,
2575         unsigned min_partition_order,
2576         unsigned max_partition_order,
2577         FLAC__bool precompute_partition_sums,
2578         FLAC__bool do_escape_coding,
2579         unsigned rice_parameter_search_dist,
2580         FLAC__EntropyCodingMethod_PartitionedRice *best_partitioned_rice
2581 )
2582 {
2583         FLAC__int32 r;
2584         unsigned residual_bits, best_residual_bits = 0;
2585         unsigned residual_sample;
2586         unsigned best_parameters_index = 0;
2587         const unsigned blocksize = residual_samples + predictor_order;
2588
2589         /* compute abs(residual) for use later */
2590         for(residual_sample = 0; residual_sample < residual_samples; residual_sample++) {
2591                 r = residual[residual_sample];
2592                 abs_residual[residual_sample] = (FLAC__uint32)(r<0? -r : r);
2593         }
2594
2595         max_partition_order = FLAC__format_get_max_rice_partition_order_from_blocksize_limited_max_and_predictor_order(max_partition_order, blocksize, predictor_order);
2596         min_partition_order = min(min_partition_order, max_partition_order);
2597
2598         if(precompute_partition_sums) {
2599                 int partition_order;
2600                 unsigned sum;
2601
2602                 precompute_partition_info_sums_(abs_residual, abs_residual_partition_sums, residual_samples, predictor_order, min_partition_order, max_partition_order);
2603
2604                 if(do_escape_coding)
2605                         precompute_partition_info_escapes_(residual, raw_bits_per_partition, residual_samples, predictor_order, min_partition_order, max_partition_order);
2606
2607                 for(partition_order = (int)max_partition_order, sum = 0; partition_order >= (int)min_partition_order; partition_order--) {
2608 #ifdef DONT_ESTIMATE_RICE_BITS
2609                         if(!
2610                                 set_partitioned_rice_with_precompute_(
2611                                         residual,
2612                                         abs_residual_partition_sums+sum,
2613                                         raw_bits_per_partition+sum,
2614                                         residual_samples,
2615                                         predictor_order,
2616                                         rice_parameter,
2617                                         rice_parameter_search_dist,
2618                                         (unsigned)partition_order,
2619                                         do_escape_coding,
2620                                         &private_->partitioned_rice_contents_extra[!best_parameters_index],
2621                                         &residual_bits
2622                                 )
2623                         )
2624 #else
2625                         if(!
2626                                 set_partitioned_rice_with_precompute_(
2627                                         abs_residual,
2628                                         abs_residual_partition_sums+sum,
2629                                         raw_bits_per_partition+sum,
2630                                         residual_samples,
2631                                         predictor_order,
2632                                         rice_parameter,
2633                                         rice_parameter_search_dist,
2634                                         (unsigned)partition_order,
2635                                         do_escape_coding,
2636                                         &private_->partitioned_rice_contents_extra[!best_parameters_index],
2637                                         &residual_bits
2638                                 )
2639                         )
2640 #endif
2641                         {
2642                                 FLAC__ASSERT(best_residual_bits != 0);
2643                                 break;
2644                         }
2645                         sum += 1u << partition_order;
2646                         if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2647                                 best_residual_bits = residual_bits;
2648                                 best_parameters_index = !best_parameters_index;
2649                                 best_partitioned_rice->order = partition_order;
2650                         }
2651                 }
2652         }
2653         else {
2654                 unsigned partition_order;
2655                 for(partition_order = min_partition_order; partition_order <= max_partition_order; partition_order++) {
2656 #ifdef DONT_ESTIMATE_RICE_BITS
2657                         if(!
2658                                 set_partitioned_rice_(
2659                                         abs_residual,
2660                                         residual,
2661                                         residual_samples,
2662                                         predictor_order,
2663                                         rice_parameter,
2664                                         rice_parameter_search_dist,
2665                                         partition_order,
2666                                         &private_->partitioned_rice_contents_extra[!best_parameters_index],
2667                                         &residual_bits
2668                                 )
2669                         )
2670 #else
2671                         if(!
2672                                 set_partitioned_rice_(
2673                                         abs_residual,
2674                                         residual_samples,
2675                                         predictor_order,
2676                                         rice_parameter,
2677                                         rice_parameter_search_dist,
2678                                         partition_order,
2679                                         &private_->partitioned_rice_contents_extra[!best_parameters_index],
2680                                         &residual_bits
2681                                 )
2682                         )
2683 #endif
2684                         {
2685                                 FLAC__ASSERT(best_residual_bits != 0);
2686                                 break;
2687                         }
2688                         if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2689                                 best_residual_bits = residual_bits;
2690                                 best_parameters_index = !best_parameters_index;
2691                                 best_partitioned_rice->order = partition_order;
2692                         }
2693                 }
2694         }
2695
2696         /*
2697          * We are allowed to de-const the pointer based on our special knowledge;
2698          * it is const to the outside world.
2699          */
2700         {
2701                 FLAC__EntropyCodingMethod_PartitionedRiceContents* best_partitioned_rice_contents = (FLAC__EntropyCodingMethod_PartitionedRiceContents*)best_partitioned_rice->contents;
2702                 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(best_partitioned_rice_contents, max(6, best_partitioned_rice->order));
2703                 memcpy(best_partitioned_rice_contents->parameters, private_->partitioned_rice_contents_extra[best_parameters_index].parameters, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2704                 memcpy(best_partitioned_rice_contents->raw_bits, private_->partitioned_rice_contents_extra[best_parameters_index].raw_bits, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2705         }
2706
2707         return best_residual_bits;
2708 }
2709
2710 void precompute_partition_info_sums_(
2711         const FLAC__uint32 abs_residual[],
2712         FLAC__uint64 abs_residual_partition_sums[],
2713         unsigned residual_samples,
2714         unsigned predictor_order,
2715         unsigned min_partition_order,
2716         unsigned max_partition_order
2717 )
2718 {
2719         int partition_order;
2720         unsigned from_partition, to_partition = 0;
2721         const unsigned blocksize = residual_samples + predictor_order;
2722
2723         /* first do max_partition_order */
2724         for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
2725                 FLAC__uint64 abs_residual_partition_sum;
2726                 FLAC__uint32 abs_r;
2727                 unsigned partition, partition_sample, partition_samples, residual_sample;
2728                 const unsigned partitions = 1u << partition_order;
2729                 const unsigned default_partition_samples = blocksize >> partition_order;
2730
2731                 FLAC__ASSERT(default_partition_samples > predictor_order);
2732
2733                 for(partition = residual_sample = 0; partition < partitions; partition++) {
2734                         partition_samples = default_partition_samples;
2735                         if(partition == 0)
2736                                 partition_samples -= predictor_order;
2737                         abs_residual_partition_sum = 0;
2738                         for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2739                                 abs_r = abs_residual[residual_sample];
2740                                 abs_residual_partition_sum += abs_r;
2741                                 residual_sample++;
2742                         }
2743                         abs_residual_partition_sums[partition] = abs_residual_partition_sum;
2744                 }
2745                 to_partition = partitions;
2746                 break;
2747         }
2748
2749         /* now merge partitions for lower orders */
2750         for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2751                 FLAC__uint64 s;
2752                 unsigned i;
2753                 const unsigned partitions = 1u << partition_order;
2754                 for(i = 0; i < partitions; i++) {
2755                         s = abs_residual_partition_sums[from_partition];
2756                         from_partition++;
2757                         abs_residual_partition_sums[to_partition] = s + abs_residual_partition_sums[from_partition];
2758                         from_partition++;
2759                         to_partition++;
2760                 }
2761         }
2762 }
2763
2764 void precompute_partition_info_escapes_(
2765         const FLAC__int32 residual[],
2766         unsigned raw_bits_per_partition[],
2767         unsigned residual_samples,
2768         unsigned predictor_order,
2769         unsigned min_partition_order,
2770         unsigned max_partition_order
2771 )
2772 {
2773         int partition_order;
2774         unsigned from_partition, to_partition = 0;
2775         const unsigned blocksize = residual_samples + predictor_order;
2776
2777         /* first do max_partition_order */
2778         for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
2779                 FLAC__int32 r, residual_partition_min, residual_partition_max;
2780                 unsigned silog2_min, silog2_max;
2781                 unsigned partition, partition_sample, partition_samples, residual_sample;
2782                 const unsigned partitions = 1u << partition_order;
2783                 const unsigned default_partition_samples = blocksize >> partition_order;
2784
2785                 FLAC__ASSERT(default_partition_samples > predictor_order);
2786
2787                 for(partition = residual_sample = 0; partition < partitions; partition++) {
2788                         partition_samples = default_partition_samples;
2789                         if(partition == 0)
2790                                 partition_samples -= predictor_order;
2791                         residual_partition_min = residual_partition_max = 0;
2792                         for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2793                                 r = residual[residual_sample];
2794                                 if(r < residual_partition_min)
2795                                         residual_partition_min = r;
2796                                 else if(r > residual_partition_max)
2797                                         residual_partition_max = r;
2798                                 residual_sample++;
2799                         }
2800                         silog2_min = FLAC__bitmath_silog2(residual_partition_min);
2801                         silog2_max = FLAC__bitmath_silog2(residual_partition_max);
2802                         raw_bits_per_partition[partition] = max(silog2_min, silog2_max);
2803                 }
2804                 to_partition = partitions;
2805                 break;
2806         }
2807
2808         /* now merge partitions for lower orders */
2809         for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2810                 unsigned m;
2811                 unsigned i;
2812                 const unsigned partitions = 1u << partition_order;
2813                 for(i = 0; i < partitions; i++) {
2814                         m = raw_bits_per_partition[from_partition];
2815                         from_partition++;
2816                         raw_bits_per_partition[to_partition] = max(m, raw_bits_per_partition[from_partition]);
2817                         from_partition++;
2818                         to_partition++;
2819                 }
2820         }
2821 }
2822
2823 #ifdef VARIABLE_RICE_BITS
2824 #undef VARIABLE_RICE_BITS
2825 #endif
2826 #ifndef DONT_ESTIMATE_RICE_BITS
2827 #define VARIABLE_RICE_BITS(value, parameter) ((value) >> (parameter))
2828 #endif
2829
2830 #ifdef DONT_ESTIMATE_RICE_BITS
2831 FLAC__bool set_partitioned_rice_(
2832         const FLAC__uint32 abs_residual[],
2833         const FLAC__int32 residual[],
2834         const unsigned residual_samples,
2835         const unsigned predictor_order,
2836         const unsigned suggested_rice_parameter,
2837         const unsigned rice_parameter_search_dist,
2838         const unsigned partition_order,
2839         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2840         unsigned *bits
2841 )
2842 #else
2843 FLAC__bool set_partitioned_rice_(
2844         const FLAC__uint32 abs_residual[],
2845         const unsigned residual_samples,
2846         const unsigned predictor_order,
2847         const unsigned suggested_rice_parameter,
2848         const unsigned rice_parameter_search_dist,
2849         const unsigned partition_order,
2850         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2851         unsigned *bits
2852 )
2853 #endif
2854 {
2855         unsigned rice_parameter, partition_bits;
2856 #ifndef NO_RICE_SEARCH
2857         unsigned best_partition_bits;
2858         unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
2859 #endif
2860         unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
2861         unsigned *parameters;
2862
2863         FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
2864
2865         FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
2866         parameters = partitioned_rice_contents->parameters;
2867
2868         if(partition_order == 0) {
2869                 unsigned i;
2870
2871 #ifndef NO_RICE_SEARCH
2872                 if(rice_parameter_search_dist) {
2873                         if(suggested_rice_parameter < rice_parameter_search_dist)
2874                                 min_rice_parameter = 0;
2875                         else
2876                                 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
2877                         max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
2878                         if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2879 #ifdef DEBUG_VERBOSE
2880                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @2\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2881 #endif
2882                                 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2883                         }
2884                 }
2885                 else
2886                         min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
2887
2888                 best_partition_bits = 0xffffffff;
2889                 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2890 #endif
2891 #ifdef VARIABLE_RICE_BITS
2892 #ifdef FLAC__SYMMETRIC_RICE
2893                         partition_bits = (2+rice_parameter) * residual_samples;
2894 #else
2895                         const unsigned rice_parameter_estimate = rice_parameter-1;
2896                         partition_bits = (1+rice_parameter) * residual_samples;
2897 #endif
2898 #else
2899                         partition_bits = 0;
2900 #endif
2901                         partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2902                         for(i = 0; i < residual_samples; i++) {
2903 #ifdef VARIABLE_RICE_BITS
2904 #ifdef FLAC__SYMMETRIC_RICE
2905                                 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
2906 #else
2907                                 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
2908 #endif
2909 #else
2910                                 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
2911 #endif
2912                         }
2913 #ifndef NO_RICE_SEARCH
2914                         if(partition_bits < best_partition_bits) {
2915                                 best_rice_parameter = rice_parameter;
2916                                 best_partition_bits = partition_bits;
2917                         }
2918                 }
2919 #endif
2920                 parameters[0] = best_rice_parameter;
2921                 bits_ += best_partition_bits;
2922         }
2923         else {
2924                 unsigned partition, residual_sample, save_residual_sample, partition_sample;
2925                 unsigned partition_samples;
2926                 FLAC__uint64 mean, k;
2927                 const unsigned partitions = 1u << partition_order;
2928                 for(partition = residual_sample = 0; partition < partitions; partition++) {
2929                         partition_samples = (residual_samples+predictor_order) >> partition_order;
2930                         if(partition == 0) {
2931                                 if(partition_samples <= predictor_order)
2932                                         return false;
2933                                 else
2934                                         partition_samples -= predictor_order;
2935                         }
2936                         mean = 0;
2937                         save_residual_sample = residual_sample;
2938                         for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++)
2939                                 mean += abs_residual[residual_sample];
2940                         residual_sample = save_residual_sample;
2941 #ifdef FLAC__SYMMETRIC_RICE
2942                         mean += partition_samples >> 1; /* for rounding effect */
2943                         mean /= partition_samples;
2944
2945                         /* calc rice_parameter = floor(log2(mean)) */
2946                         rice_parameter = 0;
2947                         mean>>=1;
2948                         while(mean) {
2949                                 rice_parameter++;
2950                                 mean >>= 1;
2951                         }
2952 #else
2953                         /* calc rice_parameter ala LOCO-I */
2954                         for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
2955                                 ;
2956 #endif
2957                         if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2958 #ifdef DEBUG_VERBOSE
2959                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @3\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2960 #endif
2961                                 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2962                         }
2963
2964 #ifndef NO_RICE_SEARCH
2965                         if(rice_parameter_search_dist) {
2966                                 if(rice_parameter < rice_parameter_search_dist)
2967                                         min_rice_parameter = 0;
2968                                 else
2969                                         min_rice_parameter = rice_parameter - rice_parameter_search_dist;
2970                                 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
2971                                 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2972 #ifdef DEBUG_VERBOSE
2973                                         fprintf(stderr, "clipping rice_parameter (%u -> %u) @4\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2974 #endif
2975                                         max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2976                                 }
2977                         }
2978                         else
2979                                 min_rice_parameter = max_rice_parameter = rice_parameter;
2980
2981                         best_partition_bits = 0xffffffff;
2982                         for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2983 #endif
2984 #ifdef VARIABLE_RICE_BITS
2985 #ifdef FLAC__SYMMETRIC_RICE
2986                                 partition_bits = (2+rice_parameter) * partition_samples;
2987 #else
2988                                 const unsigned rice_parameter_estimate = rice_parameter-1;
2989                                 partition_bits = (1+rice_parameter) * partition_samples;
2990 #endif
2991 #else
2992                                 partition_bits = 0;
2993 #endif
2994                                 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2995                                 save_residual_sample = residual_sample;
2996                                 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
2997 #ifdef VARIABLE_RICE_BITS
2998 #ifdef FLAC__SYMMETRIC_RICE
2999                                         partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
3000 #else
3001                                         partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
3002 #endif
3003 #else
3004                                         partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
3005 #endif
3006                                 }
3007 #ifndef NO_RICE_SEARCH
3008                                 if(rice_parameter != max_rice_parameter)
3009                                         residual_sample = save_residual_sample;
3010                                 if(partition_bits < best_partition_bits) {
3011                                         best_rice_parameter = rice_parameter;
3012                                         best_partition_bits = partition_bits;
3013                                 }
3014                         }
3015 #endif
3016                         parameters[partition] = best_rice_parameter;
3017                         bits_ += best_partition_bits;
3018                 }
3019         }
3020
3021         *bits = bits_;
3022         return true;
3023 }
3024
3025 #ifdef DONT_ESTIMATE_RICE_BITS
3026 FLAC__bool set_partitioned_rice_with_precompute_(
3027         const FLAC__int32 residual[],
3028         const FLAC__uint64 abs_residual_partition_sums[],
3029         const unsigned raw_bits_per_partition[],
3030         const unsigned residual_samples,
3031         const unsigned predictor_order,
3032         const unsigned suggested_rice_parameter,
3033         const unsigned rice_parameter_search_dist,
3034         const unsigned partition_order,
3035         const FLAC__bool search_for_escapes,
3036         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
3037         unsigned *bits
3038 )
3039 #else
3040 FLAC__bool set_partitioned_rice_with_precompute_(
3041         const FLAC__uint32 abs_residual[],
3042         const FLAC__uint64 abs_residual_partition_sums[],
3043         const unsigned raw_bits_per_partition[],
3044         const unsigned residual_samples,
3045         const unsigned predictor_order,
3046         const unsigned suggested_rice_parameter,
3047         const unsigned rice_parameter_search_dist,
3048         const unsigned partition_order,
3049         const FLAC__bool search_for_escapes,
3050         FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
3051         unsigned *bits
3052 )
3053 #endif
3054 {
3055         unsigned rice_parameter, partition_bits;
3056 #ifndef NO_RICE_SEARCH
3057         unsigned best_partition_bits;
3058         unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
3059 #endif
3060         unsigned flat_bits;
3061         unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
3062         unsigned *parameters, *raw_bits;
3063
3064         FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
3065
3066         FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
3067         parameters = partitioned_rice_contents->parameters;
3068         raw_bits = partitioned_rice_contents->raw_bits;
3069
3070         if(partition_order == 0) {
3071                 unsigned i;
3072
3073 #ifndef NO_RICE_SEARCH
3074                 if(rice_parameter_search_dist) {
3075                         if(suggested_rice_parameter < rice_parameter_search_dist)
3076                                 min_rice_parameter = 0;
3077                         else
3078                                 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
3079                         max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
3080                         if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
3081 #ifdef DEBUG_VERBOSE
3082                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @5\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3083 #endif
3084                                 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3085                         }
3086                 }
3087                 else
3088                         min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
3089
3090                 best_partition_bits = 0xffffffff;
3091                 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3092 #endif
3093 #ifdef VARIABLE_RICE_BITS
3094 #ifdef FLAC__SYMMETRIC_RICE
3095                         partition_bits = (2+rice_parameter) * residual_samples;
3096 #else
3097                         const unsigned rice_parameter_estimate = rice_parameter-1;
3098                         partition_bits = (1+rice_parameter) * residual_samples;
3099 #endif
3100 #else
3101                         partition_bits = 0;
3102 #endif
3103                         partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
3104                         for(i = 0; i < residual_samples; i++) {
3105 #ifdef VARIABLE_RICE_BITS
3106 #ifdef FLAC__SYMMETRIC_RICE
3107                                 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
3108 #else
3109                                 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
3110 #endif
3111 #else
3112                                 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
3113 #endif
3114                         }
3115 #ifndef NO_RICE_SEARCH
3116                         if(partition_bits < best_partition_bits) {
3117                                 best_rice_parameter = rice_parameter;
3118                                 best_partition_bits = partition_bits;
3119                         }
3120                 }
3121 #endif
3122                 if(search_for_escapes) {
3123                         flat_bits = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_RAW_LEN + raw_bits_per_partition[0] * residual_samples;
3124                         if(flat_bits <= best_partition_bits) {
3125                                 raw_bits[0] = raw_bits_per_partition[0];
3126                                 best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3127                                 best_partition_bits = flat_bits;
3128                         }
3129                 }
3130                 parameters[0] = best_rice_parameter;
3131                 bits_ += best_partition_bits;
3132         }
3133         else {
3134                 unsigned partition, residual_sample, save_residual_sample, partition_sample;
3135                 unsigned partition_samples;
3136                 FLAC__uint64 mean, k;
3137                 const unsigned partitions = 1u << partition_order;
3138                 for(partition = residual_sample = 0; partition < partitions; partition++) {
3139                         partition_samples = (residual_samples+predictor_order) >> partition_order;
3140                         if(partition == 0) {
3141                                 if(partition_samples <= predictor_order)
3142                                         return false;
3143                                 else
3144                                         partition_samples -= predictor_order;
3145                         }
3146                         mean = abs_residual_partition_sums[partition];
3147 #ifdef FLAC__SYMMETRIC_RICE
3148                         mean += partition_samples >> 1; /* for rounding effect */
3149                         mean /= partition_samples;
3150
3151                         /* calc rice_parameter = floor(log2(mean)) */
3152                         rice_parameter = 0;
3153                         mean>>=1;
3154                         while(mean) {
3155                                 rice_parameter++;
3156                                 mean >>= 1;
3157                         }
3158 #else
3159                         /* calc rice_parameter ala LOCO-I */
3160                         for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
3161                                 ;
3162 #endif
3163                         if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
3164 #ifdef DEBUG_VERBOSE
3165                                 fprintf(stderr, "clipping rice_parameter (%u -> %u) @6\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3166 #endif
3167                                 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3168                         }
3169
3170 #ifndef NO_RICE_SEARCH
3171                         if(rice_parameter_search_dist) {
3172                                 if(rice_parameter < rice_parameter_search_dist)
3173                                         min_rice_parameter = 0;
3174                                 else
3175                                         min_rice_parameter = rice_parameter - rice_parameter_search_dist;
3176                                 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
3177                                 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
3178 #ifdef DEBUG_VERBOSE
3179                                         fprintf(stderr, "clipping rice_parameter (%u -> %u) @7\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3180 #endif
3181                                         max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3182                                 }
3183                         }
3184                         else
3185                                 min_rice_parameter = max_rice_parameter = rice_parameter;
3186
3187                         best_partition_bits = 0xffffffff;
3188                         for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3189 #endif
3190 #ifdef VARIABLE_RICE_BITS
3191 #ifdef FLAC__SYMMETRIC_RICE
3192                                 partition_bits = (2+rice_parameter) * partition_samples;
3193 #else
3194                                 const unsigned rice_parameter_estimate = rice_parameter-1;
3195                                 partition_bits = (1+rice_parameter) * partition_samples;
3196 #endif
3197 #else
3198                                 partition_bits = 0;
3199 #endif
3200                                 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
3201                                 save_residual_sample = residual_sample;
3202                                 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
3203 #ifdef VARIABLE_RICE_BITS
3204 #ifdef FLAC__SYMMETRIC_RICE
3205                                         partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
3206 #else
3207                                         partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
3208 #endif
3209 #else
3210                                         partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
3211 #endif
3212                                 }
3213 #ifndef NO_RICE_SEARCH
3214                                 if(rice_parameter != max_rice_parameter)
3215                                         residual_sample = save_residual_sample;
3216                                 if(partition_bits < best_partition_bits) {
3217                                         best_rice_parameter = rice_parameter;
3218                                         best_partition_bits = partition_bits;
3219                                 }
3220                         }
3221 #endif
3222                         if(search_for_escapes) {
3223                                 flat_bits = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_RAW_LEN + raw_bits_per_partition[partition] * partition_samples;
3224                                 if(flat_bits <= best_partition_bits) {
3225                                         raw_bits[partition] = raw_bits_per_partition[partition];
3226                                         best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3227                                         best_partition_bits = flat_bits;
3228                                 }
3229                         }
3230                         parameters[partition] = best_rice_parameter;
3231                         bits_ += best_partition_bits;
3232                 }
3233         }
3234
3235         *bits = bits_;
3236         return true;
3237 }
3238
3239 unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples)
3240 {
3241         unsigned i, shift;
3242         FLAC__int32 x = 0;
3243
3244         for(i = 0; i < samples && !(x&1); i++)
3245                 x |= signal[i];
3246
3247         if(x == 0) {
3248                 shift = 0;
3249         }
3250         else {
3251                 for(shift = 0; !(x&1); shift++)
3252                         x >>= 1;
3253         }
3254
3255         if(shift > 0) {
3256                 for(i = 0; i < samples; i++)
3257                          signal[i] >>= shift;
3258         }
3259
3260         return shift;
3261 }
3262
3263 void append_to_verify_fifo_(verify_input_fifo *fifo, const FLAC__int32 * const input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3264 {
3265         unsigned channel;
3266
3267         for(channel = 0; channel < channels; channel++)
3268                 memcpy(&fifo->data[channel][fifo->tail], &input[channel][input_offset], sizeof(FLAC__int32) * wide_samples);
3269
3270         fifo->tail += wide_samples;
3271
3272         FLAC__ASSERT(fifo->tail <= fifo->size);
3273 }
3274
3275 void append_to_verify_fifo_interleaved_(verify_input_fifo *fifo, const FLAC__int32 input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3276 {
3277         unsigned channel;
3278         unsigned sample, wide_sample;
3279         unsigned tail = fifo->tail;
3280
3281         sample = input_offset * channels;
3282         for(wide_sample = 0; wide_sample < wide_samples; wide_sample++) {
3283                 for(channel = 0; channel < channels; channel++)
3284                         fifo->data[channel][tail] = input[sample++];
3285                 tail++;
3286         }
3287         fifo->tail = tail;
3288
3289         FLAC__ASSERT(fifo->tail <= fifo->size);
3290 }
3291
3292 FLAC__StreamDecoderReadStatus verify_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], unsigned *bytes, void *client_data)
3293 {
3294         FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3295         const unsigned encoded_bytes = encoder->private_->verify.output.bytes;
3296         (void)decoder;
3297
3298         if(encoder->private_->verify.needs_magic_hack) {
3299                 FLAC__ASSERT(*bytes >= FLAC__STREAM_SYNC_LENGTH);
3300                 *bytes = FLAC__STREAM_SYNC_LENGTH;
3301                 memcpy(buffer, FLAC__STREAM_SYNC_STRING, *bytes);
3302                 encoder->private_->verify.needs_magic_hack = false;
3303         }
3304         else {
3305                 if(encoded_bytes == 0) {
3306                         /*
3307                          * If we get here, a FIFO underflow has occurred,
3308                          * which means there is a bug somewhere.
3309                          */
3310                         FLAC__ASSERT(0);
3311                         return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
3312                 }
3313                 else if(encoded_bytes < *bytes)
3314                         *bytes = encoded_bytes;
3315                 memcpy(buffer, encoder->private_->verify.output.data, *bytes);
3316                 encoder->private_->verify.output.data += *bytes;
3317                 encoder->private_->verify.output.bytes -= *bytes;
3318         }
3319
3320         return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
3321 }
3322
3323 FLAC__StreamDecoderWriteStatus verify_write_callback_(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
3324 {
3325         FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder *)client_data;
3326         unsigned channel;
3327         const unsigned channels = FLAC__stream_decoder_get_channels(decoder);
3328         const unsigned blocksize = frame->header.blocksize;
3329         const unsigned bytes_per_block = sizeof(FLAC__int32) * blocksize;
3330
3331         for(channel = 0; channel < channels; channel++) {
3332                 if(0 != memcmp(buffer[channel], encoder->private_->verify.input_fifo.data[channel], bytes_per_block)) {
3333                         unsigned i, sample = 0;
3334                         FLAC__int32 expect = 0, got = 0;
3335
3336                         for(i = 0; i < blocksize; i++) {
3337                                 if(buffer[channel][i] != encoder->private_->verify.input_fifo.data[channel][i]) {
3338                                         sample = i;
3339                                         expect = (FLAC__int32)encoder->private_->verify.input_fifo.data[channel][i];
3340                                         got = (FLAC__int32)buffer[channel][i];
3341                                         break;
3342                                 }
3343                         }
3344                         FLAC__ASSERT(i < blocksize);
3345                         FLAC__ASSERT(frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
3346                         encoder->private_->verify.error_stats.absolute_sample = frame->header.number.sample_number + sample;
3347                         encoder->private_->verify.error_stats.frame_number = (unsigned)(frame->header.number.sample_number / blocksize);
3348                         encoder->private_->verify.error_stats.channel = channel;
3349                         encoder->private_->verify.error_stats.sample = sample;
3350                         encoder->private_->verify.error_stats.expected = expect;
3351                         encoder->private_->verify.error_stats.got = got;
3352                         encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA;
3353                         return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
3354                 }
3355         }
3356         /* dequeue the frame from the fifo */
3357         for(channel = 0; channel < channels; channel++) {
3358                 memmove(&encoder->private_->verify.input_fifo.data[channel][0], &encoder->private_->verify.input_fifo.data[channel][blocksize], encoder->private_->verify.input_fifo.tail - blocksize);
3359         }
3360         encoder->private_->verify.input_fifo.tail -= blocksize;
3361         return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
3362 }
3363
3364 void verify_metadata_callback_(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
3365 {
3366         (void)decoder, (void)metadata, (void)client_data;
3367 }
3368
3369 void verify_error_callback_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
3370 {
3371         FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3372         (void)decoder, (void)status;
3373         encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
3374 }