iiiypuk/cantata
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Files
d9bc2ddfcfd91df830ad020a332ae259643b128d
cantata/devices/tagreader.cpp
craig 9bda8bab76 Fix MTP various-artists workaround
2012-01-20 20:20:33 +00:00

281 lines
8.8 KiB
C++
Raw Blame History

/*
* Cantata
*
* Copyright (c) 2011-2012 Craig Drummond <craig.p.drummond@gmail.com>
*
* ----
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; see the file COPYING. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#include "tagreader.h"
#include "filetyperesolver.h"
#include <QtCore/QFile>
#include <QtCore/QMutex>
#include <QtCore/QMutexLocker>
#include <QtCore/QTextCodec>
#include <taglib/fileref.h>
#include <taglib/aifffile.h>
#include <taglib/asffile.h>
#include <taglib/flacfile.h>
#include <taglib/mp4file.h>
#include <taglib/mpcfile.h>
#include <taglib/mpegfile.h>
#include <taglib/oggfile.h>
#include <taglib/oggflacfile.h>
#include <taglib/rifffile.h>
#include <taglib/speexfile.h>
#include <taglib/trueaudiofile.h>
#include <taglib/vorbisfile.h>
#include <taglib/wavfile.h>
#include <taglib/wavpackfile.h>
#include <taglib/asftag.h>
#include <taglib/apetag.h>
#include <taglib/id3v2tag.h>
#include <taglib/mp4tag.h>
#include <taglib/xiphcomment.h>
#ifdef TAGLIB_EXTRAS_FOUND
#include <taglib-extras/audiblefiletyperesolver.h>
#include <taglib-extras/realmediafiletyperesolver.h>
#endif
namespace TagReader
{
static TagLib::FileRef getFileRef(const QString &path)
{
#ifdef Q_OS_WIN32
const wchar_t *encodedName = reinterpret_cast< const wchar_t * >(path.utf16());
#else
#ifdef COMPLEX_TAGLIB_FILENAME
const wchar_t *encodedName = reinterpret_cast< const wchar_t * >(path.utf16());
#else
QByteArray fileName = QFile::encodeName(path);
const char *encodedName = fileName.constData(); // valid as long as fileName exists
#endif
#endif
return TagLib::FileRef(encodedName, true, TagLib::AudioProperties::Fast);
}
static void ensureFileTypeResolvers()
{
static bool alreadyAdded = false;
if(!alreadyAdded) {
alreadyAdded = true;
#ifdef TAGLIB_FOUND
#ifdef TAGLIB_EXTRAS_FOUND
TagLib::FileRef::addFileTypeResolver(new AudibleFileTypeResolver);
TagLib::FileRef::addFileTypeResolver(new RealMediaFileTypeResolver);
#endif
TagLib::FileRef::addFileTypeResolver(new FileTypeResolver());
#endif
}
}
static QString tString2QString(const TagLib::String &str)
{
static QTextCodec *codec = QTextCodec::codecForName( "UTF-8" );
return codec->toUnicode(str.toCString(true)).trimmed();
}
TagLib::String qString2TString(const QString &str)
{
QString val = str.trimmed();
return val.isEmpty() ? TagLib::String::null : TagLib::String(val.toUtf8().data(), TagLib::String::UTF8);
}
// static void readID3v1Tags(const TagLib::FileRef fileref, Song &song)
// {
// }
static void readID3v2Tags(TagLib::ID3v2::Tag *tag, Song &song)
{
TagLib::ID3v2::FrameList list = tag->frameListMap()["TPE2"];
if (!list.isEmpty()) {
song.albumartist=tString2QString(list.front()->toString());
}
}
static void readAPETags(TagLib::APE::Tag *tag, Song &song)
{
TagLib::APE::ItemListMap map = tag->itemListMap();
if (map.contains("Album Artist")) {
song.albumartist=tString2QString(map["Album Artist"].toString());
}
}
static void readVorbisCommentTags(TagLib::Ogg::XiphComment *tag, Song &song)
{
if (!tag->contains("ALBUMARTIST")) {
return;
}
TagLib::StringList list = tag->fieldListMap()["ALBUMARTIST"];
if (!list.isEmpty()) {
song.albumartist=tString2QString(list.front());
}
}
static void readMP4Tags(TagLib::MP4::Tag *tag, Song &song)
{
TagLib::MP4::ItemListMap map = tag->itemListMap();
if (map.contains("aART") && !map["aART"].toStringList().isEmpty()) {
song.albumartist=tString2QString(map["aART"].toStringList().front());
}
}
static void readASFTags(TagLib::ASF::Tag *tag, Song &song)
{
TagLib::ASF::AttributeListMap map = tag->attributeListMap();
if (map.contains("WM/AlbumTitle") && !map["WM/AlbumTitle"].isEmpty()) {
song.albumartist=tString2QString(map["WM/AlbumTitle"].front().toString());
}
}
static Song readTags(const TagLib::FileRef fileref)
{
Song song;
TagLib::Tag *tag=fileref.tag();
song.title=tString2QString(tag->title());
song.artist=tString2QString(tag->artist());
song.album=tString2QString(tag->album());
song.genre=tString2QString(tag->genre());
song.track=tag->track();
song.year=tag->year();
if(TagLib::MPEG::File *file = dynamic_cast< TagLib::MPEG::File * >(fileref.file())) {
if(file->ID3v2Tag()) {
readID3v2Tags(file->ID3v2Tag(), song);
} else if(file->APETag()) {
readAPETags(file->APETag(), song);
// } else if(file->ID3v1Tag()) {
// readID3v1Tags(fileref, song);
}
} else if(TagLib::Ogg::Vorbis::File *file = dynamic_cast< TagLib::Ogg::Vorbis::File * >(fileref.file())) {
if(file->tag()) {
readVorbisCommentTags(file->tag(), song);
}
} else if(TagLib::Ogg::FLAC::File *file = dynamic_cast< TagLib::Ogg::FLAC::File * >(fileref.file())) {
if(file->tag()) {
readVorbisCommentTags(file->tag(), song);
}
} else if(TagLib::Ogg::Speex::File *file = dynamic_cast< TagLib::Ogg::Speex::File * >(fileref.file())) {
if(file->tag()) {
readVorbisCommentTags(file->tag(), song);
}
} else if(TagLib::FLAC::File *file = dynamic_cast< TagLib::FLAC::File * >(fileref.file())) {
if(file->xiphComment()) {
readVorbisCommentTags(file->xiphComment(), song);
} else if(file->ID3v2Tag()) {
readID3v2Tags(file->ID3v2Tag(), song);
// } else if(file->ID3v1Tag()) {
// readID3v1Tags(fileref, song);
}
} else if(TagLib::MP4::File *file = dynamic_cast< TagLib::MP4::File * >(fileref.file())) {
TagLib::MP4::Tag *tag = dynamic_cast< TagLib::MP4::Tag * >(file->tag());
if(tag) {
readMP4Tags(tag, song);
}
} else if(TagLib::MPC::File *file = dynamic_cast< TagLib::MPC::File * >(fileref.file())) {
if(file->APETag()) {
readAPETags(file->APETag(), song);
// } else if(file->ID3v1Tag()) {
// readID3v1Tags(fileref, song);
}
} else if(TagLib::RIFF::AIFF::File *file = dynamic_cast< TagLib::RIFF::AIFF::File * >(fileref.file())) {
if(file->tag()) {
readID3v2Tags(file->tag(), song);
}
} else if(TagLib::RIFF::WAV::File *file = dynamic_cast< TagLib::RIFF::WAV::File * >(fileref.file())) {
if(file->tag()) {
readID3v2Tags(file->tag(), song);
}
} else if(TagLib::ASF::File *file = dynamic_cast< TagLib::ASF::File * >(fileref.file())) {
TagLib::ASF::Tag *tag = dynamic_cast< TagLib::ASF::Tag * >(file->tag());
if(tag) {
readASFTags(tag, song);
}
} else if(TagLib::TrueAudio::File *file = dynamic_cast< TagLib::TrueAudio::File * >(fileref.file())) {
if(file->ID3v2Tag(false)) {
readID3v2Tags(file->ID3v2Tag(false), song);
// } else if(file->ID3v1Tag()) {
// readID3v1Tags(fileref, song);
}
} else if(TagLib::WavPack::File *file = dynamic_cast< TagLib::WavPack::File * >(fileref.file())) {
if(file->APETag()) {
readAPETags(file->APETag(), song);
// } else if(file->ID3v1Tag()) {
// readID3v1Tags(fileref, song);
}
}
return song;
}
static QMutex mutex;
Song read(const QString &fileName)
{
QMutexLocker locker(&mutex);
ensureFileTypeResolvers();
Song song;
TagLib::FileRef fileref = getFileRef(fileName);
if (fileref.isNull()) {
return song;
}
song=readTags(fileref);
song.file=fileName;
song.time=fileref.audioProperties() ? (fileref.audioProperties()->length() * 1000) : 0;
if (!song.albumartist.isEmpty() && song.albumartist != song.artist) {
song.modifiedtitle = song.artist + QLatin1String(" - ") + song.title;
}
return song;
}
bool updateArtistAndTitleTags(const QString &fileName, const Song &song)
{
QMutexLocker locker(&mutex);
ensureFileTypeResolvers();
TagLib::FileRef fileref = getFileRef(fileName);
if (fileref.isNull()) {
return false;
}
TagLib::Tag *tag=fileref.tag();
tag->setTitle(qString2TString(song.title));
tag->setArtist(qString2TString(song.artist));
TagLib::MPEG::File *mpeg=dynamic_cast<TagLib::MPEG::File *>(fileref.file());
return mpeg ? mpeg->save(TagLib::MPEG::File::ID3v2) : fileref.file()->save();
}
}
Reference in New Issue View Git Blame Copy Permalink