【Unity Shader】对反射探针的采样实现反射效果

最新推荐文章于 2023-01-14 21:50:43 发布

流年Bast

最新推荐文章于 2023-01-14 21:50:43 发布

阅读量1.3k

点赞数 2

本文链接：https://blog.csdn.net/ep661215/article/details/117778243

版权

Unity Shader 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

最近突然有小的展厅小程序需求，所以又拾起来Unity开发，发一个项目中用到的反射着色器吧，设置的时候勾选反射探针的“Box Projection”效果更好。直接上代码了：

Shader "SepShader/Reflection"
{
    Properties
    {
        _Color ("Color", Color) = (1,1,1,1)
        _CubeMap("CubeMap", Cube) = "white"{}
        _Smoothness("Smoothness", Range(0,1)) = 0.5
    }
        SubShader
        {
            Tags { "RenderType" = "Opaque" }
            LOD 200

            Pass
            {
                Tags { "LightMode" = "ForwardBase" }
                CGPROGRAM

                #pragma vertex vert
                #pragma fragment frag
                #include "UnityCG.cginc"
                #include "Lighting.cginc"

                sampler3D _CubeMap;

                half _Smoothness;
                fixed4 _Color;

                struct a2v
                {
                    float4 vertex : POSITION;
                    float3 normal : NORMAL;
                };
                struct v2f
                {
                    float4 pos : SV_POSITION;
                    float3 worldPos : TEXCOORD0;
                    float3 worldNormal : TEXCOORD1;
                    float3 worldViewDir : TEXCOORD2;
                    float3 worldRefDir : TEXCOORD3;
                };

                v2f vert(a2v v)
                {
                    v2f f;
                    f.pos = UnityObjectToClipPos(v.vertex);
                    f.worldPos = mul(unity_ObjectToWorld, v.vertex).xyz;
                    f.worldNormal = UnityObjectToWorldNormal(v.normal);
                    f.worldViewDir = UnityWorldSpaceViewDir(f.worldPos);
                    f.worldRefDir = reflect(-f.worldViewDir, f.worldNormal);

                    return f;
                }

                fixed4 frag(v2f f) : SV_Target
                {
                    fixed3 worldNormal = normalize(f.worldNormal);
                    fixed3 worldViewDir = normalize(f.worldViewDir);
                    fixed3 worldLightDir = normalize(UnityWorldSpaceLightDir(f.worldPos));

                    fixed3 ambient = UNITY_LIGHTMODEL_AMBIENT.xyz;
                    fixed3 diffuse = _Color.xyz * ambient * max(0, dot(worldNormal, worldLightDir));

                    //fixed3 reflection = texCUBE(_CubeMap, f.worldRefDir).rgb;
                    //通过BoxProjectedCubemapDirection函数修正reflectDir
                    f.worldRefDir = BoxProjectedCubemapDirection(f.worldRefDir, f.worldPos, unity_SpecCube0_ProbePosition, unity_SpecCube0_BoxMin, unity_SpecCube0_BoxMax);
                    half4 rgbm = UNITY_SAMPLE_TEXCUBE(unity_SpecCube0, f.worldRefDir);
                    half3 reflection = DecodeHDR(rgbm, unity_SpecCube0_HDR);

                    fixed3 outColor = ambient + lerp(diffuse, reflection, _Smoothness);
                    return fixed4(outColor, 1.0);
                }
                ENDCG
        }
    }
    FallBack "Diffuse"
}