mirror of
https://github.com/TES3MP/openmw-tes3mp.git
synced 2025-01-16 02:19:55 +00:00
344 lines
9.3 KiB
C++
344 lines
9.3 KiB
C++
#ifdef OPENMW_USE_FFMPEG
|
|
|
|
|
|
#include "ffmpeg_decoder.hpp"
|
|
|
|
// auto_ptr
|
|
#include <memory>
|
|
|
|
#include <stdexcept>
|
|
|
|
namespace MWSound
|
|
{
|
|
|
|
static void fail(const std::string &msg)
|
|
{ throw std::runtime_error("FFmpeg exception: "+msg); }
|
|
|
|
|
|
struct FFmpeg_Decoder::MyStream {
|
|
AVCodecContext *mCodecCtx;
|
|
int mStreamIdx;
|
|
|
|
AVPacket mPacket;
|
|
AVFrame *mFrame;
|
|
|
|
int mFrameSize;
|
|
int mFramePos;
|
|
|
|
FFmpeg_Decoder *mParent;
|
|
|
|
bool getAVAudioData();
|
|
size_t readAVAudioData(void *data, size_t length);
|
|
};
|
|
|
|
|
|
int FFmpeg_Decoder::readPacket(void *user_data, uint8_t *buf, int buf_size)
|
|
{
|
|
Ogre::DataStreamPtr stream = static_cast<FFmpeg_Decoder*>(user_data)->mDataStream;
|
|
return stream->read(buf, buf_size);
|
|
}
|
|
|
|
int FFmpeg_Decoder::writePacket(void *user_data, uint8_t *buf, int buf_size)
|
|
{
|
|
Ogre::DataStreamPtr stream = static_cast<FFmpeg_Decoder*>(user_data)->mDataStream;
|
|
return stream->write(buf, buf_size);
|
|
}
|
|
|
|
int64_t FFmpeg_Decoder::seek(void *user_data, int64_t offset, int whence)
|
|
{
|
|
Ogre::DataStreamPtr stream = static_cast<FFmpeg_Decoder*>(user_data)->mDataStream;
|
|
|
|
whence &= ~AVSEEK_FORCE;
|
|
if(whence == AVSEEK_SIZE)
|
|
return stream->size();
|
|
if(whence == SEEK_SET)
|
|
stream->seek(offset);
|
|
else if(whence == SEEK_CUR)
|
|
stream->seek(stream->tell()+offset);
|
|
else if(whence == SEEK_END)
|
|
stream->seek(stream->size()+offset);
|
|
else
|
|
return -1;
|
|
|
|
return stream->tell();
|
|
}
|
|
|
|
|
|
/* Used by getAV*Data to search for more compressed data, and buffer it in the
|
|
* correct stream. It won't buffer data for streams that the app doesn't have a
|
|
* handle for. */
|
|
bool FFmpeg_Decoder::getNextPacket()
|
|
{
|
|
if(!mStream.get())
|
|
return false;
|
|
|
|
while(av_read_frame(mFormatCtx, &mStream->mPacket) >= 0)
|
|
{
|
|
/* Check if the packet belongs to this stream */
|
|
if(mStream->mStreamIdx == mStream->mPacket.stream_index)
|
|
return true;
|
|
|
|
/* Free the packet and look for another */
|
|
av_free_packet(&mStream->mPacket);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool FFmpeg_Decoder::MyStream::getAVAudioData()
|
|
{
|
|
int got_frame, len;
|
|
|
|
if(mCodecCtx->codec_type != AVMEDIA_TYPE_AUDIO)
|
|
return false;
|
|
|
|
do {
|
|
if(mPacket.size == 0 && !mParent->getNextPacket())
|
|
return false;
|
|
|
|
/* Decode some data, and check for errors */
|
|
if((len=avcodec_decode_audio4(mCodecCtx, mFrame, &got_frame, &mPacket)) < 0)
|
|
return false;
|
|
|
|
/* Move the unread data to the front and clear the end bits */
|
|
int remaining = mPacket.size - len;
|
|
if(remaining <= 0)
|
|
av_free_packet(&mPacket);
|
|
else
|
|
{
|
|
memmove(mPacket.data, &mPacket.data[len], remaining);
|
|
av_shrink_packet(&mPacket, remaining);
|
|
}
|
|
} while(got_frame == 0 || mFrame->nb_samples == 0);
|
|
|
|
return true;
|
|
}
|
|
|
|
size_t FFmpeg_Decoder::MyStream::readAVAudioData(void *data, size_t length)
|
|
{
|
|
size_t dec = 0;
|
|
|
|
while(dec < length)
|
|
{
|
|
/* If there's no decoded data, find some */
|
|
if(mFramePos >= mFrameSize)
|
|
{
|
|
if(!getAVAudioData())
|
|
break;
|
|
mFramePos = 0;
|
|
mFrameSize = mFrame->nb_samples * mCodecCtx->channels *
|
|
av_get_bytes_per_sample(mCodecCtx->sample_fmt);
|
|
}
|
|
|
|
/* Get the amount of bytes remaining to be written, and clamp to
|
|
* the amount of decoded data we have */
|
|
size_t rem = std::min<size_t>(length-dec, mFrameSize-mFramePos);
|
|
|
|
/* Copy the data to the app's buffer and increment */
|
|
memcpy(data, mFrame->data[0]+mFramePos, rem);
|
|
data = (char*)data + rem;
|
|
dec += rem;
|
|
mFramePos += rem;
|
|
}
|
|
|
|
/* Return the number of bytes we were able to get */
|
|
return dec;
|
|
}
|
|
|
|
|
|
void FFmpeg_Decoder::open(const std::string &fname)
|
|
{
|
|
close();
|
|
mDataStream = mResourceMgr.openResource(fname);
|
|
|
|
if((mFormatCtx=avformat_alloc_context()) == NULL)
|
|
fail("Failed to allocate context");
|
|
|
|
mFormatCtx->pb = avio_alloc_context(NULL, 0, 0, this, readPacket, writePacket, seek);
|
|
if(!mFormatCtx->pb || avformat_open_input(&mFormatCtx, fname.c_str(), NULL, NULL) != 0)
|
|
{
|
|
avformat_free_context(mFormatCtx);
|
|
mFormatCtx = NULL;
|
|
fail("Failed to allocate input stream");
|
|
}
|
|
|
|
try
|
|
{
|
|
if(avformat_find_stream_info(mFormatCtx, NULL) < 0)
|
|
fail("Failed to find stream info in "+fname);
|
|
|
|
int audio_idx = -1;
|
|
for(size_t j = 0;j < mFormatCtx->nb_streams;j++)
|
|
{
|
|
if(mFormatCtx->streams[j]->codec->codec_type == AVMEDIA_TYPE_AUDIO)
|
|
{
|
|
audio_idx = j;
|
|
break;
|
|
}
|
|
}
|
|
if(audio_idx == -1)
|
|
fail("No audio streams in "+fname);
|
|
|
|
std::auto_ptr<MyStream> stream(new MyStream);
|
|
stream->mCodecCtx = mFormatCtx->streams[audio_idx]->codec;
|
|
stream->mStreamIdx = audio_idx;
|
|
memset(&stream->mPacket, 0, sizeof(stream->mPacket));
|
|
|
|
AVCodec *codec = avcodec_find_decoder(stream->mCodecCtx->codec_id);
|
|
if(!codec)
|
|
{
|
|
std::stringstream ss("No codec found for id ");
|
|
ss << stream->mCodecCtx->codec_id;
|
|
fail(ss.str());
|
|
}
|
|
if(avcodec_open2(stream->mCodecCtx, codec, NULL) < 0)
|
|
fail("Failed to open audio codec " + std::string(codec->long_name));
|
|
|
|
stream->mFrame = avcodec_alloc_frame();
|
|
|
|
stream->mParent = this;
|
|
mStream = stream;
|
|
}
|
|
catch(std::exception &e)
|
|
{
|
|
avformat_close_input(&mFormatCtx);
|
|
mFormatCtx = NULL;
|
|
throw;
|
|
}
|
|
}
|
|
|
|
void FFmpeg_Decoder::close()
|
|
{
|
|
if(mStream.get())
|
|
{
|
|
av_free_packet(&mStream->mPacket);
|
|
avcodec_close(mStream->mCodecCtx);
|
|
av_free(mStream->mFrame);
|
|
}
|
|
mStream.reset();
|
|
|
|
if(mFormatCtx)
|
|
{
|
|
AVIOContext* context = mFormatCtx->pb;
|
|
avformat_close_input(&mFormatCtx);
|
|
av_free(context);
|
|
}
|
|
mFormatCtx = NULL;
|
|
|
|
mDataStream.setNull();
|
|
}
|
|
|
|
std::string FFmpeg_Decoder::getName()
|
|
{
|
|
return mFormatCtx->filename;
|
|
}
|
|
|
|
void FFmpeg_Decoder::getInfo(int *samplerate, ChannelConfig *chans, SampleType *type)
|
|
{
|
|
if(!mStream.get())
|
|
fail("No audio stream info");
|
|
|
|
if(mStream->mCodecCtx->sample_fmt == AV_SAMPLE_FMT_U8)
|
|
*type = SampleType_UInt8;
|
|
else if(mStream->mCodecCtx->sample_fmt == AV_SAMPLE_FMT_S16)
|
|
*type = SampleType_Int16;
|
|
else
|
|
fail(std::string("Unsupported sample format: ")+
|
|
av_get_sample_fmt_name(mStream->mCodecCtx->sample_fmt));
|
|
|
|
if(mStream->mCodecCtx->channel_layout == AV_CH_LAYOUT_MONO)
|
|
*chans = ChannelConfig_Mono;
|
|
else if(mStream->mCodecCtx->channel_layout == AV_CH_LAYOUT_STEREO)
|
|
*chans = ChannelConfig_Stereo;
|
|
else if(mStream->mCodecCtx->channel_layout == AV_CH_LAYOUT_QUAD)
|
|
*chans = ChannelConfig_Quad;
|
|
else if(mStream->mCodecCtx->channel_layout == AV_CH_LAYOUT_5POINT1)
|
|
*chans = ChannelConfig_5point1;
|
|
else if(mStream->mCodecCtx->channel_layout == AV_CH_LAYOUT_7POINT1)
|
|
*chans = ChannelConfig_7point1;
|
|
else if(mStream->mCodecCtx->channel_layout == 0)
|
|
{
|
|
/* Unknown channel layout. Try to guess. */
|
|
if(mStream->mCodecCtx->channels == 1)
|
|
*chans = ChannelConfig_Mono;
|
|
else if(mStream->mCodecCtx->channels == 2)
|
|
*chans = ChannelConfig_Stereo;
|
|
else
|
|
{
|
|
std::stringstream sstr("Unsupported raw channel count: ");
|
|
sstr << mStream->mCodecCtx->channels;
|
|
fail(sstr.str());
|
|
}
|
|
}
|
|
else
|
|
{
|
|
char str[1024];
|
|
av_get_channel_layout_string(str, sizeof(str), mStream->mCodecCtx->channels,
|
|
mStream->mCodecCtx->channel_layout);
|
|
fail(std::string("Unsupported channel layout: ")+str);
|
|
}
|
|
|
|
*samplerate = mStream->mCodecCtx->sample_rate;
|
|
}
|
|
|
|
size_t FFmpeg_Decoder::read(char *buffer, size_t bytes)
|
|
{
|
|
if(!mStream.get())
|
|
fail("No audio stream");
|
|
|
|
size_t got = mStream->readAVAudioData(buffer, bytes);
|
|
mSamplesRead += got / mStream->mCodecCtx->channels /
|
|
av_get_bytes_per_sample(mStream->mCodecCtx->sample_fmt);
|
|
return got;
|
|
}
|
|
|
|
void FFmpeg_Decoder::readAll(std::vector<char> &output)
|
|
{
|
|
if(!mStream.get())
|
|
fail("No audio stream");
|
|
|
|
while(mStream->getAVAudioData())
|
|
{
|
|
size_t got = mStream->mFrame->nb_samples * mStream->mCodecCtx->channels *
|
|
av_get_bytes_per_sample(mStream->mCodecCtx->sample_fmt);
|
|
const char *inbuf = reinterpret_cast<char*>(mStream->mFrame->data[0]);
|
|
output.insert(output.end(), inbuf, inbuf+got);
|
|
mSamplesRead += mStream->mFrame->nb_samples;
|
|
}
|
|
}
|
|
|
|
void FFmpeg_Decoder::rewind()
|
|
{
|
|
av_seek_frame(mFormatCtx, -1, 0, 0);
|
|
if(mStream.get())
|
|
av_free_packet(&mStream->mPacket);
|
|
mSamplesRead = 0;
|
|
}
|
|
|
|
size_t FFmpeg_Decoder::getSampleOffset()
|
|
{
|
|
return mSamplesRead;
|
|
}
|
|
|
|
FFmpeg_Decoder::FFmpeg_Decoder() : mFormatCtx(NULL), mSamplesRead(0)
|
|
{
|
|
static bool done_init = false;
|
|
|
|
/* We need to make sure ffmpeg is initialized. Optionally silence warning
|
|
* output from the lib */
|
|
if(!done_init)
|
|
{
|
|
av_register_all();
|
|
av_log_set_level(AV_LOG_ERROR);
|
|
done_init = true;
|
|
}
|
|
}
|
|
|
|
FFmpeg_Decoder::~FFmpeg_Decoder()
|
|
{
|
|
close();
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|