Ticket #859: 859.patch

binaries/data/mods/public/gui/session/selection.js

@@ -392 +392 @@
 
 EntityGroupsContainer.prototype.update = function()
 {
+    this.checkRenamedEntities();
     for each (var group in this.groups)
     {
         for (var ent in group.ents)
@@ -407 +408 @@
     }
 }
 
+/**
+ * Update control groups if some entities in them was renamed
+ * (in case of unit promotion or finishing building structure)
+ */
+EntityGroupsContainer.prototype.checkRenamedEntities = function()
+{
+    var renamedEntities = Engine.GuiInterfaceCall("GetRenamedEntities", true);
+    if (renamedEntities.length > 0)
+    {
+        for each (var group in this.groups)
+        {
+            var addToGroup = [];
+            for each (var renamedEntity in renamedEntities)
+            {
+                if (renamedEntity.entity in group.ents)
+                {
+                    group.removeEnt(renamedEntity.entity);
+                    addToGroup.push(renamedEntity.newentity);
+                }
+            }
+            group.add(addToGroup);
+        }
+    }
+}
+
 var g_Groups = new EntityGroupsContainer();