#if UNITY_EDITOR using System.Collections; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using UnityEditor; using UnityEngine; namespace Poi.Tools { public class PoiOutlineUtilWindow : EditorWindow { private static Vector2 scrollPosition = new Vector2(0,0); private static GameObject avatar; private static readonly Dictionary meshSettings = new Dictionary(); // > private static Dictionary bakedMeshes = new Dictionary(); private static int lang = -1; private static bool isCancelled = false; private static readonly Color emptyColor = new Color(0.5f, 0.5f, 1.0f, 1.0f); private static GUIStyle marginBox; private struct MeshSettings { public string name; public bool isBakeTarget; public float shrinkTipStrength; } //------------------------------------------------------------------------------------------------------------------------------ // GUI [MenuItem("Poi/Outline Vertex Color Baker")] static void Init() { PoiOutlineUtilWindow window = (PoiOutlineUtilWindow)GetWindow(typeof(PoiOutlineUtilWindow), false, TEXT_WINDOW_NAME); window.Show(); } void OnGUI() { scrollPosition = EditorGUILayout.BeginScrollView(scrollPosition); // Language if(lang == -1) { lang = Application.systemLanguage == SystemLanguage.Japanese ? 1 : 0; } lang = EditorGUILayout.Popup("Language", lang, TEXT_LANGUAGES); marginBox = new GUIStyle(EditorStyles.helpBox); marginBox.margin.left = 30; //------------------------------------------------------------------------------------------------------------------------------ // 1. Select the mesh EditorGUILayout.LabelField(TEXT_STEP_SELECT_AVATAR[lang], EditorStyles.boldLabel); avatar = (GameObject)EditorGUILayout.ObjectField(TEXT_ITEM_DD_AVATAR[lang], avatar, typeof(GameObject), true); if(avatar == null) { EditorGUILayout.EndScrollView(); return; } if(AssetDatabase.Contains(avatar)) { EditorGUILayout.HelpBox(TEXT_WARN_SELECT_FROM_SCENE[lang], MessageType.Error); EditorGUILayout.EndScrollView(); return; } EditorGUILayout.Space(); //------------------------------------------------------------------------------------------------------------------------------ // 2. Select the modify target EditorGUILayout.LabelField(TEXT_STEP_SELECT_SUBMESH[lang], EditorStyles.boldLabel); if (GUILayout.Button("Select All")) { foreach (var item in meshSettings) { for (int i = 0; i < item.Value.Length; i++) { item.Value[i].isBakeTarget = true; } } } Component[] skinnedMeshRenderers = avatar.GetComponentsInChildren(typeof(SkinnedMeshRenderer), true); Component[] meshRenderers = avatar.GetComponentsInChildren(typeof(MeshRenderer), true); DrawModifyTargetsGUI(skinnedMeshRenderers, meshRenderers); EditorGUILayout.Space(); //------------------------------------------------------------------------------------------------------------------------------ // 3. Generate the mesh, test it, then save GameObject bakedAvatar = FindBakedAvatar(); EditorGUILayout.LabelField(TEXT_STEP_GENERATE_AND_SAVE[lang], EditorStyles.boldLabel); EditorGUILayout.BeginHorizontal(); if(GUILayout.Button(TEXT_BUTTON_GENERATE_AND_TEST[lang])) { GenerateMeshes(bakedAvatar, skinnedMeshRenderers, meshRenderers); } if(bakedAvatar == null) { EditorGUILayout.EndHorizontal(); EditorGUILayout.EndScrollView(); return; } // Save bakedMeshes = new Dictionary(); GetBakedMeshes(bakedAvatar, skinnedMeshRenderers, meshRenderers); bool isSaved = true; foreach(Mesh bakedMesh in bakedMeshes.Values) { if(!isSaved) break; if(bakedMesh == null) continue; isSaved = AssetDatabase.Contains(bakedMesh); } GUIStyle saveButton = new GUIStyle(GUI.skin.button); if(!isSaved) { saveButton.normal.textColor = Color.red; saveButton.fontStyle = FontStyle.Bold; } if(GUILayout.Button(TEXT_BUTTON_SAVE[lang], saveButton)) { SaveMeshes(); } EditorGUILayout.EndHorizontal(); if(!isSaved) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_NOT_SAVED[lang], MessageType.Warning); } EditorGUILayout.EndScrollView(); } //------------------------------------------------------------------------------------------------------------------------------ // 2. Select the modify target private static void DrawModifyTargetsGUI(Component[] skinnedMeshRenderers, Component[] meshRenderers) { foreach(SkinnedMeshRenderer skinnedMeshRenderer in skinnedMeshRenderers) { EditorGUILayout.LabelField(skinnedMeshRenderer.gameObject.name, EditorStyles.boldLabel); int id = skinnedMeshRenderer.gameObject.GetInstanceID(); Mesh sharedMesh = skinnedMeshRenderer.sharedMesh; Material[] materials = skinnedMeshRenderer.sharedMaterials; EditorGUI.indentLevel++; DrawGUIPerComponent(id, sharedMesh, materials); EditorGUI.indentLevel--; } foreach(MeshRenderer meshRenderer in meshRenderers) { MeshFilter meshFilter = meshRenderer.gameObject.GetComponent(); if(meshFilter == null) { continue; } EditorGUILayout.LabelField(meshRenderer.gameObject.name, EditorStyles.boldLabel); int id = meshRenderer.gameObject.GetInstanceID(); Mesh sharedMesh = meshFilter.sharedMesh; Material[] materials = meshRenderer.sharedMaterials; EditorGUI.indentLevel++; DrawGUIPerComponent(id, sharedMesh, materials); EditorGUI.indentLevel--; } } private static void DrawGUIPerComponent(int id, Mesh sharedMesh, Material[] materials) { Vector3[] vertices = sharedMesh?.vertices; Vector3[] normals = sharedMesh?.normals; Vector4[] tangents = sharedMesh?.tangents; Color[] colors = sharedMesh?.colors; Vector2[] uv = sharedMesh?.uv; bool hasColors = colors != null && colors.Length > 2; bool hasUV0 = uv != null || uv.Length > 2; // Draw error messages if(sharedMesh == null) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_IS_EMPTY[lang], MessageType.Error); return; } if(!sharedMesh.isReadable) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_NOT_READABLE[lang], MessageType.Error); return; } if(vertices == null || vertices.Length < 2) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_HAS_NO_VERT[lang], MessageType.Error); return; } if(normals == null && normals.Length < 2) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_HAS_NO_NORM[lang], MessageType.Error); return; } if(tangents == null && tangents.Length < 2) { EditorGUILayout.HelpBox(TEXT_WARN_MESH_HAS_NO_TANJ[lang], MessageType.Error); return; } // Generate empty settings if(!meshSettings.ContainsKey(id)) meshSettings[id] = null; if(meshSettings[id] == null || meshSettings[id].Length != sharedMesh.subMeshCount) { meshSettings[id] = new MeshSettings[sharedMesh.subMeshCount]; for(int i = 0; i < sharedMesh.subMeshCount; i++) { meshSettings[id][i] = new MeshSettings { name = null, isBakeTarget = false, shrinkTipStrength = 0.0f }; } } // Draw settings for(int i = 0; i < sharedMesh.subMeshCount; i++) { if(string.IsNullOrEmpty(meshSettings[id][i].name)) { meshSettings[id][i].name = i + ": "; if(i < materials.Length && materials[i] != null && !string.IsNullOrEmpty(materials[i].name)) { meshSettings[id][i].name += materials[i].name; } } DrawMeshSettingsGUI(id, i, hasColors, hasUV0); } } private static void DrawMeshSettingsGUI(int id, int i, bool hasColors, bool hasUV0) { meshSettings[id][i].isBakeTarget = EditorGUILayout.ToggleLeft(meshSettings[id][i].name, meshSettings[id][i].isBakeTarget); int indentCopy = EditorGUI.indentLevel; EditorGUI.indentLevel = 0; if(meshSettings[id][i].isBakeTarget) { EditorGUILayout.BeginVertical(marginBox); meshSettings[id][i].shrinkTipStrength = EditorGUILayout.FloatField(TEXT_ITEM_SHRINK_TIP[lang], meshSettings[id][i].shrinkTipStrength); EditorGUILayout.EndVertical(); } EditorGUI.indentLevel = indentCopy; } //------------------------------------------------------------------------------------------------------------------------------ // 3. Generate the mesh, test it, then save private static int[] GetChildIndices(GameObject root, GameObject child) { var indices = new List(); indices.Add(child.transform.GetSiblingIndex()); Transform parent = child.transform.parent; while(parent != null && parent != root.transform) { indices.Add(parent.GetSiblingIndex()); parent = parent.parent; } return indices.ToArray(); } private static GameObject GetChild(GameObject root, int[] indices) { Transform current = root.transform; for(int i = indices.Length - 1; i >= 0; i--) { current = current.GetChild(indices[i]); if(current == null) return null; } return current.gameObject; } private static GameObject GetChildInstance(GameObject root, GameObject rootInstance, GameObject child) { return GetChild(rootInstance, GetChildIndices(root, child)); } private static void GenerateMeshes(GameObject bakedAvatar, Component[] skinnedMeshRenderers, Component[] meshRenderers) { if(bakedAvatar == null) { bakedAvatar = Instantiate(avatar); bakedAvatar.name = avatar.name + " (VertexColorBaked)"; bakedAvatar.transform.parent = avatar.transform.parent; bakedAvatar.SetActive(true); } isCancelled = false; foreach(SkinnedMeshRenderer skinnedMeshRenderer in skinnedMeshRenderers) { GameObject child = GetChildInstance(avatar, bakedAvatar, skinnedMeshRenderer.gameObject); if(child == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Child is not found"); continue; } SkinnedMeshRenderer bakedSkinnedMeshRenderer = child.GetComponent(); if(bakedSkinnedMeshRenderer == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Component is not found"); continue; } int id = skinnedMeshRenderer.gameObject.GetInstanceID(); Mesh sharedMesh = skinnedMeshRenderer.sharedMesh; Mesh bakedMesh = bakedSkinnedMeshRenderer.sharedMesh; if(bakedMesh == null || !bakedMesh.name.Contains("(Clone)")) { bakedMesh = Instantiate(sharedMesh); } BakeVertexColors(ref bakedMesh, sharedMesh, id); if(isCancelled) break; if(bakedMesh == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Mesh is not found"); continue; } bakedSkinnedMeshRenderer.sharedMesh = bakedMesh; } foreach(MeshRenderer meshRenderer in meshRenderers) { MeshFilter meshFilter = meshRenderer.gameObject.GetComponent(); if(meshFilter == null) { continue; } GameObject child = GetChildInstance(avatar, bakedAvatar, meshRenderer.gameObject); if(child == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Child is not found"); continue; } MeshFilter bakedMeshFilter = child.GetComponent(); if(bakedMeshFilter == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Component is not found"); continue; } int id = meshRenderer.gameObject.GetInstanceID(); Mesh sharedMesh = meshFilter.sharedMesh; Mesh bakedMesh = bakedMeshFilter.sharedMesh; if(bakedMesh == null || !bakedMesh.name.Contains("(Clone)")) { bakedMesh = Instantiate(sharedMesh); } BakeVertexColors(ref bakedMesh, sharedMesh, id); if(isCancelled) break; if(bakedMesh == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Mesh is not found"); continue; } bakedMeshFilter.sharedMesh = bakedMesh; } if(!isCancelled) EditorUtility.DisplayDialog(TEXT_WINDOW_NAME, "Complete!", "OK"); } private static void GetBakedMeshes(GameObject bakedAvatar, Component[] skinnedMeshRenderers, Component[] meshRenderers) { foreach(SkinnedMeshRenderer skinnedMeshRenderer in skinnedMeshRenderers) { Mesh sharedMesh = skinnedMeshRenderer.sharedMesh; if(sharedMesh == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Mesh is not found"); continue; } GameObject child = GetChildInstance(avatar, bakedAvatar, skinnedMeshRenderer.gameObject); if(child == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Child is not found"); continue; } SkinnedMeshRenderer bakedSkinnedMeshRenderer = child.GetComponent(); if(bakedSkinnedMeshRenderer == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Component is not found"); continue; } bakedMeshes[sharedMesh] = bakedSkinnedMeshRenderer.sharedMesh; } foreach(MeshRenderer meshRenderer in meshRenderers) { MeshFilter meshFilter = meshRenderer.gameObject.GetComponent(); if(meshFilter == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Component is not found"); continue; } Mesh sharedMesh = meshFilter.sharedMesh; if(sharedMesh == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Mesh is not found"); continue; } GameObject child = GetChildInstance(avatar, bakedAvatar, meshRenderer.gameObject); if(child == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Child is not found"); continue; } MeshFilter bakedMeshFilter = child.GetComponent(); if(bakedMeshFilter == null) { Debug.LogWarning($"[{TEXT_WINDOW_NAME}] Component is not found"); continue; } bakedMeshes[sharedMesh] = bakedMeshFilter.sharedMesh; } } private static void SaveMeshes() { foreach(KeyValuePair bakedMesh in bakedMeshes) { if(bakedMesh.Value == null || string.IsNullOrEmpty(bakedMesh.Value.name)) continue; string path = AssetDatabase.GetAssetPath(bakedMesh.Value); if(string.IsNullOrEmpty(path)) { path = AssetDatabase.GetAssetPath(bakedMesh.Key); if(string.IsNullOrEmpty(path) || !path.StartsWith("Assets/")) { path = "Assets/BakedMeshes/" + bakedMesh.Value.name + ".asset"; } else { path = Path.GetDirectoryName(path) + "/BakedMeshes/" + bakedMesh.Value.name + ".asset"; } path = GetUniqueName(path); } string saveDirectory = Path.GetDirectoryName(path); if(!Directory.Exists(saveDirectory)) { Directory.CreateDirectory(saveDirectory); } if(!File.Exists(path)) { Debug.Log($"[{TEXT_WINDOW_NAME}] Create asset to: " + path); AssetDatabase.CreateAsset(bakedMesh.Value, path); } else { Debug.Log($"[{TEXT_WINDOW_NAME}] Overwrite mesh to: " + path); } } AssetDatabase.SaveAssets(); EditorUtility.DisplayDialog(TEXT_WINDOW_NAME, "Complete!", "OK"); } private static string GetUniqueName(string path) { if(!File.Exists(path)) return path; string baseName = Path.GetDirectoryName(path) + "/" + Path.GetFileNameWithoutExtension(path); string outPath; int i = 1; while(true) { outPath = baseName + " " + i.ToString() + ".asset"; if(!File.Exists(outPath)) return outPath; i++; } } //------------------------------------------------------------------------------------------------------------------------------ // Mesh Generator private static void BakeVertexColors(ref Mesh mesh, Mesh sharedMesh, int id) { if(isCancelled || sharedMesh == null || !sharedMesh.isReadable) return; Vector3[] vertices = sharedMesh.vertices; Vector3[] normals = sharedMesh.normals; Vector4[] tangents = sharedMesh.tangents; if(vertices == null || vertices.Length < 2 || normals == null && normals.Length < 2 || tangents == null && tangents.Length < 2) { return; } Color[] outColors = Enumerable.Repeat(Color.white, vertices.Length).ToArray(); isCancelled = false; for(int mi = 0; mi < sharedMesh.subMeshCount; mi++) { if(!meshSettings[id][mi].isBakeTarget) continue; int[] sharedIndices = GetOptIndices(sharedMesh, mi); BakeNormalAverage(ref outColors, sharedIndices, meshSettings[id][mi], vertices, normals, tangents); EditorUtility.ClearProgressBar(); if(isCancelled) return; } FixIllegalDatas(ref outColors); mesh.SetColors(outColors); EditorUtility.SetDirty(mesh); } //------------------------------------------------------------------------------------------------------------------------------ // Bake normal to color private static void BakeNormalAverage(ref Color[] outColors, int[] sharedIndices, MeshSettings settings, Vector3[] vertices, Vector3[] normals, Vector4[] tangents) { var normalAverages = NormalGatherer.GetNormalAveragesFast(sharedIndices, vertices, normals); string message = "Run bake in " + settings.name; for(int i = 0; i < sharedIndices.Length; ++i) { int index = sharedIndices[i]; float width = 1.0f; Vector3 normal = normals[index]; Vector4 tangent = tangents[index]; Vector3 bitangent = Vector3.Cross(normal, tangent) * tangent.w; if(IsIllegalTangent(normal, tangent)) { outColors[index].r = 0.5f; outColors[index].g = 0.5f; outColors[index].b = 1.0f; outColors[index].a = 1.0f; continue; } Vector3 normalAverage = NormalGatherer.GetClosestNormal(normalAverages, vertices[index]); if(settings.shrinkTipStrength > 0) width *= Mathf.Pow(Mathf.Clamp01(Vector3.Dot(normal,normalAverage)), settings.shrinkTipStrength); outColors[index].r = Vector3.Dot(normalAverage, tangent) * 0.5f + 0.5f; outColors[index].g = Vector3.Dot(normalAverage, bitangent) * 0.5f + 0.5f; outColors[index].b = Vector3.Dot(normalAverage, normal) * 0.5f + 0.5f; outColors[index].a = width; if(DrawProgress(message, i, (float)i / (float)sharedIndices.Length)) return; } } public static bool DrawProgress(string message, int i, float progress) { if((i & 0b11111111) == 0b11111111) return isCancelled = isCancelled || EditorUtility.DisplayCancelableProgressBar(TEXT_WINDOW_NAME, message, progress); return isCancelled; } private static int[] GetOptIndices(Mesh mesh, int mi) { return mesh.GetIndices(mi).ToList().Distinct().ToArray(); } private static bool IsIllegalTangent(Vector3 normal, Vector4 tangent) { return normal.x == tangent.x && normal.y == tangent.y && normal.z == tangent.z; } //------------------------------------------------------------------------------------------------------------------------------ // Utils private static GameObject FindBakedAvatar() { if(avatar.transform.parent != null) { for(int i = 0; i < avatar.transform.parent.childCount; i++) { GameObject childObject = avatar.transform.parent.GetChild(i).gameObject; if(childObject.name.Contains(avatar.name + " (VertexColorBaked)")) { return childObject; } } } return GameObject.Find(avatar.name + " (VertexColorBaked)"); } private static void FixIllegalDatas(ref Color[] outColors) { for(int i = 0; i < outColors.Length; i++) { Color color = outColors[i]; if( color.r >= 0 && color.r <= 1 && color.g >= 0 && color.g <= 1 && color.b >= 0 && color.b <= 1 && color.a >= 0 && color.a <= 1 ) { continue; } outColors[i] = emptyColor; } } //------------------------------------------------------------------------------------------------------------------------------ // Languages private const string TEXT_WINDOW_NAME = "PoiOutlineUtil"; private static readonly string[] TEXT_LANGUAGES = new[] {"English", "Japanese"}; private static readonly string[] TEXT_STEP_SELECT_AVATAR = new[] {"1. Select the avatar", "1. アバターを選択"}; private static readonly string[] TEXT_STEP_SELECT_SUBMESH = new[] {"2. Select the modify target", "2. 編集対象を選択"}; private static readonly string[] TEXT_STEP_GENERATE_AND_SAVE = new[] {"3. Generate the mesh, test it, then save", "3. メッシュを生成・テスト・保存"}; private static readonly string[] TEXT_WARN_SELECT_FROM_SCENE = new[] {"Please select from the scene (hierarchy)", "シーン（ヒエラルキー）から選択してください"}; private static readonly string[] TEXT_WARN_MESH_NOT_READABLE = new[] {"The selected mesh is not set to \"Read/Write\" on.", "選択されたメッシュは\"Read/Write\"がオンになっていません。"}; private static readonly string[] TEXT_WARN_MESH_IS_EMPTY = new[] {"The selected mesh is empty!", "選択したメッシュは空です"}; private static readonly string[] TEXT_WARN_MESH_HAS_NO_VERT = new[] {"The selected mesh has no vertices!", "選択したメッシュは頂点がありません。"}; private static readonly string[] TEXT_WARN_MESH_HAS_NO_NORM = new[] {"The selected mesh has no normals!", "選択したメッシュは法線がありません。"}; private static readonly string[] TEXT_WARN_MESH_HAS_NO_TANJ = new[] {"The selected mesh has no tangents!", "選択したメッシュはタンジェントがありません。"}; private static readonly string[] TEXT_WARN_MESH_NOT_SAVED = new[] {"Generated mesh is not saved!", "生成されたメッシュが保存されていません。"}; private static readonly string[] TEXT_ITEM_DD_AVATAR = new[] {"Avatar (D&D from scene)", "アバター (シーンからD&D)"}; private static readonly string[] TEXT_ITEM_SHRINK_TIP = new[] {"Shrink the tip", "先端を細くする度合い"}; private static readonly string[] TEXT_BUTTON_GENERATE_AND_TEST = new[] {"Generate & Test", "生成 & テスト"}; private static readonly string[] TEXT_BUTTON_SAVE = new[] {"Save", "保存"}; } public class NormalGatherer { public static Dictionary GetNormalAveragesFast(int[] sharedIndices, Vector3[] vertices, Vector3[] normals) { var normalAverages = new Dictionary(); string message = "Generating averages"; for(int i = 0; i < sharedIndices.Length; i++) { int index = sharedIndices[i]; Vector3 pos = vertices[index]; if(!normalAverages.ContainsKey(pos)) { normalAverages[pos] = normals[index]; continue; } normalAverages[pos] += normals[index]; if(PoiOutlineUtilWindow.DrawProgress(message, i, (float)i / (float)vertices.Length)) return normalAverages; } var keys = normalAverages.Keys.ToArray(); for(int j = 0; j < keys.Length; j++) { normalAverages[keys[j]] = Vector3.Normalize(normalAverages[keys[j]]); } return normalAverages; } public static Vector3 GetClosestNormal(Dictionary normalAverages, Vector3 pos) { if(normalAverages.ContainsKey(pos)) return normalAverages[pos]; float closestDist = 1000.0f; Vector3 closestNormal = new Vector3(0,0,0); foreach(KeyValuePair normalAverage in normalAverages) { float dist = Vector3.Distance(pos, normalAverage.Key); closestDist = dist < closestDist ? dist : closestDist; closestNormal = dist < closestDist ? normalAverage.Value : closestNormal; } return closestNormal; } } } #endif