367 lines
12 KiB
C#
367 lines
12 KiB
C#
// Crest Water System
|
|
// Copyright © 2024 Wave Harmonic. All rights reserved.
|
|
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using WaveHarmonic.Crest.Internal;
|
|
|
|
namespace WaveHarmonic.Crest
|
|
{
|
|
interface IReportsHeight
|
|
{
|
|
bool ReportHeight(WaterRenderer water, ref Rect bounds, ref float minimum, ref float maximum);
|
|
}
|
|
|
|
interface IReportsDisplacement
|
|
{
|
|
bool ReportDisplacement(WaterRenderer water, ref Rect bounds, ref float horizontal, ref float vertical);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Sets shader parameters for each geometry tile/chunk.
|
|
/// </summary>
|
|
#if !CREST_DEBUG
|
|
[AddComponentMenu("")]
|
|
#endif
|
|
[@ExecuteDuringEditMode]
|
|
sealed partial class WaterChunkRenderer : ManagedBehaviour<WaterRenderer>
|
|
{
|
|
[SerializeField]
|
|
internal bool _DrawRenderBounds = false;
|
|
|
|
internal const string k_UpdateMeshBoundsMarker = "Crest.WaterChunkRenderer.UpdateMeshBounds";
|
|
|
|
static readonly Unity.Profiling.ProfilerMarker s_UpdateMeshBoundsMarker = new(k_UpdateMeshBoundsMarker);
|
|
|
|
internal Transform _Transform;
|
|
internal Mesh _Mesh;
|
|
public Renderer Rend { get; private set; }
|
|
internal MaterialPropertyBlock _MaterialPropertyBlock;
|
|
Matrix4x4 _CurrentObjectToWorld;
|
|
Matrix4x4 _PreviousObjectToWorld;
|
|
internal Material _MotionVectorMaterial;
|
|
internal int _SortingOrder;
|
|
internal int _SiblingIndex;
|
|
|
|
internal Rect _UnexpandedBoundsXZ = new();
|
|
public Rect UnexpandedBoundsXZ => _UnexpandedBoundsXZ;
|
|
|
|
internal bool _Culled;
|
|
internal bool _Visible;
|
|
|
|
internal WaterRenderer _Water;
|
|
|
|
public bool MaterialOverridden { get; set; }
|
|
|
|
// We need to ensure that all water data has been bound for the mask to
|
|
// render properly - this is something that needs to happen irrespective
|
|
// of occlusion culling because we need the mask to render as a
|
|
// contiguous surface.
|
|
internal bool _WaterDataHasBeenBound = true;
|
|
|
|
int _LodIndex = -1;
|
|
|
|
|
|
// There is a 1-frame delay with Initialized in edit mode due to setting
|
|
// enableInEditMode in EditorApplication.update. This only really affect this
|
|
// component as it is instantiate via script, and is partial driven externally.
|
|
// So instead, call this after instantiation.
|
|
internal void Initialize(int index, Renderer renderer, Mesh mesh)
|
|
{
|
|
_LodIndex = index;
|
|
Rend = renderer;
|
|
_Mesh = mesh;
|
|
_PreviousObjectToWorld = _CurrentObjectToWorld = transform.localToWorldMatrix;
|
|
_Transform = transform;
|
|
WaterRenderer.s_OnLoadCameraData -= LoadCameraData;
|
|
WaterRenderer.s_OnLoadCameraData += LoadCameraData;
|
|
WaterRenderer.s_OnStoreCameraData -= StoreCameraData;
|
|
WaterRenderer.s_OnStoreCameraData += StoreCameraData;
|
|
WaterRenderer.s_OnRemoveCameraData -= RemoveCameraData;
|
|
WaterRenderer.s_OnRemoveCameraData += RemoveCameraData;
|
|
}
|
|
|
|
private protected override void OnStart()
|
|
{
|
|
base.OnStart();
|
|
|
|
UpdateMeshBounds();
|
|
}
|
|
|
|
internal void UpdateMeshBounds(WaterRenderer water, SurfaceRenderer surface)
|
|
{
|
|
_WaterDataHasBeenBound = false;
|
|
|
|
var count = surface.TimeSliceBoundsUpdateFrameCount;
|
|
|
|
// Time slice update to distribute the load.
|
|
if (count <= 1 || !(_SiblingIndex % count != Time.frameCount % surface.Chunks.Count % count))
|
|
{
|
|
// This needs to be called on Update because the bounds depend on transform scale which can change. Also OnWillRenderObject depends on
|
|
// the bounds being correct. This could however be called on scale change events, but would add slightly more complexity.
|
|
UpdateMeshBounds();
|
|
}
|
|
}
|
|
|
|
bool ShouldRender(bool culled)
|
|
{
|
|
// Is visible to camera.
|
|
if (!_Visible)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// If including culling, is it culled.
|
|
if (culled && _Culled)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
internal void OnLateUpdate()
|
|
{
|
|
_PreviousObjectToWorld = _CurrentObjectToWorld;
|
|
_CurrentObjectToWorld = _Transform.localToWorldMatrix;
|
|
}
|
|
|
|
internal void RenderMotionVectors(SurfaceRenderer surface, Camera camera)
|
|
{
|
|
if (!ShouldRender(culled: true))
|
|
{
|
|
return;
|
|
}
|
|
|
|
// RenderMesh will copy properties immediately, thus we need them bound.
|
|
if (!_WaterDataHasBeenBound)
|
|
{
|
|
Bind();
|
|
}
|
|
|
|
var material = MaterialOverridden ? _MotionVectorMaterial : surface._MotionVectorMaterial;
|
|
|
|
var parameters = new RenderParams(material)
|
|
{
|
|
motionVectorMode = MotionVectorGenerationMode.Object,
|
|
material = material,
|
|
matProps = _MaterialPropertyBlock,
|
|
worldBounds = Rend.bounds,
|
|
layer = surface.Layer,
|
|
renderingLayerMask = (uint)surface.Layer,
|
|
receiveShadows = false,
|
|
shadowCastingMode = UnityEngine.Rendering.ShadowCastingMode.Off,
|
|
lightProbeUsage = UnityEngine.Rendering.LightProbeUsage.Off,
|
|
reflectionProbeUsage = UnityEngine.Rendering.ReflectionProbeUsage.Off,
|
|
camera = camera,
|
|
};
|
|
|
|
Graphics.RenderMesh(parameters, _Mesh, 0, _CurrentObjectToWorld, _PreviousObjectToWorld);
|
|
}
|
|
|
|
void UpdateMeshBounds()
|
|
{
|
|
s_UpdateMeshBoundsMarker.Begin(this);
|
|
|
|
var bounds = _Mesh.bounds;
|
|
|
|
if (WaterBody.WaterBodies.Count > 0)
|
|
{
|
|
_UnexpandedBoundsXZ = ComputeBoundsXZ(_Transform, bounds);
|
|
}
|
|
|
|
bounds = ExpandBoundsForDisplacements(_Transform, bounds);
|
|
|
|
Rend.localBounds = bounds;
|
|
|
|
s_UpdateMeshBoundsMarker.End();
|
|
}
|
|
|
|
public static Rect ComputeBoundsXZ(Transform transform, Bounds bounds)
|
|
{
|
|
// Since chunks are axis-aligned it is safe to rotate the bounds.
|
|
var center = transform.rotation * bounds.center * transform.lossyScale.x + transform.position;
|
|
var size = transform.rotation * bounds.size * transform.lossyScale.x;
|
|
// Rotation can make size negative.
|
|
return new(0, 0, Mathf.Abs(size.x), Mathf.Abs(size.z))
|
|
{
|
|
center = center.XZ(),
|
|
};
|
|
}
|
|
|
|
internal void Bind()
|
|
{
|
|
_MaterialPropertyBlock = _Water.Surface.PerCascadeMPB[_LodIndex];
|
|
new PropertyWrapperMPB(_MaterialPropertyBlock).SetSHCoefficients(_Transform.position);
|
|
Rend.SetPropertyBlock(_MaterialPropertyBlock);
|
|
|
|
_WaterDataHasBeenBound = true;
|
|
}
|
|
|
|
void OnDestroy()
|
|
{
|
|
Helpers.Destroy(_Mesh);
|
|
}
|
|
|
|
// Called when visible to a camera
|
|
void OnWillRenderObject()
|
|
{
|
|
if (Rend == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (!MaterialOverridden && Rend.sharedMaterial != _Water.Surface.Material)
|
|
{
|
|
Rend.sharedMaterial = _Water.Surface.Material;
|
|
_MotionVectorMaterial = _Water.Surface._MotionVectorMaterial;
|
|
}
|
|
|
|
if (!_WaterDataHasBeenBound)
|
|
{
|
|
Bind();
|
|
}
|
|
|
|
if (_DrawRenderBounds)
|
|
{
|
|
Rend.bounds.DebugDraw();
|
|
}
|
|
}
|
|
|
|
// this is called every frame because the bounds are given in world space and depend on the transform scale, which
|
|
// can change depending on view altitude
|
|
public Bounds ExpandBoundsForDisplacements(Transform transform, Bounds bounds)
|
|
{
|
|
var extents = bounds.extents;
|
|
var center = bounds.center;
|
|
|
|
var scale = transform.lossyScale;
|
|
var rotation = transform.rotation;
|
|
|
|
// Extend the kinematic bounds slightly to give room for dynamic waves.
|
|
if (_Water._DynamicWavesLod.Enabled)
|
|
{
|
|
extents.y += 5f;
|
|
}
|
|
|
|
// Get XZ bounds. Doing this manually bypasses updating render bounds call.
|
|
Rect rect;
|
|
{
|
|
var p1 = transform.position;
|
|
var p2 = rotation * new Vector3(center.x, 0f, center.z);
|
|
var s1 = scale;
|
|
var s2 = rotation * (extents.XNZ(0f) * 2f);
|
|
|
|
rect = new(0, 0, Mathf.Abs(s1.x * s2.x), Mathf.Abs(s1.z * s2.z))
|
|
{
|
|
center = new(p1.x + p2.x, p1.z + p2.z)
|
|
};
|
|
}
|
|
|
|
// Extend bounds by local waves.
|
|
{
|
|
var totalHorizontal = 0f;
|
|
var totalVertical = 0f;
|
|
|
|
foreach (var (key, input) in AnimatedWavesLod.s_Inputs)
|
|
{
|
|
input.DisplacementReporter?.ReportDisplacement(_Water, ref rect, ref totalHorizontal, ref totalVertical);
|
|
}
|
|
|
|
var expandXZ = totalHorizontal / scale.x;
|
|
|
|
extents.x += expandXZ;
|
|
extents.y += totalVertical;
|
|
extents.z += expandXZ;
|
|
}
|
|
|
|
// Expand and offset bounds by height.
|
|
{
|
|
var minimumWaterLevelBounds = 0f;
|
|
var maximumWaterLevelBounds = 0f;
|
|
|
|
foreach (var (key, input) in LevelLod.s_Inputs)
|
|
{
|
|
input.HeightReporter?.ReportHeight(_Water, ref rect, ref minimumWaterLevelBounds, ref maximumWaterLevelBounds);
|
|
}
|
|
|
|
extents.y += Mathf.Abs((minimumWaterLevelBounds - maximumWaterLevelBounds) * 0.5f);
|
|
bounds.extents = extents;
|
|
|
|
var offset = Mathf.Lerp(minimumWaterLevelBounds, maximumWaterLevelBounds, 0.5f);
|
|
center.y += offset;
|
|
bounds.center = center;
|
|
}
|
|
|
|
return bounds;
|
|
}
|
|
}
|
|
|
|
partial class WaterChunkRenderer
|
|
{
|
|
class AdditionalCameraData
|
|
{
|
|
public Matrix4x4 _CurrentObjectToWorld;
|
|
public Matrix4x4 _PreviousObjectToWorld;
|
|
}
|
|
|
|
readonly Dictionary<Camera, AdditionalCameraData> _CameraData = new();
|
|
AdditionalCameraData _AdditionalCameraData;
|
|
|
|
void LoadCameraData(Camera camera)
|
|
{
|
|
if (!_CameraData.ContainsKey(camera))
|
|
{
|
|
_CameraData[camera] = new();
|
|
}
|
|
|
|
_AdditionalCameraData = _CameraData[camera];
|
|
_CurrentObjectToWorld = _AdditionalCameraData._CurrentObjectToWorld;
|
|
_PreviousObjectToWorld = _AdditionalCameraData._PreviousObjectToWorld;
|
|
}
|
|
|
|
void StoreCameraData(Camera camera)
|
|
{
|
|
if (_AdditionalCameraData == null) return;
|
|
_AdditionalCameraData._CurrentObjectToWorld = _CurrentObjectToWorld;
|
|
_AdditionalCameraData._PreviousObjectToWorld = _PreviousObjectToWorld;
|
|
}
|
|
|
|
void RemoveCameraData(Camera camera)
|
|
{
|
|
if (_CameraData.ContainsKey(camera))
|
|
{
|
|
_CameraData.Remove(camera);
|
|
}
|
|
}
|
|
}
|
|
|
|
static class BoundsHelper
|
|
{
|
|
internal static void DebugDraw(this Bounds b)
|
|
{
|
|
var xmin = b.min.x;
|
|
var ymin = b.min.y;
|
|
var zmin = b.min.z;
|
|
var xmax = b.max.x;
|
|
var ymax = b.max.y;
|
|
var zmax = b.max.z;
|
|
|
|
Debug.DrawLine(new(xmin, ymin, zmin), new(xmin, ymin, zmax));
|
|
Debug.DrawLine(new(xmin, ymin, zmin), new(xmax, ymin, zmin));
|
|
Debug.DrawLine(new(xmax, ymin, zmax), new(xmin, ymin, zmax));
|
|
Debug.DrawLine(new(xmax, ymin, zmax), new(xmax, ymin, zmin));
|
|
|
|
Debug.DrawLine(new(xmin, ymax, zmin), new(xmin, ymax, zmax));
|
|
Debug.DrawLine(new(xmin, ymax, zmin), new(xmax, ymax, zmin));
|
|
Debug.DrawLine(new(xmax, ymax, zmax), new(xmin, ymax, zmax));
|
|
Debug.DrawLine(new(xmax, ymax, zmax), new(xmax, ymax, zmin));
|
|
|
|
Debug.DrawLine(new(xmax, ymax, zmax), new(xmax, ymin, zmax));
|
|
Debug.DrawLine(new(xmin, ymin, zmin), new(xmin, ymax, zmin));
|
|
Debug.DrawLine(new(xmax, ymin, zmin), new(xmax, ymax, zmin));
|
|
Debug.DrawLine(new(xmin, ymax, zmax), new(xmin, ymin, zmax));
|
|
}
|
|
}
|
|
}
|