annotate data/fileio/MP3FileReader.cpp @ 231:0031495aba07

...
author Chris Cannam
date Mon, 12 Feb 2007 18:15:49 +0000
parents 2854dc429310
children 71dfc6ab3b54
rev   line source
Chris@148 1
Chris@148 2 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@148 3
Chris@148 4 /*
Chris@148 5 Sonic Visualiser
Chris@148 6 An audio file viewer and annotation editor.
Chris@148 7 Centre for Digital Music, Queen Mary, University of London.
Chris@148 8 This file copyright 2006 Chris Cannam.
Chris@148 9
Chris@148 10 This program is free software; you can redistribute it and/or
Chris@148 11 modify it under the terms of the GNU General Public License as
Chris@148 12 published by the Free Software Foundation; either version 2 of the
Chris@148 13 License, or (at your option) any later version. See the file
Chris@148 14 COPYING included with this distribution for more information.
Chris@148 15 */
Chris@148 16
Chris@148 17 #ifdef HAVE_MAD
Chris@148 18
Chris@148 19 #include "MP3FileReader.h"
Chris@150 20 #include "system/System.h"
Chris@148 21
Chris@148 22 #include <sys/types.h>
Chris@148 23 #include <sys/stat.h>
Chris@148 24 #include <fcntl.h>
Chris@148 25
Chris@148 26 #include <iostream>
Chris@148 27
Chris@148 28 #include <QApplication>
Chris@186 29 #include <QFileInfo>
Chris@148 30 #include <QProgressDialog>
Chris@148 31
Chris@148 32 MP3FileReader::MP3FileReader(QString path, bool showProgress, CacheMode mode) :
Chris@148 33 CodedAudioFileReader(mode),
Chris@148 34 m_path(path)
Chris@148 35 {
Chris@148 36 m_frameCount = 0;
Chris@148 37 m_channelCount = 0;
Chris@148 38 m_sampleRate = 0;
Chris@148 39 m_fileSize = 0;
Chris@148 40 m_bitrateNum = 0;
Chris@148 41 m_bitrateDenom = 0;
Chris@148 42 m_frameCount = 0;
Chris@148 43 m_cancelled = false;
Chris@148 44
Chris@148 45 struct stat stat;
Chris@148 46 if (::stat(path.toLocal8Bit().data(), &stat) == -1 || stat.st_size == 0) {
Chris@148 47 m_error = QString("File %1 does not exist.").arg(path);
Chris@148 48 return;
Chris@148 49 }
Chris@148 50
Chris@148 51 m_fileSize = stat.st_size;
Chris@148 52
Chris@229 53 int fd = -1;
Chris@231 54 if ((fd = ::open(path.toLocal8Bit().data(), O_RDONLY
Chris@231 55 #ifdef _WIN32
Chris@231 56 | O_BINARY
Chris@231 57 #endif
Chris@231 58 , 0)) < 0) {
Chris@148 59 m_error = QString("Failed to open file %1 for reading.").arg(path);
Chris@148 60 return;
Chris@148 61 }
Chris@148 62
Chris@148 63 unsigned char *filebuffer = 0;
Chris@148 64
Chris@148 65 try {
Chris@229 66 filebuffer = new unsigned char[m_fileSize];
Chris@148 67 } catch (...) {
Chris@148 68 m_error = QString("Out of memory");
Chris@148 69 ::close(fd);
Chris@148 70 return;
Chris@148 71 }
Chris@148 72
Chris@229 73 ssize_t sz = 0;
Chris@229 74 size_t offset = 0;
Chris@229 75 while (offset < m_fileSize) {
Chris@229 76 sz = ::read(fd, filebuffer + offset, m_fileSize - offset);
Chris@229 77 if (sz < 0) {
Chris@231 78 m_error = QString("Read error for file %1 (after %2 bytes)")
Chris@231 79 .arg(path).arg(offset);
Chris@229 80 delete[] filebuffer;
Chris@229 81 ::close(fd);
Chris@229 82 return;
Chris@230 83 } else if (sz == 0) {
Chris@231 84 std::cerr << QString("MP3FileReader::MP3FileReader: Warning: reached EOF after only %1 of %2 bytes")
Chris@231 85 .arg(offset).arg(m_fileSize).toStdString() << std::endl;
Chris@231 86 m_fileSize = offset;
Chris@230 87 break;
Chris@229 88 }
Chris@229 89 offset += sz;
Chris@148 90 }
Chris@148 91
Chris@148 92 ::close(fd);
Chris@148 93
Chris@148 94 if (showProgress) {
Chris@148 95 m_progress = new QProgressDialog
Chris@186 96 (QObject::tr("Decoding %1...").arg(QFileInfo(path).fileName()),
Chris@148 97 QObject::tr("Stop"), 0, 100);
Chris@148 98 m_progress->hide();
Chris@148 99 }
Chris@148 100
Chris@229 101 if (!decode(filebuffer, m_fileSize)) {
Chris@148 102 m_error = QString("Failed to decode file %1.").arg(path);
Chris@148 103 delete[] filebuffer;
Chris@148 104 return;
Chris@148 105 }
Chris@148 106
Chris@148 107 if (isDecodeCacheInitialised()) finishDecodeCache();
Chris@148 108
Chris@148 109 if (showProgress) {
Chris@148 110 delete m_progress;
Chris@148 111 m_progress = 0;
Chris@148 112 }
Chris@148 113
Chris@148 114 delete[] filebuffer;
Chris@148 115 }
Chris@148 116
Chris@148 117 MP3FileReader::~MP3FileReader()
Chris@148 118 {
Chris@148 119 }
Chris@148 120
Chris@148 121 bool
Chris@148 122 MP3FileReader::decode(void *mm, size_t sz)
Chris@148 123 {
Chris@148 124 DecoderData data;
Chris@148 125 struct mad_decoder decoder;
Chris@148 126
Chris@148 127 data.start = (unsigned char const *)mm;
Chris@148 128 data.length = (unsigned long)sz;
Chris@148 129 data.reader = this;
Chris@148 130
Chris@148 131 mad_decoder_init(&decoder, &data, input, 0, 0, output, error, 0);
Chris@148 132 mad_decoder_run(&decoder, MAD_DECODER_MODE_SYNC);
Chris@148 133 mad_decoder_finish(&decoder);
Chris@148 134
Chris@148 135 return true;
Chris@148 136 }
Chris@148 137
Chris@148 138 enum mad_flow
Chris@148 139 MP3FileReader::input(void *dp, struct mad_stream *stream)
Chris@148 140 {
Chris@148 141 DecoderData *data = (DecoderData *)dp;
Chris@148 142
Chris@148 143 if (!data->length) return MAD_FLOW_STOP;
Chris@148 144 mad_stream_buffer(stream, data->start, data->length);
Chris@148 145 data->length = 0;
Chris@148 146
Chris@148 147 return MAD_FLOW_CONTINUE;
Chris@148 148 }
Chris@148 149
Chris@148 150 enum mad_flow
Chris@148 151 MP3FileReader::output(void *dp,
Chris@148 152 struct mad_header const *header,
Chris@148 153 struct mad_pcm *pcm)
Chris@148 154 {
Chris@148 155 DecoderData *data = (DecoderData *)dp;
Chris@148 156 return data->reader->accept(header, pcm);
Chris@148 157 }
Chris@148 158
Chris@148 159 enum mad_flow
Chris@148 160 MP3FileReader::accept(struct mad_header const *header,
Chris@148 161 struct mad_pcm *pcm)
Chris@148 162 {
Chris@148 163 int channels = pcm->channels;
Chris@148 164 int frames = pcm->length;
Chris@148 165
Chris@148 166 if (header) {
Chris@148 167 m_bitrateNum += header->bitrate;
Chris@148 168 m_bitrateDenom ++;
Chris@148 169 }
Chris@148 170
Chris@148 171 if (frames < 1) return MAD_FLOW_CONTINUE;
Chris@148 172
Chris@148 173 if (m_channelCount == 0) {
Chris@148 174 m_channelCount = channels;
Chris@148 175 m_sampleRate = pcm->samplerate;
Chris@148 176 }
Chris@148 177
Chris@148 178 if (m_bitrateDenom > 0) {
Chris@148 179 double bitrate = m_bitrateNum / m_bitrateDenom;
Chris@148 180 double duration = double(m_fileSize * 8) / bitrate;
Chris@148 181 double elapsed = double(m_frameCount) / m_sampleRate;
Chris@148 182 double percent = ((elapsed * 100.0) / duration);
Chris@148 183 int progress = int(percent);
Chris@148 184 if (progress < 1) progress = 1;
Chris@148 185 if (progress > 99) progress = 99;
Chris@148 186 if (progress > m_progress->value()) {
Chris@148 187 m_progress->setValue(progress);
Chris@148 188 m_progress->show();
Chris@148 189 m_progress->raise();
Chris@148 190 qApp->processEvents();
Chris@148 191 if (m_progress->wasCanceled()) {
Chris@148 192 m_cancelled = true;
Chris@148 193 }
Chris@148 194 }
Chris@148 195 }
Chris@148 196
Chris@148 197 if (m_cancelled) return MAD_FLOW_STOP;
Chris@148 198
Chris@148 199 m_frameCount += frames;
Chris@148 200
Chris@148 201 if (!isDecodeCacheInitialised()) {
Chris@148 202 initialiseDecodeCache();
Chris@148 203 }
Chris@148 204
Chris@148 205 for (int i = 0; i < frames; ++i) {
Chris@148 206
Chris@148 207 for (int ch = 0; ch < channels; ++ch) {
Chris@148 208 mad_fixed_t sample = 0;
Chris@148 209 if (ch < int(sizeof(pcm->samples) / sizeof(pcm->samples[0]))) {
Chris@148 210 sample = pcm->samples[ch][i];
Chris@148 211 }
Chris@148 212 float fsample = float(sample) / float(MAD_F_ONE);
Chris@148 213 addSampleToDecodeCache(fsample);
Chris@148 214 }
Chris@148 215
Chris@148 216 if (! (i & 0xffff)) {
Chris@148 217 // periodically munlock to ensure we don't exhaust real memory
Chris@148 218 // if running with memory locked down
Chris@148 219 MUNLOCK_SAMPLEBLOCK(m_data);
Chris@148 220 }
Chris@148 221 }
Chris@148 222
Chris@148 223 if (frames > 0) {
Chris@148 224 MUNLOCK_SAMPLEBLOCK(m_data);
Chris@148 225 }
Chris@148 226
Chris@148 227 return MAD_FLOW_CONTINUE;
Chris@148 228 }
Chris@148 229
Chris@148 230 enum mad_flow
Chris@148 231 MP3FileReader::error(void *dp,
Chris@148 232 struct mad_stream *stream,
Chris@148 233 struct mad_frame *)
Chris@148 234 {
Chris@148 235 DecoderData *data = (DecoderData *)dp;
Chris@148 236
Chris@148 237 fprintf(stderr, "decoding error 0x%04x (%s) at byte offset %u\n",
Chris@148 238 stream->error, mad_stream_errorstr(stream),
Chris@148 239 stream->this_frame - data->start);
Chris@148 240
Chris@148 241 return MAD_FLOW_CONTINUE;
Chris@148 242 }
Chris@148 243
Chris@157 244 void
Chris@157 245 MP3FileReader::getSupportedExtensions(std::set<QString> &extensions)
Chris@157 246 {
Chris@157 247 extensions.insert("mp3");
Chris@157 248 }
Chris@157 249
Chris@148 250 #endif