82 lines
2.8 KiB
C#
82 lines
2.8 KiB
C#
|
using UnityEngine;
|
|||
|
|
using UnityEditor;
|
|||
|
|
|
|||
|
|
public class FlattenTerrainLayerToWorldHeight : EditorWindow
|
|||
|
|
{
|
|||
|
|
Terrain terrain;
|
|||
|
|
string targetLayerName = "Road";
|
|||
|
|
float targetWorldHeight = 10f; // 米
|
|||
|
|
float layerThreshold = 0.5f;
|
|||
|
|
|
|||
|
|
[MenuItem("Tools/Terrain/Flatten Layer To World Height")]
|
|||
|
|
public static void ShowWindow()
|
|||
|
|
{
|
|||
|
|
GetWindow<FlattenTerrainLayerToWorldHeight>("Flatten Layer Height");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void OnGUI()
|
|||
|
|
{
|
|||
|
|
terrain = (Terrain)EditorGUILayout.ObjectField("目标 Terrain", terrain, typeof(Terrain), true);
|
|||
|
|
targetLayerName = EditorGUILayout.TextField("目标 Terrain Layer 名", targetLayerName);
|
|||
|
|
targetWorldHeight = EditorGUILayout.FloatField("世界空间高度 (米)", targetWorldHeight);
|
|||
|
|
layerThreshold = EditorGUILayout.Slider("图层混合阈值", layerThreshold, 0f, 1f);
|
|||
|
|
|
|||
|
|
if (GUILayout.Button("拉平图层区域至目标高度"))
|
|||
|
|
{
|
|||
|
|
if (terrain == null) terrain = Terrain.activeTerrain;
|
|||
|
|
if (terrain != null) FlattenLayerArea();
|
|||
|
|
else Debug.LogError("未找到 Terrain");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void FlattenLayerArea()
|
|||
|
|
{
|
|||
|
|
TerrainData data = terrain.terrainData;
|
|||
|
|
int heightmapWidth = data.heightmapResolution;
|
|||
|
|
int heightmapHeight = data.heightmapResolution;
|
|||
|
|
float[,] heights = data.GetHeights(0, 0, heightmapWidth, heightmapHeight);
|
|||
|
|
|
|||
|
|
int alphaWidth = data.alphamapWidth;
|
|||
|
|
int alphaHeight = data.alphamapHeight;
|
|||
|
|
float[,,] alphaMaps = data.GetAlphamaps(0, 0, alphaWidth, alphaHeight);
|
|||
|
|
|
|||
|
|
int layerIndex = -1;
|
|||
|
|
for (int i = 0; i < data.terrainLayers.Length; i++)
|
|||
|
|
{
|
|||
|
|
if (data.terrainLayers[i].name == targetLayerName)
|
|||
|
|
{
|
|||
|
|
layerIndex = i;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (layerIndex == -1)
|
|||
|
|
{
|
|||
|
|
Debug.LogError("找不到指定的 Terrain Layer：" + targetLayerName);
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
float normalizedHeight = targetWorldHeight / data.size.y;
|
|||
|
|
int changed = 0;
|
|||
|
|
|
|||
|
|
for (int y = 0; y < heightmapHeight; y++)
|
|||
|
|
{
|
|||
|
|
for (int x = 0; x < heightmapWidth; x++)
|
|||
|
|
{
|
|||
|
|
int alphaX = Mathf.Clamp(Mathf.RoundToInt((float)x / heightmapWidth * alphaWidth), 0, alphaWidth - 1);
|
|||
|
|
int alphaY = Mathf.Clamp(Mathf.RoundToInt((float)y / heightmapHeight * alphaHeight), 0, alphaHeight - 1);
|
|||
|
|
float weight = alphaMaps[alphaY, alphaX, layerIndex];
|
|||
|
|
|
|||
|
|
if (weight > layerThreshold)
|
|||
|
|
{
|
|||
|
|
heights[y, x] = normalizedHeight;
|
|||
|
|
changed++;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
data.SetHeights(0, 0, heights);
|
|||
|
|
Debug.Log($"✅ 拉平完成！将 {changed} 个点高度设置为 {targetWorldHeight} 米");
|
|||
|
|
}
|
|||
|
|
}
|