WaterFlush/Assets/Art/ThirdPart/Calm Water/Scripts/MirrorReflection.cs

using System;
using System.Collections.Generic;
using UnityEngine;

namespace CalmWater
{
[ExecuteInEditMode]
[RequireComponent(typeof(MeshRenderer))]
public class MirrorReflection : MonoBehaviour
{

	//private RenderTexture rt;
	public LayerMask reflectionMask = -1;

	private enum QualityLevels
	{
		High = 1,
		Medium = 2,
		Low = 4,
		VeryLow = 8
	};

	[SerializeField]
	private QualityLevels Quality = QualityLevels.Medium;

	[Tooltip("Color used instead of skybox if you choose to not render it.")]
	public Color clearColor = Color.grey;
	public bool reflectSkybox = true;
	public bool m_DisablePixelLights = false;
	[Tooltip("You won't be able to select objects in the scene when thi is active.")]
	public bool UpdateSceneView = true;
	public float clipPlaneOffset = 0.07F;

	private String reflectionSampler = "_ReflectionTex";


	Vector3 m_Oldpos;
	Camera m_ReflectionCamera;
	Material m_SharedMaterial;
	Dictionary<Camera, bool> m_HelperCameras;


	void OnEnable(){
		gameObject.layer = LayerMask.NameToLayer ("Water");
		setMaterial();
	}

	void OnDisable()
	{
		if(m_ReflectionCamera != null)
		{
			DestroyImmediate (m_ReflectionCamera);
		}
	}

	void Start()
	{
		gameObject.layer = LayerMask.NameToLayer ("Water");
		setMaterial ();
	}

	public void setMaterial()
	{
		m_SharedMaterial = GetComponent<Renderer>().sharedMaterial;
	}

	Camera CreateReflectionCameraFor(Camera cam)
	{
		String reflName = gameObject.name + "Reflection" + cam.name;
		GameObject go = GameObject.Find(reflName);

		if (!go)
		{
			go = new GameObject(reflName, typeof(Camera));
			go.hideFlags = HideFlags.HideAndDontSave;
		}
		if (!go.GetComponent(typeof(Camera)))
		{
			go.AddComponent(typeof(Camera));
		}

		Camera reflectCamera = go.GetComponent<Camera>();

		reflectCamera.backgroundColor = clearColor;
		reflectCamera.clearFlags 	= reflectSkybox ? CameraClearFlags.Skybox : CameraClearFlags.SolidColor;

		SetStandardCameraParameter(reflectCamera, reflectionMask);

		if (!reflectCamera.targetTexture)
		{
			reflectCamera.targetTexture = CreateTextureFor(cam);
		}

		return reflectCamera;
	}

	void SetStandardCameraParameter(Camera cam, LayerMask mask)
	{
		cam.cullingMask 		= mask & ~(1 << LayerMask.NameToLayer("Water"));
		cam.backgroundColor 	= Color.black;
		cam.enabled 			= false;
	}


	RenderTexture CreateTextureFor(Camera cam)
	{
		int rtW = Mathf.FloorToInt(cam.pixelWidth / (int)Quality);
		int rtH = Mathf.FloorToInt(cam.pixelHeight / (int)Quality);

		RenderTexture rt = new RenderTexture(rtW,rtH, 24);
		rt.hideFlags = HideFlags.DontSave;
		return rt;
	}


	public void RenderHelpCameras(Camera currentCam)
	{
		if (null == m_HelperCameras)
		{
			m_HelperCameras = new Dictionary<Camera, bool>();
		}

		if (!m_HelperCameras.ContainsKey(currentCam))
		{
			m_HelperCameras.Add(currentCam, false);
		}

		if (m_HelperCameras[currentCam] && !UpdateSceneView)
		{
			return;
		}

		if (!m_ReflectionCamera)
		{
			m_ReflectionCamera = CreateReflectionCameraFor(currentCam);
		}

		RenderReflectionFor(currentCam, m_ReflectionCamera);

		m_HelperCameras[currentCam] = true;
	}


	public void LateUpdate()
	{
		if (null != m_HelperCameras)
		{
			m_HelperCameras.Clear();
		}
	}


	public void WaterTileBeingRendered(Transform tr, Camera currentCam)
	{
		RenderHelpCameras(currentCam);

		if (m_ReflectionCamera && m_SharedMaterial)
		{
			m_SharedMaterial.SetTexture(reflectionSampler, m_ReflectionCamera.targetTexture);
		}
	}


	public void OnWillRenderObject()
	{
		WaterTileBeingRendered(transform, Camera.current);
	}


	void RenderReflectionFor(Camera cam, Camera reflectCamera)
	{
		if (!reflectCamera)
		{
			return;
		}

		if (m_SharedMaterial && !m_SharedMaterial.HasProperty(reflectionSampler))
		{
			return;
		}

		#if UNITY_EDITOR
		int rtW = Mathf.FloorToInt(cam.pixelWidth / (int)Quality);
		int rtH = Mathf.FloorToInt(cam.pixelHeight / (int)Quality);

		if (reflectCamera.targetTexture.width != rtW || reflectCamera.targetTexture.width != rtH) 
		{
			DestroyImmediate (reflectCamera.targetTexture);
			reflectCamera.targetTexture = CreateTextureFor (cam);
		}
		#endif

		// Optionally disable pixel lights for reflection
		int oldPixelLightCount = QualitySettings.pixelLightCount;
		if (m_DisablePixelLights) 
		{
			QualitySettings.pixelLightCount = 0;
		}

		reflectCamera.cullingMask = reflectionMask & ~(1 << LayerMask.NameToLayer("Water"));
		//reflectCamera.cullingMask = ~(1<<4) & reflectionMask.value; // never render water layer

		SaneCameraSettings(reflectCamera);

		reflectCamera.backgroundColor = clearColor;
		reflectCamera.clearFlags = reflectSkybox ? CameraClearFlags.Skybox : CameraClearFlags.SolidColor;
		if (reflectSkybox)
		{
			if (cam.gameObject.GetComponent(typeof(Skybox)))
			{
				Skybox sb = (Skybox)reflectCamera.gameObject.GetComponent(typeof(Skybox));
				if (!sb)
				{
					sb = (Skybox)reflectCamera.gameObject.AddComponent(typeof(Skybox));
				}
				sb.material = ((Skybox)cam.GetComponent(typeof(Skybox))).material;
			}
		}

		GL.invertCulling = true;

		Transform reflectiveSurface = transform; //waterHeight;

		Vector3 eulerA = cam.transform.eulerAngles;

		reflectCamera.transform.eulerAngles = new Vector3(-eulerA.x, eulerA.y, eulerA.z);
		reflectCamera.transform.position = cam.transform.position;

		Vector3 pos = reflectiveSurface.transform.position;
		pos.y = reflectiveSurface.position.y;
		Vector3 normal = reflectiveSurface.transform.up;
		float d = -Vector3.Dot(normal, pos) - clipPlaneOffset;
		Vector4 reflectionPlane = new Vector4(normal.x, normal.y, normal.z, d);

		Matrix4x4 reflection = Matrix4x4.zero;
		reflection = CalculateReflectionMatrix(reflection, reflectionPlane);
		m_Oldpos = cam.transform.position;
		Vector3 newpos = reflection.MultiplyPoint(m_Oldpos);

		reflectCamera.worldToCameraMatrix = cam.worldToCameraMatrix * reflection;

		Vector4 clipPlane = CameraSpacePlane(reflectCamera, pos, normal, 1.0f);

		Matrix4x4 projection = cam.projectionMatrix;
		projection = CalculateObliqueMatrix(projection, clipPlane);
		reflectCamera.projectionMatrix = projection;

		reflectCamera.transform.position = newpos;
		Vector3 euler = cam.transform.eulerAngles;
		reflectCamera.transform.eulerAngles = new Vector3(-euler.x, euler.y, euler.z);

		reflectCamera.Render();

		GL.invertCulling = false;

		// Restore pixel light count
		if (m_DisablePixelLights) 
		{
			QualitySettings.pixelLightCount = oldPixelLightCount;
		}
	}


	void SaneCameraSettings(Camera helperCam)
	{
		helperCam.depthTextureMode 	= DepthTextureMode.None;
		helperCam.backgroundColor 	= Color.black;
		helperCam.clearFlags 		= CameraClearFlags.SolidColor;
		helperCam.renderingPath 	= RenderingPath.Forward;
	}


	static Matrix4x4 CalculateObliqueMatrix(Matrix4x4 projection, Vector4 clipPlane)
	{
		Vector4 q = projection.inverse * new Vector4(
			Sgn(clipPlane.x),
			Sgn(clipPlane.y),
			1.0F,
			1.0F
		);
		Vector4 c = clipPlane * (2.0F / (Vector4.Dot(clipPlane, q)));
		// third row = clip plane - fourth row
		projection[2] = c.x - projection[3];
		projection[6] = c.y - projection[7];
		projection[10] = c.z - projection[11];
		projection[14] = c.w - projection[15];

		return projection;
	}


	static Matrix4x4 CalculateReflectionMatrix(Matrix4x4 reflectionMat, Vector4 plane)
	{
		reflectionMat.m00 = (1.0F - 2.0F * plane[0] * plane[0]);
		reflectionMat.m01 = (- 2.0F * plane[0] * plane[1]);
		reflectionMat.m02 = (- 2.0F * plane[0] * plane[2]);
		reflectionMat.m03 = (- 2.0F * plane[3] * plane[0]);

		reflectionMat.m10 = (- 2.0F * plane[1] * plane[0]);
		reflectionMat.m11 = (1.0F - 2.0F * plane[1] * plane[1]);
		reflectionMat.m12 = (- 2.0F * plane[1] * plane[2]);
		reflectionMat.m13 = (- 2.0F * plane[3] * plane[1]);

		reflectionMat.m20 = (- 2.0F * plane[2] * plane[0]);
		reflectionMat.m21 = (- 2.0F * plane[2] * plane[1]);
		reflectionMat.m22 = (1.0F - 2.0F * plane[2] * plane[2]);
		reflectionMat.m23 = (- 2.0F * plane[3] * plane[2]);

		reflectionMat.m30 = 0.0F;
		reflectionMat.m31 = 0.0F;
		reflectionMat.m32 = 0.0F;
		reflectionMat.m33 = 1.0F;

		return reflectionMat;
	}


	static float Sgn(float a)
	{
		if (a > 0.0F)
		{
			return 1.0F;
		}
		if (a < 0.0F)
		{
			return -1.0F;
		}
		return 0.0F;
	}


	Vector4 CameraSpacePlane(Camera cam, Vector3 pos, Vector3 normal, float sideSign)
	{
		Vector3 offsetPos = pos + normal * clipPlaneOffset;
		Matrix4x4 m = cam.worldToCameraMatrix;
		Vector3 cpos = m.MultiplyPoint(offsetPos);
		Vector3 cnormal = m.MultiplyVector(normal).normalized * sideSign;

		return new Vector4(cnormal.x, cnormal.y, cnormal.z, -Vector3.Dot(cpos, cnormal));
	}
}

}