Added abandoned KDE3 version of K3B

git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/k3b@1084400 283d02a7-25f6-0310-bc7c-ecb5cbfe19da

1 files changed, 375 insertions, 0 deletions

diff --git a/plugins/decoder/ffmpeg/k3bffmpegwrapper.cpp b/plugins/decoder/ffmpeg/k3bffmpegwrapper.cpp
new file mode 100644
index 0000000..514fd67
--- /dev/null
+++ b/plugins/decoder/ffmpeg/k3bffmpegwrapper.cpp
@@ -0,0 +1,375 @@
+/* 
+ *
+ * $Id: k3bffmpegwrapper.cpp 641819 2007-03-12 17:29:23Z trueg $
+ * Copyright (C) 2004-2007 Sebastian Trueg <[email protected]>
+ *
+ * This file is part of the K3b project.
+ * Copyright (C) 1998-2007 Sebastian Trueg <[email protected]>
+ *
+ * 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 <config.h>
+
+#include "k3bffmpegwrapper.h"
+
+extern "C" {
+#include <libavcodec/avcodec.h>
+#include <libavformat/avformat.h>
+}
+
+#include <string.h>
+
+#include <klocale.h>
+
+
+#if LIBAVFORMAT_BUILD < 4629
+#define FFMPEG_BUILD_PRE_4629
+#endif
+
+
+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;
+  }
+
+  // analyze 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
+#ifdef FFMPEG_BUILD_PRE_4629
+  AVCodecContext* codecContext =  &d->formatContext->streams[0]->codec;
+#else
+  AVCodecContext* codecContext =  d->formatContext->streams[0]->codec;
+#endif
+  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->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 ) {
+#ifdef FFMPEG_BUILD_PRE_4629
+    avcodec_close( &d->formatContext->streams[0]->codec );
+#else
+    avcodec_close( d->formatContext->streams[0]->codec );
+#endif
+    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
+{
+#ifdef FFMPEG_BUILD_PRE_4629
+  return d->formatContext->streams[0]->codec.sample_rate;
+#else
+  return d->formatContext->streams[0]->codec->sample_rate;
+#endif
+}
+
+
+int K3bFFMpegFile::channels() const
+{
+#ifdef FFMPEG_BUILD_PRE_4629
+  return d->formatContext->streams[0]->codec.channels;
+#else
+  return d->formatContext->streams[0]->codec->channels;
+#endif
+}
+
+
+int K3bFFMpegFile::type() const
+{
+#ifdef FFMPEG_BUILD_PRE_4629
+  return d->formatContext->streams[0]->codec.codec_id;
+#else
+  return d->formatContext->streams[0]->codec->codec_id;
+#endif
+}
+
+
+QString K3bFFMpegFile::typeComment() const
+{
+  switch( type() ) {
+  case CODEC_ID_WMAV1:
+    return i18n("Windows Media v1");
+  case CODEC_ID_WMAV2:
+    return i18n("Windows Media v2");
+  case CODEC_ID_MP3:
+    return i18n("MPEG 1 Layer III");
+  case CODEC_ID_AAC:
+    return i18n("Advanced Audio Coding (AAC)");
+  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;
+
+#ifdef FFMPEG_BUILD_PRE_4629
+    int len = avcodec_decode_audio2( &d->formatContext->streams[0]->codec,
+#else
+    int len = avcodec_decode_audio2( d->formatContext->streams[0]->codec,
+#endif
+				    (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?
+#if LIBAVFORMAT_BUILD >= 4619
+  return ( av_seek_frame( d->formatContext, -1, timestamp + d->formatContext->start_time, 0 ) >= 0 );
+#else
+  return ( av_seek_frame( d->formatContext, -1, timestamp + d->formatContext->start_time ) >= 0 );
+#endif
+}
+
+
+
+
+
+
+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() ) {
+#ifndef K3B_FFMPEG_ALL_CODECS
+    //
+    // only allow tested formats. ffmpeg seems not to be too reliable with every format.
+    // mp3 being one of them sadly. Most importantly: allow the libsndfile decoder to do
+    // its thing.
+    //
+    if( file->type() == CODEC_ID_WMAV1 ||
+	file->type() == CODEC_ID_WMAV2 ||
+	file->type() == CODEC_ID_AAC )
+#endif
+      return file;
+  }
+
+  delete file;
+  return 0;
+}