annotate data/fileio/MP3FileReader.cpp @ 282:d9319859a4cf tip

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