104 lines
4.3 KiB
C#
104 lines
4.3 KiB
C#
|
using UnityEngine;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using UnityEditor;
|
|||
|
|
[CustomEditor(typeof(SceneBlockDivide))]
|
|||
|
|
public class SceneBlockDivideEditor : Editor
|
|||
|
|
{
|
|||
|
|
private const string rootFormat = "Root_{0} ({1}, {2})";
|
|||
|
|
private const string multiMaterialRoot = "MultipleMaterials";
|
|||
|
|
private SceneBlockDivide mTarget;
|
|||
|
|
private int mLastBlockWidth;
|
|||
|
|
private Rect mapRect;
|
|||
|
|
private MonoScript script;
|
|||
|
|
private void OnEnable()
|
|||
|
|
{
|
|||
|
|
mTarget = (SceneBlockDivide)target;
|
|||
|
|
script = MonoScript.FromMonoBehaviour(mTarget);
|
|||
|
|
mLastBlockWidth = Mathf.Max(1, mTarget.blockWidth);
|
|||
|
|
RefreshSceneSize();
|
|||
|
|
}
|
|||
|
|
public override void OnInspectorGUI()
|
|||
|
|
{
|
|||
|
|
script = EditorGUILayout.ObjectField("Script", script, typeof(MonoScript), false) as MonoScript;
|
|||
|
|
mTarget.blockWidth = EditorGUILayout.IntField("Block Width", mTarget.blockWidth);
|
|||
|
|
if (mTarget.blockWidth >= 1 && mTarget.blockWidth != mLastBlockWidth)
|
|||
|
|
mLastBlockWidth = mTarget.blockWidth;
|
|||
|
|
var widthTiles = Mathf.CeilToInt(mapRect.width / mLastBlockWidth);
|
|||
|
|
var heightTiles = Mathf.CeilToInt(mapRect.height / mLastBlockWidth);
|
|||
|
|
EditorGUILayout.LabelField(string.Format("Width: {0}", widthTiles));
|
|||
|
|
EditorGUILayout.LabelField(string.Format("Height: {0}", heightTiles));
|
|||
|
|
if (GUILayout.Button("Refresh Scene", GUILayout.Width(100f)))
|
|||
|
|
RefreshSceneSize();
|
|||
|
|
GUILayout.Space(10f);
|
|||
|
|
if (GUILayout.Button("Divide Blocks", GUILayout.Width(100f)))
|
|||
|
|
{
|
|||
|
|
var meshRenderers = GetMeshRenderers();
|
|||
|
|
for (int i = 0; i < meshRenderers.Length; i++)
|
|||
|
|
{
|
|||
|
|
var multiShaders = false;
|
|||
|
|
var meshRenderer = meshRenderers[i];
|
|||
|
|
if (meshRenderer.sharedMaterials.Length > 1)
|
|||
|
|
{
|
|||
|
|
// 如果多个材质是同一个Shader，则视为普通物体，否则视为特殊物体
|
|||
|
|
for (int j = 1; j < meshRenderer.sharedMaterials.Length; j++)
|
|||
|
|
{
|
|||
|
|
if (meshRenderer.sharedMaterials[j - 1].shader != meshRenderer.sharedMaterials[j].shader)
|
|||
|
|
{
|
|||
|
|
multiShaders = true;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (multiShaders)
|
|||
|
|
{
|
|||
|
|
var root = GetRoot(multiMaterialRoot, 0, 0);
|
|||
|
|
meshRenderer.transform.SetParent(root);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
var x = Mathf.FloorToInt((meshRenderer.bounds.min.x - mapRect.min.x) / mLastBlockWidth);
|
|||
|
|
var z = Mathf.FloorToInt((meshRenderer.bounds.min.z - mapRect.min.y) / mLastBlockWidth);
|
|||
|
|
var root = GetRoot(meshRenderer.sharedMaterial.shader.name, x, z);
|
|||
|
|
meshRenderer.transform.SetParent(root);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
private void RefreshSceneSize()
|
|||
|
|
{
|
|||
|
|
var meshRenderers = GetMeshRenderers();
|
|||
|
|
if (meshRenderers.Length == 0)
|
|||
|
|
mapRect = new Rect();
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
var bounds = meshRenderers[0].bounds;
|
|||
|
|
for (int i = 1; i < meshRenderers.Length; i++)
|
|||
|
|
bounds.Encapsulate(meshRenderers[i].bounds);
|
|||
|
|
mapRect = new Rect(bounds.min.x, bounds.min.z, bounds.size.x, bounds.size.z);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
private MeshRenderer[] GetMeshRenderers()
|
|||
|
|
{
|
|||
|
|
var meshRenderers = Object.FindObjectsOfType<MeshRenderer>();
|
|||
|
|
var meshRenderList = new List<MeshRenderer>();
|
|||
|
|
for (int i = 0; i < meshRenderers.Length; i++)
|
|||
|
|
{
|
|||
|
|
if (meshRenderers[i].gameObject.tag != "SkyBox")
|
|||
|
|
meshRenderList.Add(meshRenderers[i]);
|
|||
|
|
}
|
|||
|
|
return meshRenderList.ToArray();
|
|||
|
|
}
|
|||
|
|
private static Transform GetRoot(string name, int x, int z)
|
|||
|
|
{
|
|||
|
|
var rootName = string.Format(rootFormat, name, x, z);
|
|||
|
|
var rootObj = GameObject.Find(rootName);
|
|||
|
|
if (rootObj == null)
|
|||
|
|
{
|
|||
|
|
rootObj = new GameObject(rootName);
|
|||
|
|
rootObj.transform.localPosition = Vector3.zero;
|
|||
|
|
rootObj.transform.localRotation = Quaternion.identity;
|
|||
|
|
rootObj.transform.localScale = Vector3.one;
|
|||
|
|
}
|
|||
|
|
return rootObj.transform;
|
|||
|
|
}
|
|||
|
|
}
|