/* test_libFLAC++ - Unit tester for libFLAC++
- * Copyright (C) 2002,2003,2004,2005,2006 Josh Coalson
+ * Copyright (C) 2002,2003,2004,2005,2006,2007,2008,2009 Josh Coalson
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "encoders.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
+#include "share/compat.h"
typedef enum {
LAYER_STREAM = 0, /* FLAC__stream_encoder_init_stream() without seeking */
static const char *flacfilename(bool is_ogg)
{
- return is_ogg? "metadata.ogg" : "metadata.flac";
+ return is_ogg? "metadata.oga" : "metadata.flac";
}
static bool die_(const char *msg)
class StreamEncoder : public FLAC::Encoder::Stream {
public:
Layer layer_;
+ FILE *file_;
- StreamEncoder(Layer layer): FLAC::Encoder::Stream(), layer_(layer) { }
+ StreamEncoder(Layer layer): FLAC::Encoder::Stream(), layer_(layer), file_(0) { }
~StreamEncoder() { }
// from FLAC::Encoder::Stream
::FLAC__StreamEncoderReadStatus StreamEncoder::read_callback(FLAC__byte buffer[], size_t *bytes)
{
- (void)buffer, (void)bytes;
-
- return ::FLAC__STREAM_ENCODER_READ_STATUS_CONTINUE;
+ if(*bytes > 0) {
+ *bytes = fread(buffer, sizeof(FLAC__byte), *bytes, file_);
+ if(ferror(file_))
+ return ::FLAC__STREAM_ENCODER_READ_STATUS_ABORT;
+ else if(*bytes == 0)
+ return ::FLAC__STREAM_ENCODER_READ_STATUS_END_OF_STREAM;
+ else
+ return ::FLAC__STREAM_ENCODER_READ_STATUS_CONTINUE;
+ }
+ else
+ return ::FLAC__STREAM_ENCODER_READ_STATUS_ABORT;
}
::FLAC__StreamEncoderWriteStatus StreamEncoder::write_callback(const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame)
{
- (void)buffer, (void)bytes, (void)samples, (void)current_frame;
+ (void)samples, (void)current_frame;
- return ::FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
+ if(fwrite(buffer, 1, bytes, file_) != bytes)
+ return ::FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
+ else
+ return ::FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
}
::FLAC__StreamEncoderSeekStatus StreamEncoder::seek_callback(FLAC__uint64 absolute_byte_offset)
{
- (void)absolute_byte_offset;
-
- return layer_==LAYER_STREAM? ::FLAC__STREAM_ENCODER_SEEK_STATUS_UNSUPPORTED : ::FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
+ if(layer_==LAYER_STREAM)
+ return ::FLAC__STREAM_ENCODER_SEEK_STATUS_UNSUPPORTED;
+ else if(fseeko(file_, (FLAC__off_t)absolute_byte_offset, SEEK_SET) < 0)
+ return FLAC__STREAM_ENCODER_SEEK_STATUS_ERROR;
+ else
+ return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
}
::FLAC__StreamEncoderTellStatus StreamEncoder::tell_callback(FLAC__uint64 *absolute_byte_offset)
{
- *absolute_byte_offset = 0;
-
- return layer_==LAYER_STREAM? ::FLAC__STREAM_ENCODER_TELL_STATUS_UNSUPPORTED : ::FLAC__STREAM_ENCODER_TELL_STATUS_OK;
+ FLAC__off_t pos;
+ if(layer_==LAYER_STREAM)
+ return ::FLAC__STREAM_ENCODER_TELL_STATUS_UNSUPPORTED;
+ else if((pos = ftello(file_)) < 0)
+ return FLAC__STREAM_ENCODER_TELL_STATUS_ERROR;
+ else {
+ *absolute_byte_offset = (FLAC__uint64)pos;
+ return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
+ }
}
void StreamEncoder::metadata_callback(const ::FLAC__StreamMetadata *metadata)
return die_s_("returned false", encoder);
printf("OK\n");
- printf("testing set_do_mid_side_stereo()... ");
- if(!encoder->set_do_mid_side_stereo(false))
- return die_s_("returned false", encoder);
- printf("OK\n");
-
- printf("testing set_loose_mid_side_stereo()... ");
- if(!encoder->set_loose_mid_side_stereo(false))
- return die_s_("returned false", encoder);
- printf("OK\n");
-
printf("testing set_channels()... ");
if(!encoder->set_channels(streaminfo_.data.stream_info.channels))
return die_s_("returned false", encoder);
return die_s_("returned false", encoder);
printf("OK\n");
+ printf("testing set_compression_level()... ");
+ if(!encoder->set_compression_level((unsigned)(-1)))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
printf("testing set_blocksize()... ");
if(!encoder->set_blocksize(streaminfo_.data.stream_info.min_blocksize))
return die_s_("returned false", encoder);
printf("OK\n");
+ printf("testing set_do_mid_side_stereo()... ");
+ if(!encoder->set_do_mid_side_stereo(false))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
+ printf("testing set_loose_mid_side_stereo()... ");
+ if(!encoder->set_loose_mid_side_stereo(false))
+ return die_s_("returned false", encoder);
+ printf("OK\n");
+
printf("testing set_max_lpc_order()... ");
if(!encoder->set_max_lpc_order(0))
return die_s_("returned false", encoder);
return die_s_("returned false", encoder);
printf("OK\n");
+ if(layer < LAYER_FILENAME) {
+ printf("opening file for FLAC output... ");
+ file = ::fopen(flacfilename(is_ogg), "w+b");
+ if(0 == file) {
+ printf("ERROR (%s)\n", strerror(errno));
+ return false;
+ }
+ printf("OK\n");
+ if(layer < LAYER_FILE)
+ dynamic_cast<StreamEncoder*>(encoder)->file_ = file;
+ }
+
switch(layer) {
case LAYER_STREAM:
case LAYER_SEEKABLE_STREAM:
init_status = is_ogg? encoder->init_ogg() : encoder->init();
break;
case LAYER_FILE:
- printf("opening file for FLAC output... ");
- file = ::fopen(flacfilename(is_ogg), "w+b");
- if(0 == file) {
- printf("ERROR (%s)\n", strerror(errno));
- return false;
- }
- printf("OK\n");
-
printf("testing init%s()... ", is_ogg? "_ogg":"");
init_status = is_ogg?
dynamic_cast<FLAC::Encoder::File*>(encoder)->init_ogg(file) :
printf("testing get_total_samples_estimate()... ");
if(encoder->get_total_samples_estimate() != streaminfo_.data.stream_info.total_samples) {
- printf("FAILED, expected %llu, got %llu\n", streaminfo_.data.stream_info.total_samples, encoder->get_total_samples_estimate());
+ printf("FAILED, expected %" PRIu64 ", got %" PRIu64 "\n", streaminfo_.data.stream_info.total_samples, encoder->get_total_samples_estimate());
return false;
}
printf("OK\n");
printf("OK\n");
printf("testing finish()... ");
- encoder->finish();
+ if(!encoder->finish()) {
+ state = encoder->get_state();
+ printf("FAILED, returned false, state = %u (%s)\n", (unsigned)((::FLAC__StreamEncoderState)state), state.as_cstring());
+ return false;
+ }
printf("OK\n");
+ if(layer < LAYER_FILE)
+ ::fclose(dynamic_cast<StreamEncoder*>(encoder)->file_);
+
printf("freeing encoder instance... ");
delete encoder;
printf("OK\n");