forked from teamnwah/openmw-tes3coop
first FFmpeg / avcodoc commit (does NOT compile)
git-svn-id: https://openmw.svn.sourceforge.net/svnroot/openmw/trunk@24 ea6a568a-9f4f-0410-981a-c910a81bb256actorid
nkorslund
17 years ago
parent
1974b68dcd
commit
a99139cd2f
@ -0,0 +1,40 @@
|
|||||||
|
extern (C):
|
||||||
|
|
||||||
|
// A unique handle that represents an AV file
|
||||||
|
typedef void* AVFile;
|
||||||
|
|
||||||
|
// A unique handle representing an audio stream
|
||||||
|
typedef void* AVAudio;
|
||||||
|
|
||||||
|
// In case we ever decide to implement more codec backends, here's what
|
||||||
|
// these functions need to do...
|
||||||
|
|
||||||
|
// Open the named file, and return a unique handle representing it.
|
||||||
|
// Returns NULL on error
|
||||||
|
AVFile cpp_openAVFile(char *fname);
|
||||||
|
|
||||||
|
// Close the file handle, invalidating all streams taken from it
|
||||||
|
void cpp_closeAVFile(AVFile file);
|
||||||
|
|
||||||
|
// Get a unique handle to an audio stream in the file. The given number
|
||||||
|
// is for files that can contain multiple audio streams (generally you
|
||||||
|
// would pass 0, for the first audio stream)
|
||||||
|
void *cpp_getAVAudioStream(AVFile file, int streamnum);
|
||||||
|
|
||||||
|
// Get audio info representing the current stream. Returns 0 for success
|
||||||
|
// (not likely to fail)
|
||||||
|
int cpp_getAVAudioInfo(AVAudio stream, int *rate, int *channels, int *bits);
|
||||||
|
|
||||||
|
// Decode the next bit of data for the given audio stream. The function
|
||||||
|
// must provide no less than the requested number of bytes, except for
|
||||||
|
// end-of-stream conditions, and is responsible for buffering data. For
|
||||||
|
// files with multiple streams, it must take care to preserve data for
|
||||||
|
// any stream that has had a stream handle returned.
|
||||||
|
// eg. if a file has one video stream and 2 audio streams and the app
|
||||||
|
// gets a handle to the video stream and one audio stream, it must
|
||||||
|
// not destroy video data for subsequent calls to cpp_getAVVideoData if
|
||||||
|
// it has to read over it while decoding the audio stream. The other
|
||||||
|
// audio stream's data, however, may be discarded.
|
||||||
|
// Returns the number of bytes written to the buffer, which will be no
|
||||||
|
// more than the provided length.
|
||||||
|
int cpp_getAVAudioData(AVAudio stream, void *data, int length);
|
||||||
@ -0,0 +1,209 @@
|
|||||||
|
/*
|
||||||
|
OpenMW - The completely unofficial reimplementation of Morrowind
|
||||||
|
Copyright (C) 2008 Nicolay Korslund
|
||||||
|
Email: < korslund@gmail.com >
|
||||||
|
WWW: http://openmw.snaptoad.com/
|
||||||
|
|
||||||
|
This file (cpp_avcodec.cpp) is part of the OpenMW package.
|
||||||
|
|
||||||
|
OpenMW is distributed as free software: you can redistribute it
|
||||||
|
and/or modify it under the terms of the GNU General Public License
|
||||||
|
version 3, as published by the Free Software Foundation.
|
||||||
|
|
||||||
|
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
|
||||||
|
General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
version 3 along with this program. If not, see
|
||||||
|
http://www.gnu.org/licenses/ .
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
extern "C" { // the headers don't do this..
|
||||||
|
#include <ffmpeg/avcodec.h>
|
||||||
|
#include <ffmpeg/avformat.h>
|
||||||
|
}
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
using std::vector;
|
||||||
|
|
||||||
|
struct MyFile {
|
||||||
|
AVFormatContext *FmtCtx;
|
||||||
|
struct MyStream {
|
||||||
|
MyFile *parent;
|
||||||
|
|
||||||
|
AVCodecContext *CodecCtx;
|
||||||
|
int StreamNum;
|
||||||
|
|
||||||
|
vector<uint8_t> Data;
|
||||||
|
vector<char> DecodedData;
|
||||||
|
};
|
||||||
|
vector<MyStream> Streams;
|
||||||
|
};
|
||||||
|
|
||||||
|
// TODO:
|
||||||
|
// extern "C" MyFile::MyStream *cpp_getAVVideoStream(MyFile *file, int streamnum);
|
||||||
|
// extern "C" int cpp_getAVVideoInfo(MyFile::MyStream *stream, float *fps, int *width, int * height);
|
||||||
|
// extern "C" int cpp_getAVVideoData(MyFile::MyStream *stream, char *data, int length);
|
||||||
|
|
||||||
|
extern "C" MyFile *cpp_openAVFile(char *fname)
|
||||||
|
{
|
||||||
|
static bool done = false;
|
||||||
|
if(!done) { av_register_all();
|
||||||
|
av_log_set_level(AV_LOG_ERROR);}
|
||||||
|
done = true;
|
||||||
|
|
||||||
|
MyFile *file = new MyFile;
|
||||||
|
if(av_open_input_file(&file->FmtCtx, fname, NULL, 0, NULL) == 0)
|
||||||
|
{
|
||||||
|
if(av_find_stream_info(file->FmtCtx) >= 0)
|
||||||
|
return file;
|
||||||
|
av_close_input_file(file->FmtCtx);
|
||||||
|
}
|
||||||
|
delete file;
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" void cpp_closeAVFile(MyFile *file)
|
||||||
|
{
|
||||||
|
if(!file) return;
|
||||||
|
|
||||||
|
for(size_t i = 0;i < file->Streams.size();i++)
|
||||||
|
{
|
||||||
|
avcodec_close(file->Streams[i].CodecCtx);
|
||||||
|
file->Streams[i].Data.clear();
|
||||||
|
file->Streams[i].DecodedData.clear();
|
||||||
|
}
|
||||||
|
file->Streams.clear();
|
||||||
|
|
||||||
|
av_close_input_file(file->FmtCtx);
|
||||||
|
delete file;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" MyFile::MyStream *cpp_getAVAudioStream(MyFile *file, int streamnum)
|
||||||
|
{
|
||||||
|
if(!file) return NULL;
|
||||||
|
for(unsigned int i = 0;i < file->FmtCtx->nb_streams;i++)
|
||||||
|
{
|
||||||
|
if(file->FmtCtx->streams[i]->codec->codec_type != CODEC_TYPE_AUDIO)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
if(streamnum == 0)
|
||||||
|
{
|
||||||
|
MyFile::MyStream stream;
|
||||||
|
stream.parent = file;
|
||||||
|
stream.CodecCtx = file->FmtCtx->streams[i]->codec;
|
||||||
|
stream.StreamNum = i;
|
||||||
|
|
||||||
|
AVCodec *codec = avcodec_find_decoder(stream.CodecCtx->codec_id);
|
||||||
|
if(!codec) return NULL;
|
||||||
|
if(avcodec_open(stream.CodecCtx, codec) < 0)
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
file->Streams.push_back(stream);
|
||||||
|
return &file->Streams[file->Streams.size()-1];
|
||||||
|
}
|
||||||
|
streamnum--;
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" int cpp_getAVAudioInfo(MyFile::MyStream *stream,
|
||||||
|
int *rate, int *channels, int *bits)
|
||||||
|
{
|
||||||
|
if(!stream) return 1;
|
||||||
|
|
||||||
|
if(rate) *rate = stream->CodecCtx->sample_rate;
|
||||||
|
if(channels) *channels = stream->CodecCtx->channels;
|
||||||
|
if(bits) *bits = 16;
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int getNextPacket(MyFile *file)
|
||||||
|
{
|
||||||
|
AVPacket packet;
|
||||||
|
while(av_read_frame(file->FmtCtx, &packet) >= 0)
|
||||||
|
{
|
||||||
|
for(vector<MyFile::MyStream>::iterator i = file->Streams.begin();
|
||||||
|
i != file->Streams.end();i++)
|
||||||
|
{
|
||||||
|
if(i->StreamNum == packet.stream_index)
|
||||||
|
{
|
||||||
|
size_t idx = i->Data.size();
|
||||||
|
i->Data.resize(idx + packet.size);
|
||||||
|
memcpy(&i->Data[idx], packet.data, packet.size);
|
||||||
|
av_free_packet(&packet);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
av_free_packet(&packet);
|
||||||
|
}
|
||||||
|
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
extern "C" int cpp_getAVAudioData(MyFile::MyStream *stream, char *data, int length)
|
||||||
|
{
|
||||||
|
if(!stream) return 0;
|
||||||
|
|
||||||
|
int dec = 0;
|
||||||
|
while(dec < length)
|
||||||
|
{
|
||||||
|
if(stream->DecodedData.size() == 0)
|
||||||
|
{
|
||||||
|
while(stream->Data.size() == 0)
|
||||||
|
{
|
||||||
|
if(getNextPacket(stream->parent) != 0)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
int insize = stream->Data.size();
|
||||||
|
if(insize == 0)
|
||||||
|
break;
|
||||||
|
|
||||||
|
// Temporarilly add padding to the input data since some
|
||||||
|
// codecs read in larger chunks and may accidently read
|
||||||
|
// past the end of the allocated buffer
|
||||||
|
stream->Data.resize(insize + FF_INPUT_BUFFER_PADDING_SIZE);
|
||||||
|
stream->DecodedData.resize(AVCODEC_MAX_AUDIO_FRAME_SIZE);
|
||||||
|
|
||||||
|
int16_t *ptr = (int16_t*)&stream->DecodedData[0];
|
||||||
|
int size = stream->DecodedData.size();
|
||||||
|
int len = avcodec_decode_audio2(stream->CodecCtx, ptr, &size,
|
||||||
|
&stream->Data[0], insize);
|
||||||
|
if(len < 0)
|
||||||
|
{
|
||||||
|
stream->Data.resize(insize);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
int datarem = insize-len;
|
||||||
|
if(datarem)
|
||||||
|
memmove(&stream->Data[0], &stream->Data[len], datarem);
|
||||||
|
stream->Data.resize(datarem);
|
||||||
|
|
||||||
|
stream->DecodedData.resize(size);
|
||||||
|
if(stream->DecodedData.size() == 0)
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t rem = length-dec;
|
||||||
|
if(rem > stream->DecodedData.size())
|
||||||
|
rem = stream->DecodedData.size();
|
||||||
|
|
||||||
|
memcpy(data, &stream->DecodedData[0], rem);
|
||||||
|
data += rem;
|
||||||
|
dec += rem;
|
||||||
|
if(rem < stream->DecodedData.size())
|
||||||
|
memmove(&stream->DecodedData[0], &stream->DecodedData[rem],
|
||||||
|
stream->DecodedData.size() - rem);
|
||||||
|
stream->DecodedData.resize(stream->DecodedData.size()-rem);
|
||||||
|
}
|
||||||
|
return dec;
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue