/* ** Copyright (C) 2004-2011 Erik de Castro Lopo ** Copyright (C) 2004 Tobias Gehrig ** ** This program is free software ; you can redistribute it and/or modify ** it under the terms of the GNU Lesser General Public License as published by ** the Free Software Foundation ; either version 2.1 of the License, or ** (at your option) any later version. ** ** This program is distributed in the hope that it will be useful, ** but WITHOUT ANY WARRANTY ; without even the implied warranty of ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ** GNU Lesser General Public License for more details. ** ** You should have received a copy of the GNU Lesser 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. */ #include "sfconfig.h" #include #include #include #include #include #include #include "sndfile.h" #include "common.h" #if HAVE_EXTERNAL_LIBS #include #include #include /*------------------------------------------------------------------------------ ** Private static functions. */ #define ENC_BUFFER_SIZE 8192 typedef enum { PFLAC_PCM_SHORT = 50, PFLAC_PCM_INT = 51, PFLAC_PCM_FLOAT = 52, PFLAC_PCM_DOUBLE = 53 } PFLAC_PCM ; typedef struct { FLAC__StreamDecoder *fsd ; FLAC__StreamEncoder *fse ; PFLAC_PCM pcmtype ; void* ptr ; unsigned pos, len, remain ; FLAC__StreamMetadata *metadata ; const FLAC__int32 * const * wbuffer ; FLAC__int32 * rbuffer [FLAC__MAX_CHANNELS] ; FLAC__int32* encbuffer ; unsigned bufferpos ; const FLAC__Frame *frame ; FLAC__bool bufferbackup ; } FLAC_PRIVATE ; typedef struct { const char *tag ; int type ; } FLAC_TAG ; static sf_count_t flac_seek (SF_PRIVATE *psf, int mode, sf_count_t offset) ; static int flac_close (SF_PRIVATE *psf) ; static int flac_enc_init (SF_PRIVATE *psf) ; static int flac_read_header (SF_PRIVATE *psf) ; static sf_count_t flac_read_flac2s (SF_PRIVATE *psf, short *ptr, sf_count_t len) ; static sf_count_t flac_read_flac2i (SF_PRIVATE *psf, int *ptr, sf_count_t len) ; static sf_count_t flac_read_flac2f (SF_PRIVATE *psf, float *ptr, sf_count_t len) ; static sf_count_t flac_read_flac2d (SF_PRIVATE *psf, double *ptr, sf_count_t len) ; static sf_count_t flac_write_s2flac (SF_PRIVATE *psf, const short *ptr, sf_count_t len) ; static sf_count_t flac_write_i2flac (SF_PRIVATE *psf, const int *ptr, sf_count_t len) ; static sf_count_t flac_write_f2flac (SF_PRIVATE *psf, const float *ptr, sf_count_t len) ; static sf_count_t flac_write_d2flac (SF_PRIVATE *psf, const double *ptr, sf_count_t len) ; static void f2flac8_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void f2flac16_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void f2flac24_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void f2flac8_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void f2flac16_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void f2flac24_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac8_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac16_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac24_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac8_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac16_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static void d2flac24_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) ; static int flac_command (SF_PRIVATE *psf, int command, void *data, int datasize) ; /* Decoder Callbacks */ static FLAC__StreamDecoderReadStatus sf_flac_read_callback (const FLAC__StreamDecoder *decoder, FLAC__byte buffer [], size_t *bytes, void *client_data) ; static FLAC__StreamDecoderSeekStatus sf_flac_seek_callback (const FLAC__StreamDecoder *decoder, FLAC__uint64 absolute_byte_offset, void *client_data) ; static FLAC__StreamDecoderTellStatus sf_flac_tell_callback (const FLAC__StreamDecoder *decoder, FLAC__uint64 *absolute_byte_offset, void *client_data) ; static FLAC__StreamDecoderLengthStatus sf_flac_length_callback (const FLAC__StreamDecoder *decoder, FLAC__uint64 *stream_length, void *client_data) ; static FLAC__bool sf_flac_eof_callback (const FLAC__StreamDecoder *decoder, void *client_data) ; static FLAC__StreamDecoderWriteStatus sf_flac_write_callback (const FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer [], void *client_data) ; static void sf_flac_meta_callback (const FLAC__StreamDecoder *decoder, const FLAC__StreamMetadata *metadata, void *client_data) ; static void sf_flac_error_callback (const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status, void *client_data) ; /* Encoder Callbacks */ static FLAC__StreamEncoderSeekStatus sf_flac_enc_seek_callback (const FLAC__StreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data) ; static FLAC__StreamEncoderTellStatus sf_flac_enc_tell_callback (const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data) ; static FLAC__StreamEncoderWriteStatus sf_flac_enc_write_callback (const FLAC__StreamEncoder *encoder, const FLAC__byte buffer [], size_t bytes, unsigned samples, unsigned current_frame, void *client_data) ; static void s2flac8_array (const short *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] >> 8 ; } /* s2flac8_array */ static void s2flac16_array (const short *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] ; } /* s2flac16_array */ static void s2flac24_array (const short *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] << 8 ; } /* s2flac24_array */ static void i2flac8_array (const int *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] >> 24 ; } /* i2flac8_array */ static void i2flac16_array (const int *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] >> 16 ; } /* i2flac16_array */ static void i2flac24_array (const int *src, FLAC__int32 *dest, int count) { while (--count >= 0) dest [count] = src [count] >> 8 ; } /* i2flac24_array */ static sf_count_t flac_buffer_copy (SF_PRIVATE *psf) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; const FLAC__Frame *frame = pflac->frame ; const FLAC__int32* const *buffer = pflac->wbuffer ; unsigned i = 0, j, offset ; /* ** frame->header.blocksize is variable and we're using a constant blocksize ** of FLAC__MAX_BLOCK_SIZE. ** Check our assumptions here. */ if (frame->header.blocksize > FLAC__MAX_BLOCK_SIZE) { psf_log_printf (psf, "Ooops : frame->header.blocksize (%d) > FLAC__MAX_BLOCK_SIZE (%d)\n", __func__, __LINE__, frame->header.blocksize, FLAC__MAX_BLOCK_SIZE) ; psf->error = SFE_INTERNAL ; return 0 ; } ; if (pflac->ptr == NULL) { /* ** Not sure why this code is here and not elsewhere. ** Removing it causes valgrind errors. */ pflac->bufferbackup = SF_TRUE ; for (i = 0 ; i < frame->header.channels ; i++) { if (pflac->rbuffer [i] == NULL) pflac->rbuffer [i] = calloc (FLAC__MAX_BLOCK_SIZE, sizeof (FLAC__int32)) ; memcpy (pflac->rbuffer [i], buffer [i], frame->header.blocksize * sizeof (FLAC__int32)) ; } ; pflac->wbuffer = (const FLAC__int32* const*) pflac->rbuffer ; return 0 ; } ; switch (pflac->pcmtype) { case PFLAC_PCM_SHORT : { short *retpcm = (short*) pflac->ptr ; int shift = 16 - frame->header.bits_per_sample ; if (shift < 0) { shift = abs (shift) ; for (i = 0 ; i < frame->header.blocksize && pflac->remain > 0 ; i++) { offset = pflac->pos + i * frame->header.channels ; if (pflac->bufferpos >= frame->header.blocksize) break ; for (j = 0 ; j < frame->header.channels ; j++) retpcm [offset + j] = buffer [j][pflac->bufferpos] >> shift ; pflac->remain -= frame->header.channels ; pflac->bufferpos++ ; } } else { for (i = 0 ; i < frame->header.blocksize && pflac->remain > 0 ; i++) { offset = pflac->pos + i * frame->header.channels ; if (pflac->bufferpos >= frame->header.blocksize) break ; for (j = 0 ; j < frame->header.channels ; j++) retpcm [offset + j] = (buffer [j][pflac->bufferpos]) << shift ; pflac->remain -= frame->header.channels ; pflac->bufferpos++ ; } ; } ; } ; break ; case PFLAC_PCM_INT : { int *retpcm = (int*) pflac->ptr ; int shift = 32 - frame->header.bits_per_sample ; for (i = 0 ; i < frame->header.blocksize && pflac->remain > 0 ; i++) { offset = pflac->pos + i * frame->header.channels ; if (pflac->bufferpos >= frame->header.blocksize) break ; for (j = 0 ; j < frame->header.channels ; j++) retpcm [offset + j] = buffer [j][pflac->bufferpos] << shift ; pflac->remain -= frame->header.channels ; pflac->bufferpos++ ; } ; } ; break ; case PFLAC_PCM_FLOAT : { float *retpcm = (float*) pflac->ptr ; float norm = (psf->norm_float == SF_TRUE) ? 1.0 / (1 << (frame->header.bits_per_sample - 1)) : 1.0 ; for (i = 0 ; i < frame->header.blocksize && pflac->remain > 0 ; i++) { offset = pflac->pos + i * frame->header.channels ; if (pflac->bufferpos >= frame->header.blocksize) break ; for (j = 0 ; j < frame->header.channels ; j++) retpcm [offset + j] = buffer [j][pflac->bufferpos] * norm ; pflac->remain -= frame->header.channels ; pflac->bufferpos++ ; } ; } ; break ; case PFLAC_PCM_DOUBLE : { double *retpcm = (double*) pflac->ptr ; double norm = (psf->norm_double == SF_TRUE) ? 1.0 / (1 << (frame->header.bits_per_sample - 1)) : 1.0 ; for (i = 0 ; i < frame->header.blocksize && pflac->remain > 0 ; i++) { offset = pflac->pos + i * frame->header.channels ; if (pflac->bufferpos >= frame->header.blocksize) break ; for (j = 0 ; j < frame->header.channels ; j++) retpcm [offset + j] = buffer [j][pflac->bufferpos] * norm ; pflac->remain -= frame->header.channels ; pflac->bufferpos++ ; } ; } ; break ; default : return 0 ; } ; offset = i * frame->header.channels ; pflac->pos += i * frame->header.channels ; return offset ; } /* flac_buffer_copy */ static FLAC__StreamDecoderReadStatus sf_flac_read_callback (const FLAC__StreamDecoder * UNUSED (decoder), FLAC__byte buffer [], size_t *bytes, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; *bytes = psf_fread (buffer, 1, *bytes, psf) ; if (*bytes > 0 && psf->error == 0) return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE ; return FLAC__STREAM_DECODER_READ_STATUS_ABORT ; } /* sf_flac_read_callback */ static FLAC__StreamDecoderSeekStatus sf_flac_seek_callback (const FLAC__StreamDecoder * UNUSED (decoder), FLAC__uint64 absolute_byte_offset, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; psf_fseek (psf, absolute_byte_offset, SEEK_SET) ; if (psf->error) return FLAC__STREAM_DECODER_SEEK_STATUS_ERROR ; return FLAC__STREAM_DECODER_SEEK_STATUS_OK ; } /* sf_flac_seek_callback */ static FLAC__StreamDecoderTellStatus sf_flac_tell_callback (const FLAC__StreamDecoder * UNUSED (decoder), FLAC__uint64 *absolute_byte_offset, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; *absolute_byte_offset = psf_ftell (psf) ; if (psf->error) return FLAC__STREAM_DECODER_TELL_STATUS_ERROR ; return FLAC__STREAM_DECODER_TELL_STATUS_OK ; } /* sf_flac_tell_callback */ static FLAC__StreamDecoderLengthStatus sf_flac_length_callback (const FLAC__StreamDecoder * UNUSED (decoder), FLAC__uint64 *stream_length, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; if ((*stream_length = psf->filelength) == 0) return FLAC__STREAM_DECODER_LENGTH_STATUS_ERROR ; return FLAC__STREAM_DECODER_LENGTH_STATUS_OK ; } /* sf_flac_length_callback */ static FLAC__bool sf_flac_eof_callback (const FLAC__StreamDecoder *UNUSED (decoder), void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; if (psf_ftell (psf) == psf->filelength) return SF_TRUE ; return SF_FALSE ; } /* sf_flac_eof_callback */ static FLAC__StreamDecoderWriteStatus sf_flac_write_callback (const FLAC__StreamDecoder * UNUSED (decoder), const FLAC__Frame *frame, const FLAC__int32 * const buffer [], void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; pflac->frame = frame ; pflac->bufferpos = 0 ; pflac->bufferbackup = SF_FALSE ; pflac->wbuffer = buffer ; flac_buffer_copy (psf) ; return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE ; } /* sf_flac_write_callback */ static void sf_flac_meta_get_vorbiscomments (SF_PRIVATE *psf, const FLAC__StreamMetadata *metadata) { FLAC_TAG tags [] = { { "title", SF_STR_TITLE }, { "copyright", SF_STR_COPYRIGHT }, { "software", SF_STR_SOFTWARE }, { "artist", SF_STR_ARTIST }, { "comment", SF_STR_COMMENT }, { "date", SF_STR_DATE }, { "album", SF_STR_ALBUM }, { "license", SF_STR_LICENSE }, { "tracknumber", SF_STR_TRACKNUMBER }, { "genre", SF_STR_GENRE } } ; const char *value, *cptr ; int k, tag_num ; for (k = 0 ; k < ARRAY_LEN (tags) ; k++) { tag_num = FLAC__metadata_object_vorbiscomment_find_entry_from (metadata, 0, tags [k].tag) ; if (tag_num < 0) continue ; value = (const char*) metadata->data.vorbis_comment.comments [tag_num].entry ; if ((cptr = strchr (value, '=')) != NULL) value = cptr + 1 ; psf_log_printf (psf, " %-10s : %s\n", tags [k].tag, value) ; psf_store_string (psf, tags [k].type, value) ; } ; return ; } /* sf_flac_meta_get_vorbiscomments */ static void sf_flac_meta_callback (const FLAC__StreamDecoder * UNUSED (decoder), const FLAC__StreamMetadata *metadata, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; int bitwidth = 0 ; switch (metadata->type) { case FLAC__METADATA_TYPE_STREAMINFO : psf->sf.channels = metadata->data.stream_info.channels ; psf->sf.samplerate = metadata->data.stream_info.sample_rate ; psf->sf.frames = metadata->data.stream_info.total_samples ; psf_log_printf (psf, "FLAC Stream Metadata\n Channels : %d\n Sample rate : %d\n", psf->sf.channels, psf->sf.samplerate) ; if (psf->sf.frames == 0) { psf_log_printf (psf, " Frames : 0 (bumping to SF_COUNT_MAX)\n") ; psf->sf.frames = SF_COUNT_MAX ; } else psf_log_printf (psf, " Frames : %D\n", psf->sf.frames) ; switch (metadata->data.stream_info.bits_per_sample) { case 8 : psf->sf.format |= SF_FORMAT_PCM_S8 ; bitwidth = 8 ; break ; case 16 : psf->sf.format |= SF_FORMAT_PCM_16 ; bitwidth = 16 ; break ; case 24 : psf->sf.format |= SF_FORMAT_PCM_24 ; bitwidth = 24 ; break ; default : psf_log_printf (psf, "sf_flac_meta_callback : bits_per_sample %d not yet implemented.\n", metadata->data.stream_info.bits_per_sample) ; break ; } ; if (bitwidth > 0) psf_log_printf (psf, " Bit width : %d\n", bitwidth) ; break ; case FLAC__METADATA_TYPE_VORBIS_COMMENT : psf_log_printf (psf, "Vorbis Comment Metadata\n") ; sf_flac_meta_get_vorbiscomments (psf, metadata) ; break ; case FLAC__METADATA_TYPE_PADDING : psf_log_printf (psf, "Padding Metadata\n") ; break ; case FLAC__METADATA_TYPE_APPLICATION : psf_log_printf (psf, "Application Metadata\n") ; break ; case FLAC__METADATA_TYPE_SEEKTABLE : psf_log_printf (psf, "Seektable Metadata\n") ; break ; case FLAC__METADATA_TYPE_CUESHEET : psf_log_printf (psf, "Cuesheet Metadata\n") ; break ; case FLAC__METADATA_TYPE_PICTURE : psf_log_printf (psf, "Picture Metadata\n") ; break ; case FLAC__METADATA_TYPE_UNDEFINED : psf_log_printf (psf, "Undefined Metadata\n") ; break ; default : psf_log_printf (psf, "sf_flac_meta_callback : metadata-type %d not yet implemented.\n", metadata->type) ; break ; } ; return ; } /* sf_flac_meta_callback */ static void sf_flac_error_callback (const FLAC__StreamDecoder * UNUSED (decoder), FLAC__StreamDecoderErrorStatus status, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; psf_log_printf (psf, "ERROR : %s\n", FLAC__StreamDecoderErrorStatusString [status]) ; switch (status) { case FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC : psf->error = SFE_FLAC_LOST_SYNC ; break ; case FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER : psf->error = SFE_FLAC_BAD_HEADER ; break ; default : psf->error = SFE_FLAC_UNKOWN_ERROR ; break ; } ; return ; } /* sf_flac_error_callback */ static FLAC__StreamEncoderSeekStatus sf_flac_enc_seek_callback (const FLAC__StreamEncoder * UNUSED (encoder), FLAC__uint64 absolute_byte_offset, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; psf_fseek (psf, absolute_byte_offset, SEEK_SET) ; if (psf->error) return FLAC__STREAM_ENCODER_SEEK_STATUS_ERROR ; return FLAC__STREAM_ENCODER_SEEK_STATUS_OK ; } /* sf_flac_enc_seek_callback */ static FLAC__StreamEncoderTellStatus sf_flac_enc_tell_callback (const FLAC__StreamEncoder *UNUSED (encoder), FLAC__uint64 *absolute_byte_offset, void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; *absolute_byte_offset = psf_ftell (psf) ; if (psf->error) return FLAC__STREAM_ENCODER_TELL_STATUS_ERROR ; return FLAC__STREAM_ENCODER_TELL_STATUS_OK ; } /* sf_flac_enc_tell_callback */ static FLAC__StreamEncoderWriteStatus sf_flac_enc_write_callback (const FLAC__StreamEncoder * UNUSED (encoder), const FLAC__byte buffer [], size_t bytes, unsigned UNUSED (samples), unsigned UNUSED (current_frame), void *client_data) { SF_PRIVATE *psf = (SF_PRIVATE*) client_data ; if (psf_fwrite (buffer, 1, bytes, psf) == (sf_count_t) bytes && psf->error == 0) return FLAC__STREAM_ENCODER_WRITE_STATUS_OK ; return FLAC__STREAM_ENCODER_WRITE_STATUS_FATAL_ERROR ; } /* sf_flac_enc_write_callback */ static void flac_write_strings (SF_PRIVATE *psf, FLAC_PRIVATE* pflac) { FLAC__StreamMetadata_VorbisComment_Entry entry ; int k, string_count = 0 ; for (k = 0 ; k < SF_MAX_STRINGS ; k++) { if (psf->strings [k].type != 0) string_count ++ ; } ; if (string_count == 0) return ; if (pflac->metadata == NULL && (pflac->metadata = FLAC__metadata_object_new (FLAC__METADATA_TYPE_VORBIS_COMMENT)) == NULL) { psf_log_printf (psf, "FLAC__metadata_object_new returned NULL\n") ; return ; } ; for (k = 0 ; k < SF_MAX_STRINGS && psf->strings [k].type != 0 ; k++) { const char * key, * value ; switch (psf->strings [k].type) { case SF_STR_SOFTWARE : key = "software" ; break ; case SF_STR_TITLE : key = "title" ; break ; case SF_STR_COPYRIGHT : key = "copyright" ; break ; case SF_STR_ARTIST : key = "artist" ; break ; case SF_STR_COMMENT : key = "comment" ; break ; case SF_STR_DATE : key = "date" ; break ; case SF_STR_ALBUM : key = "album" ; break ; case SF_STR_LICENSE : key = "license" ; break ; case SF_STR_TRACKNUMBER : key = "tracknumber" ; break ; case SF_STR_GENRE : key = "genre" ; break ; default : continue ; } ; value = psf->strings [k].str ; FLAC__metadata_object_vorbiscomment_entry_from_name_value_pair (&entry, key, value) ; FLAC__metadata_object_vorbiscomment_append_comment (pflac->metadata, entry, /* copy */ SF_FALSE) ; } ; if (! FLAC__stream_encoder_set_metadata (pflac->fse, &pflac->metadata, 1)) { printf ("%s %d : fail\n", __func__, __LINE__) ; return ; } ; return ; } /* flac_write_strings */ static int flac_write_header (SF_PRIVATE *psf, int UNUSED (calc_length)) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; int err ; flac_write_strings (psf, pflac) ; if ((err = FLAC__stream_encoder_init_stream (pflac->fse, sf_flac_enc_write_callback, sf_flac_enc_seek_callback, sf_flac_enc_tell_callback, NULL, psf)) != FLAC__STREAM_DECODER_INIT_STATUS_OK) { psf_log_printf (psf, "Error : FLAC encoder init returned error : %s\n", FLAC__StreamEncoderInitStatusString [err]) ; return SFE_FLAC_INIT_DECODER ; } ; if (psf->error == 0) psf->dataoffset = psf_ftell (psf) ; pflac->encbuffer = calloc (ENC_BUFFER_SIZE, sizeof (FLAC__int32)) ; return psf->error ; } /* flac_write_header */ /*------------------------------------------------------------------------------ ** Public function. */ int flac_open (SF_PRIVATE *psf) { int subformat ; int error = 0 ; FLAC_PRIVATE* pflac = calloc (1, sizeof (FLAC_PRIVATE)) ; psf->codec_data = pflac ; if (psf->file.mode == SFM_RDWR) return SFE_BAD_MODE_RW ; if (psf->file.mode == SFM_READ) { if ((error = flac_read_header (psf))) return error ; } ; subformat = SF_CODEC (psf->sf.format) ; if (psf->file.mode == SFM_WRITE) { if ((SF_CONTAINER (psf->sf.format)) != SF_FORMAT_FLAC) return SFE_BAD_OPEN_FORMAT ; psf->endian = SF_ENDIAN_BIG ; psf->sf.seekable = 0 ; psf->str_flags = SF_STR_ALLOW_START ; if ((error = flac_enc_init (psf))) return error ; psf->write_header = flac_write_header ; } ; psf->datalength = psf->filelength ; psf->dataoffset = 0 ; psf->blockwidth = 0 ; psf->bytewidth = 1 ; psf->container_close = flac_close ; psf->seek = flac_seek ; psf->command = flac_command ; psf->blockwidth = psf->bytewidth * psf->sf.channels ; switch (subformat) { case SF_FORMAT_PCM_S8 : /* 8-bit FLAC. */ case SF_FORMAT_PCM_16 : /* 16-bit FLAC. */ case SF_FORMAT_PCM_24 : /* 24-bit FLAC. */ error = flac_init (psf) ; break ; default : return SFE_UNIMPLEMENTED ; } ; return error ; } /* flac_open */ /*------------------------------------------------------------------------------ */ static int flac_close (SF_PRIVATE *psf) { FLAC_PRIVATE* pflac ; int k ; if ((pflac = (FLAC_PRIVATE*) psf->codec_data) == NULL) return 0 ; if (pflac->metadata != NULL) FLAC__metadata_object_delete (pflac->metadata) ; if (psf->file.mode == SFM_WRITE) { FLAC__stream_encoder_finish (pflac->fse) ; FLAC__stream_encoder_delete (pflac->fse) ; if (pflac->encbuffer) free (pflac->encbuffer) ; } ; if (psf->file.mode == SFM_READ) { FLAC__stream_decoder_finish (pflac->fsd) ; FLAC__stream_decoder_delete (pflac->fsd) ; } ; for (k = 0 ; k < ARRAY_LEN (pflac->rbuffer) ; k++) free (pflac->rbuffer [k]) ; free (pflac) ; psf->codec_data = NULL ; return 0 ; } /* flac_close */ static int flac_enc_init (SF_PRIVATE *psf) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; unsigned bps ; /* To cite the flac FAQ at ** http://flac.sourceforge.net/faq.html#general__samples ** "FLAC supports linear sample rates from 1Hz - 655350Hz in 1Hz ** increments." */ if ( psf->sf.samplerate < 1 || psf->sf.samplerate > 655350 ) { psf_log_printf (psf, "flac sample rate out of range.\n", psf->sf.samplerate) ; return SFE_FLAC_BAD_SAMPLE_RATE ; } ; psf_fseek (psf, 0, SEEK_SET) ; switch (SF_CODEC (psf->sf.format)) { case SF_FORMAT_PCM_S8 : bps = 8 ; break ; case SF_FORMAT_PCM_16 : bps = 16 ; break ; case SF_FORMAT_PCM_24 : bps = 24 ; break ; default : bps = 0 ; break ; } ; if ((pflac->fse = FLAC__stream_encoder_new ()) == NULL) return SFE_FLAC_NEW_DECODER ; if (! FLAC__stream_encoder_set_channels (pflac->fse, psf->sf.channels)) { psf_log_printf (psf, "FLAC__stream_encoder_set_channels (%d) return false.\n", psf->sf.channels) ; return SFE_FLAC_INIT_DECODER ; } ; if (! FLAC__stream_encoder_set_sample_rate (pflac->fse, psf->sf.samplerate)) { psf_log_printf (psf, "FLAC__stream_encoder_set_sample_rate (%d) returned false.\n", psf->sf.samplerate) ; return SFE_FLAC_BAD_SAMPLE_RATE ; } ; if (! FLAC__stream_encoder_set_bits_per_sample (pflac->fse, bps)) { psf_log_printf (psf, "FLAC__stream_encoder_set_bits_per_sample (%d) return false.\n", bps) ; return SFE_FLAC_INIT_DECODER ; } ; return 0 ; } /* flac_enc_init */ static int flac_read_header (SF_PRIVATE *psf) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; psf_fseek (psf, 0, SEEK_SET) ; if ((pflac->fsd = FLAC__stream_decoder_new ()) == NULL) return SFE_FLAC_NEW_DECODER ; FLAC__stream_decoder_set_metadata_respond_all (pflac->fsd) ; if (FLAC__stream_decoder_init_stream (pflac->fsd, sf_flac_read_callback, sf_flac_seek_callback, sf_flac_tell_callback, sf_flac_length_callback, sf_flac_eof_callback, sf_flac_write_callback, sf_flac_meta_callback, sf_flac_error_callback, psf) != FLAC__STREAM_DECODER_INIT_STATUS_OK) return SFE_FLAC_INIT_DECODER ; FLAC__stream_decoder_process_until_end_of_metadata (pflac->fsd) ; psf_log_printf (psf, "End\n") ; if (psf->error == 0) { FLAC__uint64 position ; FLAC__stream_decoder_get_decode_position (pflac->fsd, &position) ; psf->dataoffset = position ; } ; return psf->error ; } /* flac_read_header */ static int flac_command (SF_PRIVATE * UNUSED (psf), int UNUSED (command), void * UNUSED (data), int UNUSED (datasize)) { return 0 ; } /* flac_command */ int flac_init (SF_PRIVATE *psf) { if (psf->file.mode == SFM_RDWR) return SFE_BAD_MODE_RW ; if (psf->file.mode == SFM_READ) { psf->read_short = flac_read_flac2s ; psf->read_int = flac_read_flac2i ; psf->read_float = flac_read_flac2f ; psf->read_double = flac_read_flac2d ; } ; if (psf->file.mode == SFM_WRITE) { psf->write_short = flac_write_s2flac ; psf->write_int = flac_write_i2flac ; psf->write_float = flac_write_f2flac ; psf->write_double = flac_write_d2flac ; } ; psf->bytewidth = 1 ; psf->blockwidth = psf->sf.channels ; if (psf->filelength > psf->dataoffset) psf->datalength = (psf->dataend) ? psf->dataend - psf->dataoffset : psf->filelength - psf->dataoffset ; else psf->datalength = 0 ; return 0 ; } /* flac_init */ static unsigned flac_read_loop (SF_PRIVATE *psf, unsigned len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; pflac->pos = 0 ; pflac->len = len ; pflac->remain = len ; if (pflac->frame != NULL && pflac->bufferpos < pflac->frame->header.blocksize) flac_buffer_copy (psf) ; while (pflac->pos < pflac->len) { if (FLAC__stream_decoder_process_single (pflac->fsd) == 0) break ; if (FLAC__stream_decoder_get_state (pflac->fsd) >= FLAC__STREAM_DECODER_END_OF_STREAM) break ; } ; pflac->ptr = NULL ; return pflac->pos ; } /* flac_read_loop */ static sf_count_t flac_read_flac2s (SF_PRIVATE *psf, short *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; sf_count_t total = 0, current ; unsigned readlen ; pflac->pcmtype = PFLAC_PCM_SHORT ; while (total < len) { pflac->ptr = ptr + total ; readlen = (len - total > 0x1000000) ? 0x1000000 : (unsigned) (len - total) ; current = flac_read_loop (psf, readlen) ; if (current == 0) break ; total += current ; } ; return total ; } /* flac_read_flac2s */ static sf_count_t flac_read_flac2i (SF_PRIVATE *psf, int *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; sf_count_t total = 0, current ; unsigned readlen ; pflac->pcmtype = PFLAC_PCM_INT ; while (total < len) { pflac->ptr = ptr + total ; readlen = (len - total > 0x1000000) ? 0x1000000 : (unsigned) (len - total) ; current = flac_read_loop (psf, readlen) ; if (current == 0) break ; total += current ; } ; return total ; } /* flac_read_flac2i */ static sf_count_t flac_read_flac2f (SF_PRIVATE *psf, float *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; sf_count_t total = 0, current ; unsigned readlen ; pflac->pcmtype = PFLAC_PCM_FLOAT ; while (total < len) { pflac->ptr = ptr + total ; readlen = (len - total > 0x1000000) ? 0x1000000 : (unsigned) (len - total) ; current = flac_read_loop (psf, readlen) ; if (current == 0) break ; total += current ; } ; return total ; } /* flac_read_flac2f */ static sf_count_t flac_read_flac2d (SF_PRIVATE *psf, double *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; sf_count_t total = 0, current ; unsigned readlen ; pflac->pcmtype = PFLAC_PCM_DOUBLE ; while (total < len) { pflac->ptr = ptr + total ; readlen = (len - total > 0x1000000) ? 0x1000000 : (unsigned) (len - total) ; current = flac_read_loop (psf, readlen) ; if (current == 0) break ; total += current ; } ; return total ; } /* flac_read_flac2d */ static sf_count_t flac_write_s2flac (SF_PRIVATE *psf, const short *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; void (*convert) (const short *, FLAC__int32 *, int) ; int bufferlen, writecount, thiswrite ; sf_count_t total = 0 ; FLAC__int32* buffer = pflac->encbuffer ; switch (SF_CODEC (psf->sf.format)) { case SF_FORMAT_PCM_S8 : convert = s2flac8_array ; break ; case SF_FORMAT_PCM_16 : convert = s2flac16_array ; break ; case SF_FORMAT_PCM_24 : convert = s2flac24_array ; break ; default : return -1 ; } ; bufferlen = ENC_BUFFER_SIZE / (sizeof (FLAC__int32) * psf->sf.channels) ; bufferlen *= psf->sf.channels ; while (len > 0) { writecount = (len >= bufferlen) ? bufferlen : (int) len ; convert (ptr + total, buffer, writecount) ; if (FLAC__stream_encoder_process_interleaved (pflac->fse, buffer, writecount/psf->sf.channels)) thiswrite = writecount ; else break ; total += thiswrite ; if (thiswrite < writecount) break ; len -= thiswrite ; } ; return total ; } /* flac_write_s2flac */ static sf_count_t flac_write_i2flac (SF_PRIVATE *psf, const int *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; void (*convert) (const int *, FLAC__int32 *, int) ; int bufferlen, writecount, thiswrite ; sf_count_t total = 0 ; FLAC__int32* buffer = pflac->encbuffer ; switch (SF_CODEC (psf->sf.format)) { case SF_FORMAT_PCM_S8 : convert = i2flac8_array ; break ; case SF_FORMAT_PCM_16 : convert = i2flac16_array ; break ; case SF_FORMAT_PCM_24 : convert = i2flac24_array ; break ; default : return -1 ; } ; bufferlen = ENC_BUFFER_SIZE / (sizeof (FLAC__int32) * psf->sf.channels) ; bufferlen *= psf->sf.channels ; while (len > 0) { writecount = (len >= bufferlen) ? bufferlen : (int) len ; convert (ptr + total, buffer, writecount) ; if (FLAC__stream_encoder_process_interleaved (pflac->fse, buffer, writecount/psf->sf.channels)) thiswrite = writecount ; else break ; total += thiswrite ; if (thiswrite < writecount) break ; len -= thiswrite ; } ; return total ; } /* flac_write_i2flac */ static sf_count_t flac_write_f2flac (SF_PRIVATE *psf, const float *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; void (*convert) (const float *, FLAC__int32 *, int, int) ; int bufferlen, writecount, thiswrite ; sf_count_t total = 0 ; FLAC__int32* buffer = pflac->encbuffer ; switch (SF_CODEC (psf->sf.format)) { case SF_FORMAT_PCM_S8 : convert = (psf->add_clipping) ? f2flac8_clip_array : f2flac8_array ; break ; case SF_FORMAT_PCM_16 : convert = (psf->add_clipping) ? f2flac16_clip_array : f2flac16_array ; break ; case SF_FORMAT_PCM_24 : convert = (psf->add_clipping) ? f2flac24_clip_array : f2flac24_array ; break ; default : return -1 ; } ; bufferlen = ENC_BUFFER_SIZE / (sizeof (FLAC__int32) * psf->sf.channels) ; bufferlen *= psf->sf.channels ; while (len > 0) { writecount = (len >= bufferlen) ? bufferlen : (int) len ; convert (ptr + total, buffer, writecount, psf->norm_float) ; if (FLAC__stream_encoder_process_interleaved (pflac->fse, buffer, writecount/psf->sf.channels)) thiswrite = writecount ; else break ; total += thiswrite ; if (thiswrite < writecount) break ; len -= thiswrite ; } ; return total ; } /* flac_write_f2flac */ static void f2flac8_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact, scaled_value ; normfact = normalize ? (8.0 * 0x10) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7F)) { dest [count] = 0x7F ; continue ; } ; if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x10)) { dest [count] = 0x80 ; continue ; } ; dest [count] = lrintf (scaled_value) ; } ; return ; } /* f2flac8_clip_array */ static void f2flac16_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact, scaled_value ; normfact = normalize ? (8.0 * 0x1000) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7FFF)) { dest [count] = 0x7FFF ; continue ; } if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x1000)) { dest [count] = 0x8000 ; continue ; } dest [count] = lrintf (scaled_value) ; } } /* f2flac16_clip_array */ static void f2flac24_clip_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact, scaled_value ; normfact = normalize ? (8.0 * 0x100000) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7FFFFF)) { dest [count] = 0x7FFFFF ; continue ; } ; if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x100000)) { dest [count] = 0x800000 ; continue ; } dest [count] = lrintf (scaled_value) ; } ; return ; } /* f2flac24_clip_array */ static void f2flac8_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact = normalize ? (1.0 * 0x7F) : 1.0 ; while (--count >= 0) dest [count] = lrintf (src [count] * normfact) ; } /* f2flac8_array */ static void f2flac16_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact = normalize ? (1.0 * 0x7FFF) : 1.0 ; while (--count >= 0) dest [count] = lrintf (src [count] * normfact) ; } /* f2flac16_array */ static void f2flac24_array (const float *src, FLAC__int32 *dest, int count, int normalize) { float normfact = normalize ? (1.0 * 0x7FFFFF) : 1.0 ; while (--count >= 0) dest [count] = lrintf (src [count] * normfact) ; } /* f2flac24_array */ static sf_count_t flac_write_d2flac (SF_PRIVATE *psf, const double *ptr, sf_count_t len) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; void (*convert) (const double *, FLAC__int32 *, int, int) ; int bufferlen, writecount, thiswrite ; sf_count_t total = 0 ; FLAC__int32* buffer = pflac->encbuffer ; switch (SF_CODEC (psf->sf.format)) { case SF_FORMAT_PCM_S8 : convert = (psf->add_clipping) ? d2flac8_clip_array : d2flac8_array ; break ; case SF_FORMAT_PCM_16 : convert = (psf->add_clipping) ? d2flac16_clip_array : d2flac16_array ; break ; case SF_FORMAT_PCM_24 : convert = (psf->add_clipping) ? d2flac24_clip_array : d2flac24_array ; break ; default : return -1 ; } ; bufferlen = ENC_BUFFER_SIZE / (sizeof (FLAC__int32) * psf->sf.channels) ; bufferlen *= psf->sf.channels ; while (len > 0) { writecount = (len >= bufferlen) ? bufferlen : (int) len ; convert (ptr + total, buffer, writecount, psf->norm_double) ; if (FLAC__stream_encoder_process_interleaved (pflac->fse, buffer, writecount/psf->sf.channels)) thiswrite = writecount ; else break ; total += thiswrite ; if (thiswrite < writecount) break ; len -= thiswrite ; } ; return total ; } /* flac_write_d2flac */ static void d2flac8_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact, scaled_value ; normfact = normalize ? (8.0 * 0x10) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7F)) { dest [count] = 0x7F ; continue ; } ; if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x10)) { dest [count] = 0x80 ; continue ; } ; dest [count] = lrint (scaled_value) ; } ; return ; } /* d2flac8_clip_array */ static void d2flac16_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact, scaled_value ; normfact = normalize ? (8.0 * 0x1000) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7FFF)) { dest [count] = 0x7FFF ; continue ; } ; if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x1000)) { dest [count] = 0x8000 ; continue ; } ; dest [count] = lrint (scaled_value) ; } ; return ; } /* d2flac16_clip_array */ static void d2flac24_clip_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact, scaled_value ; normfact = normalize ? (8.0 * 0x100000) : 1.0 ; while (--count >= 0) { scaled_value = src [count] * normfact ; if (CPU_CLIPS_POSITIVE == 0 && scaled_value >= (1.0 * 0x7FFFFF)) { dest [count] = 0x7FFFFF ; continue ; } ; if (CPU_CLIPS_NEGATIVE == 0 && scaled_value <= (-8.0 * 0x100000)) { dest [count] = 0x800000 ; continue ; } ; dest [count] = lrint (scaled_value) ; } ; return ; } /* d2flac24_clip_array */ static void d2flac8_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact = normalize ? (1.0 * 0x7F) : 1.0 ; while (--count >= 0) dest [count] = lrint (src [count] * normfact) ; } /* d2flac8_array */ static void d2flac16_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact = normalize ? (1.0 * 0x7FFF) : 1.0 ; while (--count >= 0) dest [count] = lrint (src [count] * normfact) ; } /* d2flac16_array */ static void d2flac24_array (const double *src, FLAC__int32 *dest, int count, int normalize) { double normfact = normalize ? (1.0 * 0x7FFFFF) : 1.0 ; while (--count >= 0) dest [count] = lrint (src [count] * normfact) ; } /* d2flac24_array */ static sf_count_t flac_seek (SF_PRIVATE *psf, int UNUSED (mode), sf_count_t offset) { FLAC_PRIVATE* pflac = (FLAC_PRIVATE*) psf->codec_data ; if (pflac == NULL) return 0 ; if (psf->dataoffset < 0) { psf->error = SFE_BAD_SEEK ; return ((sf_count_t) -1) ; } ; pflac->frame = NULL ; if (psf->file.mode == SFM_READ) { FLAC__uint64 position ; if (FLAC__stream_decoder_seek_absolute (pflac->fsd, offset)) { FLAC__stream_decoder_get_decode_position (pflac->fsd, &position) ; return offset ; } ; return ((sf_count_t) -1) ; } ; /* Seeking in write mode not yet supported. */ psf->error = SFE_BAD_SEEK ; return ((sf_count_t) -1) ; } /* flac_seek */ #else /* HAVE_EXTERNAL_LIBS */ int flac_open (SF_PRIVATE *psf) { psf_log_printf (psf, "This version of libsndfile was compiled without FLAC support.\n") ; return SFE_UNIMPLEMENTED ; } /* flac_open */ #endif