Unity烘焙材质到贴图脚本
这个脚本可以将复杂的材质,比如有法线贴图的材质进行"烘焙",转变为单一的贴图,可用来将Unity的游戏移植到移动平台时候使用.
讲脚本放Editor文件夹里,使用时选择一个Material材质,然后在菜单种"Custom/Bake Material"打开并调整照明和其他参数,点击Bake按钮,就会生成一个单一的贴图.
脚本:BakeMaterial.js
js 代码
讲脚本放Editor文件夹里,使用时选择一个Material材质,然后在菜单种"Custom/Bake Material"打开并调整照明和其他参数,点击Bake按钮,就会生成一个单一的贴图.
脚本:BakeMaterial.js
![程序代码](http://blog.1vr.cn/images/code.gif)
001 |
class BakeMaterialSettings |
002 |
{ |
003 |
private static var kEditorPrefsName = "BakeMaterialSettings" ; |
004 |
|
005 |
static var kBakingLayerShouldBeUnusedInScene = 30; |
006 |
static var kStandardTexNames = new Array ( "_MainTex" , "_BumpMap" , "_Detail" , "_ParallaxMap" , "_Parallax" ); |
007 |
008 |
var bakeAlpha = false ; |
009 |
var bakeMainTexAsWhite = false ; |
010 |
var minTextureResolution = 8; |
011 |
var maxTextureResolution = 2048; |
012 |
013 |
var emptyScene = false ; |
014 |
var useCustomLights = false ; |
015 |
var ambient = Color.black; |
016 |
|
017 |
static var kLights = 3; |
018 |
var enableLight = new boolean[kLights]; |
019 |
var colorLight = new Color[kLights]; |
020 |
var dirLight = new Vector2[kLights]; |
021 |
|
022 |
function BakeMaterialSettings () |
023 |
{ |
024 |
Load (); |
025 |
} |
026 |
|
027 |
function Load () |
028 |
{ |
029 |
bakeAlpha = EditorPrefs.GetBool(kEditorPrefsName + ".bakeAlpha" ); |
030 |
bakeMainTexAsWhite = EditorPrefs.GetBool(kEditorPrefsName + ".bakeMainTexAsWhite" ); |
031 |
minTextureResolution = EditorPrefs.GetInt(kEditorPrefsName + ".minTextureResolution" , 8); |
032 |
maxTextureResolution = EditorPrefs.GetInt(kEditorPrefsName + ".maxTextureResolution" , 2048); |
033 |
034 |
emptyScene = EditorPrefs.GetBool(kEditorPrefsName + ".emptyScene" ); |
035 |
useCustomLights = EditorPrefs.GetBool(kEditorPrefsName + ".useCustomLights" ); |
036 |
ambient.r = EditorPrefs.GetFloat(kEditorPrefsName + ".ambient.r" ); |
037 |
ambient.g = EditorPrefs.GetFloat(kEditorPrefsName + ".ambient.g" ); |
038 |
ambient.b = EditorPrefs.GetFloat(kEditorPrefsName + ".ambient.b" ); |
039 |
ambient.a = EditorPrefs.GetFloat(kEditorPrefsName + ".ambient.a" , 1.0f); |
040 |
|
041 |
for ( var q = 0; q < kLights; ++q) |
042 |
{ |
043 |
enableLight[q] = EditorPrefs.GetBool(kEditorPrefsName + ".enableLight" + q); |
044 |
colorLight[q].r = EditorPrefs.GetFloat(kEditorPrefsName + ".color.r" + q, 0.5f); |
045 |
colorLight[q].g = EditorPrefs.GetFloat(kEditorPrefsName + ".color.g" + q, 0.5f); |
046 |
colorLight[q].b = EditorPrefs.GetFloat(kEditorPrefsName + ".color.b" + q, 0.5f); |
047 |
colorLight[q].a = EditorPrefs.GetFloat(kEditorPrefsName + ".color.a" + q, 1.0f); |
048 |
dirLight[q].x = EditorPrefs.GetFloat(kEditorPrefsName + ".dir.x" + q); |
049 |
dirLight[q].y = EditorPrefs.GetFloat(kEditorPrefsName + ".dir.y" + q); |
050 |
} |
051 |
} |
052 |
|
053 |
function Save () |
054 |
{ |
055 |
EditorPrefs.SetBool(kEditorPrefsName + ".bakeAlpha" , bakeAlpha); |
056 |
EditorPrefs.SetBool(kEditorPrefsName + ".bakeMainTexAsWhite" , bakeMainTexAsWhite); |
057 |
EditorPrefs.SetInt(kEditorPrefsName + ".minTextureResolution" , minTextureResolution); |
058 |
EditorPrefs.SetInt(kEditorPrefsName + ".maxTextureResolution" , maxTextureResolution); |
059 |
060 |
EditorPrefs.GetBool(kEditorPrefsName + ".emptyScene" , emptyScene); |
061 |
EditorPrefs.SetBool(kEditorPrefsName + ".useCustomLights" , useCustomLights); |
062 |
EditorPrefs.SetFloat(kEditorPrefsName + ".ambient.r" , ambient.r); |
063 |
EditorPrefs.SetFloat(kEditorPrefsName + ".ambient.g" , ambient.g); |
064 |
EditorPrefs.SetFloat(kEditorPrefsName + ".ambient.b" , ambient.b); |
065 |
EditorPrefs.SetFloat(kEditorPrefsName + ".ambient.a" , ambient.a); |
066 |
067 |
for ( var q = 0; q < kLights; ++q) |
068 |
{ |
069 |
EditorPrefs.SetBool(kEditorPrefsName + ".enableLight" + q, enableLight[q]); |
070 |
EditorPrefs.SetFloat(kEditorPrefsName + ".color.r" + q, colorLight[q].r); |
071 |
EditorPrefs.SetFloat(kEditorPrefsName + ".color.g" + q, colorLight[q].g); |
072 |
EditorPrefs.SetFloat(kEditorPrefsName + ".color.b" + q, colorLight[q].b); |
073 |
EditorPrefs.SetFloat(kEditorPrefsName + ".color.a" + q, colorLight[q].a); |
074 |
EditorPrefs.SetFloat(kEditorPrefsName + ".dir.x" + q, dirLight[q].x); |
075 |
EditorPrefs.SetFloat(kEditorPrefsName + ".dir.y" + q, dirLight[q].y); |
076 |
} |
077 |
} |
078 |
} |
079 |
080 |
class BakeMaterial extends EditorWindow |
081 |
{ |
082 |
private static var kMateriBakeNodeName = "__MateriaBakeSetup" ; |
083 |
private static var kWindowMinSize = Vector2 (300, 386); |
084 |
|
085 |
private static var settings : BakeMaterialSettings; |
086 |
private static var visible : boolean = false ; |
087 |
|
088 |
private var camera : GameObject; |
089 |
private var plane : GameObject; |
090 |
private var previewTexture : Texture; |
091 |
private var lights : GameObject[] = new GameObject[BakeMaterialSettings.kLights]; |
092 |
private var stateChanged = false ; |
093 |
|
094 |
private var texViewScrollPosition = Vector2.zero; |
095 |
private var lastMaterial : Material; |
096 |
|
097 |
private var originalScene = "" ; |
098 |
|
099 |
private var scheduleBakeOnNextUpdate = false ; |
100 |
101 |
|
102 |
private function SetupScene () |
103 |
{ |
104 |
DestroyScene (); |
105 |
var oldGo = GameObject.Find(kMateriBakeNodeName); |
106 |
if (oldGo) |
107 |
DestroyImmediate (oldGo); |
108 |
camera = new GameObject (kMateriBakeNodeName, Camera); |
109 |
plane = GameObject.CreatePrimitive (PrimitiveType.Plane); |
110 |
111 |
var cam = camera; |
112 |
cam.camera.backgroundColor = Color.black; |
113 |
cam.camera.clearFlags = CameraClearFlags.SolidColor; |
114 |
cam.camera.orthographic = true ; |
115 |
cam.camera.orthographicSize = 5.0; |
116 |
cam.camera.cullingMask = 1 << settings.kBakingLayerShouldBeUnusedInScene; |
117 |
|
118 |
plane.transform.parent = cam.transform; |
119 |
plane.transform.position = Vector3.forward * 10.0; |
120 |
plane.transform.rotation = Quaternion.Euler (0, 0, 180) * Quaternion.Euler (-90, 0, 0); |
121 |
plane.layer = settings.kBakingLayerShouldBeUnusedInScene; |
122 |
|
123 |
for ( var l in lights) |
124 |
{ |
125 |
l = new GameObject ( "Light" , Light); |
126 |
l.light.type = LightType.Directional; |
127 |
l.light.cullingMask = 1 << settings.kBakingLayerShouldBeUnusedInScene; |
128 |
l.transform.parent = cam.transform; |
129 |
l.active = false ; |
130 |
} |
131 |
} |
132 |
|
133 |
private function UpdateScene (m : Material) |
134 |
{ |
135 |
for (q = 0; q < settings.kLights; ++q) |
136 |
{ |
137 |
lights[q].active = settings.useCustomLights & settings.enableLight[q]; |
138 |
lights[q].light.color = settings.colorLight[q]; |
139 |
lights[q].transform.rotation = |
140 |
Quaternion.AngleAxis(settings.dirLight[q].x, Vector3.up) * |
141 |
Quaternion.AngleAxis(settings.dirLight[q].y, Vector3.right); |
142 |
} |
143 |
|
144 |
if (settings.useCustomLights) |
145 |
RenderSettings.ambientLight = settings.ambient; |
146 |
else if (settings.emptyScene) |
147 |
RenderSettings.ambientLight = Color.white; |
148 |
|
149 |
plane.renderer.material = m; |
150 |
} |
151 |
|
152 |
private function DestroyScene () |
153 |
{ |
154 |
GameObject.DestroyImmediate (camera); |
155 |
GameObject.DestroyImmediate (plane); |
156 |
GameObject.DestroyImmediate (previewTexture); |
157 |
} |
158 |
159 |
function UpdateMaterialPreview (m : Material) : RenderTexture |
160 |
{ |
161 |
if (!m) |
162 |
return ; |
163 |
|
164 |
var saveAmbientLight = RenderSettings.ambientLight; |
165 |
var saveMainTexture = m.mainTexture; |
166 |
if (settings.bakeMainTexAsWhite) |
167 |
m.mainTexture = null ; |
168 |
|
169 |
|
170 |
// setup |
171 |
if (!camera) |
172 |
SetupScene (); |
173 |
camera.SetActiveRecursively( true ); |
174 |
UpdateScene (m); |
175 |
|
176 |
var res = FindLargestTextureResolution (plane.renderer.sharedMaterial, settings.minTextureResolution, settings.maxTextureResolution); |
177 |
var rt = RenderCameraToRenderTexture (camera.camera, res.x, res.y); |
178 |
|
179 |
// restore |
180 |
camera.SetActiveRecursively( false ); |
181 |
RenderSettings.ambientLight = saveAmbientLight; |
182 |
m.mainTexture = saveMainTexture; |
183 |
|
184 |
previewTexture = rt; |
185 |
return rt; |
186 |
} |
187 |
|
188 |
function CaptureMaterial(m : Material) |
189 |
{ |
190 |
var matAssetPath = AssetDatabase.GetAssetPath (m); |
191 |
var assetPath = System.IO.Path.Combine (System.IO.Path.GetDirectoryName (matAssetPath), System.IO.Path.GetFileNameWithoutExtension (matAssetPath)); |
192 |
193 |
var rt = UpdateMaterialPreview (m); |
194 |
RenderTextureToPNG (rt, settings.bakeAlpha, assetPath + ".png" ); |
195 |
} |
196 |
197 |
function OnEnable () |
198 |
{ |
199 |
if (!settings) |
200 |
settings = new BakeMaterialSettings (); |
201 |
SetupScene (); |
202 |
visible = true ; |
203 |
} |
204 |
|
205 |
function OnDisable () |
206 |
{ |
207 |
DestroyScene (); |
208 |
settings.Save (); |
209 |
visible = false ; |
210 |
} |
211 |
212 |
static function GetTargetMaterial () : Material |
213 |
{ |
214 |
return EditorUtility.InstanceIDToObject (Selection.activeInstanceID) as Material; |
215 |
} |
216 |
217 |
function OnSelectionChange () |
218 |
{ |
219 |
Repaint (); |
220 |
} |
221 |
222 |
function Update () |
223 |
{ |
224 |
var rebuildScene = false ; |
225 |
if (scheduleBakeOnNextUpdate) |
226 |
{ |
227 |
Bake (); |
228 |
scheduleBakeOnNextUpdate = false ; |
229 |
rebuildScene = true ; |
230 |
} |
231 |
|
232 |
if (originalScene == "" && EditorApplication.currentScene == "" ) |
233 |
settings.emptyScene = true ; |
234 |
|
235 |
if (settings.emptyScene && originalScene == "" && EditorApplication.currentScene != "" ) |
236 |
{ |
237 |
DestroyScene (); |
238 |
if (EditorApplication.SaveCurrentSceneIfUserWantsTo ()) |
239 |
{ |
240 |
originalScene = EditorApplication.currentScene; |
241 |
EditorApplication.NewScene (); |
242 |
} |
243 |
else |
244 |
settings.emptyScene = false ; |
245 |
rebuildScene = true ; |
246 |
} |
247 |
else if (!settings.emptyScene && originalScene != "" ) |
248 |
{ |
249 |
EditorApplication.OpenScene (originalScene); |
250 |
rebuildScene = true ; |
251 |
originalScene = "" ; |
252 |
} |
253 |
|
254 |
if (rebuildScene) |
255 |
{ |
256 |
SetupScene (); |
257 |
} |
258 |
|
259 |
if (rebuildScene || stateChanged || !settings.emptyScene) |
260 |
{ |
261 |
UpdateMaterialPreview (lastMaterial); |
262 |
Repaint (); |
263 |
stateChanged = false ; |
264 |
} |
265 |
} |
266 |
|
267 |
function OnGUI () |
268 |
{ |
269 |
var material = GetTargetMaterial (); |
270 |
if (lastMaterial != material) |
271 |
UpdateMaterialPreview (material); |
272 |
if (material) |
273 |
lastMaterial = material; |
274 |
|
275 |
EditorGUILayout.BeginHorizontal(); |
276 |
EditorGUILayout.BeginVertical(GUILayout.MaxWidth(200)); |
277 |
if (!(originalScene == "" && EditorApplication.currentScene == "" )) |
278 |
{ |
279 |
settings.emptyScene = !EditorGUILayout.BeginToggleGroup( "Scene ligthing" , !settings.emptyScene); |
280 |
EditorGUILayout.EndToggleGroup(); |
281 |
} |
282 |
settings.useCustomLights = EditorGUILayout.BeginToggleGroup( "Custom lighting" , settings.useCustomLights); |
283 |
if (settings.useCustomLights) |
284 |
{ |
285 |
EditorGUI.indentLevel = 1; |
286 |
settings.ambient = EditorGUILayout.ColorField( "Ambient" , settings.ambient); |
287 |
for ( var q = 0; q < settings.kLights; ++q) |
288 |
{ |
289 |
settings.enableLight[q] = EditorGUILayout.BeginToggleGroup( "Light" , settings.enableLight[q]); |
290 |
EditorGUI.indentLevel = 2; |
291 |
settings.colorLight[q] = EditorGUILayout.ColorField( "Color" , settings.colorLight[q]); |
292 |
settings.dirLight[q] = EditorGUILayout.Vector2Field( "Direction" , settings.dirLight[q]); |
293 |
EditorGUILayout.EndToggleGroup(); |
294 |
} |
295 |
} |
296 |
EditorGUI.indentLevel = 0; |
297 |
EditorGUILayout.EndToggleGroup(); |
298 |
|
299 |
settings.bakeAlpha = EditorGUILayout.Toggle( "Bake Alpha" , settings.bakeAlpha); |
300 |
settings.bakeMainTexAsWhite = !EditorGUILayout.Toggle( "MainTex" , !settings.bakeMainTexAsWhite); |
301 |
settings.minTextureResolution = EditorGUILayout.IntField( "Min Resolution" , settings.minTextureResolution); |
302 |
settings.maxTextureResolution = EditorGUILayout.IntField( "Max Resolution" , settings.maxTextureResolution); |
303 |
settings.minTextureResolution = Mathf.Max(2, settings.minTextureResolution); |
304 |
settings.maxTextureResolution = Mathf.Max(settings.minTextureResolution, settings.maxTextureResolution); |
305 |
306 |
EditorGUILayout.BeginHorizontal(); |
307 |
if (GUILayout.Button( "Bake" )) |
308 |
{ |
309 |
CaptureMaterial (lastMaterial); |
310 |
} |
311 |
if (GUILayout.Button( "Bake Selected" )) |
312 |
{ |
313 |
scheduleBakeOnNextUpdate = true ; |
314 |
} |
315 |
EditorGUILayout.EndHorizontal(); |
316 |
|
317 |
EditorGUILayout.EndVertical(); |
318 |
|
319 |
texViewScrollPosition = EditorGUILayout.BeginScrollView (texViewScrollPosition); |
320 |
var r = GUILayoutUtility.GetAspectRect(1.0f); |
321 |
if (previewTexture) |
322 |
EditorGUI.DrawPreviewTexture(r, previewTexture); |
323 |
EditorGUILayout.EndScrollView(); |
324 |
EditorGUILayout.EndHorizontal(); |
325 |
|
326 |
if (GUI.changed) |
327 |
{ |
328 |
stateChanged = true ; |
329 |
} |
330 |
} |
331 |
|
332 |
@MenuItem( "Custom/Bake Material ..." , false , 5) |
333 |
static function CreateBakeEditor() |
334 |
{ |
335 |
var window = EditorWindow.GetWindow(BakeMaterial); |
336 |
window.title = "Bake Material" ; |
337 |
window.minSize = kWindowMinSize; |
338 |
window.Show(); |
339 |
} |
340 |
341 |
@MenuItem( "Custom/Bake Selected Materials" , false , 4) |
342 |
static function Bake() |
343 |
{ |
344 |
var instanceIDs = Selection.instanceIDs; |
345 |
var currentScene = EditorApplication.currentScene; |
346 |
|
347 |
var wasAlreadyVisible = BakeMaterial.visible; |
348 |
var window = EditorWindow.GetWindow(BakeMaterial); |
349 |
|
350 |
if (window.settings.emptyScene) |
351 |
{ |
352 |
if (!EditorApplication.SaveCurrentSceneIfUserWantsTo ()) |
353 |
return ; |
354 |
EditorApplication.NewScene (); |
355 |
} |
356 |
|
357 |
window.SetupScene (); |
358 |
for ( var i in instanceIDs) |
359 |
{ |
360 |
var m : Material = EditorUtility.InstanceIDToObject (i) as Material; |
361 |
if (m) |
362 |
window.CaptureMaterial (m); |
363 |
} |
364 |
window.DestroyScene (); |
365 |
|
366 |
if (window.settings.emptyScene && currentScene) |
367 |
{ |
368 |
EditorApplication.OpenScene (currentScene); |
369 |
} |
370 |
|
371 |
if (!wasAlreadyVisible) |
372 |
window.Close (); |
373 |
} |
374 |
|
375 |
static function FindLargestTextureResolution (m : Material, minTexRes : int, maxTexRes : int) : Vector2 |
376 |
{ |
377 |
var res = Vector2 (minTexRes, minTexRes); |
378 |
for ( var n in BakeMaterialSettings.kStandardTexNames) |
379 |
{ |
380 |
if (!m.HasProperty (n)) |
381 |
continue ; |
382 |
|
383 |
var t : Texture = m.GetTexture (n); |
384 |
if (!t) |
385 |
continue ; |
386 |
|
387 |
res.x = Mathf.Max (res.x, t.width); |
388 |
res.y = Mathf.Max (res.y, t.height); |
389 |
} |
390 |
res.x = Mathf.Min (res.x, maxTexRes); |
391 |
res.y = Mathf.Min (res.y, maxTexRes); |
392 |
return res; |
393 |
} |
394 |
|
395 |
static function RenderCameraToRenderTexture (cam : Camera, width : int, height : int) : RenderTexture |
396 |
{ |
397 |
var rt = cam.camera.targetTexture; |
398 |
if (rt && rt.width != width && rt.height != height) |
399 |
DestroyImmediate(rt); |
400 |
if (!rt) |
401 |
rt = new RenderTexture (width, height, 24); |
402 |
cam.camera.targetTexture = rt; |
403 |
cam.camera.Render (); |
404 |
return rt; |
405 |
} |
406 |
|
407 |
static function RenderTextureToPNG (rt : RenderTexture, bakeAlpha : boolean, assetPath : String) |
408 |
{ |
409 |
RenderTexture.active = rt; |
410 |
|
411 |
var screenShot = new Texture2D (rt.width, rt.height, bakeAlpha? TextureFormat.ARGB32 : TextureFormat.RGB24, false ); |
412 |
screenShot.ReadPixels (Rect (0, 0, rt.width, rt.height), 0, 0); |
413 |
|
414 |
RenderTexture.active = null ; |
415 |
|
416 |
var bytes = screenShot.EncodeToPNG (); |
417 |
System.IO.File.WriteAllBytes (assetPath, bytes); |
418 |
|
419 |
AssetDatabase.ImportAsset (assetPath, ImportAssetOptions.ForceUpdate); |
420 |
} |
421 |
} |