81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
#include "CategorizedView.h"
|
|
#include <QApplication>
|
|
#include <QStandardItemModel>
|
|
#include <QPainter>
|
|
|
|
#include "CategorizedProxyModel.h"
|
|
|
|
QPixmap icon(const Qt::GlobalColor color)
|
|
{
|
|
QPixmap p = QPixmap(32, 32);
|
|
p.fill(QColor(color));
|
|
return p;
|
|
}
|
|
QPixmap icon(const int number)
|
|
{
|
|
QPixmap p = icon(Qt::white);
|
|
QPainter painter(&p);
|
|
QFont font = painter.font();
|
|
font.setBold(true);
|
|
font.setPixelSize(28);
|
|
painter.setFont(font);
|
|
painter.drawText(QRect(QPoint(0, 0), p.size()), Qt::AlignVCenter | Qt::AlignHCenter, QString::number(number));
|
|
painter.end();
|
|
return p;
|
|
}
|
|
QStandardItem *createItem(const Qt::GlobalColor color, const QString &text, const QString &category)
|
|
{
|
|
QStandardItem *item = new QStandardItem;
|
|
item->setText(text);
|
|
item->setData(icon(color), Qt::DecorationRole);
|
|
item->setData(category, CategorizedView::CategoryRole);
|
|
item->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
|
|
return item;
|
|
}
|
|
QStandardItem *createItem(const int index, const QString &category)
|
|
{
|
|
QStandardItem *item = new QStandardItem;
|
|
item->setText(QString("Item #%1").arg(index));
|
|
item->setData(icon(index), Qt::DecorationRole);
|
|
item->setData(category, CategorizedView::CategoryRole);
|
|
item->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
|
|
return item;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
QApplication a(argc, argv);
|
|
|
|
QStandardItemModel model;
|
|
model.setRowCount(10);
|
|
model.setColumnCount(1);
|
|
|
|
model.setItem(0, createItem(Qt::red, "Red", "Colorful"));
|
|
model.setItem(1, createItem(Qt::blue, "Blue", "Colorful"));
|
|
model.setItem(2, createItem(Qt::yellow, "Yellow", "Colorful"));
|
|
|
|
model.setItem(3, createItem(Qt::black, "Black", "Not Colorful"));
|
|
model.setItem(4, createItem(Qt::darkGray, "Dark Gray", "Not Colorful"));
|
|
model.setItem(5, createItem(Qt::gray, "Gray", "Not Colorful"));
|
|
model.setItem(6, createItem(Qt::lightGray, "Light Gray", "Not Colorful"));
|
|
model.setItem(7, createItem(Qt::white, "White", "Not Colorful"));
|
|
|
|
model.setItem(8, createItem(Qt::darkGreen, "Dark Green", ""));
|
|
model.setItem(9, createItem(Qt::green, "Green", ""));
|
|
|
|
for (int i = 0; i < 21; ++i)
|
|
{
|
|
model.setItem(i + 10, createItem(i+1, "Items 1-20"));
|
|
}
|
|
|
|
CategorizedProxyModel pModel;
|
|
pModel.setSourceModel(&model);
|
|
|
|
CategorizedView w;
|
|
w.setModel(&pModel);
|
|
w.resize(640, 480);
|
|
w.show();
|
|
|
|
return a.exec();
|
|
}
|