From 0c5424aa3f92707e99ed32896429f8887f53102a Mon Sep 17 00:00:00 2001
From: Ansis Brammanis <ansis.brammanis@gmail.com>
Date: Tue, 2 Apr 2013 13:51:32 -0400
Subject: [PATCH] re-add 'selected' class after each redraw

Performance seems to be fine, but we could move this into
svg/* and only run it on entering elements.

fix #1236
---
 js/id/modes/select.js | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/js/id/modes/select.js b/js/id/modes/select.js
index 04ee6db7d..de5d4ffe2 100644
--- a/js/id/modes/select.js
+++ b/js/id/modes/select.js
@@ -131,10 +131,15 @@ iD.modes.Select = function(context, selection, initial) {
         d3.select(document)
             .call(keybinding);
 
-        context.surface()
-            .selectAll("*")
-            .filter(selected)
-            .classed('selected', true);
+        function selectElements() {
+            context.surface()
+                .selectAll("*")
+                .filter(selected)
+                .classed('selected', true);
+        }
+
+        context.map().on('drawn.select', selectElements);
+        selectElements();
 
         radialMenu = iD.ui.RadialMenu(operations);
         var show = d3.event && !initial;
@@ -176,6 +181,8 @@ iD.modes.Select = function(context, selection, initial) {
             .on('dblclick.select', null)
             .selectAll(".selected")
             .classed('selected', false);
+
+        context.map().on('drawn.select', null);
     };
 
     return mode;