Chris@148
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@148
|
2
|
Chris@148
|
3 /*
|
Chris@148
|
4 Sonic Visualiser
|
Chris@148
|
5 An audio file viewer and annotation editor.
|
Chris@148
|
6 Centre for Digital Music, Queen Mary, University of London.
|
Chris@148
|
7 This file copyright 2006 Chris Cannam.
|
Chris@148
|
8
|
Chris@148
|
9 This program is free software; you can redistribute it and/or
|
Chris@148
|
10 modify it under the terms of the GNU General Public License as
|
Chris@148
|
11 published by the Free Software Foundation; either version 2 of the
|
Chris@148
|
12 License, or (at your option) any later version. See the file
|
Chris@148
|
13 COPYING included with this distribution for more information.
|
Chris@148
|
14 */
|
Chris@148
|
15
|
Chris@148
|
16 #ifdef HAVE_OGGZ
|
Chris@148
|
17 #ifdef HAVE_FISHSOUND
|
Chris@148
|
18
|
Chris@148
|
19 #include "OggVorbisFileReader.h"
|
Chris@148
|
20 #include "base/Profiler.h"
|
Chris@150
|
21 #include "system/System.h"
|
Chris@148
|
22
|
Chris@148
|
23 #include <sys/types.h>
|
Chris@148
|
24 #include <sys/stat.h>
|
Chris@148
|
25 #include <sys/mman.h>
|
Chris@148
|
26 #include <fcntl.h>
|
Chris@148
|
27 #include <cmath>
|
Chris@148
|
28
|
Chris@148
|
29 #include <QApplication>
|
Chris@148
|
30 #include <QFileInfo>
|
Chris@148
|
31 #include <QProgressDialog>
|
Chris@148
|
32
|
Chris@148
|
33 static int instances = 0;
|
Chris@148
|
34
|
Chris@317
|
35 OggVorbisFileReader::OggVorbisFileReader(FileSource source,
|
Chris@263
|
36 DecodeMode decodeMode,
|
Chris@297
|
37 CacheMode mode,
|
Chris@297
|
38 size_t targetRate) :
|
Chris@297
|
39 CodedAudioFileReader(mode, targetRate),
|
Chris@316
|
40 m_source(source),
|
Chris@316
|
41 m_path(source.getLocalFilename()),
|
Chris@148
|
42 m_progress(0),
|
Chris@148
|
43 m_fileSize(0),
|
Chris@148
|
44 m_bytesRead(0),
|
Chris@271
|
45 m_commentsRead(false),
|
Chris@263
|
46 m_cancelled(false),
|
Chris@265
|
47 m_completion(0),
|
Chris@263
|
48 m_decodeThread(0)
|
Chris@148
|
49 {
|
Chris@148
|
50 m_channelCount = 0;
|
Chris@297
|
51 m_fileRate = 0;
|
Chris@148
|
52
|
Chris@316
|
53 std::cerr << "OggVorbisFileReader::OggVorbisFileReader(" << m_path.toLocal8Bit().data() << "): now have " << (++instances) << " instances" << std::endl;
|
Chris@148
|
54
|
Chris@148
|
55 Profiler profiler("OggVorbisFileReader::OggVorbisFileReader", true);
|
Chris@148
|
56
|
Chris@316
|
57 QFileInfo info(m_path);
|
Chris@148
|
58 m_fileSize = info.size();
|
Chris@148
|
59
|
Chris@316
|
60 if (!(m_oggz = oggz_open(m_path.toLocal8Bit().data(), OGGZ_READ))) {
|
Chris@316
|
61 m_error = QString("File %1 is not an OGG file.").arg(m_path);
|
Chris@148
|
62 return;
|
Chris@148
|
63 }
|
Chris@148
|
64
|
Chris@148
|
65 FishSoundInfo fsinfo;
|
Chris@148
|
66 m_fishSound = fish_sound_new(FISH_SOUND_DECODE, &fsinfo);
|
Chris@148
|
67
|
Chris@148
|
68 fish_sound_set_decoded_callback(m_fishSound, acceptFrames, this);
|
Chris@263
|
69 oggz_set_read_callback(m_oggz, -1, readPacket, this);
|
Chris@148
|
70
|
Chris@263
|
71 if (decodeMode == DecodeAtOnce) {
|
Chris@263
|
72
|
Chris@327
|
73 if (dynamic_cast<QApplication *>(QCoreApplication::instance())) {
|
Chris@327
|
74 m_progress = new QProgressDialog
|
Chris@327
|
75 (QObject::tr("Decoding %1...").arg(QFileInfo(m_path).fileName()),
|
Chris@327
|
76 QObject::tr("Stop"), 0, 100);
|
Chris@327
|
77 m_progress->hide();
|
Chris@327
|
78 }
|
Chris@148
|
79
|
Chris@263
|
80 while (oggz_read(m_oggz, 1024) > 0);
|
Chris@263
|
81
|
Chris@263
|
82 fish_sound_delete(m_fishSound);
|
Chris@263
|
83 m_fishSound = 0;
|
Chris@263
|
84 oggz_close(m_oggz);
|
Chris@263
|
85 m_oggz = 0;
|
Chris@148
|
86
|
Chris@263
|
87 if (isDecodeCacheInitialised()) finishDecodeCache();
|
Chris@148
|
88
|
Chris@281
|
89 delete m_progress;
|
Chris@281
|
90 m_progress = 0;
|
Chris@148
|
91
|
Chris@263
|
92 } else {
|
Chris@263
|
93
|
Chris@263
|
94 while (oggz_read(m_oggz, 1024) > 0 &&
|
Chris@263
|
95 m_channelCount == 0);
|
Chris@263
|
96
|
Chris@263
|
97 if (m_channelCount > 0) {
|
Chris@263
|
98 m_decodeThread = new DecodeThread(this);
|
Chris@263
|
99 m_decodeThread->start();
|
Chris@263
|
100 }
|
Chris@148
|
101 }
|
Chris@148
|
102 }
|
Chris@148
|
103
|
Chris@148
|
104 OggVorbisFileReader::~OggVorbisFileReader()
|
Chris@148
|
105 {
|
Chris@290
|
106 std::cerr << "OggVorbisFileReader::~OggVorbisFileReader(" << m_path.toLocal8Bit().data() << "): now have " << (--instances) << " instances" << std::endl;
|
Chris@263
|
107 if (m_decodeThread) {
|
Chris@265
|
108 m_cancelled = true;
|
Chris@263
|
109 m_decodeThread->wait();
|
Chris@263
|
110 delete m_decodeThread;
|
Chris@263
|
111 }
|
Chris@148
|
112 }
|
Chris@148
|
113
|
Chris@263
|
114 void
|
Chris@263
|
115 OggVorbisFileReader::DecodeThread::run()
|
Chris@263
|
116 {
|
Chris@297
|
117 if (m_reader->m_cacheMode == CacheInTemporaryFile) {
|
Chris@297
|
118 m_reader->m_completion = 1;
|
Chris@297
|
119 m_reader->startSerialised("OggVorbisFileReader::Decode");
|
Chris@297
|
120 }
|
Chris@297
|
121
|
Chris@263
|
122 while (oggz_read(m_reader->m_oggz, 1024) > 0);
|
Chris@263
|
123
|
Chris@263
|
124 fish_sound_delete(m_reader->m_fishSound);
|
Chris@263
|
125 m_reader->m_fishSound = 0;
|
Chris@263
|
126 oggz_close(m_reader->m_oggz);
|
Chris@263
|
127 m_reader->m_oggz = 0;
|
Chris@263
|
128
|
Chris@263
|
129 if (m_reader->isDecodeCacheInitialised()) m_reader->finishDecodeCache();
|
Chris@265
|
130 m_reader->m_completion = 100;
|
Chris@297
|
131
|
Chris@297
|
132 m_reader->endSerialised();
|
Chris@263
|
133 }
|
Chris@263
|
134
|
Chris@148
|
135 int
|
Chris@148
|
136 OggVorbisFileReader::readPacket(OGGZ *, ogg_packet *packet, long, void *data)
|
Chris@148
|
137 {
|
Chris@148
|
138 OggVorbisFileReader *reader = (OggVorbisFileReader *)data;
|
Chris@148
|
139 FishSound *fs = reader->m_fishSound;
|
Chris@148
|
140
|
Chris@148
|
141 fish_sound_prepare_truncation(fs, packet->granulepos, packet->e_o_s);
|
Chris@148
|
142 fish_sound_decode(fs, packet->packet, packet->bytes);
|
Chris@148
|
143
|
Chris@148
|
144 reader->m_bytesRead += packet->bytes;
|
Chris@265
|
145
|
Chris@265
|
146 // The number of bytes read by this function is smaller than
|
Chris@265
|
147 // the file size because of the packet headers
|
Chris@265
|
148 int progress = lrint(double(reader->m_bytesRead) * 114 /
|
Chris@265
|
149 double(reader->m_fileSize));
|
Chris@265
|
150 if (progress > 99) progress = 99;
|
Chris@265
|
151 reader->m_completion = progress;
|
Chris@148
|
152
|
Chris@148
|
153 if (reader->m_fileSize > 0 && reader->m_progress) {
|
Chris@148
|
154 if (progress > reader->m_progress->value()) {
|
Chris@148
|
155 reader->m_progress->setValue(progress);
|
Chris@148
|
156 reader->m_progress->show();
|
Chris@148
|
157 reader->m_progress->raise();
|
Chris@148
|
158 qApp->processEvents();
|
Chris@148
|
159 if (reader->m_progress->wasCanceled()) {
|
Chris@148
|
160 reader->m_cancelled = true;
|
Chris@148
|
161 }
|
Chris@148
|
162 }
|
Chris@265
|
163 }
|
Chris@148
|
164
|
Chris@148
|
165 if (reader->m_cancelled) return 1;
|
Chris@148
|
166 return 0;
|
Chris@148
|
167 }
|
Chris@148
|
168
|
Chris@148
|
169 int
|
Chris@148
|
170 OggVorbisFileReader::acceptFrames(FishSound *fs, float **frames, long nframes,
|
Chris@148
|
171 void *data)
|
Chris@148
|
172 {
|
Chris@148
|
173 OggVorbisFileReader *reader = (OggVorbisFileReader *)data;
|
Chris@148
|
174
|
Chris@271
|
175 if (!reader->m_commentsRead) {
|
Chris@333
|
176 {
|
Chris@333
|
177 const FishSoundComment *comment = fish_sound_comment_first_byname
|
Chris@333
|
178 (fs, "TITLE");
|
Chris@333
|
179 if (comment && comment->value) {
|
Chris@333
|
180 reader->m_title = QString::fromUtf8(comment->value);
|
Chris@333
|
181 }
|
Chris@333
|
182 }
|
Chris@333
|
183 {
|
Chris@333
|
184 const FishSoundComment *comment = fish_sound_comment_first_byname
|
Chris@333
|
185 (fs, "ARTIST");
|
Chris@333
|
186 if (comment && comment->value) {
|
Chris@333
|
187 reader->m_maker = QString::fromUtf8(comment->value);
|
Chris@333
|
188 }
|
Chris@271
|
189 }
|
Chris@271
|
190 reader->m_commentsRead = true;
|
Chris@271
|
191 }
|
Chris@271
|
192
|
Chris@148
|
193 if (reader->m_channelCount == 0) {
|
Chris@148
|
194 FishSoundInfo fsinfo;
|
Chris@148
|
195 fish_sound_command(fs, FISH_SOUND_GET_INFO,
|
Chris@148
|
196 &fsinfo, sizeof(FishSoundInfo));
|
Chris@297
|
197 reader->m_fileRate = fsinfo.samplerate;
|
Chris@148
|
198 reader->m_channelCount = fsinfo.channels;
|
Chris@148
|
199 reader->initialiseDecodeCache();
|
Chris@148
|
200 }
|
Chris@148
|
201
|
Chris@148
|
202 if (nframes > 0) {
|
Chris@297
|
203 reader->addSamplesToDecodeCache(frames, nframes);
|
Chris@148
|
204 }
|
Chris@148
|
205
|
Chris@148
|
206 if (reader->m_cancelled) return 1;
|
Chris@148
|
207 return 0;
|
Chris@148
|
208 }
|
Chris@148
|
209
|
Chris@157
|
210 void
|
Chris@290
|
211 OggVorbisFileReader::getSupportedExtensions(std::set<QString> &extensions)
|
Chris@157
|
212 {
|
Chris@157
|
213 extensions.insert("ogg");
|
Chris@157
|
214 }
|
Chris@157
|
215
|
Chris@316
|
216 bool
|
Chris@316
|
217 OggVorbisFileReader::supportsExtension(QString extension)
|
Chris@316
|
218 {
|
Chris@316
|
219 std::set<QString> extensions;
|
Chris@316
|
220 getSupportedExtensions(extensions);
|
Chris@316
|
221 return (extensions.find(extension.toLower()) != extensions.end());
|
Chris@316
|
222 }
|
Chris@316
|
223
|
Chris@316
|
224 bool
|
Chris@316
|
225 OggVorbisFileReader::supportsContentType(QString type)
|
Chris@316
|
226 {
|
Chris@316
|
227 return (type == "application/ogg");
|
Chris@316
|
228 }
|
Chris@316
|
229
|
Chris@316
|
230 bool
|
Chris@317
|
231 OggVorbisFileReader::supports(FileSource &source)
|
Chris@316
|
232 {
|
Chris@316
|
233 return (supportsExtension(source.getExtension()) ||
|
Chris@316
|
234 supportsContentType(source.getContentType()));
|
Chris@316
|
235 }
|
Chris@316
|
236
|
Chris@148
|
237 #endif
|
Chris@148
|
238 #endif
|