Fixing several d&d bugs

1 files changed, 27 insertions, 0 deletions

diff --git a/main.cpp b/main.cpp
index 24d7075e..427ae17a 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,6 +1,7 @@
 #include "CategorizedView.h"
 #include <QApplication>
 #include <QStandardItemModel>
+#include <QPainter>
 
 #include "CategorizedProxyModel.h"
 
@@ -10,6 +11,18 @@ QPixmap icon(const Qt::GlobalColor color)
 	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;
@@ -19,6 +32,15 @@ QStandardItem *createItem(const Qt::GlobalColor color, const QString &text, cons
 	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[])
 {
@@ -41,6 +63,11 @@ int main(int argc, char *argv[])
 	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);