PrefabLightmapData.cs 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. using UnityEngine;
  2. using System.Collections;
  3. using System.Collections.Generic;
  4. public class PrefabLightmapData : MonoBehaviour
  5. {
  6. [System.Serializable]
  7. struct RendererInfo
  8. {
  9. public Renderer renderer;
  10. public int lightmapIndex;
  11. public Vector4 lightmapOffsetScale;
  12. }
  13. [SerializeField]
  14. RendererInfo[] m_RendererInfo;
  15. [SerializeField]
  16. Texture2D[] m_Lightmaps;
  17. void Awake()
  18. {
  19. }
  20. void init()
  21. {
  22. if (m_RendererInfo == null || m_RendererInfo.Length == 0)
  23. return;
  24. var lightmaps = LightmapSettings.lightmaps;
  25. var combinedLightmaps = new LightmapData[lightmaps.Length + m_Lightmaps.Length];
  26. lightmaps.CopyTo(combinedLightmaps, 0);
  27. for (int i = 0; i < m_Lightmaps.Length; i++)
  28. {
  29. combinedLightmaps[i + lightmaps.Length] = new LightmapData();
  30. combinedLightmaps[i + lightmaps.Length].lightmapColor = m_Lightmaps[i];
  31. }
  32. ApplyRendererInfo(m_RendererInfo, lightmaps.Length);
  33. LightmapSettings.lightmaps = combinedLightmaps;
  34. }
  35. private void OnEnable()
  36. {
  37. init();
  38. Debug.Log("11");
  39. }
  40. static void ApplyRendererInfo(RendererInfo[] infos, int lightmapOffsetIndex)
  41. {
  42. for (int i = 0; i < infos.Length; i++)
  43. {
  44. var info = infos[i];
  45. info.renderer.lightmapIndex = info.lightmapIndex + lightmapOffsetIndex;
  46. info.renderer.lightmapScaleOffset = info.lightmapOffsetScale;
  47. }
  48. }
  49. #if UNITY_EDITOR
  50. [UnityEditor.MenuItem("Assets/Bake Prefab Lightmaps")]
  51. static void GenerateLightmapInfo()
  52. {
  53. if (UnityEditor.Lightmapping.giWorkflowMode != UnityEditor.Lightmapping.GIWorkflowMode.OnDemand)
  54. {
  55. Debug.LogError("ExtractLightmapData requires that you have baked you lightmaps and Auto mode is disabled.");
  56. return;
  57. }
  58. UnityEditor.Lightmapping.Bake();
  59. PrefabLightmapData[] prefabs = FindObjectsOfType<PrefabLightmapData>();
  60. foreach (var instance in prefabs)
  61. {
  62. var gameObject = instance.gameObject;
  63. var rendererInfos = new List<RendererInfo>();
  64. var lightmaps = new List<Texture2D>();
  65. GenerateLightmapInfo(gameObject, rendererInfos, lightmaps);
  66. instance.m_RendererInfo = rendererInfos.ToArray();
  67. instance.m_Lightmaps = lightmaps.ToArray();
  68. var targetPrefab = UnityEditor.PrefabUtility.GetPrefabParent(gameObject) as GameObject;
  69. if (targetPrefab != null)
  70. {
  71. //UnityEditor.Prefab
  72. UnityEditor.PrefabUtility.ReplacePrefab(gameObject, targetPrefab);
  73. }
  74. }
  75. }
  76. static void GenerateLightmapInfo(GameObject root, List<RendererInfo> rendererInfos, List<Texture2D> lightmaps)
  77. {
  78. var renderers = root.GetComponentsInChildren<MeshRenderer>();
  79. foreach (MeshRenderer renderer in renderers)
  80. {
  81. if (renderer.lightmapIndex != -1)
  82. {
  83. RendererInfo info = new RendererInfo();
  84. info.renderer = renderer;
  85. info.lightmapOffsetScale = renderer.lightmapScaleOffset;
  86. Texture2D lightmap = LightmapSettings.lightmaps[renderer.lightmapIndex].lightmapColor;
  87. info.lightmapIndex = lightmaps.IndexOf(lightmap);
  88. if (info.lightmapIndex == -1)
  89. {
  90. info.lightmapIndex = lightmaps.Count;
  91. lightmaps.Add(lightmap);
  92. }
  93. rendererInfos.Add(info);
  94. }
  95. }
  96. }
  97. #endif
  98. }