-/*
-** Copyright (C) 2002-2011 Erik de Castro Lopo <erikd@mega-nerd.com>
-** Copyright (C) 2002-2005 Michael Smith <msmith@xiph.org>
-** Copyright (C) 2007 John ffitch
-**
-** 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.
-*/
-
-/*
-** Much of this code is based on the examples in libvorbis from the
-** XIPHOPHORUS Company http://www.xiph.org/ which has a BSD-style Licence
-** Copyright (c) 2002, Xiph.org Foundation
-**
-** Redistribution and use in source and binary forms, with or without
-** modification, are permitted provided that the following conditions
-** are met:
-**
-** - Redistributions of source code must retain the above copyright
-** notice, this list of conditions and the following disclaimer.
-**
-** - Redistributions in binary form must reproduce the above copyright
-** notice, this list of conditions and the following disclaimer in the
-** documentation and/or other materials provided with the distribution.
-**
-** - Neither the name of the Xiph.org Foundation nor the names of its
-** contributors may be used to endorse or promote products derived from
-** this software without specific prior written permission.
-**
-** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
-** ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
-** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
-** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION
-** OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
-** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
-** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ; LOSS OF USE,
-** DATA, OR PROFITS ; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
-** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
-** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
-** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-*/
-
-#include "sfconfig.h"
-
-#include <stdio.h>
-#include <fcntl.h>
-#include <string.h>
-#include <ctype.h>
-#include <time.h>
-#include <math.h>
-
-#if HAVE_UNISTD_H
-#include <unistd.h>
-#endif
-
-#include "sndfile.h"
-#include "sfendian.h"
-#include "common.h"
-
-#if HAVE_EXTERNAL_LIBS
-
-#include <ogg/ogg.h>
-#include <vorbis/codec.h>
-#include <vorbis/vorbisenc.h>
-
-#include "ogg.h"
-
-typedef int convert_func (int, void *, int, int, float **) ;
-
-static int vorbis_read_header (SF_PRIVATE *psf, int log_data) ;
-static int vorbis_write_header (SF_PRIVATE *psf, int calc_length) ;
-static int vorbis_close (SF_PRIVATE *psf) ;
-static int vorbis_command (SF_PRIVATE *psf, int command, void *data, int datasize) ;
-static sf_count_t vorbis_seek (SF_PRIVATE *psf, int mode, sf_count_t offset) ;
-static sf_count_t vorbis_read_s (SF_PRIVATE *psf, short *ptr, sf_count_t len) ;
-static sf_count_t vorbis_read_i (SF_PRIVATE *psf, int *ptr, sf_count_t len) ;
-static sf_count_t vorbis_read_f (SF_PRIVATE *psf, float *ptr, sf_count_t len) ;
-static sf_count_t vorbis_read_d (SF_PRIVATE *psf, double *ptr, sf_count_t len) ;
-static sf_count_t vorbis_write_s (SF_PRIVATE *psf, const short *ptr, sf_count_t len) ;
-static sf_count_t vorbis_write_i (SF_PRIVATE *psf, const int *ptr, sf_count_t len) ;
-static sf_count_t vorbis_write_f (SF_PRIVATE *psf, const float *ptr, sf_count_t len) ;
-static sf_count_t vorbis_write_d (SF_PRIVATE *psf, const double *ptr, sf_count_t len) ;
-static sf_count_t vorbis_read_sample (SF_PRIVATE *psf, void *ptr, sf_count_t lens, convert_func *transfn) ;
-static sf_count_t vorbis_length (SF_PRIVATE *psf) ;
-
-typedef struct
-{ int id ;
- const char *name ;
-} STR_PAIRS ;
-
-static STR_PAIRS vorbis_metatypes [] =
-{ { SF_STR_TITLE, "Title" },
- { SF_STR_COPYRIGHT, "Copyright" },
- { SF_STR_SOFTWARE, "Software" },
- { SF_STR_ARTIST, "Artist" },
- { SF_STR_COMMENT, "Comment" },
- { SF_STR_DATE, "Date" },
- { SF_STR_ALBUM, "Album" },
- { SF_STR_LICENSE, "License" },
-} ;
-
-typedef struct
-{ /* Count current location */
- sf_count_t loc ;
- /* Struct that stores all the static vorbis bitstream settings */
- vorbis_info vinfo ;
- /* Struct that stores all the bitstream user comments */
- vorbis_comment vcomment ;
- /* Ventral working state for the packet->PCM decoder */
- vorbis_dsp_state vdsp ;
- /* Local working space for packet->PCM decode */
- vorbis_block vblock ;
-
- /* Encoding quality in range [0.0, 1.0]. */
- double quality ;
-} VORBIS_PRIVATE ;
-
-static int
-vorbis_read_header (SF_PRIVATE *psf, int log_data)
-{
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- char *buffer ;
- int bytes ;
- int i, nn ;
-
- odata->eos = 0 ;
-
- /* Weird stuff happens if these aren't called. */
- ogg_stream_reset (&odata->ostream) ;
- ogg_sync_reset (&odata->osync) ;
-
- /*
- ** Grab some data at the head of the stream. We want the first page
- ** (which is guaranteed to be small and only contain the Vorbis
- ** stream initial header) We need the first page to get the stream
- ** serialno.
- */
-
- /* Expose the buffer */
- buffer = ogg_sync_buffer (&odata->osync, 4096L) ;
-
- /* Grab the part of the header that has already been read. */
- memcpy (buffer, psf->header, psf->headindex) ;
- bytes = psf->headindex ;
-
- /* Submit a 4k block to libvorbis' Ogg layer */
- bytes += psf_fread (buffer + psf->headindex, 1, 4096 - psf->headindex, psf) ;
- ogg_sync_wrote (&odata->osync, bytes) ;
-
- /* Get the first page. */
- if ((nn = ogg_sync_pageout (&odata->osync, &odata->opage)) != 1)
- {
- /* Have we simply run out of data? If so, we're done. */
- if (bytes < 4096)
- return 0 ;
-
- /* Error case. Must not be Vorbis data */
- psf_log_printf (psf, "Input does not appear to be an Ogg bitstream.\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
-
- /*
- ** Get the serial number and set up the rest of decode.
- ** Serialno first ; use it to set up a logical stream.
- */
- ogg_stream_clear (&odata->ostream) ;
- ogg_stream_init (&odata->ostream, ogg_page_serialno (&odata->opage)) ;
-
- if (ogg_stream_pagein (&odata->ostream, &odata->opage) < 0)
- { /* Error ; stream version mismatch perhaps. */
- psf_log_printf (psf, "Error reading first page of Ogg bitstream data\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
-
- if (ogg_stream_packetout (&odata->ostream, &odata->opacket) != 1)
- { /* No page? must not be vorbis. */
- psf_log_printf (psf, "Error reading initial header packet.\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
-
- /*
- ** This function (vorbis_read_header) gets called multiple times, so the OGG
- ** and vorbis structs have to be cleared every time we pass through to
- ** prevent memory leaks.
- */
- vorbis_block_clear (&vdata->vblock) ;
- vorbis_dsp_clear (&vdata->vdsp) ;
- vorbis_comment_clear (&vdata->vcomment) ;
- vorbis_info_clear (&vdata->vinfo) ;
-
- /*
- ** Extract the initial header from the first page and verify that the
- ** Ogg bitstream is in fact Vorbis data.
- **
- ** I handle the initial header first instead of just having the code
- ** read all three Vorbis headers at once because reading the initial
- ** header is an easy way to identify a Vorbis bitstream and it's
- ** useful to see that functionality seperated out.
- */
- vorbis_info_init (&vdata->vinfo) ;
- vorbis_comment_init (&vdata->vcomment) ;
-
- if (vorbis_synthesis_headerin (&vdata->vinfo, &vdata->vcomment, &odata->opacket) < 0)
- { /* Error case ; not a vorbis header. */
- psf_log_printf (psf, "Found Vorbis in stream header, but vorbis_synthesis_headerin failed.\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
-
- /*
- ** Common Ogg metadata fields?
- ** TITLE, VERSION, ALBUM, TRACKNUMBER, ARTIST, PERFORMER, COPYRIGHT, LICENSE,
- ** ORGANIZATION, DESCRIPTION, GENRE, DATE, LOCATION, CONTACT, ISRC,
- */
-
- if (log_data)
- { int k ;
-
- for (k = 0 ; k < ARRAY_LEN (vorbis_metatypes) ; k++)
- { char *dd ;
-
- dd = vorbis_comment_query (&vdata->vcomment, vorbis_metatypes [k].name, 0) ;
- if (dd == NULL)
- continue ;
- psf_store_string (psf, vorbis_metatypes [k].id, dd) ;
- } ;
- } ;
-
- /*
- ** At this point, we're sure we're Vorbis. We've set up the logical (Ogg)
- ** bitstream decoder. Get the comment and codebook headers and set up the
- ** Vorbis decoder.
- **
- ** The next two packets in order are the comment and codebook headers.
- ** They're likely large and may span multiple pages. Thus we reead
- ** and submit data until we get our two pacakets, watching that no
- ** pages are missing. If a page is missing, error out ; losing a
- ** header page is the only place where missing data is fatal.
- */
-
- i = 0 ; /* Count of number of packets read */
- while (i < 2)
- { int result = ogg_sync_pageout (&odata->osync, &odata->opage) ;
- if (result == 0)
- { /* Need more data */
- buffer = ogg_sync_buffer (&odata->osync, 4096) ;
- bytes = psf_fread (buffer, 1, 4096, psf) ;
-
- if (bytes == 0 && i < 2)
- { psf_log_printf (psf, "End of file before finding all Vorbis headers!\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
- nn = ogg_sync_wrote (&odata->osync, bytes) ;
- }
- else if (result == 1)
- { /*
- ** Don't complain about missing or corrupt data yet. We'll
- ** catch it at the packet output phase.
- **
- ** We can ignore any errors here as they'll also become apparent
- ** at packetout.
- */
- nn = ogg_stream_pagein (&odata->ostream, &odata->opage) ;
- while (i < 2)
- { result = ogg_stream_packetout (&odata->ostream, &odata->opacket) ;
- if (result == 0)
- break ;
- if (result < 0)
- { /* Uh oh ; data at some point was corrupted or missing!
- ** We can't tolerate that in a header. Die. */
- psf_log_printf (psf, "Corrupt secondary header. Exiting.\n") ;
- return SFE_MALFORMED_FILE ;
- } ;
-
- vorbis_synthesis_headerin (&vdata->vinfo, &vdata->vcomment, &odata->opacket) ;
- i++ ;
- } ;
- } ;
- } ;
-
- if (log_data)
- { int printed_metadata_msg = 0 ;
- int k ;
-
- psf_log_printf (psf, "Bitstream is %d channel, %D Hz\n", vdata->vinfo.channels, vdata->vinfo.rate) ;
- psf_log_printf (psf, "Encoded by : %s\n", vdata->vcomment.vendor) ;
-
- /* Throw the comments plus a few lines about the bitstream we're decoding. */
- for (k = 0 ; k < ARRAY_LEN (vorbis_metatypes) ; k++)
- { char *dd ;
-
- dd = vorbis_comment_query (&vdata->vcomment, vorbis_metatypes [k].name, 0) ;
- if (dd == NULL)
- continue ;
-
- if (printed_metadata_msg == 0)
- { psf_log_printf (psf, "Metadata :\n") ;
- printed_metadata_msg = 1 ;
- } ;
-
- psf_store_string (psf, vorbis_metatypes [k].id, dd) ;
- psf_log_printf (psf, " %-10s : %s\n", vorbis_metatypes [k].name, dd) ;
- } ;
-
- psf_log_printf (psf, "End\n") ;
- } ;
-
- psf->sf.samplerate = vdata->vinfo.rate ;
- psf->sf.channels = vdata->vinfo.channels ;
- psf->sf.format = SF_FORMAT_OGG | SF_FORMAT_VORBIS ;
-
- /* OK, got and parsed all three headers. Initialize the Vorbis
- ** packet->PCM decoder.
- ** Central decode state. */
- vorbis_synthesis_init (&vdata->vdsp, &vdata->vinfo) ;
-
- /* Local state for most of the decode so multiple block decodes can
- ** proceed in parallel. We could init multiple vorbis_block structures
- ** for vd here. */
- vorbis_block_init (&vdata->vdsp, &vdata->vblock) ;
-
- vdata->loc = 0 ;
-
- return 0 ;
-} /* vorbis_read_header */
-
-static int
-vorbis_write_header (SF_PRIVATE *psf, int UNUSED (calc_length))
-{
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- int k, ret ;
-
- vorbis_info_init (&vdata->vinfo) ;
-
- /* The style of encoding should be selectable here, VBR quality mode. */
- ret = vorbis_encode_init_vbr (&vdata->vinfo, psf->sf.channels, psf->sf.samplerate, vdata->quality) ;
-
-#if 0
- ret = vorbis_encode_init (&vdata->vinfo, psf->sf.channels, psf->sf.samplerate, -1, 128000, -1) ; /* average bitrate mode */
- ret = ( vorbis_encode_setup_managed (&vdata->vinfo, psf->sf.channels,
- psf->sf.samplerate, -1, 128000, -1) ||
- vorbis_encode_ctl (&vdata->vinfo, OV_ECTL_RATEMANAGE_AVG, NULL) ||
- vorbis_encode_setup_init (&vdata->vinfo)) ;
-#endif
- if (ret)
- return SFE_BAD_OPEN_FORMAT ;
-
- vdata->loc = 0 ;
-
- /* add a comment */
- vorbis_comment_init (&vdata->vcomment) ;
-
- vorbis_comment_add_tag (&vdata->vcomment, "ENCODER", "libsndfile") ;
- for (k = 0 ; k < SF_MAX_STRINGS ; k++)
- { const char * name ;
-
- if (psf->strings [k].type == 0)
- break ;
-
- switch (psf->strings [k].type)
- { case SF_STR_TITLE : name = "TITLE" ; break ;
- case SF_STR_COPYRIGHT : name = "COPYRIGHT" ; break ;
- case SF_STR_SOFTWARE : name = "SOFTWARE" ; break ;
- case SF_STR_ARTIST : name = "ARTIST" ; break ;
- case SF_STR_COMMENT : name = "COMMENT" ; break ;
- case SF_STR_DATE : name = "DATE" ; break ;
- case SF_STR_ALBUM : name = "ALBUM" ; break ;
- case SF_STR_LICENSE : name = "LICENSE" ; break ;
- default : continue ;
- } ;
-
- vorbis_comment_add_tag (&vdata->vcomment, name, psf->strings [k].str) ;
- } ;
-
- /* set up the analysis state and auxiliary encoding storage */
- vorbis_analysis_init (&vdata->vdsp, &vdata->vinfo) ;
- vorbis_block_init (&vdata->vdsp, &vdata->vblock) ;
-
- /*
- ** Set up our packet->stream encoder.
- ** Pick a random serial number ; that way we can more likely build
- ** chained streams just by concatenation.
- */
-
- ogg_stream_init (&odata->ostream, psf_rand_int32 ()) ;
-
- /* Vorbis streams begin with three headers ; the initial header (with
- most of the codec setup parameters) which is mandated by the Ogg
- bitstream spec. The second header holds any comment fields. The
- third header holds the bitstream codebook. We merely need to
- make the headers, then pass them to libvorbis one at a time ;
- libvorbis handles the additional Ogg bitstream constraints */
-
- { ogg_packet header ;
- ogg_packet header_comm ;
- ogg_packet header_code ;
- int result ;
-
- vorbis_analysis_headerout (&vdata->vdsp, &vdata->vcomment, &header, &header_comm, &header_code) ;
- ogg_stream_packetin (&odata->ostream, &header) ; /* automatically placed in its own page */
- ogg_stream_packetin (&odata->ostream, &header_comm) ;
- ogg_stream_packetin (&odata->ostream, &header_code) ;
-
- /* This ensures the actual
- * audio data will start on a new page, as per spec
- */
- while ((result = ogg_stream_flush (&odata->ostream, &odata->opage)) != 0)
- { psf_fwrite (odata->opage.header, 1, odata->opage.header_len, psf) ;
- psf_fwrite (odata->opage.body, 1, odata->opage.body_len, psf) ;
- } ;
- }
-
- return 0 ;
-} /* vorbis_write_header */
-
-static int
-vorbis_close (SF_PRIVATE *psf)
-{ OGG_PRIVATE* odata = psf->container_data ;
- VORBIS_PRIVATE *vdata = psf->codec_data ;
-
- if (odata == NULL || vdata == NULL)
- return 0 ;
-
- /* Clean up this logical bitstream ; before exit we shuld see if we're
- ** followed by another [chained]. */
-
- if (psf->file.mode == SFM_WRITE)
- {
- if (psf->write_current <= 0)
- vorbis_write_header (psf, 0) ;
-
- vorbis_analysis_wrote (&vdata->vdsp, 0) ;
- while (vorbis_analysis_blockout (&vdata->vdsp, &vdata->vblock) == 1)
- {
-
- /* analysis, assume we want to use bitrate management */
- vorbis_analysis (&vdata->vblock, NULL) ;
- vorbis_bitrate_addblock (&vdata->vblock) ;
-
- while (vorbis_bitrate_flushpacket (&vdata->vdsp, &odata->opacket))
- { /* weld the packet into the bitstream */
- ogg_stream_packetin (&odata->ostream, &odata->opacket) ;
-
- /* write out pages (if any) */
- while (!odata->eos)
- { int result = ogg_stream_pageout (&odata->ostream, &odata->opage) ;
- if (result == 0) break ;
- psf_fwrite (odata->opage.header, 1, odata->opage.header_len, psf) ;
- psf_fwrite (odata->opage.body, 1, odata->opage.body_len, psf) ;
-
- /* this could be set above, but for illustrative purposes, I do
- it here (to show that vorbis does know where the stream ends) */
-
- if (ogg_page_eos (&odata->opage)) odata->eos = 1 ;
- }
- }
- }
- }
-
- /* ogg_page and ogg_packet structs always point to storage in
- libvorbis. They are never freed or manipulated directly */
-
- vorbis_block_clear (&vdata->vblock) ;
- vorbis_dsp_clear (&vdata->vdsp) ;
- vorbis_comment_clear (&vdata->vcomment) ;
- vorbis_info_clear (&vdata->vinfo) ;
-
- return 0 ;
-} /* vorbis_close */
-
-int
-ogg_vorbis_open (SF_PRIVATE *psf)
-{ OGG_PRIVATE* odata = psf->container_data ;
- VORBIS_PRIVATE* vdata = calloc (1, sizeof (VORBIS_PRIVATE)) ;
- int error = 0 ;
-
- if (odata == NULL)
- { psf_log_printf (psf, "%s : odata is NULL???\n", __func__) ;
- return SFE_INTERNAL ;
- } ;
-
- psf->codec_data = vdata ;
-
- if (psf->file.mode == SFM_RDWR)
- return SFE_BAD_MODE_RW ;
-
- psf_log_printf (psf, "Vorbis library version : %s\n", vorbis_version_string ()) ;
-
- if (psf->file.mode == SFM_READ)
- { /* Call this here so it only gets called once, so no memory is leaked. */
- ogg_sync_init (&odata->osync) ;
-
- if ((error = vorbis_read_header (psf, 1)))
- return error ;
-
- psf->read_short = vorbis_read_s ;
- psf->read_int = vorbis_read_i ;
- psf->read_float = vorbis_read_f ;
- psf->read_double = vorbis_read_d ;
- psf->sf.frames = vorbis_length (psf) ;
- } ;
-
- psf->codec_close = vorbis_close ;
- if (psf->file.mode == SFM_WRITE)
- {
- /* Set the default vorbis quality here. */
- vdata->quality = 0.4 ;
-
- psf->write_header = vorbis_write_header ;
- psf->write_short = vorbis_write_s ;
- psf->write_int = vorbis_write_i ;
- psf->write_float = vorbis_write_f ;
- psf->write_double = vorbis_write_d ;
-
- psf->sf.frames = SF_COUNT_MAX ; /* Unknown really */
- psf->str_flags = SF_STR_ALLOW_START ;
- } ;
-
- psf->bytewidth = 1 ;
- psf->blockwidth = psf->bytewidth * psf->sf.channels ;
-
- psf->seek = vorbis_seek ;
- psf->command = vorbis_command ;
-
- /* FIXME, FIXME, FIXME : Hack these here for now and correct later. */
- psf->sf.format = SF_FORMAT_OGG | SF_FORMAT_VORBIS ;
- psf->sf.sections = 1 ;
-
- psf->datalength = 1 ;
- psf->dataoffset = 0 ;
- /* End FIXME. */
-
- return error ;
-} /* ogg_vorbis_open */
-
-static int
-vorbis_command (SF_PRIVATE *psf, int command, void * data, int datasize)
-{ VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
-
- switch (command)
- { case SFC_SET_VBR_ENCODING_QUALITY :
- if (data == NULL || datasize != sizeof (double))
- return SF_FALSE ;
-
- if (psf->have_written)
- return SF_FALSE ;
-
- vdata->quality = *((double *) data) ;
-
- /* Clip range. */
- vdata->quality = SF_MAX (0.0, SF_MIN (1.0, vdata->quality)) ;
-
- psf_log_printf (psf, "%s : Setting SFC_SET_VBR_ENCODING_QUALITY to %f.\n", __func__, vdata->quality) ;
- return SF_TRUE ;
-
- default :
- return SF_FALSE ;
- } ;
-
- return SF_FALSE ;
-} /* vorbis_command */
-
-static int
-vorbis_rnull (int samples, void *UNUSED (vptr), int UNUSED (off) , int channels, float **UNUSED (pcm))
-{
- return samples * channels ;
-} /* vorbis_rnull */
-
-static int
-vorbis_rshort (int samples, void *vptr, int off, int channels, float **pcm)
-{
- short *ptr = (short*) vptr + off ;
- int i = 0, j, n ;
- for (j = 0 ; j < samples ; j++)
- for (n = 0 ; n < channels ; n++)
- ptr [i++] = lrintf (pcm [n][j] * 32767.0f) ;
- return i ;
-} /* vorbis_rshort */
-
-static int
-vorbis_rint (int samples, void *vptr, int off, int channels, float **pcm)
-{
- int *ptr = (int*) vptr + off ;
- int i = 0, j, n ;
-
- for (j = 0 ; j < samples ; j++)
- for (n = 0 ; n < channels ; n++)
- ptr [i++] = lrintf (pcm [n][j] * 2147483647.0f) ;
- return i ;
-} /* vorbis_rint */
-
-static int
-vorbis_rfloat (int samples, void *vptr, int off, int channels, float **pcm)
-{
- float *ptr = (float*) vptr + off ;
- int i = 0, j, n ;
- for (j = 0 ; j < samples ; j++)
- for (n = 0 ; n < channels ; n++)
- ptr [i++] = pcm [n][j] ;
- return i ;
-} /* vorbis_rfloat */
-
-static int
-vorbis_rdouble (int samples, void *vptr, int off, int channels, float **pcm)
-{
- double *ptr = (double*) vptr + off ;
- int i = 0, j, n ;
- for (j = 0 ; j < samples ; j++)
- for (n = 0 ; n < channels ; n++)
- ptr [i++] = pcm [n][j] ;
- return i ;
-} /* vorbis_rdouble */
-
-
-static sf_count_t
-vorbis_read_sample (SF_PRIVATE *psf, void *ptr, sf_count_t lens, convert_func *transfn)
-{
- VORBIS_PRIVATE *vdata = psf->codec_data ;
- OGG_PRIVATE *odata = psf->container_data ;
- int len, samples, i = 0 ;
- float **pcm ;
-
- len = lens / psf->sf.channels ;
-
- while ((samples = vorbis_synthesis_pcmout (&vdata->vdsp, &pcm)) > 0)
- { if (samples > len) samples = len ;
- i += transfn (samples, ptr, i, psf->sf.channels, pcm) ;
- len -= samples ;
- /* tell libvorbis how many samples we actually consumed */
- vorbis_synthesis_read (&vdata->vdsp, samples) ;
- vdata->loc += samples ;
- if (len == 0)
- return i ; /* Is this necessary */
- }
- goto start0 ; /* Jump into the nasty nest */
- while (len > 0 && !odata->eos)
- {
- while (len > 0 && !odata->eos)
- { int result = ogg_sync_pageout (&odata->osync, &odata->opage) ;
- if (result == 0) break ; /* need more data */
- if (result < 0)
- { /* missing or corrupt data at this page position */
- psf_log_printf (psf, "Corrupt or missing data in bitstream ; continuing...\n") ;
- }
- else
- { /* can safely ignore errors at this point */
- ogg_stream_pagein (&odata->ostream, &odata->opage) ;
- start0:
- while (1)
- { result = ogg_stream_packetout (&odata->ostream, &odata->opacket) ;
- if (result == 0)
- break ; /* need more data */
- if (result < 0)
- { /* missing or corrupt data at this page position */
- /* no reason to complain ; already complained above */
- }
- else
- { /* we have a packet. Decode it */
- if (vorbis_synthesis (&vdata->vblock, &odata->opacket) == 0) /* test for success! */
- vorbis_synthesis_blockin (&vdata->vdsp, &vdata->vblock) ;
- /*
- **pcm is a multichannel float vector. In stereo, for
- example, pcm [0] is left, and pcm [1] is right. samples is
- the size of each channel. Convert the float values
- (-1.<=range<=1.) to whatever PCM format and write it out */
-
- while ((samples = vorbis_synthesis_pcmout (&vdata->vdsp, &pcm)) > 0)
- { if (samples>len) samples = len ;
- i += transfn (samples, ptr, i, psf->sf.channels, pcm) ;
- len -= samples ;
- /* tell libvorbis how many samples we actually consumed */
- vorbis_synthesis_read (&vdata->vdsp, samples) ;
- vdata->loc += samples ;
- if (len == 0)
- return i ; /* Is this necessary */
- } ;
- }
- }
- if (ogg_page_eos (&odata->opage)) odata->eos = 1 ;
- }
- }
- if (!odata->eos)
- { char *buffer ;
- int bytes ;
- buffer = ogg_sync_buffer (&odata->osync, 4096) ;
- bytes = psf_fread (buffer, 1, 4096, psf) ;
- ogg_sync_wrote (&odata->osync, bytes) ;
- if (bytes == 0) odata->eos = 1 ;
- }
- }
- return i ;
-} /* vorbis_read_sample */
-
-static sf_count_t
-vorbis_read_s (SF_PRIVATE *psf, short *ptr, sf_count_t lens)
-{ return vorbis_read_sample (psf, (void*) ptr, lens, vorbis_rshort) ;
-} /* vorbis_read_s */
-
-static sf_count_t
-vorbis_read_i (SF_PRIVATE *psf, int *ptr, sf_count_t lens)
-{ return vorbis_read_sample (psf, (void*) ptr, lens, vorbis_rint) ;
-} /* vorbis_read_i */
-
-static sf_count_t
-vorbis_read_f (SF_PRIVATE *psf, float *ptr, sf_count_t lens)
-{ return vorbis_read_sample (psf, (void*) ptr, lens, vorbis_rfloat) ;
-} /* vorbis_read_f */
-
-static sf_count_t
-vorbis_read_d (SF_PRIVATE *psf, double *ptr, sf_count_t lens)
-{ return vorbis_read_sample (psf, (void*) ptr, lens, vorbis_rdouble) ;
-} /* vorbis_read_d */
-
-/*==============================================================================
-*/
-
-static void
-vorbis_write_samples (SF_PRIVATE *psf, OGG_PRIVATE *odata, VORBIS_PRIVATE *vdata, int in_frames)
-{
- vorbis_analysis_wrote (&vdata->vdsp, in_frames) ;
-
- /*
- ** Vorbis does some data preanalysis, then divvies up blocks for
- ** more involved (potentially parallel) processing. Get a single
- ** block for encoding now.
- */
- while (vorbis_analysis_blockout (&vdata->vdsp, &vdata->vblock) == 1)
- {
- /* analysis, assume we want to use bitrate management */
- vorbis_analysis (&vdata->vblock, NULL) ;
- vorbis_bitrate_addblock (&vdata->vblock) ;
-
- while (vorbis_bitrate_flushpacket (&vdata->vdsp, &odata->opacket))
- {
- /* weld the packet into the bitstream */
- ogg_stream_packetin (&odata->ostream, &odata->opacket) ;
-
- /* write out pages (if any) */
- while (!odata->eos)
- { int result = ogg_stream_pageout (&odata->ostream, &odata->opage) ;
- if (result == 0)
- break ;
- psf_fwrite (odata->opage.header, 1, odata->opage.header_len, psf) ;
- psf_fwrite (odata->opage.body, 1, odata->opage.body_len, psf) ;
-
- /* This could be set above, but for illustrative purposes, I do
- ** it here (to show that vorbis does know where the stream ends) */
- if (ogg_page_eos (&odata->opage))
- odata->eos = 1 ;
- } ;
- } ;
- } ;
-
- vdata->loc += in_frames ;
-} /* vorbis_write_data */
-
-
-static sf_count_t
-vorbis_write_s (SF_PRIVATE *psf, const short *ptr, sf_count_t lens)
-{
- int i, m, j = 0 ;
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- int in_frames = lens / psf->sf.channels ;
- float **buffer = vorbis_analysis_buffer (&vdata->vdsp, in_frames) ;
- for (i = 0 ; i < in_frames ; i++)
- for (m = 0 ; m < psf->sf.channels ; m++)
- buffer [m][i] = (float) (ptr [j++]) / 32767.0f ;
-
- vorbis_write_samples (psf, odata, vdata, in_frames) ;
-
- return lens ;
-} /* vorbis_write_s */
-
-static sf_count_t
-vorbis_write_i (SF_PRIVATE *psf, const int *ptr, sf_count_t lens)
-{ int i, m, j = 0 ;
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- int in_frames = lens / psf->sf.channels ;
- float **buffer = vorbis_analysis_buffer (&vdata->vdsp, in_frames) ;
- for (i = 0 ; i < in_frames ; i++)
- for (m = 0 ; m < psf->sf.channels ; m++)
- buffer [m][i] = (float) (ptr [j++]) / 2147483647.0f ;
-
- vorbis_write_samples (psf, odata, vdata, in_frames) ;
-
- return lens ;
-} /* vorbis_write_i */
-
-static sf_count_t
-vorbis_write_f (SF_PRIVATE *psf, const float *ptr, sf_count_t lens)
-{ int i, m, j = 0 ;
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- int in_frames = lens / psf->sf.channels ;
- float **buffer = vorbis_analysis_buffer (&vdata->vdsp, in_frames) ;
- for (i = 0 ; i < in_frames ; i++)
- for (m = 0 ; m < psf->sf.channels ; m++)
- buffer [m][i] = ptr [j++] ;
-
- vorbis_write_samples (psf, odata, vdata, in_frames) ;
-
- return lens ;
-} /* vorbis_write_f */
-
-static sf_count_t
-vorbis_write_d (SF_PRIVATE *psf, const double *ptr, sf_count_t lens)
-{ int i, m, j = 0 ;
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
- int in_frames = lens / psf->sf.channels ;
- float **buffer = vorbis_analysis_buffer (&vdata->vdsp, in_frames) ;
- for (i = 0 ; i < in_frames ; i++)
- for (m = 0 ; m < psf->sf.channels ; m++)
- buffer [m][i] = (float) ptr [j++] ;
-
- vorbis_write_samples (psf, odata, vdata, in_frames) ;
-
- return lens ;
-} /* vorbis_write_d */
-
-static sf_count_t
-vorbis_seek (SF_PRIVATE *psf, int UNUSED (mode), sf_count_t offset)
-{
- OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
- VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
-
- if (odata == NULL || vdata == NULL)
- return 0 ;
-
- if (offset < 0)
- { psf->error = SFE_BAD_SEEK ;
- return ((sf_count_t) -1) ;
- } ;
-
- if (psf->file.mode == SFM_READ)
- { sf_count_t target = offset - vdata->loc ;
-
- if (target < 0)
- { /* 12 to allow for OggS bit */
- psf_fseek (psf, 12, SEEK_SET) ;
- vorbis_read_header (psf, 0) ; /* Reset state */
- target = offset ;
- } ;
-
- while (target > 0)
- { sf_count_t m = target > 4096 ? 4096 : target ;
-
- /*
- ** Need to multiply by channels here because the seek is done in
- ** terms of frames and the read function is done in terms of
- ** samples.
- */
- vorbis_read_sample (psf, (void *) NULL, m * psf->sf.channels, vorbis_rnull) ;
-
- target -= m ;
- } ;
-
- return vdata->loc ;
- } ;
-
- return 0 ;
-} /* vorbis_seek */
-
-/*==============================================================================
-** Most of the following code was snipped from Mike Smith's ogginfo utility
-** which is part of vorbis-tools.
-** Vorbis tools is released under the GPL but Mike has kindly allowed the
-** following to be relicensed as LGPL for libsndfile.
-*/
-
-typedef struct
-{
- int isillegal ;
- int shownillegal ;
- int isnew ;
- int end ;
-
- uint32_t serial ; /* must be 32 bit unsigned */
- ogg_stream_state ostream ;
-
- vorbis_info vinfo ;
- vorbis_comment vcomment ;
- sf_count_t lastgranulepos ;
- int doneheaders ;
-} stream_processor ;
-
-typedef struct
-{
- stream_processor *streams ;
- int allocated ;
- int used ;
- int in_headers ;
-} stream_set ;
-
-static stream_set *
-create_stream_set (void)
-{ stream_set *set = calloc (1, sizeof (stream_set)) ;
-
- set->streams = calloc (5, sizeof (stream_processor)) ;
- set->allocated = 5 ;
- set->used = 0 ;
-
- return set ;
-} /* create_stream_set */
-
-static void
-vorbis_end (stream_processor *stream, sf_count_t * len)
-{ *len += stream->lastgranulepos ;
- vorbis_comment_clear (&stream->vcomment) ;
- vorbis_info_clear (&stream->vinfo) ;
-} /* vorbis_end */
-
-static void
-free_stream_set (stream_set *set, sf_count_t * len)
-{ int i ;
-
- for (i = 0 ; i < set->used ; i++)
- { if (!set->streams [i].end)
- vorbis_end (&set->streams [i], len) ;
- ogg_stream_clear (&set->streams [i].ostream) ;
- } ;
-
- free (set->streams) ;
- free (set) ;
-} /* free_stream_set */
-
-static int
-streams_open (stream_set *set)
-{ int i, res = 0 ;
-
- for (i = 0 ; i < set->used ; i++)
- if (!set->streams [i].end)
- res ++ ;
- return res ;
-} /* streams_open */
-
-static stream_processor *
-find_stream_processor (stream_set *set, ogg_page *page)
-{ uint32_t serial = ogg_page_serialno (page) ;
- int i, invalid = 0 ;
-
- stream_processor *stream ;
-
- for (i = 0 ; i < set->used ; i++)
- {
- if (serial == set->streams [i].serial)
- { /* We have a match! */
- stream = & (set->streams [i]) ;
-
- set->in_headers = 0 ;
- /* if we have detected EOS, then this can't occur here. */
- if (stream->end)
- { stream->isillegal = 1 ;
- return stream ;
- }
-
- stream->isnew = 0 ;
- stream->end = ogg_page_eos (page) ;
- stream->serial = serial ;
- return stream ;
- } ;
- } ;
-
- /* If there are streams open, and we've reached the end of the
- ** headers, then we can't be starting a new stream.
- ** XXX: might this sometimes catch ok streams if EOS flag is missing,
- ** but the stream is otherwise ok?
- */
- if (streams_open (set) && !set->in_headers)
- invalid = 1 ;
-
- set->in_headers = 1 ;
-
- if (set->allocated < set->used)
- stream = &set->streams [set->used] ;
- else
- { set->allocated += 5 ;
- set->streams = realloc (set->streams, sizeof (stream_processor) * set->allocated) ;
- stream = &set->streams [set->used] ;
- } ;
-
- set->used++ ;
-
- stream->isnew = 1 ;
- stream->isillegal = invalid ;
-
- {
- int res ;
- ogg_packet packet ;
-
- /* We end up processing the header page twice, but that's ok. */
- ogg_stream_init (&stream->ostream, serial) ;
- ogg_stream_pagein (&stream->ostream, page) ;
- res = ogg_stream_packetout (&stream->ostream, &packet) ;
- if (res <= 0)
- return NULL ;
- else if (packet.bytes >= 7 && memcmp (packet.packet, "\x01vorbis", 7) == 0)
- {
- stream->lastgranulepos = 0 ;
- vorbis_comment_init (&stream->vcomment) ;
- vorbis_info_init (&stream->vinfo) ;
- } ;
-
- res = ogg_stream_packetout (&stream->ostream, &packet) ;
-
- /* re-init, ready for processing */
- ogg_stream_clear (&stream->ostream) ;
- ogg_stream_init (&stream->ostream, serial) ;
- }
-
- stream->end = ogg_page_eos (page) ;
- stream->serial = serial ;
-
- return stream ;
-} /* find_stream_processor */
-
-static int
-vorbis_length_get_next_page (SF_PRIVATE *psf, ogg_sync_state * osync, ogg_page *page)
-{ static const int CHUNK_SIZE = 4500 ;
-
- while (ogg_sync_pageout (osync, page) <= 0)
- { char * buffer = ogg_sync_buffer (osync, CHUNK_SIZE) ;
- int bytes = psf_fread (buffer, 1, 4096, psf) ;
-
- if (bytes <= 0)
- { ogg_sync_wrote (osync, 0) ;
- return 0 ;
- } ;
-
- ogg_sync_wrote (osync, bytes) ;
- } ;
-
- return 1 ;
-} /* vorbis_length_get_next_page */
-
-static sf_count_t
-vorbis_length_aux (SF_PRIVATE * psf)
-{
- ogg_sync_state osync ;
- ogg_page page ;
- sf_count_t len = 0 ;
- stream_set *processors ;
-
- processors = create_stream_set () ;
- if (processors == NULL)
- return 0 ; // out of memory?
-
- ogg_sync_init (&osync) ;
-
- while (vorbis_length_get_next_page (psf, &osync, &page))
- {
- stream_processor *p = find_stream_processor (processors, &page) ;
-
- if (!p)
- { len = 0 ;
- break ;
- } ;
-
- if (p->isillegal && !p->shownillegal)
- {
- p->shownillegal = 1 ;
- /* If it's a new stream, we want to continue processing this page
- ** anyway to suppress additional spurious errors
- */
- if (!p->isnew) continue ;
- } ;
-
- if (!p->isillegal)
- { ogg_packet packet ;
- int header = 0 ;
-
- ogg_stream_pagein (&p->ostream, &page) ;
- if (p->doneheaders < 3)
- header = 1 ;
-
- while (ogg_stream_packetout (&p->ostream, &packet) > 0)
- {
- if (p->doneheaders < 3)
- { if (vorbis_synthesis_headerin (&p->vinfo, &p->vcomment, &packet) < 0)
- continue ;
- p->doneheaders ++ ;
- } ;
- } ;
- if (!header)
- { sf_count_t gp = ogg_page_granulepos (&page) ;
- if (gp > 0) p->lastgranulepos = gp ;
- } ;
- if (p->end)
- { vorbis_end (p, &len) ;
- p->isillegal = 1 ;
- } ;
- } ;
- } ;
-
- ogg_sync_clear (&osync) ;
- free_stream_set (processors, &len) ;
-
- return len ;
-} /* vorbis_length_aux */
-
-static sf_count_t
-vorbis_length (SF_PRIVATE *psf)
-{ sf_count_t length ;
- int error ;
-
- if (psf->sf.seekable == 0)
- return SF_COUNT_MAX ;
-
- psf_fseek (psf, 0, SEEK_SET) ;
- length = vorbis_length_aux (psf) ;
-
- psf_fseek (psf, 12, SEEK_SET) ;
- if ((error = vorbis_read_header (psf, 0)) != 0)
- psf->error = error ;
-
- return length ;
-} /* vorbis_length */
-
-#else /* HAVE_EXTERNAL_LIBS */
-
-int
-ogg_vorbis_open (SF_PRIVATE *psf)
-{
- psf_log_printf (psf, "This version of libsndfile was compiled without Ogg/Vorbis support.\n") ;
- return SFE_UNIMPLEMENTED ;
-} /* ogg_vorbis_open */
-
-#endif