Add custom roles to the post model and export them to QML

This commit is contained in:
Luca Beltrame 2015-02-05 00:29:41 +01:00
parent 4fbfdf753e
commit b0185c136b
2 changed files with 50 additions and 0 deletions

View file

@ -98,6 +98,31 @@ QVariant DanbooruPostModel::data(const QModelIndex &index, int role) const
return post->fileUrl().fileName();
}
if (role == DanbooruPostModel::PixmapRole) {
QPixmap pixmap = post->pixmap();
return pixmap;
}
if (role == FileUrlRole) {
return post->fileUrl();
}
if (role == ThumbUrlRole) {
return post->thumbnailUrl();
}
if (role == SizeRole) {
return post->size();
}
if (role == ResolutionRole) {
return QSize(post->width(), post->height());
}
if (role == RatingRole) {
return post->rating();
}
return QVariant();
}
@ -115,4 +140,18 @@ void DanbooruPostModel::clear()
reset();
}
QHash< int, QByteArray > DanbooruPostModel::roleNames() const
{
QHash<int, QByteArray> roles;
roles[PixmapRole] = "thumbPix";
roles[FileUrlRole] = "fileUrl";
roles[ThumbUrlRole] = "thumbnailUrl";
roles[RatingRole] = "rating";
roles[TagRole] = "tags";
roles[SizeRole] = "fileSize";
roles[ResolutionRole] = "resolution";
return roles;
}
};

View file

@ -59,6 +59,17 @@ public:
int rowCount(const QModelIndex &parent = QModelIndex()) const;
QVariant data(const QModelIndex &index, int role) const;
void clear();
QHash<int, QByteArray> roleNames() const;
enum PostRoles {
PixmapRole = Qt::UserRole + 1000,
FileUrlRole = Qt::UserRole + 1001,
ThumbUrlRole = Qt::UserRole + 1002,
SizeRole = Qt::UserRole + 1003,
RatingRole = Qt::UserRole + 1004,
TagRole = Qt::UserRole + 1005,
ResolutionRole = Qt::UserRole + 1006
};
private:
QVector<DanbooruPost *> m_items;