220 lines
9.3 KiB
C#
220 lines
9.3 KiB
C#
|
using UnityEngine;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using Pinwheel.Griffin.TextureTool;
|
||
|
|
#if UNITY_EDITOR
|
||
|
|
using UnityEditor;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
namespace Pinwheel.Griffin.PaintTool
|
||
|
|
{
|
||
|
|
public class GNoisePainter : IGTexturePainter, IGTexturePainterWithCustomParams, IGTexturePainterWithLivePreview
|
||
|
|
{
|
||
|
|
public string HistoryPrefix
|
||
|
|
{
|
||
|
|
get
|
||
|
|
{
|
||
|
|
return "Noise Painting";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public string Instruction
|
||
|
|
{
|
||
|
|
get
|
||
|
|
{
|
||
|
|
string s = string.Format(
|
||
|
|
"Raise or lower terrain surface using noise.\n" +
|
||
|
|
" - Hold Left Mouse to raise.\n" +
|
||
|
|
" - Hold Ctrl & Left Mouse to lower.");
|
||
|
|
return s;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public List<GTerrainResourceFlag> GetResourceFlagForHistory(GTexturePainterArgs args)
|
||
|
|
{
|
||
|
|
return GCommon.HeightMapAndFoliageResourceFlags;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Paint(Pinwheel.Griffin.GStylizedTerrain terrain, GTexturePainterArgs args)
|
||
|
|
{
|
||
|
|
if (terrain.TerrainData == null)
|
||
|
|
return;
|
||
|
|
if (args.MouseEventType == GPainterMouseEventType.Down)
|
||
|
|
{
|
||
|
|
terrain.ForceLOD(0);
|
||
|
|
GRuntimeSettings.Instance.isEditingGeometry = true;
|
||
|
|
}
|
||
|
|
if (args.MouseEventType == GPainterMouseEventType.Up)
|
||
|
|
{
|
||
|
|
terrain.ForceLOD(-1);
|
||
|
|
GRuntimeSettings.Instance.isEditingGeometry = false;
|
||
|
|
terrain.UpdateTreesPosition();
|
||
|
|
terrain.UpdateGrassPatches();
|
||
|
|
terrain.TerrainData.Foliage.ClearTreeDirtyRegions();
|
||
|
|
terrain.TerrainData.Foliage.ClearGrassDirtyRegions();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
Vector2[] uvCorners = new Vector2[args.WorldPointCorners.Length];
|
||
|
|
for (int i = 0; i < uvCorners.Length; ++i)
|
||
|
|
{
|
||
|
|
uvCorners[i] = terrain.WorldPointToUV(args.WorldPointCorners[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
Rect dirtyRect = GUtilities.GetRectContainsPoints(uvCorners);
|
||
|
|
if (!dirtyRect.Overlaps(new Rect(0, 0, 1, 1)))
|
||
|
|
return;
|
||
|
|
|
||
|
|
Texture2D bg = terrain.TerrainData.Geometry.HeightMap;
|
||
|
|
int heightMapResolution = terrain.TerrainData.Geometry.HeightMapResolution;
|
||
|
|
RenderTexture rt = GTerrainTexturePainter.Internal_GetRenderTexture(heightMapResolution);
|
||
|
|
GCommon.CopyToRT(bg, rt);
|
||
|
|
|
||
|
|
RenderTexture noiseMap = GTerrainTexturePainter.Internal_GetRenderTexture(terrain, heightMapResolution, 1);
|
||
|
|
RenderNoiseTexture(noiseMap, terrain);
|
||
|
|
|
||
|
|
Material mat = GInternalMaterials.NoisePainterMaterial;
|
||
|
|
mat.SetTexture("_MainTex", bg);
|
||
|
|
mat.SetTexture("_Mask", args.BrushMask);
|
||
|
|
mat.SetFloat("_Opacity", Mathf.Pow(args.Opacity, GTerrainTexturePainter.GEOMETRY_OPACITY_EXPONENT));
|
||
|
|
mat.SetTexture("_NoiseMap", noiseMap);
|
||
|
|
if (args.EnableTerrainMask)
|
||
|
|
{
|
||
|
|
mat.SetTexture("_TerrainMask", terrain.TerrainData.Mask.MaskMapOrDefault);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
mat.SetTexture("_TerrainMask", Texture2D.blackTexture);
|
||
|
|
}
|
||
|
|
GCommon.SetMaterialKeywordActive(mat, "USE_WORLD_SPACE", GTexturePainterCustomParams.Instance.Noise.UseWorldSpace);
|
||
|
|
int pass =
|
||
|
|
args.ActionType == GPainterActionType.Normal ? 0 :
|
||
|
|
args.ActionType == GPainterActionType.Negative ? 1 : 0;
|
||
|
|
GCommon.DrawQuad(rt, uvCorners, mat, pass);
|
||
|
|
|
||
|
|
RenderTexture.active = rt;
|
||
|
|
terrain.TerrainData.Geometry.HeightMap.ReadPixels(
|
||
|
|
new Rect(0, 0, heightMapResolution, heightMapResolution), 0, 0);
|
||
|
|
terrain.TerrainData.Geometry.HeightMap.Apply();
|
||
|
|
RenderTexture.active = null;
|
||
|
|
|
||
|
|
terrain.TerrainData.Geometry.SetRegionDirty(dirtyRect);
|
||
|
|
terrain.TerrainData.Foliage.SetTreeRegionDirty(dirtyRect);
|
||
|
|
terrain.TerrainData.Foliage.SetGrassRegionDirty(dirtyRect);
|
||
|
|
terrain.TerrainData.SetDirty(GTerrainData.DirtyFlags.Geometry);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Editor_DrawCustomParamsGUI()
|
||
|
|
{
|
||
|
|
#if UNITY_EDITOR
|
||
|
|
string label = "Noise Painting";
|
||
|
|
string id = "noise-painter";
|
||
|
|
|
||
|
|
GCommonGUI.Foldout(label, true, id, () =>
|
||
|
|
{
|
||
|
|
GNoisePainterParams param = GTexturePainterCustomParams.Instance.Noise;
|
||
|
|
|
||
|
|
EditorGUILayout.GetControlRect(GUILayout.Height(1));
|
||
|
|
Rect r = EditorGUILayout.GetControlRect(GUILayout.Height(200));
|
||
|
|
RenderTexture preview = GTerrainTexturePainter.Internal_GetRenderTexture(null, 200, 0);
|
||
|
|
RenderNoiseTexture(preview, null);
|
||
|
|
EditorGUI.DrawPreviewTexture(r, preview, null, ScaleMode.ScaleToFit);
|
||
|
|
EditorGUILayout.GetControlRect(GUILayout.Height(1));
|
||
|
|
|
||
|
|
param.Type = (GNoiseType)EditorGUILayout.EnumPopup("Type", param.Type);
|
||
|
|
param.Origin = GCommonGUI.InlineVector2Field("Origin", param.Origin);
|
||
|
|
param.Frequency = EditorGUILayout.FloatField("Frequency", param.Frequency);
|
||
|
|
param.Lacunarity = EditorGUILayout.FloatField("Lacunarity", param.Lacunarity);
|
||
|
|
param.Persistence = EditorGUILayout.FloatField("Persistence", param.Persistence);
|
||
|
|
param.Octaves = EditorGUILayout.IntField("Octaves", param.Octaves);
|
||
|
|
param.Seed = EditorGUILayout.FloatField("Seed", param.Seed);
|
||
|
|
param.UseWorldSpace = EditorGUILayout.Toggle("World Space", param.UseWorldSpace);
|
||
|
|
|
||
|
|
GTexturePainterCustomParams.Instance.Noise = param;
|
||
|
|
|
||
|
|
EditorUtility.SetDirty(GTexturePainterCustomParams.Instance);
|
||
|
|
});
|
||
|
|
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Editor_DrawLivePreview(GStylizedTerrain terrain, GTexturePainterArgs args, Camera cam)
|
||
|
|
{
|
||
|
|
#if UNITY_EDITOR
|
||
|
|
Vector2[] uvCorners = new Vector2[args.WorldPointCorners.Length];
|
||
|
|
for (int i = 0; i < uvCorners.Length; ++i)
|
||
|
|
{
|
||
|
|
uvCorners[i] = terrain.WorldPointToUV(args.WorldPointCorners[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
Rect dirtyRect = GUtilities.GetRectContainsPoints(uvCorners);
|
||
|
|
if (!dirtyRect.Overlaps(new Rect(0, 0, 1, 1)))
|
||
|
|
return;
|
||
|
|
|
||
|
|
Texture2D bg = terrain.TerrainData.Geometry.HeightMap;
|
||
|
|
int heightMapResolution = terrain.TerrainData.Geometry.HeightMapResolution;
|
||
|
|
RenderTexture rt = GTerrainTexturePainter.Internal_GetRenderTexture(terrain, heightMapResolution);
|
||
|
|
GCommon.CopyToRT(bg, rt);
|
||
|
|
|
||
|
|
RenderTexture noiseMap = GTerrainTexturePainter.Internal_GetRenderTexture(terrain, heightMapResolution, 1);
|
||
|
|
RenderNoiseTexture(noiseMap, terrain);
|
||
|
|
|
||
|
|
Material mat = GInternalMaterials.NoisePainterMaterial;
|
||
|
|
mat.SetTexture("_MainTex", bg);
|
||
|
|
mat.SetTexture("_Mask", args.BrushMask);
|
||
|
|
mat.SetFloat("_Opacity", Mathf.Pow(args.Opacity, GTerrainTexturePainter.GEOMETRY_OPACITY_EXPONENT));
|
||
|
|
mat.SetTexture("_NoiseMap", noiseMap);
|
||
|
|
if (args.EnableTerrainMask)
|
||
|
|
{
|
||
|
|
mat.SetTexture("_TerrainMask", terrain.TerrainData.Mask.MaskMapOrDefault);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
mat.SetTexture("_TerrainMask", Texture2D.blackTexture);
|
||
|
|
}
|
||
|
|
GCommon.SetMaterialKeywordActive(mat, "USE_WORLD_SPACE", GTexturePainterCustomParams.Instance.Noise.UseWorldSpace);
|
||
|
|
int pass =
|
||
|
|
args.ActionType == GPainterActionType.Normal ? 0 :
|
||
|
|
args.ActionType == GPainterActionType.Negative ? 1 : 0;
|
||
|
|
GCommon.DrawQuad(rt, uvCorners, mat, pass);
|
||
|
|
|
||
|
|
GLivePreviewDrawer.DrawGeometryLivePreview(terrain, cam, rt, dirtyRect);
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
private void RenderNoiseTexture(RenderTexture noiseMap, GStylizedTerrain terrain)
|
||
|
|
{
|
||
|
|
GNoisePainterParams param = GTexturePainterCustomParams.Instance.Noise;
|
||
|
|
GNoiseMapGenerator gen = new GNoiseMapGenerator();
|
||
|
|
GNoiseMapGeneratorParams genParam = new GNoiseMapGeneratorParams();
|
||
|
|
genParam.Type = param.Type;
|
||
|
|
if (param.UseWorldSpace && terrain != null)
|
||
|
|
{
|
||
|
|
Vector3 terrainSize = new Vector3(
|
||
|
|
terrain.TerrainData.Geometry.Width,
|
||
|
|
terrain.TerrainData.Geometry.Height,
|
||
|
|
terrain.TerrainData.Geometry.Length);
|
||
|
|
Vector3 terrainPos = terrain.transform.position;
|
||
|
|
|
||
|
|
if (genParam.Type == GNoiseType.Voronoi)
|
||
|
|
{
|
||
|
|
genParam.Origin = param.Origin + new Vector2(terrainPos.x / terrainSize.x, terrainPos.z / terrainSize.z);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
genParam.Origin = param.Origin + new Vector2(terrainPos.x * param.Frequency / terrainSize.x, terrainPos.z * param.Frequency / terrainSize.z);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
genParam.Origin = param.Origin;
|
||
|
|
}
|
||
|
|
genParam.Frequency = param.Frequency;
|
||
|
|
genParam.Lacunarity = param.Lacunarity;
|
||
|
|
genParam.Persistence = param.Persistence;
|
||
|
|
genParam.Octaves = param.Octaves;
|
||
|
|
genParam.Seed = param.Seed;
|
||
|
|
gen.Generate(noiseMap, genParam);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|