BITFALL/Assets/Polaris - Low Poly Ecosystem/Polaris - Low Poly Terrain .../Editor/Scripts/StampTool/GTextureStamperInspector.cs

597 lines
25 KiB
C#
Raw Normal View History

2023-12-30 17:37:48 +08:00
using Pinwheel.Griffin.BackupTool;
using Pinwheel.Griffin.TextureTool;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
namespace Pinwheel.Griffin.StampTool
{
[CustomEditor(typeof(GTextureStamper))]
public class GTextureStamperInspector : Editor
{
private GTextureStamper instance;
private Dictionary<string, RenderTexture> previewTextures;
private MaterialPropertyBlock previewPropertyBlock;
private const string INSTRUCTION = "Stamp texture or color onto the terrain surface.";
private const string HISTORY_PREFIX_COLOR = "Stamp AMS";
private const string HISTORY_PREFIX_TEXTURE = "Stamp Splat";
private readonly Vector3[] worldPoints = new Vector3[4];
private readonly Vector2[] normalizedPoints = new Vector2[4];
private readonly Vector3[] worldBox = new Vector3[8];
private void OnEnable()
{
Undo.undoRedoPerformed += OnUndoRedo;
SceneView.duringSceneGui += DuringSceneGUI;
instance = target as GTextureStamper;
Tools.hidden = true;
instance.Internal_UpdateFalloffTexture();
instance.Internal_UpdateLayerTransitionTextures();
previewPropertyBlock = new MaterialPropertyBlock();
GCommon.RegisterBeginRender(OnCameraRender);
GCommon.RegisterBeginRenderSRP(OnCameraRenderSRP);
GCommon.UpdateMaterials(instance.GroupId);
UpdatePreview();
}
private void OnDisable()
{
Undo.undoRedoPerformed -= OnUndoRedo;
SceneView.duringSceneGui -= DuringSceneGUI;
Tools.hidden = false;
GCommon.UnregisterBeginRender(OnCameraRender);
GCommon.UnregisterBeginRenderSRP(OnCameraRenderSRP);
if (previewTextures != null)
{
foreach (string k in previewTextures.Keys)
{
RenderTexture rt = previewTextures[k];
if (rt == null)
continue;
rt.Release();
Object.DestroyImmediate(rt);
}
}
}
private void OnUndoRedo()
{
if (Selection.activeGameObject != instance.gameObject)
return;
if (string.IsNullOrEmpty(GUndoCompatibleBuffer.Instance.CurrentBackupName))
return;
GBackup.Restore(GUndoCompatibleBuffer.Instance.CurrentBackupName);
UpdatePreview();
}
public override void OnInspectorGUI()
{
EditorGUI.BeginChangeCheck();
instance.GroupId = GEditorCommon.ActiveTerrainGroupPopupWithAllOption("Group Id", instance.GroupId);
instance.EnableTerrainMask = EditorGUILayout.Toggle("Enable Terrain Mask", instance.EnableTerrainMask);
GEditorSettings.Instance.topographic.enable = EditorGUILayout.Toggle("Enable Topographic", GEditorSettings.Instance.topographic.enable);
DrawInstructionGUI();
DrawTransformGUI();
DrawStampGUI();
DrawStampLayersGUI();
DrawGizmosGUI();
DrawActionGUI();
GEditorCommon.DrawBackupHelpBox();
if (EditorGUI.EndChangeCheck())
{
UpdatePreview();
}
}
public override bool RequiresConstantRepaint()
{
return false;
}
private void DrawInstructionGUI()
{
string label = "Instruction";
string id = "instruction" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
EditorGUILayout.LabelField(INSTRUCTION, GEditorCommon.WordWrapItalicLabel);
});
}
private void DrawTransformGUI()
{
string label = "Transform";
string id = "transform" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
instance.Position = GEditorCommon.InlineVector3Field("Position", instance.Position);
instance.Rotation = Quaternion.Euler(GEditorCommon.InlineVector3Field("Rotation", instance.Rotation.eulerAngles));
instance.Scale = GEditorCommon.InlineVector3Field("Scale", instance.Scale);
Vector3 euler = instance.Rotation.eulerAngles;
euler = new Vector3(0, euler.y, 0);
instance.Rotation = Quaternion.Euler(euler);
});
}
private void DrawStampGUI()
{
string label = "Stamp";
string id = "stamp" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
instance.Mask = EditorGUILayout.ObjectField("Mask", instance.Mask, typeof(Texture2D), false) as Texture2D;
EditorGUI.BeginChangeCheck();
instance.Falloff = EditorGUILayout.CurveField("Falloff", instance.Falloff, Color.red, new Rect(0, 0, 1, 1));
if (EditorGUI.EndChangeCheck())
{
instance.Internal_UpdateFalloffTexture();
}
instance.Channel = (GTextureStampChannel)EditorGUILayout.EnumPopup("Channel", instance.Channel);
});
}
private void DrawStampLayersGUI()
{
string label = "Layers";
string id = "layers" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
List<GTextureStampLayer> layers = instance.Layers;
for (int i = 0; i < layers.Count; ++i)
{
DrawLayer(layers[i], i);
}
if (layers.Count > 0)
{
GEditorCommon.Separator();
}
if (GUILayout.Button("Add Layer"))
{
GTextureStampLayer layer = GTextureStampLayer.Create();
layers.Add(layer);
}
});
}
private void DrawLayer(GTextureStampLayer layer, int index)
{
string label = string.Format("Layer: {0} {1}",
!string.IsNullOrEmpty(layer.Name) ? layer.Name : index.ToString(),
layer.Ignore ? "[Ignored]" : string.Empty);
string id = "stamperlayer" + index + instance.GetInstanceID().ToString();
GenericMenu menu = new GenericMenu();
menu.AddItem(
new GUIContent("Remove"),
false,
() =>
{
ConfirmAndRemoveLayerAt(index);
});
GEditorCommon.Foldout(label, false, id, () =>
{
EditorGUI.indentLevel -= 1;
layer.Ignore = EditorGUILayout.Toggle("Ignore", layer.Ignore);
layer.Name = EditorGUILayout.TextField("Name", layer.Name);
if (instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness)
{
layer.Color = EditorGUILayout.ColorField("Color", layer.Color);
layer.Metallic = EditorGUILayout.Slider("Metallic", layer.Metallic, 0f, 1f);
layer.Smoothness = EditorGUILayout.Slider("Smoothness", layer.Smoothness, 0f, 1f);
}
else if (instance.Channel == GTextureStampChannel.Splat)
{
layer.SplatIndex = GEditorCommon.SplatSetSelectionGrid(instance.GroupId, layer.SplatIndex);
}
layer.BlendHeight = EditorGUILayout.Toggle("Blend Height", layer.BlendHeight);
if (layer.BlendHeight)
{
EditorGUI.indentLevel += 1;
layer.MinHeight = EditorGUILayout.FloatField("Min", layer.MinHeight);
layer.MaxHeight = EditorGUILayout.FloatField("Max", layer.MaxHeight);
EditorGUI.BeginChangeCheck();
layer.HeightTransition = EditorGUILayout.CurveField("Transition", layer.HeightTransition, Color.red, new Rect(0, 0, 1, 1));
if (EditorGUI.EndChangeCheck())
{
layer.UpdateCurveTextures();
}
EditorGUI.indentLevel -= 1;
}
layer.BlendSlope = EditorGUILayout.Toggle("Blend Slope", layer.BlendSlope);
if (layer.BlendSlope)
{
EditorGUI.indentLevel += 1;
layer.NormalMapMode = (GNormalMapMode)EditorGUILayout.EnumPopup("Mode", layer.NormalMapMode);
layer.MinSlope = EditorGUILayout.FloatField("Min", layer.MinSlope);
layer.MaxSlope = EditorGUILayout.FloatField("Max", layer.MaxSlope);
EditorGUI.BeginChangeCheck();
layer.SlopeTransition = EditorGUILayout.CurveField("Transition", layer.SlopeTransition, Color.red, new Rect(0, 0, 1, 1));
if (EditorGUI.EndChangeCheck())
{
layer.UpdateCurveTextures();
}
EditorGUI.indentLevel -= 1;
}
layer.BlendNoise = EditorGUILayout.Toggle("Blend Noise", layer.BlendNoise);
if (layer.BlendNoise)
{
EditorGUI.indentLevel += 1;
layer.NoiseOrigin = GEditorCommon.InlineVector2Field("Origin", layer.NoiseOrigin);
layer.NoiseFrequency = EditorGUILayout.FloatField("Frequency", layer.NoiseFrequency);
layer.NoiseLacunarity = EditorGUILayout.FloatField("Lacunarity", layer.NoiseLacunarity);
layer.NoisePersistence = EditorGUILayout.Slider("Persistence", layer.NoisePersistence, 0.01f, 1f);
layer.NoiseOctaves = EditorGUILayout.IntSlider("Octaves", layer.NoiseOctaves, 1, 4);
EditorGUI.BeginChangeCheck();
layer.NoiseRemap = EditorGUILayout.CurveField("Remap", layer.NoiseRemap, Color.red, new Rect(0, 0, 1, 1));
if (EditorGUI.EndChangeCheck())
{
layer.UpdateCurveTextures();
}
EditorGUI.indentLevel -= 1;
}
EditorGUI.indentLevel += 1;
},
menu);
}
private void ConfirmAndRemoveLayerAt(int index)
{
GTextureStampLayer layer = instance.Layers[index];
string layerName = string.IsNullOrEmpty(layer.Name) ? ("Layer " + index) : ("Layer " + layer.Name);
if (EditorUtility.DisplayDialog(
"Confirm",
"Remove " + layerName + "?",
"Yes", "Cancel"))
{
instance.Layers.RemoveAt(index);
}
}
private void DrawGizmosGUI()
{
string label = "Gizmos";
string id = "gizmos" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
instance.Editor_ShowLivePreview = EditorGUILayout.Toggle("Live Preview", instance.Editor_ShowLivePreview);
instance.Editor_ShowBounds = EditorGUILayout.Toggle("Bounds", instance.Editor_ShowBounds);
});
}
private void DrawActionGUI()
{
string label = "Action";
string id = "action" + instance.GetInstanceID().ToString();
GEditorCommon.Foldout(label, true, id, () =>
{
EditorGUILayout.BeginHorizontal();
if (GUILayout.Button("Snap To Terrain"))
{
IEnumerator<GStylizedTerrain> terrains = GStylizedTerrain.ActiveTerrains.GetEnumerator();
while (terrains.MoveNext())
{
GStylizedTerrain t = terrains.Current;
Bounds b = t.Bounds;
Rect r = new Rect(new Vector2(b.min.x, b.min.z), new Vector2(b.size.x, b.size.z));
Vector2 p = new Vector2(instance.Position.x, instance.Position.z);
if (r.Contains(p))
{
instance.Position = new Vector3(r.center.x, b.min.y, r.center.y);
instance.Rotation = Quaternion.identity;
instance.Scale = new Vector3(b.size.x, b.size.y, b.size.z);
break;
}
}
}
if (GUILayout.Button("Snap To Level Bounds"))
{
Bounds b = GCommon.GetLevelBounds();
instance.Position = new Vector3(b.center.x, b.min.y, b.center.z);
instance.Rotation = Quaternion.identity;
instance.Scale = new Vector3(b.size.x, b.size.y, b.size.z);
}
EditorGUILayout.EndHorizontal();
if (GUILayout.Button("Apply"))
{
GAnalytics.Record(GAnalytics.STAMPER_TEXTURE);
CreateInitialBackup();
ApplyStamp();
CreateAfterStampBackup();
}
});
}
private void CreateInitialBackup()
{
string historyPrefix =
instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness ? HISTORY_PREFIX_COLOR :
instance.Channel == GTextureStampChannel.Splat ? HISTORY_PREFIX_TEXTURE : "Unknown Action";
List<GTerrainResourceFlag> resourcesFlag = new List<GTerrainResourceFlag>();
if (instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness)
{
resourcesFlag.Add(GTerrainResourceFlag.AlbedoMap);
resourcesFlag.Add(GTerrainResourceFlag.MetallicMap);
}
else if (instance.Channel == GTextureStampChannel.Splat)
{
resourcesFlag.Add(GTerrainResourceFlag.SplatControlMaps);
}
//GBackup.TryCreateInitialBackup(historyPrefix, instance.GroupId, resourcesFlag);
List<GStylizedTerrain> terrains = GStampToolUtilities.CheckOverlap(instance.GroupId, instance.Rect);
GBackupInternal.TryCreateAndMergeInitialBackup(historyPrefix, terrains, resourcesFlag, true);
}
private void ApplyStamp()
{
EditorUtility.DisplayProgressBar("Applying", "Stamping terrain textures...", 1f);
try
{
instance.Apply();
}
catch (System.Exception e)
{
Debug.LogError(e.ToString());
}
EditorUtility.ClearProgressBar();
}
private void CreateAfterStampBackup()
{
string historyPrefix =
instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness ? HISTORY_PREFIX_COLOR :
instance.Channel == GTextureStampChannel.Splat ? HISTORY_PREFIX_TEXTURE : "Unknown Action";
List<GTerrainResourceFlag> resourcesFlag = new List<GTerrainResourceFlag>();
if (instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness)
{
resourcesFlag.Add(GTerrainResourceFlag.AlbedoMap);
resourcesFlag.Add(GTerrainResourceFlag.MetallicMap);
}
else if (instance.Channel == GTextureStampChannel.Splat)
{
resourcesFlag.Add(GTerrainResourceFlag.SplatControlMaps);
}
//GBackup.TryCreateBackup(historyPrefix, instance.GroupId, resourcesFlag);
List<GStylizedTerrain> terrains = GStampToolUtilities.CheckOverlap(instance.GroupId, instance.Rect);
GBackupInternal.TryCreateAndMergeBackup(historyPrefix, terrains, resourcesFlag, true);
}
private void DuringSceneGUI(SceneView sv)
{
EditorGUI.BeginChangeCheck();
HandleEditingTransform();
DrawStampBounds();
if (EditorGUI.EndChangeCheck())
{
UpdatePreview();
}
}
private void HandleEditingTransform()
{
if (Tools.current == Tool.Move)
{
instance.Position = Handles.PositionHandle(instance.Position, instance.Rotation);
}
else if (Tools.current == Tool.Rotate)
{
instance.Rotation = Handles.RotationHandle(instance.Rotation, instance.Position);
Vector3 euler = instance.Rotation.eulerAngles;
euler = new Vector3(0, euler.y, 0);
instance.Rotation = Quaternion.Euler(euler);
}
else if (Tools.current == Tool.Scale)
{
instance.Scale = Handles.ScaleHandle(instance.Scale, instance.Position, instance.Rotation, HandleUtility.GetHandleSize(instance.Position));
}
}
private void OnCameraRender(Camera cam)
{
if (instance.Editor_ShowLivePreview)
DrawLivePreview(cam);
if (instance.EnableTerrainMask)
DrawTerrainMask(cam);
}
private void DrawTerrainMask(Camera cam)
{
GCommon.ForEachTerrain(instance.GroupId, (t) =>
{
GLivePreviewDrawer.DrawTerrainMask(t, cam);
});
}
private void OnCameraRenderSRP(UnityEngine.Rendering.ScriptableRenderContext context, Camera cam)
{
OnCameraRender(cam);
}
private void DrawLivePreview(Camera cam)
{
IEnumerator<GStylizedTerrain> terrains = GStylizedTerrain.ActiveTerrains.GetEnumerator();
while (terrains.MoveNext())
{
GStylizedTerrain t = terrains.Current;
if (t.TerrainData == null)
continue;
if (instance.GroupId >= 0 &&
instance.GroupId != t.GroupId)
continue;
DrawLivePreview(t, cam);
}
}
private void DrawLivePreview(GStylizedTerrain t, Camera cam)
{
if (t.transform.rotation != Quaternion.identity ||
t.transform.lossyScale != Vector3.one)
return;
instance.GetQuad(worldPoints);
for (int i = 0; i < normalizedPoints.Length; ++i)
{
normalizedPoints[i] = t.WorldPointToUV(worldPoints[i]);
}
Rect dirtyRect = GUtilities.GetRectContainsPoints(normalizedPoints);
if (instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness)
{
int albedoResolution = t.TerrainData.Shading.AlbedoMapResolution;
FilterMode albedoFilter = t.TerrainData.Shading.AlbedoMapOrDefault.filterMode;
int metallicResolution = t.TerrainData.Shading.MetallicMapResolution;
FilterMode metallicFilter = t.TerrainData.Shading.MetallicMapOrDefault.filterMode;
Texture albedo = GetPreviewTexture(t, "albedo", albedoResolution, albedoFilter);
Texture metallic = GetPreviewTexture(t, "metallic", metallicResolution, metallicFilter);
GLivePreviewDrawer.DrawAMSLivePreview(t, cam, albedo, metallic, dirtyRect);
}
else if (instance.Channel == GTextureStampChannel.Splat)
{
int controlMapResolution = t.TerrainData.Shading.SplatControlResolution;
Texture[] controls = new Texture[t.TerrainData.Shading.SplatControlMapCount];
for (int i = 0; i < controls.Length; ++i)
{
Texture currentControl = t.TerrainData.Shading.GetSplatControlOrDefault(i);
controls[i] = GetPreviewTexture(t, "splatControl" + i, controlMapResolution, currentControl.filterMode);
}
GLivePreviewDrawer.DrawSplatLivePreview(t, cam, controls, dirtyRect);
}
}
private void UpdatePreview()
{
IEnumerator<GStylizedTerrain> terrains = GStylizedTerrain.ActiveTerrains.GetEnumerator();
while (terrains.MoveNext())
{
GStylizedTerrain t = terrains.Current;
if (t.TerrainData == null)
continue;
if (instance.GroupId >= 0 &&
instance.GroupId != t.GroupId)
continue;
if (instance.Channel == GTextureStampChannel.AlbedoMetallicSmoothness)
{
SetupAlbedoMetallicSmoothnessPreview(t);
}
else if (instance.Channel == GTextureStampChannel.Splat)
{
SetupSplatPreview(t);
}
}
}
private void SetupSplatPreview(GStylizedTerrain t)
{
Material mat = t.TerrainData.Shading.MaterialToRender;
if (mat == null)
return;
int controlMapResolution = t.TerrainData.Shading.SplatControlResolution;
int controlMapCount = t.TerrainData.Shading.SplatControlMapCount;
if (controlMapCount == 0)
return;
RenderTexture[] rtControls = new RenderTexture[controlMapCount];
for (int i = 0; i < controlMapCount; ++i)
{
Texture2D splatControl = t.TerrainData.Shading.GetSplatControlOrDefault(i);
rtControls[i] = GetPreviewTexture(t, "splatControl" + i, controlMapResolution, splatControl.filterMode);
GCommon.ClearRT(rtControls[i]);
}
instance.Internal_ApplySplat(t, rtControls);
previewPropertyBlock.Clear();
}
private void SetupAlbedoMetallicSmoothnessPreview(GStylizedTerrain t)
{
Material mat = t.TerrainData.Shading.MaterialToRender;
if (mat == null)
return;
int albedoResolution = t.TerrainData.Shading.AlbedoMapResolution;
RenderTexture previewAlbedo = GetPreviewTexture(t, "albedo", albedoResolution, t.TerrainData.Shading.AlbedoMapOrDefault.filterMode);
GCommon.ClearRT(previewAlbedo);
int metallicResolution = t.TerrainData.Shading.MetallicMapResolution;
RenderTexture previewMetallic = GetPreviewTexture(t, "metallic", metallicResolution, t.TerrainData.Shading.MetallicMapOrDefault.filterMode);
GCommon.ClearRT(previewMetallic);
instance.Internal_ApplyAlbedoMetallicSmoothness(t, previewAlbedo, previewMetallic);
}
private RenderTexture GetPreviewTexture(GStylizedTerrain t, string mapName, int resolution, FilterMode filter)
{
if (previewTextures == null)
{
previewTextures = new Dictionary<string, RenderTexture>();
}
string key = string.Format("{0}_{1}", t.GetInstanceID(), mapName);
if (!previewTextures.ContainsKey(key) ||
previewTextures[key] == null)
{
RenderTexture rt = new RenderTexture(resolution, resolution, 0, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Linear);
rt.wrapMode = TextureWrapMode.Clamp;
previewTextures[key] = rt;
}
else if (previewTextures[key].width != resolution || previewTextures[key].height != resolution)
{
previewTextures[key].Release();
Object.DestroyImmediate(previewTextures[key]);
RenderTexture rt = new RenderTexture(resolution, resolution, 0, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Linear);
rt.wrapMode = TextureWrapMode.Clamp;
previewTextures[key] = rt;
}
previewTextures[key].filterMode = filter;
return previewTextures[key];
}
private void DrawStampBounds()
{
if (!instance.Editor_ShowBounds)
return;
instance.GetBox(worldBox);
Vector3[] b = worldBox;
Handles.color = Color.yellow;
Handles.zTest = UnityEngine.Rendering.CompareFunction.LessEqual;
Handles.DrawLines(new Vector3[]
{
//bottom quad
b[0], b[1],
b[1], b[2],
b[2], b[3],
b[3], b[0],
//top quad
b[4], b[5],
b[5], b[6],
b[6], b[7],
b[7], b[4],
//vertical lines
b[0], b[4],
b[1], b[5],
b[2], b[6],
b[3], b[7]
});
Handles.zTest = UnityEngine.Rendering.CompareFunction.Always;
}
}
}