Save the status of the tree before trying to reset the display.

This is a try: when the user changes the orientation on the columns
by clicking on the title of them and organizing by something that's
not the number, we lost track of the opened branches. this commit
doesn't fix that, but it fixes a possible similar issue.

Signed-off-by: Tomaz Canabrava <tcanabrava@kde.org>

1 files changed, 20 insertions, 4 deletions

diff --git a/qt-ui/divelistview.cpp b/qt-ui/divelistview.cpp
index 0a7c2f2e3..1217e0d25 100644
--- a/qt-ui/divelistview.cpp
+++ b/qt-ui/divelistview.cpp
@@ -65,6 +65,16 @@ 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);
+			}
+		}
+	}
+
 	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 */
@@ -77,12 +87,18 @@ void DiveListView::setupUi(){
 			setColumnWidth(i, width.toInt());
 		else
 			setColumnWidth(i, 100);
-
 	}
 	settings.endGroup();
-	collapseAll();
-	expand(model()->index(0,0));
-	scrollTo(model()->index(0,0), QAbstractItemView::PositionAtCenter);
+
+	if(firstRun){
+		Q_FOREACH(const int &i, expandedColumns){
+			setExpanded( model()->index(i, 0), true );
+		}
+	}else{
+		collapseAll();
+	}
+
+	firstRun = false;
 }
 
 void DiveListView::fixMessyQtModelBehaviour()