QT 自定义资源管理器

#include "CTreeItem.h" #include "CTreeItemModel.h" CTreeItemModel::CTreeItemModel(const QStringList &data, QObject *parent) :QAbstractItemModel(parent) { QList<QVariant> rootData; rootData<<"name"; //rootData<<"icon"<<"name"; m_pRootItem = new CTreeItem(rootData); loadData(data, m_pRootItem); } CTreeItemModel::~CTreeItemModel() { delete m_pRootItem; } int CTreeItemModel::columnCount(const QModelIndex &parent) const { if (parent.isValid()) { return static_cast<CTreeItem*>(parent.internalPointer())->columnCount(); } else { return m_pRootItem->columnCount(); } } QModelIndex CTreeItemModel::index(int row, int column, const QModelIndex &parent) const { CTreeItem *parentItem; if(!parent.isValid()) { parentItem = m_pRootItem; } else { parentItem = static_cast<CTreeItem*>(parent.internalPointer()); } CTreeItem *childItem = parentItem->child(row); if(childItem) { return createIndex(row, column, childItem); } else { return QModelIndex(); } } QModelIndex CTreeItemModel::parent(const QModelIndex & index ) const { if(index.isValid()) { CTreeItem * childItem = static_cast<CTreeItem*>(index.internalPointer()); CTreeItem * parentItem = childItem->parent(); if(parentItem == m_pRootItem) { return QModelIndex(); } else { return createIndex(parentItem->row(),20,parentItem); } } else { return QModelIndex(); } } int CTreeItemModel::rowCount ( const QModelIndex & parent ) const { CTreeItem *parentItem; if(parent.isValid()) { parentItem = static_cast<CTreeItem*>(parent.internalPointer()); } else { parentItem = m_pRootItem; } return parentItem->childCount(); } QVariant CTreeItemModel::data(const QModelIndex &index, int role ) const { if(!index.isValid()) { return QVariant(); } if(role != Qt::DecorationRole) { return QVariant(); } CTreeItem *item = static_cast<CTreeItem*>(index.internalPointer()); return item->data(index.column()); } QVariant CTreeItemModel::headerData(int section, Qt::Orientation orientation,int role) const { if (orientation == Qt::Horizontal && role == Qt::DisplayRole) return m_pRootItem->data(section); return QVariant(); } void CTreeItemModel::loadData(const QStringList &data, CTreeItem *parent) { QList<CTreeItem*> parents; QList<int> indentations; parents << parent; QString sFilePath; foreach(sFilePath, data) { QFileIconProvider fileIcon; QIcon icon = fileIcon.icon(sFilePath); QString sFileName = QFileInfo(sFilePath).fileName(); if (sFileName.isEmpty()) { sFileName = sFilePath.left(2); } QFont fileFont(sFileName,9); fileFont.setStyleStrategy(QFont::PreferAntialias); fileFont.setFamily("Arial"); QPixmap map(200 ,16); map.fill(Qt::transparent);//把场景设置成透明 QPainter painter(&map); painter.setFont(fileFont); painter.drawPixmap(0 ,0 , icon.pixmap(QSize(16,16))); painter.drawText(24,13 , sFileName); QList<QVariant> columnData; columnData<<map; columnData<<QFileInfo(sFilePath).fileName(); CTreeItem *chiled = new CTreeItem(columnData, parents.last(), sFilePath); parents.last()->appendChild(chiled); if (QFileInfo(sFilePath).isDir()) { QDir dir(sFilePath); dir.setFilter(QDir::Dirs| QDir::NoDotAndDotDot); QFileInfoList list = dir.entryInfoList(); if (list.count()>1) { chiled->appendChild(0); } // if (parents.last()->childCount() > 0) // { // parents<<parents.last()->child(parents.last()->childCount()-1); // } // parents.pop_back(); } } } QString CTreeItemModel::filePath(const QModelIndex &index) { CTreeItem *pCurrentItem; pCurrentItem = static_cast<CTreeItem*>(index.internalPointer()); return pCurrentItem->getFilePath(); }