feedc0de/qt-creator
1
0
Fork 0
forked from qt-creator/qt-creator
Code Pull Requests Activity

TreeModel: Make TreeModelVisitor useful

Browse Source 
Functions that are meant to be overridden should be virtual.
Also add a convience level() function, seems to be a common need.

Change-Id: I09d452ab12f0ac29fb31de853057511418a47e7e
Reviewed-by: Christian Stenger <christian.stenger@theqtcompany.com>
This commit is contained in:
hjk
2015-02-06 21:00:56 +01:00
parent 10f36eada5
commit a4b2801fed
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/libs/utils/treemodel.cpp
View File

@@ -777,10 +777,13 @@ void TreeItem::setModel(TreeModel *model)
void TreeItem::walkTree(TreeItemVisitor *visitor) void TreeItem::walkTree(TreeItemVisitor *visitor)
{ {
if (visitor->preVisit(this)) { if (visitor->preVisit(this)) {
++visitor->m_level;
visitor->visit(this); visitor->visit(this);
foreach (TreeItem *item, m_children) foreach (TreeItem *item, m_children)
item->walkTree(visitor); item->walkTree(visitor);
--visitor->m_level;
} }
visitor->postVisit(this);
} }
void TreeItem::walkTree(std::function<void (TreeItem *)> f) void TreeItem::walkTree(std::function<void (TreeItem *)> f)

13
src/libs/utils/treemodel.h
View File

@@ -49,11 +49,18 @@ class TreeModel;
class QTCREATOR_UTILS_EXPORT TreeItemVisitor class QTCREATOR_UTILS_EXPORT TreeItemVisitor
{ {
public: public:
TreeItemVisitor() {} TreeItemVisitor() : m_level(0) {}
virtual ~TreeItemVisitor() {} virtual ~TreeItemVisitor() {}
bool preVisit(TreeItem *) { return true; } virtual bool preVisit(TreeItem *) { return true; }
void visit(TreeItem *) {} virtual void visit(TreeItem *) {}
virtual void postVisit(TreeItem *) {}
int level() const { return m_level; }
private:
friend class TreeItem;
int m_level;
}; };
class QTCREATOR_UTILS_EXPORT TreeItem class QTCREATOR_UTILS_EXPORT TreeItem