From 0d3d149c4f3bb3197bc2f4488ba8f856f0f35889 Mon Sep 17 00:00:00 2001
From: Miodrag Milanovic <mmicko@gmail.com>
Date: Sat, 28 Jul 2018 16:56:16 +0200
Subject: Clean highlight selection if removed from tree

---
 gui/designwidget.cc | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'gui/designwidget.cc')

diff --git a/gui/designwidget.cc b/gui/designwidget.cc
index b321aef1..ee6a14b4 100644
--- a/gui/designwidget.cc
+++ b/gui/designwidget.cc
@@ -223,6 +223,18 @@ void DesignWidget::updateTree()
 
     clearProperties();
 
+    QMap<ContextTreeItem *, int>::iterator i = highlightSelected.begin();
+    while (i != highlightSelected.end()) {
+        QMap<ContextTreeItem *, int>::iterator prev = i;
+        ++i;
+        if (prev.key()->type() == ElementType::NET && ctx->nets.find(prev.key()->id()) == ctx->nets.end()) {
+            highlightSelected.erase(prev);
+        }
+        if (prev.key()->type() == ElementType::CELL && ctx->cells.find(prev.key()->id()) == ctx->cells.end()) {
+            highlightSelected.erase(prev);
+        }
+    }
+
     treeModel->updateData(ctx);
 }
 QtProperty *DesignWidget::addTopLevelProperty(const QString &id)
-- 
cgit v1.2.3