lbajardsilogic@0: 
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_MAD
lbajardsilogic@0: 
lbajardsilogic@0: #include "MP3FileReader.h"
lbajardsilogic@0: #include "system/System.h"
lbajardsilogic@0: 
lbajardsilogic@0: #include <sys/types.h>
lbajardsilogic@0: #include <sys/stat.h>
lbajardsilogic@0: #include <fcntl.h>
lbajardsilogic@0: 
lbajardsilogic@0: #include <iostream>
lbajardsilogic@0: 
lbajardsilogic@0: #include <QApplication>
lbajardsilogic@0: #include <QFileInfo>
lbajardsilogic@0: #include <QProgressDialog>
lbajardsilogic@0: 
lbajardsilogic@0: MP3FileReader::MP3FileReader(QString path, bool showProgress, CacheMode mode) :
lbajardsilogic@0:     CodedAudioFileReader(mode),
lbajardsilogic@0:     m_path(path)
lbajardsilogic@0: {
lbajardsilogic@0:     m_frameCount = 0;
lbajardsilogic@0:     m_channelCount = 0;
lbajardsilogic@0:     m_sampleRate = 0;
lbajardsilogic@0:     m_fileSize = 0;
lbajardsilogic@0:     m_bitrateNum = 0;
lbajardsilogic@0:     m_bitrateDenom = 0;
lbajardsilogic@0:     m_frameCount = 0;
lbajardsilogic@0:     m_cancelled = false;
lbajardsilogic@0: 
lbajardsilogic@0:     struct stat stat;
lbajardsilogic@0:     if (::stat(path.toLocal8Bit().data(), &stat) == -1 || stat.st_size == 0) {
lbajardsilogic@0: 	m_error = QString("File %1 does not exist.").arg(path);
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     m_fileSize = stat.st_size;
lbajardsilogic@0: 
lbajardsilogic@0:     int fd = -1;
lbajardsilogic@0:     if ((fd = ::open(path.toLocal8Bit().data(), O_RDONLY
lbajardsilogic@0: #ifdef _WIN32
lbajardsilogic@0:                      | O_BINARY
lbajardsilogic@0: #endif
lbajardsilogic@0:                      , 0)) < 0) {
lbajardsilogic@0: 	m_error = QString("Failed to open file %1 for reading.").arg(path);
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }	
lbajardsilogic@0: 
lbajardsilogic@0:     unsigned char *filebuffer = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     try {
lbajardsilogic@0:         filebuffer = new unsigned char[m_fileSize];
lbajardsilogic@0:     } catch (...) {
lbajardsilogic@0:         m_error = QString("Out of memory");
lbajardsilogic@0:         ::close(fd);
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     ssize_t sz = 0;
lbajardsilogic@0:     size_t offset = 0;
lbajardsilogic@0:     while (offset < m_fileSize) {
lbajardsilogic@0:         sz = ::read(fd, filebuffer + offset, m_fileSize - offset);
lbajardsilogic@0:         if (sz < 0) {
lbajardsilogic@0:             m_error = QString("Read error for file %1 (after %2 bytes)")
lbajardsilogic@0:                 .arg(path).arg(offset);
lbajardsilogic@0:             delete[] filebuffer;
lbajardsilogic@0:             ::close(fd);
lbajardsilogic@0:             return;
lbajardsilogic@0:         } else if (sz == 0) {
lbajardsilogic@0:             std::cerr << QString("MP3FileReader::MP3FileReader: Warning: reached EOF after only %1 of %2 bytes")
lbajardsilogic@0:                 .arg(offset).arg(m_fileSize).toStdString() << std::endl;
lbajardsilogic@0:             m_fileSize = offset;
lbajardsilogic@0:             break;
lbajardsilogic@0:         }
lbajardsilogic@0:         offset += sz;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     ::close(fd);
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:     if (!decode(filebuffer, m_fileSize)) {
lbajardsilogic@0: 	m_error = QString("Failed to decode file %1.").arg(path);
lbajardsilogic@0:         delete[] filebuffer;
lbajardsilogic@0: 	return;
lbajardsilogic@0:     }
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:     delete[] filebuffer;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: MP3FileReader::~MP3FileReader()
lbajardsilogic@0: {
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: bool
lbajardsilogic@0: MP3FileReader::decode(void *mm, size_t sz)
lbajardsilogic@0: {
lbajardsilogic@0:     DecoderData data;
lbajardsilogic@0:     struct mad_decoder decoder;
lbajardsilogic@0: 
lbajardsilogic@0:     data.start = (unsigned char const *)mm;
lbajardsilogic@0:     data.length = (unsigned long)sz;
lbajardsilogic@0:     data.reader = this;
lbajardsilogic@0: 
lbajardsilogic@0:     mad_decoder_init(&decoder, &data, input, 0, 0, output, error, 0);
lbajardsilogic@0:     mad_decoder_run(&decoder, MAD_DECODER_MODE_SYNC);
lbajardsilogic@0:     mad_decoder_finish(&decoder);
lbajardsilogic@0: 
lbajardsilogic@0:     return true;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: enum mad_flow
lbajardsilogic@0: MP3FileReader::input(void *dp, struct mad_stream *stream)
lbajardsilogic@0: {
lbajardsilogic@0:     DecoderData *data = (DecoderData *)dp;
lbajardsilogic@0: 
lbajardsilogic@0:     if (!data->length) return MAD_FLOW_STOP;
lbajardsilogic@0:     mad_stream_buffer(stream, data->start, data->length);
lbajardsilogic@0:     data->length = 0;
lbajardsilogic@0: 
lbajardsilogic@0:     return MAD_FLOW_CONTINUE;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: enum mad_flow
lbajardsilogic@0: MP3FileReader::output(void *dp,
lbajardsilogic@0: 		      struct mad_header const *header,
lbajardsilogic@0: 		      struct mad_pcm *pcm)
lbajardsilogic@0: {
lbajardsilogic@0:     DecoderData *data = (DecoderData *)dp;
lbajardsilogic@0:     return data->reader->accept(header, pcm);
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: enum mad_flow
lbajardsilogic@0: MP3FileReader::accept(struct mad_header const *header,
lbajardsilogic@0: 		      struct mad_pcm *pcm)
lbajardsilogic@0: {
lbajardsilogic@0:     int channels = pcm->channels;
lbajardsilogic@0:     int frames = pcm->length;
lbajardsilogic@0: 
lbajardsilogic@0:     if (header) {
lbajardsilogic@0:         m_bitrateNum += header->bitrate;
lbajardsilogic@0:         m_bitrateDenom ++;
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (frames < 1) return MAD_FLOW_CONTINUE;
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_channelCount == 0) {
lbajardsilogic@0:         m_channelCount = channels;
lbajardsilogic@0:         m_sampleRate = pcm->samplerate;
lbajardsilogic@0:     }
lbajardsilogic@0:     
lbajardsilogic@0:     if (m_bitrateDenom > 0) {
lbajardsilogic@0:         double bitrate = m_bitrateNum / m_bitrateDenom;
lbajardsilogic@0:         double duration = double(m_fileSize * 8) / bitrate;
lbajardsilogic@0:         double elapsed = double(m_frameCount) / m_sampleRate;
lbajardsilogic@0:         double percent = ((elapsed * 100.0) / duration);
lbajardsilogic@0:         int progress = int(percent);
lbajardsilogic@0:         if (progress < 1) progress = 1;
lbajardsilogic@0:         if (progress > 99) progress = 99;
lbajardsilogic@0:         if (progress > m_progress->value()) {
lbajardsilogic@0:             m_progress->setValue(progress);
lbajardsilogic@0:             m_progress->show();
lbajardsilogic@0:             m_progress->raise();
lbajardsilogic@0:             qApp->processEvents();
lbajardsilogic@0:             if (m_progress->wasCanceled()) {
lbajardsilogic@0:                 m_cancelled = true;
lbajardsilogic@0:             }
lbajardsilogic@0:         }
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (m_cancelled) return MAD_FLOW_STOP;
lbajardsilogic@0: 
lbajardsilogic@0:     m_frameCount += frames;
lbajardsilogic@0: 
lbajardsilogic@0:     if (!isDecodeCacheInitialised()) {
lbajardsilogic@0:         initialiseDecodeCache();
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     for (int i = 0; i < frames; ++i) {
lbajardsilogic@0: 
lbajardsilogic@0: 	for (int ch = 0; ch < channels; ++ch) {
lbajardsilogic@0: 	    mad_fixed_t sample = 0;
lbajardsilogic@0: 	    if (ch < int(sizeof(pcm->samples) / sizeof(pcm->samples[0]))) {
lbajardsilogic@0: 		sample = pcm->samples[ch][i];
lbajardsilogic@0: 	    }
lbajardsilogic@0: 	    float fsample = float(sample) / float(MAD_F_ONE);
lbajardsilogic@0:             addSampleToDecodeCache(fsample);
lbajardsilogic@0: 	}
lbajardsilogic@0: 
lbajardsilogic@0: 	if (! (i & 0xffff)) {
lbajardsilogic@0: 	    // periodically munlock to ensure we don't exhaust real memory
lbajardsilogic@0: 	    // if running with memory locked down
lbajardsilogic@0: 	    MUNLOCK_SAMPLEBLOCK(m_data);
lbajardsilogic@0: 	}
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     if (frames > 0) {
lbajardsilogic@0: 	MUNLOCK_SAMPLEBLOCK(m_data);
lbajardsilogic@0:     }
lbajardsilogic@0: 
lbajardsilogic@0:     return MAD_FLOW_CONTINUE;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: enum mad_flow
lbajardsilogic@0: MP3FileReader::error(void *dp,
lbajardsilogic@0: 		     struct mad_stream *stream,
lbajardsilogic@0: 		     struct mad_frame *)
lbajardsilogic@0: {
lbajardsilogic@0:     DecoderData *data = (DecoderData *)dp;
lbajardsilogic@0: 
lbajardsilogic@0:     fprintf(stderr, "decoding error 0x%04x (%s) at byte offset %u\n",
lbajardsilogic@0: 	    stream->error, mad_stream_errorstr(stream),
lbajardsilogic@0: 	    stream->this_frame - data->start);
lbajardsilogic@0: 
lbajardsilogic@0:     return MAD_FLOW_CONTINUE;
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: void
lbajardsilogic@0: MP3FileReader::getSupportedExtensions(std::set<QString> &extensions)
lbajardsilogic@0: {
lbajardsilogic@0:     extensions.insert("mp3");
lbajardsilogic@0: }
lbajardsilogic@0: 
lbajardsilogic@0: #endif