QML TreeView 自定义样式加上节点可拖动

#include "TreeModel.h" #if _MSC_VER >= 1600 #pragma execution_character_set("utf-8"); #endif TreeModel::TreeModel(QObject *parent) : QAbstractItemModel(parent) { int dataNum[20] = {0,1,2,3,4,5,6,7,8}; m_rootItem = new TreeItem({ "所有通道" },dataNum+0); auto item1 = new TreeItem({ "四川省", "b" },dataNum+1); auto item2 = new TreeItem({ "标签", "bb" },dataNum+2); auto item3 = new TreeItem({ "下属机构","bbb"},dataNum+3); auto item4 = new TreeItem({ "熊猫","b1" },dataNum+4); auto item5 = new TreeItem({ "美食","b1" },dataNum+9); QList<QString> names; names.push_back("成都"); names.push_back("绵阳"); names.push_back("南充"); names.push_back("乐山"); QList<QString> foods; foods.push_back("火锅"); foods.push_back("水煮鱼"); foods.push_back("回锅肉"); foods.push_back("麻婆豆腐"); m_rootItem->appendChild(item1); item1->appendChild(item2); item1->appendChild(item3); item1->appendChild(item5); item2->appendChild(item4); for (int i= 0;i<names.count();i++) { QString name = names[i]; item3->appendChild(new TreeItem({name},dataNum+i+5)); QString food=foods[i]; item5->appendChild(new TreeItem({food},dataNum+i+10)); } } TreeModel::~TreeModel() { delete m_rootItem; } int TreeModel::columnCount(const QModelIndex &parent) const { if (parent.isValid()) { return static_cast<TreeItem*>(parent.internalPointer())->columnCount(); } else { return m_rootItem->columnCount(); } } QHash<int, QByteArray> TreeModel::roleNames() const { /* roleName可以自己定义，我这里是随便写的 */ QHash<int, QByteArray> names(QAbstractItemModel::roleNames()); names[12345] = "text"; names[12346] = "text2"; return names; } QVariant TreeModel::data(const QModelIndex &index, int role) const { if (!index.isValid()) { return QVariant(); } switch (role) { case 12345: { return static_cast<TreeItem*>(index.internalPointer())->data(0); } case 12346: { return static_cast<TreeItem*>(index.internalPointer())->data(1); } case Qt::DisplayRole: { return static_cast<TreeItem*>(index.internalPointer())->data(index.column()); } default: { return QVariant(); } } } Qt::ItemFlags TreeModel::flags(const QModelIndex &index) const { if (!index.isValid()) { return 0; } return Qt::ItemIsEnabled | Qt::ItemIsSelectable; } QVariant TreeModel::headerData(int section, Qt::Orientation orientation, int role) const { if (orientation == Qt::Horizontal && role == Qt::DisplayRole) { return m_rootItem->data(section); } return QVariant(); } QModelIndex TreeModel::index(int row, int column, const QModelIndex &parent) const { if (!hasIndex(row, column, parent)) { return QModelIndex(); } TreeItem *parentItem; if (!parent.isValid()) { parentItem = m_rootItem; } else { parentItem = static_cast<TreeItem*>(parent.internalPointer()); } TreeItem *childItem = parentItem->child(row); if (childItem) { return createIndex(row, column, childItem); } else { return QModelIndex(); } } QModelIndex TreeModel::parent(const QModelIndex &index) const { if (!index.isValid()) { return QModelIndex(); } TreeItem *childItem = static_cast<TreeItem*>(index.internalPointer()); TreeItem *parentItem = childItem->parent(); if (parentItem == m_rootItem) { return QModelIndex(); } return createIndex(parentItem->row(), 0, parentItem); } int TreeModel::rowCount(const QModelIndex &parent) const { TreeItem *parentItem; if (parent.column() > 0) { return 0; } if (!parent.isValid()) { parentItem = m_rootItem; } else { parentItem = static_cast<TreeItem*>(parent.internalPointer()); } return parentItem->childCount(); } QAbstractItemModel *TreeModel::model() { return this; } int TreeModel::getData(const QModelIndex &index) { void* a= static_cast<TreeItem*>(index.internalPointer())->getRecordData(); int data = *(int*)a; return data; }