lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #ifdef HAVE_OGGZ
lbajardsilogic@0: #ifdef HAVE_FISHSOUND
lbajardsilogic@0: 
lbajardsilogic@0: #include "OggVorbisFileReader.h"
lbajardsilogic@0: #include "base/Profiler.h"
lbajardsilogic@0: #include "system/System.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <sys/types.h>
lbajardsilogic@0: #include <sys/stat.h>
lbajardsilogic@0: //#include <sys/mman.h>
lbajardsilogic@0: #include <fcntl.h>
lbajardsilogic@0: #include <cmath>
lbajardsilogic@0: 
lbajardsilogic@0: #include <QApplication>
lbajardsilogic@0: #include <QFileInfo>
lbajardsilogic@0: #include <QProgressDialog>
lbajardsilogic@0: 
lbajardsilogic@0: static int instances = 0;
lbajardsilogic@0: 
lbajardsilogic@0: OggVorbisFileReader::OggVorbisFileReader(QString path, bool showProgress,
lbajardsilogic@0:                                          CacheMode mode) :
lbajardsilogic@0:     CodedAudioFileReader(mode),
lbajardsilogic@0:     m_path(path),
lbajardsilogic@0:     m_progress(0),
lbajardsilogic@0:     m_fileSize(0),
lbajardsilogic@0:     m_bytesRead(0),
lbajardsilogic@0:     m_cancelled(false)
lbajardsilogic@0: {
lbajardsilogic@0:     m_frameCount = 0;
lbajardsilogic@0:     m_channelCount = 0;
lbajardsilogic@0:     m_sampleRate = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     std::cerr << "OggVorbisFileReader::OggVorbisFileReader(" << path.toLocal8Bit().data() << "): now have " << (++instances) << " instances" << std::endl;
lbajardsilogic@0: 
lbajardsilogic@0:     Profiler profiler("OggVorbisFileReader::OggVorbisFileReader", true);
lbajardsilogic@0: 
lbajardsilogic@0:     QFileInfo info(path);
lbajardsilogic@0:     m_fileSize = info.size();
lbajardsilogic@0: 
lbajardsilogic@0:     OGGZ *oggz;
lbajardsilogic@0:     if (!(oggz = oggz_open(path.toLocal8Bit().data(), OGGZ_READ))) {
lbajardsilogic@0: 	m_error = QString("File %1 is not an OGG file.").arg(path);
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     FishSoundInfo fsinfo;
lbajardsilogic@0:     m_fishSound = fish_sound_new(FISH_SOUND_DECODE, &fsinfo);
lbajardsilogic@0: 
lbajardsilogic@0:     fish_sound_set_decoded_callback(m_fishSound, acceptFrames, this);
lbajardsilogic@0:     oggz_set_read_callback(oggz, -1, readPacket, this);
lbajardsilogic@0: 
lbajardsilogic@0:     if (showProgress) {
lbajardsilogic@0: 	m_progress = new QProgressDialog
lbajardsilogic@0: 	    (QObject::tr("Decoding %1...").arg(QFileInfo(path).fileName()),
lbajardsilogic@0: 	     QObject::tr("Stop"), 0, 100);
lbajardsilogic@0: 	m_progress->hide();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     while (oggz_read(oggz, 1024) > 0);
lbajardsilogic@0: 
lbajardsilogic@0:     fish_sound_delete(m_fishSound);
lbajardsilogic@0:     m_fishSound = 0;
lbajardsilogic@0:     oggz_close(oggz);
lbajardsilogic@0: 
lbajardsilogic@0:     if (isDecodeCacheInitialised()) finishDecodeCache();
lbajardsilogic@0: 
lbajardsilogic@0:     if (showProgress) {
lbajardsilogic@0: 	delete m_progress;
lbajardsilogic@0: 	m_progress = 0;
lbajardsilogic@0:     }
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: OggVorbisFileReader::~OggVorbisFileReader()
lbajardsilogic@0: {
lbajardsilogic@0:     std::cerr << "OggVorbisFileReader::~OggVorbisFileReader(" << m_path.toLocal8Bit().data() << "): now have " << (--instances) << " instances" << std::endl;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: OggVorbisFileReader::readPacket(OGGZ *, ogg_packet *packet, long, void *data)
lbajardsilogic@0: {
lbajardsilogic@0:     OggVorbisFileReader *reader = (OggVorbisFileReader *)data;
lbajardsilogic@0:     FishSound *fs = reader->m_fishSound;
lbajardsilogic@0: 
lbajardsilogic@0:     fish_sound_prepare_truncation(fs, packet->granulepos, packet->e_o_s);
lbajardsilogic@0:     fish_sound_decode(fs, packet->packet, packet->bytes);
lbajardsilogic@0: 
lbajardsilogic@0:     reader->m_bytesRead += packet->bytes;
lbajardsilogic@0:     
lbajardsilogic@0:     if (reader->m_fileSize > 0 && reader->m_progress) {
lbajardsilogic@0: 	// The number of bytes read by this function is smaller than
lbajardsilogic@0: 	// the file size because of the packet headers
lbajardsilogic@0: 	int progress = lrint(double(reader->m_bytesRead) * 114 /
lbajardsilogic@0: 			     double(reader->m_fileSize));
lbajardsilogic@0: 	if (progress > 99) progress = 99;
lbajardsilogic@0: 	if (progress > reader->m_progress->value()) {
lbajardsilogic@0: 	    reader->m_progress->setValue(progress);
lbajardsilogic@0: 	    reader->m_progress->show();
lbajardsilogic@0: 	    reader->m_progress->raise();
lbajardsilogic@0: 	    qApp->processEvents();
lbajardsilogic@0: 	    if (reader->m_progress->wasCanceled()) {
lbajardsilogic@0: 		reader->m_cancelled = true;
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0:     } 
lbajardsilogic@0: 
lbajardsilogic@0:     if (reader->m_cancelled) return 1;
lbajardsilogic@0:     return 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: int
lbajardsilogic@0: OggVorbisFileReader::acceptFrames(FishSound *fs, float **frames, long nframes,
lbajardsilogic@0: 				  void *data)
lbajardsilogic@0: {
lbajardsilogic@0:     OggVorbisFileReader *reader = (OggVorbisFileReader *)data;
lbajardsilogic@0: 
lbajardsilogic@0:     if (reader->m_channelCount == 0) {
lbajardsilogic@0: 	FishSoundInfo fsinfo;
lbajardsilogic@0: 	fish_sound_command(fs, FISH_SOUND_GET_INFO,
lbajardsilogic@0: 			   &fsinfo, sizeof(FishSoundInfo));
lbajardsilogic@0: 	reader->m_channelCount = fsinfo.channels;
lbajardsilogic@0: 	reader->m_sampleRate = fsinfo.samplerate;
lbajardsilogic@0:         reader->initialiseDecodeCache();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (nframes > 0) {
lbajardsilogic@0: 
lbajardsilogic@0: 	reader->m_frameCount += nframes;
lbajardsilogic@0:     
lbajardsilogic@0: 	for (long i = 0; i < nframes; ++i) {
lbajardsilogic@0: 	    for (size_t c = 0; c < reader->m_channelCount; ++c) {
lbajardsilogic@0:                 reader->addSampleToDecodeCache(frames[c][i]);
lbajardsilogic@0: //		reader->m_data.push_back(frames[c][i]);
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	MUNLOCK_SAMPLEBLOCK(reader->m_data);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (reader->m_cancelled) return 1;
lbajardsilogic@0:     return 0;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: OggVorbisFileReader::getSupportedExtensions(std::set<QString> &extensions)
lbajardsilogic@0: {
lbajardsilogic@0:     extensions.insert("ogg");
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #endif
lbajardsilogic@0: #endif