89 lines
3.0 KiB
C#
89 lines
3.0 KiB
C#
|
using UnityEngine;
|
|||
|
|
using UnityEditor;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
|
|||
|
|
public class ReplacePrefabsWithTerrainTrees : EditorWindow
|
|||
|
|
{
|
|||
|
|
Terrain targetTerrain;
|
|||
|
|
|
|||
|
|
[MenuItem("Tools/Terrain/Replace Prefabs with Trees")]
|
|||
|
|
static void ShowWindow()
|
|||
|
|
{
|
|||
|
|
GetWindow<ReplacePrefabsWithTerrainTrees>("Replace Prefabs");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void OnGUI()
|
|||
|
|
{
|
|||
|
|
EditorGUILayout.HelpBox("先在 Project 里选中你想替换的 Prefabs，然后点击按钮。", MessageType.Info);
|
|||
|
|
|
|||
|
|
targetTerrain = (Terrain)EditorGUILayout.ObjectField("目标 Terrain", targetTerrain, typeof(Terrain), true);
|
|||
|
|
|
|||
|
|
if (GUILayout.Button("替换选中 Prefab 实例为 Terrain 树"))
|
|||
|
|
{
|
|||
|
|
if (targetTerrain == null)
|
|||
|
|
targetTerrain = Terrain.activeTerrain;
|
|||
|
|
|
|||
|
|
if (targetTerrain == null)
|
|||
|
|
{
|
|||
|
|
Debug.LogError("未指定 Terrain，也未检测到活动地形。");
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ReplacePrefabs();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void ReplacePrefabs()
|
|||
|
|
{
|
|||
|
|
Object[] selectedPrefabs = Selection.GetFiltered(typeof(GameObject), SelectionMode.Assets);
|
|||
|
|
if (selectedPrefabs.Length == 0)
|
|||
|
|
{
|
|||
|
|
Debug.LogWarning("你需要在 Project 面板中选择 Prefab！");
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
TerrainData terrainData = targetTerrain.terrainData;
|
|||
|
|
Vector3 terrainPos = targetTerrain.transform.position;
|
|||
|
|
|
|||
|
|
List<TreeInstance> newTrees = new List<TreeInstance>(terrainData.treeInstances);
|
|||
|
|
int count = 0;
|
|||
|
|
|
|||
|
|
foreach (GameObject prefab in selectedPrefabs)
|
|||
|
|
{
|
|||
|
|
GameObject[] instances = FindObjectsOfType<GameObject>();
|
|||
|
|
foreach (GameObject obj in instances)
|
|||
|
|
{
|
|||
|
|
if (PrefabUtility.GetCorrespondingObjectFromSource(obj) == prefab)
|
|||
|
|
{
|
|||
|
|
Vector3 worldPos = obj.transform.position;
|
|||
|
|
|
|||
|
|
// 转换为 Terrain 本地坐标（归一化）
|
|||
|
|
Vector3 normalizedPos = (worldPos - terrainPos);
|
|||
|
|
normalizedPos.x /= terrainData.size.x;
|
|||
|
|
normalizedPos.z /= terrainData.size.z;
|
|||
|
|
normalizedPos.y = 0f;
|
|||
|
|
|
|||
|
|
// 选择一个随机树种（TreePrototype）
|
|||
|
|
int prototypeIndex = Random.Range(0, terrainData.treePrototypes.Length);
|
|||
|
|
TreeInstance tree = new TreeInstance
|
|||
|
|
{
|
|||
|
|
position = new Vector3(normalizedPos.x, 0, normalizedPos.z),
|
|||
|
|
prototypeIndex = prototypeIndex,
|
|||
|
|
heightScale = 1f,
|
|||
|
|
widthScale = 1f,
|
|||
|
|
color = Color.white,
|
|||
|
|
lightmapColor = Color.white
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
newTrees.Add(tree);
|
|||
|
|
Undo.DestroyObjectImmediate(obj);
|
|||
|
|
count++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
terrainData.treeInstances = newTrees.ToArray();
|
|||
|
|
Debug.Log($"✅ 替换完成，处理了 {count} 个实例。添加到 Terrain 的 Tree 中。");
|
|||
|
|
}
|
|||
|
|
}
|