320 changes: 320 additions & 0 deletions plugins/decoder/ffmpeg/k3bffmpegwrapper.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,320 @@
/*
*
* $Id$
* Copyright (C) 2004 Sebastian Trueg <trueg@k3b.org>
*
* This file is part of the K3b project.
* Copyright (C) 1998-2004 Sebastian Trueg <trueg@k3b.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
* See the file "COPYING" for the exact licensing terms.
*/

#include "k3bffmpegwrapper.h"

#include <ffmpeg/avcodec.h>
#include <ffmpeg/avformat.h>

#include <string.h>

#include <klocale.h>



K3bFFMpegWrapper* K3bFFMpegWrapper::s_instance = 0;


class K3bFFMpegFile::Private
{
public:
AVFormatContext* formatContext;
AVCodec* codec;

K3b::Msf length;

// for decoding
char outputBuffer[AVCODEC_MAX_AUDIO_FRAME_SIZE];
char* outputBufferPos;
int outputBufferSize;
AVPacket packet;
Q_UINT8* packetData;
int packetSize;
};


K3bFFMpegFile::K3bFFMpegFile( const QString& filename )
: m_filename(filename)
{
d = new Private;
d->formatContext = 0;
d->codec = 0;
}


K3bFFMpegFile::~K3bFFMpegFile()
{
close();
delete d;
}


bool K3bFFMpegFile::open()
{
close();

// open the file
int err = av_open_input_file( &d->formatContext, m_filename.local8Bit(), 0, 0, 0 );
if( err < 0 ) {
kdDebug() << "(K3bFFMpegFile) unable to open " << m_filename << " with error " << err << endl;
return false;
}

// analyse the streams
av_find_stream_info( d->formatContext );

// we only handle files containing one audio stream
if( d->formatContext->nb_streams != 1 ) {
kdDebug() << "(K3bFFMpegFile) more than one stream in " << m_filename << endl;
return false;
}

// urgh... ugly
AVCodecContext* codecContext = &d->formatContext->streams[0]->codec;
if( codecContext->codec_type != CODEC_TYPE_AUDIO ) {
kdDebug() << "(K3bFFMpegFile) not a simple audio stream: " << m_filename << endl;
return false;
}

// get the codec
d->codec = avcodec_find_decoder(codecContext->codec_id);
if( !d->codec ) {
kdDebug() << "(K3bFFMpegFile) no codec found for " << m_filename << endl;
return false;
}

// open the codec on our context
kdDebug() << "(K3bFFMpegFile) found codec for " << m_filename << endl;
if( avcodec_open( codecContext, d->codec ) < 0 ) {
kdDebug() << "(K3bFFMpegDecoderFactory) could not open codec." << endl;
return false;
}

// determine the length of the stream
d->length = K3b::Msf::fromSeconds( (double)d->formatContext->streams[0]->duration / (double)AV_TIME_BASE );

if( d->length == 0 ) {
kdDebug() << "(K3bFFMpegDecoderFactory) invalid length." << endl;
return false;
}

// dump some debugging info
dump_format( d->formatContext, 0, m_filename.local8Bit(), 0 );

return true;
}


void K3bFFMpegFile::close()
{
d->outputBufferSize = 0;
d->packetSize = 0;
d->packetData = 0;

if( d->codec ) {
avcodec_close( &d->formatContext->streams[0]->codec );
d->codec = 0;
}

if( d->formatContext ) {
av_close_input_file( d->formatContext );
d->formatContext = 0;
}
}


K3b::Msf K3bFFMpegFile::length() const
{
return d->length;
}


int K3bFFMpegFile::sampleRate() const
{
return d->formatContext->streams[0]->codec.sample_rate;
}


int K3bFFMpegFile::channels() const
{
return d->formatContext->streams[0]->codec.channels;
}


QString K3bFFMpegFile::type() const
{
switch( d->formatContext->streams[0]->codec.codec_id ) {
case CODEC_ID_WMAV1:
return i18n("Windows Media v1");
case CODEC_ID_WMAV2:
return i18n("Windows Media v2");
case CODEC_ID_MP3LAME:
return i18n("MPEG 1 Layer III");
default:
return QString::fromLocal8Bit( d->codec->name );
}
}


QString K3bFFMpegFile::title() const
{
// FIXME: is this UTF8 or something??
if( d->formatContext->title[0] != '\0' )
return QString::fromLocal8Bit( d->formatContext->title );
else
return QString::null;
}


QString K3bFFMpegFile::author() const
{
// FIXME: is this UTF8 or something??
if( d->formatContext->author[0] != '\0' )
return QString::fromLocal8Bit( d->formatContext->author );
else
return QString::null;
}


QString K3bFFMpegFile::comment() const
{
// FIXME: is this UTF8 or something??
if( d->formatContext->comment[0] != '\0' )
return QString::fromLocal8Bit( d->formatContext->comment );
else
return QString::null;
}


int K3bFFMpegFile::read( char* buf, int bufLen )
{
if( fillOutputBuffer() > 0 ) {
int len = QMIN(bufLen, d->outputBufferSize);
::memcpy( buf, d->outputBufferPos, len );

// TODO: only swap if needed
for( int i = 0; i < len-1; i+=2 ) {
char a = buf[i];
buf[i] = buf[i+1];
buf[i+1] = a;
}

d->outputBufferPos += len;
d->outputBufferSize -= len;
return len;
}
else
return 0;
}


// fill d->packetData with data to decode
int K3bFFMpegFile::readPacket()
{
if( d->packetSize <= 0 ) {
av_init_packet( &d->packet );

if( av_read_frame( d->formatContext, &d->packet ) < 0 ) {
return 0;
}

d->packetSize = d->packet.size;
d->packetData = d->packet.data;
}

return d->packetSize;
}


// decode data in d->packetData and fill d->outputBuffer
int K3bFFMpegFile::fillOutputBuffer()
{
// decode if the output buffer is empty
if( d->outputBufferSize <= 0 ) {

// make sure we have data to decode
if( readPacket() == 0 ) {
return 0;
}

d->outputBufferPos = d->outputBuffer;

int len = avcodec_decode_audio( &d->formatContext->streams[0]->codec,
(short*)d->outputBuffer, &d->outputBufferSize,
d->packetData, d->packetSize );

d->packetSize -= len;
d->packetData += len;

if( d->packetSize <= 0 )
av_free_packet( &d->packet );
}

// if it is still empty try again
if( d->outputBufferSize <= 0 )
return fillOutputBuffer();
else
return d->outputBufferSize;
}


bool K3bFFMpegFile::seek( const K3b::Msf& msf )
{
d->outputBufferSize = 0;
d->packetSize = 0;

double seconds = (double)msf.totalFrames()/75.0;
Q_UINT64 timestamp = (Q_UINT64)(seconds * (double)AV_TIME_BASE);

// FIXME: do we really need the start_time and why?
return ( av_seek_frame( d->formatContext, -1, timestamp + d->formatContext->start_time ) >= 0 );
}






K3bFFMpegWrapper::K3bFFMpegWrapper()
{
av_register_all();
}


K3bFFMpegWrapper::~K3bFFMpegWrapper()
{
s_instance = 0;
}


K3bFFMpegWrapper* K3bFFMpegWrapper::instance()
{
if( !s_instance ) {
s_instance = new K3bFFMpegWrapper();
}

return s_instance;
}


K3bFFMpegFile* K3bFFMpegWrapper::open( const QString& filename ) const
{
K3bFFMpegFile* file = new K3bFFMpegFile( filename );
if( file->open() )
return file;

delete file;
return 0;
}
81 changes: 81 additions & 0 deletions plugins/decoder/ffmpeg/k3bffmpegwrapper.h
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
/*
*
* $Id$
* Copyright (C) 2004 Sebastian Trueg <trueg@k3b.org>
*
* This file is part of the K3b project.
* Copyright (C) 1998-2004 Sebastian Trueg <trueg@k3b.org>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
* See the file "COPYING" for the exact licensing terms.
*/

#ifndef _K3B_FFMPEG_WRAPPER_H_
#define _K3B_FFMPEG_WRAPPER_H_

#include <k3bmsf.h>



/**
* Create with K3bFFMpegWrapper::open
*/
class K3bFFMpegFile
{
friend class K3bFFMpegWrapper;

public:
~K3bFFMpegFile();

const QString& filename() const { return m_filename; }

bool open();
void close();

K3b::Msf length() const;
int sampleRate() const;
int channels() const;

QString type() const;

QString title() const;
QString author() const;
QString comment() const;

int read( char* buf, int bufLen );
bool seek( const K3b::Msf& );

private:
K3bFFMpegFile( const QString& filename );
int readPacket();
int fillOutputBuffer();

QString m_filename;

class Private;
Private* d;
};


class K3bFFMpegWrapper
{
public:
~K3bFFMpegWrapper();

/**
* returns 0 on failure.
*/
K3bFFMpegFile* open( const QString& filename ) const;

static K3bFFMpegWrapper* instance();

private:
K3bFFMpegWrapper();

static K3bFFMpegWrapper* s_instance;
};

#endif
9 changes: 9 additions & 0 deletions plugins/decoder/ffmpeg/k3bxffmpegdecoder.plugin
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
[K3b Plugin]
Lib=libk3bffmpegdecoder
Group=AudioDecoder
Name=K3b FFMpeg Decoder
Author=Sebastian Trueg
Email=trueg@k3b.org
Version=0.9
Comment=Decoding module to decode wma files
License=GPL
2 changes: 1 addition & 1 deletion plugins/decoder/skeleton.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ K3b<name>Decoder::~K3b<name>Decoder()
}


QString K3b<name>Decoder::fileType() const;
QString K3b<name>Decoder::fileType() const
{
// PUT YOUR CODE HERE
}
Expand Down
2 changes: 1 addition & 1 deletion plugins/decoder/skeleton.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ class K3b<name>DecoderFactory : public K3bAudioDecoderFactory

bool canDecode( const KURL& filename );

int pluginSystemVersion() const { return 1; }
int pluginSystemVersion() const { return 2; }

K3bPlugin* createPluginObject( QObject* parent = 0,
const char* name = 0,
Expand Down