Fixed baking one object several times

wilk-polarny · Jul 28, 2018 · eb2dd35 · eb2dd35
1 parent c017a98
commit eb2dd35
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/SimplestMeshBaker/Editor/MeshBaker.cs b/SimplestMeshBaker/Editor/MeshBaker.cs
@@ -1,4 +1,5 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 using System.Linq;
 using UnityEditor;
 using UnityEngine;
@@ -26,6 +27,7 @@ enum Resolving
         [MenuItem("GameObject/Bake Meshes", false, 0)]
         private static void BakeMeshes(MenuCommand menuCommand)
         {
+
             if (Selection.objects.Length == 0)
             {
                 return;
@@ -131,11 +133,13 @@ private static void BakeMeshes(MenuCommand menuCommand)
             if (EditorUtility.DisplayDialog("Simplest Mesh Baker",
                 "Do you want to remove sources?", "Yes", "No"))
             {
+                HashSet<GameObject> gos = new HashSet<GameObject>();
                 foreach (GameObject selected in Selection.gameObjects)
                 {
-                    if (selected != null)
+                    if (selected != null && !gos.Contains(selected))
                     {
                         Undo.DestroyObjectImmediate(selected);
+                        gos.Add(selected);
                     }
                 }
             }