blob: 126384023cd1dbd82d0879c5bc8bb7a4d73f19c1 [file] [log] [blame]
Josh Coalson26560dd2001-02-08 00:38:41 +00001/* libFLAC - Free Lossless Audio Codec library
Josh Coalsona78fac62005-01-25 04:17:55 +00002 * Copyright (C) 2000,2001,2002,2003,2004,2005 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[]);
Josh Coalson7446e182005-01-26 04:04:38 +0000361 void (*local_lpc_compute_residual_from_qlp_coefficients)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
362 void (*local_lpc_compute_residual_from_qlp_coefficients_64bit)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
363 void (*local_lpc_compute_residual_from_qlp_coefficients_16bit)(const FLAC__int32 *data, unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 residual[]);
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
Josh Coalson8ddf7fb2004-12-30 00:58:50 +0000920 * that metadata_has_vorbis_comment is true it will have already
Josh Coalson69cfda72004-09-10 00:38:21 +0000921 * 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;
Josh Coalson47c7b142005-01-29 06:08:58 +0000997#ifndef FLAC__MANDATORY_VERIFY_WHILE_ENCODING
Josh Coalsond86e03b2002-08-03 21:56:15 +0000998 encoder->protected_->verify = value;
Josh Coalson47c7b142005-01-29 06:08:58 +0000999#endif
Josh Coalsond86e03b2002-08-03 21:56:15 +00001000 return true;
1001}
1002
Josh Coalson6afed9f2002-10-16 22:29:47 +00001003FLAC_API FLAC__bool FLAC__stream_encoder_set_streamable_subset(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001004{
Josh Coalson92031602002-07-24 06:02:11 +00001005 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001006 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001007 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001008 encoder->protected_->streamable_subset = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001009 return true;
1010}
1011
Josh Coalson6afed9f2002-10-16 22:29:47 +00001012FLAC_API FLAC__bool FLAC__stream_encoder_set_do_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001013{
Josh Coalson92031602002-07-24 06:02:11 +00001014 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001015 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001016 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001017 encoder->protected_->do_mid_side_stereo = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001018 return true;
1019}
1020
Josh Coalson6afed9f2002-10-16 22:29:47 +00001021FLAC_API FLAC__bool FLAC__stream_encoder_set_loose_mid_side_stereo(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001022{
Josh Coalson92031602002-07-24 06:02:11 +00001023 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001024 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001025 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001026 encoder->protected_->loose_mid_side_stereo = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001027 return true;
1028}
1029
Josh Coalson6afed9f2002-10-16 22:29:47 +00001030FLAC_API FLAC__bool FLAC__stream_encoder_set_channels(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001031{
Josh Coalson92031602002-07-24 06:02:11 +00001032 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001033 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001034 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001035 encoder->protected_->channels = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001036 return true;
1037}
1038
Josh Coalson6afed9f2002-10-16 22:29:47 +00001039FLAC_API FLAC__bool FLAC__stream_encoder_set_bits_per_sample(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001040{
Josh Coalson92031602002-07-24 06:02:11 +00001041 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001042 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001043 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001044 encoder->protected_->bits_per_sample = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001045 return true;
1046}
1047
Josh Coalson6afed9f2002-10-16 22:29:47 +00001048FLAC_API FLAC__bool FLAC__stream_encoder_set_sample_rate(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001049{
Josh Coalson92031602002-07-24 06:02:11 +00001050 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001051 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001052 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001053 encoder->protected_->sample_rate = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001054 return true;
1055}
1056
Josh Coalson6afed9f2002-10-16 22:29:47 +00001057FLAC_API FLAC__bool FLAC__stream_encoder_set_blocksize(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001058{
Josh Coalson92031602002-07-24 06:02:11 +00001059 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001060 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001061 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001062 encoder->protected_->blocksize = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001063 return true;
1064}
1065
Josh Coalson6afed9f2002-10-16 22:29:47 +00001066FLAC_API FLAC__bool FLAC__stream_encoder_set_max_lpc_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001067{
Josh Coalson92031602002-07-24 06:02:11 +00001068 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001069 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001070 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001071 encoder->protected_->max_lpc_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001072 return true;
1073}
1074
Josh Coalson6afed9f2002-10-16 22:29:47 +00001075FLAC_API FLAC__bool FLAC__stream_encoder_set_qlp_coeff_precision(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001076{
Josh Coalson92031602002-07-24 06:02:11 +00001077 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001078 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001079 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001080 encoder->protected_->qlp_coeff_precision = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001081 return true;
1082}
1083
Josh Coalson6afed9f2002-10-16 22:29:47 +00001084FLAC_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 +00001085{
Josh Coalson92031602002-07-24 06:02:11 +00001086 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001087 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001088 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001089 encoder->protected_->do_qlp_coeff_prec_search = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001090 return true;
1091}
1092
Josh Coalson6afed9f2002-10-16 22:29:47 +00001093FLAC_API FLAC__bool FLAC__stream_encoder_set_do_escape_coding(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson8395d022001-07-12 21:25:22 +00001094{
Josh Coalson92031602002-07-24 06:02:11 +00001095 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001096 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson8395d022001-07-12 21:25:22 +00001097 return false;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001098#if 0
1099 /*@@@ deprecated: */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001100 encoder->protected_->do_escape_coding = value;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001101#else
1102 (void)value;
1103#endif
Josh Coalson8395d022001-07-12 21:25:22 +00001104 return true;
1105}
1106
Josh Coalson6afed9f2002-10-16 22:29:47 +00001107FLAC_API FLAC__bool FLAC__stream_encoder_set_do_exhaustive_model_search(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalson00e53872001-06-16 07:32:25 +00001108{
Josh Coalson92031602002-07-24 06:02:11 +00001109 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001110 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001111 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001112 encoder->protected_->do_exhaustive_model_search = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001113 return true;
1114}
1115
Josh Coalson6afed9f2002-10-16 22:29:47 +00001116FLAC_API FLAC__bool FLAC__stream_encoder_set_min_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001117{
Josh Coalson92031602002-07-24 06:02:11 +00001118 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001119 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001120 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001121 encoder->protected_->min_residual_partition_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001122 return true;
1123}
1124
Josh Coalson6afed9f2002-10-16 22:29:47 +00001125FLAC_API FLAC__bool FLAC__stream_encoder_set_max_residual_partition_order(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001126{
Josh Coalson92031602002-07-24 06:02:11 +00001127 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001128 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001129 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001130 encoder->protected_->max_residual_partition_order = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001131 return true;
1132}
1133
Josh Coalson6afed9f2002-10-16 22:29:47 +00001134FLAC_API FLAC__bool FLAC__stream_encoder_set_rice_parameter_search_dist(FLAC__StreamEncoder *encoder, unsigned value)
Josh Coalson00e53872001-06-16 07:32:25 +00001135{
Josh Coalson92031602002-07-24 06:02:11 +00001136 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001137 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001138 return false;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001139#if 0
1140 /*@@@ deprecated: */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001141 encoder->protected_->rice_parameter_search_dist = value;
Josh Coalson680e3aa2002-08-01 07:32:17 +00001142#else
1143 (void)value;
1144#endif
Josh Coalson00e53872001-06-16 07:32:25 +00001145 return true;
1146}
1147
Josh Coalson6afed9f2002-10-16 22:29:47 +00001148FLAC_API FLAC__bool FLAC__stream_encoder_set_total_samples_estimate(FLAC__StreamEncoder *encoder, FLAC__uint64 value)
Josh Coalson00e53872001-06-16 07:32:25 +00001149{
Josh Coalson92031602002-07-24 06:02:11 +00001150 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001151 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001152 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001153 encoder->protected_->total_samples_estimate = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001154 return true;
1155}
1156
Josh Coalson6afed9f2002-10-16 22:29:47 +00001157FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata(FLAC__StreamEncoder *encoder, FLAC__StreamMetadata **metadata, unsigned num_blocks)
Josh Coalson00e53872001-06-16 07:32:25 +00001158{
Josh Coalson92031602002-07-24 06:02:11 +00001159 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001160 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001161 return false;
Josh Coalson66075c12002-06-01 05:39:38 +00001162 encoder->protected_->metadata = metadata;
1163 encoder->protected_->num_metadata_blocks = num_blocks;
Josh Coalson00e53872001-06-16 07:32:25 +00001164 return true;
1165}
1166
Josh Coalson6afed9f2002-10-16 22:29:47 +00001167FLAC_API FLAC__bool FLAC__stream_encoder_set_write_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderWriteCallback value)
Josh Coalson00e53872001-06-16 07:32:25 +00001168{
Josh Coalson92031602002-07-24 06:02:11 +00001169 FLAC__ASSERT(0 != encoder);
1170 FLAC__ASSERT(0 != value);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001171 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001172 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001173 encoder->private_->write_callback = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001174 return true;
1175}
1176
Josh Coalson6afed9f2002-10-16 22:29:47 +00001177FLAC_API FLAC__bool FLAC__stream_encoder_set_metadata_callback(FLAC__StreamEncoder *encoder, FLAC__StreamEncoderMetadataCallback value)
Josh Coalson00e53872001-06-16 07:32:25 +00001178{
Josh Coalson92031602002-07-24 06:02:11 +00001179 FLAC__ASSERT(0 != encoder);
1180 FLAC__ASSERT(0 != value);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001181 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001182 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001183 encoder->private_->metadata_callback = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001184 return true;
1185}
1186
Josh Coalson6afed9f2002-10-16 22:29:47 +00001187FLAC_API FLAC__bool FLAC__stream_encoder_set_client_data(FLAC__StreamEncoder *encoder, void *value)
Josh Coalson00e53872001-06-16 07:32:25 +00001188{
Josh Coalson92031602002-07-24 06:02:11 +00001189 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001190 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
Josh Coalson00e53872001-06-16 07:32:25 +00001191 return false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001192 encoder->private_->client_data = value;
Josh Coalson00e53872001-06-16 07:32:25 +00001193 return true;
1194}
1195
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001196/*
1197 * These three functions are not static, but not publically exposed in
1198 * include/FLAC/ either. They are used by the test suite.
1199 */
Josh Coalson6afed9f2002-10-16 22:29:47 +00001200FLAC_API FLAC__bool FLAC__stream_encoder_disable_constant_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001201{
1202 FLAC__ASSERT(0 != encoder);
1203 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1204 return false;
1205 encoder->private_->disable_constant_subframes = value;
1206 return true;
1207}
1208
Josh Coalson6afed9f2002-10-16 22:29:47 +00001209FLAC_API FLAC__bool FLAC__stream_encoder_disable_fixed_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001210{
1211 FLAC__ASSERT(0 != encoder);
1212 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1213 return false;
1214 encoder->private_->disable_fixed_subframes = value;
1215 return true;
1216}
1217
Josh Coalson6afed9f2002-10-16 22:29:47 +00001218FLAC_API FLAC__bool FLAC__stream_encoder_disable_verbatim_subframes(FLAC__StreamEncoder *encoder, FLAC__bool value)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001219{
1220 FLAC__ASSERT(0 != encoder);
1221 if(encoder->protected_->state != FLAC__STREAM_ENCODER_UNINITIALIZED)
1222 return false;
1223 encoder->private_->disable_verbatim_subframes = value;
1224 return true;
1225}
1226
Josh Coalson6afed9f2002-10-16 22:29:47 +00001227FLAC_API FLAC__StreamEncoderState FLAC__stream_encoder_get_state(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001228{
Josh Coalson92031602002-07-24 06:02:11 +00001229 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001230 return encoder->protected_->state;
Josh Coalson0a15c142001-06-13 17:59:57 +00001231}
1232
Josh Coalson6afed9f2002-10-16 22:29:47 +00001233FLAC_API FLAC__StreamDecoderState FLAC__stream_encoder_get_verify_decoder_state(const FLAC__StreamEncoder *encoder)
Josh Coalsond86e03b2002-08-03 21:56:15 +00001234{
1235 FLAC__ASSERT(0 != encoder);
1236 if(encoder->protected_->verify)
1237 return FLAC__stream_decoder_get_state(encoder->private_->verify.decoder);
1238 else
1239 return FLAC__STREAM_DECODER_UNINITIALIZED;
1240}
1241
Josh Coalson02954222002-11-08 06:16:31 +00001242FLAC_API const char *FLAC__stream_encoder_get_resolved_state_string(const FLAC__StreamEncoder *encoder)
1243{
1244 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR)
1245 return FLAC__StreamEncoderStateString[encoder->protected_->state];
1246 else
Josh Coalson807140d2003-09-24 22:10:51 +00001247 return FLAC__stream_decoder_get_resolved_state_string(encoder->private_->verify.decoder);
Josh Coalson02954222002-11-08 06:16:31 +00001248}
1249
Josh Coalson6afed9f2002-10-16 22:29:47 +00001250FLAC_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 +00001251{
1252 FLAC__ASSERT(0 != encoder);
1253 if(0 != absolute_sample)
1254 *absolute_sample = encoder->private_->verify.error_stats.absolute_sample;
1255 if(0 != frame_number)
1256 *frame_number = encoder->private_->verify.error_stats.frame_number;
1257 if(0 != channel)
1258 *channel = encoder->private_->verify.error_stats.channel;
1259 if(0 != sample)
1260 *sample = encoder->private_->verify.error_stats.sample;
1261 if(0 != expected)
1262 *expected = encoder->private_->verify.error_stats.expected;
1263 if(0 != got)
1264 *got = encoder->private_->verify.error_stats.got;
1265}
1266
Josh Coalson6afed9f2002-10-16 22:29:47 +00001267FLAC_API FLAC__bool FLAC__stream_encoder_get_verify(const FLAC__StreamEncoder *encoder)
Josh Coalsond86e03b2002-08-03 21:56:15 +00001268{
1269 FLAC__ASSERT(0 != encoder);
1270 return encoder->protected_->verify;
1271}
1272
Josh Coalson6afed9f2002-10-16 22:29:47 +00001273FLAC_API FLAC__bool FLAC__stream_encoder_get_streamable_subset(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001274{
Josh Coalson92031602002-07-24 06:02:11 +00001275 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001276 return encoder->protected_->streamable_subset;
Josh Coalson0a15c142001-06-13 17:59:57 +00001277}
1278
Josh Coalson6afed9f2002-10-16 22:29:47 +00001279FLAC_API FLAC__bool FLAC__stream_encoder_get_do_mid_side_stereo(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001280{
Josh Coalson92031602002-07-24 06:02:11 +00001281 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001282 return encoder->protected_->do_mid_side_stereo;
Josh Coalson0a15c142001-06-13 17:59:57 +00001283}
1284
Josh Coalson6afed9f2002-10-16 22:29:47 +00001285FLAC_API FLAC__bool FLAC__stream_encoder_get_loose_mid_side_stereo(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001286{
Josh Coalson92031602002-07-24 06:02:11 +00001287 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001288 return encoder->protected_->loose_mid_side_stereo;
Josh Coalson0a15c142001-06-13 17:59:57 +00001289}
1290
Josh Coalson6afed9f2002-10-16 22:29:47 +00001291FLAC_API unsigned FLAC__stream_encoder_get_channels(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001292{
Josh Coalson92031602002-07-24 06:02:11 +00001293 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001294 return encoder->protected_->channels;
Josh Coalson0a15c142001-06-13 17:59:57 +00001295}
1296
Josh Coalson6afed9f2002-10-16 22:29:47 +00001297FLAC_API unsigned FLAC__stream_encoder_get_bits_per_sample(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001298{
Josh Coalson92031602002-07-24 06:02:11 +00001299 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001300 return encoder->protected_->bits_per_sample;
Josh Coalson0a15c142001-06-13 17:59:57 +00001301}
1302
Josh Coalson6afed9f2002-10-16 22:29:47 +00001303FLAC_API unsigned FLAC__stream_encoder_get_sample_rate(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001304{
Josh Coalson92031602002-07-24 06:02:11 +00001305 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001306 return encoder->protected_->sample_rate;
Josh Coalson0a15c142001-06-13 17:59:57 +00001307}
1308
Josh Coalson6afed9f2002-10-16 22:29:47 +00001309FLAC_API unsigned FLAC__stream_encoder_get_blocksize(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001310{
Josh Coalson92031602002-07-24 06:02:11 +00001311 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001312 return encoder->protected_->blocksize;
Josh Coalson0a15c142001-06-13 17:59:57 +00001313}
1314
Josh Coalson6afed9f2002-10-16 22:29:47 +00001315FLAC_API unsigned FLAC__stream_encoder_get_max_lpc_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001316{
Josh Coalson92031602002-07-24 06:02:11 +00001317 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001318 return encoder->protected_->max_lpc_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001319}
1320
Josh Coalson6afed9f2002-10-16 22:29:47 +00001321FLAC_API unsigned FLAC__stream_encoder_get_qlp_coeff_precision(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001322{
Josh Coalson92031602002-07-24 06:02:11 +00001323 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001324 return encoder->protected_->qlp_coeff_precision;
Josh Coalson0a15c142001-06-13 17:59:57 +00001325}
1326
Josh Coalson6afed9f2002-10-16 22:29:47 +00001327FLAC_API FLAC__bool FLAC__stream_encoder_get_do_qlp_coeff_prec_search(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001328{
Josh Coalson92031602002-07-24 06:02:11 +00001329 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001330 return encoder->protected_->do_qlp_coeff_prec_search;
Josh Coalson0a15c142001-06-13 17:59:57 +00001331}
1332
Josh Coalson6afed9f2002-10-16 22:29:47 +00001333FLAC_API FLAC__bool FLAC__stream_encoder_get_do_escape_coding(const FLAC__StreamEncoder *encoder)
Josh Coalson8395d022001-07-12 21:25:22 +00001334{
Josh Coalson92031602002-07-24 06:02:11 +00001335 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001336 return encoder->protected_->do_escape_coding;
Josh Coalson8395d022001-07-12 21:25:22 +00001337}
1338
Josh Coalson6afed9f2002-10-16 22:29:47 +00001339FLAC_API FLAC__bool FLAC__stream_encoder_get_do_exhaustive_model_search(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001340{
Josh Coalson92031602002-07-24 06:02:11 +00001341 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001342 return encoder->protected_->do_exhaustive_model_search;
Josh Coalson0a15c142001-06-13 17:59:57 +00001343}
1344
Josh Coalson6afed9f2002-10-16 22:29:47 +00001345FLAC_API unsigned FLAC__stream_encoder_get_min_residual_partition_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001346{
Josh Coalson92031602002-07-24 06:02:11 +00001347 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001348 return encoder->protected_->min_residual_partition_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001349}
1350
Josh Coalson6afed9f2002-10-16 22:29:47 +00001351FLAC_API unsigned FLAC__stream_encoder_get_max_residual_partition_order(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001352{
Josh Coalson92031602002-07-24 06:02:11 +00001353 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001354 return encoder->protected_->max_residual_partition_order;
Josh Coalson0a15c142001-06-13 17:59:57 +00001355}
1356
Josh Coalson6afed9f2002-10-16 22:29:47 +00001357FLAC_API unsigned FLAC__stream_encoder_get_rice_parameter_search_dist(const FLAC__StreamEncoder *encoder)
Josh Coalson0a15c142001-06-13 17:59:57 +00001358{
Josh Coalson92031602002-07-24 06:02:11 +00001359 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001360 return encoder->protected_->rice_parameter_search_dist;
Josh Coalson0a15c142001-06-13 17:59:57 +00001361}
1362
Josh Coalson6afed9f2002-10-16 22:29:47 +00001363FLAC_API FLAC__uint64 FLAC__stream_encoder_get_total_samples_estimate(const FLAC__StreamEncoder *encoder)
Josh Coalson3a7b2c92002-08-02 07:38:20 +00001364{
1365 FLAC__ASSERT(0 != encoder);
1366 return encoder->protected_->total_samples_estimate;
1367}
1368
Josh Coalson6afed9f2002-10-16 22:29:47 +00001369FLAC_API FLAC__bool FLAC__stream_encoder_process(FLAC__StreamEncoder *encoder, const FLAC__int32 * const buffer[], unsigned samples)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001370{
1371 unsigned i, j, channel;
Josh Coalson77e3f312001-06-23 03:03:24 +00001372 FLAC__int32 x, mid, side;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001373 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001374
Josh Coalsonf1eff452002-07-31 07:05:33 +00001375 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001376 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001377
1378 j = 0;
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001379 /*
1380 * we have several flavors of the same basic loop, optimized for
1381 * different conditions:
1382 */
1383 if(encoder->protected_->max_lpc_order > 0) {
1384 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1385 /*
1386 * stereo coding: unroll channel loop
1387 * with LPC: calculate floating point version of signal
1388 */
1389 do {
1390 if(encoder->protected_->verify)
1391 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
Josh Coalsond86e03b2002-08-03 21:56:15 +00001392
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001393 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1394 x = mid = side = buffer[0][j];
1395 encoder->private_->integer_signal[0][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001396#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001397 encoder->private_->real_signal[0][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001398#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001399 x = buffer[1][j];
1400 encoder->private_->integer_signal[1][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001401#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001402 encoder->private_->real_signal[1][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001403#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001404 mid += x;
1405 side -= x;
1406 mid >>= 1; /* NOTE: not the same as 'mid = (buffer[0][j] + buffer[1][j]) / 2' ! */
1407 encoder->private_->integer_signal_mid_side[1][i] = side;
1408 encoder->private_->integer_signal_mid_side[0][i] = mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001409#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001410 encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1411 encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001412#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001413 encoder->private_->current_sample_number++;
1414 }
1415 if(i == blocksize) {
1416 if(!process_frame_(encoder, false)) /* false => not last frame */
1417 return false;
1418 }
1419 } while(j < samples);
1420 }
1421 else {
1422 /*
1423 * independent channel coding: buffer each channel in inner loop
1424 * with LPC: calculate floating point version of signal
1425 */
1426 do {
1427 if(encoder->protected_->verify)
1428 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1429
1430 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1431 for(channel = 0; channel < channels; channel++) {
1432 x = buffer[channel][j];
1433 encoder->private_->integer_signal[channel][i] = x;
1434#ifndef FLAC__INTEGER_ONLY_LIBRARY
1435 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
1436#endif
1437 }
1438 encoder->private_->current_sample_number++;
1439 }
1440 if(i == blocksize) {
1441 if(!process_frame_(encoder, false)) /* false => not last frame */
1442 return false;
1443 }
1444 } while(j < samples);
1445 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001446 }
1447 else {
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001448 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1449 /*
1450 * stereo coding: unroll channel loop
1451 * without LPC: no need to calculate floating point version of signal
1452 */
1453 do {
1454 if(encoder->protected_->verify)
1455 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
Josh Coalsond86e03b2002-08-03 21:56:15 +00001456
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001457 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1458 encoder->private_->integer_signal[0][i] = mid = side = buffer[0][j];
1459 x = buffer[1][j];
1460 encoder->private_->integer_signal[1][i] = x;
1461 mid += x;
1462 side -= x;
1463 mid >>= 1; /* NOTE: not the same as 'mid = (buffer[0][j] + buffer[1][j]) / 2' ! */
1464 encoder->private_->integer_signal_mid_side[1][i] = side;
1465 encoder->private_->integer_signal_mid_side[0][i] = mid;
1466 encoder->private_->current_sample_number++;
Josh Coalsonaa255362001-05-31 06:17:41 +00001467 }
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001468 if(i == blocksize) {
1469 if(!process_frame_(encoder, false)) /* false => not last frame */
1470 return false;
1471 }
1472 } while(j < samples);
1473 }
1474 else {
1475 /*
1476 * independent channel coding: buffer each channel in inner loop
1477 * without LPC: no need to calculate floating point version of signal
1478 */
1479 do {
1480 if(encoder->protected_->verify)
1481 append_to_verify_fifo_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1482
1483 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1484 for(channel = 0; channel < channels; channel++)
1485 encoder->private_->integer_signal[channel][i] = buffer[channel][j];
1486 encoder->private_->current_sample_number++;
1487 }
1488 if(i == blocksize) {
1489 if(!process_frame_(encoder, false)) /* false => not last frame */
1490 return false;
1491 }
1492 } while(j < samples);
1493 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001494 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001495
1496 return true;
1497}
1498
Josh Coalson6afed9f2002-10-16 22:29:47 +00001499FLAC_API FLAC__bool FLAC__stream_encoder_process_interleaved(FLAC__StreamEncoder *encoder, const FLAC__int32 buffer[], unsigned samples)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001500{
1501 unsigned i, j, k, channel;
Josh Coalson77e3f312001-06-23 03:03:24 +00001502 FLAC__int32 x, mid, side;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001503 const unsigned channels = encoder->protected_->channels, blocksize = encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001504
Josh Coalsonf1eff452002-07-31 07:05:33 +00001505 FLAC__ASSERT(0 != encoder);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001506 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001507
1508 j = k = 0;
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001509 /*
1510 * we have several flavors of the same basic loop, optimized for
1511 * different conditions:
1512 */
1513 if(encoder->protected_->max_lpc_order > 0) {
1514 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1515 /*
1516 * stereo coding: unroll channel loop
1517 * with LPC: calculate floating point version of signal
1518 */
1519 do {
1520 if(encoder->protected_->verify)
1521 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
Josh Coalsond86e03b2002-08-03 21:56:15 +00001522
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001523 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1524 x = mid = side = buffer[k++];
1525 encoder->private_->integer_signal[0][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001526#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001527 encoder->private_->real_signal[0][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001528#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001529 x = buffer[k++];
1530 encoder->private_->integer_signal[1][i] = x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001531#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001532 encoder->private_->real_signal[1][i] = (FLAC__real)x;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001533#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001534 mid += x;
1535 side -= x;
1536 mid >>= 1; /* NOTE: not the same as 'mid = (left + right) / 2' ! */
1537 encoder->private_->integer_signal_mid_side[1][i] = side;
1538 encoder->private_->integer_signal_mid_side[0][i] = mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001539#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001540 encoder->private_->real_signal_mid_side[1][i] = (FLAC__real)side;
1541 encoder->private_->real_signal_mid_side[0][i] = (FLAC__real)mid;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001542#endif
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001543 encoder->private_->current_sample_number++;
1544 }
1545 if(i == blocksize) {
1546 if(!process_frame_(encoder, false)) /* false => not last frame */
1547 return false;
1548 }
1549 } while(j < samples);
1550 }
1551 else {
1552 /*
1553 * independent channel coding: buffer each channel in inner loop
1554 * with LPC: calculate floating point version of signal
1555 */
1556 do {
1557 if(encoder->protected_->verify)
1558 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1559
1560 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1561 for(channel = 0; channel < channels; channel++) {
1562 x = buffer[k++];
1563 encoder->private_->integer_signal[channel][i] = x;
1564#ifndef FLAC__INTEGER_ONLY_LIBRARY
1565 encoder->private_->real_signal[channel][i] = (FLAC__real)x;
1566#endif
1567 }
1568 encoder->private_->current_sample_number++;
1569 }
1570 if(i == blocksize) {
1571 if(!process_frame_(encoder, false)) /* false => not last frame */
1572 return false;
1573 }
1574 } while(j < samples);
1575 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001576 }
1577 else {
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001578 if(encoder->protected_->do_mid_side_stereo && channels == 2) {
1579 /*
1580 * stereo coding: unroll channel loop
1581 * without LPC: no need to calculate floating point version of signal
1582 */
1583 do {
1584 if(encoder->protected_->verify)
1585 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
Josh Coalsond86e03b2002-08-03 21:56:15 +00001586
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001587 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1588 encoder->private_->integer_signal[0][i] = mid = side = buffer[k++];
Josh Coalson57ba6f42002-06-07 05:27:37 +00001589 x = buffer[k++];
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001590 encoder->private_->integer_signal[1][i] = x;
1591 mid += x;
1592 side -= x;
1593 mid >>= 1; /* NOTE: not the same as 'mid = (left + right) / 2' ! */
1594 encoder->private_->integer_signal_mid_side[1][i] = side;
1595 encoder->private_->integer_signal_mid_side[0][i] = mid;
1596 encoder->private_->current_sample_number++;
Josh Coalsonaa255362001-05-31 06:17:41 +00001597 }
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001598 if(i == blocksize) {
1599 if(!process_frame_(encoder, false)) /* false => not last frame */
1600 return false;
1601 }
1602 } while(j < samples);
1603 }
1604 else {
1605 /*
1606 * independent channel coding: buffer each channel in inner loop
1607 * without LPC: no need to calculate floating point version of signal
1608 */
1609 do {
1610 if(encoder->protected_->verify)
1611 append_to_verify_fifo_interleaved_(&encoder->private_->verify.input_fifo, buffer, j, channels, min(blocksize-encoder->private_->current_sample_number, samples-j));
1612
1613 for(i = encoder->private_->current_sample_number; i < blocksize && j < samples; i++, j++) {
1614 for(channel = 0; channel < channels; channel++)
1615 encoder->private_->integer_signal[channel][i] = buffer[k++];
1616 encoder->private_->current_sample_number++;
1617 }
1618 if(i == blocksize) {
1619 if(!process_frame_(encoder, false)) /* false => not last frame */
1620 return false;
1621 }
1622 } while(j < samples);
1623 }
Josh Coalsonaa255362001-05-31 06:17:41 +00001624 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001625
1626 return true;
1627}
1628
Josh Coalsonf1eff452002-07-31 07:05:33 +00001629/***********************************************************************
1630 *
1631 * Private class methods
1632 *
1633 ***********************************************************************/
1634
1635void set_defaults_(FLAC__StreamEncoder *encoder)
Josh Coalson92031602002-07-24 06:02:11 +00001636{
1637 FLAC__ASSERT(0 != encoder);
1638
Josh Coalson47c7b142005-01-29 06:08:58 +00001639#ifdef FLAC__MANDATORY_VERIFY_WHILE_ENCODING
1640 encoder->protected_->verify = true;
1641#else
Josh Coalsond86e03b2002-08-03 21:56:15 +00001642 encoder->protected_->verify = false;
Josh Coalson47c7b142005-01-29 06:08:58 +00001643#endif
Josh Coalson92031602002-07-24 06:02:11 +00001644 encoder->protected_->streamable_subset = true;
1645 encoder->protected_->do_mid_side_stereo = false;
1646 encoder->protected_->loose_mid_side_stereo = false;
1647 encoder->protected_->channels = 2;
1648 encoder->protected_->bits_per_sample = 16;
1649 encoder->protected_->sample_rate = 44100;
1650 encoder->protected_->blocksize = 1152;
1651 encoder->protected_->max_lpc_order = 0;
1652 encoder->protected_->qlp_coeff_precision = 0;
1653 encoder->protected_->do_qlp_coeff_prec_search = false;
1654 encoder->protected_->do_exhaustive_model_search = false;
1655 encoder->protected_->do_escape_coding = false;
1656 encoder->protected_->min_residual_partition_order = 0;
1657 encoder->protected_->max_residual_partition_order = 0;
1658 encoder->protected_->rice_parameter_search_dist = 0;
1659 encoder->protected_->total_samples_estimate = 0;
1660 encoder->protected_->metadata = 0;
1661 encoder->protected_->num_metadata_blocks = 0;
1662
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00001663 encoder->private_->disable_constant_subframes = false;
1664 encoder->private_->disable_fixed_subframes = false;
1665 encoder->private_->disable_verbatim_subframes = false;
Josh Coalson92031602002-07-24 06:02:11 +00001666 encoder->private_->write_callback = 0;
1667 encoder->private_->metadata_callback = 0;
1668 encoder->private_->client_data = 0;
1669}
1670
Josh Coalsonf1eff452002-07-31 07:05:33 +00001671void free_(FLAC__StreamEncoder *encoder)
Josh Coalson639aeb02002-07-25 05:38:23 +00001672{
1673 unsigned i, channel;
1674
Josh Coalsonf1eff452002-07-31 07:05:33 +00001675 FLAC__ASSERT(0 != encoder);
Josh Coalson639aeb02002-07-25 05:38:23 +00001676 for(i = 0; i < encoder->protected_->channels; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001677 if(0 != encoder->private_->integer_signal_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001678 free(encoder->private_->integer_signal_unaligned[i]);
1679 encoder->private_->integer_signal_unaligned[i] = 0;
1680 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001681#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonf1eff452002-07-31 07:05:33 +00001682 if(0 != encoder->private_->real_signal_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001683 free(encoder->private_->real_signal_unaligned[i]);
1684 encoder->private_->real_signal_unaligned[i] = 0;
1685 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001686#endif
Josh Coalson639aeb02002-07-25 05:38:23 +00001687 }
1688 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001689 if(0 != encoder->private_->integer_signal_mid_side_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001690 free(encoder->private_->integer_signal_mid_side_unaligned[i]);
1691 encoder->private_->integer_signal_mid_side_unaligned[i] = 0;
1692 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001693#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonf1eff452002-07-31 07:05:33 +00001694 if(0 != encoder->private_->real_signal_mid_side_unaligned[i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001695 free(encoder->private_->real_signal_mid_side_unaligned[i]);
1696 encoder->private_->real_signal_mid_side_unaligned[i] = 0;
1697 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001698#endif
Josh Coalson639aeb02002-07-25 05:38:23 +00001699 }
1700 for(channel = 0; channel < encoder->protected_->channels; channel++) {
1701 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001702 if(0 != encoder->private_->residual_workspace_unaligned[channel][i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001703 free(encoder->private_->residual_workspace_unaligned[channel][i]);
1704 encoder->private_->residual_workspace_unaligned[channel][i] = 0;
1705 }
1706 }
1707 }
1708 for(channel = 0; channel < 2; channel++) {
1709 for(i = 0; i < 2; i++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00001710 if(0 != encoder->private_->residual_workspace_mid_side_unaligned[channel][i]) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001711 free(encoder->private_->residual_workspace_mid_side_unaligned[channel][i]);
1712 encoder->private_->residual_workspace_mid_side_unaligned[channel][i] = 0;
1713 }
1714 }
1715 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001716 if(0 != encoder->private_->abs_residual_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001717 free(encoder->private_->abs_residual_unaligned);
1718 encoder->private_->abs_residual_unaligned = 0;
1719 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001720 if(0 != encoder->private_->abs_residual_partition_sums_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001721 free(encoder->private_->abs_residual_partition_sums_unaligned);
1722 encoder->private_->abs_residual_partition_sums_unaligned = 0;
1723 }
Josh Coalsonf1eff452002-07-31 07:05:33 +00001724 if(0 != encoder->private_->raw_bits_per_partition_unaligned) {
Josh Coalson639aeb02002-07-25 05:38:23 +00001725 free(encoder->private_->raw_bits_per_partition_unaligned);
1726 encoder->private_->raw_bits_per_partition_unaligned = 0;
1727 }
Josh Coalsond86e03b2002-08-03 21:56:15 +00001728 if(encoder->protected_->verify) {
1729 for(i = 0; i < encoder->protected_->channels; i++) {
1730 if(0 != encoder->private_->verify.input_fifo.data[i]) {
1731 free(encoder->private_->verify.input_fifo.data[i]);
1732 encoder->private_->verify.input_fifo.data[i] = 0;
1733 }
1734 }
1735 }
Josh Coalson639aeb02002-07-25 05:38:23 +00001736 FLAC__bitbuffer_free(encoder->private_->frame);
1737}
1738
Josh Coalsonf1eff452002-07-31 07:05:33 +00001739FLAC__bool resize_buffers_(FLAC__StreamEncoder *encoder, unsigned new_size)
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001740{
Josh Coalson77e3f312001-06-23 03:03:24 +00001741 FLAC__bool ok;
Josh Coalson0a15c142001-06-13 17:59:57 +00001742 unsigned i, channel;
1743
1744 FLAC__ASSERT(new_size > 0);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001745 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
1746 FLAC__ASSERT(encoder->private_->current_sample_number == 0);
Josh Coalson0a15c142001-06-13 17:59:57 +00001747
1748 /* To avoid excessive malloc'ing, we only grow the buffer; no shrinking. */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001749 if(new_size <= encoder->private_->input_capacity)
Josh Coalson0a15c142001-06-13 17:59:57 +00001750 return true;
1751
1752 ok = true;
Josh Coalson8395d022001-07-12 21:25:22 +00001753
Josh Coalsonc9c0d132002-10-04 05:29:05 +00001754 /* WATCHOUT: FLAC__lpc_compute_residual_from_qlp_coefficients_asm_ia32_mmx()
1755 * requires that the input arrays (in our case the integer signals)
1756 * have a buffer of up to 3 zeroes in front (at negative indices) for
1757 * alignment purposes; we use 4 to keep the data well-aligned.
1758 */
Josh Coalson8395d022001-07-12 21:25:22 +00001759
Josh Coalsonfa697a92001-08-16 20:07:29 +00001760 for(i = 0; ok && i < encoder->protected_->channels; i++) {
1761 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 +00001762#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001763 if(encoder->protected_->max_lpc_order > 0)
1764 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 +00001765#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001766 memset(encoder->private_->integer_signal[i], 0, sizeof(FLAC__int32)*4);
1767 encoder->private_->integer_signal[i] += 4;
Josh Coalson0a15c142001-06-13 17:59:57 +00001768 }
1769 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001770 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 +00001771#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalsonc549f0f2004-12-30 03:47:49 +00001772 if(encoder->protected_->max_lpc_order > 0)
1773 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 +00001774#endif
Josh Coalsonfa697a92001-08-16 20:07:29 +00001775 memset(encoder->private_->integer_signal_mid_side[i], 0, sizeof(FLAC__int32)*4);
1776 encoder->private_->integer_signal_mid_side[i] += 4;
Josh Coalson0a15c142001-06-13 17:59:57 +00001777 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001778 for(channel = 0; ok && channel < encoder->protected_->channels; channel++) {
Josh Coalson0a15c142001-06-13 17:59:57 +00001779 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001780 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 +00001781 }
1782 }
1783 for(channel = 0; ok && channel < 2; channel++) {
1784 for(i = 0; ok && i < 2; i++) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001785 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 +00001786 }
1787 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001788 ok = ok && FLAC__memory_alloc_aligned_uint32_array(new_size, &encoder->private_->abs_residual_unaligned, &encoder->private_->abs_residual);
1789 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 */
1790 ok = ok && FLAC__memory_alloc_aligned_uint64_array(new_size * 2, &encoder->private_->abs_residual_partition_sums_unaligned, &encoder->private_->abs_residual_partition_sums);
1791 if(encoder->protected_->do_escape_coding)
1792 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 +00001793
1794 if(ok)
Josh Coalsonfa697a92001-08-16 20:07:29 +00001795 encoder->private_->input_capacity = new_size;
Josh Coalson0a15c142001-06-13 17:59:57 +00001796 else
Josh Coalsonfa697a92001-08-16 20:07:29 +00001797 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalson0a15c142001-06-13 17:59:57 +00001798
1799 return ok;
1800}
1801
Josh Coalsond86e03b2002-08-03 21:56:15 +00001802FLAC__bool write_bitbuffer_(FLAC__StreamEncoder *encoder, unsigned samples)
Josh Coalson5c491a12002-08-01 06:39:40 +00001803{
1804 const FLAC__byte *buffer;
1805 unsigned bytes;
1806
1807 FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1808
1809 FLAC__bitbuffer_get_buffer(encoder->private_->frame, &buffer, &bytes);
1810
Josh Coalsond86e03b2002-08-03 21:56:15 +00001811 if(encoder->protected_->verify) {
1812 encoder->private_->verify.output.data = buffer;
1813 encoder->private_->verify.output.bytes = bytes;
1814 if(encoder->private_->verify.state_hint == ENCODER_IN_MAGIC) {
1815 encoder->private_->verify.needs_magic_hack = true;
1816 }
1817 else {
1818 if(!FLAC__stream_decoder_process_single(encoder->private_->verify.decoder)) {
1819 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1820 if(encoder->protected_->state != FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA)
1821 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
1822 return false;
1823 }
1824 }
1825 }
1826
1827 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 +00001828 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
Josh Coalsond86e03b2002-08-03 21:56:15 +00001829 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_WRITING;
Josh Coalson5c491a12002-08-01 06:39:40 +00001830 return false;
Josh Coalsond86e03b2002-08-03 21:56:15 +00001831 }
Josh Coalson5c491a12002-08-01 06:39:40 +00001832
1833 FLAC__bitbuffer_release_buffer(encoder->private_->frame);
1834
Josh Coalsond86e03b2002-08-03 21:56:15 +00001835 if(samples > 0) {
1836 encoder->private_->metadata.data.stream_info.min_framesize = min(bytes, encoder->private_->metadata.data.stream_info.min_framesize);
1837 encoder->private_->metadata.data.stream_info.max_framesize = max(bytes, encoder->private_->metadata.data.stream_info.max_framesize);
1838 }
1839
Josh Coalson5c491a12002-08-01 06:39:40 +00001840 return true;
1841}
1842
Josh Coalsonf1eff452002-07-31 07:05:33 +00001843FLAC__bool process_frame_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
Josh Coalson0a15c142001-06-13 17:59:57 +00001844{
Josh Coalsonfa697a92001-08-16 20:07:29 +00001845 FLAC__ASSERT(encoder->protected_->state == FLAC__STREAM_ENCODER_OK);
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001846
1847 /*
Josh Coalsonfa37f1c2001-01-12 23:55:11 +00001848 * Accumulate raw signal to the MD5 signature
1849 */
Josh Coalson57ba6f42002-06-07 05:27:37 +00001850 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 +00001851 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonfa37f1c2001-01-12 23:55:11 +00001852 return false;
1853 }
1854
1855 /*
Josh Coalson94e02cd2001-01-25 10:41:06 +00001856 * Process the frame header and subframes into the frame bitbuffer
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001857 */
Josh Coalsonf1eff452002-07-31 07:05:33 +00001858 if(!process_subframes_(encoder, is_last_frame)) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00001859 /* the above function sets the state for us in case of an error */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001860 return false;
1861 }
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001862
1863 /*
1864 * Zero-pad the frame to a byte_boundary
1865 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001866 if(!FLAC__bitbuffer_zero_pad_to_byte_boundary(encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001867 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001868 return false;
1869 }
1870
1871 /*
Josh Coalson215af572001-03-27 01:15:58 +00001872 * CRC-16 the whole thing
1873 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001874 FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(encoder->private_->frame));
1875 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 +00001876
1877 /*
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001878 * Write it
1879 */
Josh Coalsond86e03b2002-08-03 21:56:15 +00001880 if(!write_bitbuffer_(encoder, encoder->protected_->blocksize)) {
1881 /* the above function sets the state for us in case of an error */
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001882 return false;
1883 }
1884
1885 /*
1886 * Get ready for the next frame
1887 */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001888 encoder->private_->current_sample_number = 0;
1889 encoder->private_->current_frame_number++;
1890 encoder->private_->metadata.data.stream_info.total_samples += (FLAC__uint64)encoder->protected_->blocksize;
Josh Coalsonbb7f6b92000-12-10 04:09:52 +00001891
1892 return true;
1893}
1894
Josh Coalsonf1eff452002-07-31 07:05:33 +00001895FLAC__bool process_subframes_(FLAC__StreamEncoder *encoder, FLAC__bool is_last_frame)
Josh Coalson94e02cd2001-01-25 10:41:06 +00001896{
1897 FLAC__FrameHeader frame_header;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001898 unsigned channel, min_partition_order = encoder->protected_->min_residual_partition_order, max_partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00001899 FLAC__bool do_independent, do_mid_side, precompute_partition_sums;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001900
1901 /*
Josh Coalson60f77d72001-04-25 02:16:36 +00001902 * Calculate the min,max Rice partition orders
Josh Coalson94e02cd2001-01-25 10:41:06 +00001903 */
1904 if(is_last_frame) {
1905 max_partition_order = 0;
1906 }
1907 else {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001908 max_partition_order = FLAC__format_get_max_rice_partition_order_from_blocksize(encoder->protected_->blocksize);
1909 max_partition_order = min(max_partition_order, encoder->protected_->max_residual_partition_order);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001910 }
Josh Coalson60f77d72001-04-25 02:16:36 +00001911 min_partition_order = min(min_partition_order, max_partition_order);
Josh Coalson94e02cd2001-01-25 10:41:06 +00001912
Josh Coalsonfa697a92001-08-16 20:07:29 +00001913 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 +00001914
Josh Coalson94e02cd2001-01-25 10:41:06 +00001915 /*
1916 * Setup the frame
1917 */
Josh Coalsonaec256b2002-03-12 16:19:54 +00001918 if(!FLAC__bitbuffer_clear(encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001919 encoder->protected_->state = FLAC__STREAM_ENCODER_MEMORY_ALLOCATION_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001920 return false;
1921 }
Josh Coalsonfa697a92001-08-16 20:07:29 +00001922 frame_header.blocksize = encoder->protected_->blocksize;
1923 frame_header.sample_rate = encoder->protected_->sample_rate;
1924 frame_header.channels = encoder->protected_->channels;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001925 frame_header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT; /* the default unless the encoder determines otherwise */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001926 frame_header.bits_per_sample = encoder->protected_->bits_per_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00001927 frame_header.number_type = FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER;
Josh Coalsonfa697a92001-08-16 20:07:29 +00001928 frame_header.number.frame_number = encoder->private_->current_frame_number;
Josh Coalson94e02cd2001-01-25 10:41:06 +00001929
1930 /*
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001931 * Figure out what channel assignments to try
1932 */
Josh Coalsonfa697a92001-08-16 20:07:29 +00001933 if(encoder->protected_->do_mid_side_stereo) {
1934 if(encoder->protected_->loose_mid_side_stereo) {
1935 if(encoder->private_->loose_mid_side_stereo_frame_count == 0) {
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001936 do_independent = true;
1937 do_mid_side = true;
1938 }
1939 else {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001940 do_independent = (encoder->private_->last_channel_assignment == FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001941 do_mid_side = !do_independent;
1942 }
1943 }
1944 else {
1945 do_independent = true;
1946 do_mid_side = true;
1947 }
1948 }
1949 else {
1950 do_independent = true;
1951 do_mid_side = false;
1952 }
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001953
Josh Coalson1b689822001-05-31 20:11:02 +00001954 FLAC__ASSERT(do_independent || do_mid_side);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001955
1956 /*
Josh Coalson82b73242001-03-28 22:17:05 +00001957 * Check for wasted bits; set effective bps for each subframe
Josh Coalson859bc542001-03-27 22:22:27 +00001958 */
1959 if(do_independent) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001960 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001961 const unsigned w = get_wasted_bits_(encoder->private_->integer_signal[channel], encoder->protected_->blocksize);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001962 encoder->private_->subframe_workspace[channel][0].wasted_bits = encoder->private_->subframe_workspace[channel][1].wasted_bits = w;
1963 encoder->private_->subframe_bps[channel] = encoder->protected_->bits_per_sample - w;
Josh Coalson82b73242001-03-28 22:17:05 +00001964 }
Josh Coalson859bc542001-03-27 22:22:27 +00001965 }
1966 if(do_mid_side) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001967 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson82b73242001-03-28 22:17:05 +00001968 for(channel = 0; channel < 2; channel++) {
Josh Coalsonb7023aa2002-08-17 15:23:43 +00001969 const unsigned w = get_wasted_bits_(encoder->private_->integer_signal_mid_side[channel], encoder->protected_->blocksize);
Josh Coalsonfa697a92001-08-16 20:07:29 +00001970 encoder->private_->subframe_workspace_mid_side[channel][0].wasted_bits = encoder->private_->subframe_workspace_mid_side[channel][1].wasted_bits = w;
1971 encoder->private_->subframe_bps_mid_side[channel] = encoder->protected_->bits_per_sample - w + (channel==0? 0:1);
Josh Coalson82b73242001-03-28 22:17:05 +00001972 }
Josh Coalson859bc542001-03-27 22:22:27 +00001973 }
1974
1975 /*
Josh Coalson94e02cd2001-01-25 10:41:06 +00001976 * First do a normal encoding pass of each independent channel
1977 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001978 if(do_independent) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00001979 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalson6fe72f72002-08-20 04:01:59 +00001980 if(!
1981 process_subframe_(
1982 encoder,
1983 min_partition_order,
1984 max_partition_order,
1985 precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00001986 &frame_header,
1987 encoder->private_->subframe_bps[channel],
1988 encoder->private_->integer_signal[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001989#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00001990 encoder->private_->real_signal[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00001991#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00001992 encoder->private_->subframe_workspace_ptr[channel],
1993 encoder->private_->partitioned_rice_contents_workspace_ptr[channel],
1994 encoder->private_->residual_workspace[channel],
1995 encoder->private_->best_subframe+channel,
1996 encoder->private_->best_subframe_bits+channel
1997 )
1998 )
Josh Coalsonb5e60e52001-01-28 09:27:27 +00001999 return false;
2000 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002001 }
2002
2003 /*
2004 * Now do mid and side channels if requested
2005 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002006 if(do_mid_side) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002007 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002008
2009 for(channel = 0; channel < 2; channel++) {
Josh Coalson6fe72f72002-08-20 04:01:59 +00002010 if(!
2011 process_subframe_(
2012 encoder,
2013 min_partition_order,
2014 max_partition_order,
2015 precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00002016 &frame_header,
2017 encoder->private_->subframe_bps_mid_side[channel],
2018 encoder->private_->integer_signal_mid_side[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002019#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00002020 encoder->private_->real_signal_mid_side[channel],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002021#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00002022 encoder->private_->subframe_workspace_ptr_mid_side[channel],
2023 encoder->private_->partitioned_rice_contents_workspace_ptr_mid_side[channel],
2024 encoder->private_->residual_workspace_mid_side[channel],
2025 encoder->private_->best_subframe_mid_side+channel,
2026 encoder->private_->best_subframe_bits_mid_side+channel
2027 )
2028 )
Josh Coalson94e02cd2001-01-25 10:41:06 +00002029 return false;
2030 }
2031 }
2032
2033 /*
2034 * Compose the frame bitbuffer
2035 */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002036 if(do_mid_side) {
Josh Coalson82b73242001-03-28 22:17:05 +00002037 unsigned left_bps = 0, right_bps = 0; /* initialized only to prevent superfluous compiler warning */
2038 FLAC__Subframe *left_subframe = 0, *right_subframe = 0; /* initialized only to prevent superfluous compiler warning */
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002039 FLAC__ChannelAssignment channel_assignment;
2040
Josh Coalsonfa697a92001-08-16 20:07:29 +00002041 FLAC__ASSERT(encoder->protected_->channels == 2);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002042
Josh Coalsonfa697a92001-08-16 20:07:29 +00002043 if(encoder->protected_->loose_mid_side_stereo && encoder->private_->loose_mid_side_stereo_frame_count > 0) {
2044 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 +00002045 }
2046 else {
2047 unsigned bits[4]; /* WATCHOUT - indexed by FLAC__ChannelAssignment */
2048 unsigned min_bits;
2049 FLAC__ChannelAssignment ca;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002050
Josh Coalson1b689822001-05-31 20:11:02 +00002051 FLAC__ASSERT(do_independent && do_mid_side);
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002052
2053 /* We have to figure out which channel assignent results in the smallest frame */
Josh Coalsonfa697a92001-08-16 20:07:29 +00002054 bits[FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT] = encoder->private_->best_subframe_bits [0] + encoder->private_->best_subframe_bits [1];
2055 bits[FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE ] = encoder->private_->best_subframe_bits [0] + encoder->private_->best_subframe_bits_mid_side[1];
2056 bits[FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE ] = encoder->private_->best_subframe_bits [1] + encoder->private_->best_subframe_bits_mid_side[1];
2057 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 +00002058
Josh Coalson7424d2f2002-11-06 07:10:38 +00002059 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 +00002060 if(bits[ca] < min_bits) {
2061 min_bits = bits[ca];
2062 channel_assignment = ca;
2063 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002064 }
2065 }
2066
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002067 frame_header.channel_assignment = channel_assignment;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002068
Josh Coalson3e7a96e2004-07-23 05:18:22 +00002069 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002070 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002071 return false;
2072 }
2073
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002074 switch(channel_assignment) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00002075 case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002076 left_subframe = &encoder->private_->subframe_workspace [0][encoder->private_->best_subframe [0]];
2077 right_subframe = &encoder->private_->subframe_workspace [1][encoder->private_->best_subframe [1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002078 break;
2079 case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002080 left_subframe = &encoder->private_->subframe_workspace [0][encoder->private_->best_subframe [0]];
2081 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002082 break;
2083 case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002084 left_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
2085 right_subframe = &encoder->private_->subframe_workspace [1][encoder->private_->best_subframe [1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002086 break;
2087 case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002088 left_subframe = &encoder->private_->subframe_workspace_mid_side[0][encoder->private_->best_subframe_mid_side[0]];
2089 right_subframe = &encoder->private_->subframe_workspace_mid_side[1][encoder->private_->best_subframe_mid_side[1]];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002090 break;
2091 default:
Josh Coalson1b689822001-05-31 20:11:02 +00002092 FLAC__ASSERT(0);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002093 }
Josh Coalson82b73242001-03-28 22:17:05 +00002094
2095 switch(channel_assignment) {
2096 case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002097 left_bps = encoder->private_->subframe_bps [0];
2098 right_bps = encoder->private_->subframe_bps [1];
Josh Coalson82b73242001-03-28 22:17:05 +00002099 break;
2100 case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002101 left_bps = encoder->private_->subframe_bps [0];
2102 right_bps = encoder->private_->subframe_bps_mid_side[1];
Josh Coalson82b73242001-03-28 22:17:05 +00002103 break;
2104 case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002105 left_bps = encoder->private_->subframe_bps_mid_side[1];
2106 right_bps = encoder->private_->subframe_bps [1];
Josh Coalson82b73242001-03-28 22:17:05 +00002107 break;
2108 case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
Josh Coalsonfa697a92001-08-16 20:07:29 +00002109 left_bps = encoder->private_->subframe_bps_mid_side[0];
2110 right_bps = encoder->private_->subframe_bps_mid_side[1];
Josh Coalson82b73242001-03-28 22:17:05 +00002111 break;
2112 default:
Josh Coalson1b689822001-05-31 20:11:02 +00002113 FLAC__ASSERT(0);
Josh Coalson82b73242001-03-28 22:17:05 +00002114 }
2115
2116 /* note that encoder_add_subframe_ sets the state for us in case of an error */
Josh Coalsonf1eff452002-07-31 07:05:33 +00002117 if(!add_subframe_(encoder, &frame_header, left_bps , left_subframe , encoder->private_->frame))
Josh Coalson82b73242001-03-28 22:17:05 +00002118 return false;
Josh Coalsonf1eff452002-07-31 07:05:33 +00002119 if(!add_subframe_(encoder, &frame_header, right_bps, right_subframe, encoder->private_->frame))
Josh Coalson82b73242001-03-28 22:17:05 +00002120 return false;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002121 }
2122 else {
Josh Coalson3e7a96e2004-07-23 05:18:22 +00002123 if(!FLAC__frame_add_header(&frame_header, encoder->protected_->streamable_subset, encoder->private_->frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002124 encoder->protected_->state = FLAC__STREAM_ENCODER_FRAMING_ERROR;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002125 return false;
2126 }
2127
Josh Coalsonfa697a92001-08-16 20:07:29 +00002128 for(channel = 0; channel < encoder->protected_->channels; channel++) {
Josh Coalsonf1eff452002-07-31 07:05:33 +00002129 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 +00002130 /* the above function sets the state for us in case of an error */
2131 return false;
2132 }
2133 }
2134 }
2135
Josh Coalsonfa697a92001-08-16 20:07:29 +00002136 if(encoder->protected_->loose_mid_side_stereo) {
2137 encoder->private_->loose_mid_side_stereo_frame_count++;
2138 if(encoder->private_->loose_mid_side_stereo_frame_count >= encoder->private_->loose_mid_side_stereo_frames)
2139 encoder->private_->loose_mid_side_stereo_frame_count = 0;
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002140 }
2141
Josh Coalsonfa697a92001-08-16 20:07:29 +00002142 encoder->private_->last_channel_assignment = frame_header.channel_assignment;
Josh Coalsonb5e60e52001-01-28 09:27:27 +00002143
Josh Coalson94e02cd2001-01-25 10:41:06 +00002144 return true;
2145}
2146
Josh Coalson6fe72f72002-08-20 04:01:59 +00002147FLAC__bool process_subframe_(
2148 FLAC__StreamEncoder *encoder,
2149 unsigned min_partition_order,
2150 unsigned max_partition_order,
2151 FLAC__bool precompute_partition_sums,
Josh Coalson6fe72f72002-08-20 04:01:59 +00002152 const FLAC__FrameHeader *frame_header,
2153 unsigned subframe_bps,
2154 const FLAC__int32 integer_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002155#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00002156 const FLAC__real real_signal[],
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002157#endif
Josh Coalson6fe72f72002-08-20 04:01:59 +00002158 FLAC__Subframe *subframe[2],
2159 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents[2],
2160 FLAC__int32 *residual[2],
2161 unsigned *best_subframe,
2162 unsigned *best_bits
2163)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002164{
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002165#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002166 FLAC__float fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002167#else
2168 FLAC__fixedpoint fixed_residual_bits_per_sample[FLAC__MAX_FIXED_ORDER+1];
2169#endif
2170#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002171 FLAC__double lpc_residual_bits_per_sample;
Josh Coalsonfa697a92001-08-16 20:07:29 +00002172 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 +00002173 FLAC__double lpc_error[FLAC__MAX_LPC_ORDER];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002174 unsigned min_lpc_order, max_lpc_order, lpc_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002175 unsigned min_qlp_coeff_precision, max_qlp_coeff_precision, qlp_coeff_precision;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002176#endif
2177 unsigned min_fixed_order, max_fixed_order, guess_fixed_order, fixed_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002178 unsigned rice_parameter;
2179 unsigned _candidate_bits, _best_bits;
2180 unsigned _best_subframe;
2181
2182 /* verbatim subframe is the baseline against which we measure other compressed subframes */
2183 _best_subframe = 0;
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002184 if(encoder->private_->disable_verbatim_subframes && frame_header->blocksize >= FLAC__MAX_FIXED_ORDER)
2185 _best_bits = UINT_MAX;
2186 else
2187 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002188
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002189 if(frame_header->blocksize >= FLAC__MAX_FIXED_ORDER) {
2190 unsigned signal_is_constant = false;
Josh Coalsonfa697a92001-08-16 20:07:29 +00002191 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 +00002192 /* check for constant subframe */
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002193 if(
2194 !encoder->private_->disable_constant_subframes &&
2195#ifndef FLAC__INTEGER_ONLY_LIBRARY
2196 fixed_residual_bits_per_sample[1] == 0.0
2197#else
2198 fixed_residual_bits_per_sample[1] == FLAC__FP_ZERO
2199#endif
2200 ) {
2201 /* the above means it's possible all samples are the same value; now double-check it: */
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002202 unsigned i;
2203 signal_is_constant = true;
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002204 for(i = 1; i < frame_header->blocksize; i++) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00002205 if(integer_signal[0] != integer_signal[i]) {
2206 signal_is_constant = false;
2207 break;
2208 }
2209 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002210 }
2211 if(signal_is_constant) {
2212 _candidate_bits = evaluate_constant_subframe_(integer_signal[0], subframe_bps, subframe[!_best_subframe]);
2213 if(_candidate_bits < _best_bits) {
2214 _best_subframe = !_best_subframe;
2215 _best_bits = _candidate_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002216 }
2217 }
2218 else {
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002219 if(!encoder->private_->disable_fixed_subframes || (encoder->protected_->max_lpc_order == 0 && _best_bits == UINT_MAX)) {
2220 /* encode fixed */
2221 if(encoder->protected_->do_exhaustive_model_search) {
2222 min_fixed_order = 0;
2223 max_fixed_order = FLAC__MAX_FIXED_ORDER;
Josh Coalson8395d022001-07-12 21:25:22 +00002224 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002225 else {
2226 min_fixed_order = max_fixed_order = guess_fixed_order;
2227 }
2228 for(fixed_order = min_fixed_order; fixed_order <= max_fixed_order; fixed_order++) {
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002229#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson09758432004-10-20 00:21:50 +00002230 if(fixed_residual_bits_per_sample[fixed_order] >= (FLAC__float)subframe_bps)
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002231 continue; /* don't even try */
2232 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 +00002233#else
2234 if(FLAC__fixedpoint_trunc(fixed_residual_bits_per_sample[fixed_order]) >= (int)subframe_bps)
2235 continue; /* don't even try */
2236 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 */
2237#endif
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002238#ifndef FLAC__SYMMETRIC_RICE
2239 rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
2240#endif
2241 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
2242#ifdef DEBUG_VERBOSE
2243 fprintf(stderr, "clipping rice_parameter (%u -> %u) @0\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2244#endif
2245 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2246 }
2247 _candidate_bits =
2248 evaluate_fixed_subframe_(
2249 encoder,
2250 integer_signal,
2251 residual[!_best_subframe],
2252 encoder->private_->abs_residual,
2253 encoder->private_->abs_residual_partition_sums,
2254 encoder->private_->raw_bits_per_partition,
2255 frame_header->blocksize,
2256 subframe_bps,
2257 fixed_order,
2258 rice_parameter,
2259 min_partition_order,
2260 max_partition_order,
2261 precompute_partition_sums,
2262 encoder->protected_->do_escape_coding,
2263 encoder->protected_->rice_parameter_search_dist,
2264 subframe[!_best_subframe],
2265 partitioned_rice_contents[!_best_subframe]
2266 );
2267 if(_candidate_bits < _best_bits) {
2268 _best_subframe = !_best_subframe;
2269 _best_bits = _candidate_bits;
2270 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002271 }
2272 }
2273
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002274#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson94e02cd2001-01-25 10:41:06 +00002275 /* encode lpc */
Josh Coalsonfa697a92001-08-16 20:07:29 +00002276 if(encoder->protected_->max_lpc_order > 0) {
2277 if(encoder->protected_->max_lpc_order >= frame_header->blocksize)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002278 max_lpc_order = frame_header->blocksize-1;
2279 else
Josh Coalsonfa697a92001-08-16 20:07:29 +00002280 max_lpc_order = encoder->protected_->max_lpc_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002281 if(max_lpc_order > 0) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002282 encoder->private_->local_lpc_compute_autocorrelation(real_signal, frame_header->blocksize, max_lpc_order+1, autoc);
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002283 /* if autoc[0] == 0.0, the signal is constant and we usually won't get here, but it can happen */
2284 if(autoc[0] != 0.0) {
Josh Coalson8084b052001-11-01 00:27:29 +00002285 FLAC__lpc_compute_lp_coefficients(autoc, max_lpc_order, encoder->private_->lp_coeff, lpc_error);
Josh Coalsonfa697a92001-08-16 20:07:29 +00002286 if(encoder->protected_->do_exhaustive_model_search) {
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002287 min_lpc_order = 1;
2288 }
2289 else {
Josh Coalson82b73242001-03-28 22:17:05 +00002290 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 +00002291 min_lpc_order = max_lpc_order = guess_lpc_order;
2292 }
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002293 for(lpc_order = min_lpc_order; lpc_order <= max_lpc_order; lpc_order++) {
2294 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 +00002295 if(lpc_residual_bits_per_sample >= (FLAC__double)subframe_bps)
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002296 continue; /* don't even try */
2297 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 +00002298#ifndef FLAC__SYMMETRIC_RICE
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002299 rice_parameter++; /* to account for the signed->unsigned conversion during rice coding */
Josh Coalsonb9433f92001-03-17 01:07:00 +00002300#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002301 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002302#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002303 fprintf(stderr, "clipping rice_parameter (%u -> %u) @1\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2304#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00002305 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00002306 }
Josh Coalsonc9c0d132002-10-04 05:29:05 +00002307 if(encoder->protected_->do_qlp_coeff_prec_search) {
2308 min_qlp_coeff_precision = FLAC__MIN_QLP_COEFF_PRECISION;
2309 /* ensure a 32-bit datapath throughout for 16bps or less */
2310 if(subframe_bps <= 16)
2311 max_qlp_coeff_precision = min(32 - subframe_bps - lpc_order, FLAC__MAX_QLP_COEFF_PRECISION);
2312 else
2313 max_qlp_coeff_precision = FLAC__MAX_QLP_COEFF_PRECISION;
2314 }
2315 else {
2316 min_qlp_coeff_precision = max_qlp_coeff_precision = encoder->protected_->qlp_coeff_precision;
2317 }
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002318 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 +00002319 _candidate_bits =
2320 evaluate_lpc_subframe_(
2321 encoder,
2322 integer_signal,
2323 residual[!_best_subframe],
2324 encoder->private_->abs_residual,
2325 encoder->private_->abs_residual_partition_sums,
2326 encoder->private_->raw_bits_per_partition,
2327 encoder->private_->lp_coeff[lpc_order-1],
2328 frame_header->blocksize,
2329 subframe_bps,
2330 lpc_order,
2331 qlp_coeff_precision,
2332 rice_parameter,
2333 min_partition_order,
2334 max_partition_order,
2335 precompute_partition_sums,
2336 encoder->protected_->do_escape_coding,
2337 encoder->protected_->rice_parameter_search_dist,
2338 subframe[!_best_subframe],
2339 partitioned_rice_contents[!_best_subframe]
2340 );
Josh Coalsonf4ce50b2001-02-28 23:45:15 +00002341 if(_candidate_bits > 0) { /* if == 0, there was a problem quantizing the lpcoeffs */
2342 if(_candidate_bits < _best_bits) {
2343 _best_subframe = !_best_subframe;
2344 _best_bits = _candidate_bits;
2345 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002346 }
2347 }
2348 }
2349 }
2350 }
2351 }
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002352#endif /* !defined FLAC__INTEGER_ONLY_LIBRARY */
Josh Coalson94e02cd2001-01-25 10:41:06 +00002353 }
2354 }
2355
Josh Coalson72695802002-10-11 06:25:16 +00002356 /* under rare circumstances this can happen when all but lpc subframe types are disabled: */
2357 if(_best_bits == UINT_MAX) {
2358 FLAC__ASSERT(_best_subframe == 0);
2359 _best_bits = evaluate_verbatim_subframe_(integer_signal, frame_header->blocksize, subframe_bps, subframe[_best_subframe]);
2360 }
Josh Coalsone6b3bbe2002-10-08 06:03:25 +00002361
Josh Coalson94e02cd2001-01-25 10:41:06 +00002362 *best_subframe = _best_subframe;
2363 *best_bits = _best_bits;
2364
2365 return true;
2366}
2367
Josh Coalson6fe72f72002-08-20 04:01:59 +00002368FLAC__bool add_subframe_(
2369 FLAC__StreamEncoder *encoder,
2370 const FLAC__FrameHeader *frame_header,
2371 unsigned subframe_bps,
2372 const FLAC__Subframe *subframe,
2373 FLAC__BitBuffer *frame
2374)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002375{
2376 switch(subframe->type) {
2377 case FLAC__SUBFRAME_TYPE_CONSTANT:
Josh Coalson82b73242001-03-28 22:17:05 +00002378 if(!FLAC__subframe_add_constant(&(subframe->data.constant), subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002379 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002380 return false;
2381 }
2382 break;
2383 case FLAC__SUBFRAME_TYPE_FIXED:
Josh Coalson82b73242001-03-28 22:17:05 +00002384 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 +00002385 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002386 return false;
2387 }
2388 break;
2389 case FLAC__SUBFRAME_TYPE_LPC:
Josh Coalson82b73242001-03-28 22:17:05 +00002390 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 +00002391 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002392 return false;
2393 }
2394 break;
2395 case FLAC__SUBFRAME_TYPE_VERBATIM:
Josh Coalson82b73242001-03-28 22:17:05 +00002396 if(!FLAC__subframe_add_verbatim(&(subframe->data.verbatim), frame_header->blocksize, subframe_bps, subframe->wasted_bits, frame)) {
Josh Coalsonfa697a92001-08-16 20:07:29 +00002397 encoder->protected_->state = FLAC__STREAM_ENCODER_FATAL_ERROR_WHILE_ENCODING;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002398 return false;
2399 }
2400 break;
2401 default:
Josh Coalson1b689822001-05-31 20:11:02 +00002402 FLAC__ASSERT(0);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002403 }
2404
2405 return true;
2406}
2407
Josh Coalson6fe72f72002-08-20 04:01:59 +00002408unsigned evaluate_constant_subframe_(
2409 const FLAC__int32 signal,
2410 unsigned subframe_bps,
2411 FLAC__Subframe *subframe
2412)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002413{
2414 subframe->type = FLAC__SUBFRAME_TYPE_CONSTANT;
2415 subframe->data.constant.value = signal;
2416
Josh Coalson82b73242001-03-28 22:17:05 +00002417 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 +00002418}
2419
Josh Coalson6fe72f72002-08-20 04:01:59 +00002420unsigned evaluate_fixed_subframe_(
2421 FLAC__StreamEncoder *encoder,
2422 const FLAC__int32 signal[],
2423 FLAC__int32 residual[],
2424 FLAC__uint32 abs_residual[],
2425 FLAC__uint64 abs_residual_partition_sums[],
2426 unsigned raw_bits_per_partition[],
2427 unsigned blocksize,
2428 unsigned subframe_bps,
2429 unsigned order,
2430 unsigned rice_parameter,
2431 unsigned min_partition_order,
2432 unsigned max_partition_order,
2433 FLAC__bool precompute_partition_sums,
2434 FLAC__bool do_escape_coding,
2435 unsigned rice_parameter_search_dist,
2436 FLAC__Subframe *subframe,
2437 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2438)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002439{
2440 unsigned i, residual_bits;
2441 const unsigned residual_samples = blocksize - order;
2442
2443 FLAC__fixed_compute_residual(signal+order, residual_samples, order, residual);
2444
2445 subframe->type = FLAC__SUBFRAME_TYPE_FIXED;
2446
2447 subframe->data.fixed.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
Josh Coalsona37ba462002-08-19 21:36:39 +00002448 subframe->data.fixed.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002449 subframe->data.fixed.residual = residual;
2450
Josh Coalson6fe72f72002-08-20 04:01:59 +00002451 residual_bits =
2452 find_best_partition_order_(
2453 encoder->private_,
2454 residual,
2455 abs_residual,
2456 abs_residual_partition_sums,
2457 raw_bits_per_partition,
2458 residual_samples,
2459 order,
2460 rice_parameter,
2461 min_partition_order,
2462 max_partition_order,
2463 precompute_partition_sums,
2464 do_escape_coding,
2465 rice_parameter_search_dist,
2466 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2467 );
Josh Coalson94e02cd2001-01-25 10:41:06 +00002468
2469 subframe->data.fixed.order = order;
2470 for(i = 0; i < order; i++)
2471 subframe->data.fixed.warmup[i] = signal[i];
2472
Josh Coalson82b73242001-03-28 22:17:05 +00002473 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 +00002474}
2475
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002476#ifndef FLAC__INTEGER_ONLY_LIBRARY
Josh Coalson6fe72f72002-08-20 04:01:59 +00002477unsigned evaluate_lpc_subframe_(
2478 FLAC__StreamEncoder *encoder,
2479 const FLAC__int32 signal[],
2480 FLAC__int32 residual[],
2481 FLAC__uint32 abs_residual[],
2482 FLAC__uint64 abs_residual_partition_sums[],
2483 unsigned raw_bits_per_partition[],
2484 const FLAC__real lp_coeff[],
2485 unsigned blocksize,
2486 unsigned subframe_bps,
2487 unsigned order,
2488 unsigned qlp_coeff_precision,
2489 unsigned rice_parameter,
2490 unsigned min_partition_order,
2491 unsigned max_partition_order,
2492 FLAC__bool precompute_partition_sums,
2493 FLAC__bool do_escape_coding,
2494 unsigned rice_parameter_search_dist,
2495 FLAC__Subframe *subframe,
2496 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents
2497)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002498{
Josh Coalson77e3f312001-06-23 03:03:24 +00002499 FLAC__int32 qlp_coeff[FLAC__MAX_LPC_ORDER];
Josh Coalson94e02cd2001-01-25 10:41:06 +00002500 unsigned i, residual_bits;
2501 int quantization, ret;
2502 const unsigned residual_samples = blocksize - order;
2503
Josh Coalson20ac2c12002-08-30 05:47:14 +00002504 /* try to keep qlp coeff precision such that only 32-bit math is required for decode of <=16bps streams */
2505 if(subframe_bps <= 16) {
2506 FLAC__ASSERT(order > 0);
2507 FLAC__ASSERT(order <= FLAC__MAX_LPC_ORDER);
2508 qlp_coeff_precision = min(qlp_coeff_precision, 32 - subframe_bps - FLAC__bitmath_ilog2(order));
2509 }
2510
Josh Coalsonc9c0d132002-10-04 05:29:05 +00002511 ret = FLAC__lpc_quantize_coefficients(lp_coeff, order, qlp_coeff_precision, qlp_coeff, &quantization);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002512 if(ret != 0)
2513 return 0; /* this is a hack to indicate to the caller that we can't do lp at this order on this subframe */
2514
Josh Coalsonfb9d18f2002-10-21 07:04:07 +00002515 if(subframe_bps + qlp_coeff_precision + FLAC__bitmath_ilog2(order) <= 32)
2516 if(subframe_bps <= 16 && qlp_coeff_precision <= 16)
2517 encoder->private_->local_lpc_compute_residual_from_qlp_coefficients_16bit(signal+order, residual_samples, qlp_coeff, order, quantization, residual);
2518 else
2519 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 +00002520 else
2521 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 +00002522
2523 subframe->type = FLAC__SUBFRAME_TYPE_LPC;
2524
2525 subframe->data.lpc.entropy_coding_method.type = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE;
Josh Coalsona37ba462002-08-19 21:36:39 +00002526 subframe->data.lpc.entropy_coding_method.data.partitioned_rice.contents = partitioned_rice_contents;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002527 subframe->data.lpc.residual = residual;
2528
Josh Coalson6fe72f72002-08-20 04:01:59 +00002529 residual_bits =
2530 find_best_partition_order_(
2531 encoder->private_,
2532 residual,
2533 abs_residual,
2534 abs_residual_partition_sums,
2535 raw_bits_per_partition,
2536 residual_samples,
2537 order,
2538 rice_parameter,
2539 min_partition_order,
2540 max_partition_order,
2541 precompute_partition_sums,
2542 do_escape_coding,
2543 rice_parameter_search_dist,
2544 &subframe->data.fixed.entropy_coding_method.data.partitioned_rice
2545 );
Josh Coalson94e02cd2001-01-25 10:41:06 +00002546
2547 subframe->data.lpc.order = order;
2548 subframe->data.lpc.qlp_coeff_precision = qlp_coeff_precision;
2549 subframe->data.lpc.quantization_level = quantization;
Josh Coalson77e3f312001-06-23 03:03:24 +00002550 memcpy(subframe->data.lpc.qlp_coeff, qlp_coeff, sizeof(FLAC__int32)*FLAC__MAX_LPC_ORDER);
Josh Coalson94e02cd2001-01-25 10:41:06 +00002551 for(i = 0; i < order; i++)
2552 subframe->data.lpc.warmup[i] = signal[i];
2553
Josh Coalson82b73242001-03-28 22:17:05 +00002554 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 +00002555}
Josh Coalson5f2b46d2004-11-09 01:34:01 +00002556#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002557
Josh Coalson6fe72f72002-08-20 04:01:59 +00002558unsigned evaluate_verbatim_subframe_(
2559 const FLAC__int32 signal[],
2560 unsigned blocksize,
2561 unsigned subframe_bps,
2562 FLAC__Subframe *subframe
2563)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002564{
2565 subframe->type = FLAC__SUBFRAME_TYPE_VERBATIM;
2566
2567 subframe->data.verbatim.data = signal;
2568
Josh Coalson82b73242001-03-28 22:17:05 +00002569 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 +00002570}
2571
Josh Coalson6fe72f72002-08-20 04:01:59 +00002572unsigned find_best_partition_order_(
2573 FLAC__StreamEncoderPrivate *private_,
2574 const FLAC__int32 residual[],
2575 FLAC__uint32 abs_residual[],
2576 FLAC__uint64 abs_residual_partition_sums[],
2577 unsigned raw_bits_per_partition[],
2578 unsigned residual_samples,
2579 unsigned predictor_order,
2580 unsigned rice_parameter,
2581 unsigned min_partition_order,
2582 unsigned max_partition_order,
2583 FLAC__bool precompute_partition_sums,
2584 FLAC__bool do_escape_coding,
2585 unsigned rice_parameter_search_dist,
2586 FLAC__EntropyCodingMethod_PartitionedRice *best_partitioned_rice
2587)
Josh Coalson94e02cd2001-01-25 10:41:06 +00002588{
Josh Coalson77e3f312001-06-23 03:03:24 +00002589 FLAC__int32 r;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002590 unsigned residual_bits, best_residual_bits = 0;
Josh Coalsonafcd8772001-04-18 22:59:25 +00002591 unsigned residual_sample;
Josh Coalson8084b052001-11-01 00:27:29 +00002592 unsigned best_parameters_index = 0;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002593 const unsigned blocksize = residual_samples + predictor_order;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002594
Josh Coalson2051dd42001-04-12 22:22:34 +00002595 /* compute abs(residual) for use later */
2596 for(residual_sample = 0; residual_sample < residual_samples; residual_sample++) {
2597 r = residual[residual_sample];
Josh Coalson77e3f312001-06-23 03:03:24 +00002598 abs_residual[residual_sample] = (FLAC__uint32)(r<0? -r : r);
Josh Coalson2051dd42001-04-12 22:22:34 +00002599 }
2600
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002601 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 +00002602 min_partition_order = min(min_partition_order, max_partition_order);
2603
Josh Coalson8395d022001-07-12 21:25:22 +00002604 if(precompute_partition_sums) {
2605 int partition_order;
2606 unsigned sum;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002607
Josh Coalsonf1eff452002-07-31 07:05:33 +00002608 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 +00002609
2610 if(do_escape_coding)
Josh Coalsonf1eff452002-07-31 07:05:33 +00002611 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 +00002612
2613 for(partition_order = (int)max_partition_order, sum = 0; partition_order >= (int)min_partition_order; partition_order--) {
2614#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002615 if(!
2616 set_partitioned_rice_with_precompute_(
2617 residual,
2618 abs_residual_partition_sums+sum,
2619 raw_bits_per_partition+sum,
2620 residual_samples,
2621 predictor_order,
2622 rice_parameter,
2623 rice_parameter_search_dist,
2624 (unsigned)partition_order,
2625 do_escape_coding,
2626 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2627 &residual_bits
2628 )
2629 )
Josh Coalsonafcd8772001-04-18 22:59:25 +00002630#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002631 if(!
2632 set_partitioned_rice_with_precompute_(
2633 abs_residual,
2634 abs_residual_partition_sums+sum,
2635 raw_bits_per_partition+sum,
2636 residual_samples,
2637 predictor_order,
2638 rice_parameter,
2639 rice_parameter_search_dist,
2640 (unsigned)partition_order,
2641 do_escape_coding,
2642 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2643 &residual_bits
2644 )
2645 )
Josh Coalson8395d022001-07-12 21:25:22 +00002646#endif
2647 {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002648 FLAC__ASSERT(best_residual_bits != 0);
2649 break;
Josh Coalson8395d022001-07-12 21:25:22 +00002650 }
2651 sum += 1u << partition_order;
2652 if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2653 best_residual_bits = residual_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00002654 best_parameters_index = !best_parameters_index;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002655 best_partitioned_rice->order = partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00002656 }
Josh Coalsonafcd8772001-04-18 22:59:25 +00002657 }
2658 }
Josh Coalson8395d022001-07-12 21:25:22 +00002659 else {
2660 unsigned partition_order;
2661 for(partition_order = min_partition_order; partition_order <= max_partition_order; partition_order++) {
2662#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002663 if(!
2664 set_partitioned_rice_(
2665 abs_residual,
2666 residual,
2667 residual_samples,
2668 predictor_order,
2669 rice_parameter,
2670 rice_parameter_search_dist,
2671 partition_order,
2672 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2673 &residual_bits
2674 )
2675 )
Josh Coalson8395d022001-07-12 21:25:22 +00002676#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002677 if(!
2678 set_partitioned_rice_(
2679 abs_residual,
2680 residual_samples,
2681 predictor_order,
2682 rice_parameter,
2683 rice_parameter_search_dist,
2684 partition_order,
2685 &private_->partitioned_rice_contents_extra[!best_parameters_index],
2686 &residual_bits
2687 )
2688 )
Josh Coalsonafcd8772001-04-18 22:59:25 +00002689#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002690 {
2691 FLAC__ASSERT(best_residual_bits != 0);
2692 break;
2693 }
2694 if(best_residual_bits == 0 || residual_bits < best_residual_bits) {
2695 best_residual_bits = residual_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00002696 best_parameters_index = !best_parameters_index;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002697 best_partitioned_rice->order = partition_order;
Josh Coalson8395d022001-07-12 21:25:22 +00002698 }
2699 }
2700 }
2701
Josh Coalsona37ba462002-08-19 21:36:39 +00002702 /*
Josh Coalson20ac2c12002-08-30 05:47:14 +00002703 * We are allowed to de-const the pointer based on our special knowledge;
Josh Coalsona37ba462002-08-19 21:36:39 +00002704 * it is const to the outside world.
2705 */
2706 {
2707 FLAC__EntropyCodingMethod_PartitionedRiceContents* best_partitioned_rice_contents = (FLAC__EntropyCodingMethod_PartitionedRiceContents*)best_partitioned_rice->contents;
2708 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(best_partitioned_rice_contents, max(6, best_partitioned_rice->order));
2709 memcpy(best_partitioned_rice_contents->parameters, private_->partitioned_rice_contents_extra[best_parameters_index].parameters, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2710 memcpy(best_partitioned_rice_contents->raw_bits, private_->partitioned_rice_contents_extra[best_parameters_index].raw_bits, sizeof(unsigned)*(1<<(best_partitioned_rice->order)));
2711 }
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002712
2713 return best_residual_bits;
2714}
2715
Josh Coalson6fe72f72002-08-20 04:01:59 +00002716void precompute_partition_info_sums_(
2717 const FLAC__uint32 abs_residual[],
2718 FLAC__uint64 abs_residual_partition_sums[],
2719 unsigned residual_samples,
2720 unsigned predictor_order,
2721 unsigned min_partition_order,
2722 unsigned max_partition_order
2723)
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002724{
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002725 int partition_order;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002726 unsigned from_partition, to_partition = 0;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002727 const unsigned blocksize = residual_samples + predictor_order;
2728
Josh Coalsonaef013c2001-04-24 01:25:42 +00002729 /* first do max_partition_order */
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002730 for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002731 FLAC__uint64 abs_residual_partition_sum;
Josh Coalson77e3f312001-06-23 03:03:24 +00002732 FLAC__uint32 abs_r;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002733 unsigned partition, partition_sample, partition_samples, residual_sample;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002734 const unsigned partitions = 1u << partition_order;
2735 const unsigned default_partition_samples = blocksize >> partition_order;
2736
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002737 FLAC__ASSERT(default_partition_samples > predictor_order);
2738
2739 for(partition = residual_sample = 0; partition < partitions; partition++) {
2740 partition_samples = default_partition_samples;
2741 if(partition == 0)
2742 partition_samples -= predictor_order;
2743 abs_residual_partition_sum = 0;
2744 for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2745 abs_r = abs_residual[residual_sample];
2746 abs_residual_partition_sum += abs_r;
2747 residual_sample++;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002748 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002749 abs_residual_partition_sums[partition] = abs_residual_partition_sum;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002750 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002751 to_partition = partitions;
2752 break;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002753 }
Josh Coalsonf76a3612001-04-18 02:28:11 +00002754
Josh Coalson8395d022001-07-12 21:25:22 +00002755 /* now merge partitions for lower orders */
Josh Coalson6bd17572001-05-25 19:02:01 +00002756 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002757 FLAC__uint64 s;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002758 unsigned i;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002759 const unsigned partitions = 1u << partition_order;
2760 for(i = 0; i < partitions; i++) {
Josh Coalsonaef013c2001-04-24 01:25:42 +00002761 s = abs_residual_partition_sums[from_partition];
Josh Coalsonaef013c2001-04-24 01:25:42 +00002762 from_partition++;
Josh Coalsonaef013c2001-04-24 01:25:42 +00002763 abs_residual_partition_sums[to_partition] = s + abs_residual_partition_sums[from_partition];
Josh Coalsonaef013c2001-04-24 01:25:42 +00002764 from_partition++;
2765 to_partition++;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00002766 }
2767 }
Josh Coalson94e02cd2001-01-25 10:41:06 +00002768}
Josh Coalson8395d022001-07-12 21:25:22 +00002769
Josh Coalson6fe72f72002-08-20 04:01:59 +00002770void precompute_partition_info_escapes_(
2771 const FLAC__int32 residual[],
2772 unsigned raw_bits_per_partition[],
2773 unsigned residual_samples,
2774 unsigned predictor_order,
2775 unsigned min_partition_order,
2776 unsigned max_partition_order
2777)
Josh Coalson8395d022001-07-12 21:25:22 +00002778{
2779 int partition_order;
2780 unsigned from_partition, to_partition = 0;
2781 const unsigned blocksize = residual_samples + predictor_order;
2782
2783 /* first do max_partition_order */
2784 for(partition_order = (int)max_partition_order; partition_order >= 0; partition_order--) {
2785 FLAC__int32 r, residual_partition_min, residual_partition_max;
2786 unsigned silog2_min, silog2_max;
2787 unsigned partition, partition_sample, partition_samples, residual_sample;
2788 const unsigned partitions = 1u << partition_order;
2789 const unsigned default_partition_samples = blocksize >> partition_order;
2790
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002791 FLAC__ASSERT(default_partition_samples > predictor_order);
2792
2793 for(partition = residual_sample = 0; partition < partitions; partition++) {
2794 partition_samples = default_partition_samples;
2795 if(partition == 0)
2796 partition_samples -= predictor_order;
2797 residual_partition_min = residual_partition_max = 0;
2798 for(partition_sample = 0; partition_sample < partition_samples; partition_sample++) {
2799 r = residual[residual_sample];
2800 if(r < residual_partition_min)
2801 residual_partition_min = r;
2802 else if(r > residual_partition_max)
2803 residual_partition_max = r;
2804 residual_sample++;
Josh Coalson8395d022001-07-12 21:25:22 +00002805 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002806 silog2_min = FLAC__bitmath_silog2(residual_partition_min);
2807 silog2_max = FLAC__bitmath_silog2(residual_partition_max);
2808 raw_bits_per_partition[partition] = max(silog2_min, silog2_max);
Josh Coalson8395d022001-07-12 21:25:22 +00002809 }
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002810 to_partition = partitions;
2811 break;
Josh Coalson8395d022001-07-12 21:25:22 +00002812 }
2813
2814 /* now merge partitions for lower orders */
2815 for(from_partition = 0, --partition_order; partition_order >= (int)min_partition_order; partition_order--) {
2816 unsigned m;
2817 unsigned i;
2818 const unsigned partitions = 1u << partition_order;
2819 for(i = 0; i < partitions; i++) {
2820 m = raw_bits_per_partition[from_partition];
2821 from_partition++;
2822 raw_bits_per_partition[to_partition] = max(m, raw_bits_per_partition[from_partition]);
2823 from_partition++;
2824 to_partition++;
2825 }
2826 }
2827}
Josh Coalson94e02cd2001-01-25 10:41:06 +00002828
Josh Coalson352e0f62001-03-20 22:55:50 +00002829#ifdef VARIABLE_RICE_BITS
2830#undef VARIABLE_RICE_BITS
2831#endif
Josh Coalson8395d022001-07-12 21:25:22 +00002832#ifndef DONT_ESTIMATE_RICE_BITS
Josh Coalson352e0f62001-03-20 22:55:50 +00002833#define VARIABLE_RICE_BITS(value, parameter) ((value) >> (parameter))
Josh Coalson8395d022001-07-12 21:25:22 +00002834#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00002835
Josh Coalson8395d022001-07-12 21:25:22 +00002836#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00002837FLAC__bool set_partitioned_rice_(
2838 const FLAC__uint32 abs_residual[],
2839 const FLAC__int32 residual[],
2840 const unsigned residual_samples,
2841 const unsigned predictor_order,
2842 const unsigned suggested_rice_parameter,
2843 const unsigned rice_parameter_search_dist,
2844 const unsigned partition_order,
2845 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2846 unsigned *bits
2847)
Josh Coalson8395d022001-07-12 21:25:22 +00002848#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00002849FLAC__bool set_partitioned_rice_(
2850 const FLAC__uint32 abs_residual[],
2851 const unsigned residual_samples,
2852 const unsigned predictor_order,
2853 const unsigned suggested_rice_parameter,
2854 const unsigned rice_parameter_search_dist,
2855 const unsigned partition_order,
2856 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
2857 unsigned *bits
2858)
Josh Coalson8395d022001-07-12 21:25:22 +00002859#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002860{
Josh Coalson034dfab2001-04-27 19:10:23 +00002861 unsigned rice_parameter, partition_bits;
2862#ifndef NO_RICE_SEARCH
2863 unsigned best_partition_bits;
2864 unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
2865#endif
Josh Coalson94e02cd2001-01-25 10:41:06 +00002866 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002867 unsigned *parameters;
Josh Coalson94e02cd2001-01-25 10:41:06 +00002868
Josh Coalson1b689822001-05-31 20:11:02 +00002869 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
Josh Coalson2051dd42001-04-12 22:22:34 +00002870
Josh Coalsona37ba462002-08-19 21:36:39 +00002871 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
2872 parameters = partitioned_rice_contents->parameters;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00002873
Josh Coalson94e02cd2001-01-25 10:41:06 +00002874 if(partition_order == 0) {
2875 unsigned i;
Josh Coalson352e0f62001-03-20 22:55:50 +00002876
Josh Coalson034dfab2001-04-27 19:10:23 +00002877#ifndef NO_RICE_SEARCH
Josh Coalson60f77d72001-04-25 02:16:36 +00002878 if(rice_parameter_search_dist) {
Josh Coalson034dfab2001-04-27 19:10:23 +00002879 if(suggested_rice_parameter < rice_parameter_search_dist)
Josh Coalson60f77d72001-04-25 02:16:36 +00002880 min_rice_parameter = 0;
2881 else
Josh Coalson034dfab2001-04-27 19:10:23 +00002882 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
2883 max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
Josh Coalson8395d022001-07-12 21:25:22 +00002884 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002885#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002886 fprintf(stderr, "clipping rice_parameter (%u -> %u) @2\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2887#endif
Josh Coalson60f77d72001-04-25 02:16:36 +00002888 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00002889 }
2890 }
2891 else
2892 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
2893
2894 best_partition_bits = 0xffffffff;
2895 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2896#endif
2897#ifdef VARIABLE_RICE_BITS
2898#ifdef FLAC__SYMMETRIC_RICE
2899 partition_bits = (2+rice_parameter) * residual_samples;
2900#else
2901 const unsigned rice_parameter_estimate = rice_parameter-1;
2902 partition_bits = (1+rice_parameter) * residual_samples;
2903#endif
2904#else
2905 partition_bits = 0;
2906#endif
2907 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2908 for(i = 0; i < residual_samples; i++) {
2909#ifdef VARIABLE_RICE_BITS
2910#ifdef FLAC__SYMMETRIC_RICE
2911 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
2912#else
2913 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
2914#endif
2915#else
2916 partition_bits += FLAC__bitbuffer_rice_bits(residual[i], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
2917#endif
2918 }
2919#ifndef NO_RICE_SEARCH
2920 if(partition_bits < best_partition_bits) {
2921 best_rice_parameter = rice_parameter;
2922 best_partition_bits = partition_bits;
2923 }
2924 }
2925#endif
2926 parameters[0] = best_rice_parameter;
2927 bits_ += best_partition_bits;
2928 }
2929 else {
2930 unsigned partition, residual_sample, save_residual_sample, partition_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002931 unsigned partition_samples;
2932 FLAC__uint64 mean, k;
Josh Coalson8395d022001-07-12 21:25:22 +00002933 const unsigned partitions = 1u << partition_order;
2934 for(partition = residual_sample = 0; partition < partitions; partition++) {
2935 partition_samples = (residual_samples+predictor_order) >> partition_order;
2936 if(partition == 0) {
2937 if(partition_samples <= predictor_order)
2938 return false;
2939 else
2940 partition_samples -= predictor_order;
2941 }
2942 mean = 0;
2943 save_residual_sample = residual_sample;
2944 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++)
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002945 mean += abs_residual[residual_sample];
Josh Coalson8395d022001-07-12 21:25:22 +00002946 residual_sample = save_residual_sample;
2947#ifdef FLAC__SYMMETRIC_RICE
2948 mean += partition_samples >> 1; /* for rounding effect */
2949 mean /= partition_samples;
2950
2951 /* calc rice_parameter = floor(log2(mean)) */
2952 rice_parameter = 0;
2953 mean>>=1;
2954 while(mean) {
2955 rice_parameter++;
2956 mean >>= 1;
2957 }
2958#else
2959 /* calc rice_parameter ala LOCO-I */
Josh Coalsonb3347bd2001-07-16 18:06:41 +00002960 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
Josh Coalson8395d022001-07-12 21:25:22 +00002961 ;
2962#endif
2963 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002964#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002965 fprintf(stderr, "clipping rice_parameter (%u -> %u) @3\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2966#endif
2967 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2968 }
2969
2970#ifndef NO_RICE_SEARCH
2971 if(rice_parameter_search_dist) {
2972 if(rice_parameter < rice_parameter_search_dist)
2973 min_rice_parameter = 0;
2974 else
2975 min_rice_parameter = rice_parameter - rice_parameter_search_dist;
2976 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
2977 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00002978#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00002979 fprintf(stderr, "clipping rice_parameter (%u -> %u) @4\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
2980#endif
2981 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
2982 }
2983 }
2984 else
2985 min_rice_parameter = max_rice_parameter = rice_parameter;
2986
2987 best_partition_bits = 0xffffffff;
2988 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
2989#endif
2990#ifdef VARIABLE_RICE_BITS
2991#ifdef FLAC__SYMMETRIC_RICE
2992 partition_bits = (2+rice_parameter) * partition_samples;
2993#else
2994 const unsigned rice_parameter_estimate = rice_parameter-1;
2995 partition_bits = (1+rice_parameter) * partition_samples;
2996#endif
2997#else
2998 partition_bits = 0;
2999#endif
3000 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
3001 save_residual_sample = residual_sample;
3002 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
3003#ifdef VARIABLE_RICE_BITS
3004#ifdef FLAC__SYMMETRIC_RICE
3005 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
3006#else
3007 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
3008#endif
3009#else
3010 partition_bits += FLAC__bitbuffer_rice_bits(residual[residual_sample], rice_parameter); /* NOTE: we will need to pass in residual[] in addition to abs_residual[] */
3011#endif
3012 }
3013#ifndef NO_RICE_SEARCH
3014 if(rice_parameter != max_rice_parameter)
3015 residual_sample = save_residual_sample;
3016 if(partition_bits < best_partition_bits) {
3017 best_rice_parameter = rice_parameter;
3018 best_partition_bits = partition_bits;
3019 }
3020 }
3021#endif
3022 parameters[partition] = best_rice_parameter;
3023 bits_ += best_partition_bits;
3024 }
3025 }
3026
3027 *bits = bits_;
3028 return true;
3029}
3030
3031#ifdef DONT_ESTIMATE_RICE_BITS
Josh Coalson6fe72f72002-08-20 04:01:59 +00003032FLAC__bool set_partitioned_rice_with_precompute_(
3033 const FLAC__int32 residual[],
3034 const FLAC__uint64 abs_residual_partition_sums[],
3035 const unsigned raw_bits_per_partition[],
3036 const unsigned residual_samples,
3037 const unsigned predictor_order,
3038 const unsigned suggested_rice_parameter,
3039 const unsigned rice_parameter_search_dist,
3040 const unsigned partition_order,
3041 const FLAC__bool search_for_escapes,
3042 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
3043 unsigned *bits
3044)
Josh Coalson8395d022001-07-12 21:25:22 +00003045#else
Josh Coalson6fe72f72002-08-20 04:01:59 +00003046FLAC__bool set_partitioned_rice_with_precompute_(
3047 const FLAC__uint32 abs_residual[],
3048 const FLAC__uint64 abs_residual_partition_sums[],
3049 const unsigned raw_bits_per_partition[],
3050 const unsigned residual_samples,
3051 const unsigned predictor_order,
3052 const unsigned suggested_rice_parameter,
3053 const unsigned rice_parameter_search_dist,
3054 const unsigned partition_order,
3055 const FLAC__bool search_for_escapes,
3056 FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents,
3057 unsigned *bits
3058)
Josh Coalson8395d022001-07-12 21:25:22 +00003059#endif
3060{
3061 unsigned rice_parameter, partition_bits;
3062#ifndef NO_RICE_SEARCH
3063 unsigned best_partition_bits;
3064 unsigned min_rice_parameter, max_rice_parameter, best_rice_parameter = 0;
3065#endif
3066 unsigned flat_bits;
3067 unsigned bits_ = FLAC__ENTROPY_CODING_METHOD_TYPE_LEN + FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00003068 unsigned *parameters, *raw_bits;
Josh Coalson8395d022001-07-12 21:25:22 +00003069
3070 FLAC__ASSERT(suggested_rice_parameter < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER);
3071
Josh Coalsona37ba462002-08-19 21:36:39 +00003072 FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, max(6, partition_order));
3073 parameters = partitioned_rice_contents->parameters;
3074 raw_bits = partitioned_rice_contents->raw_bits;
Josh Coalsonb7023aa2002-08-17 15:23:43 +00003075
Josh Coalson8395d022001-07-12 21:25:22 +00003076 if(partition_order == 0) {
3077 unsigned i;
3078
3079#ifndef NO_RICE_SEARCH
3080 if(rice_parameter_search_dist) {
3081 if(suggested_rice_parameter < rice_parameter_search_dist)
3082 min_rice_parameter = 0;
3083 else
3084 min_rice_parameter = suggested_rice_parameter - rice_parameter_search_dist;
3085 max_rice_parameter = suggested_rice_parameter + rice_parameter_search_dist;
3086 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00003087#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00003088 fprintf(stderr, "clipping rice_parameter (%u -> %u) @5\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3089#endif
3090 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
3091 }
Josh Coalson60f77d72001-04-25 02:16:36 +00003092 }
3093 else
Josh Coalson034dfab2001-04-27 19:10:23 +00003094 min_rice_parameter = max_rice_parameter = suggested_rice_parameter;
Josh Coalson2051dd42001-04-12 22:22:34 +00003095
Josh Coalson034dfab2001-04-27 19:10:23 +00003096 best_partition_bits = 0xffffffff;
3097 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3098#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00003099#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003100#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson034dfab2001-04-27 19:10:23 +00003101 partition_bits = (2+rice_parameter) * residual_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003102#else
Josh Coalson352e0f62001-03-20 22:55:50 +00003103 const unsigned rice_parameter_estimate = rice_parameter-1;
Josh Coalson034dfab2001-04-27 19:10:23 +00003104 partition_bits = (1+rice_parameter) * residual_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003105#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003106#else
3107 partition_bits = 0;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003108#endif
Josh Coalson2051dd42001-04-12 22:22:34 +00003109 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
Josh Coalson352e0f62001-03-20 22:55:50 +00003110 for(i = 0; i < residual_samples; i++) {
3111#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003112#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson2051dd42001-04-12 22:22:34 +00003113 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter);
Josh Coalson94e02cd2001-01-25 10:41:06 +00003114#else
Josh Coalson2051dd42001-04-12 22:22:34 +00003115 partition_bits += VARIABLE_RICE_BITS(abs_residual[i], rice_parameter_estimate);
Josh Coalsonb9433f92001-03-17 01:07:00 +00003116#endif
3117#else
Josh Coalson2051dd42001-04-12 22:22:34 +00003118 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 +00003119#endif
Josh Coalson2051dd42001-04-12 22:22:34 +00003120 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003121#ifndef NO_RICE_SEARCH
3122 if(partition_bits < best_partition_bits) {
3123 best_rice_parameter = rice_parameter;
3124 best_partition_bits = partition_bits;
Josh Coalson352e0f62001-03-20 22:55:50 +00003125 }
3126 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003127#endif
Josh Coalson8395d022001-07-12 21:25:22 +00003128 if(search_for_escapes) {
3129 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;
3130 if(flat_bits <= best_partition_bits) {
3131 raw_bits[0] = raw_bits_per_partition[0];
3132 best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3133 best_partition_bits = flat_bits;
3134 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003135 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003136 parameters[0] = best_rice_parameter;
3137 bits_ += best_partition_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003138 }
3139 else {
Josh Coalson4dacd192001-06-06 21:11:44 +00003140 unsigned partition, residual_sample, save_residual_sample, partition_sample;
Josh Coalsonb3347bd2001-07-16 18:06:41 +00003141 unsigned partition_samples;
3142 FLAC__uint64 mean, k;
Josh Coalsond4e0ddb2001-04-18 02:20:52 +00003143 const unsigned partitions = 1u << partition_order;
Josh Coalson4dacd192001-06-06 21:11:44 +00003144 for(partition = residual_sample = 0; partition < partitions; partition++) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00003145 partition_samples = (residual_samples+predictor_order) >> partition_order;
Josh Coalson034dfab2001-04-27 19:10:23 +00003146 if(partition == 0) {
Josh Coalson94e02cd2001-01-25 10:41:06 +00003147 if(partition_samples <= predictor_order)
3148 return false;
3149 else
3150 partition_samples -= predictor_order;
3151 }
Josh Coalson05d20792001-06-29 23:12:26 +00003152 mean = abs_residual_partition_sums[partition];
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003153#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson05d20792001-06-29 23:12:26 +00003154 mean += partition_samples >> 1; /* for rounding effect */
3155 mean /= partition_samples;
3156
Josh Coalson034dfab2001-04-27 19:10:23 +00003157 /* calc rice_parameter = floor(log2(mean)) */
3158 rice_parameter = 0;
3159 mean>>=1;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003160 while(mean) {
Josh Coalson034dfab2001-04-27 19:10:23 +00003161 rice_parameter++;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003162 mean >>= 1;
3163 }
Josh Coalsonb9433f92001-03-17 01:07:00 +00003164#else
Josh Coalson05d20792001-06-29 23:12:26 +00003165 /* calc rice_parameter ala LOCO-I */
Josh Coalsonb3347bd2001-07-16 18:06:41 +00003166 for(rice_parameter = 0, k = partition_samples; k < mean; rice_parameter++, k <<= 1)
Josh Coalson05d20792001-06-29 23:12:26 +00003167 ;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003168#endif
Josh Coalson8395d022001-07-12 21:25:22 +00003169 if(rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00003170#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00003171 fprintf(stderr, "clipping rice_parameter (%u -> %u) @6\n", rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3172#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003173 rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00003174 }
Josh Coalson60f77d72001-04-25 02:16:36 +00003175
Josh Coalson034dfab2001-04-27 19:10:23 +00003176#ifndef NO_RICE_SEARCH
Josh Coalson60f77d72001-04-25 02:16:36 +00003177 if(rice_parameter_search_dist) {
Josh Coalson034dfab2001-04-27 19:10:23 +00003178 if(rice_parameter < rice_parameter_search_dist)
Josh Coalson60f77d72001-04-25 02:16:36 +00003179 min_rice_parameter = 0;
3180 else
Josh Coalson034dfab2001-04-27 19:10:23 +00003181 min_rice_parameter = rice_parameter - rice_parameter_search_dist;
3182 max_rice_parameter = rice_parameter + rice_parameter_search_dist;
Josh Coalson8395d022001-07-12 21:25:22 +00003183 if(max_rice_parameter >= FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
Josh Coalson31209492001-07-18 23:43:01 +00003184#ifdef DEBUG_VERBOSE
Josh Coalson8395d022001-07-12 21:25:22 +00003185 fprintf(stderr, "clipping rice_parameter (%u -> %u) @7\n", max_rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1);
3186#endif
Josh Coalson60f77d72001-04-25 02:16:36 +00003187 max_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER - 1;
Josh Coalson8395d022001-07-12 21:25:22 +00003188 }
Josh Coalson60f77d72001-04-25 02:16:36 +00003189 }
3190 else
3191 min_rice_parameter = max_rice_parameter = rice_parameter;
Josh Coalson60f77d72001-04-25 02:16:36 +00003192
Josh Coalson034dfab2001-04-27 19:10:23 +00003193 best_partition_bits = 0xffffffff;
3194 for(rice_parameter = min_rice_parameter; rice_parameter <= max_rice_parameter; rice_parameter++) {
3195#endif
Josh Coalson352e0f62001-03-20 22:55:50 +00003196#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003197#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson034dfab2001-04-27 19:10:23 +00003198 partition_bits = (2+rice_parameter) * partition_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003199#else
Josh Coalson034dfab2001-04-27 19:10:23 +00003200 const unsigned rice_parameter_estimate = rice_parameter-1;
3201 partition_bits = (1+rice_parameter) * partition_samples;
Josh Coalsonb9433f92001-03-17 01:07:00 +00003202#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003203#else
3204 partition_bits = 0;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003205#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003206 partition_bits += FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
Josh Coalson4dacd192001-06-06 21:11:44 +00003207 save_residual_sample = residual_sample;
3208 for(partition_sample = 0; partition_sample < partition_samples; residual_sample++, partition_sample++) {
Josh Coalson352e0f62001-03-20 22:55:50 +00003209#ifdef VARIABLE_RICE_BITS
Josh Coalsonbb6712e2001-04-24 22:54:07 +00003210#ifdef FLAC__SYMMETRIC_RICE
Josh Coalson4dacd192001-06-06 21:11:44 +00003211 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter);
Josh Coalson94e02cd2001-01-25 10:41:06 +00003212#else
Josh Coalson4dacd192001-06-06 21:11:44 +00003213 partition_bits += VARIABLE_RICE_BITS(abs_residual[residual_sample], rice_parameter_estimate);
Josh Coalsonb9433f92001-03-17 01:07:00 +00003214#endif
3215#else
Josh Coalson4dacd192001-06-06 21:11:44 +00003216 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 +00003217#endif
Josh Coalson034dfab2001-04-27 19:10:23 +00003218 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003219#ifndef NO_RICE_SEARCH
Josh Coalson4dacd192001-06-06 21:11:44 +00003220 if(rice_parameter != max_rice_parameter)
3221 residual_sample = save_residual_sample;
Josh Coalson034dfab2001-04-27 19:10:23 +00003222 if(partition_bits < best_partition_bits) {
3223 best_rice_parameter = rice_parameter;
3224 best_partition_bits = partition_bits;
3225 }
Josh Coalson2051dd42001-04-12 22:22:34 +00003226 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003227#endif
Josh Coalson8395d022001-07-12 21:25:22 +00003228 if(search_for_escapes) {
3229 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;
3230 if(flat_bits <= best_partition_bits) {
3231 raw_bits[partition] = raw_bits_per_partition[partition];
3232 best_rice_parameter = FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
3233 best_partition_bits = flat_bits;
3234 }
Josh Coalson2051dd42001-04-12 22:22:34 +00003235 }
Josh Coalson034dfab2001-04-27 19:10:23 +00003236 parameters[partition] = best_rice_parameter;
3237 bits_ += best_partition_bits;
Josh Coalson94e02cd2001-01-25 10:41:06 +00003238 }
3239 }
3240
3241 *bits = bits_;
3242 return true;
3243}
Josh Coalson859bc542001-03-27 22:22:27 +00003244
Josh Coalsonf1eff452002-07-31 07:05:33 +00003245unsigned get_wasted_bits_(FLAC__int32 signal[], unsigned samples)
Josh Coalson859bc542001-03-27 22:22:27 +00003246{
3247 unsigned i, shift;
Josh Coalson77e3f312001-06-23 03:03:24 +00003248 FLAC__int32 x = 0;
Josh Coalson859bc542001-03-27 22:22:27 +00003249
3250 for(i = 0; i < samples && !(x&1); i++)
3251 x |= signal[i];
3252
3253 if(x == 0) {
3254 shift = 0;
3255 }
3256 else {
3257 for(shift = 0; !(x&1); shift++)
3258 x >>= 1;
3259 }
3260
3261 if(shift > 0) {
3262 for(i = 0; i < samples; i++)
3263 signal[i] >>= shift;
3264 }
3265
3266 return shift;
3267}
Josh Coalsond86e03b2002-08-03 21:56:15 +00003268
3269void append_to_verify_fifo_(verify_input_fifo *fifo, const FLAC__int32 * const input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3270{
3271 unsigned channel;
3272
3273 for(channel = 0; channel < channels; channel++)
3274 memcpy(&fifo->data[channel][fifo->tail], &input[channel][input_offset], sizeof(FLAC__int32) * wide_samples);
3275
3276 fifo->tail += wide_samples;
3277
3278 FLAC__ASSERT(fifo->tail <= fifo->size);
3279}
3280
3281void append_to_verify_fifo_interleaved_(verify_input_fifo *fifo, const FLAC__int32 input[], unsigned input_offset, unsigned channels, unsigned wide_samples)
3282{
3283 unsigned channel;
3284 unsigned sample, wide_sample;
3285 unsigned tail = fifo->tail;
3286
3287 sample = input_offset * channels;
3288 for(wide_sample = 0; wide_sample < wide_samples; wide_sample++) {
3289 for(channel = 0; channel < channels; channel++)
3290 fifo->data[channel][tail] = input[sample++];
3291 tail++;
3292 }
3293 fifo->tail = tail;
3294
3295 FLAC__ASSERT(fifo->tail <= fifo->size);
3296}
3297
3298FLAC__StreamDecoderReadStatus verify_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], unsigned *bytes, void *client_data)
3299{
3300 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3301 const unsigned encoded_bytes = encoder->private_->verify.output.bytes;
3302 (void)decoder;
3303
3304 if(encoder->private_->verify.needs_magic_hack) {
3305 FLAC__ASSERT(*bytes >= FLAC__STREAM_SYNC_LENGTH);
3306 *bytes = FLAC__STREAM_SYNC_LENGTH;
3307 memcpy(buffer, FLAC__STREAM_SYNC_STRING, *bytes);
3308 encoder->private_->verify.needs_magic_hack = false;
3309 }
3310 else {
3311 if(encoded_bytes == 0) {
Josh Coalsonfc2b7372002-08-16 05:39:34 +00003312 /*
3313 * If we get here, a FIFO underflow has occurred,
3314 * which means there is a bug somewhere.
3315 */
3316 FLAC__ASSERT(0);
Josh Coalsond86e03b2002-08-03 21:56:15 +00003317 return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
3318 }
3319 else if(encoded_bytes < *bytes)
3320 *bytes = encoded_bytes;
3321 memcpy(buffer, encoder->private_->verify.output.data, *bytes);
3322 encoder->private_->verify.output.data += *bytes;
3323 encoder->private_->verify.output.bytes -= *bytes;
3324 }
3325
3326 return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
3327}
3328
3329FLAC__StreamDecoderWriteStatus verify_write_callback_(const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[], void *client_data)
3330{
3331 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder *)client_data;
3332 unsigned channel;
3333 const unsigned channels = FLAC__stream_decoder_get_channels(decoder);
3334 const unsigned blocksize = frame->header.blocksize;
3335 const unsigned bytes_per_block = sizeof(FLAC__int32) * blocksize;
3336
3337 for(channel = 0; channel < channels; channel++) {
3338 if(0 != memcmp(buffer[channel], encoder->private_->verify.input_fifo.data[channel], bytes_per_block)) {
3339 unsigned i, sample = 0;
3340 FLAC__int32 expect = 0, got = 0;
3341
3342 for(i = 0; i < blocksize; i++) {
3343 if(buffer[channel][i] != encoder->private_->verify.input_fifo.data[channel][i]) {
3344 sample = i;
3345 expect = (FLAC__int32)encoder->private_->verify.input_fifo.data[channel][i];
3346 got = (FLAC__int32)buffer[channel][i];
3347 break;
3348 }
3349 }
3350 FLAC__ASSERT(i < blocksize);
3351 FLAC__ASSERT(frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
3352 encoder->private_->verify.error_stats.absolute_sample = frame->header.number.sample_number + sample;
Josh Coalson5f39e9f2002-08-21 05:27:01 +00003353 encoder->private_->verify.error_stats.frame_number = (unsigned)(frame->header.number.sample_number / blocksize);
Josh Coalsond86e03b2002-08-03 21:56:15 +00003354 encoder->private_->verify.error_stats.channel = channel;
3355 encoder->private_->verify.error_stats.sample = sample;
3356 encoder->private_->verify.error_stats.expected = expect;
3357 encoder->private_->verify.error_stats.got = got;
3358 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_MISMATCH_IN_AUDIO_DATA;
3359 return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
3360 }
3361 }
3362 /* dequeue the frame from the fifo */
3363 for(channel = 0; channel < channels; channel++) {
3364 memmove(&encoder->private_->verify.input_fifo.data[channel][0], &encoder->private_->verify.input_fifo.data[channel][blocksize], encoder->private_->verify.input_fifo.tail - blocksize);
3365 }
3366 encoder->private_->verify.input_fifo.tail -= blocksize;
3367 return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
3368}
3369
3370void verify_metadata_callback_(const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data)
3371{
3372 (void)decoder, (void)metadata, (void)client_data;
3373}
3374
3375void verify_error_callback_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data)
3376{
3377 FLAC__StreamEncoder *encoder = (FLAC__StreamEncoder*)client_data;
3378 (void)decoder, (void)status;
3379 encoder->protected_->state = FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR;
3380}