From f40de88f8fbe02aded668eb8bd8479fdc70bbe3d Mon Sep 17 00:00:00 2001
From: adityapk00 <adityapk@gmail.com>
Date: Tue, 30 Oct 2018 13:33:02 -0700
Subject: [PATCH] #17 Remember table column states as well

---
 src/mainwindow.cpp | 19 +++++++++++++++----
 src/mainwindow.h   |  2 ++
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/mainwindow.cpp b/src/mainwindow.cpp
index 7f2e8f8..6e83887 100644
--- a/src/mainwindow.cpp
+++ b/src/mainwindow.cpp
@@ -22,9 +22,6 @@ MainWindow::MainWindow(QWidget *parent) :
 {
     ui->setupUi(this);
 
-	QSettings s;
-	restoreGeometry(s.value("geometry").toByteArray());
-
 	// Status Bar
 	setupStatusBar();
     
@@ -69,12 +66,26 @@ MainWindow::MainWindow(QWidget *parent) :
     rpc->refreshZECPrice();
 
     rpc->refresh(true);  // Force refresh first time
+
+	restoreSavedStates();
+}
+
+void MainWindow::restoreSavedStates() {
+	QSettings s;
+	restoreGeometry(s.value("geometry").toByteArray());
+
+	ui->balancesTable->horizontalHeader()->restoreState(s.value("baltablegeometry").toByteArray());
+	ui->transactionsTable->horizontalHeader()->restoreState(s.value("tratablegeometry").toByteArray());
 }
 
 void MainWindow::closeEvent(QCloseEvent* event) {
 	QSettings s;
+
 	s.setValue("geometry", saveGeometry());
-	QWidget::closeEvent(event);
+	s.setValue("baltablegeometry", ui->balancesTable->horizontalHeader()->saveState());
+	s.setValue("tratablegeometry", ui->transactionsTable->horizontalHeader()->saveState());
+
+	QMainWindow::closeEvent(event);
 }
 
 void MainWindow::turnstileProgress() {
diff --git a/src/mainwindow.h b/src/mainwindow.h
index b8e2dd3..a604999 100644
--- a/src/mainwindow.h
+++ b/src/mainwindow.h
@@ -84,6 +84,8 @@ private:
     void importPrivKey();
     void doImport(QList<QString>* keys);
 
+	void restoreSavedStates();
+
     RPC*                rpc;
 
     QMovie*             loadingMovie;