239 lines
7.6 KiB
C++
239 lines
7.6 KiB
C++
/*
|
|
* Strawberry Music Player
|
|
* This file was part of Clementine.
|
|
* Copyright 2010, David Sansome <me@davidsansome.com>
|
|
* Copyright 2018-2021, Jonas Kvinge <jonas@jkvinge.net>
|
|
*
|
|
* Strawberry 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Strawberry 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 Strawberry. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <QFile>
|
|
#include <QSize>
|
|
#include <QString>
|
|
#include <QImage>
|
|
|
|
#include "core/song.h"
|
|
#include "core/tagreaderclient.h"
|
|
#include "albumcoverloaderoptions.h"
|
|
#include "albumcoverexport.h"
|
|
#include "coverexportrunnable.h"
|
|
|
|
CoverExportRunnable::CoverExportRunnable(const AlbumCoverExport::DialogResult &dialog_result, const AlbumCoverLoaderOptions::Types &cover_types, const Song &song, QObject *parent)
|
|
: QObject(parent),
|
|
dialog_result_(dialog_result),
|
|
cover_types_(cover_types),
|
|
song_(song) {}
|
|
|
|
void CoverExportRunnable::run() {
|
|
|
|
if (song_.art_unset() || (!song_.art_embedded() && !song_.art_automatic_is_valid() && !song_.art_manual_is_valid())) {
|
|
EmitCoverSkipped();
|
|
}
|
|
else {
|
|
if (dialog_result_.RequiresCoverProcessing()) {
|
|
ProcessAndExportCover();
|
|
}
|
|
else {
|
|
ExportCover();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// Exports a single album cover using a "save QImage to file" approach.
|
|
// For performance reasons this method will be invoked only if loading and in memory processing of images is necessary for current settings which means that:
|
|
// - either the force size flag is being used
|
|
// - or the "overwrite smaller" mode is used
|
|
// In all other cases, the faster ExportCover() method will be used.
|
|
void CoverExportRunnable::ProcessAndExportCover() {
|
|
|
|
QImage image;
|
|
QString extension;
|
|
|
|
for (const AlbumCoverLoaderOptions::Type cover_type : cover_types_) {
|
|
switch (cover_type) {
|
|
case AlbumCoverLoaderOptions::Type::Unset:
|
|
if (song_.art_unset()) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Embedded:
|
|
if (song_.art_embedded() && dialog_result_.export_embedded_) {
|
|
image = TagReaderClient::Instance()->LoadEmbeddedArtAsImageBlocking(song_.url().toLocalFile());
|
|
if (!image.isNull()) {
|
|
extension = QStringLiteral("jpg");
|
|
}
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Manual:
|
|
if (dialog_result_.export_downloaded_ && song_.art_manual_is_valid()) {
|
|
const QString cover_path = song_.art_manual().toLocalFile();
|
|
if (image.load(cover_path)) {
|
|
extension = cover_path.section('.', -1);
|
|
}
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Automatic:
|
|
if (dialog_result_.export_downloaded_ && song_.art_automatic_is_valid()) {
|
|
const QString cover_path = song_.art_automatic().toLocalFile();
|
|
if (image.load(cover_path)) {
|
|
extension = cover_path.section('.', -1);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
if (!image.isNull() && !extension.isEmpty()) break;
|
|
}
|
|
|
|
if (image.isNull() || extension.isEmpty()) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
|
|
// Rescale if necessary
|
|
if (dialog_result_.IsSizeForced()) {
|
|
image = image.scaled(QSize(dialog_result_.width_, dialog_result_.height_), Qt::IgnoreAspectRatio);
|
|
}
|
|
|
|
QString cover_dir = song_.url().toLocalFile().section('/', 0, -2);
|
|
QString new_file = cover_dir + '/' + dialog_result_.filename_ + '.' + (song_.art_embedded() ? QStringLiteral("jpg") : extension);
|
|
|
|
// If the file exists, do not override!
|
|
if (dialog_result_.overwrite_ == AlbumCoverExport::OverwriteMode::None && QFile::exists(new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
|
|
// We're handling overwrite as remove + copy so we need to delete the old file first
|
|
if (QFile::exists(new_file) && dialog_result_.overwrite_ != AlbumCoverExport::OverwriteMode::None) {
|
|
|
|
// if the mode is "overwrite smaller" then skip the cover if a bigger one is already available in the folder
|
|
if (dialog_result_.overwrite_ == AlbumCoverExport::OverwriteMode::Smaller) {
|
|
QImage image_existing;
|
|
image_existing.load(new_file);
|
|
|
|
if (image_existing.isNull() || image_existing.size().height() >= image.size().height() || image_existing.size().width() >= image.size().width()) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (!QFile::remove(new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (image.save(new_file)) {
|
|
EmitCoverExported();
|
|
}
|
|
else {
|
|
EmitCoverSkipped();
|
|
}
|
|
|
|
}
|
|
|
|
// Exports a single album cover using a "copy file" approach.
|
|
void CoverExportRunnable::ExportCover() {
|
|
|
|
QImage image;
|
|
QString extension;
|
|
QString cover_path;
|
|
bool embedded_cover = false;
|
|
|
|
for (const AlbumCoverLoaderOptions::Type cover_type : cover_types_) {
|
|
switch (cover_type) {
|
|
case AlbumCoverLoaderOptions::Type::Unset:
|
|
if (song_.art_unset()) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Embedded:
|
|
if (song_.art_embedded() && dialog_result_.export_embedded_) {
|
|
image = TagReaderClient::Instance()->LoadEmbeddedArtAsImageBlocking(song_.url().toLocalFile());
|
|
if (!image.isNull()) {
|
|
embedded_cover = true;
|
|
extension = QStringLiteral("jpg");
|
|
}
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Manual:
|
|
if (dialog_result_.export_downloaded_ && song_.art_manual_is_valid()) {
|
|
cover_path = song_.art_manual().toLocalFile();
|
|
if (image.load(cover_path)) {
|
|
extension = cover_path.section('.', -1);
|
|
}
|
|
}
|
|
break;
|
|
case AlbumCoverLoaderOptions::Type::Automatic:
|
|
if (dialog_result_.export_downloaded_ && song_.art_automatic_is_valid()) {
|
|
cover_path = song_.art_automatic().toLocalFile();
|
|
if (image.load(cover_path)) {
|
|
extension = cover_path.section('.', -1);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
if (!image.isNull() && !extension.isEmpty() && (embedded_cover || !cover_path.isEmpty())) break;
|
|
}
|
|
|
|
if (image.isNull() || extension.isEmpty()) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
|
|
QString cover_dir = song_.url().toLocalFile().section('/', 0, -2);
|
|
QString new_file = cover_dir + '/' + dialog_result_.filename_ + '.' + extension;
|
|
|
|
// If the file exists, do not override!
|
|
if (dialog_result_.overwrite_ == AlbumCoverExport::OverwriteMode::None && QFile::exists(new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
|
|
// We're handling overwrite as remove + copy so we need to delete the old file first
|
|
if (dialog_result_.overwrite_ != AlbumCoverExport::OverwriteMode::None && QFile::exists(new_file)) {
|
|
if (!QFile::remove(new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (embedded_cover) {
|
|
if (!image.save(new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
// Automatic or manual cover, available in an image file
|
|
if (!QFile::copy(cover_path, new_file)) {
|
|
EmitCoverSkipped();
|
|
return;
|
|
}
|
|
}
|
|
|
|
EmitCoverExported();
|
|
|
|
}
|
|
|
|
void CoverExportRunnable::EmitCoverExported() { emit CoverExported(); }
|
|
|
|
void CoverExportRunnable::EmitCoverSkipped() { emit CoverSkipped(); }
|