Firstborn/Library/PackageCache/com.unity.terrain-tools@4.0.3/Editor/TerrainToolbox/TerrainToolboxVisualization.cs
Schaken-Mods b486678290 Library -Artifacts
Library -Artifacts
2023-03-28 12:24:16 -05:00

629 lines
24 KiB
C#

using System.Collections.Generic;
using System.IO;
using UnityEditor.SceneManagement;
using UnityEngine;
using UnityEngine.SceneManagement;
using System.Linq;
namespace UnityEditor.TerrainTools
{
[ExecuteInEditMode]
internal class TerrainToolboxVisualization
{
TerrainVisualizationSettings m_Settings = ScriptableObject.CreateInstance<TerrainVisualizationSettings>();
enum VISUALIZERMODE { None, AltitudeHeatmap };
VISUALIZERMODE m_selectedMode = VISUALIZERMODE.None;
VISUALIZERMODE m_previousMode = VISUALIZERMODE.None;
List<Terrain> m_Terrains = new List<Terrain>();
List<Material> m_TerrainMaterials = new List<Material>();
static Material m_VisualizationMaterial;
#if UNITY_2019_2_OR_NEWER
#else
Terrain.MaterialType m_TerrainMaterialType;
float m_TerrainLegacyShininess;
Color m_TerrainLegacySpecular;
#endif
bool m_ModeWarning = false;
string m_PresetPath = string.Empty;
float m_TerrainMaxHeight = 0f;
//Preset
TerrainVisualizationSettings m_SelectedPreset;
//Visualization shader paths
const string k_HDRPShaderPath = "Packages/com.unity.terrain-tools/Shaders/Visualization/HDRP/HDRP_TerrainVisualization.hdrpterraintoolshader";
const string k_URPShaderPath = "Packages/com.unity.terrain-tools/Shaders/Visualization/URP/URP_TerrainVisualizationLit.urpterraintoolshader";
static class Styles
{
//General
public static readonly GUIContent VisualizationSettings = EditorGUIUtility.TrTextContent("Visualization Modes", "Select from visualization modes.");
public static readonly string ModeWarningSettings = "There are no terrains within the scene. Add a terrain to use the visualization tool";
public static readonly string CompatabilityWarningSettings = "Terrain Visualization isn't compatible with HDRP at this time";
//Heatmap
public static readonly GUIContent ReferenceSpace = EditorGUIUtility.TrTextContent("Reference Space", "Select to either visualize in world space or local space.");
public static readonly GUIContent SeaLevel = EditorGUIUtility.TrTextContent("Sea Level", "Height of sea level.");
public static readonly GUIContent HeatLevels = EditorGUIUtility.TrTextContent("Levels", "The number of heat levels.");
public static readonly GUIContent MeasurementUnit = EditorGUIUtility.TrTextContent("Measurement Unit", "The measurement unit used to determine heat map altitude.");
//Settings
public static readonly GUIContent Preset = EditorGUIUtility.TrTextContent("Preset", "Preset used to visualize terrain. Select a pre-saved visualization preset asset or create a new preset.");
public static readonly GUIContent SavePreset = EditorGUIUtility.TrTextContent("Save", "Save the current preset with current settings.");
public static readonly GUIContent SaveAsPreset = EditorGUIUtility.TrTextContent("Save As", "Save the current preset as a new preset asset file.");
public static readonly GUIContent RefreshPreset = EditorGUIUtility.TrTextContent("Refresh", "Load selected preset and apply to current visualization settings");
}
public void OnGUI()
{
//Scroll view of settings
EditorGUIUtility.hierarchyMode = true;
TerrainToolboxUtilities.DrawSeperatorLine();
EditorGUILayout.BeginHorizontal();
EditorGUI.BeginChangeCheck();
EditorGUILayout.LabelField(Styles.VisualizationSettings, EditorStyles.boldLabel);
m_selectedMode = (VISUALIZERMODE)EditorGUILayout.EnumPopup(m_selectedMode);
bool hadVisualModeChange = false;
if (EditorGUI.EndChangeCheck())
{
ToggleVisualization();
hadVisualModeChange = true;
}
EditorGUILayout.EndHorizontal();
if (m_selectedMode == VISUALIZERMODE.AltitudeHeatmap)
{
ShowHeatmapGUI(hadVisualModeChange);
}
if (m_ModeWarning)
{
EditorGUILayout.HelpBox(Styles.ModeWarningSettings, MessageType.Warning);
}
ShowPresetGUI();
}
public void ToggleVisualization()
{
if (GameObject.FindObjectOfType<Terrain>() == null)
{
m_selectedMode = VISUALIZERMODE.None;
m_Settings.ModeWarning = true;
return;
}
if (m_Terrains == null || GameObject.FindObjectsOfType<Terrain>().Length != m_Terrains.Count || m_Terrains[0] == null)
{
m_Terrains.Clear();
m_Terrains.AddRange(ToolboxHelper.GetAllTerrainsInScene());
m_Settings.TerrainMaxHeight = m_Terrains[0].terrainData.size.y;
}
switch (m_selectedMode)
{
case VISUALIZERMODE.AltitudeHeatmap:
RevertMaterial();
RefreshTerrainInScene();
UpdateHeatmapSettings();
break;
case VISUALIZERMODE.None:
RevertMaterial();
break;
}
m_ModeWarning = false;
m_previousMode = m_selectedMode;
}
void ShowPresetGUI()
{
TerrainToolboxUtilities.DrawSeperatorLine();
EditorGUILayout.BeginHorizontal();
EditorGUILayout.LabelField(Styles.Preset, EditorStyles.boldLabel);
EditorGUI.BeginChangeCheck();
m_SelectedPreset = (TerrainVisualizationSettings)EditorGUILayout.ObjectField(m_SelectedPreset, typeof(TerrainVisualizationSettings), false);
if (EditorGUI.EndChangeCheck() && m_SelectedPreset != null)
{
if (EditorUtility.DisplayDialog("Confirm", "Load terrain creation settings from selected preset?", "OK", "Cancel"))
{
LoadVisualizationSettings();
}
}
if (GUILayout.Button(Styles.SavePreset))
{
if (m_SelectedPreset == null)
{
if (EditorUtility.DisplayDialog("Confirm", "No preset selected. Create a new preset?", "Continue", "Cancel"))
{
CreateNewPreset();
}
}
else
{
TransferSettings(m_Settings, m_SelectedPreset);
AssetDatabase.SaveAssets();
AssetDatabase.Refresh();
ToggleVisualization();
}
}
if (GUILayout.Button(Styles.SaveAsPreset))
{
CreateNewPreset();
}
if (GUILayout.Button(Styles.RefreshPreset))
{
LoadVisualizationSettings();
}
EditorGUILayout.EndHorizontal();
}
void ShowHeatmapGUI(bool needsUpdate = false)
{
EditorGUILayout.BeginFadeGroup(Mathf.Clamp((int)m_selectedMode, 0, 1));
//Color chooser GUI
EditorGUI.indentLevel++;
EditorGUI.BeginChangeCheck();
m_Settings.ReferenceSpace = (TerrainVisualizationSettings.REFERENCESPACE)EditorGUILayout.EnumPopup(Styles.ReferenceSpace, m_Settings.ReferenceSpace);
needsUpdate = needsUpdate || EditorGUI.EndChangeCheck();
if (m_Settings.ReferenceSpace == TerrainVisualizationSettings.REFERENCESPACE.WorldSpace)
{
EditorGUI.BeginChangeCheck();
m_Settings.SeaLevel = EditorGUILayout.FloatField(Styles.SeaLevel, m_Settings.SeaLevel, GUILayout.ExpandWidth(false));
needsUpdate = needsUpdate || EditorGUI.EndChangeCheck();
}
//Measure GUI
EditorGUI.BeginChangeCheck();
m_Settings.CurrentMeasure = (TerrainVisualizationSettings.MEASUREMENTS)EditorGUILayout.EnumPopup(Styles.MeasurementUnit, m_Settings.CurrentMeasure);
if (EditorGUI.EndChangeCheck())
{
ConvertUnits();
}
//Heat Levels GUI
EditorGUILayout.BeginHorizontal();
EditorGUILayout.LabelField(Styles.HeatLevels, GUILayout.MaxWidth(100));
EditorGUI.BeginChangeCheck();
m_Settings.HeatLevels = EditorGUILayout.IntSlider(m_Settings.HeatLevels, 1, 8);
EditorGUILayout.EndHorizontal();
GUILayout.BeginHorizontal();
GUILayout.Space(20f);
EditorGUILayout.BeginVertical("Box");
float min = m_Settings.DistanceSelection[0];
float max = m_Settings.DistanceSelection[m_Settings.HeatLevels - 1];
for (int i = 0; i < m_Settings.HeatLevels; i++)
{
float distance = (float)System.Math.Round(m_Settings.DistanceSelection[i], 2);
EditorGUILayout.BeginHorizontal();
if (m_Settings.ReferenceSpace == TerrainVisualizationSettings.REFERENCESPACE.WorldSpace)
m_Settings.DistanceSelection[i] = EditorGUILayout.FloatField(distance, GUILayout.Width(70));
else
m_Settings.DistanceSelection[i] = Mathf.Clamp(EditorGUILayout.FloatField(distance, GUILayout.Width(70)), 0, m_TerrainMaxHeight);
float height = m_Settings.DistanceSelection[i];
//Compare the distances
EditorGUI.indentLevel--;
if (m_Settings.CurrentMeasure == TerrainVisualizationSettings.MEASUREMENTS.Feet)
{
EditorGUILayout.LabelField(new GUIContent("ft"), GUILayout.Width(30));
height /= TerrainVisualizationSettings.CONVERSIONNUM;
}
else
{
EditorGUILayout.LabelField(new GUIContent("m"), GUILayout.Width(30));
}
m_Settings.MaxDistance = Mathf.Max(height, max);
m_Settings.MinDistance = Mathf.Min(height, min);
EditorGUI.indentLevel--;
m_Settings.ColorSelection[i] = EditorGUILayout.ColorField(m_Settings.ColorSelection[i]);
EditorGUI.indentLevel += 2;
EditorGUILayout.EndHorizontal();
}
if (EditorGUI.EndChangeCheck() || needsUpdate)
{
ConfigureHeatLevels();
UpdateHeatmapSettings();
}
EditorGUILayout.EndVertical();
EditorGUILayout.EndHorizontal();
EditorGUILayout.EndFadeGroup();
EditorGUI.indentLevel--;
}
void UpdateHeatmapSettings()
{
GetAndSetActiveRenderPipelineSettings();
MaterialPropertyBlock heatmapBlock = new MaterialPropertyBlock();
Vector4 shaderParams = new Vector4(m_Settings.MinDistance, m_Settings.MaxDistance, m_Settings.SeaLevel, 0);
if (m_Settings.ReferenceSpace == TerrainVisualizationSettings.REFERENCESPACE.WorldSpace
&& m_Settings.CurrentMeasure == TerrainVisualizationSettings.MEASUREMENTS.Feet)
{
shaderParams /= TerrainVisualizationSettings.CONVERSIONNUM;
}
m_VisualizationMaterial.EnableKeyword("_HEATMAP");
m_VisualizationMaterial.DisableKeyword("_SPLATMAP_PREVIEW");
m_VisualizationMaterial.SetTexture("_HeatmapGradient", CreateGradient());
m_VisualizationMaterial.SetVector("_HeatmapData", shaderParams);
if (m_Settings.ReferenceSpace == TerrainVisualizationSettings.REFERENCESPACE.WorldSpace)
{
m_VisualizationMaterial.DisableKeyword("LOCAL_SPACE");
m_VisualizationMaterial.EnableKeyword("WORLD_SPACE");
}
else
{
m_VisualizationMaterial.DisableKeyword("WORLD_SPACE");
m_VisualizationMaterial.EnableKeyword("LOCAL_SPACE");
}
foreach (Terrain terrain in m_Terrains)
{
#if UNITY_2019_2_OR_NEWER
#else
terrain.materialType = Terrain.MaterialType.Custom;
#endif
terrain.materialTemplate = m_VisualizationMaterial;
heatmapBlock.Clear();
heatmapBlock.SetTexture("_HeatHeightmap", terrain.terrainData.heightmapTexture);
terrain.SetSplatMaterialPropertyBlock(heatmapBlock);
EditorUtility.SetDirty(terrain);
}
}
void ConvertUnits()
{
if (m_Settings.CurrentMeasure == TerrainVisualizationSettings.MEASUREMENTS.Feet)
{
for (int i = 0; i < m_Settings.HeatLevels; i++)
{
m_Settings.DistanceSelection[i] *= TerrainVisualizationSettings.CONVERSIONNUM;
}
m_TerrainMaxHeight *= TerrainVisualizationSettings.CONVERSIONNUM;
}
else
{
for (int i = 0; i < m_Settings.HeatLevels; i++)
{
m_Settings.DistanceSelection[i] /= TerrainVisualizationSettings.CONVERSIONNUM;
}
m_TerrainMaxHeight /= TerrainVisualizationSettings.CONVERSIONNUM;
}
}
void ConfigureHeatLevels()
{
m_Settings.MaxDistance = float.MinValue;
m_Settings.MinDistance = float.MaxValue;
int num = m_Settings.HeatLevels;
for (int i = 0; i < num; i++)
{
var height = m_Settings.DistanceSelection[i] + 100;
//Compare the distances
if (m_Settings.CurrentMeasure == TerrainVisualizationSettings.MEASUREMENTS.Feet)
height /= TerrainVisualizationSettings.CONVERSIONNUM;
m_Settings.MaxDistance = Mathf.Max(height, m_Settings.MaxDistance);
m_Settings.MinDistance = Mathf.Min(height, m_Settings.MinDistance);
}
}
Texture2D CreateGradient()
{
Color[] colors = m_Settings.ColorSelection.ToArray();
if (colors.Length == 0 || colors == null)
return null;
int textureWidth = 256;
int textureHeight = 1;
int length = m_Settings.HeatLevels;
GradientColorKey[] colorKeys = new GradientColorKey[length];
GradientAlphaKey[] alphaKeys = new GradientAlphaKey[length];
//Configure new gradient data
for (int i = 0; i < length; i++)
{
float distance = m_Settings.DistanceSelection[i];
float step;
if (m_Settings.ReferenceSpace == TerrainVisualizationSettings.REFERENCESPACE.WorldSpace)
{
step = (distance - m_Settings.MinDistance) / (m_Settings.MaxDistance - m_Settings.MinDistance);
}
else
{
step = distance / m_TerrainMaxHeight;
}
colorKeys[i].color = colors[i];
colorKeys[i].time = step;
alphaKeys[i].alpha = colors[i].a;
alphaKeys[i].time = step;
}
//Create gradient
Gradient gradient = new Gradient();
gradient.SetKeys(colorKeys, alphaKeys);
//Create texture
Texture2D texture = new Texture2D(textureWidth, textureHeight, TextureFormat.ARGB32, false, false);
texture.wrapMode = TextureWrapMode.Clamp;
for (int i = 0; i < textureWidth; i++)
{
texture.SetPixel(i, 0, gradient.Evaluate((float)i / (float)textureWidth));
}
texture.Apply(false);
return texture;
}
public void RevertMaterial()
{
GetAndSetActiveRenderPipelineSettings();
if (m_VisualizationMaterial != null)
{
m_VisualizationMaterial.DisableKeyword("_HEATMAP");
}
for (int i = 0; i < m_Terrains.Count && m_TerrainMaterials.Count > 0; i++)
{
if (m_Terrains[i] != null)
{
#if UNITY_2019_2_OR_NEWER
m_Terrains[i].materialTemplate = m_TerrainMaterials.Count > i ? m_TerrainMaterials[i] : m_TerrainMaterials.First();
#else
m_Terrains[i].materialType = m_TerrainMaterialType;
if (m_TerrainMaterialType == Terrain.MaterialType.Custom)
{
m_Terrains[i].materialTemplate = m_TerrainMaterials.Count > i ? m_TerrainMaterials[i] : m_TerrainMaterials[0];
}
else if (m_TerrainMaterialType == Terrain.MaterialType.BuiltInLegacySpecular)
{
m_Terrains[i].legacyShininess = m_TerrainLegacyShininess;
m_Terrains[i].legacySpecular = m_TerrainLegacySpecular;
m_Terrains[i].materialTemplate = null;
}
else
{
m_Terrains[i].materialTemplate = null;
}
#endif
}
}
SceneView.RepaintAll();
}
void RefreshTerrainInScene()
{
m_Terrains.Clear();
m_Terrains.AddRange(ToolboxHelper.GetAllTerrainsInScene());
if (m_Terrains.Count == 0)
{
m_ModeWarning = true;
return;
}
m_TerrainMaxHeight = m_Terrains.Max(t => t.terrainData.size.y);
m_TerrainMaterials.Clear();
foreach (Terrain terrain in m_Terrains)
{
m_TerrainMaterials.Add(terrain.materialTemplate);
}
m_ModeWarning = false;
}
public static Material GetVizMaterial(Shader vizShader)
{
if(m_VisualizationMaterial != null) return m_VisualizationMaterial;
return (m_VisualizationMaterial = new Material(vizShader));
}
private void VisualizationSetup()
{
ToolboxHelper.RenderPipeline m_ActiveRenderPipeline = ToolboxHelper.GetRenderPipeline();
Shader vizShader = Shader.Find("Hidden/Builtin_TerrainVisualization");
switch (m_ActiveRenderPipeline)
{
case ToolboxHelper.RenderPipeline.HD:
vizShader = AssetDatabase.LoadAssetAtPath<Shader>(k_HDRPShaderPath);
break;
case ToolboxHelper.RenderPipeline.Universal:
vizShader = AssetDatabase.LoadAssetAtPath<Shader>(k_URPShaderPath);
break;
default:
if (m_Terrains == null || m_Terrains.Count == 0)
{
break;
}
#if UNITY_2019_2_OR_NEWER
#else
m_TerrainMaterialType = m_Terrains[0].materialType;
if (m_TerrainMaterialType == Terrain.MaterialType.BuiltInLegacySpecular)
{
m_TerrainLegacyShininess = m_Terrains[0].legacyShininess;
m_TerrainLegacySpecular = m_Terrains[0].legacySpecular;
}
#endif
break;
}
m_VisualizationMaterial = GetVizMaterial(vizShader);
}
void GetAndSetActiveRenderPipelineSettings()
{
VisualizationSetup();
}
void CreateNewPreset()
{
string filePath = EditorUtility.SaveFilePanelInProject("Create Terrain Visualization Settings", "New Terrain Visualization.asset", "asset", "");
if (string.IsNullOrEmpty(filePath))
{
return;
}
m_SelectedPreset = ScriptableObject.CreateInstance<TerrainVisualizationSettings>();
TransferSettings(m_Settings, m_SelectedPreset);
AssetDatabase.CreateAsset(m_SelectedPreset, filePath);
AssetDatabase.SaveAssets();
AssetDatabase.Refresh();
ToggleVisualization();
}
bool GetVisualizationSettingsPreset()
{
if (m_SelectedPreset == null)
{
if (EditorUtility.DisplayDialog("Error", "No terrain visualization setting found, create a new one?", "OK", "Cancel"))
{
CreateNewPreset();
return true;
}
else
{
return false;
}
}
return true;
}
void LoadVisualizationSettings()
{
if (m_SelectedPreset == null)
{
EditorUtility.DisplayDialog("Error", "No selected preset found. Select one to continue.", "OK");
return;
}
else
{
TransferSettings(m_SelectedPreset, m_Settings);
if (m_selectedMode == VISUALIZERMODE.AltitudeHeatmap)
{
ConfigureHeatLevels();
UpdateHeatmapSettings();
}
}
}
public void SaveSettings()
{
if (m_SelectedPreset != null)
{
m_PresetPath = AssetDatabase.GetAssetPath(m_SelectedPreset);
}
else
{
m_PresetPath = string.Empty;
}
string filePath = ToolboxHelper.GetPrefFilePath(ToolboxHelper.ToolboxPrefsVisualization);
string settingsData = JsonUtility.ToJson(m_Settings);
File.WriteAllText(filePath, settingsData);
SceneView.RepaintAll();
EditorSceneManager.sceneSaving -= OnSceneSaving;
EditorSceneManager.sceneSaved -= OnSceneSaved;
EditorSceneManager.sceneOpened -= OnSceneOpened;
EditorApplication.playModeStateChanged -= PlayModeChanged;
Undo.undoRedoPerformed -= OnUndo;
}
public void LoadSettings()
{
string filePath = ToolboxHelper.GetPrefFilePath(ToolboxHelper.ToolboxPrefsVisualization);
if (File.Exists(filePath))
{
string settingsData = File.ReadAllText(filePath);
JsonUtility.FromJsonOverwrite(settingsData, m_Settings);
}
if (m_PresetPath == string.Empty)
{
m_SelectedPreset = null;
}
else
{
m_SelectedPreset = AssetDatabase.LoadAssetAtPath(m_PresetPath, typeof(TerrainVisualizationSettings)) as TerrainVisualizationSettings;
}
EditorSceneManager.sceneSaving += OnSceneSaving;
EditorSceneManager.sceneSaved += OnSceneSaved;
EditorSceneManager.sceneOpened += OnSceneOpened;
EditorApplication.playModeStateChanged += PlayModeChanged;
Undo.undoRedoPerformed += OnUndo;
}
void TransferSettings(TerrainVisualizationSettings fromSettings, TerrainVisualizationSettings toSettings)
{
fromSettings.ColorSelection.CopyTo(toSettings.ColorSelection, 0);
fromSettings.DistanceSelection.CopyTo(toSettings.DistanceSelection, 0);
toSettings.CurrentMeasure = fromSettings.CurrentMeasure;
toSettings.HeatLevels = fromSettings.HeatLevels;
toSettings.SeaLevel = fromSettings.SeaLevel;
toSettings.MaxDistance = fromSettings.MaxDistance;
toSettings.MinDistance = fromSettings.MinDistance;
toSettings.ReferenceSpace = fromSettings.ReferenceSpace;
toSettings.WorldSpace = fromSettings.WorldSpace;
}
void Reset()
{
m_selectedMode = VISUALIZERMODE.None;
m_VisualizationMaterial = null;
m_Terrains.Clear();
m_TerrainMaterials.Clear();
}
void OnSceneSaving(Scene scene, string path)
{
if (m_selectedMode != VISUALIZERMODE.None)
{
RevertMaterial();
}
}
void OnSceneSaved(Scene scene)
{
Reset();
}
void OnSceneOpened(Scene scene, OpenSceneMode open)
{
Reset();
}
void OnUndo()
{
if (m_selectedMode == VISUALIZERMODE.AltitudeHeatmap)
{
UpdateHeatmapSettings();
}
}
void PlayModeChanged(PlayModeStateChange state)
{
if (m_selectedMode != VISUALIZERMODE.None)
{
RevertMaterial();
}
}
}
}