104 lines
3.2 KiB
C#
104 lines
3.2 KiB
C#
|
using UnityEngine;
|
|||
|
|
using UnityEditor;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
|
|||
|
|
public class BatchAddTreesToTerrain : EditorWindow
|
|||
|
|
{
|
|||
|
|
Terrain targetTerrain;
|
|||
|
|
List<GameObject> treePrefabs = new List<GameObject>();
|
|||
|
|
|
|||
|
|
[MenuItem("Tools/Terrain/Batch Add Trees")]
|
|||
|
|
static void OpenWindow()
|
|||
|
|
{
|
|||
|
|
GetWindow<BatchAddTreesToTerrain>("Batch Add Trees");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void OnGUI()
|
|||
|
|
{
|
|||
|
|
GUILayout.Label("🌲 批量添加树到 Terrain", EditorStyles.boldLabel);
|
|||
|
|
targetTerrain = (Terrain)EditorGUILayout.ObjectField("目标 Terrain", targetTerrain, typeof(Terrain), true);
|
|||
|
|
|
|||
|
|
EditorGUILayout.Space();
|
|||
|
|
GUILayout.Label("树 Prefab 列表", EditorStyles.boldLabel);
|
|||
|
|
|
|||
|
|
// 拖拽区域
|
|||
|
|
var dropArea = GUILayoutUtility.GetRect(0, 50, GUILayout.ExpandWidth(true));
|
|||
|
|
GUI.Box(dropArea, "将 Tree Prefab 拖到这里");
|
|||
|
|
|
|||
|
|
// 检测拖入
|
|||
|
|
Event evt = Event.current;
|
|||
|
|
if (evt.type == EventType.DragUpdated || evt.type == EventType.DragPerform)
|
|||
|
|
{
|
|||
|
|
if (dropArea.Contains(evt.mousePosition))
|
|||
|
|
{
|
|||
|
|
DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
|
|||
|
|
|
|||
|
|
if (evt.type == EventType.DragPerform)
|
|||
|
|
{
|
|||
|
|
DragAndDrop.AcceptDrag();
|
|||
|
|
foreach (var obj in DragAndDrop.objectReferences)
|
|||
|
|
{
|
|||
|
|
if (obj is GameObject go)
|
|||
|
|
{
|
|||
|
|
if (!treePrefabs.Contains(go))
|
|||
|
|
treePrefabs.Add(go);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
evt.Use();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 展示列表 + 删除按钮
|
|||
|
|
int removeIndex = -1;
|
|||
|
|
for (int i = 0; i < treePrefabs.Count; i++)
|
|||
|
|
{
|
|||
|
|
EditorGUILayout.BeginHorizontal();
|
|||
|
|
treePrefabs[i] = (GameObject)EditorGUILayout.ObjectField(treePrefabs[i], typeof(GameObject), false);
|
|||
|
|
if (GUILayout.Button("X", GUILayout.Width(20)))
|
|||
|
|
removeIndex = i;
|
|||
|
|
EditorGUILayout.EndHorizontal();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (removeIndex >= 0)
|
|||
|
|
treePrefabs.RemoveAt(removeIndex);
|
|||
|
|
|
|||
|
|
EditorGUILayout.Space();
|
|||
|
|
|
|||
|
|
if (GUILayout.Button("✅ 添加到 Terrain"))
|
|||
|
|
{
|
|||
|
|
AddTreesToTerrain();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void AddTreesToTerrain()
|
|||
|
|
{
|
|||
|
|
if (targetTerrain == null)
|
|||
|
|
{
|
|||
|
|
Debug.LogWarning("请指定目标 Terrain！");
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
TerrainData terrainData = targetTerrain.terrainData;
|
|||
|
|
List<TreePrototype> existingPrototypes = new List<TreePrototype>(terrainData.treePrototypes);
|
|||
|
|
|
|||
|
|
int added = 0;
|
|||
|
|
foreach (var prefab in treePrefabs)
|
|||
|
|
{
|
|||
|
|
if (prefab == null) continue;
|
|||
|
|
|
|||
|
|
bool alreadyExists = existingPrototypes.Exists(p => p.prefab == prefab);
|
|||
|
|
if (!alreadyExists)
|
|||
|
|
{
|
|||
|
|
TreePrototype newProto = new TreePrototype { prefab = prefab };
|
|||
|
|
existingPrototypes.Add(newProto);
|
|||
|
|
added++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
terrainData.treePrototypes = existingPrototypes.ToArray();
|
|||
|
|
|
|||
|
|
Debug.Log($"✅ 已添加 {added} 个新树种到 Terrain！");
|
|||
|
|
}
|
|||
|
|
}
|