Torque3D/Engine/source/sfx/media/sfxVorbisStream.cpp
2020-05-11 15:24:49 -05:00

281 lines
7.4 KiB
C++

//-----------------------------------------------------------------------------
// Copyright (c) 2012 GarageGames, LLC
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.
//-----------------------------------------------------------------------------
#ifdef TORQUE_OGGVORBIS
#include "sfx/media/sfxVorbisStream.h"
#include "core/stream/stream.h"
#include "console/console.h"
SFXVorbisStream* SFXVorbisStream::create( Stream *stream )
{
SFXVorbisStream *sfxStream = new SFXVorbisStream();
if ( sfxStream->open( stream, true ) )
return sfxStream;
delete sfxStream;
return NULL;
}
SFXVorbisStream::SFXVorbisStream()
: mVF( NULL ),
mBitstream(-1),
mBytesRead( 0 )
{
}
SFXVorbisStream::SFXVorbisStream( const SFXVorbisStream& cloneFrom )
: Parent( cloneFrom )
{
mVF = NULL;
mBitstream = -1;
mBytesRead = 0;
if( !mStream->hasCapability( Stream::StreamPosition ) )
{
Con::errorf( "SFXVorbisStream::SFXVorbisStream() - Source stream does not allow seeking" );
return;
}
mStream->setPosition( 0 );
if( !_readHeader() )
{
Con::errorf( "SFXVorbisStream::SFXVorbisStream() - Opening Vorbis stream failed" );
return;
}
ov_pcm_seek( mVF, ov_pcm_tell( cloneFrom.mVF ) );
mBitstream = cloneFrom.mBitstream;
mBytesRead = cloneFrom.mBytesRead;
}
SFXVorbisStream::~SFXVorbisStream()
{
// We must call close from our own destructor
// and not the base class... as it causes a
// pure virtual runtime assertion.
close();
}
size_t SFXVorbisStream::_read_func( void *ptr, size_t size, size_t nmemb, void *datasource )
{
Stream *stream = reinterpret_cast<Stream*>( datasource );
// Stream::read() returns true if any data was
// read, so we must track the read bytes ourselves.
U32 startByte = stream->getPosition();
stream->read( size * nmemb, ptr );
U32 endByte = stream->getPosition();
// How many did we actually read?
U32 readBytes = ( endByte - startByte );
U32 readItems = readBytes / size;
return readItems;
}
S32 SFXVorbisStream::_seek_func( void *datasource, ogg_int64_t offset, S32 whence )
{
Stream *stream = reinterpret_cast<Stream*>( datasource );
U32 newPos = 0;
if ( whence == SEEK_CUR )
newPos = stream->getPosition() + (U32)offset;
else if ( whence == SEEK_END )
newPos = stream->getStreamSize() - (U32)offset;
else
newPos = (U32)offset;
return stream->setPosition( newPos ) ? 0 : -1;
}
long SFXVorbisStream::_tell_func( void *datasource )
{
Stream *stream = reinterpret_cast<Stream*>( datasource );
return stream->getPosition();
}
bool SFXVorbisStream::_openVorbis()
{
mVF = new OggVorbis_File;
dMemset( mVF, 0, sizeof( OggVorbis_File ) );
const bool canSeek = mStream->hasCapability( Stream::StreamPosition );
ov_callbacks cb;
cb.read_func = _read_func;
cb.seek_func = canSeek ? _seek_func : NULL;
cb.close_func = NULL;
cb.tell_func = canSeek ? _tell_func : NULL;
// Open it.
S32 ovResult = ov_open_callbacks( mStream, mVF, NULL, 0, cb );
if( ovResult != 0 )
return false;
return true;
}
bool SFXVorbisStream::_readHeader()
{
if( !_openVorbis() )
return false;
// Fill in the format info.
const vorbis_info *vi = getInfo();
mFormat.set( vi->channels, vi->channels * 16, vi->rate );
// Set the sample count.
mSamples = getPcmTotal();
// Reset the bitstream.
mBitstream = 0;
return true;
}
void SFXVorbisStream::_close()
{
if ( !mVF )
return;
ov_clear( mVF );
delete mVF;
mVF = NULL;
mBitstream = -1;
}
const vorbis_info* SFXVorbisStream::getInfo( S32 link )
{
AssertFatal( mVF, "SFXVorbisStream::getInfo() - Stream is closed!" );
return ov_info( mVF, link );
}
const vorbis_comment* SFXVorbisStream::getComment( S32 link )
{
AssertFatal( mVF, "SFXVorbisStream::getComment() - Stream is closed!" );
return ov_comment( mVF, link );
}
U64 SFXVorbisStream::getPcmTotal( S32 link )
{
AssertFatal( mVF, "SFXVorbisStream::getInfo() - Stream is closed!" );
return ov_pcm_total( mVF, link );
}
S32 SFXVorbisStream::read( U8 *buffer,
U32 length,
S32 *bitstream )
{
AssertFatal( mVF, "SFXVorbisStream::read() - Stream is closed!" );
mBitstream = *bitstream;
#ifdef TORQUE_BIG_ENDIAN
static const S32 isBigEndian = 1;
#else
static const S32 isBigEndian = 0;
#endif
// Vorbis doesn't seem to like reading
// requests longer than this.
const U32 MAXREAD = 4096;
S64 bytesRead = 0;
U32 offset = 0;
U32 bytesToRead = 0;
// Since it only returns the result of one packet
// per call, you generally have to loop to read it all.
while( offset < length )
{
if ( ( length - offset ) < MAXREAD )
bytesToRead = length - offset;
else
bytesToRead = MAXREAD;
bytesRead = ov_read( mVF, (char*)buffer, bytesToRead, isBigEndian, 2, 1, bitstream );
if( bytesRead == 0 ) // EOF
return offset;
else if( bytesRead < 0 )
{
// We got an error... return the result.
return bytesRead;
}
offset += bytesRead;
buffer += bytesRead;
mBytesRead += bytesRead;
}
// Return the total data read.
return offset;
}
bool SFXVorbisStream::isEOS() const
{
return ( Parent::isEOS() || ( mStream && ov_pcm_tell( mVF ) == mSamples ) );
}
void SFXVorbisStream::reset()
{
AssertFatal( mVF, "SFXVorbisStream::reset() - Stream is closed!" );
// There's a bug in libvorbis 1.2.0 that will cause the
// ov_pcm_seek* functions to go into an infinite loop on
// some files (apparently if they contain Theora streams).
// Avoiding to seek when not necessary seems to do the trick
// but if it deadlocks here, that's why.
if( mBytesRead > 0 )
{
ov_pcm_seek_page( mVF, 0 );
mBitstream = 0;
mBytesRead = 0;
}
}
U32 SFXVorbisStream::getPosition() const
{
AssertFatal( mVF, "SFXVorbisStream::getPosition() - Stream is closed!" );
return U32( ov_pcm_tell( mVF ) ) * mFormat.getBytesPerSample();
}
void SFXVorbisStream::setPosition( U32 offset )
{
AssertFatal( mVF, "SFXVorbisStream::setPosition() - Stream is closed!" );
ov_pcm_seek( mVF, offset / mFormat.getBytesPerSample() );
}
U32 SFXVorbisStream::read( U8 *buffer, U32 length )
{
S32 result = read( buffer, length, &mBitstream );
if ( result < 0 )
return 0;
return result;
}
#endif // TORQUE_OGGVORBIS