1 /* test_libFLAC - Unit tester for libFLAC
2 * Copyright (C) 2002,2003,2004,2005,2006 Josh Coalson
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
28 #include "FLAC/assert.h"
29 #include "FLAC/stream_encoder.h"
30 #include "share/grabbag.h"
31 #include "test_libs_common/file_utils_flac.h"
32 #include "test_libs_common/metadata_utils.h"
35 LAYER_STREAM = 0, /* FLAC__stream_encoder_init_stream() without seeking */
36 LAYER_SEEKABLE_STREAM, /* FLAC__stream_encoder_init_stream() with seeking */
37 LAYER_FILE, /* FLAC__stream_encoder_init_FILE() */
38 LAYER_FILENAME /* FLAC__stream_encoder_init_file() */
41 static const char * const LayerString[] = {
48 static FLAC__StreamMetadata streaminfo_, padding_, seektable_, application1_, application2_, vorbiscomment_, cuesheet_, picture_, unknown_;
49 static FLAC__StreamMetadata *metadata_sequence_[] = { &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_ };
50 static const unsigned num_metadata_ = sizeof(metadata_sequence_) / sizeof(metadata_sequence_[0]);
51 static const char *flacfilename_ = "metadata.flac";
53 static FLAC__bool die_(const char *msg)
55 printf("ERROR: %s\n", msg);
59 static FLAC__bool die_s_(const char *msg, const FLAC__StreamEncoder *encoder)
61 FLAC__StreamEncoderState state = FLAC__stream_encoder_get_state(encoder);
64 printf("FAILED, %s", msg);
68 printf(", state = %u (%s)\n", (unsigned)state, FLAC__StreamEncoderStateString[state]);
69 if(state == FLAC__STREAM_ENCODER_VERIFY_DECODER_ERROR) {
70 FLAC__StreamDecoderState dstate = FLAC__stream_encoder_get_verify_decoder_state(encoder);
71 printf(" verify decoder state = %u (%s)\n", (unsigned)dstate, FLAC__StreamDecoderStateString[dstate]);
77 static void init_metadata_blocks_()
79 mutils__init_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
82 static void free_metadata_blocks_()
84 mutils__free_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
87 static FLAC__StreamEncoderWriteStatus stream_encoder_write_callback_(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
89 (void)encoder, (void)buffer, (void)bytes, (void)samples, (void)current_frame, (void)client_data;
90 return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
93 static FLAC__StreamEncoderSeekStatus stream_encoder_seek_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data)
95 (void)encoder, (void)absolute_byte_offset, (void)client_data;
96 return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
99 static FLAC__StreamEncoderTellStatus stream_encoder_tell_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data)
101 (void)encoder, (void)client_data;
102 *absolute_byte_offset = 0;
103 return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
106 static void stream_encoder_metadata_callback_(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
108 (void)encoder, (void)metadata, (void)client_data;
111 static void stream_encoder_progress_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data)
113 (void)encoder, (void)bytes_written, (void)samples_written, (void)frames_written, (void)total_frames_estimate, (void)client_data;
116 static FLAC__bool test_stream_encoder(Layer layer)
118 FLAC__StreamEncoder *encoder;
119 FLAC__StreamEncoderState state;
120 FLAC__StreamDecoderState dstate;
122 FLAC__int32 samples[1024];
123 FLAC__int32 *samples_array[1];
126 samples_array[0] = samples;
128 printf("\n+++ libFLAC unit test: FLAC__StreamEncoder (layer: %s)\n\n", LayerString[layer]);
130 printf("testing FLAC__stream_encoder_new()... ");
131 encoder = FLAC__stream_encoder_new();
133 printf("FAILED, returned NULL\n");
138 printf("testing FLAC__stream_encoder_set_verify()... ");
139 if(!FLAC__stream_encoder_set_verify(encoder, true))
140 return die_s_("returned false", encoder);
143 printf("testing FLAC__stream_encoder_set_streamable_subset()... ");
144 if(!FLAC__stream_encoder_set_streamable_subset(encoder, true))
145 return die_s_("returned false", encoder);
148 printf("testing FLAC__stream_encoder_set_do_mid_side_stereo()... ");
149 if(!FLAC__stream_encoder_set_do_mid_side_stereo(encoder, false))
150 return die_s_("returned false", encoder);
153 printf("testing FLAC__stream_encoder_set_loose_mid_side_stereo()... ");
154 if(!FLAC__stream_encoder_set_loose_mid_side_stereo(encoder, false))
155 return die_s_("returned false", encoder);
158 printf("testing FLAC__stream_encoder_set_channels()... ");
159 if(!FLAC__stream_encoder_set_channels(encoder, streaminfo_.data.stream_info.channels))
160 return die_s_("returned false", encoder);
163 printf("testing FLAC__stream_encoder_set_bits_per_sample()... ");
164 if(!FLAC__stream_encoder_set_bits_per_sample(encoder, streaminfo_.data.stream_info.bits_per_sample))
165 return die_s_("returned false", encoder);
168 printf("testing FLAC__stream_encoder_set_sample_rate()... ");
169 if(!FLAC__stream_encoder_set_sample_rate(encoder, streaminfo_.data.stream_info.sample_rate))
170 return die_s_("returned false", encoder);
173 printf("testing FLAC__stream_encoder_set_blocksize()... ");
174 if(!FLAC__stream_encoder_set_blocksize(encoder, streaminfo_.data.stream_info.min_blocksize))
175 return die_s_("returned false", encoder);
178 printf("testing FLAC__stream_encoder_set_max_lpc_order()... ");
179 if(!FLAC__stream_encoder_set_max_lpc_order(encoder, 0))
180 return die_s_("returned false", encoder);
183 printf("testing FLAC__stream_encoder_set_qlp_coeff_precision()... ");
184 if(!FLAC__stream_encoder_set_qlp_coeff_precision(encoder, 0))
185 return die_s_("returned false", encoder);
188 printf("testing FLAC__stream_encoder_set_do_qlp_coeff_prec_search()... ");
189 if(!FLAC__stream_encoder_set_do_qlp_coeff_prec_search(encoder, false))
190 return die_s_("returned false", encoder);
193 printf("testing FLAC__stream_encoder_set_do_escape_coding()... ");
194 if(!FLAC__stream_encoder_set_do_escape_coding(encoder, false))
195 return die_s_("returned false", encoder);
198 printf("testing FLAC__stream_encoder_set_do_exhaustive_model_search()... ");
199 if(!FLAC__stream_encoder_set_do_exhaustive_model_search(encoder, false))
200 return die_s_("returned false", encoder);
203 printf("testing FLAC__stream_encoder_set_min_residual_partition_order()... ");
204 if(!FLAC__stream_encoder_set_min_residual_partition_order(encoder, 0))
205 return die_s_("returned false", encoder);
208 printf("testing FLAC__stream_encoder_set_max_residual_partition_order()... ");
209 if(!FLAC__stream_encoder_set_max_residual_partition_order(encoder, 0))
210 return die_s_("returned false", encoder);
213 printf("testing FLAC__stream_encoder_set_rice_parameter_search_dist()... ");
214 if(!FLAC__stream_encoder_set_rice_parameter_search_dist(encoder, 0))
215 return die_s_("returned false", encoder);
218 printf("testing FLAC__stream_encoder_set_total_samples_estimate()... ");
219 if(!FLAC__stream_encoder_set_total_samples_estimate(encoder, streaminfo_.data.stream_info.total_samples))
220 return die_s_("returned false", encoder);
223 printf("testing FLAC__stream_encoder_set_metadata()... ");
224 if(!FLAC__stream_encoder_set_metadata(encoder, metadata_sequence_, num_metadata_))
225 return die_s_("returned false", encoder);
230 printf("testing FLAC__stream_encoder_init_stream()... ");
231 if(FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, /*seek_callback=*/0, /*tell_callback=*/0, stream_encoder_metadata_callback_, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
232 return die_s_(0, encoder);
234 case LAYER_SEEKABLE_STREAM:
235 printf("testing FLAC__stream_encoder_init_stream()... ");
236 if(FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, stream_encoder_seek_callback_, stream_encoder_tell_callback_, /*metadata_callback=*/0, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
237 return die_s_(0, encoder);
240 printf("opening file for FLAC output... ");
241 file = fopen(flacfilename_, "w+b");
243 printf("ERROR (%s)\n", strerror(errno));
248 printf("testing FLAC__stream_encoder_init_FILE()... ");
249 if(FLAC__stream_encoder_init_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
250 return die_s_(0, encoder);
253 printf("testing FLAC__stream_encoder_init_file()... ");
254 if(FLAC__stream_encoder_init_file(encoder, flacfilename_, stream_encoder_progress_callback_, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
255 return die_s_(0, encoder);
258 die_("internal error 001");
263 printf("testing FLAC__stream_encoder_get_state()... ");
264 state = FLAC__stream_encoder_get_state(encoder);
265 printf("returned state = %u (%s)... OK\n", (unsigned)state, FLAC__StreamEncoderStateString[state]);
267 printf("testing FLAC__stream_encoder_get_verify_decoder_state()... ");
268 dstate = FLAC__stream_encoder_get_verify_decoder_state(encoder);
269 printf("returned state = %u (%s)... OK\n", (unsigned)dstate, FLAC__StreamDecoderStateString[dstate]);
272 FLAC__uint64 absolute_sample;
273 unsigned frame_number;
276 FLAC__int32 expected;
279 printf("testing FLAC__stream_encoder_get_verify_decoder_error_stats()... ");
280 FLAC__stream_encoder_get_verify_decoder_error_stats(encoder, &absolute_sample, &frame_number, &channel, &sample, &expected, &got);
284 printf("testing FLAC__stream_encoder_get_verify()... ");
285 if(FLAC__stream_encoder_get_verify(encoder) != true) {
286 printf("FAILED, expected true, got false\n");
291 printf("testing FLAC__stream_encoder_get_streamable_subset()... ");
292 if(FLAC__stream_encoder_get_streamable_subset(encoder) != true) {
293 printf("FAILED, expected true, got false\n");
298 printf("testing FLAC__stream_encoder_get_do_mid_side_stereo()... ");
299 if(FLAC__stream_encoder_get_do_mid_side_stereo(encoder) != false) {
300 printf("FAILED, expected false, got true\n");
305 printf("testing FLAC__stream_encoder_get_loose_mid_side_stereo()... ");
306 if(FLAC__stream_encoder_get_loose_mid_side_stereo(encoder) != false) {
307 printf("FAILED, expected false, got true\n");
312 printf("testing FLAC__stream_encoder_get_channels()... ");
313 if(FLAC__stream_encoder_get_channels(encoder) != streaminfo_.data.stream_info.channels) {
314 printf("FAILED, expected %u, got %u\n", streaminfo_.data.stream_info.channels, FLAC__stream_encoder_get_channels(encoder));
319 printf("testing FLAC__stream_encoder_get_bits_per_sample()... ");
320 if(FLAC__stream_encoder_get_bits_per_sample(encoder) != streaminfo_.data.stream_info.bits_per_sample) {
321 printf("FAILED, expected %u, got %u\n", streaminfo_.data.stream_info.bits_per_sample, FLAC__stream_encoder_get_bits_per_sample(encoder));
326 printf("testing FLAC__stream_encoder_get_sample_rate()... ");
327 if(FLAC__stream_encoder_get_sample_rate(encoder) != streaminfo_.data.stream_info.sample_rate) {
328 printf("FAILED, expected %u, got %u\n", streaminfo_.data.stream_info.sample_rate, FLAC__stream_encoder_get_sample_rate(encoder));
333 printf("testing FLAC__stream_encoder_get_blocksize()... ");
334 if(FLAC__stream_encoder_get_blocksize(encoder) != streaminfo_.data.stream_info.min_blocksize) {
335 printf("FAILED, expected %u, got %u\n", streaminfo_.data.stream_info.min_blocksize, FLAC__stream_encoder_get_blocksize(encoder));
340 printf("testing FLAC__stream_encoder_get_max_lpc_order()... ");
341 if(FLAC__stream_encoder_get_max_lpc_order(encoder) != 0) {
342 printf("FAILED, expected %u, got %u\n", 0, FLAC__stream_encoder_get_max_lpc_order(encoder));
347 printf("testing FLAC__stream_encoder_get_qlp_coeff_precision()... ");
348 (void)FLAC__stream_encoder_get_qlp_coeff_precision(encoder);
349 /* we asked the encoder to auto select this so we accept anything */
352 printf("testing FLAC__stream_encoder_get_do_qlp_coeff_prec_search()... ");
353 if(FLAC__stream_encoder_get_do_qlp_coeff_prec_search(encoder) != false) {
354 printf("FAILED, expected false, got true\n");
359 printf("testing FLAC__stream_encoder_get_do_escape_coding()... ");
360 if(FLAC__stream_encoder_get_do_escape_coding(encoder) != false) {
361 printf("FAILED, expected false, got true\n");
366 printf("testing FLAC__stream_encoder_get_do_exhaustive_model_search()... ");
367 if(FLAC__stream_encoder_get_do_exhaustive_model_search(encoder) != false) {
368 printf("FAILED, expected false, got true\n");
373 printf("testing FLAC__stream_encoder_get_min_residual_partition_order()... ");
374 if(FLAC__stream_encoder_get_min_residual_partition_order(encoder) != 0) {
375 printf("FAILED, expected %u, got %u\n", 0, FLAC__stream_encoder_get_min_residual_partition_order(encoder));
380 printf("testing FLAC__stream_encoder_get_max_residual_partition_order()... ");
381 if(FLAC__stream_encoder_get_max_residual_partition_order(encoder) != 0) {
382 printf("FAILED, expected %u, got %u\n", 0, FLAC__stream_encoder_get_max_residual_partition_order(encoder));
387 printf("testing FLAC__stream_encoder_get_rice_parameter_search_dist()... ");
388 if(FLAC__stream_encoder_get_rice_parameter_search_dist(encoder) != 0) {
389 printf("FAILED, expected %u, got %u\n", 0, FLAC__stream_encoder_get_rice_parameter_search_dist(encoder));
394 printf("testing FLAC__stream_encoder_get_total_samples_estimate()... ");
395 if(FLAC__stream_encoder_get_total_samples_estimate(encoder) != streaminfo_.data.stream_info.total_samples) {
396 printf("FAILED, expected %llu, got %llu\n", streaminfo_.data.stream_info.total_samples, FLAC__stream_encoder_get_total_samples_estimate(encoder));
401 /* init the dummy sample buffer */
402 for(i = 0; i < sizeof(samples) / sizeof(FLAC__int32); i++)
405 printf("testing FLAC__stream_encoder_process()... ");
406 if(!FLAC__stream_encoder_process(encoder, (const FLAC__int32 * const *)samples_array, sizeof(samples) / sizeof(FLAC__int32)))
407 return die_s_("returned false", encoder);
410 printf("testing FLAC__stream_encoder_process_interleaved()... ");
411 if(!FLAC__stream_encoder_process_interleaved(encoder, samples, sizeof(samples) / sizeof(FLAC__int32)))
412 return die_s_("returned false", encoder);
415 printf("testing FLAC__stream_encoder_finish()... ");
416 FLAC__stream_encoder_finish(encoder);
419 printf("testing FLAC__stream_encoder_delete()... ");
420 FLAC__stream_encoder_delete(encoder);
423 printf("\nPASSED!\n");
428 FLAC__bool test_encoders()
430 init_metadata_blocks_();
432 if(!test_stream_encoder(LAYER_STREAM))
435 if(!test_stream_encoder(LAYER_SEEKABLE_STREAM))
438 if(!test_stream_encoder(LAYER_FILE))
441 if(!test_stream_encoder(LAYER_FILENAME))
444 (void) grabbag__file_remove_file(flacfilename_);
445 free_metadata_blocks_();