From c7ae18df4defe4776931541151fc86ef437893a3 Mon Sep 17 00:00:00 2001
From: Teo Mrnjavac <teo@kde.org>
Date: Wed, 30 Dec 2015 17:25:48 +0100
Subject: [PATCH] Keep selection in sync between tree and bars view in manual
 partitioning

---
 src/modules/partition/gui/PartitionPage.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/modules/partition/gui/PartitionPage.cpp b/src/modules/partition/gui/PartitionPage.cpp
index 7542a51ef..d5b6e08cd 100644
--- a/src/modules/partition/gui/PartitionPage.cpp
+++ b/src/modules/partition/gui/PartitionPage.cpp
@@ -293,6 +293,26 @@ PartitionPage::updateFromCurrentDevice()
     m_ui->partitionTreeView->setModel( model );
     m_ui->partitionTreeView->expandAll();
 
+    // Make both views use the same selection model.
+    if ( m_ui->partitionBarsView->selectionModel() !=
+         m_ui->partitionTreeView->selectionModel() )
+    {
+        QItemSelectionModel* selectionModel = m_ui->partitionTreeView->selectionModel();
+        m_ui->partitionTreeView->setSelectionModel( m_ui->partitionBarsView->selectionModel() );
+        selectionModel->deleteLater();
+    }
+
+    // This is necessary because even with the same selection model it might happen that
+    // a !=0 column is selected in the tree view, which for some reason doesn't trigger a
+    // timely repaint in the bars view.
+    connect( m_ui->partitionBarsView->selectionModel(), &QItemSelectionModel::currentChanged,
+             this, [=]
+    {
+        QModelIndex selectedIndex = m_ui->partitionBarsView->selectionModel()->currentIndex();
+        selectedIndex = selectedIndex.sibling( selectedIndex.row(), 0 );
+        m_ui->partitionBarsView->setCurrentIndex( selectedIndex );
+    }, Qt::UniqueConnection );
+
     // Must be done here because we need to have a model set to define
     // individual column resize mode
     QHeaderView* header = m_ui->partitionTreeView->header();