qDeleteAll(childItems);
}
+uint TreeItem::id() const {
+ return (uint)this;
+}
+
void TreeItem::appendChild(TreeItem *item) {
childItems.append(item);
+ childHash[item->id()] = item;
}
void TreeItem::removeChild(int row) {
+ if(row >= childItems.size())
+ return;
+ TreeItem *treeitem = childItems.value(row);
childItems.removeAt(row);
+ childHash.remove(childHash.key(treeitem));
}
-TreeItem *TreeItem::child(int row) {
- return childItems.value(row);
+TreeItem *TreeItem::child(int row) const {
+ if(row < childItems.size())
+ return childItems.value(row);
+ else
+ return 0;
+}
+
+TreeItem *TreeItem::childById(const uint &id) const {
+ if(childHash.contains(id))
+ return childHash.value(id);
+ else
+ return 0;
}
int TreeItem::childCount() const {
return QVariant();
}
+Qt::ItemFlags TreeItem::flags() const {
+ // some sane defaults
+ return Qt::ItemIsSelectable | Qt::ItemIsEnabled;
+}
/*****************************************
* TreeModel
*****************************************/
-TreeModel::TreeModel(const QList<QVariant> &data, QObject *parent) : QAbstractItemModel(parent) {
+TreeModel::TreeModel(const QList<QVariant> &data, QObject *parent)
+ : QAbstractItemModel(parent)
+{
rootItem = new TreeItem(data, 0);
}
return QModelIndex();
}
+QModelIndex TreeModel::indexById(uint id, const QModelIndex &parent) const {
+ TreeItem *parentItem;
+
+ if(!parent.isValid())
+ parentItem = rootItem;
+ else
+ parentItem = static_cast<TreeItem *>(parent.internalPointer());
+
+ TreeItem *childItem = parentItem->childById(id);
+ if(childItem)
+ return createIndex(childItem->row(), 0, childItem);
+ else
+ return QModelIndex();
+}
+
QModelIndex TreeModel::parent(const QModelIndex &index) const {
if(!index.isValid())
return QModelIndex();
}
Qt::ItemFlags TreeModel::flags(const QModelIndex &index) const {
+ TreeItem *item;
if(!index.isValid())
- return 0;
+ item = rootItem;
else
- return Qt::ItemIsEnabled | Qt::ItemIsSelectable;
+ item = static_cast<TreeItem *>(index.internalPointer());
+ return item->flags();
}
QVariant TreeModel::headerData(int section, Qt::Orientation orientation, int role) const {
}
bool TreeModel::removeRow(int row, const QModelIndex &parent) {
+ if(row > rowCount(parent))
+ return false;
+
+ TreeItem *item;
+ if(!parent.isValid())
+ item = rootItem;
+ else
+ item = static_cast<TreeItem*>(parent.internalPointer());
+
beginRemoveRows(parent, row, row);
- TreeItem *item = static_cast<TreeItem*>(parent.internalPointer());
item->removeChild(row);
endRemoveRows();
return true;
}
+bool TreeModel::removeRows(int row, int count, const QModelIndex &parent) {
+ // check if there is work to be done
+ if(count == 0)
+ return true;
+
+ // out of range check
+ if(row + count - 1 > rowCount(parent) || row < 0 || count < 0)
+ return false;
+
+ TreeItem *item;
+ if(!parent.isValid())
+ item = rootItem;
+ else
+ item = static_cast<TreeItem*>(parent.internalPointer());
+
+
+ beginRemoveRows(parent, row, row + count - 1);
+
+ for(int i = row + count - 1; i >= 0; i--) {
+ item->removeChild(i);
+ }
+ endRemoveRows();
+ return true;
+}
+
+void TreeModel::clear() {
+ removeRows(0, rowCount());
+}