Fix issue#265: crash after two identical setGroup commands

Author: lelandaisb

src/Core/Mesh/CommandAddRemoveGroupName.cpp

@@ -70,12 +70,18 @@ void CommandAddRemoveGroupName::internalExecute()
 			updateMesh(ge, grp->getName(), false);
 		} else if (m_ope == set){
 			// on retire tous les groupes
+            bool ge_already_in_group = false;
 			for (Group::GroupEntity* grp : groups)
-				m_group_helper.removeFromGroup(grp->getName(), ge);
+                if (grp->getName() == m_groupName)
+                    ge_already_in_group = true;
+                else
+    				m_group_helper.removeFromGroup(grp->getName(), ge);
 
-			// on ajoute le groupe
-			Group::GroupEntity* grp = m_group_helper.addToGroup(m_groupName, ge);
-			updateMesh(ge, grp->getName(), true);
+			// on ajoute le groupe s'il n'était pas déjà présent
+            if (!ge_already_in_group) {
+                Group::GroupEntity* grp = m_group_helper.addToGroup(m_groupName, ge);
+                updateMesh(ge, grp->getName(), true);
+            }
 		} else {
 			TkUtil::Exception ("[Erreur interne] Opération non prévue");
 		}
@@ -97,12 +103,18 @@ void CommandAddRemoveGroupName::internalExecute()
 			updateMesh(te, grp->getName(), false);
 		} else if (m_ope == set){
 			// on retire tous les groupes
+            bool te_already_in_group = false;
 			for (Group::GroupEntity* grp : groups)
-				m_group_helper.removeFromGroup(grp->getName(), te);
+                if (grp->getName() == m_groupName)
+                    te_already_in_group = true;
+                else
+    				m_group_helper.removeFromGroup(grp->getName(), te);
 
-			// on ajoute le groupe
-			Group::GroupEntity* grp = m_group_helper.addToGroup(m_groupName, te);
-			updateMesh(te, grp->getName(), true);
+			// on ajoute le groupe s'il n'était pas déjà présent
+            if (!te_already_in_group) {
+                Group::GroupEntity* grp = m_group_helper.addToGroup(m_groupName, te);
+                updateMesh(te, grp->getName(), true);
+            }
 		} else {
 			TkUtil::Exception ("[Erreur interne] Opération non prévue");
 		}

test_link/test_groups.py

@@ -334,3 +334,21 @@ def test_topo_surface():
     assert ctx.getGroupManager().getTopoFaces("aaa", 2) == ["Fa0000"]
     ctx.undo()
     assert ctx.getGroupManager().getTopoFaces("aaa", 2) == []
+
+# issue#265: crash after two identical setGroup commands
+def test_issue265():
+    ctx = Mgx3D.getStdContext()
+    ctx.clearSession() # Clean the session after the previous test
+    gm = ctx.getGeomManager()
+
+    gm.newBox (Mgx3D.Point(0, 0, 0), Mgx3D.Point(1, 1, 1))
+    gm.setGroup(["Vol0000"], 3, "bbb")
+    assert gm.getInfos("Vol0000", 3).groups() == ["bbb"]
+    gm.setGroup(["Vol0000"], 3, "aaa")
+    assert gm.getInfos("Vol0000", 3).groups() == ["aaa"]
+    gm.setGroup(["Vol0000"], 3, "aaa")
+    assert gm.getInfos("Vol0000", 3).groups() == ["aaa"]
+    gm.newBox (Mgx3D.Point(1, 0, 0), Mgx3D.Point(2, 1, 1))
+    gm.setGroup(["Vol0001"], 3, "aaa")
+    assert gm.getInfos("Vol0000", 3).groups() == ["aaa"]
+    assert gm.getInfos("Vol0001", 3).groups() == ["aaa"]