Add drawBranches() to QtGraphicsTreeViewItem to separate drawing the branches from the main paint() function.

1 files changed, 2 insertions, 0 deletions

diff --git a/src/qgraphicstreeview.h b/src/qgraphicstreeview.h
index 83bc388..55cf522 100644
--- a/src/qgraphicstreeview.h
+++ b/src/qgraphicstreeview.h
@@ -27,6 +27,7 @@
 #include "qtreemodelinterface.h"
 
 #include <QtCore/qsize.h>
+#include <QtCore/qstack.h>
 #include <QtGui/qtransform.h>
 #include <QtGui/qgraphicswidget.h>
 #include <QtGui/qstyleoption.h>
@@ -67,6 +68,7 @@ public:
     virtual QSizeF sizeHint(const QTreeModelIterator &it, int column, const QStyleOptionViewItemV4 *option, Qt::SizeHint which, const QSizeF &constraint = QSizeF()) const;
 
     virtual void paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget = 0);
+    virtual void paintBranches(QPainter *painter, QStyleOptionViewItemV4 *option, const QStack<QTreeModelIterator> &stack);
     virtual void itemChanged(const QList<int> &roles = QList<int>());
 
     QHash<int, QVariant> data(int column = 0, const QList<int> &roles = QList<int>()) const;