288 lines
9.7 KiB
C#
288 lines
9.7 KiB
C#
//Source: https://github.com/Ayfel/PrefabLightmapping
|
||
using UnityEngine;
|
||
using UnityEditor;
|
||
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using UnityEngine.SceneManagement;
|
||
|
||
[ExecuteAlways]
|
||
public class PrefabLightmapData : MonoBehaviour
|
||
{
|
||
[System.Serializable]
|
||
struct RendererInfo
|
||
{
|
||
public Renderer renderer;
|
||
public int lightmapIndex;
|
||
public Vector4 lightmapOffsetScale;
|
||
}
|
||
[System.Serializable]
|
||
struct LightInfo
|
||
{
|
||
public Light light;
|
||
public int lightmapBaketype;
|
||
public int mixedLightingMode;
|
||
}
|
||
|
||
[SerializeField]
|
||
RendererInfo[] m_RendererInfo;
|
||
[SerializeField]
|
||
Texture2D[] m_Lightmaps;
|
||
[SerializeField]
|
||
Texture2D[] m_LightmapsDir;
|
||
[SerializeField]
|
||
Texture2D[] m_ShadowMasks;
|
||
[SerializeField]
|
||
LightInfo[] m_LightInfo;
|
||
|
||
|
||
void Awake()
|
||
{
|
||
Init();
|
||
}
|
||
|
||
public void Init()
|
||
{
|
||
if (m_RendererInfo == null || m_RendererInfo.Length == 0)
|
||
return;
|
||
|
||
var lightmaps = LightmapSettings.lightmaps;
|
||
int[] offsetsindexes = new int[m_Lightmaps.Length];
|
||
int counttotal = lightmaps.Length;
|
||
List<LightmapData> combinedLightmaps = new List<LightmapData>();
|
||
|
||
for (int i = 0; i < m_Lightmaps.Length; i++)
|
||
{
|
||
bool exists = false;
|
||
for (int j = 0; j < lightmaps.Length; j++)
|
||
{
|
||
|
||
if (m_Lightmaps[i] == lightmaps[j].lightmapColor)
|
||
{
|
||
exists = true;
|
||
offsetsindexes[i] = j;
|
||
|
||
}
|
||
|
||
}
|
||
if (!exists)
|
||
{
|
||
offsetsindexes[i] = counttotal;
|
||
var newlightmapdata = new LightmapData
|
||
{
|
||
lightmapColor = m_Lightmaps[i],
|
||
lightmapDir = m_LightmapsDir.Length == m_Lightmaps.Length ? m_LightmapsDir[i] : default(Texture2D),
|
||
shadowMask = m_ShadowMasks.Length == m_Lightmaps.Length ? m_ShadowMasks[i] : default(Texture2D),
|
||
};
|
||
|
||
combinedLightmaps.Add(newlightmapdata);
|
||
|
||
counttotal += 1;
|
||
|
||
|
||
}
|
||
|
||
}
|
||
|
||
var combinedLightmaps2 = new LightmapData[counttotal];
|
||
|
||
lightmaps.CopyTo(combinedLightmaps2, 0);
|
||
combinedLightmaps.ToArray().CopyTo(combinedLightmaps2, lightmaps.Length);
|
||
|
||
bool directional=true;
|
||
|
||
foreach(Texture2D t in m_LightmapsDir)
|
||
{
|
||
if (t == null)
|
||
{
|
||
directional = false;
|
||
break;
|
||
}
|
||
}
|
||
|
||
LightmapSettings.lightmapsMode = (m_LightmapsDir.Length == m_Lightmaps.Length && directional) ? LightmapsMode.CombinedDirectional : LightmapsMode.NonDirectional;
|
||
ApplyRendererInfo(m_RendererInfo, offsetsindexes, m_LightInfo);
|
||
LightmapSettings.lightmaps = combinedLightmaps2;
|
||
}
|
||
|
||
void OnEnable()
|
||
{
|
||
|
||
SceneManager.sceneLoaded += OnSceneLoaded;
|
||
|
||
}
|
||
|
||
// called second
|
||
void OnSceneLoaded(Scene scene, LoadSceneMode mode)
|
||
{
|
||
Init();
|
||
}
|
||
|
||
// called when the game is terminated
|
||
void OnDisable()
|
||
{
|
||
SceneManager.sceneLoaded -= OnSceneLoaded;
|
||
}
|
||
|
||
|
||
|
||
static void ApplyRendererInfo(RendererInfo[] infos, int[] lightmapOffsetIndex, LightInfo[] lightsInfo)
|
||
{
|
||
for (int i = 0; i < infos.Length; i++)
|
||
{
|
||
var info = infos[i];
|
||
|
||
info.renderer.lightmapIndex = lightmapOffsetIndex[info.lightmapIndex];
|
||
info.renderer.lightmapScaleOffset = info.lightmapOffsetScale;
|
||
|
||
// You have to release shaders.
|
||
Material[] mat = info.renderer.sharedMaterials;
|
||
for (int j = 0; j < mat.Length; j++)
|
||
{
|
||
if (mat[j] != null && Shader.Find(mat[j].shader.name) != null)
|
||
mat[j].shader = Shader.Find(mat[j].shader.name);
|
||
}
|
||
|
||
}
|
||
|
||
for (int i = 0; i < lightsInfo.Length; i++)
|
||
{
|
||
LightBakingOutput bakingOutput = new LightBakingOutput();
|
||
bakingOutput.isBaked = true;
|
||
bakingOutput.lightmapBakeType = (LightmapBakeType)lightsInfo[i].lightmapBaketype;
|
||
bakingOutput.mixedLightingMode = (MixedLightingMode)lightsInfo[i].mixedLightingMode;
|
||
|
||
lightsInfo[i].light.bakingOutput = bakingOutput;
|
||
|
||
}
|
||
|
||
|
||
}
|
||
|
||
#if UNITY_EDITOR
|
||
[UnityEditor.MenuItem("Assets/Bake Prefab Lightmaps")]
|
||
static void GenerateLightmapInfo()
|
||
{
|
||
if (UnityEditor.Lightmapping.giWorkflowMode != UnityEditor.Lightmapping.GIWorkflowMode.OnDemand)
|
||
{
|
||
Debug.LogError("ExtractLightmapData requires that you have baked you lightmaps and Auto mode is disabled.");
|
||
return;
|
||
}
|
||
UnityEditor.Lightmapping.Bake();
|
||
|
||
PrefabLightmapData[] prefabs = FindObjectsOfType<PrefabLightmapData>();
|
||
|
||
foreach (var instance in prefabs)
|
||
{
|
||
var gameObject = instance.gameObject;
|
||
var rendererInfos = new List<RendererInfo>();
|
||
var lightmaps = new List<Texture2D>();
|
||
var lightmapsDir = new List<Texture2D>();
|
||
var shadowMasks = new List<Texture2D>();
|
||
var lightsInfos = new List<LightInfo>();
|
||
|
||
GenerateLightmapInfo(gameObject, rendererInfos, lightmaps, lightmapsDir, shadowMasks, lightsInfos);
|
||
|
||
instance.m_RendererInfo = rendererInfos.ToArray();
|
||
instance.m_Lightmaps = lightmaps.ToArray();
|
||
instance.m_LightmapsDir = lightmapsDir.ToArray();
|
||
instance.m_LightInfo = lightsInfos.ToArray();
|
||
instance.m_ShadowMasks = shadowMasks.ToArray();
|
||
#if UNITY_2018_3_OR_NEWER
|
||
var targetPrefab = PrefabUtility.GetCorrespondingObjectFromOriginalSource(instance.gameObject) as GameObject;
|
||
if (targetPrefab != null)
|
||
{
|
||
GameObject root = PrefabUtility.GetOutermostPrefabInstanceRoot(instance.gameObject);// 根结点
|
||
//如果当前预制体是是某个嵌套预制体的一部分（IsPartOfPrefabInstance）
|
||
if (root != null)
|
||
{
|
||
GameObject rootPrefab = PrefabUtility.GetCorrespondingObjectFromSource(instance.gameObject);
|
||
string rootPath = AssetDatabase.GetAssetPath(rootPrefab);
|
||
//打开根部预制体
|
||
PrefabUtility.UnpackPrefabInstanceAndReturnNewOutermostRoots(root, PrefabUnpackMode.OutermostRoot);
|
||
try
|
||
{
|
||
//Apply各个子预制体的改变
|
||
PrefabUtility.ApplyPrefabInstance(instance.gameObject, InteractionMode.AutomatedAction);
|
||
}
|
||
catch { }
|
||
finally
|
||
{
|
||
//重新更新根预制体
|
||
PrefabUtility.SaveAsPrefabAssetAndConnect(root, rootPath, InteractionMode.AutomatedAction);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
PrefabUtility.ApplyPrefabInstance(instance.gameObject, InteractionMode.AutomatedAction);
|
||
}
|
||
}
|
||
#else
|
||
var targetPrefab = UnityEditor.PrefabUtility.GetPrefabParent(gameObject) as GameObject;
|
||
if (targetPrefab != null)
|
||
{
|
||
//UnityEditor.Prefab
|
||
UnityEditor.PrefabUtility.ReplacePrefab(gameObject, targetPrefab);
|
||
}
|
||
#endif
|
||
}
|
||
|
||
|
||
}
|
||
|
||
static void GenerateLightmapInfo(GameObject root, List<RendererInfo> rendererInfos, List<Texture2D> lightmaps, List<Texture2D> lightmapsDir, List<Texture2D> shadowMasks, List<LightInfo> lightsInfo)
|
||
{
|
||
var renderers = root.GetComponentsInChildren<MeshRenderer>();
|
||
foreach (MeshRenderer renderer in renderers)
|
||
{
|
||
if (renderer.lightmapIndex != -1)
|
||
{
|
||
RendererInfo info = new RendererInfo();
|
||
info.renderer = renderer;
|
||
|
||
if (renderer.lightmapScaleOffset != Vector4.zero)
|
||
{
|
||
//1ibrium's pointed out this issue : https://docs.unity3d.com/ScriptReference/Renderer-lightmapIndex.html
|
||
if(renderer.lightmapIndex < 0 || renderer.lightmapIndex == 0xFFFE) continue;
|
||
info.lightmapOffsetScale = renderer.lightmapScaleOffset;
|
||
|
||
Texture2D lightmap = LightmapSettings.lightmaps[renderer.lightmapIndex].lightmapColor;
|
||
Texture2D lightmapDir = LightmapSettings.lightmaps[renderer.lightmapIndex].lightmapDir;
|
||
Texture2D shadowMask = LightmapSettings.lightmaps[renderer.lightmapIndex].shadowMask;
|
||
|
||
info.lightmapIndex = lightmaps.IndexOf(lightmap);
|
||
if (info.lightmapIndex == -1)
|
||
{
|
||
info.lightmapIndex = lightmaps.Count;
|
||
lightmaps.Add(lightmap);
|
||
lightmapsDir.Add(lightmapDir);
|
||
shadowMasks.Add(shadowMask);
|
||
}
|
||
|
||
rendererInfos.Add(info);
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
var lights = root.GetComponentsInChildren<Light>(true);
|
||
|
||
foreach (Light l in lights)
|
||
{
|
||
LightInfo lightInfo = new LightInfo();
|
||
lightInfo.light = l;
|
||
lightInfo.lightmapBaketype = (int)l.lightmapBakeType;
|
||
#if UNITY_2020_1_OR_NEWER
|
||
lightInfo.mixedLightingMode = (int)UnityEditor.Lightmapping.lightingSettings.mixedBakeMode;
|
||
#elif UNITY_2018_1_OR_NEWER
|
||
lightInfo.mixedLightingMode = (int)UnityEditor.LightmapEditorSettings.mixedBakeMode;
|
||
#else
|
||
lightInfo.mixedLightingMode = (int)l.bakingOutput.lightmapBakeType;
|
||
#endif
|
||
lightsInfo.Add(lightInfo);
|
||
|
||
}
|
||
}
|
||
#endif
|
||
|
||
}
|