configure.ac : Detect the size of off_t.
[platform/upstream/flac.git] / src / test_libFLAC / encoders.c
index 053695f..d05c674 100644 (file)
@@ -1,5 +1,5 @@
 /* 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
@@ -11,9 +11,9 @@
  * 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.
  */
 
 #if HAVE_CONFIG_H
 #include "FLAC/assert.h"
 #include "FLAC/stream_encoder.h"
 #include "share/grabbag.h"
+#include "share/compat.h"
 #include "test_libs_common/file_utils_flac.h"
 #include "test_libs_common/metadata_utils.h"
 
 typedef enum {
-       LAYER_STREAM = 0, /* FLAC__stream_encoder_init_stream() without seeking */
-       LAYER_SEEKABLE_STREAM, /* FLAC__stream_encoder_init_stream() with seeking */
-       LAYER_FILE, /* FLAC__stream_encoder_init_FILE() */
-       LAYER_FILENAME /* FLAC__stream_encoder_init_file() */
+       LAYER_STREAM = 0, /* FLAC__stream_encoder_init_[ogg_]stream() without seeking */
+       LAYER_SEEKABLE_STREAM, /* FLAC__stream_encoder_init_[ogg_]stream() with seeking */
+       LAYER_FILE, /* FLAC__stream_encoder_init_[ogg_]FILE() */
+       LAYER_FILENAME /* FLAC__stream_encoder_init_[ogg_]file() */
 } Layer;
 
 static const char * const LayerString[] = {
@@ -45,10 +46,14 @@ static const char * const LayerString[] = {
        "Filename"
 };
 
-static FLAC__StreamMetadata streaminfo_, padding_, seektable_, application1_, application2_, vorbiscomment_, cuesheet_, unknown_;
-static FLAC__StreamMetadata *metadata_sequence_[] = { &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &unknown_ };
+static FLAC__StreamMetadata streaminfo_, padding_, seektable_, application1_, application2_, vorbiscomment_, cuesheet_, picture_, unknown_;
+static FLAC__StreamMetadata *metadata_sequence_[] = { &vorbiscomment_, &padding_, &seektable_, &application1_, &application2_, &cuesheet_, &picture_, &unknown_ };
 static const unsigned num_metadata_ = sizeof(metadata_sequence_) / sizeof(metadata_sequence_[0]);
-static const char *flacfilename_ = "metadata.flac";
+
+static const char *flacfilename(FLAC__bool is_ogg)
+{
+       return is_ogg? "metadata.oga" : "metadata.flac";
+}
 
 static FLAC__bool die_(const char *msg)
 {
@@ -74,33 +79,64 @@ static FLAC__bool die_s_(const char *msg, const FLAC__StreamEncoder *encoder)
        return false;
 }
 
-static void init_metadata_blocks_()
+static void init_metadata_blocks_(void)
 {
-       mutils__init_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &unknown_);
+       mutils__init_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
 }
 
-static void free_metadata_blocks_()
+static void free_metadata_blocks_(void)
 {
-       mutils__free_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &unknown_);
+       mutils__free_metadata_blocks(&streaminfo_, &padding_, &seektable_, &application1_, &application2_, &vorbiscomment_, &cuesheet_, &picture_, &unknown_);
+}
+
+static FLAC__StreamEncoderReadStatus stream_encoder_read_callback_(const FLAC__StreamEncoder *encoder, FLAC__byte buffer[], size_t *bytes, void *client_data)
+{
+       FILE *f = (FILE*)client_data;
+       (void)encoder;
+       if(*bytes > 0) {
+               *bytes = fread(buffer, sizeof(FLAC__byte), *bytes, f);
+               if(ferror(f))
+                       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;
 }
 
-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)
+static FLAC__StreamEncoderWriteStatus stream_encoder_write_callback_(const FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame, void *client_data)
 {
-       (void)encoder, (void)buffer, (void)bytes, (void)samples, (void)current_frame, (void)client_data;
-       return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
+       FILE *f = (FILE*)client_data;
+       (void)encoder, (void)samples, (void)current_frame;
+       if(fwrite(buffer, 1, bytes, f) != bytes)
+               return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR;
+       else
+               return FLAC__STREAM_ENCODER_WRITE_STATUS_OK;
 }
 
 static FLAC__StreamEncoderSeekStatus stream_encoder_seek_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data)
 {
-       (void)encoder, (void)absolute_byte_offset, (void)client_data;
-       return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
+       FILE *f = (FILE*)client_data;
+       (void)encoder;
+       if(fseek(f, (long)absolute_byte_offset, SEEK_SET) < 0)
+               return FLAC__STREAM_ENCODER_SEEK_STATUS_ERROR;
+       else
+               return FLAC__STREAM_ENCODER_SEEK_STATUS_OK;
 }
 
 static FLAC__StreamEncoderTellStatus stream_encoder_tell_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data)
 {
-       (void)encoder, (void)client_data;
-       *absolute_byte_offset = 0;
-       return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
+       FILE *f = (FILE*)client_data;
+       long pos;
+       (void)encoder;
+       if((pos = ftell(f)) < 0)
+               return FLAC__STREAM_ENCODER_TELL_STATUS_ERROR;
+       else {
+               *absolute_byte_offset = (FLAC__uint64)pos;
+               return FLAC__STREAM_ENCODER_TELL_STATUS_OK;
+       }
 }
 
 static void stream_encoder_metadata_callback_(const FLAC__StreamEncoder *encoder, const FLAC__StreamMetadata *metadata, void *client_data)
@@ -113,9 +149,10 @@ static void stream_encoder_progress_callback_(const FLAC__StreamEncoder *encoder
        (void)encoder, (void)bytes_written, (void)samples_written, (void)frames_written, (void)total_frames_estimate, (void)client_data;
 }
 
-static FLAC__bool test_stream_encoder(Layer layer)
+static FLAC__bool test_stream_encoder(Layer layer, FLAC__bool is_ogg)
 {
        FLAC__StreamEncoder *encoder;
+       FLAC__StreamEncoderInitStatus init_status;
        FLAC__StreamEncoderState state;
        FLAC__StreamDecoderState dstate;
        FILE *file = 0;
@@ -125,7 +162,7 @@ static FLAC__bool test_stream_encoder(Layer layer)
 
        samples_array[0] = samples;
 
-       printf("\n+++ libFLAC unit test: FLAC__StreamEncoder (layer: %s)\n\n", LayerString[layer]);
+       printf("\n+++ libFLAC unit test: FLAC__StreamEncoder (layer: %s, format: %s)\n\n", LayerString[layer], is_ogg? "Ogg FLAC":"FLAC");
 
        printf("testing FLAC__stream_encoder_new()... ");
        encoder = FLAC__stream_encoder_new();
@@ -135,6 +172,13 @@ static FLAC__bool test_stream_encoder(Layer layer)
        }
        printf("OK\n");
 
+       if(is_ogg) {
+               printf("testing FLAC__stream_encoder_set_ogg_serial_number()... ");
+               if(!FLAC__stream_encoder_set_ogg_serial_number(encoder, file_utils__ogg_serial_number))
+                       return die_s_("returned false", encoder);
+               printf("OK\n");
+       }
+
        printf("testing FLAC__stream_encoder_set_verify()... ");
        if(!FLAC__stream_encoder_set_verify(encoder, true))
                return die_s_("returned false", encoder);
@@ -145,16 +189,6 @@ static FLAC__bool test_stream_encoder(Layer layer)
                return die_s_("returned false", encoder);
        printf("OK\n");
 
-       printf("testing FLAC__stream_encoder_set_do_mid_side_stereo()... ");
-       if(!FLAC__stream_encoder_set_do_mid_side_stereo(encoder, false))
-               return die_s_("returned false", encoder);
-       printf("OK\n");
-
-       printf("testing FLAC__stream_encoder_set_loose_mid_side_stereo()... ");
-       if(!FLAC__stream_encoder_set_loose_mid_side_stereo(encoder, false))
-               return die_s_("returned false", encoder);
-       printf("OK\n");
-
        printf("testing FLAC__stream_encoder_set_channels()... ");
        if(!FLAC__stream_encoder_set_channels(encoder, streaminfo_.data.stream_info.channels))
                return die_s_("returned false", encoder);
@@ -170,11 +204,26 @@ static FLAC__bool test_stream_encoder(Layer layer)
                return die_s_("returned false", encoder);
        printf("OK\n");
 
+       printf("testing FLAC__stream_encoder_set_compression_level()... ");
+       if(!FLAC__stream_encoder_set_compression_level(encoder, (unsigned)(-1)))
+               return die_s_("returned false", encoder);
+       printf("OK\n");
+
        printf("testing FLAC__stream_encoder_set_blocksize()... ");
        if(!FLAC__stream_encoder_set_blocksize(encoder, streaminfo_.data.stream_info.min_blocksize))
                return die_s_("returned false", encoder);
        printf("OK\n");
 
+       printf("testing FLAC__stream_encoder_set_do_mid_side_stereo()... ");
+       if(!FLAC__stream_encoder_set_do_mid_side_stereo(encoder, false))
+               return die_s_("returned false", encoder);
+       printf("OK\n");
+
+       printf("testing FLAC__stream_encoder_set_loose_mid_side_stereo()... ");
+       if(!FLAC__stream_encoder_set_loose_mid_side_stereo(encoder, false))
+               return die_s_("returned false", encoder);
+       printf("OK\n");
+
        printf("testing FLAC__stream_encoder_set_max_lpc_order()... ");
        if(!FLAC__stream_encoder_set_max_lpc_order(encoder, 0))
                return die_s_("returned false", encoder);
@@ -225,39 +274,47 @@ static FLAC__bool test_stream_encoder(Layer layer)
                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");
+       }
+
        switch(layer) {
                case LAYER_STREAM:
-                       printf("testing FLAC__stream_encoder_init_stream()... ");
-                       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)
-                               return die_s_(0, encoder);
+                       printf("testing FLAC__stream_encoder_init_%sstream()... ", is_ogg? "ogg_":"");
+                       init_status = is_ogg?
+                               FLAC__stream_encoder_init_ogg_stream(encoder, /*read_callback=*/0, stream_encoder_write_callback_, /*seek_callback=*/0, /*tell_callback=*/0, stream_encoder_metadata_callback_, /*client_data=*/file) :
+                               FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, /*seek_callback=*/0, /*tell_callback=*/0, stream_encoder_metadata_callback_, /*client_data=*/file);
                        break;
                case LAYER_SEEKABLE_STREAM:
-                       printf("testing FLAC__stream_encoder_init_stream()... ");
-                       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)
-                               return die_s_(0, encoder);
+                       printf("testing FLAC__stream_encoder_init_%sstream()... ", is_ogg? "ogg_":"");
+                       init_status = is_ogg?
+                               FLAC__stream_encoder_init_ogg_stream(encoder, stream_encoder_read_callback_, stream_encoder_write_callback_, stream_encoder_seek_callback_, stream_encoder_tell_callback_, /*metadata_callback=*/0, /*client_data=*/file) :
+                               FLAC__stream_encoder_init_stream(encoder, stream_encoder_write_callback_, stream_encoder_seek_callback_, stream_encoder_tell_callback_, /*metadata_callback=*/0, /*client_data=*/file);
                        break;
                case LAYER_FILE:
-                       printf("opening file for FLAC output... ");
-                       file = fopen(flacfilename_, "w+b");
-                       if(0 == file) {
-                               printf("ERROR (%s)\n", strerror(errno));
-                               return false;
-                       }
-                       printf("OK\n");
-
-                       printf("testing FLAC__stream_encoder_init_FILE()... ");
-                       if(FLAC__stream_encoder_init_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
-                               return die_s_(0, encoder);
+                       printf("testing FLAC__stream_encoder_init_%sFILE()... ", is_ogg? "ogg_":"");
+                       init_status = is_ogg?
+                               FLAC__stream_encoder_init_ogg_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0) :
+                               FLAC__stream_encoder_init_FILE(encoder, file, stream_encoder_progress_callback_, /*client_data=*/0);
                        break;
                case LAYER_FILENAME:
-                       printf("testing FLAC__stream_encoder_init_file()... ");
-                       if(FLAC__stream_encoder_init_file(encoder, flacfilename_, stream_encoder_progress_callback_, /*client_data=*/0) != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
-                               return die_s_(0, encoder);
+                       printf("testing FLAC__stream_encoder_init_%sfile()... ", is_ogg? "ogg_":"");
+                       init_status = is_ogg?
+                               FLAC__stream_encoder_init_ogg_file(encoder, flacfilename(is_ogg), stream_encoder_progress_callback_, /*client_data=*/0) :
+                               FLAC__stream_encoder_init_file(encoder, flacfilename(is_ogg), stream_encoder_progress_callback_, /*client_data=*/0);
                        break;
                default:
                        die_("internal error 001");
                        return false;
        }
+       if(init_status != FLAC__STREAM_ENCODER_INIT_STATUS_OK)
+               return die_s_(0, encoder);
        printf("OK\n");
 
        printf("testing FLAC__stream_encoder_get_state()... ");
@@ -393,7 +450,7 @@ static FLAC__bool test_stream_encoder(Layer layer)
 
        printf("testing FLAC__stream_encoder_get_total_samples_estimate()... ");
        if(FLAC__stream_encoder_get_total_samples_estimate(encoder) != streaminfo_.data.stream_info.total_samples) {
-               printf("FAILED, expected %llu, got %llu\n", streaminfo_.data.stream_info.total_samples, FLAC__stream_encoder_get_total_samples_estimate(encoder));
+               printf("FAILED, expected %" PRIu64 ", got %" PRIu64 "\n", streaminfo_.data.stream_info.total_samples, FLAC__stream_encoder_get_total_samples_estimate(encoder));
                return false;
        }
        printf("OK\n");
@@ -413,9 +470,13 @@ static FLAC__bool test_stream_encoder(Layer layer)
        printf("OK\n");
 
        printf("testing FLAC__stream_encoder_finish()... ");
-       FLAC__stream_encoder_finish(encoder);
+       if(!FLAC__stream_encoder_finish(encoder))
+               return die_s_("returned false", encoder);
        printf("OK\n");
 
+       if(layer < LAYER_FILE)
+               fclose(file);
+
        printf("testing FLAC__stream_encoder_delete()... ");
        FLAC__stream_encoder_delete(encoder);
        printf("OK\n");
@@ -425,24 +486,33 @@ static FLAC__bool test_stream_encoder(Layer layer)
        return true;
 }
 
-FLAC__bool test_encoders()
+FLAC__bool test_encoders(void)
 {
-       init_metadata_blocks_();
+       FLAC__bool is_ogg = false;
 
-       if(!test_stream_encoder(LAYER_STREAM))
-               return false;
+       while(1) {
+               init_metadata_blocks_();
 
-       if(!test_stream_encoder(LAYER_SEEKABLE_STREAM))
-               return false;
+               if(!test_stream_encoder(LAYER_STREAM, is_ogg))
+                       return false;
 
-       if(!test_stream_encoder(LAYER_FILE))
-               return false;
+               if(!test_stream_encoder(LAYER_SEEKABLE_STREAM, is_ogg))
+                       return false;
 
-       if(!test_stream_encoder(LAYER_FILENAME))
-               return false;
+               if(!test_stream_encoder(LAYER_FILE, is_ogg))
+                       return false;
+
+               if(!test_stream_encoder(LAYER_FILENAME, is_ogg))
+                       return false;
+
+               (void) grabbag__file_remove_file(flacfilename(is_ogg));
 
-       (void) grabbag__file_remove_file(flacfilename_);
-       free_metadata_blocks_();
+               free_metadata_blocks_();
+
+               if(!FLAC_API_SUPPORTS_OGG_FLAC || is_ogg)
+                       break;
+               is_ogg = true;
+       }
 
        return true;
 }