http://download.csdn.net/detail/liuguangzhou123/5175389
模型主要代码如下：
//TreeModel.h
#ifndef
TREEMODEL_H
#define TREEMODEL_H
#include <QAbstractItemModel>
#include <QStandardItemModel>
#include "DevRoot.h"
class TreeModel : public QAbstractItemModel
{
Q_OBJECT
public:
TreeModel(QObject *parent = 0);
~TreeModel();
QVariant data(const QModelIndex &index, int role) const;
Qt::ItemFlags flags(const QModelIndex &index) const;
QVariant headerData(int section, Qt::Orientation orientation,
int role = Qt::DisplayRole) const;
QModelIndex index(int row, int column,
const QModelIndex &parent = QModelIndex()) const;
QModelIndex parent(const QModelIndex &index) const;
int rowCount(const QModelIndex &parent = QModelIndex()) const;
int columnCount(const QModelIndex &parent = QModelIndex()) const;
private:
void setupModelData(/*const QStringList &lines, TreeItem *parent*/);
CDevRoot *m_pDevRoot;
};
#endif // TREEMODEL_H
//TreeModel.cpp
#include
"TreeModel.h"
#include <QStringList>
#include <qDebug>
#include "NodeAA.h"
#include "NodeBB.h"
#include "NodeCC.h"
TreeModel::TreeModel(QObject *parent)
: QAbstractItemModel(parent)
{
m_pDevRoot = new CDevRoot("根目录");
setupModelData();
}
TreeModel::~TreeModel()
{
delete m_pDevRoot;
}
QVariant TreeModel::data(const QModelIndex &index, int role) const
{
if (!index.isValid())
{
return QVariant();
}
else if (role == Qt::DisplayRole)
{
CNodeBase *item = static_cast<CNodeBase*>(index.internalPointer());
return item->getName();
}
else if(role == Qt::DecorationRole)
{
CNodeBase *item = static_cast<CNodeBase*>(index.internalPointer());
if(item->getNodeType() == MAX_LEN_NODE_NAME)
return QIcon(":/images/window.bmp");
return QIcon(":/images/test.png");
}
else if(role == Qt::CheckStateRole)
{
}
else if(role == Qt::UserRole)
{
}
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_pDevRoot->getName();
return QVariant();
}
QModelIndex TreeModel::index(int row, int column, const QModelIndex &parent) const
{
if (!hasIndex(row, column, parent))
return QModelIndex();
CNodeBase *pNodeBase;
if (!parent.isValid())//ROOT下挂装置NodeMac
{
CNodeBase *pNodeMac = m_pDevRoot->getNodeMac(row);
return createIndex(row, column, pNodeMac);
}
pNodeBase = static_cast<CNodeBase*>(parent.internalPointer());
if (pNodeBase->getNodeType() == NODE_TYPE_MAC && row == 0)//装置NodeMac下挂装置AA
{
CNodeMac *pNodeMac = static_cast<CNodeMac*>(parent.internalPointer());
CNodeAA *P = pNodeMac->getChildDeviceAA();
return createIndex(row,column,P);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_MAC && row == 1)//装置NodeMac下挂装置BB
{
CNodeMac *pNodeMac = static_cast<CNodeMac*>(parent.internalPointer());
CNodeBB *pNodeDeviceGroupFolder = pNodeMac->getChildDeviceBB();
return createIndex(row,column,pNodeDeviceGroupFolder);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_MAC && row == 2)//装置NodeMac下挂装置CC
{
CNodeMac *pNodeMac = static_cast<CNodeMac*>(parent.internalPointer());
CNodeCC *pNodeDeviceGroupFolder = pNodeMac->getChildDeviceCC();
return createIndex(row,column,pNodeDeviceGroupFolder);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_AA)//装置AA下挂装置AA-CHILD
{
CNodeAA *pNodeDeviceFolder = static_cast<CNodeAA*>(parent.internalPointer());
CNodeAAChildA *pNodeBase = pNodeDeviceFolder->getChildDevice(row);
return createIndex(row,column,pNodeBase);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_AA_CHILD)//装置AA-CHILD下挂装置
{
CNodeAAChildA *pNodeDeviceFolder = static_cast<CNodeAAChildA*>(parent.internalPointer());
CNodeBase *pNodeBase = pNodeDeviceFolder->getChildDevice(row);
return createIndex(row,column,pNodeBase);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_BB)//装置BB下挂装置BB-CHILD
{
CNodeBB *pNodeDeviceGroupFolder = static_cast<CNodeBB*>(parent.internalPointer());
CNodeBase *pNodeBase = pNodeDeviceGroupFolder->getChildDeviceGroup(row);
return createIndex(row,column,pNodeBase);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_CC)//装置CC下挂装置CC-CHILD
{
CNodeCC *pNodeDeviceGroupFolder = static_cast<CNodeCC*>(parent.internalPointer());
CNodeCCChild *pNodeBase = pNodeDeviceGroupFolder->getChildDevice(row);
return createIndex(row,column,pNodeBase);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_CC_CHILD)//装置CC-CHILD下挂装置CC-CHILD-CHILD
{
CNodeCCChild *pNodeDeviceGroupFolder = static_cast<CNodeCCChild*>(parent.internalPointer());
CNodeCCChildChild *pNodeBase = pNodeDeviceGroupFolder->getChildDevice(row);
return createIndex(row,column,pNodeBase);
}
else if (pNodeBase->getNodeType() == NODE_TYPE_CC_CHILD_CHILD)//装置CC-CHILD-CHILD下挂装置
{
CNodeCCChildChild *pNodeDeviceGroupFolder = static_cast<CNodeCCChildChild*>(parent.internalPointer());
CNodeBase *pNodeBase = pNodeDeviceGroupFolder->getChildDevice(row);
return createIndex(row,column,pNodeBase);
}
return QModelIndex();
}
QModelIndex TreeModel::parent(const QModelIndex &index) const
{
if (!index.isValid())
{
return QModelIndex();
}
CNodeBase *childItem = static_cast<CNodeBase*>(index.internalPointer());
CNodeBase *parentItem = childItem->getParent();
if (parentItem->getNodeType() == NODE_TYPE_MAC_MANAGER)//设备管理
{
return QModelIndex();
}
else if(parentItem->getNodeType() == NODE_TYPE_MAC)//设备
{
return createIndex(0, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_AA)//设备下挂装置AA
{
return createIndex(0, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_AA_CHILD)//装置AA下挂装置
{
return createIndex(0, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_BB)//设备下挂装置BB
{
return createIndex(1, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_CC)//设备下挂装置CC
{
return createIndex(2, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_CC_CHILD)//装置CC下挂装置CC-CHILD
{
return createIndex(0, 0, parentItem);
}
else if(parentItem->getNodeType() == NODE_TYPE_CC_CHILD_CHILD)//装置CC-CHILD-CHILD下挂装置
{
return createIndex(0, 0, parentItem);
}
return QModelIndex();
}
int TreeModel::rowCount(const QModelIndex &parent) const
{
CNodeBase *parentItem;
if (parent.column() > 0)
{
return 0;
}
if (!parent.isValid())
{
return m_pDevRoot->getMacCount();
}
parentItem = static_cast<CNodeBase*>(parent.internalPointer());
if(parentItem->getNodeType() == NODE_TYPE_MAC)//设备下挂数量
{
CNodeMac *p = static_cast<CNodeMac*>(parent.internalPointer());
return