在shader调用unity内置lightmap和Light Probes

在编写unity shader的时候，有时候会有需求，希望写的shader能支持unity内置lightmap或者light probe （灯光探测器）。如果是用surface 编写的话，这些自然不用考虑，unity会自动编译支持，但如果是用vert&frag编写shader，这些需要自己添加相关代码调用了 。

unity内置lightmap的调用

为了使unity内置数据和各种宏定义（如本文的LIGHTMAP_OFF）起作用，需要添加#pragma指令：

pragma multi_compile LIGHTMAP_OFF LIGHTMAP_ON

在unity5.0之前的版本需要声明一下两个内置参数，unity5.0的话就不需要这两句了：

half4 unity_LightmapST;
sampler2D unity_Lightmap;

Lightmap是使用模型的uv2 ，因此接下来在顶点输入结构体里声明uv2：

float2 texcoord1 : TEXCOORD1;

在另一个顶点结构里面定义用来接收uv2的uv值：

ifndef LIGHTMAP_OFF

half2 uvLM : TEXCOORD4;

endif

在vert函数里面给uvLM赋值：

ifndef LIGHTMAP_OFF

o.uvLM = v.texcoord1.xy * unity_LightmapST.xy + unity_LightmapST.zw;

endif

然后frag函数里对lightmap贴图采样并叠加给主颜色：

ifndef LIGHTMAP_OFF

fixed3 lm = DecodeLightmap (UNITY_SAMPLE_TEX2D(unity_Lightmap, i.uvLM.xy));
col.rgb*=lm;

endif

在上面的代码中使用了DecodeLightmap解码unity的内置光照贴图，这是因为 Unity烘焙的LightMap是32bit的HDR图，在桌面端，光照贴图的编码为RGBM，而在移动端，大部分情况下，光照贴图的编码为double-LDR，因此需要针对不同平台提供不同的编码方式。DecodeLightmap作用就在这里，它可以针对不同的平台对光照贴图进行解码

VF版本代码01:

Shader “PengLu/Unlit/TextureLM” {
Properties {
_MainTex (“Base (RGB)”, 2D) = “white” {}
}

SubShader {
Tags { “RenderType”=”Opaque” }
LOD 100

Pass {
CGPROGRAM
#pragma vertex vert
#pragma fragment frag
#pragma multi_compile_fog
#pragma multi_compile LIGHTMAP_OFF LIGHTMAP_ON
#include “UnityCG.cginc”

    struct appdata_t {
        float4 vertex : POSITION;
        float2 texcoord : TEXCOORD0;
        float2 texcoord1 : TEXCOORD1;
    };

    struct v2f {
        float4 vertex : SV_POSITION;
        half2 texcoord : TEXCOORD0;
        #ifndef LIGHTMAP_OFF
        half2 uvLM : TEXCOORD1;
        #endif 
        UNITY_FOG_COORDS(1)
    };

    sampler2D _MainTex;
    float4 _MainTex_ST;

    v2f vert (appdata_t v)
    {
        v2f o;
        o.vertex = mul(UNITY_MATRIX_MVP, v.vertex);
        o.texcoord = TRANSFORM_TEX(v.texcoord, _MainTex);
        #ifndef LIGHTMAP_OFF
        o.uvLM = v.texcoord1.xy * unity_LightmapST.xy + unity_LightmapST.zw;
        #endif
        UNITY_TRANSFER_FOG(o,o.vertex);
        return o;
    }

    fixed4 frag (v2f i) : SV_Target
    {
        fixed4 col = tex2D(_MainTex, i.texcoord);
        UNITY_APPLY_FOG(i.fogCoord, col);
        UNITY_OPAQUE_ALPHA(col.a);
        #ifndef LIGHTMAP_OFF
        fixed3 lm = DecodeLightmap (UNITY_SAMPLE_TEX2D(unity_Lightmap, i.uvLM.xy));
        col.rgb*=lm;
        #endif
        return col;
    }
ENDCG

}

unity内置Light Probes的调用

在shader中我们是通过unity定义的half3 ShadeSH9(half4 normal)来调用Light Probes的，Light Probes照明使用的是一种叫球谐光照（Sphere Harmonic）的模拟，简称SH,因此在ShadeSH9函数需要一个世界坐标中的Normal来决定物体表面的光照。

首先我们在顶点输出结构定义一个参数SHLighting：

fixed3 SHLighting : COLOR;

然后在顶点函数里为它赋值：

float3 worldNormal = mul((float3x3)_Object2World, v.normal);//获得世界坐标中的normal

o.SHLighting= ShadeSH9(float4(worldNormal,1)) ;

VF版本代码02:

Shader “PengLu/Unlit/TextureLM” {
Properties {
_MainTex (“Base (RGB)”, 2D) = “white” {}
_SHLightingScale(“LightProbe influence scale”,float) = 1
}

SubShader {
Tags { “Queue”=”Geometry”“LightMode”=”ForwardBase”“RenderType”=”Opaque” }
LOD 100

Pass {
CGPROGRAM
#pragma vertex vert
#pragma fragment frag
#pragma multi_compile_fog

    #include "UnityCG.cginc"



    struct v2f {
        float4 vertex : SV_POSITION;
        half2 texcoord : TEXCOORD0;
        fixed3  SHLighting : COLOR;
        UNITY_FOG_COORDS(1)
    };

    sampler2D _MainTex;
    float4 _MainTex_ST;
    float _SHLightingScale;

    v2f vert (appdata_base v)
    {
        v2f o;
        o.vertex = mul(UNITY_MATRIX_MVP, v.vertex);
        o.texcoord = TRANSFORM_TEX(v.texcoord, _MainTex);
        float3 worldNormal = mul((float3x3)_Object2World, v.normal);
        o.SHLighting= ShadeSH9(float4(worldNormal,1)) ;
                        UNITY_TRANSFER_FOG(o,o.vertex);
        return o;
    }

    fixed4 frag (v2f i) : SV_Target
    {
        fixed4 col = tex2D(_MainTex, i.texcoord);
        col.rgb*=i.SHLighting;
        UNITY_APPLY_FOG(i.fogCoord, col);
        UNITY_OPAQUE_ALPHA(col.a);

        return col*_SHLightingScale;
    }
ENDCG

}
}

}