1 files changed, 51 insertions, 0 deletions
diff --git a/Swift/QtUI/QtCheckBoxStyledItemDelegate.cpp b/Swift/QtUI/QtCheckBoxStyledItemDelegate.cpp
new file mode 100644
index 0000000..5f71ed4
--- /dev/null
+++ b/Swift/QtUI/QtCheckBoxStyledItemDelegate.cpp
@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) 2017 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
+ */
+
+#include <Swift/QtUI/QtCheckBoxStyledItemDelegate.h>
+
+#include <array>
+
+#include <QApplication>
+#include <QEvent>
+#include <QPainter>
+
+namespace Swift {
+
+QtCheckBoxStyledItemDelegate::QtCheckBoxStyledItemDelegate(QObject* parent) : QStyledItemDelegate(parent) {
+
+}
+
+void QtCheckBoxStyledItemDelegate::paint(QPainter* painter, const QStyleOptionViewItem& option, const QModelIndex& index) const {
+    QStyleOptionButton cbOpt;
+    cbOpt.rect = option.rect;
+    cbOpt.state = QStyle::State_Active;
+
+    auto copyFlags = std::array<QStyle::StateFlag, 2>({{QStyle::State_Enabled/*, QStyle::State_Sunken*/}});
+    for (auto flag : copyFlags) {
+        if (option.state && flag) {
+            cbOpt.state |= flag;
+        }
+    }
+
+    bool isChecked = index.data(DATA_ROLE).toBool();
+    if (isChecked) {
+        cbOpt.state |= QStyle::State_On;
+    }
+    else {
+        cbOpt.state |= QStyle::State_Off;
+    }
+    painter->fillRect(option.rect, option.state & QStyle::State_Selected ? option.palette.highlight() : option.palette.base());
+    QApplication::style()->drawControl(QStyle::CE_CheckBox, &cbOpt, painter);
+}
+
+bool QtCheckBoxStyledItemDelegate::editorEvent(QEvent* event, QAbstractItemModel* model, const QStyleOptionViewItem& /*option*/, const QModelIndex& index) {
+    if (event->type() == QEvent::MouseButtonRelease) {
+        model->setData(index, !index.data(DATA_ROLE).toBool(), DATA_ROLE);
+    }
+    return true;
+}
+
+};