using System.Reflection; using System; using System.Linq; using System.Collections.Generic; using UnityEngine; using UnityEngine.SceneManagement; using UnityEditor; using UnityEditor.IMGUI.Controls; namespace GameplayIngredients.Editor { public class FindAndReplaceWindow : EditorWindow { [MenuItem("Edit/Find And Replace... %&#F", priority = 144)] static void OpenWindow() { GetWindow(); } static readonly Dictionary s_assemblyTypes = GetTypes(); private static Dictionary GetTypes() { Dictionary all = new Dictionary(); foreach(var assembly in AppDomain.CurrentDomain.GetAssemblies()) { foreach(Type t in assembly.GetTypes()) { if(typeof(Component).IsAssignableFrom(t) && !all.ContainsKey(t.Name)) all.Add(t.Name, t); } } return all; } public enum SearchBy { Name, ComponentType, Tag, Layer, Mesh, Material, } private void OnEnable() { titleContent = Contents.title; minSize = new Vector2(640, 280); } private void OnGUI() { using (new GUILayout.HorizontalScope()) { using (new GUILayout.VerticalScope(GUILayout.Width(320))) { SearchControlsGUI(); } using (new GUILayout.VerticalScope(GUILayout.Width(4))) { GUILayout.FlexibleSpace(); Rect r = GUILayoutUtility.GetLastRect(); EditorGUI.DrawRect(r, Color.black); } using (new GUILayout.VerticalScope()) { SearchResultsGUI(); } } } [SerializeField] GameObject prefabReplacement; [SerializeField] SearchBy searchBy; [SerializeField] string nameSearch = "GameObject"; [SerializeField] string tagSearch = "Player"; [SerializeField] string layerSearch = "PostProcessing"; [SerializeField] string componentSearch = "Light"; [SerializeField] Mesh meshSearch; [SerializeField] Material materialSearch; [SerializeField] bool replacePosition = true; [SerializeField] bool replaceRotation = true; [SerializeField] bool replaceScale = false; [SerializeField] bool replaceParenting = true; [SerializeField] bool replaceName = true; [SerializeField] bool replaceTag = false; [SerializeField] bool replaceLayer = true; [SerializeField] bool replaceStatic = false; [SerializeField] bool unpackPrefab = false; enum SearchOp { Find, Add, Refine } void SearchControlsGUI() { EditorGUIUtility.labelWidth = 120; GUILayout.Space(4); GUILayout.Label("Search and Filter", Styles.boldLabel); searchBy = (SearchBy)EditorGUILayout.EnumPopup(Contents.searchBy, searchBy); switch(searchBy) { case SearchBy.Name: nameSearch = EditorGUILayout.TextField(Contents.nameSearch, nameSearch); SearchButtonsGUI(searchBy, nameSearch); break; case SearchBy.Tag: tagSearch = EditorGUILayout.TextField(Contents.tagSearch, tagSearch); SearchButtonsGUI(searchBy, tagSearch); break; case SearchBy.Layer: layerSearch = EditorGUILayout.TextField(Contents.layerSearch, layerSearch); SearchButtonsGUI(searchBy, layerSearch); break; case SearchBy.ComponentType: componentSearch = EditorGUILayout.TextField(Contents.componentSearch, componentSearch); SearchButtonsGUI(searchBy, componentSearch); break; case SearchBy.Mesh: meshSearch = (Mesh)EditorGUILayout.ObjectField(Contents.meshSearch, meshSearch, typeof(Mesh), true); SearchButtonsGUI(searchBy, meshSearch); break; case SearchBy.Material: materialSearch = (Material)EditorGUILayout.ObjectField(Contents.materialSearch, materialSearch, typeof(Material), true); SearchButtonsGUI(searchBy, materialSearch); break; } GUILayout.FlexibleSpace(); GUILayout.Label("Replace By", Styles.boldLabel); prefabReplacement = (GameObject)EditorGUILayout.ObjectField(Contents.prefabReplacement, prefabReplacement, typeof(GameObject), true); using (new GUILayout.HorizontalScope()) { replacePosition = GUILayout.Toggle(replacePosition, "Position", EditorStyles.miniButtonLeft, GUILayout.Height(16)); replaceRotation = GUILayout.Toggle(replaceRotation, "Rotation", EditorStyles.miniButtonMid, GUILayout.Height(16)); replaceScale = GUILayout.Toggle(replaceScale, "Scale", EditorStyles.miniButtonMid, GUILayout.Height(16)); replaceParenting = GUILayout.Toggle(replaceParenting, "Parenting", EditorStyles.miniButtonRight, GUILayout.Height(16)); } using (new GUILayout.HorizontalScope()) { replaceName = GUILayout.Toggle(replaceName, "Name", EditorStyles.miniButtonLeft, GUILayout.Height(16)); replaceTag = GUILayout.Toggle(replaceTag, "Tag", EditorStyles.miniButtonMid, GUILayout.Height(16)); replaceLayer = GUILayout.Toggle(replaceLayer, "Layer", EditorStyles.miniButtonMid, GUILayout.Height(16)); replaceStatic = GUILayout.Toggle(replaceStatic, "Static", EditorStyles.miniButtonMid, GUILayout.Height(16)); unpackPrefab = GUILayout.Toggle(unpackPrefab, "Unpack Prefab", EditorStyles.miniButtonRight, GUILayout.Height(16)); } if (GUILayout.Button("Replace All", Styles.bigButton, GUILayout.Height(24)) && prefabReplacement != null) { Undo.RecordObjects(searchResults.ToArray(), "Replace Objects"); for(int i = 0; i < searchResults.Count; i++) { var obj = searchResults[i]; var newObj = SwapObject(obj, prefabReplacement, searchResults); searchResults[i] = newObj; } } GUILayout.Space(8); } GameObject SwapObject(GameObject toReplace, GameObject replacement, List others) { GameObject newObj; if (PrefabUtility.GetPrefabAssetType(replacement) != PrefabAssetType.NotAPrefab && !unpackPrefab) newObj = (GameObject)PrefabUtility.InstantiatePrefab(replacement); else newObj = (GameObject)Instantiate(replacement); if (replaceName) newObj.name = replacement.name; else newObj.name = toReplace.name; if(replacePosition) newObj.transform.position = toReplace.transform.position; if(replaceRotation) newObj.transform.rotation = toReplace.transform.rotation; if(replaceParenting) newObj.transform.parent = toReplace.transform.parent; if(replaceScale) newObj.transform.localScale = toReplace.transform.localScale; if(replaceTag) newObj.tag = toReplace.tag; if(replaceLayer) newObj.layer = toReplace.layer; if(replaceStatic) newObj.isStatic = toReplace.isStatic; foreach(var other in others) { if(other.transform.parent == toReplace.transform) { other.transform.parent = newObj.transform; } } DestroyImmediate(toReplace); return newObj; } void SearchButtonsGUI(SearchBy by, object criteria) { using (new GUILayout.HorizontalScope()) { if (GUILayout.Button("Find", Styles.bigButton, GUILayout.Height(24))) Search(SearchOp.Find, by, criteria); if (GUILayout.Button("Add", Styles.bigButton, GUILayout.Height(24))) Search(SearchOp.Add, by, criteria); if (GUILayout.Button("Refine", Styles.bigButton, GUILayout.Height(24))) Search(SearchOp.Refine, by, criteria); } } static GameObject[] FindAllSceneObjects() { var all = Resources.FindObjectsOfTypeAll(); all = all.Where(o => o.scene.isLoaded).ToArray(); return all; } void Search(SearchOp op, SearchBy by, object criteria) { List query = new List(); var all = FindAllSceneObjects(); switch (by) { case SearchBy.Name: foreach(var go in all) { if (go.name.Contains((string)criteria)) query.Add(go); } break; case SearchBy.Tag: query.AddRange(GameObject.FindGameObjectsWithTag((string)criteria)); break; case SearchBy.Layer: foreach (var go in all) { if (go.layer == LayerMask.NameToLayer((string)criteria)) query.Add(go); } break; case SearchBy.ComponentType: if(s_assemblyTypes.ContainsKey((string)criteria)) { Type t = s_assemblyTypes[(string)criteria]; if( typeof(Component).IsAssignableFrom(t)) { Component[] components = (Component[])Resources.FindObjectsOfTypeAll(t); if(components != null) { foreach(var c in components) { if (c.gameObject.scene != null && !query.Contains(c.gameObject)) query.Add(c.gameObject); } } } } break; case SearchBy.Mesh: Mesh mesh = (Mesh)criteria; foreach (var go in all) { MeshFilter filter = go.GetComponent(); if (filter != null && filter.sharedMesh == mesh) { query.Add(go); } } break; case SearchBy.Material: Material mat = (Material)criteria; foreach (var go in all) { Renderer renderer = go.GetComponent(); if (renderer != null) { if(renderer.sharedMaterials.Contains(mat)) { query.Add(go); } } } break; } switch (op) { case SearchOp.Find: searchResults = query; break; case SearchOp.Add: foreach(var item in query) { if (!searchResults.Contains(item)) searchResults.Add(item); } break; case SearchOp.Refine: List refined = new List(); foreach (var item in searchResults) { if (query.Contains(item)) refined.Add(item); } searchResults = refined; break; } } [SerializeField] List searchResults= new List(); Vector2 scroll; void SearchResultsGUI() { using (new GUILayout.HorizontalScope()) { GUILayout.Label("Search Results", Styles.boldLabel); GUILayout.FlexibleSpace(); if(GUILayout.Button("From Selection", GUILayout.Height(24))) { searchResults = Selection.gameObjects.ToList(); } if(GUILayout.Button("Select", GUILayout.Height(24))) { Selection.objects = searchResults.ToArray(); } } scroll = GUILayout.BeginScrollView(scroll, EditorStyles.helpBox); { GameObject toRemove = null; // Trim all nulls searchResults = searchResults.Where(o => o != null).ToList(); foreach(var obj in searchResults) { using (new GUILayout.HorizontalScope(EditorStyles.textField)) { GUILayout.Label(obj.name, EditorStyles.label); if(GUILayout.Button("X", GUILayout.Width(32))) { toRemove = obj; } } } if (toRemove != null) searchResults.Remove(toRemove); } GUILayout.EndScrollView(); } static class Contents { public static GUIContent title = new GUIContent("Find and Replace", (Texture)EditorGUIUtility.LoadRequired("ViewToolZoom On")); public static GUIContent searchBy = new GUIContent("Search by"); public static GUIContent nameSearch = new GUIContent("GameObject Name"); public static GUIContent tagSearch = new GUIContent("Tag"); public static GUIContent layerSearch = new GUIContent("Layer"); public static GUIContent componentSearch = new GUIContent("Component Type"); public static GUIContent meshSearch = new GUIContent("Mesh"); public static GUIContent materialSearch = new GUIContent("Material"); public static GUIContent prefabReplacement = new GUIContent("Prefab Replacement"); } static class Styles { public static readonly GUIStyle boldLabel = GetBoldLabel(); public static readonly GUIStyle bigButton = GetBigButton(); static GUIStyle GetBoldLabel() { var style = new GUIStyle(EditorStyles.boldLabel); style.fontSize = 14; return style; } static GUIStyle GetBigButton() { var style = new GUIStyle(EditorStyles.miniButton); style.fontSize = 14; return style; } } } }