danbooru-client/src/libdanbooru/danboorupost.cpp

171 lines
4 KiB
C++

/*
* Copyright 2015 Luca Beltrame <lbeltrame@kde.org>
*
* This file is part of Danbooru Client.
*
* Danbooru Client 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.
*
* Danbooru Client 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 Danbooru Client. If not, see <http://www.gnu.org/licenses/>.
*/
// Qt
#include <QtCore/QString>
#include <QtGui/QPixmap>
// KDE
#include <QUrl>
// Own
#include "danboorupost.h"
namespace Danbooru
{
const QMap<QString, DanbooruPost::Rating> DanbooruPost::RATING_MAP = initRatingMap();
DanbooruPost::DanbooruPost(QObject *parent): QObject(parent)
{
m_pixmap = QPixmap();
m_tags = QSet<QString>();
m_height = 0;
m_width = 0;
m_url = QUrl();
m_thumbnailUrl = QUrl();
m_size = 0;
m_rating = Questionable;
}
DanbooruPost::DanbooruPost(QVariantMap postData, QPixmap pixmap,
QObject *parent):
QObject(parent),
m_pixmap(pixmap)
{
m_tags = QSet<QString>::fromList(postData.value("tags").toString().split(' '));
m_id = postData.value("id").toString().toInt();
m_height = postData.value("height").toString().toInt();
m_width = postData.value("width").toString().toInt();
m_url = postData.value("file_url").toUrl();
m_thumbnailUrl = postData.value("preview_url").toUrl();
m_size = postData.value("file_size").toInt();
m_rating = RATING_MAP.value(postData.value("rating").toString());
m_sampleUrl = postData.value("sample_url").toUrl();
}
DanbooruPost::DanbooruPost(QXmlStreamAttributes &postData, QPixmap pixmap, QObject *parent):
QObject(parent),
m_pixmap(pixmap)
{
m_tags = QSet<QString>::fromList(postData.value("tags").toString().split(' '));
m_id = postData.value("id").toString().toInt();
m_height = postData.value("height").toString().toInt();
m_width = postData.value("width").toString().toInt();
m_url = QUrl::fromUserInput(postData.value("file_url").toString());
m_thumbnailUrl = QUrl::fromUserInput(postData.value("preview_url").toString());
m_sampleUrl = QUrl::fromUserInput(postData.value("sample_url").toString());
m_size = postData.value("file_size").toString().toInt();
m_rating = RATING_MAP.value(postData.value("rating").toString());
}
DanbooruPost::~DanbooruPost()
{
}
const QMap< QString, DanbooruPost::Rating > DanbooruPost::initRatingMap()
{
QMap<QString, DanbooruPost::Rating> map;
map.insert("s", Safe);
map.insert("q", Questionable);
map.insert("e", Explicit);
return map;
}
bool DanbooruPost::operator==(const Danbooru::DanbooruPost &other)
{
return m_url == other.m_url && m_id == other.m_id;
}
void DanbooruPost::setPixmap(const QPixmap &pixmap)
{
m_pixmap = pixmap;
}
const QString DanbooruPost::toString()
{
QString display = QString("Danbooru Post with ID %1 and URL %2, width %3 height %4 ");
display = display.arg(id()).arg(fileUrl().url()).arg(width()).arg(height());
return display;
}
int DanbooruPost::id() const
{
return m_id;
}
int DanbooruPost::height() const
{
return m_height;
}
int DanbooruPost::width() const
{
return m_width;
}
const QSet< QString > DanbooruPost::tags() const
{
return m_tags;
}
const QUrl DanbooruPost::thumbnailUrl() const
{
return m_thumbnailUrl;
}
const QUrl DanbooruPost::sampleUrl() const
{
return m_sampleUrl;
}
QPixmap DanbooruPost::pixmap() const
{
return m_pixmap;
}
int DanbooruPost::size() const
{
return m_size;
}
const QUrl DanbooruPost::fileUrl() const
{
return m_url;
}
Danbooru::DanbooruPost::Rating DanbooruPost::rating() const
{
return m_rating;
}
} // namespace Danbooru