91 lines
2.8 KiB
C#
91 lines
2.8 KiB
C#
using UnityEngine;
|
||
using UnityEditor;
|
||
using System.Collections.Generic;
|
||
|
||
public class RemoveTreesUnderTerrainLayer : EditorWindow
|
||
{
|
||
public Terrain terrain;
|
||
public string targetLayerName = "Road";
|
||
public float threshold = 0.5f; // Road图层强度大于该值则认为是Road
|
||
|
||
[MenuItem("Tools/Terrain/Remove Trees Under Terrain Layer")]
|
||
public static void ShowWindow()
|
||
{
|
||
GetWindow<RemoveTreesUnderTerrainLayer>("Remove Trees by Terrain Layer");
|
||
}
|
||
|
||
private void OnGUI()
|
||
{
|
||
terrain = (Terrain)EditorGUILayout.ObjectField("目标 Terrain", terrain, typeof(Terrain), true);
|
||
targetLayerName = EditorGUILayout.TextField("目标 Terrain Layer 名", targetLayerName);
|
||
threshold = EditorGUILayout.Slider("图层权重阈值", threshold, 0f, 1f);
|
||
|
||
if (GUILayout.Button("移除指定图层下的树"))
|
||
{
|
||
if (terrain != null)
|
||
{
|
||
RemoveTreesByLayer();
|
||
}
|
||
else
|
||
{
|
||
Debug.LogError("未指定 Terrain！");
|
||
}
|
||
}
|
||
}
|
||
|
||
private void RemoveTreesByLayer()
|
||
{
|
||
TerrainData terrainData = terrain.terrainData;
|
||
TreeInstance[] allTrees = terrainData.treeInstances;
|
||
List<TreeInstance> remainingTrees = new List<TreeInstance>();
|
||
|
||
int mapWidth = terrainData.alphamapWidth;
|
||
int mapHeight = terrainData.alphamapHeight;
|
||
float[,,] alphaMaps = terrainData.GetAlphamaps(0, 0, mapWidth, mapHeight);
|
||
|
||
int targetLayerIndex = -1;
|
||
var terrainLayers = terrainData.terrainLayers;
|
||
|
||
for (int i = 0; i < terrainLayers.Length; i++)
|
||
{
|
||
if (terrainLayers[i].name == targetLayerName)
|
||
{
|
||
targetLayerIndex = i;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (targetLayerIndex == -1)
|
||
{
|
||
Debug.LogError($"找不到 Terrain Layer：{targetLayerName}");
|
||
return;
|
||
}
|
||
|
||
Vector3 size = terrainData.size;
|
||
Vector3 terrainPos = terrain.transform.position;
|
||
int removed = 0;
|
||
|
||
foreach (var tree in allTrees)
|
||
{
|
||
Vector3 worldPos = Vector3.Scale(tree.position, size) + terrainPos;
|
||
|
||
int mapX = Mathf.Clamp(Mathf.RoundToInt((tree.position.x) * mapWidth), 0, mapWidth - 1);
|
||
int mapZ = Mathf.Clamp(Mathf.RoundToInt((tree.position.z) * mapHeight), 0, mapHeight - 1);
|
||
|
||
float layerWeight = alphaMaps[mapZ, mapX, targetLayerIndex];
|
||
|
||
if (layerWeight < threshold)
|
||
{
|
||
remainingTrees.Add(tree);
|
||
}
|
||
else
|
||
{
|
||
removed++;
|
||
}
|
||
}
|
||
|
||
terrainData.treeInstances = remainingTrees.ToArray();
|
||
Debug.Log($"✅ 清除完成：移除 {removed} 棵树，保留 {remainingTrees.Count} 棵。");
|
||
}
|
||
}
|