Unity custom shader中调用内置Lightmap和Light Probes

刚开始接触unity shader的时候发现自己写的unlit shader不支持烘焙场景中的光,无论烘焙场景什么样,自己写的shader根本不接受烘焙的影响,而unity自带的standard shader及其他一系列都完美支持,那自己写的shader要如何像unity自带shader那样支持烘焙呢?
其实很简单,几行代码就能搞定。让我们先来看一个场景

场景

这是我在unity中搭建了一个临时场景并烘焙好(具体过程可看这个视频),然后这个猴脸用了standard shader,现在我把他换成我自己写的一个最简单的diffuse shader,只输出贴图颜色,然后就成了这样
失败的场景

贴图丑了点,大家无视好了。。。
现在发现,咦,为什么不像刚刚那样在猴脸的左侧有橙黄色的光叠加而右侧有蓝光的叠加呢?这是因为那几行神奇的代码没有加,我们现在来看要加什么。

Shader "Unlit/Monkey"
{
    Properties
    {
        _MainTex ("Texture", 2D) = "white" {}
        _Color ("Main Color",Color) = (1,1,1,1)
        
    }
    SubShader
    {
        Tags { "RenderType"="Opaque" }
        LOD 100

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

            struct appdata
            {
                float4 vertex : POSITION;
                float2 uv : TEXCOORD0;
                float2 uv1 : TEXCOORD1;
            };

            struct v2f
            {
                float2 uv : TEXCOORD0;
                float4 vertex : SV_POSITION;
                #ifdef LIGHTMAP_ON
                    float2 uvLM : TEXCOORD1;
                #endif
            };

            sampler2D _MainTex;
            float4 _MainTex_ST;
            fixed4 _Color;

            v2f vert (appdata v)
            {
                v2f o;
                o.vertex = UnityObjectToClipPos(v.vertex);
                o.uv = TRANSFORM_TEX(v.uv, _MainTex);
                #ifdef LIGHTMAP_ON
                    o.uvLM = v.uv1.xy * unity_LightmapST.xy + unity_LightmapST.zw;
                #endif
                return o;
            }

            fixed4 frag (v2f i) : SV_Target
            {
                // sample the texture
                fixed4 col = tex2D(_MainTex, i.uv);
                col *= _Color;
                #ifdef LIGHTMAP_ON
                    fixed3 lm = DecodeLightmap(UNITY_SAMPLE_TEX2D(unity_Lightmap,i.uvLM));
                    col.rgb *= lm;
                #endif
                return col;
            }
            ENDCG
        }
    }
}

首先第一条要在CGPROGRAM后添加一行
#pragma multi_compile LIGHTMAP_OFF LIGHTMAP_ON
这样shader就可以判断lightmap开没开了。

然后在vertex shader里添加

#ifdef LIGHTMAP_ON
    o.uvLM = v.uv1.xy * unity_LightmapST.xy + unity_LightmapST.zw;
#endif

把lightmap的uv准备好之后使用。
这里这样写是和unity的源码TRANSFORM_TEX一致

// Transforms 2D UV by scale/bias property
#define TRANSFORM_TEX(tex,name) (tex.xy * name##_ST.xy + name##_ST.zw)

不过不能直接调用TRANSFORM_TEX是因为unity_LightmapST这个东西在ST之前没有下划线_,所以没法用现成的方法了。
再来到fragment shader里就和获取主贴图一样操作了

#ifdef LIGHTMAP_ON
    fixed3 lm = DecodeLightmap(UNITY_SAMPLE_TEX2D(unity_Lightmap,i.uvLM));
    col.rgb *= lm;
#endif

这里为什么要用DecodeLightmap又要用UNITY_SAMPLE_TEX2D来解读ligtmap呢?实际上unity生成的lightmap的有两种编码格式:doubleLDR和RGBM,根据冯乐乐前辈的考证

Unity使用了两种编码方式来存储lightmap:
1、doubleLDR,需要一张rgb24贴图
2、RGBM,需要一张rgba32贴图
在移动设备上使用doubleLDR格式,可以获得更快的计算速度。它只用到了lightmap的RGB通道。
PC上则使用RGBM格式,可以获得更广的亮度范围,而牺牲一点速度。它使用了贴图的RGBA通道,而A通道是用来做乘法,所以称为RGBM格式。
这两种格式的差异就是导致不同平台下lightmap表现不同的原因,当然Unity会在切换平台时帮我们对贴图进行转换,而不需要太关心这个差异。

我们这样解读lightmap才能得到正确的颜色值而且不必担心因为平台不同造成的差异。

其实烘焙分为两种模式,有向的和无向的,如下图

上面的解析光照贴图方式是针对无向模式,有向模式应该这样解析

float4 lightmapDir = UNITY_SAMPLE_TEX2D_SAMPLER(unity_LightmapInd, unity_Lightmap, i.uvLM);
float3 lm = DecodeDirectionalLightmap(col,lightmapDir,normal);

unity_LightmapInd代表有向模式下烘焙出来的包含光照方向的光照贴图(是的!有向模式会烘焙出两张光照贴图,一张里储存颜色,一张里储存方向),然后有个方法叫DecodeDirectionalLightmap,需要传入颜色(比如说物体贴图的颜色,或者自定义的颜色,看最终想要的效果来决定),解析出来的光照贴图颜色以及世界坐标下的法线,这样出来的最终结果就是有向烘焙模式下光照贴图的信息了。

最终,我们自己写的shader也能支持烘焙了。


成功的场景

说完如何支持lightmap,我们再来看看如何支持light probes。我们在烘焙场景的时候,有时候场景中的某些东西不是static,也就是不是固定不动的,这样烘焙好的结果无法应用到这种在场景中动来动去的物件,而我们希望这样的物件依然受到烘焙的影响,那么这个时候就需要light probes了。
由于弄light probes挺麻烦的,所以我直接使用了这篇博客中的场景,然后加上了几行代码来支持light probes。我们先来看结果(这次我会做动图了!)


这个方块在动到黄色区域的时候呈现偏黄的颜色,而到了蓝色区域就会偏蓝。这其实就是在利用unity shader中自带的一个方法ShadeSH9来实现的,具体代码如下

Shader "Unlit/LightProbes"
{
    Properties
    {
        _MainTex ("Texture", 2D) = "white" {}
        _Color ("Main Color",Color) = (1,1,1,1)
    }
    SubShader
    {
        Tags { "RenderType"="Opaque" "LightMode" = "ForwardBase" }
        LOD 100

        Pass
        {
            CGPROGRAM
            #pragma vertex vert
            #pragma fragment frag

            #include "UnityCG.cginc"

            struct appdata
            {
                float4 vertex : POSITION;
                float2 uv : TEXCOORD0;
                float3 normal : NORMAL;
            };

            struct v2f
            {
                float2 uv : TEXCOORD0;
                float4 vertex : SV_POSITION;
                fixed3 SHLighting : COLOR;
            };

            sampler2D _MainTex;
            float4 _MainTex_ST;
            fixed4 _Color;

            v2f vert (appdata v)
            {
                v2f o;
                o.vertex = UnityObjectToClipPos(v.vertex);
                o.uv = TRANSFORM_TEX(v.uv, _MainTex);
                float3 worldNormal = UnityObjectToWorldNormal(v.normal);
                o.SHLighting = ShadeSH9(float4(worldNormal,1));
                return o;
            }

            fixed4 frag (v2f i) : SV_Target
            {
                // sample the texture
                fixed4 col = tex2D(_MainTex, i.uv);
                col *= _Color;
                col.rgb *= i.SHLighting;
                return col;
            }
            ENDCG
        }
    }
}

这里有一点要记住,自定义的shader一定要写上Tags{"LightMode" = "ForwardBase"},即是说要指定好LightMode,不指定的话ShadeSH9会返回黑色(全黑的那种),你完全不会想看到的颜色。

项目地址

参考
UnityShader实例07:在shader调用unity内置lightmap和light Probes

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 219,188评论 6 508
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,464评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 165,562评论 0 356
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,893评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,917评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,708评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,430评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,342评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,801评论 1 317
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,976评论 3 337
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,115评论 1 351
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,804评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,458评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,008评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,135评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,365评论 3 373
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,055评论 2 355

推荐阅读更多精彩内容