shader学习之路——折射

斯涅尔定律来计算反射角  介质1*sinA1 = 介质2*sinA2

Shader "Custom/RefractionShader" {
	Properties {
		_Color ("Color", Color) = (1,1,1,1)
		_RefractColor("Refract Color",Color) = (1,1,1,1)
		_RefractAmount("Refract Amount",Range(0.1,1)) = 0.5
		_RefractRatio("Refraction Ratio",Range(0.1,1)) = 0.5
		_Cubemap("Refraction Cubemap",cube) = "_Skybox"{}
	}

	SubShader {
		Pass{
			Tags { "RenderType"="Opaque" }
			CGPROGRAM
			#pragma vertex vert
			#pragma fragment frag
			#include "Lighting.cginc"
			#include "AutoLight.cginc"
			#include "UnityCG.cginc"

			fixed4 _Color;
			fixed4 _RefractColor;
			float _RefractAmount;
			float _RefractRatio;
			samplerCUBE _Cubemap;

			struct a2v{
				float4 vertex:POSITION;
				float3 normal:NORMAL;			
			};			

			struct v2f{
				float4 pos:SV_POSITION;
				float3 worldNormal:TEXCOORD0;
				float3 worldPos:TEXCOORD1;
				float3 worldViewDir:TEXCOORD2;
				float3 worldRefr:TEXCOORD3;
			};

			v2f vert(a2v v){
				v2f o;
				o.pos = UnityObjectToClipPos(v.vertex);
				o.worldNormal = UnityObjectToWorldNormal(v.normal);
				o.worldPos = mul(unity_ObjectToWorld,v.vertex).xyz;
				o.worldViewDir = UnityWorldSpaceViewDir(o.worldPos);
				//refract计算折射方向   第一个参数为入射光线方向  第二个参数是表面发现 
				//第三个参数是入射光线所在介质的折射率和折射光线所在介质的折射率之间的比值
				o.worldRefr = refract(-normalize(o.worldViewDir),normalize(o.worldNormal),_RefractRatio);
				TRANSFER_SHADOW(o);
				return o;
			}

			fixed4 frag(v2f i):SV_Target
			{
				fixed3 worldNormal = normalize(i.worldNormal);
				fixed3 worldLightDir = normalize(UnityWorldSpaceLightDir(i.worldPos));
				fixed3 worldViewDir = normalize(i.worldViewDir);
				fixed3 ambient = UNITY_LIGHTMODEL_AMBIENT.xyz;
				fixed3 diffuse = _LightColor0.rgb * _Color.rgb * max(0,dot(worldNormal,worldLightDir));
				fixed3 refraction = texCUBE(_Cubemap,i.worldRefr).rgb * _RefractColor.rgb;//texCUBE采样只需要提供
				UNITY_LIGHT_ATTENUATION(atten,i,i.worldPos);
				fixed3 color = ambient + lerp(diffuse , refraction ,_RefractAmount) * atten; //lerp插值函数
				return fixed4(color,1.0);
			}
			ENDCG
		}
	}
	FallBack "Diffuse"
}