Fixup packet list selection.

Be less aggressive about resetting the packet list model so that we
retain our selection when (un)coloring packets or rebuilding columns.

Correctly enable a colorization menu item while we're here.

Change-Id: I8d1c8f26dbb8a814b8344b609695b77632006e4b
Reviewed-on: https://code.wireshark.org/review/9608
Reviewed-by: Gerald Combs <gerald@wireshark.org>

1 files changed, 3 insertions, 4 deletions

diff --git a/ui/qt/packet_list_model.cpp b/ui/qt/packet_list_model.cpp
index 80563be940..8a3e443d52 100644
--- a/ui/qt/packet_list_model.cpp
+++ b/ui/qt/packet_list_model.cpp
@@ -110,22 +110,21 @@ void PacketListModel::clear() {
 
 void PacketListModel::resetColumns()
 {
-    beginResetModel();
     if (cap_file_) {
         PacketListRecord::resetColumns(&cap_file_->cinfo);
     }
-    endResetModel();
+    dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
+    headerDataChanged(Qt::Horizontal, 0, columnCount() - 1);
 }
 
 void PacketListModel::resetColorized()
 {
     PacketListRecord *record;
 
-    beginResetModel();
     foreach (record, physical_rows_) {
         record->resetColorized();
     }
-    endResetModel();
+    dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
 }
 
 void PacketListModel::setMonospaceFont(const QFont &mono_font, int row_height)