#include <QDebug>
#include <QDir>
#include <QFileInfo>
+#include <QtConcurrentRun>
#include <mpegfile.h>
+#include <flacfile.h>
+#include <oggfile.h>
#include <id3v2tag.h>
#include <mpeg/id3v2/frames/attachedpictureframe.h>
+using namespace SomePlayer::DataObjects;
+
CoverFinder::CoverFinder(QObject *parent) :
QObject(parent)
{
SUFFIXES << "png" << "jpg" << "jpeg" << "bmp" << "gif";
NAMES << "cover" << "folder" << "album";
DIRS << "cover" << "folder" << ".cover" << ".folder";
+ PRIOR_NAMES << "front" << "cover";
+ UNPRIOR_NAMES << "disk" << "back" << "booklet" << "cd";
}
-bool CoverFinder::find(QString path) {
+bool CoverFinder::_find(QString path) {
QDir dir(path);
QFileInfoList items = dir.entryInfoList(QDir::Files);
QFileInfoList dirs = dir.entryInfoList(QDir::NoDotAndDotDot | QDir::AllDirs | QDir::Hidden);
QFileInfoList pics;
+ QFileInfoList appropriate_pics;
foreach (QFileInfo item, items) {
if (SUFFIXES.contains(item.suffix().toLower())) {
pics << item;
if (NAMES.contains(item.baseName().toLower())) {
emit found(QImage(item.absoluteFilePath()));
+ emit foundPath(item.absoluteFilePath());
return true;
}
}
foreach (QFileInfo item, pics) {
QImage i(item.absoluteFilePath());
if (0.8 <= (i.width()*1.0/(i.height()+1.0)) <= 1.2) {
- emit found(i);
- return true;
+ appropriate_pics << item;
+ }
+ }
+ if (!appropriate_pics.isEmpty()) {
+ foreach (QString name, PRIOR_NAMES) {
+ foreach (QFileInfo item, appropriate_pics) {
+ if (item.baseName().toLower().contains(name)) {
+ emit found(QImage(item.absoluteFilePath()));
+ emit foundPath(item.absoluteFilePath());
+ return true;
+ }
+ }
+ }
+ QFileInfoList unprior;
+ foreach (QString name, UNPRIOR_NAMES) {
+ foreach (QFileInfo item, appropriate_pics) {
+ if (item.baseName().toLower().contains(name)) {
+ unprior << item;
+ }
+ }
}
+ if (appropriate_pics.size() > unprior.size()) {
+ foreach (QFileInfo item, appropriate_pics) {
+ if (!unprior.contains(item)) {
+ emit found(QImage(item.absoluteFilePath()));
+ emit foundPath(item.absoluteFilePath());
+ return true;
+ }
+ }
+ }
+ emit found(QImage(unprior.at(0).absoluteFilePath()));
+ emit foundPath(unprior.at(0).absoluteFilePath());
+ return true;
}
foreach(QFileInfo item, dirs) {
if (DIRS.contains(item.fileName().toLower())) {
- if (find(item.absoluteFilePath())) {
+ if (_find(item.absoluteFilePath())) {
return true;
}
}
}
- emit found(_defaultCover);
return false;
}
return _defaultCover;
}
-bool CoverFinder::extract(QString file) {
+bool CoverFinder::_extract(QString file) {
QFileInfo info(file);
QString suffix = info.suffix().toLower();
TagLib::ID3v2::Tag *tag = NULL;
+ bool todo = false;
+ TagLib::File *f = NULL;
if (suffix == "mp3") {
- TagLib::MPEG::File f(QFile::encodeName(file).data());
- tag = f.ID3v2Tag();
- if (f.isValid() && tag != NULL) {
- TagLib::ID3v2::FrameList l = tag->frameList("APIC");
- if (l.isEmpty())
- return false;
- TagLib::ID3v2::AttachedPictureFrame *pic = static_cast<TagLib::ID3v2::AttachedPictureFrame *>(l.front());
- QImage img;
- img.loadFromData((const uchar *) pic->picture().data(), pic->picture().size());
- emit found(img);
- return true;
- }
+ f = new TagLib::MPEG::File(QFile::encodeName(file).data(), true, TagLib::AudioProperties::Fast);
+ if (f == NULL) return false;
+ tag = ((TagLib::MPEG::File*)f)->ID3v2Tag();
+ todo = f->isValid();
+ } else if (suffix == "flac") {
+ f = new TagLib::FLAC::File(QFile::encodeName(file).data(), true, TagLib::AudioProperties::Fast);
+ if (f == NULL) return false;
+ tag = ((TagLib::FLAC::File*)f)->ID3v2Tag();
+ todo = f->isValid();
+ }
+ if (todo && tag != NULL) {
+ TagLib::ID3v2::FrameList l = tag->frameList("APIC");
+ if (l.isEmpty())
+ return false;
+ TagLib::ID3v2::AttachedPictureFrame *pic = static_cast<TagLib::ID3v2::AttachedPictureFrame *>(l.front());
+ QImage img;
+ img.loadFromData((const uchar *) pic->picture().data(), pic->picture().size());
+ emit found(img);
+ return true;
+ }
+ if (f != NULL) delete f;
+ return false;
+}
+
+void CoverFinder::find(Track track) {
+ QFileInfo filePath(track.source());
+ QtConcurrent::run(this, &CoverFinder::_async_find, filePath, track);
+}
+
+bool CoverFinder::_async_find(QFileInfo filePath, Track track) {
+ if (!_find(filePath.absolutePath()) && !_tfind(track.metadata().artist(), track.metadata().album()) &&
+ !_malfind(filePath.absolutePath()+"/.mediaartlocal/"+track.mediaArtLocal()) &&
+ !_extract(filePath.absoluteFilePath())) {
+ emit found(_defaultCover);
+ return false;
+ }
+ return true;
+}
+
+bool CoverFinder::_tfind(QString artist, QString album) {
+ QString aname = artist.toLower();
+ QString aalbum = album.toLower();
+ aname.replace("/", "");
+ QString aaname = aname+" - "+aalbum;
+ aaname.replace("/", "");
+ QString fname1 = QDir::homePath()+"/.covers/"+aaname+".jpg";
+ QString fname2 = QDir::homePath()+"/.covers/"+aname+".jpg";
+ if (QFile::exists(fname1)) {
+ emit found(QImage(fname1));
+ emit foundPath(fname1);
+ return true;
+ } else if (QFile::exists(fname2)) {
+ emit found(QImage(fname2));
+ emit foundPath(fname2);
+ return true;
+ }
+ return false;
+}
+
+bool CoverFinder::_malfind(QString path) {
+ if (QFile::exists(path)) {
+ emit found(QImage(path));
+ emit foundPath(path);
+ return true;
}
return false;
}