1 /* libFLAC - Free Lossless Audio Codec library
2 * Copyright (C) 2000,2001,2002,2003,2004,2005 Josh Coalson
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
8 * - Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
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.
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.
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.
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"
57 #define min(x,y) ((x)<(y)?(x):(y))
62 #define max(x,y) ((x)>(y)?(x):(y))
65 FLAC__int32 *data[FLAC__MAX_CHANNELS];
66 unsigned size; /* of each data[] in samples */
71 const FLAC__byte *data;
78 ENCODER_IN_METADATA = 1,
82 /***********************************************************************
84 * Private class method prototypes
86 ***********************************************************************/
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);
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[],
106 FLAC__Subframe *subframe[2],
107 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
108 FLAC__int32 *residual[2],
109 unsigned *best_subframe,
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
121 static unsigned evaluate_constant_subframe_(
122 const FLAC__int32 signal,
123 unsigned subframe_bps,
124 FLAC__Subframe *subframe
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[],
135 unsigned subframe_bps,
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
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[],
157 unsigned subframe_bps,
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
171 static unsigned evaluate_verbatim_subframe_(
172 const FLAC__int32 signal[],
174 unsigned subframe_bps,
175 FLAC__Subframe *subframe
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
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
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
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,
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,
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,
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,
266 static unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples);
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,
274 unsigned wide_samples
277 static void append_to_verify_fifo_interleaved_(
278 verify_input_fifo *fifo,
279 const FLAC__int32 input[],
280 unsigned input_offset,
282 unsigned wide_samples
285 static FLAC__StreamDecoderReadStatus verify_read_callback_(
286 const FLAC__StreamDecoder *decoder,
292 static FLAC__StreamDecoderWriteStatus verify_write_callback_(
293 const FLAC__StreamDecoder *decoder,
294 const FLAC__Frame *frame,
295 const FLAC__int32 * const buffer[],
299 static void verify_metadata_callback_(
300 const FLAC__StreamDecoder *decoder,
301 const FLAC__StreamMetadata *metadata,
305 static void verify_error_callback_(
306 const FLAC__StreamDecoder *decoder,
307 FLAC__StreamDecoderErrorStatus status,
312 /***********************************************************************
316 ***********************************************************************/
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) */
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]);
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]);
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[]);
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;
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];
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;
388 * These fields have been moved here from private function local
389 * declarations merely to save stack space during encoding.
391 #ifndef FLAC__INTEGER_ONLY_LIBRARY
392 FLAC__real lp_coeff[FLAC__MAX_LPC_ORDER][FLAC__MAX_LPC_ORDER]; /* from process_subframe_() */
394 FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents_extra[2]; /* from find_best_partition_order_() */
396 * The data for the verify section
399 FLAC__StreamDecoder *decoder;
400 EncoderStateHint state_hint;
401 FLAC__bool needs_magic_hack;
402 verify_input_fifo input_fifo;
403 verify_output output;
405 FLAC__uint64 absolute_sample;
406 unsigned frame_number;
409 FLAC__int32 expected;
413 FLAC__bool is_being_deleted; /* if true, call to ..._finish() from ..._delete() will not call the callbacks */
414 } FLAC__StreamEncoderPrivate;
416 /***********************************************************************
418 * Public static class data
420 ***********************************************************************/
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"
447 FLAC_API const char * const FLAC__StreamEncoderWriteStatusString[] = {
448 "FLAC__STREAM_ENCODER_WRITE_STATUS_OK",
449 "FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR"
452 /***********************************************************************
454 * Class constructor/destructor
457 FLAC_API FLAC__StreamEncoder *FLAC__stream_encoder_new()
459 FLAC__StreamEncoder *encoder;
462 FLAC__ASSERT(sizeof(int) >= 4); /* we want to die right away if this is not true */
464 encoder = (FLAC__StreamEncoder*)calloc(1, sizeof(FLAC__StreamEncoder));
469 encoder->protected_ = (FLAC__StreamEncoderProtected*)calloc(1, sizeof(FLAC__StreamEncoderProtected));
470 if(encoder->protected_ == 0) {
475 encoder->private_ = (FLAC__StreamEncoderPrivate*)calloc(1, sizeof(FLAC__StreamEncoderPrivate));
476 if(encoder->private_ == 0) {
477 free(encoder->protected_);
482 encoder->private_->frame = FLAC__bitbuffer_new();
483 if(encoder->private_->frame == 0) {
484 free(encoder->private_);
485 free(encoder->protected_);
490 set_defaults_(encoder);
492 encoder->private_->is_being_deleted = false;
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];
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];
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];
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];
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]);
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]);
519 for(i = 0; i < 2; i++)
520 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_extra[i]);
522 encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
527 FLAC_API void FLAC__stream_encoder_delete(FLAC__StreamEncoder *encoder)
531 FLAC__ASSERT(0 != encoder);
532 FLAC__ASSERT(0 != encoder->protected_);
533 FLAC__ASSERT(0 != encoder->private_);
534 FLAC__ASSERT(0 != encoder->private_->frame);
536 encoder->private_->is_being_deleted = true;
538 FLAC__stream_encoder_finish(encoder);
540 if(0 != encoder->private_->verify.decoder)
541 FLAC__stream_decoder_delete(encoder->private_->verify.decoder);
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]);
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]);
551 for(i = 0; i < 2; i++)
552 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_extra[i]);
554 FLAC__bitbuffer_delete(encoder->private_->frame);
555 free(encoder->private_);
556 free(encoder->protected_);
560 /***********************************************************************
562 * Public class methods
564 ***********************************************************************/
566 FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_init(FLAC__StreamEncoder *encoder)
569 FLAC__bool metadata_has_seektable, metadata_has_vorbis_comment;
571 FLAC__ASSERT(0 != encoder);
573 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
574 return encoder->protected_->state = FLAC__STREAM_ENCODER_ALREADY_INITIALIZED;
576 encoder->protected_->state = FLAC__STREAM_ENCODER_OK;
578 if(0 == encoder->private_->write_callback || 0 == encoder->private_->metadata_callback)
579 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_CALLBACK;
581 if(encoder->protected_->channels == 0 || encoder->protected_->channels > FLAC__MAX_CHANNELS)
582 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_NUMBER_OF_CHANNELS;
584 if(encoder->protected_->do_mid_side_stereo && encoder->protected_->channels != 2)
585 return encoder->protected_->state = FLAC__STREAM_ENCODER_MID_SIDE_CHANNELS_MISMATCH;
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;
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 */
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;
596 if(!FLAC__format_sample_rate_is_valid(encoder->protected_->sample_rate))
597 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_SAMPLE_RATE;
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;
602 if(encoder->protected_->max_lpc_order > FLAC__MAX_LPC_ORDER)
603 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_MAX_LPC_ORDER;
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;
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);
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;
628 encoder->protected_->qlp_coeff_precision = 13;
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;
636 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
638 FLAC__ASSERT(encoder->protected_->qlp_coeff_precision <= FLAC__MAX_QLP_COEFF_PRECISION);
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;
643 if(encoder->protected_->streamable_subset) {
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
658 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
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
669 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
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
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;
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;
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;
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;
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;
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;
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;
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;
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;
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);
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÷ 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);
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;
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 */
761 * get the CPU info and set the function pointers
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;
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;
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
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;
790 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32;
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;
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;
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;
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 */
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;
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;
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;
826 if(!FLAC__bitbuffer_init(encoder->private_->frame))
827 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
830 * Set up the verify stuff if necessary
832 if(encoder->protected_->verify) {
834 * First, set up the fifo which will hold the
835 * original signal to compare against
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;
842 encoder->private_->verify.input_fifo.tail = 0;
845 * Now set up a stream decoder for verification
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;
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;
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;
867 * write the stream header
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;
879 * write the STREAMINFO metadata block
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;
906 * Now that the STREAMINFO block is written, we can init this to an
907 * absurdly-high value...
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;
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).
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.)
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;
943 * write the user's metadata blocks
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;
957 if(encoder->protected_->verify)
958 encoder->private_->verify.state_hint = ENCODER_IN_AUDIO;
960 return encoder->protected_->state;
963 FLAC_API void FLAC__stream_encoder_finish(FLAC__StreamEncoder *encoder)
965 FLAC__ASSERT(0 != encoder);
967 if(encoder->protected_->state == FLAC__STREAM_ENCODER_UNINITIALIZED)
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 */
977 FLAC__MD5Final(encoder->private_->metadata.data.stream_info.md5sum, &encoder->private_->md5context);
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);
983 if(encoder->protected_->verify && 0 != encoder->private_->verify.decoder)
984 FLAC__stream_decoder_finish(encoder->private_->verify.decoder);
987 set_defaults_(encoder);
989 encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
992 FLAC_API FLAC__bool FLAC__stream_encoder_set_verify(FLAC__StreamEncoder *encoder, FLAC__bool value)
994 FLAC__ASSERT(0 != encoder);
995 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
997 encoder->protected_->verify = value;
1001 FLAC_API FLAC__bool FLAC__stream_encoder_set_streamable_subset(FLAC__StreamEncoder *encoder, FLAC__bool value)
1003 FLAC__ASSERT(0 != encoder);
1004 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1006 encoder->protected_->streamable_subset = value;
1010 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
1012 FLAC__ASSERT(0 != encoder);
1013 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1015 encoder->protected_->do_mid_side_stereo = value;
1019 FLAC_API FLAC__bool FLAC__stream_encoder_set_loose_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
1021 FLAC__ASSERT(0 != encoder);
1022 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1024 encoder->protected_->loose_mid_side_stereo = value;
1028 FLAC_API FLAC__bool FLAC__stream_encoder_set_channels(FLAC__StreamEncoder *encoder, unsigned value)
1030 FLAC__ASSERT(0 != encoder);
1031 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1033 encoder->protected_->channels = value;
1037 FLAC_API FLAC__bool FLAC__stream_encoder_set_bits_per_sample(FLAC__StreamEncoder *encoder, unsigned value)
1039 FLAC__ASSERT(0 != encoder);
1040 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1042 encoder->protected_->bits_per_sample = value;
1046 FLAC_API FLAC__bool FLAC__stream_encoder_set_sample_rate(FLAC__StreamEncoder *encoder, unsigned value)
1048 FLAC__ASSERT(0 != encoder);
1049 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1051 encoder->protected_->sample_rate = value;
1055 FLAC_API FLAC__bool FLAC__stream_encoder_set_blocksize(FLAC__StreamEncoder *encoder, unsigned value)
1057 FLAC__ASSERT(0 != encoder);
1058 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1060 encoder->protected_->blocksize = value;
1064 FLAC_API FLAC__bool FLAC__stream_encoder_set_max_lpc_order(FLAC__StreamEncoder *encoder, unsigned value)
1066 FLAC__ASSERT(0 != encoder);
1067 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1069 encoder->protected_->max_lpc_order = value;
1073 FLAC_API FLAC__bool FLAC__stream_encoder_set_qlp_coeff_precision(FLAC__StreamEncoder *encoder, unsigned value)
1075 FLAC__ASSERT(0 != encoder);
1076 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1078 encoder->protected_->qlp_coeff_precision = value;
1082 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_qlp_coeff_prec_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
1084 FLAC__ASSERT(0 != encoder);
1085 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1087 encoder->protected_->do_qlp_coeff_prec_search = value;
1091 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_escape_coding(FLAC__StreamEncoder *encoder, FLAC__bool value)
1093 FLAC__ASSERT(0 != encoder);
1094 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1097 /*@@@ deprecated: */
1098 encoder->protected_->do_escape_coding = value;
1105 FLAC_API FLAC__bool FLAC__stream_encoder_set_do_exhaustive_model_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
1107 FLAC__ASSERT(0 != encoder);
1108 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1110 encoder->protected_->do_exhaustive_model_search = value;
1114 FLAC_API FLAC__bool FLAC__stream_encoder_set_min_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
1116 FLAC__ASSERT(0 != encoder);
1117 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1119 encoder->protected_->min_residual_partition_order = value;
1123 FLAC_API FLAC__bool FLAC__stream_encoder_set_max_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
1125 FLAC__ASSERT(0 != encoder);
1126 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1128 encoder->protected_->max_residual_partition_order = value;
1132 FLAC_API FLAC__bool FLAC__stream_encoder_set_rice_parameter_search_dist(FLAC__StreamEncoder *encoder, unsigned value)
1134 FLAC__ASSERT(0 != encoder);
1135 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1138 /*@@@ deprecated: */
1139 encoder->protected_->rice_parameter_search_dist = value;
1146 FLAC_API FLAC__bool FLAC__stream_encoder_set_total_samples_estimate(FLAC__StreamEncoder *encoder, FLAC__uint64 value)
1148 FLAC__ASSERT(0 != encoder);
1149 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1151 encoder->protected_->total_samples_estimate = value;
1155 FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata(FLAC__StreamEncoder *encoder, FLAC__StreamMetadata **metadata, unsigned num_blocks)
1157 FLAC__ASSERT(0 != encoder);
1158 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1160 encoder->protected_->metadata = metadata;
1161 encoder->protected_->num_metadata_blocks = num_blocks;
1165 FLAC_API FLAC__bool FLAC__stream_encoder_set_write_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderWriteCallback value)
1167 FLAC__ASSERT(0 != encoder);
1168 FLAC__ASSERT(0 != value);
1169 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1171 encoder->private_->write_callback = value;
1175 FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderMetadataCallback value)
1177 FLAC__ASSERT(0 != encoder);
1178 FLAC__ASSERT(0 != value);
1179 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1181 encoder->private_->metadata_callback = value;
1185 FLAC_API FLAC__bool FLAC__stream_encoder_set_client_data(FLAC__StreamEncoder *encoder, void *value)
1187 FLAC__ASSERT(0 != encoder);
1188 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1190 encoder->private_->client_data = value;
1195 * These three functions are not static, but not publically exposed in
1196 * include/FLAC/ either. They are used by the test suite.
1198 FLAC_API FLAC__bool FLAC__stream_encoder_disable_constant_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1200 FLAC__ASSERT(0 != encoder);
1201 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1203 encoder->private_->disable_constant_subframes = value;
1207 FLAC_API FLAC__bool FLAC__stream_encoder_disable_fixed_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1209 FLAC__ASSERT(0 != encoder);
1210 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1212 encoder->private_->disable_fixed_subframes = value;
1216 FLAC_API FLAC__bool FLAC__stream_encoder_disable_verbatim_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
1218 FLAC__ASSERT(0 != encoder);
1219 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1221 encoder->private_->disable_verbatim_subframes = value;
1225 FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_get_state(const FLAC__StreamEncoder *encoder)
1227 FLAC__ASSERT(0 != encoder);
1228 return encoder->protected_->state;
1231 FLAC_API FLAC__StreamDecoderState FLAC__stream_encoder_get_verify_decoder_state(const FLAC__StreamEncoder *encoder)
1233 FLAC__ASSERT(0 != encoder);
1234 if(encoder->protected_->verify)
1235 return FLAC__stream_decoder_get_state(encoder->private_->verify.decoder);
1237 return FLAC__STREAM_DECODER_UNINITIALIZED;
1240 FLAC_API const char *FLAC__stream_encoder_get_resolved_state_string(const FLAC__StreamEncoder *encoder)
1242 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR)
1243 return FLAC__StreamEncoderStateString[encoder->protected_->state];
1245 return FLAC__stream_decoder_get_resolved_state_string(encoder->private_->verify.decoder);
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)
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;
1256 *channel = encoder->private_->verify.error_stats.channel;
1258 *sample = encoder->private_->verify.error_stats.sample;
1260 *expected = encoder->private_->verify.error_stats.expected;
1262 *got = encoder->private_->verify.error_stats.got;
1265 FLAC_API FLAC__bool FLAC__stream_encoder_get_verify(const FLAC__StreamEncoder *encoder)
1267 FLAC__ASSERT(0 != encoder);
1268 return encoder->protected_->verify;
1271 FLAC_API FLAC__bool FLAC__stream_encoder_get_streamable_subset(const FLAC__StreamEncoder *encoder)
1273 FLAC__ASSERT(0 != encoder);
1274 return encoder->protected_->streamable_subset;
1277 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_mid_side_stereo(const FLAC__StreamEncoder *encoder)
1279 FLAC__ASSERT(0 != encoder);
1280 return encoder->protected_->do_mid_side_stereo;
1283 FLAC_API FLAC__bool FLAC__stream_encoder_get_loose_mid_side_stereo(const FLAC__StreamEncoder *encoder)
1285 FLAC__ASSERT(0 != encoder);
1286 return encoder->protected_->loose_mid_side_stereo;
1289 FLAC_API unsigned FLAC__stream_encoder_get_channels(const FLAC__StreamEncoder *encoder)
1291 FLAC__ASSERT(0 != encoder);
1292 return encoder->protected_->channels;
1295 FLAC_API unsigned FLAC__stream_encoder_get_bits_per_sample(const FLAC__StreamEncoder *encoder)
1297 FLAC__ASSERT(0 != encoder);
1298 return encoder->protected_->bits_per_sample;
1301 FLAC_API unsigned FLAC__stream_encoder_get_sample_rate(const FLAC__StreamEncoder *encoder)
1303 FLAC__ASSERT(0 != encoder);
1304 return encoder->protected_->sample_rate;
1307 FLAC_API unsigned FLAC__stream_encoder_get_blocksize(const FLAC__StreamEncoder *encoder)
1309 FLAC__ASSERT(0 != encoder);
1310 return encoder->protected_->blocksize;
1313 FLAC_API unsigned FLAC__stream_encoder_get_max_lpc_order(const FLAC__StreamEncoder *encoder)
1315 FLAC__ASSERT(0 != encoder);
1316 return encoder->protected_->max_lpc_order;
1319 FLAC_API unsigned FLAC__stream_encoder_get_qlp_coeff_precision(const FLAC__StreamEncoder *encoder)
1321 FLAC__ASSERT(0 != encoder);
1322 return encoder->protected_->qlp_coeff_precision;
1325 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_qlp_coeff_prec_search(const FLAC__StreamEncoder *encoder)
1327 FLAC__ASSERT(0 != encoder);
1328 return encoder->protected_->do_qlp_coeff_prec_search;
1331 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_escape_coding(const FLAC__StreamEncoder *encoder)
1333 FLAC__ASSERT(0 != encoder);
1334 return encoder->protected_->do_escape_coding;
1337 FLAC_API FLAC__bool FLAC__stream_encoder_get_do_exhaustive_model_search(const FLAC__StreamEncoder *encoder)
1339 FLAC__ASSERT(0 != encoder);
1340 return encoder->protected_->do_exhaustive_model_search;
1343 FLAC_API unsigned FLAC__stream_encoder_get_min_residual_partition_order(const FLAC__StreamEncoder *encoder)
1345 FLAC__ASSERT(0 != encoder);
1346 return encoder->protected_->min_residual_partition_order;
1349 FLAC_API unsigned FLAC__stream_encoder_get_max_residual_partition_order(const FLAC__StreamEncoder *encoder)
1351 FLAC__ASSERT(0 != encoder);
1352 return encoder->protected_->max_residual_partition_order;
1355 FLAC_API unsigned FLAC__stream_encoder_get_rice_parameter_search_dist(const FLAC__StreamEncoder *encoder)
1357 FLAC__ASSERT(0 != encoder);
1358 return encoder->protected_->rice_parameter_search_dist;
1361 FLAC_API FLAC__uint64 FLAC__stream_encoder_get_total_samples_estimate(const FLAC__StreamEncoder *encoder)
1363 FLAC__ASSERT(0 != encoder);
1364 return encoder->protected_->total_samples_estimate;
1367 FLAC_API FLAC__bool FLAC__stream_encoder_process(FLAC__StreamEncoder *encoder, const FLAC__int32 * const buffer[], unsigned samples)
1369 unsigned i, j, channel;
1370 FLAC__int32 x, mid, side;
1371 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
1373 FLAC__ASSERT(0 != encoder);
1374 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1378 * we have several flavors of the same basic loop, optimized for
1379 * different conditions:
1381 if(encoder->protected_->max_lpc_order > 0) {
1382 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1384 * stereo coding: unroll channel loop
1385 * with LPC: calculate floating point version of signal
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));
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;
1398 encoder->private_->integer_signal[1][i] = x;
1399 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1400 encoder->private_->real_signal[1][i] = (FLAC__real)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;
1411 encoder->private_->current_sample_number++;
1413 if(i == blocksize) {
1414 if(!process_frame_(encoder, false)) /* false => not last frame */
1417 } while(j < samples);
1421 * independent channel coding: buffer each channel in inner loop
1422 * with LPC: calculate floating point version of signal
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));
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;
1436 encoder->private_->current_sample_number++;
1438 if(i == blocksize) {
1439 if(!process_frame_(encoder, false)) /* false => not last frame */
1442 } while(j < samples);
1446 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1448 * stereo coding: unroll channel loop
1449 * without LPC: no need to calculate floating point version of signal
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));
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];
1458 encoder->private_->integer_signal[1][i] = 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++;
1466 if(i == blocksize) {
1467 if(!process_frame_(encoder, false)) /* false => not last frame */
1470 } while(j < samples);
1474 * independent channel coding: buffer each channel in inner loop
1475 * without LPC: no need to calculate floating point version of signal
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));
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++;
1486 if(i == blocksize) {
1487 if(!process_frame_(encoder, false)) /* false => not last frame */
1490 } while(j < samples);
1497 FLAC_API FLAC__bool FLAC__stream_encoder_process_interleaved(FLAC__StreamEncoder *encoder, const FLAC__int32 buffer[], unsigned samples)
1499 unsigned i, j, k, channel;
1500 FLAC__int32 x, mid, side;
1501 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
1503 FLAC__ASSERT(0 != encoder);
1504 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1508 * we have several flavors of the same basic loop, optimized for
1509 * different conditions:
1511 if(encoder->protected_->max_lpc_order > 0) {
1512 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1514 * stereo coding: unroll channel loop
1515 * with LPC: calculate floating point version of signal
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));
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;
1528 encoder->private_->integer_signal[1][i] = x;
1529 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1530 encoder->private_->real_signal[1][i] = (FLAC__real)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;
1541 encoder->private_->current_sample_number++;
1543 if(i == blocksize) {
1544 if(!process_frame_(encoder, false)) /* false => not last frame */
1547 } while(j < samples);
1551 * independent channel coding: buffer each channel in inner loop
1552 * with LPC: calculate floating point version of signal
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));
1558 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1559 for(channel = 0; channel < channels; channel++) {
1561 encoder->private_->integer_signal[channel][i] = x;
1562 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1563 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
1566 encoder->private_->current_sample_number++;
1568 if(i == blocksize) {
1569 if(!process_frame_(encoder, false)) /* false => not last frame */
1572 } while(j < samples);
1576 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1578 * stereo coding: unroll channel loop
1579 * without LPC: no need to calculate floating point version of signal
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));
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++];
1588 encoder->private_->integer_signal[1][i] = 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++;
1596 if(i == blocksize) {
1597 if(!process_frame_(encoder, false)) /* false => not last frame */
1600 } while(j < samples);
1604 * independent channel coding: buffer each channel in inner loop
1605 * without LPC: no need to calculate floating point version of signal
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));
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++;
1616 if(i == blocksize) {
1617 if(!process_frame_(encoder, false)) /* false => not last frame */
1620 } while(j < samples);
1627 /***********************************************************************
1629 * Private class methods
1631 ***********************************************************************/
1633 void set_defaults_(FLAC__StreamEncoder *encoder)
1635 FLAC__ASSERT(0 != encoder);
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;
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;
1665 void free_(FLAC__StreamEncoder *encoder)
1667 unsigned i, channel;
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;
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;
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;
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;
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;
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;
1710 if(0 != encoder->private_->abs_residual_unaligned) {
1711 free(encoder->private_->abs_residual_unaligned);
1712 encoder->private_->abs_residual_unaligned = 0;
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;
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;
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;
1730 FLAC__bitbuffer_free(encoder->private_->frame);
1733 FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size)
1736 unsigned i, channel;
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);
1742 /* To avoid excessive malloc'ing, we only grow the buffer; no shrinking. */
1743 if(new_size <= encoder->private_->input_capacity)
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.
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]);
1760 memset(encoder->private_->integer_signal[i], 0, sizeof(FLAC__int32)*4);
1761 encoder->private_->integer_signal[i] += 4;
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]);
1769 memset(encoder->private_->integer_signal_mid_side[i], 0, sizeof(FLAC__int32)*4);
1770 encoder->private_->integer_signal_mid_side[i] += 4;
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]);
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]);
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);
1789 encoder->private_->input_capacity = new_size;
1791 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1796 FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples)
1798 const FLAC__byte *buffer;
1801 FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1803 FLAC__bitbuffer_get_buffer(encoder->private_->frame, &buffer, &bytes);
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;
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;
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;
1827 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
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);
1837 FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
1839 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1842 * Accumulate raw signal to the MD5 signature
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;
1850 * Process the frame header and subframes into the frame bitbuffer
1852 if(!process_subframes_(encoder, is_last_frame)) {
1853 /* the above function sets the state for us in case of an error */
1858 * Zero-pad the frame to a byte_boundary
1860 if(!FLAC__bitbuffer_zero_pad_to_byte_boundary(encoder->private_->frame)) {
1861 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
1866 * CRC-16 the whole thing
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);
1874 if(!write_bitbuffer_(encoder, encoder->protected_->blocksize)) {
1875 /* the above function sets the state for us in case of an error */
1880 * Get ready for the next frame
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;
1889 FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
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;
1896 * Calculate the min,max Rice partition orders
1899 max_partition_order = 0;
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);
1905 min_partition_order = min(min_partition_order, max_partition_order);
1907 precompute_partition_sums = encoder->private_->precompute_partition_sums && ((max_partition_order > min_partition_order) || encoder->protected_->do_escape_coding);
1912 if(!FLAC__bitbuffer_clear(encoder->private_->frame)) {
1913 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
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;
1925 * Figure out what channel assignments to try
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;
1934 do_independent = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT);
1935 do_mid_side = !do_independent;
1939 do_independent = true;
1944 do_independent = true;
1945 do_mid_side = false;
1948 FLAC__ASSERT(do_independent || do_mid_side);
1951 * Check for wasted bits; set effective bps for each subframe
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;
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);
1970 * First do a normal encoding pass of each independent channel
1972 if(do_independent) {
1973 for(channel = 0; channel < encoder->protected_->channels; channel++) {
1977 min_partition_order,
1978 max_partition_order,
1979 precompute_partition_sums,
1981 encoder->private_->subframe_bps[channel],
1982 encoder->private_->integer_signal[channel],
1983 #ifndef FLAC__INTEGER_ONLY_LIBRARY
1984 encoder->private_->real_signal[channel],
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
1998 * Now do mid and side channels if requested
2001 FLAC__ASSERT(encoder->protected_->channels == 2);
2003 for(channel = 0; channel < 2; channel++) {
2007 min_partition_order,
2008 max_partition_order,
2009 precompute_partition_sums,
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],
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
2028 * Compose the frame bitbuffer
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;
2035 FLAC__ASSERT(encoder->protected_->channels == 2);
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);
2041 unsigned bits[4]; /* WATCHOUT - indexed by FLAC__ChannelAssignment */
2043 FLAC__ChannelAssignment ca;
2045 FLAC__ASSERT(do_independent && do_mid_side);
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];
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;
2061 frame_header.channel_assignment = channel_assignment;
2063 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
2064 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
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]];
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]];
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]];
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]];
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];
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];
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];
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];
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))
2113 if(!add_subframe_(encoder, &frame_header, right_bps, right_subframe, encoder->private_->frame))
2117 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
2118 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
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 */
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;
2136 encoder->private_->last_channel_assignment = frame_header.channel_assignment;
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[],
2152 FLAC__Subframe *subframe[2],
2153 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
2154 FLAC__int32 *residual[2],
2155 unsigned *best_subframe,
2159 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2160 FLAC__float fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
2162 FLAC__fixedpoint fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
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;
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;
2176 /* verbatim subframe is the baseline against which we measure other compressed subframes */
2178 if(encoder->private_->disable_verbatim_subframes && frame_header->blocksize >= FLAC__MAX_FIXED_ORDER)
2179 _best_bits = UINT_MAX;
2181 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
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 */
2188 !encoder->private_->disable_constant_subframes &&
2189 #ifndef FLAC__INTEGER_ONLY_LIBRARY
2190 fixed_residual_bits_per_sample[1] == 0.0
2192 fixed_residual_bits_per_sample[1] == FLAC__FP_ZERO
2195 /* the above means it's possible all samples are the same value; now double-check it: */
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;
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;
2213 if(!encoder->private_->disable_fixed_subframes || (encoder->protected_->max_lpc_order == 0 && _best_bits == UINT_MAX)) {
2215 if(encoder->protected_->do_exhaustive_model_search) {
2216 min_fixed_order = 0;
2217 max_fixed_order = FLAC__MAX_FIXED_ORDER;
2220 min_fixed_order = max_fixed_order = guess_fixed_order;
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 */
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 */
2232 #ifndef FLAC__SYMMETRIC_RICE
2233 rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
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);
2239 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2242 evaluate_fixed_subframe_(
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,
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]
2261 if(_candidate_bits < _best_bits) {
2262 _best_subframe = !_best_subframe;
2263 _best_bits = _candidate_bits;
2268 #ifndef FLAC__INTEGER_ONLY_LIBRARY
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;
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) {
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;
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 */
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);
2299 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
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);
2307 max_qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
2310 min_qlp_coeff_precision = max_qlp_coeff_precision = encoder->protected_->qlp_coeff_precision;
2312 for(qlp_coeff_precision = min_qlp_coeff_precision; qlp_coeff_precision <= max_qlp_coeff_precision; qlp_coeff_precision++) {
2314 evaluate_lpc_subframe_(
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,
2325 qlp_coeff_precision,
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]
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;
2346 #endif /* !defined FLAC__INTEGER_ONLY_LIBRARY */
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]);
2356 *best_subframe = _best_subframe;
2357 *best_bits = _best_bits;
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
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;
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;
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;
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;
2402 unsigned evaluate_constant_subframe_(
2403 const FLAC__int32 signal,
2404 unsigned subframe_bps,
2405 FLAC__Subframe *subframe
2408 subframe->type = FLAC__SUBFRAME_TYPE_CONSTANT;
2409 subframe->data.constant.value = signal;
2411 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + subframe_bps;
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[],
2422 unsigned subframe_bps,
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
2434 unsigned i, residual_bits;
2435 const unsigned residual_samples = blocksize - order;
2437 FLAC__fixed_compute_residual(signal+order, residual_samples, order, residual);
2439 subframe->type = FLAC__SUBFRAME_TYPE_FIXED;
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;
2446 find_best_partition_order_(
2450 abs_residual_partition_sums,
2451 raw_bits_per_partition,
2455 min_partition_order,
2456 max_partition_order,
2457 precompute_partition_sums,
2459 rice_parameter_search_dist,
2460 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2463 subframe->data.fixed.order = order;
2464 for(i = 0; i < order; i++)
2465 subframe->data.fixed.warmup[i] = signal[i];
2467 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (order * subframe_bps) + residual_bits;
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[],
2480 unsigned subframe_bps,
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
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;
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));
2505 ret = FLAC__lpc_quantize_coefficients(lp_coeff, order, qlp_coeff_precision, qlp_coeff, &quantization);
2507 return 0; /* this is a hack to indicate to the caller that we can't do lp at this order on this subframe */
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);
2513 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2515 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_64bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2517 subframe->type = FLAC__SUBFRAME_TYPE_LPC;
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;
2524 find_best_partition_order_(
2528 abs_residual_partition_sums,
2529 raw_bits_per_partition,
2533 min_partition_order,
2534 max_partition_order,
2535 precompute_partition_sums,
2537 rice_parameter_search_dist,
2538 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
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];
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;
2552 unsigned evaluate_verbatim_subframe_(
2553 const FLAC__int32 signal[],
2555 unsigned subframe_bps,
2556 FLAC__Subframe *subframe
2559 subframe->type = FLAC__SUBFRAME_TYPE_VERBATIM;
2561 subframe->data.verbatim.data = signal;
2563 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (blocksize * subframe_bps);
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
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;
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);
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);
2598 if(precompute_partition_sums) {
2599 int partition_order;
2602 precompute_partition_info_sums_(abs_residual, abs_residual_partition_sums, residual_samples, predictor_order, min_partition_order, max_partition_order);
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);
2607 for(partition_order = (int)max_partition_order, sum = 0; partition_order >= (int)min_partition_order; partition_order--) {
2608 #ifdef DONT_ESTIMATE_RICE_BITS
2610 set_partitioned_rice_with_precompute_(
2612 abs_residual_partition_sums+sum,
2613 raw_bits_per_partition+sum,
2617 rice_parameter_search_dist,
2618 (unsigned)partition_order,
2620 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2626 set_partitioned_rice_with_precompute_(
2628 abs_residual_partition_sums+sum,
2629 raw_bits_per_partition+sum,
2633 rice_parameter_search_dist,
2634 (unsigned)partition_order,
2636 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2642 FLAC__ASSERT(best_residual_bits != 0);
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;
2654 unsigned partition_order;
2655 for(partition_order = min_partition_order; partition_order <= max_partition_order; partition_order++) {
2656 #ifdef DONT_ESTIMATE_RICE_BITS
2658 set_partitioned_rice_(
2664 rice_parameter_search_dist,
2666 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2672 set_partitioned_rice_(
2677 rice_parameter_search_dist,
2679 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2685 FLAC__ASSERT(best_residual_bits != 0);
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;
2697 * We are allowed to de-const the pointer based on our special knowledge;
2698 * it is const to the outside world.
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)));
2707 return best_residual_bits;
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
2719 int partition_order;
2720 unsigned from_partition, to_partition = 0;
2721 const unsigned blocksize = residual_samples + predictor_order;
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;
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;
2731 FLAC__ASSERT(default_partition_samples > predictor_order);
2733 for(partition = residual_sample = 0; partition < partitions; partition++) {
2734 partition_samples = default_partition_samples;
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;
2743 abs_residual_partition_sums[partition] = abs_residual_partition_sum;
2745 to_partition = partitions;
2749 /* now merge partitions for lower orders */
2750 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2753 const unsigned partitions = 1u << partition_order;
2754 for(i = 0; i < partitions; i++) {
2755 s = abs_residual_partition_sums[from_partition];
2757 abs_residual_partition_sums[to_partition] = s + abs_residual_partition_sums[from_partition];
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
2773 int partition_order;
2774 unsigned from_partition, to_partition = 0;
2775 const unsigned blocksize = residual_samples + predictor_order;
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;
2785 FLAC__ASSERT(default_partition_samples > predictor_order);
2787 for(partition = residual_sample = 0; partition < partitions; partition++) {
2788 partition_samples = default_partition_samples;
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;
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);
2804 to_partition = partitions;
2808 /* now merge partitions for lower orders */
2809 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2812 const unsigned partitions = 1u << partition_order;
2813 for(i = 0; i < partitions; i++) {
2814 m = raw_bits_per_partition[from_partition];
2816 raw_bits_per_partition[to_partition] = max(m, raw_bits_per_partition[from_partition]);
2823 #ifdef VARIABLE_RICE_BITS
2824 #undef VARIABLE_RICE_BITS
2826 #ifndef DONT_ESTIMATE_RICE_BITS
2827 #define VARIABLE_RICE_BITS(value, parameter) ((value) >> (parameter))
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,
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,
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;
2860 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
2861 unsigned *parameters;
2863 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
2865 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
2866 parameters = partitioned_rice_contents->parameters;
2868 if(partition_order == 0) {
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;
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);
2882 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2886 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
2888 best_partition_bits = 0xffffffff;
2889 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2891 #ifdef VARIABLE_RICE_BITS
2892 #ifdef FLAC__SYMMETRIC_RICE
2893 partition_bits = (2+rice_parameter) * residual_samples;
2895 const unsigned rice_parameter_estimate = rice_parameter-1;
2896 partition_bits = (1+rice_parameter) * residual_samples;
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);
2907 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
2910 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
2913 #ifndef NO_RICE_SEARCH
2914 if(partition_bits < best_partition_bits) {
2915 best_rice_parameter = rice_parameter;
2916 best_partition_bits = partition_bits;
2920 parameters[0] = best_rice_parameter;
2921 bits_ += best_partition_bits;
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)
2934 partition_samples -= predictor_order;
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;
2945 /* calc rice_parameter = floor(log2(mean)) */
2953 /* calc rice_parameter ala LOCO-I */
2954 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
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);
2961 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2964 #ifndef NO_RICE_SEARCH
2965 if(rice_parameter_search_dist) {
2966 if(rice_parameter < rice_parameter_search_dist)
2967 min_rice_parameter = 0;
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);
2975 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2979 min_rice_parameter = max_rice_parameter = rice_parameter;
2981 best_partition_bits = 0xffffffff;
2982 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2984 #ifdef VARIABLE_RICE_BITS
2985 #ifdef FLAC__SYMMETRIC_RICE
2986 partition_bits = (2+rice_parameter) * partition_samples;
2988 const unsigned rice_parameter_estimate = rice_parameter-1;
2989 partition_bits = (1+rice_parameter) * partition_samples;
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);
3001 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
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[] */
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;
3016 parameters[partition] = best_rice_parameter;
3017 bits_ += best_partition_bits;
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,
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,
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;
3061 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
3062 unsigned *parameters, *raw_bits;
3064 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
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;
3070 if(partition_order == 0) {
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;
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);
3084 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3088 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
3090 best_partition_bits = 0xffffffff;
3091 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3093 #ifdef VARIABLE_RICE_BITS
3094 #ifdef FLAC__SYMMETRIC_RICE
3095 partition_bits = (2+rice_parameter) * residual_samples;
3097 const unsigned rice_parameter_estimate = rice_parameter-1;
3098 partition_bits = (1+rice_parameter) * residual_samples;
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);
3109 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
3112 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
3115 #ifndef NO_RICE_SEARCH
3116 if(partition_bits < best_partition_bits) {
3117 best_rice_parameter = rice_parameter;
3118 best_partition_bits = partition_bits;
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;
3130 parameters[0] = best_rice_parameter;
3131 bits_ += best_partition_bits;
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)
3144 partition_samples -= predictor_order;
3146 mean = abs_residual_partition_sums[partition];
3147 #ifdef FLAC__SYMMETRIC_RICE
3148 mean += partition_samples >> 1; /* for rounding effect */
3149 mean /= partition_samples;
3151 /* calc rice_parameter = floor(log2(mean)) */
3159 /* calc rice_parameter ala LOCO-I */
3160 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
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);
3167 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3170 #ifndef NO_RICE_SEARCH
3171 if(rice_parameter_search_dist) {
3172 if(rice_parameter < rice_parameter_search_dist)
3173 min_rice_parameter = 0;
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);
3181 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3185 min_rice_parameter = max_rice_parameter = rice_parameter;
3187 best_partition_bits = 0xffffffff;
3188 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3190 #ifdef VARIABLE_RICE_BITS
3191 #ifdef FLAC__SYMMETRIC_RICE
3192 partition_bits = (2+rice_parameter) * partition_samples;
3194 const unsigned rice_parameter_estimate = rice_parameter-1;
3195 partition_bits = (1+rice_parameter) * partition_samples;
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);
3207 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
3210 partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
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;
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;
3230 parameters[partition] = best_rice_parameter;
3231 bits_ += best_partition_bits;
3239 unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples)
3244 for(i = 0; i < samples && !(x&1); i++)
3251 for(shift = 0; !(x&1); shift++)
3256 for(i = 0; i < samples; i++)
3257 signal[i] >>= shift;
3263 void append_to_verify_fifo_(verify_input_fifo *fifo, const FLAC__int32 * const input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3267 for(channel = 0; channel < channels; channel++)
3268 memcpy(&fifo->data[channel][fifo->tail], &input[channel][input_offset], sizeof(FLAC__int32) * wide_samples);
3270 fifo->tail += wide_samples;
3272 FLAC__ASSERT(fifo->tail <= fifo->size);
3275 void append_to_verify_fifo_interleaved_(verify_input_fifo *fifo, const FLAC__int32 input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3278 unsigned sample, wide_sample;
3279 unsigned tail = fifo->tail;
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++];
3289 FLAC__ASSERT(fifo->tail <= fifo->size);
3292 FLAC__StreamDecoderReadStatus verify_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], unsigned *bytes, void *client_data)
3294 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3295 const unsigned encoded_bytes = encoder->private_->verify.output.bytes;
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;
3305 if(encoded_bytes == 0) {
3307 * If we get here, a FIFO underflow has occurred,
3308 * which means there is a bug somewhere.
3311 return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
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;
3320 return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
3323 FLAC__StreamDecoderWriteStatus verify_write_callback_(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
3325 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder *)client_data;
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;
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;
3336 for(i = 0; i < blocksize; i++) {
3337 if(buffer[channel][i] != encoder->private_->verify.input_fifo.data[channel][i]) {
3339 expect = (FLAC__int32)encoder->private_->verify.input_fifo.data[channel][i];
3340 got = (FLAC__int32)buffer[channel][i];
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;
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);
3360 encoder->private_->verify.input_fifo.tail -= blocksize;
3361 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
3364 void verify_metadata_callback_(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
3366 (void)decoder, (void)metadata, (void)client_data;
3369 void verify_error_callback_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
3371 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3372 (void)decoder, (void)status;
3373 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;