From e805c893c6338795f8c7f02724ba23d4d9209dea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jacob=20St=C3=B8ren?= <jacob.storen@ceetronSolutions.com>
Date: Fri, 14 Aug 2015 18:03:51 +0200
Subject: [PATCH] (#378) Added an tree view expansion interface

---
 Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.cpp       | 8 ++++++++
 Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.h         | 1 +
 Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp | 9 +++++++++
 Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h   | 3 ++-
 4 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.cpp b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.cpp
index 129784dc9a..f6b60e4352 100644
--- a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.cpp
+++ b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.cpp
@@ -147,5 +147,13 @@ void PdmUiTreeView::selectAsCurrentItem(PdmUiItem* uiItem)
     m_treeViewEditor->selectAsCurrentItem(uiItem);
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void PdmUiTreeView::setExpanded(const PdmUiItem* uiItem, bool doExpand) const 
+{
+    m_treeViewEditor->setExpanded(uiItem, doExpand);
+}
+
 } //End of namespace caf
 
diff --git a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.h b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.h
index 7fbc6c933e..386f47915c 100644
--- a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.h
+++ b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeView.h
@@ -71,6 +71,7 @@ public:
 
     void        selectedObjects(std::vector<PdmUiItem*>& objects);
     void        selectAsCurrentItem(PdmUiItem* uiItem);
+    void        setExpanded(const PdmUiItem* uiItem, bool doExpand) const ;
 
 signals:
     void        selectionChanged();
diff --git a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp
index 9a0b267b1a..5955cb79dd 100644
--- a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp
+++ b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.cpp
@@ -265,5 +265,14 @@ void PdmUiTreeViewEditor::slotOnSelectionChanged(const QItemSelection & selected
     emit selectionChanged();
 }
 
+//--------------------------------------------------------------------------------------------------
+/// 
+//--------------------------------------------------------------------------------------------------
+void PdmUiTreeViewEditor::setExpanded(const PdmUiItem* uiItem, bool doExpand) const
+{
+    QModelIndex index = m_treeViewModel->findModelIndex(uiItem);
+    m_treeView->setExpanded(index, doExpand);
+}
+
 
 } // end namespace caf
diff --git a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h
index c474477087..fe1048c6fc 100644
--- a/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h
+++ b/Fwk/AppFwk/cafUserInterface/cafPdmUiTreeViewEditor.h
@@ -89,7 +89,8 @@ public:
 
     void                selectAsCurrentItem(PdmUiItem* uiItem);
     void                selectedUiItems(std::vector<PdmUiItem*>& objects);
-    
+    void                setExpanded(const PdmUiItem* uiItem, bool doExpand) const;
+
     QWidget*            createWidget(QWidget* parent);
 
 signals: