Chris@235: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@235: 
Chris@235: /*
Chris@235:     Sonic Visualiser
Chris@235:     An audio file viewer and annotation editor.
Chris@235:     Centre for Digital Music, Queen Mary, University of London.
Chris@235:     This file copyright 2006 Chris Cannam and QMUL.
Chris@235:     
Chris@235:     This program is free software; you can redistribute it and/or
Chris@235:     modify it under the terms of the GNU General Public License as
Chris@235:     published by the Free Software Foundation; either version 2 of the
Chris@235:     License, or (at your option) any later version.  See the file
Chris@235:     COPYING included with this distribution for more information.
Chris@235: */
Chris@235: 
Chris@235: #include "AudioFileReaderFactory.h"
Chris@235: 
Chris@235: #include "WavFileReader.h"
Chris@297: #include "ResamplingWavFileReader.h"
Chris@235: #include "OggVorbisFileReader.h"
Chris@235: #include "MP3FileReader.h"
Chris@281: #include "QuickTimeFileReader.h"
Chris@235: 
Chris@235: #include <QString>
Chris@241: #include <QFileInfo>
Chris@235: #include <iostream>
Chris@235: 
Chris@290: QString
Chris@235: AudioFileReaderFactory::getKnownExtensions()
Chris@235: {
Chris@290:     std::set<QString> extensions;
Chris@235: 
Chris@235:     WavFileReader::getSupportedExtensions(extensions);
Chris@235: #ifdef HAVE_MAD
Chris@235:     MP3FileReader::getSupportedExtensions(extensions);
Chris@235: #endif
Chris@235: #ifdef HAVE_OGGZ
Chris@235: #ifdef HAVE_FISHSOUND
Chris@235:     OggVorbisFileReader::getSupportedExtensions(extensions);
Chris@235: #endif
Chris@235: #endif
Chris@281: #ifdef HAVE_QUICKTIME
Chris@281:     QuickTimeFileReader::getSupportedExtensions(extensions);
Chris@281: #endif
Chris@235: 
Chris@290:     QString rv;
Chris@290:     for (std::set<QString>::const_iterator i = extensions.begin();
Chris@235:          i != extensions.end(); ++i) {
Chris@235:         if (i != extensions.begin()) rv += " ";
Chris@235:         rv += "*." + *i;
Chris@235:     }
Chris@235: 
Chris@235:     return rv;
Chris@235: }
Chris@235: 
Chris@235: AudioFileReader *
Chris@297: AudioFileReaderFactory::createReader(QString path, size_t targetRate)
Chris@235: {
Chris@290:     QString err;
Chris@235: 
Chris@297:     std::cerr << "AudioFileReaderFactory::createReader(\"" << path.toStdString() << "\"): Requested rate: " << targetRate << std::endl;
Chris@297: 
Chris@235:     AudioFileReader *reader = 0;
Chris@235: 
Chris@241:     // First try to construct a preferred reader based on the
Chris@241:     // extension.  If we can't identify one or it fails to load the
Chris@241:     // file, fall back to trying all readers in no particular order.
Chris@241: 
Chris@290:     QString ext = QFileInfo(path).suffix().toLower();
Chris@290:     std::set<QString> extensions;
Chris@241: 
Chris@241:     WavFileReader::getSupportedExtensions(extensions);
Chris@241:     if (extensions.find(ext) != extensions.end()) {
Chris@241:         reader = new WavFileReader(path);
Chris@297: 
Chris@297:         if (targetRate != 0 &&
Chris@297:             reader->isOK() &&
Chris@297:             reader->getSampleRate() != targetRate) {
Chris@297: 
Chris@297:             std::cerr << "AudioFileReaderFactory::createReader: WAV file rate: " << reader->getSampleRate() << ", creating resampling reader" << std::endl;
Chris@297: 
Chris@297:             delete reader;
Chris@297:             reader = new ResamplingWavFileReader
Chris@297:                 (path,
Chris@297:                  ResamplingWavFileReader::ResampleThreaded,
Chris@297:                  ResamplingWavFileReader::CacheInTemporaryFile,
Chris@297:                  targetRate);
Chris@297:         }
Chris@241:     }
Chris@241:     
Chris@241: #ifdef HAVE_OGGZ
Chris@241: #ifdef HAVE_FISHSOUND
Chris@241:     if (!reader) {
Chris@241:         extensions.clear();
Chris@241:         OggVorbisFileReader::getSupportedExtensions(extensions);
Chris@241:         if (extensions.find(ext) != extensions.end()) {
Chris@241:             reader = new OggVorbisFileReader
Chris@263:                 (path, 
Chris@265:                  OggVorbisFileReader::DecodeThreaded,
Chris@297:                  OggVorbisFileReader::CacheInTemporaryFile,
Chris@297:                  targetRate);
Chris@241:         }
Chris@241:     }
Chris@241: #endif
Chris@241: #endif
Chris@241: 
Chris@281: #ifdef HAVE_MAD
Chris@281:     if (!reader) {
Chris@281:         extensions.clear();
Chris@281:         MP3FileReader::getSupportedExtensions(extensions);
Chris@281:         if (extensions.find(ext) != extensions.end()) {
Chris@281:             reader = new MP3FileReader
Chris@281:                 (path,
Chris@281:                  MP3FileReader::DecodeThreaded,
Chris@297:                  MP3FileReader::CacheInTemporaryFile,
Chris@297:                  targetRate);
Chris@281:         }
Chris@281:     }
Chris@281: #endif
Chris@281: 
Chris@281: #ifdef HAVE_QUICKTIME
Chris@281:     if (!reader) {
Chris@281:         extensions.clear();
Chris@281:         QuickTimeFileReader::getSupportedExtensions(extensions);
Chris@281:         if (extensions.find(ext) != extensions.end()) {
Chris@281:             reader = new QuickTimeFileReader
Chris@281:                 (path,
Chris@281:                  QuickTimeFileReader::DecodeThreaded,
Chris@297:                  QuickTimeFileReader::CacheInTemporaryFile,
Chris@297:                  targetRate);
Chris@281:         }
Chris@281:     }
Chris@281: #endif
Chris@281: 
Chris@241:     if (reader) {
Chris@241:         if (reader->isOK()) return reader;
Chris@241:         if (reader->getError() != "") {
Chris@241:             std::cerr << "AudioFileReaderFactory: Preferred reader for "
Chris@290:                       << "extension \"" << ext.toStdString() << "\" failed: \""
Chris@290:                       << reader->getError().toStdString() << "\"" << std::endl;
Chris@241:         } else {
Chris@241:             std::cerr << "AudioFileReaderFactory: Preferred reader for "
Chris@290:                       << "extension \"" << ext.toStdString() << "\" failed"
Chris@241:                       << std::endl;
Chris@241:         }            
Chris@241:         delete reader;
Chris@241:         reader = 0;
Chris@241:     }
Chris@241: 
Chris@308:     return reader;
Chris@235: }
Chris@235: