From d3fe3494fef7b43556bfd8bd3b2128a3e0dc623d Mon Sep 17 00:00:00 2001
From: Tomaz Canabrava <tcanabrava@kde.org>
Date: Fri, 8 Nov 2013 21:08:14 -0200
Subject: Save / Restore the Tree after organizing by other columns.

This patch saves and restores the state of the TreeView after the user
clicks on another columns that will make the tree disappear. All of the
branche states are saved, this way the expanded nodes will be restored
when the user gets back to tree mode.

Signed-off-by: Tomaz Canabrava <tcanabrava@kde.org>
Signed-off-by: Dirk Hohndel <dirk@hohndel.org>
---
 qt-ui/divelistview.cpp | 40 ++++++++++++++++++++++++----------------
 1 file changed, 24 insertions(+), 16 deletions(-)

(limited to 'qt-ui/divelistview.cpp')

diff --git a/qt-ui/divelistview.cpp b/qt-ui/divelistview.cpp
index 1217e0d25..b2f33ffa1 100644
--- a/qt-ui/divelistview.cpp
+++ b/qt-ui/divelistview.cpp
@@ -66,15 +66,8 @@ DiveListView::~DiveListView()
 void DiveListView::setupUi(){
 	QSettings settings;
 	static bool firstRun = true;
-	QList<int> expandedColumns;
-	if(!firstRun){
-		for(int i = 0; i < model()->rowCount(); i++){
-			if(isExpanded( model()->index(i, 0) )){
-				expandedColumns.push_back(i);
-			}
-		}
-	}
-
+	if(firstRun)
+		backupExpandedRows();
 	settings.beginGroup("ListWidget");
 	/* if no width are set, use the calculated width for each column;
 	 * for that to work we need to temporarily expand all rows */
@@ -89,18 +82,27 @@ void DiveListView::setupUi(){
 			setColumnWidth(i, 100);
 	}
 	settings.endGroup();
+	if(firstRun)
+		restoreExpandedRows();
+	else
+		collapseAll();
+	firstRun = false;
+}
 
-	if(firstRun){
-		Q_FOREACH(const int &i, expandedColumns){
-			setExpanded( model()->index(i, 0), true );
+void DiveListView::backupExpandedRows(){
+	expandedRows.clear();
+	for(int i = 0; i < model()->rowCount(); i++){
+		if(isExpanded( model()->index(i, 0) )){
+			expandedRows.push_back(i);
 		}
-	}else{
-		collapseAll();
 	}
-
-	firstRun = false;
 }
 
+void DiveListView::restoreExpandedRows(){
+	Q_FOREACH(const int &i, expandedRows){
+		setExpanded( model()->index(i, 0), true );
+	}
+}
 void DiveListView::fixMessyQtModelBehaviour()
 {
 	QAbstractItemModel *m = model();
@@ -184,9 +186,15 @@ void DiveListView::headerClicked(int i)
 		sortByColumn(i, currentOrder);
 	} else {
 		// clear the model, repopulate with new indexes.
+		if(currentLayout == DiveTripModel::TREE){
+			backupExpandedRows();
+		}
 		reload(newLayout, false);
 		currentOrder = Qt::DescendingOrder;
 		sortByColumn(i, currentOrder);
+		if (newLayout == DiveTripModel::TREE){
+			restoreExpandedRows();
+		}
 	}
 
 	// repopulate the selections.
-- 
cgit v1.2.3-70-g09d2