blob: f9be720f9beca8094b2501005a1bdf1d63e62f56 [file] [log] [blame]
Josh Coalson26560dd2001-02-08 00:38:41 +00001/* libFLAC - Free Lossless Audio Codec library
Josh Coalson95643902004-01-17 04:14:43 +00002 * Copyright (C) 2000,2001,2002,2003,2004 Josh Coalson
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00003 *
Josh Coalsonafd81072003-01-31 23:34:56 +00004 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00007 *
Josh Coalsonafd81072003-01-31 23:34:56 +00008 * - Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000010 *
Josh Coalsonafd81072003-01-31 23:34:56 +000011 * - Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * - Neither the name of the Xiph.org Foundation nor the names of its
16 * contributors may be used to endorse or promote products derived from
17 * this software without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
23 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000030 */
31
Josh Coalsone6b3bbe2002-10-08 06:03:25 +000032#include <limits.h>
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000033#include <stdio.h>
34#include <stdlib.h> /* for malloc() */
35#include <string.h> /* for memcpy() */
Josh Coalson1b689822001-05-31 20:11:02 +000036#include "FLAC/assert.h"
Josh Coalsond86e03b2002-08-03 21:56:15 +000037#include "FLAC/stream_decoder.h"
Josh Coalson0a15c142001-06-13 17:59:57 +000038#include "protected/stream_encoder.h"
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000039#include "private/bitbuffer.h"
Josh Coalsoneef56702001-03-30 00:45:22 +000040#include "private/bitmath.h"
Josh Coalson215af572001-03-27 01:15:58 +000041#include "private/crc.h"
Josh Coalsoncf30f502001-05-23 20:57:44 +000042#include "private/cpu.h"
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000043#include "private/fixed.h"
Josh Coalsonb7023aa2002-08-17 15:23:43 +000044#include "private/format.h"
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000045#include "private/lpc.h"
Josh Coalsonfa37f1c2001-01-12 23:55:11 +000046#include "private/md5.h"
Josh Coalsond98c43d2001-05-13 05:17:01 +000047#include "private/memory.h"
Josh Coalsonb7023aa2002-08-17 15:23:43 +000048#include "private/stream_encoder_framing.h"
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000049
Josh Coalson5e31be12002-12-04 07:07:35 +000050#ifdef HAVE_CONFIG_H
51#include <config.h>
52#endif
53
Josh Coalsonbb7f6b92000-12-10 04:09:52 +000054#ifdef min
55#undef min
56#endif
57#define min(x,y) ((x)<(y)?(x):(y))
58
59#ifdef max
60#undef max
61#endif
62#define max(x,y) ((x)>(y)?(x):(y))
63
Josh Coalsond86e03b2002-08-03 21:56:15 +000064typedef struct {
65 FLAC__int32 *data[FLAC__MAX_CHANNELS];
66 unsigned size; /* of each data[] in samples */
67 unsigned tail;
68} verify_input_fifo;
69
70typedef struct {
71 const FLAC__byte *data;
72 unsigned capacity;
73 unsigned bytes;
74} verify_output;
75
76typedef enum {
77 ENCODER_IN_MAGIC = 0,
78 ENCODER_IN_METADATA = 1,
79 ENCODER_IN_AUDIO = 2
80} EncoderStateHint;
81
Josh Coalson0a15c142001-06-13 17:59:57 +000082/***********************************************************************
83 *
84 * Private class method prototypes
85 *
86 ***********************************************************************/
87
Josh Coalsonf1eff452002-07-31 07:05:33 +000088static void set_defaults_(FLAC__StreamEncoder *encoder);
89static void free_(FLAC__StreamEncoder *encoder);
90static FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size);
Josh Coalsond86e03b2002-08-03 21:56:15 +000091static FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples);
Josh Coalsonf1eff452002-07-31 07:05:33 +000092static FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame);
93static FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame);
Josh Coalson6fe72f72002-08-20 04:01:59 +000094
95static FLAC__bool process_subframe_(
96 FLAC__StreamEncoder *encoder,
97 unsigned min_partition_order,
98 unsigned max_partition_order,
99 FLAC__bool precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +0000100 const FLAC__FrameHeader *frame_header,
101 unsigned subframe_bps,
102 const FLAC__int32 integer_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000103#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +0000104 const FLAC__real real_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000105#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +0000106 FLAC__Subframe *subframe[2],
107 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
108 FLAC__int32 *residual[2],
109 unsigned *best_subframe,
110 unsigned *best_bits
111);
112
113static FLAC__bool add_subframe_(
114 FLAC__StreamEncoder *encoder,
115 const FLAC__FrameHeader *frame_header,
116 unsigned subframe_bps,
117 const FLAC__Subframe *subframe,
118 FLAC__BitBuffer *frame
119);
120
121static unsigned evaluate_constant_subframe_(
122 const FLAC__int32 signal,
123 unsigned subframe_bps,
124 FLAC__Subframe *subframe
125);
126
127static unsigned evaluate_fixed_subframe_(
128 FLAC__StreamEncoder *encoder,
129 const FLAC__int32 signal[],
130 FLAC__int32 residual[],
131 FLAC__uint32 abs_residual[],
132 FLAC__uint64 abs_residual_partition_sums[],
133 unsigned raw_bits_per_partition[],
134 unsigned blocksize,
135 unsigned subframe_bps,
136 unsigned order,
137 unsigned rice_parameter,
138 unsigned min_partition_order,
139 unsigned max_partition_order,
140 FLAC__bool precompute_partition_sums,
141 FLAC__bool do_escape_coding,
142 unsigned rice_parameter_search_dist,
143 FLAC__Subframe *subframe,
144 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
145);
146
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000147#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +0000148static unsigned evaluate_lpc_subframe_(
149 FLAC__StreamEncoder *encoder,
150 const FLAC__int32 signal[],
151 FLAC__int32 residual[],
152 FLAC__uint32 abs_residual[],
153 FLAC__uint64 abs_residual_partition_sums[],
154 unsigned raw_bits_per_partition[],
155 const FLAC__real lp_coeff[],
156 unsigned blocksize,
157 unsigned subframe_bps,
158 unsigned order,
159 unsigned qlp_coeff_precision,
160 unsigned rice_parameter,
161 unsigned min_partition_order,
162 unsigned max_partition_order,
163 FLAC__bool precompute_partition_sums,
164 FLAC__bool do_escape_coding,
165 unsigned rice_parameter_search_dist,
166 FLAC__Subframe *subframe,
167 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
168);
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000169#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +0000170
171static unsigned evaluate_verbatim_subframe_(
172 const FLAC__int32 signal[],
173 unsigned blocksize,
174 unsigned subframe_bps,
175 FLAC__Subframe *subframe
176);
177
178static unsigned find_best_partition_order_(
179 struct FLAC__StreamEncoderPrivate *private_,
180 const FLAC__int32 residual[],
181 FLAC__uint32 abs_residual[],
182 FLAC__uint64 abs_residual_partition_sums[],
183 unsigned raw_bits_per_partition[],
184 unsigned residual_samples,
185 unsigned predictor_order,
186 unsigned rice_parameter,
187 unsigned min_partition_order,
188 unsigned max_partition_order,
189 FLAC__bool precompute_partition_sums,
190 FLAC__bool do_escape_coding,
191 unsigned rice_parameter_search_dist,
192 FLAC__EntropyCodingMethod_PartitionedRice *best_partitioned_rice
193);
194
195static void precompute_partition_info_sums_(
196 const FLAC__uint32 abs_residual[],
197 FLAC__uint64 abs_residual_partition_sums[],
198 unsigned residual_samples,
199 unsigned predictor_order,
200 unsigned min_partition_order,
201 unsigned max_partition_order
202);
203
204static void precompute_partition_info_escapes_(
205 const FLAC__int32 residual[],
206 unsigned raw_bits_per_partition[],
207 unsigned residual_samples,
208 unsigned predictor_order,
209 unsigned min_partition_order,
210 unsigned max_partition_order
211);
212
Josh Coalson8395d022001-07-12 21:25:22 +0000213#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +0000214static FLAC__bool set_partitioned_rice_(
215 const FLAC__uint32 abs_residual[],
216 const FLAC__int32 residual[],
217 const unsigned residual_samples,
218 const unsigned predictor_order,
219 const unsigned suggested_rice_parameter,
220 const unsigned rice_parameter_search_dist,
221 const unsigned partition_order,
222 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
223 unsigned *bits
224);
225
226static FLAC__bool set_partitioned_rice_with_precompute_(
227 const FLAC__int32 residual[],
228 const FLAC__uint64 abs_residual_partition_sums[],
229 const unsigned raw_bits_per_partition[],
230 const unsigned residual_samples,
231 const unsigned predictor_order,
232 const unsigned suggested_rice_parameter,
233 const unsigned rice_parameter_search_dist,
234 const unsigned partition_order,
235 const FLAC__bool search_for_escapes,
236 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
237 unsigned *bits
238);
Josh Coalson8395d022001-07-12 21:25:22 +0000239#else
Josh Coalson6fe72f72002-08-20 04:01:59 +0000240static FLAC__bool set_partitioned_rice_(
241 const FLAC__uint32 abs_residual[],
242 const unsigned residual_samples,
243 const unsigned predictor_order,
244 const unsigned suggested_rice_parameter,
245 const unsigned rice_parameter_search_dist,
246 const unsigned partition_order,
247 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
248 unsigned *bits
249);
250
251static FLAC__bool set_partitioned_rice_with_precompute_(
252 const FLAC__uint32 abs_residual[],
253 const FLAC__uint64 abs_residual_partition_sums[],
254 const unsigned raw_bits_per_partition[],
255 const unsigned residual_samples,
256 const unsigned predictor_order,
257 const unsigned suggested_rice_parameter,
258 const unsigned rice_parameter_search_dist,
259 const unsigned partition_order,
260 const FLAC__bool search_for_escapes,
261 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
262 unsigned *bits
263);
Josh Coalson0a15c142001-06-13 17:59:57 +0000264#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +0000265
Josh Coalsonf1eff452002-07-31 07:05:33 +0000266static unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples);
Josh Coalson6fe72f72002-08-20 04:01:59 +0000267
Josh Coalsond86e03b2002-08-03 21:56:15 +0000268/* verify-related routines: */
Josh Coalson6fe72f72002-08-20 04:01:59 +0000269static void append_to_verify_fifo_(
270 verify_input_fifo *fifo,
271 const FLAC__int32 * const input[],
272 unsigned input_offset,
273 unsigned channels,
274 unsigned wide_samples
275);
276
277static void append_to_verify_fifo_interleaved_(
278 verify_input_fifo *fifo,
279 const FLAC__int32 input[],
280 unsigned input_offset,
281 unsigned channels,
282 unsigned wide_samples
283);
284
285static FLAC__StreamDecoderReadStatus verify_read_callback_(
286 const FLAC__StreamDecoder *decoder,
287 FLAC__byte buffer[],
288 unsigned *bytes,
289 void *client_data
290);
291
292static FLAC__StreamDecoderWriteStatus verify_write_callback_(
293 const FLAC__StreamDecoder *decoder,
294 const FLAC__Frame *frame,
295 const FLAC__int32 * const buffer[],
296 void *client_data
297);
298
299static void verify_metadata_callback_(
300 const FLAC__StreamDecoder *decoder,
301 const FLAC__StreamMetadata *metadata,
302 void *client_data
303);
304
305static void verify_error_callback_(
306 const FLAC__StreamDecoder *decoder,
307 FLAC__StreamDecoderErrorStatus status,
308 void *client_data
309);
310
Josh Coalson0a15c142001-06-13 17:59:57 +0000311
312/***********************************************************************
313 *
314 * Private class data
315 *
316 ***********************************************************************/
317
318typedef struct FLAC__StreamEncoderPrivate {
Josh Coalson8395d022001-07-12 21:25:22 +0000319 unsigned input_capacity; /* current size (in samples) of the signal and residual buffers */
Josh Coalson77e3f312001-06-23 03:03:24 +0000320 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) */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000322#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson77e3f312001-06-23 03:03:24 +0000323 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) */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000325#endif
Josh Coalson8395d022001-07-12 21:25:22 +0000326 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) */
Josh Coalson77e3f312001-06-23 03:03:24 +0000328 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];
Josh Coalson94e02cd2001-01-25 10:41:06 +0000330 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];
Josh Coalsona37ba462002-08-19 21:36:39 +0000334 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];
Josh Coalson8395d022001-07-12 21:25:22 +0000338 unsigned best_subframe[FLAC__MAX_CHANNELS]; /* index into the above workspaces */
Josh Coalson94e02cd2001-01-25 10:41:06 +0000339 unsigned best_subframe_mid_side[2];
Josh Coalson8395d022001-07-12 21:25:22 +0000340 unsigned best_subframe_bits[FLAC__MAX_CHANNELS]; /* size in bits of the best subframe for each channel */
Josh Coalson94e02cd2001-01-25 10:41:06 +0000341 unsigned best_subframe_bits_mid_side[2];
Josh Coalson77e3f312001-06-23 03:03:24 +0000342 FLAC__uint32 *abs_residual; /* workspace where abs(candidate residual) is stored */
Josh Coalsonb3347bd2001-07-16 18:06:41 +0000343 FLAC__uint64 *abs_residual_partition_sums; /* workspace where the sum of abs(candidate residual) for each partition is stored */
Josh Coalson8395d022001-07-12 21:25:22 +0000344 unsigned *raw_bits_per_partition; /* workspace where the sum of silog2(candidate residual) for each partition is stored */
Josh Coalsonaec256b2002-03-12 16:19:54 +0000345 FLAC__BitBuffer *frame; /* the current frame being worked on */
Josh Coalson8395d022001-07-12 21:25:22 +0000346 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 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +0000348 FLAC__ChannelAssignment last_channel_assignment;
Josh Coalsoncc682512002-06-08 04:53:42 +0000349 FLAC__StreamMetadata metadata;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000350 unsigned current_sample_number;
351 unsigned current_frame_number;
Josh Coalson3e7a96e2004-07-23 05:18:22 +0000352 struct FLAC__MD5Context md5context;
Josh Coalsoncf30f502001-05-23 20:57:44 +0000353 FLAC__CPUInfo cpuinfo;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000354#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +0000355 unsigned (*local_fixed_compute_best_predictor)(const FLAC__int32 data[], unsigned data_len, FLAC__float residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000356#else
357 unsigned (*local_fixed_compute_best_predictor)(const FLAC__int32 data[], unsigned data_len, FLAC__fixedpoint residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1]);
358#endif
359#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson77e3f312001-06-23 03:03:24 +0000360 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[]);
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000362 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[]);
Josh Coalson77e3f312001-06-23 03:03:24 +0000363 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[]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000364#endif
Josh Coalson3262b0d2002-08-14 20:58:42 +0000365 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 */
Josh Coalsone6b3bbe2002-10-08 06:03:25 +0000369 FLAC__bool disable_constant_subframes;
370 FLAC__bool disable_fixed_subframes;
371 FLAC__bool disable_verbatim_subframes;
Josh Coalson681c2932002-08-01 08:19:37 +0000372 FLAC__StreamEncoderWriteCallback write_callback;
373 FLAC__StreamEncoderMetadataCallback metadata_callback;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000374 void *client_data;
Josh Coalsond98c43d2001-05-13 05:17:01 +0000375 /* unaligned (original) pointers to allocated data */
Josh Coalson77e3f312001-06-23 03:03:24 +0000376 FLAC__int32 *integer_signal_unaligned[FLAC__MAX_CHANNELS];
377 FLAC__int32 *integer_signal_mid_side_unaligned[2];
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000378#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson77e3f312001-06-23 03:03:24 +0000379 FLAC__real *real_signal_unaligned[FLAC__MAX_CHANNELS];
380 FLAC__real *real_signal_mid_side_unaligned[2];
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000381#endif
Josh Coalson77e3f312001-06-23 03:03:24 +0000382 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;
Josh Coalsonb3347bd2001-07-16 18:06:41 +0000385 FLAC__uint64 *abs_residual_partition_sums_unaligned;
Josh Coalsond98c43d2001-05-13 05:17:01 +0000386 unsigned *raw_bits_per_partition_unaligned;
Josh Coalson8084b052001-11-01 00:27:29 +0000387 /*
388 * These fields have been moved here from private function local
389 * declarations merely to save stack space during encoding.
390 */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000391#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonf1eff452002-07-31 07:05:33 +0000392 FLAC__real lp_coeff[FLAC__MAX_LPC_ORDER][FLAC__MAX_LPC_ORDER]; /* from process_subframe_() */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000393#endif
Josh Coalsona37ba462002-08-19 21:36:39 +0000394 FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents_extra[2]; /* from find_best_partition_order_() */
Josh Coalsond86e03b2002-08-03 21:56:15 +0000395 /*
396 * The data for the verify section
397 */
398 struct {
399 FLAC__StreamDecoder *decoder;
400 EncoderStateHint state_hint;
401 FLAC__bool needs_magic_hack;
402 verify_input_fifo input_fifo;
403 verify_output output;
404 struct {
405 FLAC__uint64 absolute_sample;
406 unsigned frame_number;
407 unsigned channel;
408 unsigned sample;
409 FLAC__int32 expected;
410 FLAC__int32 got;
411 } error_stats;
412 } verify;
Josh Coalson3262b0d2002-08-14 20:58:42 +0000413 FLAC__bool is_being_deleted; /* if true, call to ..._finish() from ..._delete() will not call the callbacks */
Josh Coalson0a15c142001-06-13 17:59:57 +0000414} FLAC__StreamEncoderPrivate;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000415
Josh Coalson0a15c142001-06-13 17:59:57 +0000416/***********************************************************************
417 *
418 * Public static class data
419 *
420 ***********************************************************************/
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000421
Josh Coalson6afed9f2002-10-16 22:29:47 +0000422FLAC_API const char * const FLAC__StreamEncoderStateString[] = {
Josh Coalson0a15c142001-06-13 17:59:57 +0000423 "FLAC__STREAM_ENCODER_OK",
Josh Coalsond86e03b2002-08-03 21:56:15 +0000424 "FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR",
425 "FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA",
Josh Coalson00e53872001-06-16 07:32:25 +0000426 "FLAC__STREAM_ENCODER_INVALID_CALLBACK",
Josh Coalson0a15c142001-06-13 17:59:57 +0000427 "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",
Josh Coalson20ac2c12002-08-30 05:47:14 +0000431 "FLAC__STREAM_ENCODER_INVALID_MAX_LPC_ORDER",
Josh Coalson0a15c142001-06-13 17:59:57 +0000432 "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",
Josh Coalson66075c12002-06-01 05:39:38 +0000439 "FLAC__STREAM_ENCODER_INVALID_METADATA",
Josh Coalson0a15c142001-06-13 17:59:57 +0000440 "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"
Josh Coalsoncbf595f2000-12-22 22:35:33 +0000445};
446
Josh Coalson6afed9f2002-10-16 22:29:47 +0000447FLAC_API const char * const FLAC__StreamEncoderWriteStatusString[] = {
Josh Coalson5c491a12002-08-01 06:39:40 +0000448 "FLAC__STREAM_ENCODER_WRITE_STATUS_OK",
449 "FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR"
Josh Coalsoncbf595f2000-12-22 22:35:33 +0000450};
451
Josh Coalson0a15c142001-06-13 17:59:57 +0000452/***********************************************************************
453 *
454 * Class constructor/destructor
455 *
Josh Coalsond86e03b2002-08-03 21:56:15 +0000456 */
Josh Coalson6afed9f2002-10-16 22:29:47 +0000457FLAC_API FLAC__StreamEncoder *FLAC__stream_encoder_new()
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000458{
Josh Coalson0a15c142001-06-13 17:59:57 +0000459 FLAC__StreamEncoder *encoder;
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000460 unsigned i;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000461
Josh Coalson0a15c142001-06-13 17:59:57 +0000462 FLAC__ASSERT(sizeof(int) >= 4); /* we want to die right away if this is not true */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000463
Josh Coalsonea7155f2002-10-18 05:49:19 +0000464 encoder = (FLAC__StreamEncoder*)calloc(1, sizeof(FLAC__StreamEncoder));
Josh Coalson0a15c142001-06-13 17:59:57 +0000465 if(encoder == 0) {
466 return 0;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000467 }
Josh Coalsond86e03b2002-08-03 21:56:15 +0000468
Josh Coalsonea7155f2002-10-18 05:49:19 +0000469 encoder->protected_ = (FLAC__StreamEncoderProtected*)calloc(1, sizeof(FLAC__StreamEncoderProtected));
Josh Coalsonfa697a92001-08-16 20:07:29 +0000470 if(encoder->protected_ == 0) {
Josh Coalson0a15c142001-06-13 17:59:57 +0000471 free(encoder);
472 return 0;
Josh Coalsond98c43d2001-05-13 05:17:01 +0000473 }
Josh Coalsond86e03b2002-08-03 21:56:15 +0000474
Josh Coalsonea7155f2002-10-18 05:49:19 +0000475 encoder->private_ = (FLAC__StreamEncoderPrivate*)calloc(1, sizeof(FLAC__StreamEncoderPrivate));
Josh Coalsonfa697a92001-08-16 20:07:29 +0000476 if(encoder->private_ == 0) {
477 free(encoder->protected_);
Josh Coalson0a15c142001-06-13 17:59:57 +0000478 free(encoder);
479 return 0;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000480 }
Josh Coalsond86e03b2002-08-03 21:56:15 +0000481
Josh Coalsonaec256b2002-03-12 16:19:54 +0000482 encoder->private_->frame = FLAC__bitbuffer_new();
483 if(encoder->private_->frame == 0) {
484 free(encoder->private_);
485 free(encoder->protected_);
486 free(encoder);
487 return 0;
488 }
Josh Coalsond98c43d2001-05-13 05:17:01 +0000489
Josh Coalsonf1eff452002-07-31 07:05:33 +0000490 set_defaults_(encoder);
Josh Coalson92031602002-07-24 06:02:11 +0000491
Josh Coalson3262b0d2002-08-14 20:58:42 +0000492 encoder->private_->is_being_deleted = false;
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000493
494 for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
495 encoder->private_->subframe_workspace_ptr[i][0] = &encoder->private_->subframe_workspace[i][0];
496 encoder->private_->subframe_workspace_ptr[i][1] = &encoder->private_->subframe_workspace[i][1];
497 }
498 for(i = 0; i < 2; i++) {
499 encoder->private_->subframe_workspace_ptr_mid_side[i][0] = &encoder->private_->subframe_workspace_mid_side[i][0];
500 encoder->private_->subframe_workspace_ptr_mid_side[i][1] = &encoder->private_->subframe_workspace_mid_side[i][1];
501 }
502 for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000503 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];
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000505 }
506 for(i = 0; i < 2; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000507 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];
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000509 }
510
511 for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000512 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]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000514 }
515 for(i = 0; i < 2; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000516 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]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000518 }
519 for(i = 0; i < 2; i++)
Josh Coalsona37ba462002-08-19 21:36:39 +0000520 FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&encoder->private_->partitioned_rice_contents_extra[i]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000521
Josh Coalsonfa697a92001-08-16 20:07:29 +0000522 encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000523
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000524 return encoder;
525}
526
Josh Coalson6afed9f2002-10-16 22:29:47 +0000527FLAC_API void FLAC__stream_encoder_delete(FLAC__StreamEncoder *encoder)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000528{
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000529 unsigned i;
530
Josh Coalsonf1eff452002-07-31 07:05:33 +0000531 FLAC__ASSERT(0 != encoder);
532 FLAC__ASSERT(0 != encoder->protected_);
533 FLAC__ASSERT(0 != encoder->private_);
534 FLAC__ASSERT(0 != encoder->private_->frame);
Josh Coalson0a15c142001-06-13 17:59:57 +0000535
Josh Coalson3262b0d2002-08-14 20:58:42 +0000536 encoder->private_->is_being_deleted = true;
537
538 FLAC__stream_encoder_finish(encoder);
539
Josh Coalson4fa90592002-12-04 07:01:37 +0000540 if(0 != encoder->private_->verify.decoder)
Josh Coalsond86e03b2002-08-03 21:56:15 +0000541 FLAC__stream_decoder_delete(encoder->private_->verify.decoder);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000542
543 for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000544 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]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000546 }
547 for(i = 0; i < 2; i++) {
Josh Coalsona37ba462002-08-19 21:36:39 +0000548 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]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000550 }
551 for(i = 0; i < 2; i++)
Josh Coalsona37ba462002-08-19 21:36:39 +0000552 FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&encoder->private_->partitioned_rice_contents_extra[i]);
Josh Coalsonb7023aa2002-08-17 15:23:43 +0000553
Josh Coalsonaec256b2002-03-12 16:19:54 +0000554 FLAC__bitbuffer_delete(encoder->private_->frame);
Josh Coalsonfa697a92001-08-16 20:07:29 +0000555 free(encoder->private_);
556 free(encoder->protected_);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000557 free(encoder);
558}
559
Josh Coalson0a15c142001-06-13 17:59:57 +0000560/***********************************************************************
561 *
562 * Public class methods
563 *
564 ***********************************************************************/
565
Josh Coalson6afed9f2002-10-16 22:29:47 +0000566FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_init(FLAC__StreamEncoder *encoder)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000567{
568 unsigned i;
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000569 FLAC__bool metadata_has_seektable, metadata_has_vorbis_comment;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000570
Josh Coalsonf1eff452002-07-31 07:05:33 +0000571 FLAC__ASSERT(0 != encoder);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000572
Josh Coalsonfa697a92001-08-16 20:07:29 +0000573 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
574 return encoder->protected_->state = FLAC__STREAM_ENCODER_ALREADY_INITIALIZED;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000575
Josh Coalsonfa697a92001-08-16 20:07:29 +0000576 encoder->protected_->state = FLAC__STREAM_ENCODER_OK;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000577
Josh Coalsonfa697a92001-08-16 20:07:29 +0000578 if(0 == encoder->private_->write_callback || 0 == encoder->private_->metadata_callback)
579 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_CALLBACK;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000580
Josh Coalsonfa697a92001-08-16 20:07:29 +0000581 if(encoder->protected_->channels == 0 || encoder->protected_->channels > FLAC__MAX_CHANNELS)
582 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_NUMBER_OF_CHANNELS;
Josh Coalson69f1ee02001-01-24 00:54:43 +0000583
Josh Coalsonfa697a92001-08-16 20:07:29 +0000584 if(encoder->protected_->do_mid_side_stereo && encoder->protected_->channels != 2)
585 return encoder->protected_->state = FLAC__STREAM_ENCODER_MID_SIDE_CHANNELS_MISMATCH;
Josh Coalsond37d1352001-05-30 23:09:31 +0000586
Josh Coalsonfa697a92001-08-16 20:07:29 +0000587 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;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000589
Josh Coalsonfa697a92001-08-16 20:07:29 +0000590 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 */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000592
Josh Coalson76c68bc2002-05-17 06:22:02 +0000593 if(encoder->protected_->bits_per_sample < FLAC__MIN_BITS_PER_SAMPLE || encoder->protected_->bits_per_sample > FLAC__REFERENCE_CODEC_MAX_BITS_PER_SAMPLE)
Josh Coalsonfa697a92001-08-16 20:07:29 +0000594 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_BITS_PER_SAMPLE;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000595
Josh Coalson0833f342002-07-15 05:31:55 +0000596 if(!FLAC__format_sample_rate_is_valid(encoder->protected_->sample_rate))
Josh Coalsonfa697a92001-08-16 20:07:29 +0000597 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_SAMPLE_RATE;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000598
Josh Coalsonfa697a92001-08-16 20:07:29 +0000599 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;
Josh Coalson0a15c142001-06-13 17:59:57 +0000601
Josh Coalson20ac2c12002-08-30 05:47:14 +0000602 if(encoder->protected_->max_lpc_order > FLAC__MAX_LPC_ORDER)
603 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_MAX_LPC_ORDER;
604
Josh Coalsonfa697a92001-08-16 20:07:29 +0000605 if(encoder->protected_->blocksize < encoder->protected_->max_lpc_order)
606 return encoder->protected_->state = FLAC__STREAM_ENCODER_BLOCK_SIZE_TOO_SMALL_FOR_LPC_ORDER;
Josh Coalson0a15c142001-06-13 17:59:57 +0000607
Josh Coalsonfa697a92001-08-16 20:07:29 +0000608 if(encoder->protected_->qlp_coeff_precision == 0) {
609 if(encoder->protected_->bits_per_sample < 16) {
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000610 /* @@@ need some data about how to set this here w.r.t. blocksize and sample rate */
611 /* @@@ until then we'll make a guess */
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000612 encoder->protected_->qlp_coeff_precision = max(FLAC__MIN_QLP_COEFF_PRECISION, 2 + encoder->protected_->bits_per_sample / 2);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000613 }
Josh Coalsonfa697a92001-08-16 20:07:29 +0000614 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;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000627 else
Josh Coalsonfa697a92001-08-16 20:07:29 +0000628 encoder->protected_->qlp_coeff_precision = 13;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000629 }
630 else {
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000631 if(encoder->protected_->blocksize <= 384)
632 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION-2;
633 else if(encoder->protected_->blocksize <= 1152)
634 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION-1;
635 else
636 encoder->protected_->qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000637 }
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000638 FLAC__ASSERT(encoder->protected_->qlp_coeff_precision <= FLAC__MAX_QLP_COEFF_PRECISION);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000639 }
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000640 else if(encoder->protected_->qlp_coeff_precision < FLAC__MIN_QLP_COEFF_PRECISION || encoder->protected_->qlp_coeff_precision > FLAC__MAX_QLP_COEFF_PRECISION)
Josh Coalsonfa697a92001-08-16 20:07:29 +0000641 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_QLP_COEFF_PRECISION;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000642
Josh Coalsonfa697a92001-08-16 20:07:29 +0000643 if(encoder->protected_->streamable_subset) {
Josh Coalson20ac2c12002-08-30 05:47:14 +0000644 if(
645 encoder->protected_->blocksize != 192 &&
646 encoder->protected_->blocksize != 576 &&
647 encoder->protected_->blocksize != 1152 &&
648 encoder->protected_->blocksize != 2304 &&
649 encoder->protected_->blocksize != 4608 &&
650 encoder->protected_->blocksize != 256 &&
651 encoder->protected_->blocksize != 512 &&
652 encoder->protected_->blocksize != 1024 &&
653 encoder->protected_->blocksize != 2048 &&
654 encoder->protected_->blocksize != 4096 &&
655 encoder->protected_->blocksize != 8192 &&
656 encoder->protected_->blocksize != 16384
657 )
Josh Coalsonfa697a92001-08-16 20:07:29 +0000658 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
Josh Coalson20ac2c12002-08-30 05:47:14 +0000659 if(
660 encoder->protected_->sample_rate != 8000 &&
661 encoder->protected_->sample_rate != 16000 &&
662 encoder->protected_->sample_rate != 22050 &&
663 encoder->protected_->sample_rate != 24000 &&
664 encoder->protected_->sample_rate != 32000 &&
665 encoder->protected_->sample_rate != 44100 &&
666 encoder->protected_->sample_rate != 48000 &&
667 encoder->protected_->sample_rate != 96000
668 )
669 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
670 if(
671 encoder->protected_->bits_per_sample != 8 &&
672 encoder->protected_->bits_per_sample != 12 &&
673 encoder->protected_->bits_per_sample != 16 &&
674 encoder->protected_->bits_per_sample != 20 &&
675 encoder->protected_->bits_per_sample != 24
676 )
677 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
Josh Coalsonc1c8d492002-09-26 04:42:10 +0000678 if(encoder->protected_->max_residual_partition_order > FLAC__SUBSET_MAX_RICE_PARTITION_ORDER)
Josh Coalsonfa697a92001-08-16 20:07:29 +0000679 return encoder->protected_->state = FLAC__STREAM_ENCODER_NOT_STREAMABLE;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000680 }
681
Josh Coalsonfa697a92001-08-16 20:07:29 +0000682 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;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000686
Josh Coalson66075c12002-06-01 05:39:38 +0000687 /* validate metadata */
688 if(0 == encoder->protected_->metadata && encoder->protected_->num_metadata_blocks > 0)
689 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000690 metadata_has_seektable = false;
691 metadata_has_vorbis_comment = false;
Josh Coalson66075c12002-06-01 05:39:38 +0000692 for(i = 0; i < encoder->protected_->num_metadata_blocks; i++) {
Josh Coalsond0609472003-01-10 05:37:13 +0000693 if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_STREAMINFO)
Josh Coalson66075c12002-06-01 05:39:38 +0000694 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
695 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_SEEKTABLE) {
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000696 if(metadata_has_seektable) /* only one is allowed */
697 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
698 metadata_has_seektable = true;
Josh Coalson0833f342002-07-15 05:31:55 +0000699 if(!FLAC__format_seektable_is_legal(&encoder->protected_->metadata[i]->data.seek_table))
Josh Coalson66075c12002-06-01 05:39:38 +0000700 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
701 }
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000702 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_VORBIS_COMMENT) {
703 if(metadata_has_vorbis_comment) /* only one is allowed */
704 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
705 metadata_has_vorbis_comment = true;
706 }
Josh Coalsone4869382002-11-15 05:41:48 +0000707 else if(encoder->protected_->metadata[i]->type == FLAC__METADATA_TYPE_CUESHEET) {
Josh Coalson8f0c71b2002-12-05 06:37:46 +0000708 if(!FLAC__format_cuesheet_is_legal(&encoder->protected_->metadata[i]->data.cue_sheet, encoder->protected_->metadata[i]->data.cue_sheet.is_cd, /*violation=*/0))
Josh Coalsone4869382002-11-15 05:41:48 +0000709 return encoder->protected_->state = FLAC__STREAM_ENCODER_INVALID_METADATA;
710 }
Josh Coalson66075c12002-06-01 05:39:38 +0000711 }
712
Josh Coalsonfa697a92001-08-16 20:07:29 +0000713 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;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000716#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +0000717 encoder->private_->real_signal_unaligned[i] = encoder->private_->real_signal[i] = 0;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000718#endif
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000719 }
720 for(i = 0; i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +0000721 encoder->private_->integer_signal_mid_side_unaligned[i] = encoder->private_->integer_signal_mid_side[i] = 0;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000722#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +0000723 encoder->private_->real_signal_mid_side_unaligned[i] = encoder->private_->real_signal_mid_side[i] = 0;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000724#endif
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000725 }
Josh Coalsonfa697a92001-08-16 20:07:29 +0000726 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;
Josh Coalson94e02cd2001-01-25 10:41:06 +0000730 }
731 for(i = 0; i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +0000732 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;
Josh Coalson94e02cd2001-01-25 10:41:06 +0000735 }
Josh Coalsonfa697a92001-08-16 20:07:29 +0000736 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;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000739#ifndef FLAC__INTEGER_ONLY_LIBRARY
740 encoder->private_->loose_mid_side_stereo_frames = (unsigned)((FLAC__double)encoder->protected_->sample_rate * 0.4 / (FLAC__double)encoder->protected_->blocksize + 0.5);
741#else
742 /* 26214 is the approximate fixed-point equivalent to 0.4 (0.4 * 2^16) */
743 /* sample rate can be up to 655350 Hz, and thus use 20 bits, so we do the multiply&divide by hand */
744 FLAC__ASSERT(FLAC__MAX_SAMPLE_RATE <= 655350);
745 FLAC__ASSERT(FLAC__MAX_BLOCK_SIZE <= 65535);
746 FLAC__ASSERT(encoder->protected_->sample_rate <= 655350);
747 FLAC__ASSERT(encoder->protected_->blocksize <= 65535);
748 encoder->private_->loose_mid_side_stereo_frames = (unsigned)FLAC__fixedpoint_trunc((((FLAC__uint64)(encoder->protected_->sample_rate) * (FLAC__uint64)(26214)) << 16) / (encoder->protected_->blocksize<<16) + FLAC__FP_ONE_HALF);
749#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +0000750 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;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000755
Josh Coalsonfa697a92001-08-16 20:07:29 +0000756 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 */
Josh Coalson8395d022001-07-12 21:25:22 +0000759
Josh Coalsoncf30f502001-05-23 20:57:44 +0000760 /*
761 * get the CPU info and set the function pointers
762 */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000763 FLAC__cpu_info(&encoder->private_->cpuinfo);
Josh Coalsoncf30f502001-05-23 20:57:44 +0000764 /* first default to the non-asm routines */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000765#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +0000766 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000767#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +0000768 encoder->private_->local_fixed_compute_best_predictor = FLAC__fixed_compute_best_predictor;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000769#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +0000770 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients = FLAC__lpc_compute_residual_from_qlp_coefficients;
Josh Coalsonc9c0d132002-10-04 05:29:05 +0000771 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_64bit = FLAC__lpc_compute_residual_from_qlp_coefficients_wide;
Josh Coalsonfa697a92001-08-16 20:07:29 +0000772 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit = FLAC__lpc_compute_residual_from_qlp_coefficients;
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000773#endif
Josh Coalsoncf30f502001-05-23 20:57:44 +0000774 /* now override with asm where appropriate */
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000775#ifndef FLAC__INTEGER_ONLY_LIBRARY
776# ifndef FLAC__NO_ASM
Josh Coalsonfa697a92001-08-16 20:07:29 +0000777 if(encoder->private_->cpuinfo.use_asm) {
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000778# ifdef FLAC__CPU_IA32
Josh Coalsonfa697a92001-08-16 20:07:29 +0000779 FLAC__ASSERT(encoder->private_->cpuinfo.type == FLAC__CPUINFO_TYPE_IA32);
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000780# ifdef FLAC__HAS_NASM
781# ifdef FLAC__SSE_OS
Josh Coalson48cbe662002-12-30 23:38:14 +0000782 if(encoder->private_->cpuinfo.data.ia32.sse) {
Josh Coalsonfa697a92001-08-16 20:07:29 +0000783 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;
Josh Coalson021ad3b2001-07-18 00:25:52 +0000789 else
Josh Coalsonfa697a92001-08-16 20:07:29 +0000790 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32;
Josh Coalson021ad3b2001-07-18 00:25:52 +0000791 }
Josh Coalson48cbe662002-12-30 23:38:14 +0000792 else
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000793# endif /* FLAC__SSE_OS */
Josh Coalson48cbe662002-12-30 23:38:14 +0000794 if(encoder->private_->cpuinfo.data.ia32._3dnow)
Josh Coalsonfa697a92001-08-16 20:07:29 +0000795 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32_3dnow;
Josh Coalsonaa255362001-05-31 06:17:41 +0000796 else
Josh Coalsonfa697a92001-08-16 20:07:29 +0000797 encoder->private_->local_lpc_compute_autocorrelation = FLAC__lpc_compute_autocorrelation_asm_ia32;
Josh Coalsonfa697a92001-08-16 20:07:29 +0000798 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;
Josh Coalson021ad3b2001-07-18 00:25:52 +0000801 }
802 else {
Josh Coalsonfa697a92001-08-16 20:07:29 +0000803 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;
Josh Coalson021ad3b2001-07-18 00:25:52 +0000805 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000806 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 */
Josh Coalson021ad3b2001-07-18 00:25:52 +0000810 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +0000811# endif /* !FLAC__NO_ASM */
812#endif /* !FLAC__INTEGER_ONLY_LIBRARY */
Josh Coalson8395d022001-07-12 21:25:22 +0000813 /* finally override based on wide-ness if necessary */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000814 if(encoder->private_->use_wide_by_block) {
815 encoder->private_->local_fixed_compute_best_predictor = FLAC__fixed_compute_best_predictor_wide;
Josh Coalson8395d022001-07-12 21:25:22 +0000816 }
Josh Coalsoncf30f502001-05-23 20:57:44 +0000817
Josh Coalson8395d022001-07-12 21:25:22 +0000818 /* we require precompute_partition_sums if do_escape_coding because of their intertwined nature */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000819 encoder->private_->precompute_partition_sums = (encoder->protected_->max_residual_partition_order > encoder->protected_->min_residual_partition_order) || encoder->protected_->do_escape_coding;
Josh Coalsoneef56702001-03-30 00:45:22 +0000820
Josh Coalsonf1eff452002-07-31 07:05:33 +0000821 if(!resize_buffers_(encoder, encoder->protected_->blocksize)) {
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000822 /* the above function sets the state for us in case of an error */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000823 return encoder->protected_->state;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000824 }
Josh Coalsonaec256b2002-03-12 16:19:54 +0000825
826 if(!FLAC__bitbuffer_init(encoder->private_->frame))
827 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000828
829 /*
Josh Coalsond86e03b2002-08-03 21:56:15 +0000830 * Set up the verify stuff if necessary
831 */
832 if(encoder->protected_->verify) {
833 /*
834 * First, set up the fifo which will hold the
835 * original signal to compare against
836 */
837 encoder->private_->verify.input_fifo.size = encoder->protected_->blocksize;
838 for(i = 0; i < encoder->protected_->channels; i++) {
839 if(0 == (encoder->private_->verify.input_fifo.data[i] = (FLAC__int32*)malloc(sizeof(FLAC__int32) * encoder->private_->verify.input_fifo.size)))
840 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
841 }
842 encoder->private_->verify.input_fifo.tail = 0;
843
844 /*
845 * Now set up a stream decoder for verification
846 */
847 encoder->private_->verify.decoder = FLAC__stream_decoder_new();
848 if(0 == encoder->private_->verify.decoder)
849 return encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
850
851 FLAC__stream_decoder_set_read_callback(encoder->private_->verify.decoder, verify_read_callback_);
852 FLAC__stream_decoder_set_write_callback(encoder->private_->verify.decoder, verify_write_callback_);
853 FLAC__stream_decoder_set_metadata_callback(encoder->private_->verify.decoder, verify_metadata_callback_);
854 FLAC__stream_decoder_set_error_callback(encoder->private_->verify.decoder, verify_error_callback_);
855 FLAC__stream_decoder_set_client_data(encoder->private_->verify.decoder, encoder);
856 if(FLAC__stream_decoder_init(encoder->private_->verify.decoder) != FLAC__STREAM_DECODER_SEARCH_FOR_METADATA)
857 return encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
858 }
Josh Coalson589f8c72002-08-07 23:54:55 +0000859 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;
Josh Coalsond86e03b2002-08-03 21:56:15 +0000865
866 /*
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000867 * write the stream header
868 */
Josh Coalsond86e03b2002-08-03 21:56:15 +0000869 if(encoder->protected_->verify)
870 encoder->private_->verify.state_hint = ENCODER_IN_MAGIC;
Josh Coalsonaec256b2002-03-12 16:19:54 +0000871 if(!FLAC__bitbuffer_write_raw_uint32(encoder->private_->frame, FLAC__STREAM_SYNC, FLAC__STREAM_SYNC_LEN))
Josh Coalsonfa697a92001-08-16 20:07:29 +0000872 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalsond86e03b2002-08-03 21:56:15 +0000873 if(!write_bitbuffer_(encoder, 0)) {
874 /* the above function sets the state for us in case of an error */
875 return encoder->protected_->state;
876 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000877
Josh Coalson5c491a12002-08-01 06:39:40 +0000878 /*
879 * write the STREAMINFO metadata block
880 */
Josh Coalsond86e03b2002-08-03 21:56:15 +0000881 if(encoder->protected_->verify)
882 encoder->private_->verify.state_hint = ENCODER_IN_METADATA;
Josh Coalsonfa697a92001-08-16 20:07:29 +0000883 encoder->private_->metadata.type = FLAC__METADATA_TYPE_STREAMINFO;
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000884 encoder->private_->metadata.is_last = false; /* we will have at a minimum a VORBIS_COMMENT afterwards */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000885 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 */
Josh Coalson3e7a96e2004-07-23 05:18:22 +0000895 FLAC__MD5Init(&encoder->private_->md5context);
Josh Coalson7424d2f2002-11-06 07:10:38 +0000896 if(!FLAC__bitbuffer_clear(encoder->private_->frame))
897 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonaec256b2002-03-12 16:19:54 +0000898 if(!FLAC__add_metadata_block(&encoder->private_->metadata, encoder->private_->frame))
Josh Coalsonfa697a92001-08-16 20:07:29 +0000899 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalsond86e03b2002-08-03 21:56:15 +0000900 if(!write_bitbuffer_(encoder, 0)) {
901 /* the above function sets the state for us in case of an error */
902 return encoder->protected_->state;
903 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000904
Josh Coalson5c491a12002-08-01 06:39:40 +0000905 /*
906 * Now that the STREAMINFO block is written, we can init this to an
907 * absurdly-high value...
908 */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000909 encoder->private_->metadata.data.stream_info.min_framesize = (1u << FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN) - 1;
Josh Coalsoncbbbb5f2001-01-23 00:41:48 +0000910 /* ... and clear this to 0 */
Josh Coalsonfa697a92001-08-16 20:07:29 +0000911 encoder->private_->metadata.data.stream_info.total_samples = 0;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000912
Josh Coalson5c491a12002-08-01 06:39:40 +0000913 /*
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000914 * 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).
Josh Coalson69cfda72004-09-10 00:38:21 +0000917 *
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 it true it will have already
921 * insured that the metadata list is properly ordered.)
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000922 */
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;
Josh Coalson7424d2f2002-11-06 07:10:38 +0000932 if(!FLAC__bitbuffer_clear(encoder->private_->frame))
933 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsoncb9d93a2002-08-25 05:27:15 +0000934 if(!FLAC__add_metadata_block(&vorbis_comment, encoder->private_->frame))
935 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
936 if(!write_bitbuffer_(encoder, 0)) {
937 /* the above function sets the state for us in case of an error */
938 return encoder->protected_->state;
939 }
940 }
941
942 /*
Josh Coalson5c491a12002-08-01 06:39:40 +0000943 * write the user's metadata blocks
944 */
945 for(i = 0; i < encoder->protected_->num_metadata_blocks; i++) {
946 encoder->protected_->metadata[i]->is_last = (i == encoder->protected_->num_metadata_blocks - 1);
Josh Coalson7424d2f2002-11-06 07:10:38 +0000947 if(!FLAC__bitbuffer_clear(encoder->private_->frame))
948 return encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalson5c491a12002-08-01 06:39:40 +0000949 if(!FLAC__add_metadata_block(encoder->protected_->metadata[i], encoder->private_->frame))
950 return encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalsond86e03b2002-08-03 21:56:15 +0000951 if(!write_bitbuffer_(encoder, 0)) {
952 /* the above function sets the state for us in case of an error */
953 return encoder->protected_->state;
954 }
Josh Coalson5c491a12002-08-01 06:39:40 +0000955 }
956
Josh Coalsond86e03b2002-08-03 21:56:15 +0000957 if(encoder->protected_->verify)
958 encoder->private_->verify.state_hint = ENCODER_IN_AUDIO;
959
Josh Coalsonfa697a92001-08-16 20:07:29 +0000960 return encoder->protected_->state;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000961}
962
Josh Coalson6afed9f2002-10-16 22:29:47 +0000963FLAC_API void FLAC__stream_encoder_finish(FLAC__StreamEncoder *encoder)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000964{
Josh Coalsonf1eff452002-07-31 07:05:33 +0000965 FLAC__ASSERT(0 != encoder);
Josh Coalson2b245f22002-08-07 17:10:50 +0000966
Josh Coalsonfa697a92001-08-16 20:07:29 +0000967 if(encoder->protected_->state == FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000968 return;
Josh Coalson2b245f22002-08-07 17:10:50 +0000969
Josh Coalson3262b0d2002-08-14 20:58:42 +0000970 if(encoder->protected_->state == FLAC__STREAM_ENCODER_OK && !encoder->private_->is_being_deleted) {
Josh Coalson2b245f22002-08-07 17:10:50 +0000971 if(encoder->private_->current_sample_number != 0) {
972 encoder->protected_->blocksize = encoder->private_->current_sample_number;
973 process_frame_(encoder, true); /* true => is last frame */
974 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000975 }
Josh Coalson2b245f22002-08-07 17:10:50 +0000976
Josh Coalson3e7a96e2004-07-23 05:18:22 +0000977 FLAC__MD5Final(encoder->private_->metadata.data.stream_info.md5sum, &encoder->private_->md5context);
Josh Coalson2b245f22002-08-07 17:10:50 +0000978
Josh Coalson3262b0d2002-08-14 20:58:42 +0000979 if(encoder->protected_->state == FLAC__STREAM_ENCODER_OK && !encoder->private_->is_being_deleted) {
Josh Coalson2b245f22002-08-07 17:10:50 +0000980 encoder->private_->metadata_callback(encoder, &encoder->private_->metadata, encoder->private_->client_data);
981 }
Josh Coalson0a15c142001-06-13 17:59:57 +0000982
Josh Coalsond86e03b2002-08-03 21:56:15 +0000983 if(encoder->protected_->verify && 0 != encoder->private_->verify.decoder)
984 FLAC__stream_decoder_finish(encoder->private_->verify.decoder);
985
Josh Coalsonf1eff452002-07-31 07:05:33 +0000986 free_(encoder);
987 set_defaults_(encoder);
Josh Coalson92031602002-07-24 06:02:11 +0000988
Josh Coalsonfa697a92001-08-16 20:07:29 +0000989 encoder->protected_->state = FLAC__STREAM_ENCODER_UNINITIALIZED;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +0000990}
991
Josh Coalson6afed9f2002-10-16 22:29:47 +0000992FLAC_API FLAC__bool FLAC__stream_encoder_set_verify(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsond86e03b2002-08-03 21:56:15 +0000993{
994 FLAC__ASSERT(0 != encoder);
995 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
996 return false;
997 encoder->protected_->verify = value;
998 return true;
999}
1000
Josh Coalson6afed9f2002-10-16 22:29:47 +00001001FLAC_API FLAC__bool FLAC__stream_encoder_set_streamable_subset(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001002{
Josh Coalson92031602002-07-24 06:02:11 +00001003 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001004 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001005 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001006 encoder->protected_->streamable_subset = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001007 return true;
1008}
1009
Josh Coalson6afed9f2002-10-16 22:29:47 +00001010FLAC_API FLAC__bool FLAC__stream_encoder_set_do_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001011{
Josh Coalson92031602002-07-24 06:02:11 +00001012 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001013 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001014 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001015 encoder->protected_->do_mid_side_stereo = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001016 return true;
1017}
1018
Josh Coalson6afed9f2002-10-16 22:29:47 +00001019FLAC_API FLAC__bool FLAC__stream_encoder_set_loose_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001020{
Josh Coalson92031602002-07-24 06:02:11 +00001021 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001022 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001023 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001024 encoder->protected_->loose_mid_side_stereo = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001025 return true;
1026}
1027
Josh Coalson6afed9f2002-10-16 22:29:47 +00001028FLAC_API FLAC__bool FLAC__stream_encoder_set_channels(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001029{
Josh Coalson92031602002-07-24 06:02:11 +00001030 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001031 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001032 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001033 encoder->protected_->channels = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001034 return true;
1035}
1036
Josh Coalson6afed9f2002-10-16 22:29:47 +00001037FLAC_API FLAC__bool FLAC__stream_encoder_set_bits_per_sample(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001038{
Josh Coalson92031602002-07-24 06:02:11 +00001039 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001040 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001041 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001042 encoder->protected_->bits_per_sample = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001043 return true;
1044}
1045
Josh Coalson6afed9f2002-10-16 22:29:47 +00001046FLAC_API FLAC__bool FLAC__stream_encoder_set_sample_rate(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001047{
Josh Coalson92031602002-07-24 06:02:11 +00001048 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001049 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001050 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001051 encoder->protected_->sample_rate = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001052 return true;
1053}
1054
Josh Coalson6afed9f2002-10-16 22:29:47 +00001055FLAC_API FLAC__bool FLAC__stream_encoder_set_blocksize(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001056{
Josh Coalson92031602002-07-24 06:02:11 +00001057 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001058 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001059 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001060 encoder->protected_->blocksize = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001061 return true;
1062}
1063
Josh Coalson6afed9f2002-10-16 22:29:47 +00001064FLAC_API FLAC__bool FLAC__stream_encoder_set_max_lpc_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001065{
Josh Coalson92031602002-07-24 06:02:11 +00001066 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001067 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001068 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001069 encoder->protected_->max_lpc_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001070 return true;
1071}
1072
Josh Coalson6afed9f2002-10-16 22:29:47 +00001073FLAC_API FLAC__bool FLAC__stream_encoder_set_qlp_coeff_precision(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001074{
Josh Coalson92031602002-07-24 06:02:11 +00001075 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001076 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001077 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001078 encoder->protected_->qlp_coeff_precision = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001079 return true;
1080}
1081
Josh Coalson6afed9f2002-10-16 22:29:47 +00001082FLAC_API FLAC__bool FLAC__stream_encoder_set_do_qlp_coeff_prec_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001083{
Josh Coalson92031602002-07-24 06:02:11 +00001084 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001085 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001086 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001087 encoder->protected_->do_qlp_coeff_prec_search = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001088 return true;
1089}
1090
Josh Coalson6afed9f2002-10-16 22:29:47 +00001091FLAC_API FLAC__bool FLAC__stream_encoder_set_do_escape_coding(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson8395d022001-07-12 21:25:22 +00001092{
Josh Coalson92031602002-07-24 06:02:11 +00001093 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001094 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson8395d022001-07-12 21:25:22 +00001095 return false;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001096#if 0
1097 /*@@@ deprecated: */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001098 encoder->protected_->do_escape_coding = value;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001099#else
1100 (void)value;
1101#endif
Josh Coalson8395d022001-07-12 21:25:22 +00001102 return true;
1103}
1104
Josh Coalson6afed9f2002-10-16 22:29:47 +00001105FLAC_API FLAC__bool FLAC__stream_encoder_set_do_exhaustive_model_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001106{
Josh Coalson92031602002-07-24 06:02:11 +00001107 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001108 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001109 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001110 encoder->protected_->do_exhaustive_model_search = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001111 return true;
1112}
1113
Josh Coalson6afed9f2002-10-16 22:29:47 +00001114FLAC_API FLAC__bool FLAC__stream_encoder_set_min_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001115{
Josh Coalson92031602002-07-24 06:02:11 +00001116 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001117 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001118 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001119 encoder->protected_->min_residual_partition_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001120 return true;
1121}
1122
Josh Coalson6afed9f2002-10-16 22:29:47 +00001123FLAC_API FLAC__bool FLAC__stream_encoder_set_max_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001124{
Josh Coalson92031602002-07-24 06:02:11 +00001125 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001126 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001127 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001128 encoder->protected_->max_residual_partition_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001129 return true;
1130}
1131
Josh Coalson6afed9f2002-10-16 22:29:47 +00001132FLAC_API FLAC__bool FLAC__stream_encoder_set_rice_parameter_search_dist(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001133{
Josh Coalson92031602002-07-24 06:02:11 +00001134 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001135 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001136 return false;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001137#if 0
1138 /*@@@ deprecated: */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001139 encoder->protected_->rice_parameter_search_dist = value;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001140#else
1141 (void)value;
1142#endif
Josh Coalson00e53872001-06-16 07:32:25 +00001143 return true;
1144}
1145
Josh Coalson6afed9f2002-10-16 22:29:47 +00001146FLAC_API FLAC__bool FLAC__stream_encoder_set_total_samples_estimate(FLAC__StreamEncoder *encoder, FLAC__uint64 value)
Josh Coalson00e53872001-06-16 07:32:25 +00001147{
Josh Coalson92031602002-07-24 06:02:11 +00001148 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001149 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001150 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001151 encoder->protected_->total_samples_estimate = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001152 return true;
1153}
1154
Josh Coalson6afed9f2002-10-16 22:29:47 +00001155FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata(FLAC__StreamEncoder *encoder, FLAC__StreamMetadata **metadata, unsigned num_blocks)
Josh Coalson00e53872001-06-16 07:32:25 +00001156{
Josh Coalson92031602002-07-24 06:02:11 +00001157 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001158 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001159 return false;
Josh Coalson66075c12002-06-01 05:39:38 +00001160 encoder->protected_->metadata = metadata;
1161 encoder->protected_->num_metadata_blocks = num_blocks;
Josh Coalson00e53872001-06-16 07:32:25 +00001162 return true;
1163}
1164
Josh Coalson6afed9f2002-10-16 22:29:47 +00001165FLAC_API FLAC__bool FLAC__stream_encoder_set_write_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderWriteCallback value)
Josh Coalson00e53872001-06-16 07:32:25 +00001166{
Josh Coalson92031602002-07-24 06:02:11 +00001167 FLAC__ASSERT(0 != encoder);
1168 FLAC__ASSERT(0 != value);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001169 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001170 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001171 encoder->private_->write_callback = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001172 return true;
1173}
1174
Josh Coalson6afed9f2002-10-16 22:29:47 +00001175FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderMetadataCallback value)
Josh Coalson00e53872001-06-16 07:32:25 +00001176{
Josh Coalson92031602002-07-24 06:02:11 +00001177 FLAC__ASSERT(0 != encoder);
1178 FLAC__ASSERT(0 != value);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001179 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001180 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001181 encoder->private_->metadata_callback = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001182 return true;
1183}
1184
Josh Coalson6afed9f2002-10-16 22:29:47 +00001185FLAC_API FLAC__bool FLAC__stream_encoder_set_client_data(FLAC__StreamEncoder *encoder, void *value)
Josh Coalson00e53872001-06-16 07:32:25 +00001186{
Josh Coalson92031602002-07-24 06:02:11 +00001187 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001188 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001189 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001190 encoder->private_->client_data = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001191 return true;
1192}
1193
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001194/*
1195 * These three functions are not static, but not publically exposed in
1196 * include/FLAC/ either. They are used by the test suite.
1197 */
Josh Coalson6afed9f2002-10-16 22:29:47 +00001198FLAC_API FLAC__bool FLAC__stream_encoder_disable_constant_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001199{
1200 FLAC__ASSERT(0 != encoder);
1201 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1202 return false;
1203 encoder->private_->disable_constant_subframes = value;
1204 return true;
1205}
1206
Josh Coalson6afed9f2002-10-16 22:29:47 +00001207FLAC_API FLAC__bool FLAC__stream_encoder_disable_fixed_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001208{
1209 FLAC__ASSERT(0 != encoder);
1210 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1211 return false;
1212 encoder->private_->disable_fixed_subframes = value;
1213 return true;
1214}
1215
Josh Coalson6afed9f2002-10-16 22:29:47 +00001216FLAC_API FLAC__bool FLAC__stream_encoder_disable_verbatim_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001217{
1218 FLAC__ASSERT(0 != encoder);
1219 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1220 return false;
1221 encoder->private_->disable_verbatim_subframes = value;
1222 return true;
1223}
1224
Josh Coalson6afed9f2002-10-16 22:29:47 +00001225FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_get_state(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001226{
Josh Coalson92031602002-07-24 06:02:11 +00001227 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001228 return encoder->protected_->state;
Josh Coalson0a15c142001-06-13 17:59:57 +00001229}
1230
Josh Coalson6afed9f2002-10-16 22:29:47 +00001231FLAC_API FLAC__StreamDecoderState FLAC__stream_encoder_get_verify_decoder_state(const FLAC__StreamEncoder *encoder)
Josh Coalsond86e03b2002-08-03 21:56:15 +00001232{
1233 FLAC__ASSERT(0 != encoder);
1234 if(encoder->protected_->verify)
1235 return FLAC__stream_decoder_get_state(encoder->private_->verify.decoder);
1236 else
1237 return FLAC__STREAM_DECODER_UNINITIALIZED;
1238}
1239
Josh Coalson02954222002-11-08 06:16:31 +00001240FLAC_API const char *FLAC__stream_encoder_get_resolved_state_string(const FLAC__StreamEncoder *encoder)
1241{
1242 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR)
1243 return FLAC__StreamEncoderStateString[encoder->protected_->state];
1244 else
Josh Coalson807140d2003-09-24 22:10:51 +00001245 return FLAC__stream_decoder_get_resolved_state_string(encoder->private_->verify.decoder);
Josh Coalson02954222002-11-08 06:16:31 +00001246}
1247
Josh Coalson6afed9f2002-10-16 22:29:47 +00001248FLAC_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)
Josh Coalson589f8c72002-08-07 23:54:55 +00001249{
1250 FLAC__ASSERT(0 != encoder);
1251 if(0 != absolute_sample)
1252 *absolute_sample = encoder->private_->verify.error_stats.absolute_sample;
1253 if(0 != frame_number)
1254 *frame_number = encoder->private_->verify.error_stats.frame_number;
1255 if(0 != channel)
1256 *channel = encoder->private_->verify.error_stats.channel;
1257 if(0 != sample)
1258 *sample = encoder->private_->verify.error_stats.sample;
1259 if(0 != expected)
1260 *expected = encoder->private_->verify.error_stats.expected;
1261 if(0 != got)
1262 *got = encoder->private_->verify.error_stats.got;
1263}
1264
Josh Coalson6afed9f2002-10-16 22:29:47 +00001265FLAC_API FLAC__bool FLAC__stream_encoder_get_verify(const FLAC__StreamEncoder *encoder)
Josh Coalsond86e03b2002-08-03 21:56:15 +00001266{
1267 FLAC__ASSERT(0 != encoder);
1268 return encoder->protected_->verify;
1269}
1270
Josh Coalson6afed9f2002-10-16 22:29:47 +00001271FLAC_API FLAC__bool FLAC__stream_encoder_get_streamable_subset(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001272{
Josh Coalson92031602002-07-24 06:02:11 +00001273 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001274 return encoder->protected_->streamable_subset;
Josh Coalson0a15c142001-06-13 17:59:57 +00001275}
1276
Josh Coalson6afed9f2002-10-16 22:29:47 +00001277FLAC_API FLAC__bool FLAC__stream_encoder_get_do_mid_side_stereo(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001278{
Josh Coalson92031602002-07-24 06:02:11 +00001279 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001280 return encoder->protected_->do_mid_side_stereo;
Josh Coalson0a15c142001-06-13 17:59:57 +00001281}
1282
Josh Coalson6afed9f2002-10-16 22:29:47 +00001283FLAC_API FLAC__bool FLAC__stream_encoder_get_loose_mid_side_stereo(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001284{
Josh Coalson92031602002-07-24 06:02:11 +00001285 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001286 return encoder->protected_->loose_mid_side_stereo;
Josh Coalson0a15c142001-06-13 17:59:57 +00001287}
1288
Josh Coalson6afed9f2002-10-16 22:29:47 +00001289FLAC_API unsigned FLAC__stream_encoder_get_channels(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001290{
Josh Coalson92031602002-07-24 06:02:11 +00001291 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001292 return encoder->protected_->channels;
Josh Coalson0a15c142001-06-13 17:59:57 +00001293}
1294
Josh Coalson6afed9f2002-10-16 22:29:47 +00001295FLAC_API unsigned FLAC__stream_encoder_get_bits_per_sample(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001296{
Josh Coalson92031602002-07-24 06:02:11 +00001297 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001298 return encoder->protected_->bits_per_sample;
Josh Coalson0a15c142001-06-13 17:59:57 +00001299}
1300
Josh Coalson6afed9f2002-10-16 22:29:47 +00001301FLAC_API unsigned FLAC__stream_encoder_get_sample_rate(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001302{
Josh Coalson92031602002-07-24 06:02:11 +00001303 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001304 return encoder->protected_->sample_rate;
Josh Coalson0a15c142001-06-13 17:59:57 +00001305}
1306
Josh Coalson6afed9f2002-10-16 22:29:47 +00001307FLAC_API unsigned FLAC__stream_encoder_get_blocksize(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001308{
Josh Coalson92031602002-07-24 06:02:11 +00001309 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001310 return encoder->protected_->blocksize;
Josh Coalson0a15c142001-06-13 17:59:57 +00001311}
1312
Josh Coalson6afed9f2002-10-16 22:29:47 +00001313FLAC_API unsigned FLAC__stream_encoder_get_max_lpc_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001314{
Josh Coalson92031602002-07-24 06:02:11 +00001315 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001316 return encoder->protected_->max_lpc_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001317}
1318
Josh Coalson6afed9f2002-10-16 22:29:47 +00001319FLAC_API unsigned FLAC__stream_encoder_get_qlp_coeff_precision(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001320{
Josh Coalson92031602002-07-24 06:02:11 +00001321 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001322 return encoder->protected_->qlp_coeff_precision;
Josh Coalson0a15c142001-06-13 17:59:57 +00001323}
1324
Josh Coalson6afed9f2002-10-16 22:29:47 +00001325FLAC_API FLAC__bool FLAC__stream_encoder_get_do_qlp_coeff_prec_search(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001326{
Josh Coalson92031602002-07-24 06:02:11 +00001327 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001328 return encoder->protected_->do_qlp_coeff_prec_search;
Josh Coalson0a15c142001-06-13 17:59:57 +00001329}
1330
Josh Coalson6afed9f2002-10-16 22:29:47 +00001331FLAC_API FLAC__bool FLAC__stream_encoder_get_do_escape_coding(const FLAC__StreamEncoder *encoder)
Josh Coalson8395d022001-07-12 21:25:22 +00001332{
Josh Coalson92031602002-07-24 06:02:11 +00001333 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001334 return encoder->protected_->do_escape_coding;
Josh Coalson8395d022001-07-12 21:25:22 +00001335}
1336
Josh Coalson6afed9f2002-10-16 22:29:47 +00001337FLAC_API FLAC__bool FLAC__stream_encoder_get_do_exhaustive_model_search(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001338{
Josh Coalson92031602002-07-24 06:02:11 +00001339 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001340 return encoder->protected_->do_exhaustive_model_search;
Josh Coalson0a15c142001-06-13 17:59:57 +00001341}
1342
Josh Coalson6afed9f2002-10-16 22:29:47 +00001343FLAC_API unsigned FLAC__stream_encoder_get_min_residual_partition_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001344{
Josh Coalson92031602002-07-24 06:02:11 +00001345 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001346 return encoder->protected_->min_residual_partition_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001347}
1348
Josh Coalson6afed9f2002-10-16 22:29:47 +00001349FLAC_API unsigned FLAC__stream_encoder_get_max_residual_partition_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001350{
Josh Coalson92031602002-07-24 06:02:11 +00001351 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001352 return encoder->protected_->max_residual_partition_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001353}
1354
Josh Coalson6afed9f2002-10-16 22:29:47 +00001355FLAC_API unsigned FLAC__stream_encoder_get_rice_parameter_search_dist(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001356{
Josh Coalson92031602002-07-24 06:02:11 +00001357 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001358 return encoder->protected_->rice_parameter_search_dist;
Josh Coalson0a15c142001-06-13 17:59:57 +00001359}
1360
Josh Coalson6afed9f2002-10-16 22:29:47 +00001361FLAC_API FLAC__uint64 FLAC__stream_encoder_get_total_samples_estimate(const FLAC__StreamEncoder *encoder)
Josh Coalson3a7b2c92002-08-02 07:38:20 +00001362{
1363 FLAC__ASSERT(0 != encoder);
1364 return encoder->protected_->total_samples_estimate;
1365}
1366
Josh Coalson6afed9f2002-10-16 22:29:47 +00001367FLAC_API FLAC__bool FLAC__stream_encoder_process(FLAC__StreamEncoder *encoder, const FLAC__int32 * const buffer[], unsigned samples)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001368{
1369 unsigned i, j, channel;
Josh Coalson77e3f312001-06-23 03:03:24 +00001370 FLAC__int32 x, mid, side;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001371 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001372
Josh Coalsonf1eff452002-07-31 07:05:33 +00001373 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001374 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001375
1376 j = 0;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001377 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
Josh Coalsonaa255362001-05-31 06:17:41 +00001378 do {
Josh Coalsond86e03b2002-08-03 21:56:15 +00001379 if(encoder->protected_->verify)
1380 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1381
Josh Coalsonfa697a92001-08-16 20:07:29 +00001382 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
Josh Coalson57ba6f42002-06-07 05:27:37 +00001383 x = mid = side = buffer[0][j];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001384 encoder->private_->integer_signal[0][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001385#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001386 encoder->private_->real_signal[0][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001387#endif
Josh Coalson57ba6f42002-06-07 05:27:37 +00001388 x = buffer[1][j];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001389 encoder->private_->integer_signal[1][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001390#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001391 encoder->private_->real_signal[1][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001392#endif
Josh Coalsonaa255362001-05-31 06:17:41 +00001393 mid += x;
1394 side -= x;
Josh Coalson57ba6f42002-06-07 05:27:37 +00001395 mid >>= 1; /* NOTE: not the same as 'mid = (buffer[0][j] + buffer[1][j]) / 2' ! */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001396 encoder->private_->integer_signal_mid_side[1][i] = side;
1397 encoder->private_->integer_signal_mid_side[0][i] = mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001398#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001399 encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1400 encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001401#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001402 encoder->private_->current_sample_number++;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001403 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001404 if(i == blocksize) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001405 if(!process_frame_(encoder, false)) /* false => not last frame */
Josh Coalsonaa255362001-05-31 06:17:41 +00001406 return false;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001407 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001408 } while(j < samples);
1409 }
1410 else {
1411 do {
Josh Coalsond86e03b2002-08-03 21:56:15 +00001412 if(encoder->protected_->verify)
1413 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1414
Josh Coalsonfa697a92001-08-16 20:07:29 +00001415 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
Josh Coalsonaa255362001-05-31 06:17:41 +00001416 for(channel = 0; channel < channels; channel++) {
Josh Coalson57ba6f42002-06-07 05:27:37 +00001417 x = buffer[channel][j];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001418 encoder->private_->integer_signal[channel][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001419#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001420 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001421#endif
Josh Coalsonaa255362001-05-31 06:17:41 +00001422 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001423 encoder->private_->current_sample_number++;
Josh Coalsonaa255362001-05-31 06:17:41 +00001424 }
1425 if(i == blocksize) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001426 if(!process_frame_(encoder, false)) /* false => not last frame */
Josh Coalsonaa255362001-05-31 06:17:41 +00001427 return false;
1428 }
1429 } while(j < samples);
1430 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001431
1432 return true;
1433}
1434
Josh Coalson6afed9f2002-10-16 22:29:47 +00001435FLAC_API FLAC__bool FLAC__stream_encoder_process_interleaved(FLAC__StreamEncoder *encoder, const FLAC__int32 buffer[], unsigned samples)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001436{
1437 unsigned i, j, k, channel;
Josh Coalson77e3f312001-06-23 03:03:24 +00001438 FLAC__int32 x, mid, side;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001439 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001440
Josh Coalsonf1eff452002-07-31 07:05:33 +00001441 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001442 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001443
1444 j = k = 0;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001445 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
Josh Coalsonaa255362001-05-31 06:17:41 +00001446 do {
Josh Coalsond86e03b2002-08-03 21:56:15 +00001447 if(encoder->protected_->verify)
1448 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1449
Josh Coalsonfa697a92001-08-16 20:07:29 +00001450 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
Josh Coalson57ba6f42002-06-07 05:27:37 +00001451 x = mid = side = buffer[k++];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001452 encoder->private_->integer_signal[0][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001453#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001454 encoder->private_->real_signal[0][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001455#endif
Josh Coalson57ba6f42002-06-07 05:27:37 +00001456 x = buffer[k++];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001457 encoder->private_->integer_signal[1][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001458#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001459 encoder->private_->real_signal[1][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001460#endif
Josh Coalsonaa255362001-05-31 06:17:41 +00001461 mid += x;
1462 side -= x;
1463 mid >>= 1; /* NOTE: not the same as 'mid = (left + right) / 2' ! */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001464 encoder->private_->integer_signal_mid_side[1][i] = side;
1465 encoder->private_->integer_signal_mid_side[0][i] = mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001466#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001467 encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1468 encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001469#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001470 encoder->private_->current_sample_number++;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001471 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001472 if(i == blocksize) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001473 if(!process_frame_(encoder, false)) /* false => not last frame */
Josh Coalsonaa255362001-05-31 06:17:41 +00001474 return false;
1475 }
1476 } while(j < samples);
1477 }
1478 else {
1479 do {
Josh Coalsond86e03b2002-08-03 21:56:15 +00001480 if(encoder->protected_->verify)
1481 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1482
Josh Coalsonfa697a92001-08-16 20:07:29 +00001483 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
Josh Coalsonaa255362001-05-31 06:17:41 +00001484 for(channel = 0; channel < channels; channel++) {
Josh Coalson57ba6f42002-06-07 05:27:37 +00001485 x = buffer[k++];
Josh Coalsonfa697a92001-08-16 20:07:29 +00001486 encoder->private_->integer_signal[channel][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001487#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001488 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001489#endif
Josh Coalsonaa255362001-05-31 06:17:41 +00001490 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001491 encoder->private_->current_sample_number++;
Josh Coalsonaa255362001-05-31 06:17:41 +00001492 }
1493 if(i == blocksize) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001494 if(!process_frame_(encoder, false)) /* false => not last frame */
Josh Coalsonaa255362001-05-31 06:17:41 +00001495 return false;
1496 }
1497 } while(j < samples);
1498 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001499
1500 return true;
1501}
1502
Josh Coalsonf1eff452002-07-31 07:05:33 +00001503/***********************************************************************
1504 *
1505 * Private class methods
1506 *
1507 ***********************************************************************/
1508
1509void set_defaults_(FLAC__StreamEncoder *encoder)
Josh Coalson92031602002-07-24 06:02:11 +00001510{
1511 FLAC__ASSERT(0 != encoder);
1512
Josh Coalsond86e03b2002-08-03 21:56:15 +00001513 encoder->protected_->verify = false;
Josh Coalson92031602002-07-24 06:02:11 +00001514 encoder->protected_->streamable_subset = true;
1515 encoder->protected_->do_mid_side_stereo = false;
1516 encoder->protected_->loose_mid_side_stereo = false;
1517 encoder->protected_->channels = 2;
1518 encoder->protected_->bits_per_sample = 16;
1519 encoder->protected_->sample_rate = 44100;
1520 encoder->protected_->blocksize = 1152;
1521 encoder->protected_->max_lpc_order = 0;
1522 encoder->protected_->qlp_coeff_precision = 0;
1523 encoder->protected_->do_qlp_coeff_prec_search = false;
1524 encoder->protected_->do_exhaustive_model_search = false;
1525 encoder->protected_->do_escape_coding = false;
1526 encoder->protected_->min_residual_partition_order = 0;
1527 encoder->protected_->max_residual_partition_order = 0;
1528 encoder->protected_->rice_parameter_search_dist = 0;
1529 encoder->protected_->total_samples_estimate = 0;
1530 encoder->protected_->metadata = 0;
1531 encoder->protected_->num_metadata_blocks = 0;
1532
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001533 encoder->private_->disable_constant_subframes = false;
1534 encoder->private_->disable_fixed_subframes = false;
1535 encoder->private_->disable_verbatim_subframes = false;
Josh Coalson92031602002-07-24 06:02:11 +00001536 encoder->private_->write_callback = 0;
1537 encoder->private_->metadata_callback = 0;
1538 encoder->private_->client_data = 0;
1539}
1540
Josh Coalsonf1eff452002-07-31 07:05:33 +00001541void free_(FLAC__StreamEncoder *encoder)
Josh Coalson639aeb02002-07-25 05:38:23 +00001542{
1543 unsigned i, channel;
1544
Josh Coalsonf1eff452002-07-31 07:05:33 +00001545 FLAC__ASSERT(0 != encoder);
Josh Coalson639aeb02002-07-25 05:38:23 +00001546 for(i = 0; i < encoder->protected_->channels; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001547 if(0 != encoder->private_->integer_signal_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001548 free(encoder->private_->integer_signal_unaligned[i]);
1549 encoder->private_->integer_signal_unaligned[i] = 0;
1550 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001551#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonf1eff452002-07-31 07:05:33 +00001552 if(0 != encoder->private_->real_signal_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001553 free(encoder->private_->real_signal_unaligned[i]);
1554 encoder->private_->real_signal_unaligned[i] = 0;
1555 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001556#endif
Josh Coalson639aeb02002-07-25 05:38:23 +00001557 }
1558 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001559 if(0 != encoder->private_->integer_signal_mid_side_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001560 free(encoder->private_->integer_signal_mid_side_unaligned[i]);
1561 encoder->private_->integer_signal_mid_side_unaligned[i] = 0;
1562 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001563#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonf1eff452002-07-31 07:05:33 +00001564 if(0 != encoder->private_->real_signal_mid_side_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001565 free(encoder->private_->real_signal_mid_side_unaligned[i]);
1566 encoder->private_->real_signal_mid_side_unaligned[i] = 0;
1567 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001568#endif
Josh Coalson639aeb02002-07-25 05:38:23 +00001569 }
1570 for(channel = 0; channel < encoder->protected_->channels; channel++) {
1571 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001572 if(0 != encoder->private_->residual_workspace_unaligned[channel][i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001573 free(encoder->private_->residual_workspace_unaligned[channel][i]);
1574 encoder->private_->residual_workspace_unaligned[channel][i] = 0;
1575 }
1576 }
1577 }
1578 for(channel = 0; channel < 2; channel++) {
1579 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001580 if(0 != encoder->private_->residual_workspace_mid_side_unaligned[channel][i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001581 free(encoder->private_->residual_workspace_mid_side_unaligned[channel][i]);
1582 encoder->private_->residual_workspace_mid_side_unaligned[channel][i] = 0;
1583 }
1584 }
1585 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001586 if(0 != encoder->private_->abs_residual_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001587 free(encoder->private_->abs_residual_unaligned);
1588 encoder->private_->abs_residual_unaligned = 0;
1589 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001590 if(0 != encoder->private_->abs_residual_partition_sums_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001591 free(encoder->private_->abs_residual_partition_sums_unaligned);
1592 encoder->private_->abs_residual_partition_sums_unaligned = 0;
1593 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001594 if(0 != encoder->private_->raw_bits_per_partition_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001595 free(encoder->private_->raw_bits_per_partition_unaligned);
1596 encoder->private_->raw_bits_per_partition_unaligned = 0;
1597 }
Josh Coalsond86e03b2002-08-03 21:56:15 +00001598 if(encoder->protected_->verify) {
1599 for(i = 0; i < encoder->protected_->channels; i++) {
1600 if(0 != encoder->private_->verify.input_fifo.data[i]) {
1601 free(encoder->private_->verify.input_fifo.data[i]);
1602 encoder->private_->verify.input_fifo.data[i] = 0;
1603 }
1604 }
1605 }
Josh Coalson639aeb02002-07-25 05:38:23 +00001606 FLAC__bitbuffer_free(encoder->private_->frame);
1607}
1608
Josh Coalsonf1eff452002-07-31 07:05:33 +00001609FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001610{
Josh Coalson77e3f312001-06-23 03:03:24 +00001611 FLAC__bool ok;
Josh Coalson0a15c142001-06-13 17:59:57 +00001612 unsigned i, channel;
1613
1614 FLAC__ASSERT(new_size > 0);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001615 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1616 FLAC__ASSERT(encoder->private_->current_sample_number == 0);
Josh Coalson0a15c142001-06-13 17:59:57 +00001617
1618 /* To avoid excessive malloc'ing, we only grow the buffer; no shrinking. */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001619 if(new_size <= encoder->private_->input_capacity)
Josh Coalson0a15c142001-06-13 17:59:57 +00001620 return true;
1621
1622 ok = true;
Josh Coalson8395d022001-07-12 21:25:22 +00001623
Josh Coalsonc9c0d132002-10-04 05:29:05 +00001624 /* WATCHOUT: FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32_mmx()
1625 * requires that the input arrays (in our case the integer signals)
1626 * have a buffer of up to 3 zeroes in front (at negative indices) for
1627 * alignment purposes; we use 4 to keep the data well-aligned.
1628 */
Josh Coalson8395d022001-07-12 21:25:22 +00001629
Josh Coalsonfa697a92001-08-16 20:07:29 +00001630 for(i = 0; ok && i < encoder->protected_->channels; i++) {
1631 ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size+4, &encoder->private_->integer_signal_unaligned[i], &encoder->private_->integer_signal[i]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001632#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001633 ok = ok && FLAC__memory_alloc_aligned_real_array(new_size, &encoder->private_->real_signal_unaligned[i], &encoder->private_->real_signal[i]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001634#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001635 memset(encoder->private_->integer_signal[i], 0, sizeof(FLAC__int32)*4);
1636 encoder->private_->integer_signal[i] += 4;
Josh Coalson0a15c142001-06-13 17:59:57 +00001637 }
1638 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001639 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]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001640#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonfa697a92001-08-16 20:07:29 +00001641 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]);
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001642#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001643 memset(encoder->private_->integer_signal_mid_side[i], 0, sizeof(FLAC__int32)*4);
1644 encoder->private_->integer_signal_mid_side[i] += 4;
Josh Coalson0a15c142001-06-13 17:59:57 +00001645 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001646 for(channel = 0; ok && channel < encoder->protected_->channels; channel++) {
Josh Coalson0a15c142001-06-13 17:59:57 +00001647 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001648 ok = ok && FLAC__memory_alloc_aligned_int32_array(new_size, &encoder->private_->residual_workspace_unaligned[channel][i], &encoder->private_->residual_workspace[channel][i]);
Josh Coalson0a15c142001-06-13 17:59:57 +00001649 }
1650 }
1651 for(channel = 0; ok && channel < 2; channel++) {
1652 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001653 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]);
Josh Coalson0a15c142001-06-13 17:59:57 +00001654 }
1655 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001656 ok = ok && FLAC__memory_alloc_aligned_uint32_array(new_size, &encoder->private_->abs_residual_unaligned, &encoder->private_->abs_residual);
1657 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 */
1658 ok = ok && FLAC__memory_alloc_aligned_uint64_array(new_size * 2, &encoder->private_->abs_residual_partition_sums_unaligned, &encoder->private_->abs_residual_partition_sums);
1659 if(encoder->protected_->do_escape_coding)
1660 ok = ok && FLAC__memory_alloc_aligned_unsigned_array(new_size * 2, &encoder->private_->raw_bits_per_partition_unaligned, &encoder->private_->raw_bits_per_partition);
Josh Coalson0a15c142001-06-13 17:59:57 +00001661
1662 if(ok)
Josh Coalsonfa697a92001-08-16 20:07:29 +00001663 encoder->private_->input_capacity = new_size;
Josh Coalson0a15c142001-06-13 17:59:57 +00001664 else
Josh Coalsonfa697a92001-08-16 20:07:29 +00001665 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalson0a15c142001-06-13 17:59:57 +00001666
1667 return ok;
1668}
1669
Josh Coalsond86e03b2002-08-03 21:56:15 +00001670FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples)
Josh Coalson5c491a12002-08-01 06:39:40 +00001671{
1672 const FLAC__byte *buffer;
1673 unsigned bytes;
1674
1675 FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1676
1677 FLAC__bitbuffer_get_buffer(encoder->private_->frame, &buffer, &bytes);
1678
Josh Coalsond86e03b2002-08-03 21:56:15 +00001679 if(encoder->protected_->verify) {
1680 encoder->private_->verify.output.data = buffer;
1681 encoder->private_->verify.output.bytes = bytes;
1682 if(encoder->private_->verify.state_hint == ENCODER_IN_MAGIC) {
1683 encoder->private_->verify.needs_magic_hack = true;
1684 }
1685 else {
1686 if(!FLAC__stream_decoder_process_single(encoder->private_->verify.decoder)) {
1687 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1688 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA)
1689 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
1690 return false;
1691 }
1692 }
1693 }
1694
1695 if(encoder->private_->write_callback(encoder, buffer, bytes, samples, encoder->private_->current_frame_number, encoder->private_->client_data) != FLAC__STREAM_ENCODER_WRITE_STATUS_OK) {
Josh Coalsondd190232002-12-29 09:30:23 +00001696 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
Josh Coalsond86e03b2002-08-03 21:56:15 +00001697 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_WRITING;
Josh Coalson5c491a12002-08-01 06:39:40 +00001698 return false;
Josh Coalsond86e03b2002-08-03 21:56:15 +00001699 }
Josh Coalson5c491a12002-08-01 06:39:40 +00001700
1701 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1702
Josh Coalsond86e03b2002-08-03 21:56:15 +00001703 if(samples > 0) {
1704 encoder->private_->metadata.data.stream_info.min_framesize = min(bytes, encoder->private_->metadata.data.stream_info.min_framesize);
1705 encoder->private_->metadata.data.stream_info.max_framesize = max(bytes, encoder->private_->metadata.data.stream_info.max_framesize);
1706 }
1707
Josh Coalson5c491a12002-08-01 06:39:40 +00001708 return true;
1709}
1710
Josh Coalsonf1eff452002-07-31 07:05:33 +00001711FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
Josh Coalson0a15c142001-06-13 17:59:57 +00001712{
Josh Coalsonfa697a92001-08-16 20:07:29 +00001713 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001714
1715 /*
Josh Coalsonfa37f1c2001-01-12 23:55:11 +00001716 * Accumulate raw signal to the MD5 signature
1717 */
Josh Coalson57ba6f42002-06-07 05:27:37 +00001718 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)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001719 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonfa37f1c2001-01-12 23:55:11 +00001720 return false;
1721 }
1722
1723 /*
Josh Coalson94e02cd2001-01-25 10:41:06 +00001724 * Process the frame header and subframes into the frame bitbuffer
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001725 */
Josh Coalsonf1eff452002-07-31 07:05:33 +00001726 if(!process_subframes_(encoder, is_last_frame)) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00001727 /* the above function sets the state for us in case of an error */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001728 return false;
1729 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001730
1731 /*
1732 * Zero-pad the frame to a byte_boundary
1733 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001734 if(!FLAC__bitbuffer_zero_pad_to_byte_boundary(encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001735 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001736 return false;
1737 }
1738
1739 /*
Josh Coalson215af572001-03-27 01:15:58 +00001740 * CRC-16 the whole thing
1741 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001742 FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1743 FLAC__bitbuffer_write_raw_uint32(encoder->private_->frame, FLAC__bitbuffer_get_write_crc16(encoder->private_->frame), FLAC__FRAME_FOOTER_CRC_LEN);
Josh Coalson215af572001-03-27 01:15:58 +00001744
1745 /*
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001746 * Write it
1747 */
Josh Coalsond86e03b2002-08-03 21:56:15 +00001748 if(!write_bitbuffer_(encoder, encoder->protected_->blocksize)) {
1749 /* the above function sets the state for us in case of an error */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001750 return false;
1751 }
1752
1753 /*
1754 * Get ready for the next frame
1755 */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001756 encoder->private_->current_sample_number = 0;
1757 encoder->private_->current_frame_number++;
1758 encoder->private_->metadata.data.stream_info.total_samples += (FLAC__uint64)encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001759
1760 return true;
1761}
1762
Josh Coalsonf1eff452002-07-31 07:05:33 +00001763FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
Josh Coalson94e02cd2001-01-25 10:41:06 +00001764{
1765 FLAC__FrameHeader frame_header;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001766 unsigned channel, min_partition_order = encoder->protected_->min_residual_partition_order, max_partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00001767 FLAC__bool do_independent, do_mid_side, precompute_partition_sums;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001768
1769 /*
Josh Coalson60f77d72001-04-25 02:16:36 +00001770 * Calculate the min,max Rice partition orders
Josh Coalson94e02cd2001-01-25 10:41:06 +00001771 */
1772 if(is_last_frame) {
1773 max_partition_order = 0;
1774 }
1775 else {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001776 max_partition_order = FLAC__format_get_max_rice_partition_order_from_blocksize(encoder->protected_->blocksize);
1777 max_partition_order = min(max_partition_order, encoder->protected_->max_residual_partition_order);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001778 }
Josh Coalson60f77d72001-04-25 02:16:36 +00001779 min_partition_order = min(min_partition_order, max_partition_order);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001780
Josh Coalsonfa697a92001-08-16 20:07:29 +00001781 precompute_partition_sums = encoder->private_->precompute_partition_sums && ((max_partition_order > min_partition_order) || encoder->protected_->do_escape_coding);
Josh Coalson8395d022001-07-12 21:25:22 +00001782
Josh Coalson94e02cd2001-01-25 10:41:06 +00001783 /*
1784 * Setup the frame
1785 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001786 if(!FLAC__bitbuffer_clear(encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001787 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001788 return false;
1789 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001790 frame_header.blocksize = encoder->protected_->blocksize;
1791 frame_header.sample_rate = encoder->protected_->sample_rate;
1792 frame_header.channels = encoder->protected_->channels;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001793 frame_header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT; /* the default unless the encoder determines otherwise */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001794 frame_header.bits_per_sample = encoder->protected_->bits_per_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00001795 frame_header.number_type = FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001796 frame_header.number.frame_number = encoder->private_->current_frame_number;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001797
1798 /*
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001799 * Figure out what channel assignments to try
1800 */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001801 if(encoder->protected_->do_mid_side_stereo) {
1802 if(encoder->protected_->loose_mid_side_stereo) {
1803 if(encoder->private_->loose_mid_side_stereo_frame_count == 0) {
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001804 do_independent = true;
1805 do_mid_side = true;
1806 }
1807 else {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001808 do_independent = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001809 do_mid_side = !do_independent;
1810 }
1811 }
1812 else {
1813 do_independent = true;
1814 do_mid_side = true;
1815 }
1816 }
1817 else {
1818 do_independent = true;
1819 do_mid_side = false;
1820 }
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001821
Josh Coalson1b689822001-05-31 20:11:02 +00001822 FLAC__ASSERT(do_independent || do_mid_side);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001823
1824 /*
Josh Coalson82b73242001-03-28 22:17:05 +00001825 * Check for wasted bits; set effective bps for each subframe
Josh Coalson859bc542001-03-27 22:22:27 +00001826 */
1827 if(do_independent) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001828 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001829 const unsigned w = get_wasted_bits_(encoder->private_->integer_signal[channel], encoder->protected_->blocksize);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001830 encoder->private_->subframe_workspace[channel][0].wasted_bits = encoder->private_->subframe_workspace[channel][1].wasted_bits = w;
1831 encoder->private_->subframe_bps[channel] = encoder->protected_->bits_per_sample - w;
Josh Coalson82b73242001-03-28 22:17:05 +00001832 }
Josh Coalson859bc542001-03-27 22:22:27 +00001833 }
1834 if(do_mid_side) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001835 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson82b73242001-03-28 22:17:05 +00001836 for(channel = 0; channel < 2; channel++) {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001837 const unsigned w = get_wasted_bits_(encoder->private_->integer_signal_mid_side[channel], encoder->protected_->blocksize);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001838 encoder->private_->subframe_workspace_mid_side[channel][0].wasted_bits = encoder->private_->subframe_workspace_mid_side[channel][1].wasted_bits = w;
1839 encoder->private_->subframe_bps_mid_side[channel] = encoder->protected_->bits_per_sample - w + (channel==0? 0:1);
Josh Coalson82b73242001-03-28 22:17:05 +00001840 }
Josh Coalson859bc542001-03-27 22:22:27 +00001841 }
1842
1843 /*
Josh Coalson94e02cd2001-01-25 10:41:06 +00001844 * First do a normal encoding pass of each independent channel
1845 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001846 if(do_independent) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001847 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalson6fe72f72002-08-20 04:01:59 +00001848 if(!
1849 process_subframe_(
1850 encoder,
1851 min_partition_order,
1852 max_partition_order,
1853 precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00001854 &frame_header,
1855 encoder->private_->subframe_bps[channel],
1856 encoder->private_->integer_signal[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001857#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00001858 encoder->private_->real_signal[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001859#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00001860 encoder->private_->subframe_workspace_ptr[channel],
1861 encoder->private_->partitioned_rice_contents_workspace_ptr[channel],
1862 encoder->private_->residual_workspace[channel],
1863 encoder->private_->best_subframe+channel,
1864 encoder->private_->best_subframe_bits+channel
1865 )
1866 )
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001867 return false;
1868 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00001869 }
1870
1871 /*
1872 * Now do mid and side channels if requested
1873 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001874 if(do_mid_side) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001875 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001876
1877 for(channel = 0; channel < 2; channel++) {
Josh Coalson6fe72f72002-08-20 04:01:59 +00001878 if(!
1879 process_subframe_(
1880 encoder,
1881 min_partition_order,
1882 max_partition_order,
1883 precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00001884 &frame_header,
1885 encoder->private_->subframe_bps_mid_side[channel],
1886 encoder->private_->integer_signal_mid_side[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001887#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00001888 encoder->private_->real_signal_mid_side[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001889#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00001890 encoder->private_->subframe_workspace_ptr_mid_side[channel],
1891 encoder->private_->partitioned_rice_contents_workspace_ptr_mid_side[channel],
1892 encoder->private_->residual_workspace_mid_side[channel],
1893 encoder->private_->best_subframe_mid_side+channel,
1894 encoder->private_->best_subframe_bits_mid_side+channel
1895 )
1896 )
Josh Coalson94e02cd2001-01-25 10:41:06 +00001897 return false;
1898 }
1899 }
1900
1901 /*
1902 * Compose the frame bitbuffer
1903 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001904 if(do_mid_side) {
Josh Coalson82b73242001-03-28 22:17:05 +00001905 unsigned left_bps = 0, right_bps = 0; /* initialized only to prevent superfluous compiler warning */
1906 FLAC__Subframe *left_subframe = 0, *right_subframe = 0; /* initialized only to prevent superfluous compiler warning */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001907 FLAC__ChannelAssignment channel_assignment;
1908
Josh Coalsonfa697a92001-08-16 20:07:29 +00001909 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001910
Josh Coalsonfa697a92001-08-16 20:07:29 +00001911 if(encoder->protected_->loose_mid_side_stereo && encoder->private_->loose_mid_side_stereo_frame_count > 0) {
1912 channel_assignment = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT? FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT : FLAC__CHANNEL_ASSIGNMENT_MID_SIDE);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001913 }
1914 else {
1915 unsigned bits[4]; /* WATCHOUT - indexed by FLAC__ChannelAssignment */
1916 unsigned min_bits;
1917 FLAC__ChannelAssignment ca;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001918
Josh Coalson1b689822001-05-31 20:11:02 +00001919 FLAC__ASSERT(do_independent && do_mid_side);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001920
1921 /* We have to figure out which channel assignent results in the smallest frame */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001922 bits[FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT] = encoder->private_->best_subframe_bits [0] + encoder->private_->best_subframe_bits [1];
1923 bits[FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE ] = encoder->private_->best_subframe_bits [0] + encoder->private_->best_subframe_bits_mid_side[1];
1924 bits[FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE ] = encoder->private_->best_subframe_bits [1] + encoder->private_->best_subframe_bits_mid_side[1];
1925 bits[FLAC__CHANNEL_ASSIGNMENT_MID_SIDE ] = encoder->private_->best_subframe_bits_mid_side[0] + encoder->private_->best_subframe_bits_mid_side[1];
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001926
Josh Coalson7424d2f2002-11-06 07:10:38 +00001927 for(channel_assignment = (FLAC__ChannelAssignment)0, min_bits = bits[0], ca = (FLAC__ChannelAssignment)1; (int)ca <= 3; ca = (FLAC__ChannelAssignment)((int)ca + 1)) {
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001928 if(bits[ca] < min_bits) {
1929 min_bits = bits[ca];
1930 channel_assignment = ca;
1931 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00001932 }
1933 }
1934
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001935 frame_header.channel_assignment = channel_assignment;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001936
Josh Coalson3e7a96e2004-07-23 05:18:22 +00001937 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001938 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001939 return false;
1940 }
1941
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001942 switch(channel_assignment) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00001943 case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001944 left_subframe = &encoder->private_->subframe_workspace [0][encoder->private_->best_subframe [0]];
1945 right_subframe = &encoder->private_->subframe_workspace [1][encoder->private_->best_subframe [1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00001946 break;
1947 case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001948 left_subframe = &encoder->private_->subframe_workspace [0][encoder->private_->best_subframe [0]];
1949 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00001950 break;
1951 case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001952 left_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
1953 right_subframe = &encoder->private_->subframe_workspace [1][encoder->private_->best_subframe [1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00001954 break;
1955 case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001956 left_subframe = &encoder->private_->subframe_workspace_mid_side[0][encoder->private_->best_subframe_mid_side[0]];
1957 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00001958 break;
1959 default:
Josh Coalson1b689822001-05-31 20:11:02 +00001960 FLAC__ASSERT(0);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001961 }
Josh Coalson82b73242001-03-28 22:17:05 +00001962
1963 switch(channel_assignment) {
1964 case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001965 left_bps = encoder->private_->subframe_bps [0];
1966 right_bps = encoder->private_->subframe_bps [1];
Josh Coalson82b73242001-03-28 22:17:05 +00001967 break;
1968 case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001969 left_bps = encoder->private_->subframe_bps [0];
1970 right_bps = encoder->private_->subframe_bps_mid_side[1];
Josh Coalson82b73242001-03-28 22:17:05 +00001971 break;
1972 case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001973 left_bps = encoder->private_->subframe_bps_mid_side[1];
1974 right_bps = encoder->private_->subframe_bps [1];
Josh Coalson82b73242001-03-28 22:17:05 +00001975 break;
1976 case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00001977 left_bps = encoder->private_->subframe_bps_mid_side[0];
1978 right_bps = encoder->private_->subframe_bps_mid_side[1];
Josh Coalson82b73242001-03-28 22:17:05 +00001979 break;
1980 default:
Josh Coalson1b689822001-05-31 20:11:02 +00001981 FLAC__ASSERT(0);
Josh Coalson82b73242001-03-28 22:17:05 +00001982 }
1983
1984 /* note that encoder_add_subframe_ sets the state for us in case of an error */
Josh Coalsonf1eff452002-07-31 07:05:33 +00001985 if(!add_subframe_(encoder, &frame_header, left_bps , left_subframe , encoder->private_->frame))
Josh Coalson82b73242001-03-28 22:17:05 +00001986 return false;
Josh Coalsonf1eff452002-07-31 07:05:33 +00001987 if(!add_subframe_(encoder, &frame_header, right_bps, right_subframe, encoder->private_->frame))
Josh Coalson82b73242001-03-28 22:17:05 +00001988 return false;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001989 }
1990 else {
Josh Coalson3e7a96e2004-07-23 05:18:22 +00001991 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001992 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001993 return false;
1994 }
1995
Josh Coalsonfa697a92001-08-16 20:07:29 +00001996 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001997 if(!add_subframe_(encoder, &frame_header, encoder->private_->subframe_bps[channel], &encoder->private_->subframe_workspace[channel][encoder->private_->best_subframe[channel]], encoder->private_->frame)) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00001998 /* the above function sets the state for us in case of an error */
1999 return false;
2000 }
2001 }
2002 }
2003
Josh Coalsonfa697a92001-08-16 20:07:29 +00002004 if(encoder->protected_->loose_mid_side_stereo) {
2005 encoder->private_->loose_mid_side_stereo_frame_count++;
2006 if(encoder->private_->loose_mid_side_stereo_frame_count >= encoder->private_->loose_mid_side_stereo_frames)
2007 encoder->private_->loose_mid_side_stereo_frame_count = 0;
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002008 }
2009
Josh Coalsonfa697a92001-08-16 20:07:29 +00002010 encoder->private_->last_channel_assignment = frame_header.channel_assignment;
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002011
Josh Coalson94e02cd2001-01-25 10:41:06 +00002012 return true;
2013}
2014
Josh Coalson6fe72f72002-08-20 04:01:59 +00002015FLAC__bool process_subframe_(
2016 FLAC__StreamEncoder *encoder,
2017 unsigned min_partition_order,
2018 unsigned max_partition_order,
2019 FLAC__bool precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00002020 const FLAC__FrameHeader *frame_header,
2021 unsigned subframe_bps,
2022 const FLAC__int32 integer_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002023#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00002024 const FLAC__real real_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002025#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00002026 FLAC__Subframe *subframe[2],
2027 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
2028 FLAC__int32 *residual[2],
2029 unsigned *best_subframe,
2030 unsigned *best_bits
2031)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002032{
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002033#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002034 FLAC__float fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002035#else
2036 FLAC__fixedpoint fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
2037#endif
2038#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002039 FLAC__double lpc_residual_bits_per_sample;
Josh Coalsonfa697a92001-08-16 20:07:29 +00002040 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 */
Josh Coalson09758432004-10-20 00:21:50 +00002041 FLAC__double lpc_error[FLAC__MAX_LPC_ORDER];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002042 unsigned min_lpc_order, max_lpc_order, lpc_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002043 unsigned min_qlp_coeff_precision, max_qlp_coeff_precision, qlp_coeff_precision;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002044#endif
2045 unsigned min_fixed_order, max_fixed_order, guess_fixed_order, fixed_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002046 unsigned rice_parameter;
2047 unsigned _candidate_bits, _best_bits;
2048 unsigned _best_subframe;
2049
2050 /* verbatim subframe is the baseline against which we measure other compressed subframes */
2051 _best_subframe = 0;
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002052 if(encoder->private_->disable_verbatim_subframes && frame_header->blocksize >= FLAC__MAX_FIXED_ORDER)
2053 _best_bits = UINT_MAX;
2054 else
2055 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002056
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002057 if(frame_header->blocksize >= FLAC__MAX_FIXED_ORDER) {
2058 unsigned signal_is_constant = false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00002059 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);
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002060 /* check for constant subframe */
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002061 if(
2062 !encoder->private_->disable_constant_subframes &&
2063#ifndef FLAC__INTEGER_ONLY_LIBRARY
2064 fixed_residual_bits_per_sample[1] == 0.0
2065#else
2066 fixed_residual_bits_per_sample[1] == FLAC__FP_ZERO
2067#endif
2068 ) {
2069 /* the above means it's possible all samples are the same value; now double-check it: */
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002070 unsigned i;
2071 signal_is_constant = true;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002072 for(i = 1; i < frame_header->blocksize; i++) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00002073 if(integer_signal[0] != integer_signal[i]) {
2074 signal_is_constant = false;
2075 break;
2076 }
2077 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002078 }
2079 if(signal_is_constant) {
2080 _candidate_bits = evaluate_constant_subframe_(integer_signal[0], subframe_bps, subframe[!_best_subframe]);
2081 if(_candidate_bits < _best_bits) {
2082 _best_subframe = !_best_subframe;
2083 _best_bits = _candidate_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002084 }
2085 }
2086 else {
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002087 if(!encoder->private_->disable_fixed_subframes || (encoder->protected_->max_lpc_order == 0 && _best_bits == UINT_MAX)) {
2088 /* encode fixed */
2089 if(encoder->protected_->do_exhaustive_model_search) {
2090 min_fixed_order = 0;
2091 max_fixed_order = FLAC__MAX_FIXED_ORDER;
Josh Coalson8395d022001-07-12 21:25:22 +00002092 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002093 else {
2094 min_fixed_order = max_fixed_order = guess_fixed_order;
2095 }
2096 for(fixed_order = min_fixed_order; fixed_order <= max_fixed_order; fixed_order++) {
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002097#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002098 if(fixed_residual_bits_per_sample[fixed_order] >= (FLAC__float)subframe_bps)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002099 continue; /* don't even try */
2100 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 */
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002101#else
2102 if(FLAC__fixedpoint_trunc(fixed_residual_bits_per_sample[fixed_order]) >= (int)subframe_bps)
2103 continue; /* don't even try */
2104 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 */
2105#endif
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002106#ifndef FLAC__SYMMETRIC_RICE
2107 rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
2108#endif
2109 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2110#ifdef DEBUG_VERBOSE
2111 fprintf(stderr, "clipping rice_parameter (%u -> %u) @0\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2112#endif
2113 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2114 }
2115 _candidate_bits =
2116 evaluate_fixed_subframe_(
2117 encoder,
2118 integer_signal,
2119 residual[!_best_subframe],
2120 encoder->private_->abs_residual,
2121 encoder->private_->abs_residual_partition_sums,
2122 encoder->private_->raw_bits_per_partition,
2123 frame_header->blocksize,
2124 subframe_bps,
2125 fixed_order,
2126 rice_parameter,
2127 min_partition_order,
2128 max_partition_order,
2129 precompute_partition_sums,
2130 encoder->protected_->do_escape_coding,
2131 encoder->protected_->rice_parameter_search_dist,
2132 subframe[!_best_subframe],
2133 partitioned_rice_contents[!_best_subframe]
2134 );
2135 if(_candidate_bits < _best_bits) {
2136 _best_subframe = !_best_subframe;
2137 _best_bits = _candidate_bits;
2138 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002139 }
2140 }
2141
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002142#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson94e02cd2001-01-25 10:41:06 +00002143 /* encode lpc */
Josh Coalsonfa697a92001-08-16 20:07:29 +00002144 if(encoder->protected_->max_lpc_order > 0) {
2145 if(encoder->protected_->max_lpc_order >= frame_header->blocksize)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002146 max_lpc_order = frame_header->blocksize-1;
2147 else
Josh Coalsonfa697a92001-08-16 20:07:29 +00002148 max_lpc_order = encoder->protected_->max_lpc_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002149 if(max_lpc_order > 0) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002150 encoder->private_->local_lpc_compute_autocorrelation(real_signal, frame_header->blocksize, max_lpc_order+1, autoc);
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002151 /* if autoc[0] == 0.0, the signal is constant and we usually won't get here, but it can happen */
2152 if(autoc[0] != 0.0) {
Josh Coalson8084b052001-11-01 00:27:29 +00002153 FLAC__lpc_compute_lp_coefficients(autoc, max_lpc_order, encoder->private_->lp_coeff, lpc_error);
Josh Coalsonfa697a92001-08-16 20:07:29 +00002154 if(encoder->protected_->do_exhaustive_model_search) {
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002155 min_lpc_order = 1;
2156 }
2157 else {
Josh Coalson82b73242001-03-28 22:17:05 +00002158 unsigned guess_lpc_order = FLAC__lpc_compute_best_order(lpc_error, max_lpc_order, frame_header->blocksize, subframe_bps);
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002159 min_lpc_order = max_lpc_order = guess_lpc_order;
2160 }
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002161 for(lpc_order = min_lpc_order; lpc_order <= max_lpc_order; lpc_order++) {
2162 lpc_residual_bits_per_sample = FLAC__lpc_compute_expected_bits_per_residual_sample(lpc_error[lpc_order-1], frame_header->blocksize-lpc_order);
Josh Coalson09758432004-10-20 00:21:50 +00002163 if(lpc_residual_bits_per_sample >= (FLAC__double)subframe_bps)
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002164 continue; /* don't even try */
2165 rice_parameter = (lpc_residual_bits_per_sample > 0.0)? (unsigned)(lpc_residual_bits_per_sample+0.5) : 0; /* 0.5 is for rounding */
Josh Coalsonbb6712e2001-04-24 22:54:07 +00002166#ifndef FLAC__SYMMETRIC_RICE
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002167 rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
Josh Coalsonb9433f92001-03-17 01:07:00 +00002168#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002169 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002170#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002171 fprintf(stderr, "clipping rice_parameter (%u -> %u) @1\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2172#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00002173 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00002174 }
Josh Coalsonc9c0d132002-10-04 05:29:05 +00002175 if(encoder->protected_->do_qlp_coeff_prec_search) {
2176 min_qlp_coeff_precision = FLAC__MIN_QLP_COEFF_PRECISION;
2177 /* ensure a 32-bit datapath throughout for 16bps or less */
2178 if(subframe_bps <= 16)
2179 max_qlp_coeff_precision = min(32 - subframe_bps - lpc_order, FLAC__MAX_QLP_COEFF_PRECISION);
2180 else
2181 max_qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
2182 }
2183 else {
2184 min_qlp_coeff_precision = max_qlp_coeff_precision = encoder->protected_->qlp_coeff_precision;
2185 }
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002186 for(qlp_coeff_precision = min_qlp_coeff_precision; qlp_coeff_precision <= max_qlp_coeff_precision; qlp_coeff_precision++) {
Josh Coalson6fe72f72002-08-20 04:01:59 +00002187 _candidate_bits =
2188 evaluate_lpc_subframe_(
2189 encoder,
2190 integer_signal,
2191 residual[!_best_subframe],
2192 encoder->private_->abs_residual,
2193 encoder->private_->abs_residual_partition_sums,
2194 encoder->private_->raw_bits_per_partition,
2195 encoder->private_->lp_coeff[lpc_order-1],
2196 frame_header->blocksize,
2197 subframe_bps,
2198 lpc_order,
2199 qlp_coeff_precision,
2200 rice_parameter,
2201 min_partition_order,
2202 max_partition_order,
2203 precompute_partition_sums,
2204 encoder->protected_->do_escape_coding,
2205 encoder->protected_->rice_parameter_search_dist,
2206 subframe[!_best_subframe],
2207 partitioned_rice_contents[!_best_subframe]
2208 );
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002209 if(_candidate_bits > 0) { /* if == 0, there was a problem quantizing the lpcoeffs */
2210 if(_candidate_bits < _best_bits) {
2211 _best_subframe = !_best_subframe;
2212 _best_bits = _candidate_bits;
2213 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002214 }
2215 }
2216 }
2217 }
2218 }
2219 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002220#endif /* !defined FLAC__INTEGER_ONLY_LIBRARY */
Josh Coalson94e02cd2001-01-25 10:41:06 +00002221 }
2222 }
2223
Josh Coalson72695802002-10-11 06:25:16 +00002224 /* under rare circumstances this can happen when all but lpc subframe types are disabled: */
2225 if(_best_bits == UINT_MAX) {
2226 FLAC__ASSERT(_best_subframe == 0);
2227 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
2228 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002229
Josh Coalson94e02cd2001-01-25 10:41:06 +00002230 *best_subframe = _best_subframe;
2231 *best_bits = _best_bits;
2232
2233 return true;
2234}
2235
Josh Coalson6fe72f72002-08-20 04:01:59 +00002236FLAC__bool add_subframe_(
2237 FLAC__StreamEncoder *encoder,
2238 const FLAC__FrameHeader *frame_header,
2239 unsigned subframe_bps,
2240 const FLAC__Subframe *subframe,
2241 FLAC__BitBuffer *frame
2242)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002243{
2244 switch(subframe->type) {
2245 case FLAC__SUBFRAME_TYPE_CONSTANT:
Josh Coalson82b73242001-03-28 22:17:05 +00002246 if(!FLAC__subframe_add_constant(&(subframe->data.constant), subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002247 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002248 return false;
2249 }
2250 break;
2251 case FLAC__SUBFRAME_TYPE_FIXED:
Josh Coalson82b73242001-03-28 22:17:05 +00002252 if(!FLAC__subframe_add_fixed(&(subframe->data.fixed), frame_header->blocksize - subframe->data.fixed.order, subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002253 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002254 return false;
2255 }
2256 break;
2257 case FLAC__SUBFRAME_TYPE_LPC:
Josh Coalson82b73242001-03-28 22:17:05 +00002258 if(!FLAC__subframe_add_lpc(&(subframe->data.lpc), frame_header->blocksize - subframe->data.lpc.order, subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002259 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002260 return false;
2261 }
2262 break;
2263 case FLAC__SUBFRAME_TYPE_VERBATIM:
Josh Coalson82b73242001-03-28 22:17:05 +00002264 if(!FLAC__subframe_add_verbatim(&(subframe->data.verbatim), frame_header->blocksize, subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002265 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002266 return false;
2267 }
2268 break;
2269 default:
Josh Coalson1b689822001-05-31 20:11:02 +00002270 FLAC__ASSERT(0);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002271 }
2272
2273 return true;
2274}
2275
Josh Coalson6fe72f72002-08-20 04:01:59 +00002276unsigned evaluate_constant_subframe_(
2277 const FLAC__int32 signal,
2278 unsigned subframe_bps,
2279 FLAC__Subframe *subframe
2280)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002281{
2282 subframe->type = FLAC__SUBFRAME_TYPE_CONSTANT;
2283 subframe->data.constant.value = signal;
2284
Josh Coalson82b73242001-03-28 22:17:05 +00002285 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + subframe_bps;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002286}
2287
Josh Coalson6fe72f72002-08-20 04:01:59 +00002288unsigned evaluate_fixed_subframe_(
2289 FLAC__StreamEncoder *encoder,
2290 const FLAC__int32 signal[],
2291 FLAC__int32 residual[],
2292 FLAC__uint32 abs_residual[],
2293 FLAC__uint64 abs_residual_partition_sums[],
2294 unsigned raw_bits_per_partition[],
2295 unsigned blocksize,
2296 unsigned subframe_bps,
2297 unsigned order,
2298 unsigned rice_parameter,
2299 unsigned min_partition_order,
2300 unsigned max_partition_order,
2301 FLAC__bool precompute_partition_sums,
2302 FLAC__bool do_escape_coding,
2303 unsigned rice_parameter_search_dist,
2304 FLAC__Subframe *subframe,
2305 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2306)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002307{
2308 unsigned i, residual_bits;
2309 const unsigned residual_samples = blocksize - order;
2310
2311 FLAC__fixed_compute_residual(signal+order, residual_samples, order, residual);
2312
2313 subframe->type = FLAC__SUBFRAME_TYPE_FIXED;
2314
2315 subframe->data.fixed.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
Josh Coalsona37ba462002-08-19 21:36:39 +00002316 subframe->data.fixed.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002317 subframe->data.fixed.residual = residual;
2318
Josh Coalson6fe72f72002-08-20 04:01:59 +00002319 residual_bits =
2320 find_best_partition_order_(
2321 encoder->private_,
2322 residual,
2323 abs_residual,
2324 abs_residual_partition_sums,
2325 raw_bits_per_partition,
2326 residual_samples,
2327 order,
2328 rice_parameter,
2329 min_partition_order,
2330 max_partition_order,
2331 precompute_partition_sums,
2332 do_escape_coding,
2333 rice_parameter_search_dist,
2334 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2335 );
Josh Coalson94e02cd2001-01-25 10:41:06 +00002336
2337 subframe->data.fixed.order = order;
2338 for(i = 0; i < order; i++)
2339 subframe->data.fixed.warmup[i] = signal[i];
2340
Josh Coalson82b73242001-03-28 22:17:05 +00002341 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (order * subframe_bps) + residual_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002342}
2343
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002344#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00002345unsigned evaluate_lpc_subframe_(
2346 FLAC__StreamEncoder *encoder,
2347 const FLAC__int32 signal[],
2348 FLAC__int32 residual[],
2349 FLAC__uint32 abs_residual[],
2350 FLAC__uint64 abs_residual_partition_sums[],
2351 unsigned raw_bits_per_partition[],
2352 const FLAC__real lp_coeff[],
2353 unsigned blocksize,
2354 unsigned subframe_bps,
2355 unsigned order,
2356 unsigned qlp_coeff_precision,
2357 unsigned rice_parameter,
2358 unsigned min_partition_order,
2359 unsigned max_partition_order,
2360 FLAC__bool precompute_partition_sums,
2361 FLAC__bool do_escape_coding,
2362 unsigned rice_parameter_search_dist,
2363 FLAC__Subframe *subframe,
2364 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2365)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002366{
Josh Coalson77e3f312001-06-23 03:03:24 +00002367 FLAC__int32 qlp_coeff[FLAC__MAX_LPC_ORDER];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002368 unsigned i, residual_bits;
2369 int quantization, ret;
2370 const unsigned residual_samples = blocksize - order;
2371
Josh Coalson20ac2c12002-08-30 05:47:14 +00002372 /* try to keep qlp coeff precision such that only 32-bit math is required for decode of <=16bps streams */
2373 if(subframe_bps <= 16) {
2374 FLAC__ASSERT(order > 0);
2375 FLAC__ASSERT(order <= FLAC__MAX_LPC_ORDER);
2376 qlp_coeff_precision = min(qlp_coeff_precision, 32 - subframe_bps - FLAC__bitmath_ilog2(order));
2377 }
2378
Josh Coalsonc9c0d132002-10-04 05:29:05 +00002379 ret = FLAC__lpc_quantize_coefficients(lp_coeff, order, qlp_coeff_precision, qlp_coeff, &quantization);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002380 if(ret != 0)
2381 return 0; /* this is a hack to indicate to the caller that we can't do lp at this order on this subframe */
2382
Josh Coalsonfb9d18f2002-10-21 07:04:07 +00002383 if(subframe_bps + qlp_coeff_precision + FLAC__bitmath_ilog2(order) <= 32)
2384 if(subframe_bps <= 16 && qlp_coeff_precision <= 16)
2385 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2386 else
2387 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
Josh Coalsonc9c0d132002-10-04 05:29:05 +00002388 else
2389 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_64bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002390
2391 subframe->type = FLAC__SUBFRAME_TYPE_LPC;
2392
2393 subframe->data.lpc.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
Josh Coalsona37ba462002-08-19 21:36:39 +00002394 subframe->data.lpc.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002395 subframe->data.lpc.residual = residual;
2396
Josh Coalson6fe72f72002-08-20 04:01:59 +00002397 residual_bits =
2398 find_best_partition_order_(
2399 encoder->private_,
2400 residual,
2401 abs_residual,
2402 abs_residual_partition_sums,
2403 raw_bits_per_partition,
2404 residual_samples,
2405 order,
2406 rice_parameter,
2407 min_partition_order,
2408 max_partition_order,
2409 precompute_partition_sums,
2410 do_escape_coding,
2411 rice_parameter_search_dist,
2412 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2413 );
Josh Coalson94e02cd2001-01-25 10:41:06 +00002414
2415 subframe->data.lpc.order = order;
2416 subframe->data.lpc.qlp_coeff_precision = qlp_coeff_precision;
2417 subframe->data.lpc.quantization_level = quantization;
Josh Coalson77e3f312001-06-23 03:03:24 +00002418 memcpy(subframe->data.lpc.qlp_coeff, qlp_coeff, sizeof(FLAC__int32)*FLAC__MAX_LPC_ORDER);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002419 for(i = 0; i < order; i++)
2420 subframe->data.lpc.warmup[i] = signal[i];
2421
Josh Coalson82b73242001-03-28 22:17:05 +00002422 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;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002423}
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002424#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002425
Josh Coalson6fe72f72002-08-20 04:01:59 +00002426unsigned evaluate_verbatim_subframe_(
2427 const FLAC__int32 signal[],
2428 unsigned blocksize,
2429 unsigned subframe_bps,
2430 FLAC__Subframe *subframe
2431)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002432{
2433 subframe->type = FLAC__SUBFRAME_TYPE_VERBATIM;
2434
2435 subframe->data.verbatim.data = signal;
2436
Josh Coalson82b73242001-03-28 22:17:05 +00002437 return FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN + (blocksize * subframe_bps);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002438}
2439
Josh Coalson6fe72f72002-08-20 04:01:59 +00002440unsigned find_best_partition_order_(
2441 FLAC__StreamEncoderPrivate *private_,
2442 const FLAC__int32 residual[],
2443 FLAC__uint32 abs_residual[],
2444 FLAC__uint64 abs_residual_partition_sums[],
2445 unsigned raw_bits_per_partition[],
2446 unsigned residual_samples,
2447 unsigned predictor_order,
2448 unsigned rice_parameter,
2449 unsigned min_partition_order,
2450 unsigned max_partition_order,
2451 FLAC__bool precompute_partition_sums,
2452 FLAC__bool do_escape_coding,
2453 unsigned rice_parameter_search_dist,
2454 FLAC__EntropyCodingMethod_PartitionedRice *best_partitioned_rice
2455)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002456{
Josh Coalson77e3f312001-06-23 03:03:24 +00002457 FLAC__int32 r;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002458 unsigned residual_bits, best_residual_bits = 0;
Josh Coalsonafcd8772001-04-18 22:59:25 +00002459 unsigned residual_sample;
Josh Coalson8084b052001-11-01 00:27:29 +00002460 unsigned best_parameters_index = 0;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002461 const unsigned blocksize = residual_samples + predictor_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002462
Josh Coalson2051dd42001-04-12 22:22:34 +00002463 /* compute abs(residual) for use later */
2464 for(residual_sample = 0; residual_sample < residual_samples; residual_sample++) {
2465 r = residual[residual_sample];
Josh Coalson77e3f312001-06-23 03:03:24 +00002466 abs_residual[residual_sample] = (FLAC__uint32)(r<0? -r : r);
Josh Coalson2051dd42001-04-12 22:22:34 +00002467 }
2468
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002469 max_partition_order = FLAC__format_get_max_rice_partition_order_from_blocksize_limited_max_and_predictor_order(max_partition_order, blocksize, predictor_order);
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002470 min_partition_order = min(min_partition_order, max_partition_order);
2471
Josh Coalson8395d022001-07-12 21:25:22 +00002472 if(precompute_partition_sums) {
2473 int partition_order;
2474 unsigned sum;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002475
Josh Coalsonf1eff452002-07-31 07:05:33 +00002476 precompute_partition_info_sums_(abs_residual, abs_residual_partition_sums, residual_samples, predictor_order, min_partition_order, max_partition_order);
Josh Coalson8395d022001-07-12 21:25:22 +00002477
2478 if(do_escape_coding)
Josh Coalsonf1eff452002-07-31 07:05:33 +00002479 precompute_partition_info_escapes_(residual, raw_bits_per_partition, residual_samples, predictor_order, min_partition_order, max_partition_order);
Josh Coalson8395d022001-07-12 21:25:22 +00002480
2481 for(partition_order = (int)max_partition_order, sum = 0; partition_order >= (int)min_partition_order; partition_order--) {
2482#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002483 if(!
2484 set_partitioned_rice_with_precompute_(
2485 residual,
2486 abs_residual_partition_sums+sum,
2487 raw_bits_per_partition+sum,
2488 residual_samples,
2489 predictor_order,
2490 rice_parameter,
2491 rice_parameter_search_dist,
2492 (unsigned)partition_order,
2493 do_escape_coding,
2494 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2495 &residual_bits
2496 )
2497 )
Josh Coalsonafcd8772001-04-18 22:59:25 +00002498#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002499 if(!
2500 set_partitioned_rice_with_precompute_(
2501 abs_residual,
2502 abs_residual_partition_sums+sum,
2503 raw_bits_per_partition+sum,
2504 residual_samples,
2505 predictor_order,
2506 rice_parameter,
2507 rice_parameter_search_dist,
2508 (unsigned)partition_order,
2509 do_escape_coding,
2510 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2511 &residual_bits
2512 )
2513 )
Josh Coalson8395d022001-07-12 21:25:22 +00002514#endif
2515 {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002516 FLAC__ASSERT(best_residual_bits != 0);
2517 break;
Josh Coalson8395d022001-07-12 21:25:22 +00002518 }
2519 sum += 1u << partition_order;
2520 if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2521 best_residual_bits = residual_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00002522 best_parameters_index = !best_parameters_index;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002523 best_partitioned_rice->order = partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00002524 }
Josh Coalsonafcd8772001-04-18 22:59:25 +00002525 }
2526 }
Josh Coalson8395d022001-07-12 21:25:22 +00002527 else {
2528 unsigned partition_order;
2529 for(partition_order = min_partition_order; partition_order <= max_partition_order; partition_order++) {
2530#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002531 if(!
2532 set_partitioned_rice_(
2533 abs_residual,
2534 residual,
2535 residual_samples,
2536 predictor_order,
2537 rice_parameter,
2538 rice_parameter_search_dist,
2539 partition_order,
2540 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2541 &residual_bits
2542 )
2543 )
Josh Coalson8395d022001-07-12 21:25:22 +00002544#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002545 if(!
2546 set_partitioned_rice_(
2547 abs_residual,
2548 residual_samples,
2549 predictor_order,
2550 rice_parameter,
2551 rice_parameter_search_dist,
2552 partition_order,
2553 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2554 &residual_bits
2555 )
2556 )
Josh Coalsonafcd8772001-04-18 22:59:25 +00002557#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002558 {
2559 FLAC__ASSERT(best_residual_bits != 0);
2560 break;
2561 }
2562 if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2563 best_residual_bits = residual_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00002564 best_parameters_index = !best_parameters_index;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002565 best_partitioned_rice->order = partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00002566 }
2567 }
2568 }
2569
Josh Coalsona37ba462002-08-19 21:36:39 +00002570 /*
Josh Coalson20ac2c12002-08-30 05:47:14 +00002571 * We are allowed to de-const the pointer based on our special knowledge;
Josh Coalsona37ba462002-08-19 21:36:39 +00002572 * it is const to the outside world.
2573 */
2574 {
2575 FLAC__EntropyCodingMethod_PartitionedRiceContents* best_partitioned_rice_contents = (FLAC__EntropyCodingMethod_PartitionedRiceContents*)best_partitioned_rice->contents;
2576 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(best_partitioned_rice_contents, max(6, best_partitioned_rice->order));
2577 memcpy(best_partitioned_rice_contents->parameters, private_->partitioned_rice_contents_extra[best_parameters_index].parameters, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2578 memcpy(best_partitioned_rice_contents->raw_bits, private_->partitioned_rice_contents_extra[best_parameters_index].raw_bits, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2579 }
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002580
2581 return best_residual_bits;
2582}
2583
Josh Coalson6fe72f72002-08-20 04:01:59 +00002584void precompute_partition_info_sums_(
2585 const FLAC__uint32 abs_residual[],
2586 FLAC__uint64 abs_residual_partition_sums[],
2587 unsigned residual_samples,
2588 unsigned predictor_order,
2589 unsigned min_partition_order,
2590 unsigned max_partition_order
2591)
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002592{
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002593 int partition_order;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002594 unsigned from_partition, to_partition = 0;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002595 const unsigned blocksize = residual_samples + predictor_order;
2596
Josh Coalsonaef013c2001-04-24 01:25:42 +00002597 /* first do max_partition_order */
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002598 for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002599 FLAC__uint64 abs_residual_partition_sum;
Josh Coalson77e3f312001-06-23 03:03:24 +00002600 FLAC__uint32 abs_r;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002601 unsigned partition, partition_sample, partition_samples, residual_sample;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002602 const unsigned partitions = 1u << partition_order;
2603 const unsigned default_partition_samples = blocksize >> partition_order;
2604
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002605 FLAC__ASSERT(default_partition_samples > predictor_order);
2606
2607 for(partition = residual_sample = 0; partition < partitions; partition++) {
2608 partition_samples = default_partition_samples;
2609 if(partition == 0)
2610 partition_samples -= predictor_order;
2611 abs_residual_partition_sum = 0;
2612 for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2613 abs_r = abs_residual[residual_sample];
2614 abs_residual_partition_sum += abs_r;
2615 residual_sample++;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002616 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002617 abs_residual_partition_sums[partition] = abs_residual_partition_sum;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002618 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002619 to_partition = partitions;
2620 break;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002621 }
Josh Coalsonf76a3612001-04-18 02:28:11 +00002622
Josh Coalson8395d022001-07-12 21:25:22 +00002623 /* now merge partitions for lower orders */
Josh Coalson6bd17572001-05-25 19:02:01 +00002624 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002625 FLAC__uint64 s;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002626 unsigned i;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002627 const unsigned partitions = 1u << partition_order;
2628 for(i = 0; i < partitions; i++) {
Josh Coalsonaef013c2001-04-24 01:25:42 +00002629 s = abs_residual_partition_sums[from_partition];
Josh Coalsonaef013c2001-04-24 01:25:42 +00002630 from_partition++;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002631 abs_residual_partition_sums[to_partition] = s + abs_residual_partition_sums[from_partition];
Josh Coalsonaef013c2001-04-24 01:25:42 +00002632 from_partition++;
2633 to_partition++;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002634 }
2635 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002636}
Josh Coalson8395d022001-07-12 21:25:22 +00002637
Josh Coalson6fe72f72002-08-20 04:01:59 +00002638void precompute_partition_info_escapes_(
2639 const FLAC__int32 residual[],
2640 unsigned raw_bits_per_partition[],
2641 unsigned residual_samples,
2642 unsigned predictor_order,
2643 unsigned min_partition_order,
2644 unsigned max_partition_order
2645)
Josh Coalson8395d022001-07-12 21:25:22 +00002646{
2647 int partition_order;
2648 unsigned from_partition, to_partition = 0;
2649 const unsigned blocksize = residual_samples + predictor_order;
2650
2651 /* first do max_partition_order */
2652 for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
2653 FLAC__int32 r, residual_partition_min, residual_partition_max;
2654 unsigned silog2_min, silog2_max;
2655 unsigned partition, partition_sample, partition_samples, residual_sample;
2656 const unsigned partitions = 1u << partition_order;
2657 const unsigned default_partition_samples = blocksize >> partition_order;
2658
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002659 FLAC__ASSERT(default_partition_samples > predictor_order);
2660
2661 for(partition = residual_sample = 0; partition < partitions; partition++) {
2662 partition_samples = default_partition_samples;
2663 if(partition == 0)
2664 partition_samples -= predictor_order;
2665 residual_partition_min = residual_partition_max = 0;
2666 for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2667 r = residual[residual_sample];
2668 if(r < residual_partition_min)
2669 residual_partition_min = r;
2670 else if(r > residual_partition_max)
2671 residual_partition_max = r;
2672 residual_sample++;
Josh Coalson8395d022001-07-12 21:25:22 +00002673 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002674 silog2_min = FLAC__bitmath_silog2(residual_partition_min);
2675 silog2_max = FLAC__bitmath_silog2(residual_partition_max);
2676 raw_bits_per_partition[partition] = max(silog2_min, silog2_max);
Josh Coalson8395d022001-07-12 21:25:22 +00002677 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002678 to_partition = partitions;
2679 break;
Josh Coalson8395d022001-07-12 21:25:22 +00002680 }
2681
2682 /* now merge partitions for lower orders */
2683 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2684 unsigned m;
2685 unsigned i;
2686 const unsigned partitions = 1u << partition_order;
2687 for(i = 0; i < partitions; i++) {
2688 m = raw_bits_per_partition[from_partition];
2689 from_partition++;
2690 raw_bits_per_partition[to_partition] = max(m, raw_bits_per_partition[from_partition]);
2691 from_partition++;
2692 to_partition++;
2693 }
2694 }
2695}
Josh Coalson94e02cd2001-01-25 10:41:06 +00002696
Josh Coalson352e0f62001-03-20 22:55:50 +00002697#ifdef VARIABLE_RICE_BITS
2698#undef VARIABLE_RICE_BITS
2699#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002700#ifndef DONT_ESTIMATE_RICE_BITS
Josh Coalson352e0f62001-03-20 22:55:50 +00002701#define VARIABLE_RICE_BITS(value, parameter) ((value) >> (parameter))
Josh Coalson8395d022001-07-12 21:25:22 +00002702#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00002703
Josh Coalson8395d022001-07-12 21:25:22 +00002704#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002705FLAC__bool set_partitioned_rice_(
2706 const FLAC__uint32 abs_residual[],
2707 const FLAC__int32 residual[],
2708 const unsigned residual_samples,
2709 const unsigned predictor_order,
2710 const unsigned suggested_rice_parameter,
2711 const unsigned rice_parameter_search_dist,
2712 const unsigned partition_order,
2713 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2714 unsigned *bits
2715)
Josh Coalson8395d022001-07-12 21:25:22 +00002716#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002717FLAC__bool set_partitioned_rice_(
2718 const FLAC__uint32 abs_residual[],
2719 const unsigned residual_samples,
2720 const unsigned predictor_order,
2721 const unsigned suggested_rice_parameter,
2722 const unsigned rice_parameter_search_dist,
2723 const unsigned partition_order,
2724 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2725 unsigned *bits
2726)
Josh Coalson8395d022001-07-12 21:25:22 +00002727#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002728{
Josh Coalson034dfab2001-04-27 19:10:23 +00002729 unsigned rice_parameter, partition_bits;
2730#ifndef NO_RICE_SEARCH
2731 unsigned best_partition_bits;
2732 unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
2733#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002734 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002735 unsigned *parameters;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002736
Josh Coalson1b689822001-05-31 20:11:02 +00002737 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
Josh Coalson2051dd42001-04-12 22:22:34 +00002738
Josh Coalsona37ba462002-08-19 21:36:39 +00002739 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
2740 parameters = partitioned_rice_contents->parameters;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002741
Josh Coalson94e02cd2001-01-25 10:41:06 +00002742 if(partition_order == 0) {
2743 unsigned i;
Josh Coalson352e0f62001-03-20 22:55:50 +00002744
Josh Coalson034dfab2001-04-27 19:10:23 +00002745#ifndef NO_RICE_SEARCH
Josh Coalson60f77d72001-04-25 02:16:36 +00002746 if(rice_parameter_search_dist) {
Josh Coalson034dfab2001-04-27 19:10:23 +00002747 if(suggested_rice_parameter < rice_parameter_search_dist)
Josh Coalson60f77d72001-04-25 02:16:36 +00002748 min_rice_parameter = 0;
2749 else
Josh Coalson034dfab2001-04-27 19:10:23 +00002750 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
2751 max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
Josh Coalson8395d022001-07-12 21:25:22 +00002752 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002753#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002754 fprintf(stderr, "clipping rice_parameter (%u -> %u) @2\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2755#endif
Josh Coalson60f77d72001-04-25 02:16:36 +00002756 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00002757 }
2758 }
2759 else
2760 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
2761
2762 best_partition_bits = 0xffffffff;
2763 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2764#endif
2765#ifdef VARIABLE_RICE_BITS
2766#ifdef FLAC__SYMMETRIC_RICE
2767 partition_bits = (2+rice_parameter) * residual_samples;
2768#else
2769 const unsigned rice_parameter_estimate = rice_parameter-1;
2770 partition_bits = (1+rice_parameter) * residual_samples;
2771#endif
2772#else
2773 partition_bits = 0;
2774#endif
2775 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2776 for(i = 0; i < residual_samples; i++) {
2777#ifdef VARIABLE_RICE_BITS
2778#ifdef FLAC__SYMMETRIC_RICE
2779 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
2780#else
2781 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
2782#endif
2783#else
2784 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
2785#endif
2786 }
2787#ifndef NO_RICE_SEARCH
2788 if(partition_bits < best_partition_bits) {
2789 best_rice_parameter = rice_parameter;
2790 best_partition_bits = partition_bits;
2791 }
2792 }
2793#endif
2794 parameters[0] = best_rice_parameter;
2795 bits_ += best_partition_bits;
2796 }
2797 else {
2798 unsigned partition, residual_sample, save_residual_sample, partition_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002799 unsigned partition_samples;
2800 FLAC__uint64 mean, k;
Josh Coalson8395d022001-07-12 21:25:22 +00002801 const unsigned partitions = 1u << partition_order;
2802 for(partition = residual_sample = 0; partition < partitions; partition++) {
2803 partition_samples = (residual_samples+predictor_order) >> partition_order;
2804 if(partition == 0) {
2805 if(partition_samples <= predictor_order)
2806 return false;
2807 else
2808 partition_samples -= predictor_order;
2809 }
2810 mean = 0;
2811 save_residual_sample = residual_sample;
2812 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++)
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002813 mean += abs_residual[residual_sample];
Josh Coalson8395d022001-07-12 21:25:22 +00002814 residual_sample = save_residual_sample;
2815#ifdef FLAC__SYMMETRIC_RICE
2816 mean += partition_samples >> 1; /* for rounding effect */
2817 mean /= partition_samples;
2818
2819 /* calc rice_parameter = floor(log2(mean)) */
2820 rice_parameter = 0;
2821 mean>>=1;
2822 while(mean) {
2823 rice_parameter++;
2824 mean >>= 1;
2825 }
2826#else
2827 /* calc rice_parameter ala LOCO-I */
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002828 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
Josh Coalson8395d022001-07-12 21:25:22 +00002829 ;
2830#endif
2831 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002832#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002833 fprintf(stderr, "clipping rice_parameter (%u -> %u) @3\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2834#endif
2835 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2836 }
2837
2838#ifndef NO_RICE_SEARCH
2839 if(rice_parameter_search_dist) {
2840 if(rice_parameter < rice_parameter_search_dist)
2841 min_rice_parameter = 0;
2842 else
2843 min_rice_parameter = rice_parameter - rice_parameter_search_dist;
2844 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
2845 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002846#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002847 fprintf(stderr, "clipping rice_parameter (%u -> %u) @4\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2848#endif
2849 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2850 }
2851 }
2852 else
2853 min_rice_parameter = max_rice_parameter = rice_parameter;
2854
2855 best_partition_bits = 0xffffffff;
2856 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2857#endif
2858#ifdef VARIABLE_RICE_BITS
2859#ifdef FLAC__SYMMETRIC_RICE
2860 partition_bits = (2+rice_parameter) * partition_samples;
2861#else
2862 const unsigned rice_parameter_estimate = rice_parameter-1;
2863 partition_bits = (1+rice_parameter) * partition_samples;
2864#endif
2865#else
2866 partition_bits = 0;
2867#endif
2868 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2869 save_residual_sample = residual_sample;
2870 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
2871#ifdef VARIABLE_RICE_BITS
2872#ifdef FLAC__SYMMETRIC_RICE
2873 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
2874#else
2875 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
2876#endif
2877#else
2878 partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
2879#endif
2880 }
2881#ifndef NO_RICE_SEARCH
2882 if(rice_parameter != max_rice_parameter)
2883 residual_sample = save_residual_sample;
2884 if(partition_bits < best_partition_bits) {
2885 best_rice_parameter = rice_parameter;
2886 best_partition_bits = partition_bits;
2887 }
2888 }
2889#endif
2890 parameters[partition] = best_rice_parameter;
2891 bits_ += best_partition_bits;
2892 }
2893 }
2894
2895 *bits = bits_;
2896 return true;
2897}
2898
2899#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002900FLAC__bool set_partitioned_rice_with_precompute_(
2901 const FLAC__int32 residual[],
2902 const FLAC__uint64 abs_residual_partition_sums[],
2903 const unsigned raw_bits_per_partition[],
2904 const unsigned residual_samples,
2905 const unsigned predictor_order,
2906 const unsigned suggested_rice_parameter,
2907 const unsigned rice_parameter_search_dist,
2908 const unsigned partition_order,
2909 const FLAC__bool search_for_escapes,
2910 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2911 unsigned *bits
2912)
Josh Coalson8395d022001-07-12 21:25:22 +00002913#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002914FLAC__bool set_partitioned_rice_with_precompute_(
2915 const FLAC__uint32 abs_residual[],
2916 const FLAC__uint64 abs_residual_partition_sums[],
2917 const unsigned raw_bits_per_partition[],
2918 const unsigned residual_samples,
2919 const unsigned predictor_order,
2920 const unsigned suggested_rice_parameter,
2921 const unsigned rice_parameter_search_dist,
2922 const unsigned partition_order,
2923 const FLAC__bool search_for_escapes,
2924 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2925 unsigned *bits
2926)
Josh Coalson8395d022001-07-12 21:25:22 +00002927#endif
2928{
2929 unsigned rice_parameter, partition_bits;
2930#ifndef NO_RICE_SEARCH
2931 unsigned best_partition_bits;
2932 unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
2933#endif
2934 unsigned flat_bits;
2935 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002936 unsigned *parameters, *raw_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00002937
2938 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
2939
Josh Coalsona37ba462002-08-19 21:36:39 +00002940 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
2941 parameters = partitioned_rice_contents->parameters;
2942 raw_bits = partitioned_rice_contents->raw_bits;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002943
Josh Coalson8395d022001-07-12 21:25:22 +00002944 if(partition_order == 0) {
2945 unsigned i;
2946
2947#ifndef NO_RICE_SEARCH
2948 if(rice_parameter_search_dist) {
2949 if(suggested_rice_parameter < rice_parameter_search_dist)
2950 min_rice_parameter = 0;
2951 else
2952 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
2953 max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
2954 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002955#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002956 fprintf(stderr, "clipping rice_parameter (%u -> %u) @5\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2957#endif
2958 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2959 }
Josh Coalson60f77d72001-04-25 02:16:36 +00002960 }
2961 else
Josh Coalson034dfab2001-04-27 19:10:23 +00002962 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
Josh Coalson2051dd42001-04-12 22:22:34 +00002963
Josh Coalson034dfab2001-04-27 19:10:23 +00002964 best_partition_bits = 0xffffffff;
2965 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2966#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00002967#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00002968#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson034dfab2001-04-27 19:10:23 +00002969 partition_bits = (2+rice_parameter) * residual_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00002970#else
Josh Coalson352e0f62001-03-20 22:55:50 +00002971 const unsigned rice_parameter_estimate = rice_parameter-1;
Josh Coalson034dfab2001-04-27 19:10:23 +00002972 partition_bits = (1+rice_parameter) * residual_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00002973#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00002974#else
2975 partition_bits = 0;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002976#endif
Josh Coalson2051dd42001-04-12 22:22:34 +00002977 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
Josh Coalson352e0f62001-03-20 22:55:50 +00002978 for(i = 0; i < residual_samples; i++) {
2979#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00002980#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson2051dd42001-04-12 22:22:34 +00002981 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002982#else
Josh Coalson2051dd42001-04-12 22:22:34 +00002983 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
Josh Coalsonb9433f92001-03-17 01:07:00 +00002984#endif
2985#else
Josh Coalson2051dd42001-04-12 22:22:34 +00002986 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
Josh Coalson94e02cd2001-01-25 10:41:06 +00002987#endif
Josh Coalson2051dd42001-04-12 22:22:34 +00002988 }
Josh Coalson034dfab2001-04-27 19:10:23 +00002989#ifndef NO_RICE_SEARCH
2990 if(partition_bits < best_partition_bits) {
2991 best_rice_parameter = rice_parameter;
2992 best_partition_bits = partition_bits;
Josh Coalson352e0f62001-03-20 22:55:50 +00002993 }
2994 }
Josh Coalson034dfab2001-04-27 19:10:23 +00002995#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002996 if(search_for_escapes) {
2997 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;
2998 if(flat_bits <= best_partition_bits) {
2999 raw_bits[0] = raw_bits_per_partition[0];
3000 best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3001 best_partition_bits = flat_bits;
3002 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003003 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003004 parameters[0] = best_rice_parameter;
3005 bits_ += best_partition_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003006 }
3007 else {
Josh Coalson4dacd192001-06-06 21:11:44 +00003008 unsigned partition, residual_sample, save_residual_sample, partition_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00003009 unsigned partition_samples;
3010 FLAC__uint64 mean, k;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00003011 const unsigned partitions = 1u << partition_order;
Josh Coalson4dacd192001-06-06 21:11:44 +00003012 for(partition = residual_sample = 0; partition < partitions; partition++) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00003013 partition_samples = (residual_samples+predictor_order) >> partition_order;
Josh Coalson034dfab2001-04-27 19:10:23 +00003014 if(partition == 0) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00003015 if(partition_samples <= predictor_order)
3016 return false;
3017 else
3018 partition_samples -= predictor_order;
3019 }
Josh Coalson05d20792001-06-29 23:12:26 +00003020 mean = abs_residual_partition_sums[partition];
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003021#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson05d20792001-06-29 23:12:26 +00003022 mean += partition_samples >> 1; /* for rounding effect */
3023 mean /= partition_samples;
3024
Josh Coalson034dfab2001-04-27 19:10:23 +00003025 /* calc rice_parameter = floor(log2(mean)) */
3026 rice_parameter = 0;
3027 mean>>=1;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003028 while(mean) {
Josh Coalson034dfab2001-04-27 19:10:23 +00003029 rice_parameter++;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003030 mean >>= 1;
3031 }
Josh Coalsonb9433f92001-03-17 01:07:00 +00003032#else
Josh Coalson05d20792001-06-29 23:12:26 +00003033 /* calc rice_parameter ala LOCO-I */
Josh Coalsonb3347bd2001-07-16 18:06:41 +00003034 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
Josh Coalson05d20792001-06-29 23:12:26 +00003035 ;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003036#endif
Josh Coalson8395d022001-07-12 21:25:22 +00003037 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00003038#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00003039 fprintf(stderr, "clipping rice_parameter (%u -> %u) @6\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3040#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003041 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00003042 }
Josh Coalson60f77d72001-04-25 02:16:36 +00003043
Josh Coalson034dfab2001-04-27 19:10:23 +00003044#ifndef NO_RICE_SEARCH
Josh Coalson60f77d72001-04-25 02:16:36 +00003045 if(rice_parameter_search_dist) {
Josh Coalson034dfab2001-04-27 19:10:23 +00003046 if(rice_parameter < rice_parameter_search_dist)
Josh Coalson60f77d72001-04-25 02:16:36 +00003047 min_rice_parameter = 0;
3048 else
Josh Coalson034dfab2001-04-27 19:10:23 +00003049 min_rice_parameter = rice_parameter - rice_parameter_search_dist;
3050 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
Josh Coalson8395d022001-07-12 21:25:22 +00003051 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00003052#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00003053 fprintf(stderr, "clipping rice_parameter (%u -> %u) @7\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3054#endif
Josh Coalson60f77d72001-04-25 02:16:36 +00003055 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00003056 }
Josh Coalson60f77d72001-04-25 02:16:36 +00003057 }
3058 else
3059 min_rice_parameter = max_rice_parameter = rice_parameter;
Josh Coalson60f77d72001-04-25 02:16:36 +00003060
Josh Coalson034dfab2001-04-27 19:10:23 +00003061 best_partition_bits = 0xffffffff;
3062 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3063#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00003064#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003065#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson034dfab2001-04-27 19:10:23 +00003066 partition_bits = (2+rice_parameter) * partition_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003067#else
Josh Coalson034dfab2001-04-27 19:10:23 +00003068 const unsigned rice_parameter_estimate = rice_parameter-1;
3069 partition_bits = (1+rice_parameter) * partition_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003070#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003071#else
3072 partition_bits = 0;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003073#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003074 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
Josh Coalson4dacd192001-06-06 21:11:44 +00003075 save_residual_sample = residual_sample;
3076 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
Josh Coalson352e0f62001-03-20 22:55:50 +00003077#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003078#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson4dacd192001-06-06 21:11:44 +00003079 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
Josh Coalson94e02cd2001-01-25 10:41:06 +00003080#else
Josh Coalson4dacd192001-06-06 21:11:44 +00003081 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
Josh Coalsonb9433f92001-03-17 01:07:00 +00003082#endif
3083#else
Josh Coalson4dacd192001-06-06 21:11:44 +00003084 partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] instead of abs_residual[] */
Josh Coalson94e02cd2001-01-25 10:41:06 +00003085#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003086 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003087#ifndef NO_RICE_SEARCH
Josh Coalson4dacd192001-06-06 21:11:44 +00003088 if(rice_parameter != max_rice_parameter)
3089 residual_sample = save_residual_sample;
Josh Coalson034dfab2001-04-27 19:10:23 +00003090 if(partition_bits < best_partition_bits) {
3091 best_rice_parameter = rice_parameter;
3092 best_partition_bits = partition_bits;
3093 }
Josh Coalson2051dd42001-04-12 22:22:34 +00003094 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003095#endif
Josh Coalson8395d022001-07-12 21:25:22 +00003096 if(search_for_escapes) {
3097 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;
3098 if(flat_bits <= best_partition_bits) {
3099 raw_bits[partition] = raw_bits_per_partition[partition];
3100 best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3101 best_partition_bits = flat_bits;
3102 }
Josh Coalson2051dd42001-04-12 22:22:34 +00003103 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003104 parameters[partition] = best_rice_parameter;
3105 bits_ += best_partition_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003106 }
3107 }
3108
3109 *bits = bits_;
3110 return true;
3111}
Josh Coalson859bc542001-03-27 22:22:27 +00003112
Josh Coalsonf1eff452002-07-31 07:05:33 +00003113unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples)
Josh Coalson859bc542001-03-27 22:22:27 +00003114{
3115 unsigned i, shift;
Josh Coalson77e3f312001-06-23 03:03:24 +00003116 FLAC__int32 x = 0;
Josh Coalson859bc542001-03-27 22:22:27 +00003117
3118 for(i = 0; i < samples && !(x&1); i++)
3119 x |= signal[i];
3120
3121 if(x == 0) {
3122 shift = 0;
3123 }
3124 else {
3125 for(shift = 0; !(x&1); shift++)
3126 x >>= 1;
3127 }
3128
3129 if(shift > 0) {
3130 for(i = 0; i < samples; i++)
3131 signal[i] >>= shift;
3132 }
3133
3134 return shift;
3135}
Josh Coalsond86e03b2002-08-03 21:56:15 +00003136
3137void append_to_verify_fifo_(verify_input_fifo *fifo, const FLAC__int32 * const input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3138{
3139 unsigned channel;
3140
3141 for(channel = 0; channel < channels; channel++)
3142 memcpy(&fifo->data[channel][fifo->tail], &input[channel][input_offset], sizeof(FLAC__int32) * wide_samples);
3143
3144 fifo->tail += wide_samples;
3145
3146 FLAC__ASSERT(fifo->tail <= fifo->size);
3147}
3148
3149void append_to_verify_fifo_interleaved_(verify_input_fifo *fifo, const FLAC__int32 input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3150{
3151 unsigned channel;
3152 unsigned sample, wide_sample;
3153 unsigned tail = fifo->tail;
3154
3155 sample = input_offset * channels;
3156 for(wide_sample = 0; wide_sample < wide_samples; wide_sample++) {
3157 for(channel = 0; channel < channels; channel++)
3158 fifo->data[channel][tail] = input[sample++];
3159 tail++;
3160 }
3161 fifo->tail = tail;
3162
3163 FLAC__ASSERT(fifo->tail <= fifo->size);
3164}
3165
3166FLAC__StreamDecoderReadStatus verify_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], unsigned *bytes, void *client_data)
3167{
3168 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3169 const unsigned encoded_bytes = encoder->private_->verify.output.bytes;
3170 (void)decoder;
3171
3172 if(encoder->private_->verify.needs_magic_hack) {
3173 FLAC__ASSERT(*bytes >= FLAC__STREAM_SYNC_LENGTH);
3174 *bytes = FLAC__STREAM_SYNC_LENGTH;
3175 memcpy(buffer, FLAC__STREAM_SYNC_STRING, *bytes);
3176 encoder->private_->verify.needs_magic_hack = false;
3177 }
3178 else {
3179 if(encoded_bytes == 0) {
Josh Coalsonfc2b7372002-08-16 05:39:34 +00003180 /*
3181 * If we get here, a FIFO underflow has occurred,
3182 * which means there is a bug somewhere.
3183 */
3184 FLAC__ASSERT(0);
Josh Coalsond86e03b2002-08-03 21:56:15 +00003185 return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
3186 }
3187 else if(encoded_bytes < *bytes)
3188 *bytes = encoded_bytes;
3189 memcpy(buffer, encoder->private_->verify.output.data, *bytes);
3190 encoder->private_->verify.output.data += *bytes;
3191 encoder->private_->verify.output.bytes -= *bytes;
3192 }
3193
3194 return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
3195}
3196
3197FLAC__StreamDecoderWriteStatus verify_write_callback_(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
3198{
3199 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder *)client_data;
3200 unsigned channel;
3201 const unsigned channels = FLAC__stream_decoder_get_channels(decoder);
3202 const unsigned blocksize = frame->header.blocksize;
3203 const unsigned bytes_per_block = sizeof(FLAC__int32) * blocksize;
3204
3205 for(channel = 0; channel < channels; channel++) {
3206 if(0 != memcmp(buffer[channel], encoder->private_->verify.input_fifo.data[channel], bytes_per_block)) {
3207 unsigned i, sample = 0;
3208 FLAC__int32 expect = 0, got = 0;
3209
3210 for(i = 0; i < blocksize; i++) {
3211 if(buffer[channel][i] != encoder->private_->verify.input_fifo.data[channel][i]) {
3212 sample = i;
3213 expect = (FLAC__int32)encoder->private_->verify.input_fifo.data[channel][i];
3214 got = (FLAC__int32)buffer[channel][i];
3215 break;
3216 }
3217 }
3218 FLAC__ASSERT(i < blocksize);
3219 FLAC__ASSERT(frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
3220 encoder->private_->verify.error_stats.absolute_sample = frame->header.number.sample_number + sample;
Josh Coalson5f39e9f2002-08-21 05:27:01 +00003221 encoder->private_->verify.error_stats.frame_number = (unsigned)(frame->header.number.sample_number / blocksize);
Josh Coalsond86e03b2002-08-03 21:56:15 +00003222 encoder->private_->verify.error_stats.channel = channel;
3223 encoder->private_->verify.error_stats.sample = sample;
3224 encoder->private_->verify.error_stats.expected = expect;
3225 encoder->private_->verify.error_stats.got = got;
3226 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA;
3227 return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
3228 }
3229 }
3230 /* dequeue the frame from the fifo */
3231 for(channel = 0; channel < channels; channel++) {
3232 memmove(&encoder->private_->verify.input_fifo.data[channel][0], &encoder->private_->verify.input_fifo.data[channel][blocksize], encoder->private_->verify.input_fifo.tail - blocksize);
3233 }
3234 encoder->private_->verify.input_fifo.tail -= blocksize;
3235 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
3236}
3237
3238void verify_metadata_callback_(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
3239{
3240 (void)decoder, (void)metadata, (void)client_data;
3241}
3242
3243void verify_error_callback_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
3244{
3245 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3246 (void)decoder, (void)status;
3247 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
3248}