Firstborn/Assets/MeshBaker/scripts/MB3_TextureBaker.cs
Schaken-Mods 959e80cf72 assets upload
assets upload description.
2023-03-28 12:16:30 -05:00

574 lines
21 KiB
C#

//----------------------------------------------
// MeshBaker
// Copyright © 2011-2012 Ian Deane
//----------------------------------------------
using UnityEngine;
using System.Collections;
using System.Collections.Specialized;
using System;
using System.Collections.Generic;
using System.Text;
using System.IO;
using DigitalOpus.MB.Core;
/// <summary>
/// Component that handles baking materials into a combined material.
///
/// The result of the material baking process is a MB2_TextureBakeResults object, which
/// becomes the input for the mesh baking.
///
/// This class uses the MB_TextureCombiner to do the combining.
///
/// This class is a Component (MonoBehavior) so it is serialized and found using GetComponent. If
/// you want to access the texture baking functionality without creating a Component then use MB_TextureCombiner
/// directly.
/// </summary>
public class MB3_TextureBaker : MB3_MeshBakerRoot
{
public MB2_LogLevel LOG_LEVEL = MB2_LogLevel.info;
[SerializeField]
protected MB2_TextureBakeResults _textureBakeResults;
public override MB2_TextureBakeResults textureBakeResults
{
get { return _textureBakeResults; }
set { _textureBakeResults = value; }
}
[SerializeField]
protected int _atlasPadding = 1;
public virtual int atlasPadding
{
get { return _atlasPadding; }
set { _atlasPadding = value; }
}
[SerializeField]
protected int _maxAtlasSize = 4096;
public virtual int maxAtlasSize
{
get { return _maxAtlasSize; }
set { _maxAtlasSize = value; }
}
[SerializeField]
protected bool _useMaxAtlasWidthOverride = false;
public virtual bool useMaxAtlasWidthOverride
{
get { return _useMaxAtlasWidthOverride; }
set { _useMaxAtlasWidthOverride = value; }
}
[SerializeField]
protected int _maxAtlasWidthOverride = 4096;
public virtual int maxAtlasWidthOverride
{
get { return _maxAtlasWidthOverride; }
set { _maxAtlasWidthOverride = value; }
}
[SerializeField]
protected bool _useMaxAtlasHeightOverride = false;
public virtual bool useMaxAtlasHeightOverride
{
get { return _useMaxAtlasHeightOverride; }
set { _useMaxAtlasHeightOverride = value; }
}
[SerializeField]
protected int _maxAtlasHeightOverride = 4096;
public virtual int maxAtlasHeightOverride
{
get { return _maxAtlasHeightOverride; }
set { _maxAtlasHeightOverride = value; }
}
[SerializeField]
protected bool _resizePowerOfTwoTextures = false;
public virtual bool resizePowerOfTwoTextures
{
get { return _resizePowerOfTwoTextures; }
set { _resizePowerOfTwoTextures = value; }
}
[SerializeField]
protected bool _fixOutOfBoundsUVs = false; //is considerMeshUVs but can't change because it would break all existing bakers
public virtual bool fixOutOfBoundsUVs
{
get { return _fixOutOfBoundsUVs; }
set { _fixOutOfBoundsUVs = value; }
}
[SerializeField]
protected int _maxTilingBakeSize = 1024;
public virtual int maxTilingBakeSize
{
get { return _maxTilingBakeSize; }
set { _maxTilingBakeSize = value; }
}
[SerializeField]
protected MB2_PackingAlgorithmEnum _packingAlgorithm = MB2_PackingAlgorithmEnum.MeshBakerTexturePacker;
public virtual MB2_PackingAlgorithmEnum packingAlgorithm
{
get { return _packingAlgorithm; }
set { _packingAlgorithm = value; }
}
[SerializeField]
protected bool _meshBakerTexturePackerForcePowerOfTwo = true;
public bool meshBakerTexturePackerForcePowerOfTwo
{
get { return _meshBakerTexturePackerForcePowerOfTwo; }
set { _meshBakerTexturePackerForcePowerOfTwo = value; }
}
[SerializeField]
protected List<ShaderTextureProperty> _customShaderProperties = new List<ShaderTextureProperty>();
public virtual List<ShaderTextureProperty> customShaderProperties
{
get { return _customShaderProperties; }
set { _customShaderProperties = value; }
}
//this is depricated
[SerializeField]
protected List<string> _customShaderPropNames_Depricated = new List<string>();
public virtual List<string> customShaderPropNames
{
get { return _customShaderPropNames_Depricated; }
set { _customShaderPropNames_Depricated = value; }
}
[SerializeField]
protected bool _doMultiMaterial;
public virtual bool doMultiMaterial
{
get { return _doMultiMaterial; }
set { _doMultiMaterial = value; }
}
[SerializeField]
protected bool _doMultiMaterialSplitAtlasesIfTooBig = true;
public virtual bool doMultiMaterialSplitAtlasesIfTooBig
{
get { return _doMultiMaterialSplitAtlasesIfTooBig; }
set { _doMultiMaterialSplitAtlasesIfTooBig = value; }
}
[SerializeField]
protected bool _doMultiMaterialSplitAtlasesIfOBUVs = true;
public virtual bool doMultiMaterialSplitAtlasesIfOBUVs
{
get { return _doMultiMaterialSplitAtlasesIfOBUVs; }
set { _doMultiMaterialSplitAtlasesIfOBUVs = value; }
}
[SerializeField]
protected Material _resultMaterial;
public virtual Material resultMaterial
{
get { return _resultMaterial; }
set { _resultMaterial = value; }
}
[SerializeField]
protected bool _considerNonTextureProperties = false;
public bool considerNonTextureProperties
{
get { return _considerNonTextureProperties; }
set { _considerNonTextureProperties = value; }
}
[SerializeField]
protected bool _doSuggestTreatment = true;
public bool doSuggestTreatment
{
get { return _doSuggestTreatment; }
set { _doSuggestTreatment = value; }
}
private CreateAtlasesCoroutineResult _coroutineResult;
public CreateAtlasesCoroutineResult CoroutineResult
{
get
{
return _coroutineResult;
}
}
public MB_MultiMaterial[] resultMaterials = new MB_MultiMaterial[0];
public List<GameObject> objsToMesh; //todo make this Renderer
public override List<GameObject> GetObjectsToCombine()
{
if (objsToMesh == null) objsToMesh = new List<GameObject>();
return objsToMesh;
}
public MB_AtlasesAndRects[] CreateAtlases()
{
return CreateAtlases(null, false, null);
}
public delegate void OnCombinedTexturesCoroutineSuccess();
public delegate void OnCombinedTexturesCoroutineFail();
public OnCombinedTexturesCoroutineSuccess onBuiltAtlasesSuccess;
public OnCombinedTexturesCoroutineFail onBuiltAtlasesFail;
public MB_AtlasesAndRects[] OnCombinedTexturesCoroutineAtlasesAndRects;
/*
bool _CreateAtlasesCoroutineSuccess = false;
public bool CreateAtlasesCoroutineSuccess
{
get { return _CreateAtlasesCoroutineSuccess; }
}
bool _CreateAtlasesCoroutineIsFinished = false;
public bool CreateAtlasesCoroutineIsFinished
{
get { return _CreateAtlasesCoroutineIsFinished; }
}
*/
public class CreateAtlasesCoroutineResult
{
public bool success = true;
public bool isFinished = false;
}
public IEnumerator CreateAtlasesCoroutine(ProgressUpdateDelegate progressInfo, CreateAtlasesCoroutineResult coroutineResult, bool saveAtlasesAsAssets = false, MB2_EditorMethodsInterface editorMethods = null, float maxTimePerFrame = .01f)
{
MBVersionConcrete mbv = new MBVersionConcrete ();
if (!MB3_TextureCombiner._RunCorutineWithoutPauseIsRunning &&( mbv.GetMajorVersion() < 5 ||(mbv.GetMajorVersion() == 5 && mbv.GetMinorVersion() < 3))){
Debug.LogError("Running the texture combiner as a coroutine only works in Unity 5.3 and higher");
coroutineResult.success = false;
yield break;
}
this.OnCombinedTexturesCoroutineAtlasesAndRects = null;
if (maxTimePerFrame <= 0f)
{
Debug.LogError("maxTimePerFrame must be a value greater than zero");
coroutineResult.isFinished = true;
yield break;
}
MB2_ValidationLevel vl = Application.isPlaying ? MB2_ValidationLevel.quick : MB2_ValidationLevel.robust;
if (!DoCombinedValidate(this, MB_ObjsToCombineTypes.dontCare, null, vl))
{
coroutineResult.isFinished = true;
yield break;
}
if (_doMultiMaterial && !_ValidateResultMaterials())
{
coroutineResult.isFinished = true;
yield break;
}
else if (!_doMultiMaterial)
{
if (_resultMaterial == null)
{
Debug.LogError("Combined Material is null please create and assign a result material.");
coroutineResult.isFinished = true;
yield break;
}
Shader targShader = _resultMaterial.shader;
for (int i = 0; i < objsToMesh.Count; i++)
{
Material[] ms = MB_Utility.GetGOMaterials(objsToMesh[i]);
for (int j = 0; j < ms.Length; j++)
{
Material m = ms[j];
if (m != null && m.shader != targShader)
{
Debug.LogWarning("Game object " + objsToMesh[i] + " does not use shader " + targShader + " it may not have the required textures. If not small solid color textures will be generated.");
}
}
}
}
MB3_TextureCombiner combiner = CreateAndConfigureTextureCombiner();
combiner.saveAtlasesAsAssets = saveAtlasesAsAssets;
//initialize structure to store results
int numResults = 1;
if (_doMultiMaterial) numResults = resultMaterials.Length;
OnCombinedTexturesCoroutineAtlasesAndRects = new MB_AtlasesAndRects[numResults];
for (int i = 0; i < OnCombinedTexturesCoroutineAtlasesAndRects.Length; i++)
{
OnCombinedTexturesCoroutineAtlasesAndRects[i] = new MB_AtlasesAndRects();
}
//Do the material combining.
for (int i = 0; i < OnCombinedTexturesCoroutineAtlasesAndRects.Length; i++)
{
Material resMatToPass = null;
List<Material> sourceMats = null;
if (_doMultiMaterial)
{
sourceMats = resultMaterials[i].sourceMaterials;
resMatToPass = resultMaterials[i].combinedMaterial;
combiner.fixOutOfBoundsUVs = resultMaterials[i].considerMeshUVs;
}
else
{
resMatToPass = _resultMaterial;
}
MB3_TextureCombiner.CombineTexturesIntoAtlasesCoroutineResult coroutineResult2 = new MB3_TextureCombiner.CombineTexturesIntoAtlasesCoroutineResult();
yield return combiner.CombineTexturesIntoAtlasesCoroutine(progressInfo, OnCombinedTexturesCoroutineAtlasesAndRects[i], resMatToPass, objsToMesh, sourceMats, editorMethods, coroutineResult2, maxTimePerFrame);
coroutineResult.success = coroutineResult2.success;
if (!coroutineResult.success)
{
coroutineResult.isFinished = true;
yield break;
}
}
unpackMat2RectMap(textureBakeResults);
//Save the results
textureBakeResults.doMultiMaterial = _doMultiMaterial;
//textureBakeResults.resultMaterial = _resultMaterial;
if (_doMultiMaterial)
{
textureBakeResults.resultMaterials = resultMaterials;
} else
{
MB_MultiMaterial[] resMats = new MB_MultiMaterial[1];
resMats[0] = new MB_MultiMaterial();
resMats[0].combinedMaterial = _resultMaterial;
resMats[0].considerMeshUVs = _fixOutOfBoundsUVs;
resMats[0].sourceMaterials = new List<Material>();
for (int i = 0; i < textureBakeResults.materialsAndUVRects.Length; i++)
{
resMats[0].sourceMaterials.Add(textureBakeResults.materialsAndUVRects[i].material);
}
textureBakeResults.resultMaterials = resMats;
}
//textureBakeResults.fixOutOfBoundsUVs = combiner.fixOutOfBoundsUVs;
//set the texture bake resultAtlasesAndRects on the Mesh Baker component if it exists
MB3_MeshBakerCommon[] mb = GetComponentsInChildren<MB3_MeshBakerCommon>();
for (int i = 0; i < mb.Length; i++)
{
mb[i].textureBakeResults = textureBakeResults;
}
if (LOG_LEVEL >= MB2_LogLevel.info) Debug.Log("Created Atlases");
coroutineResult.isFinished = true;
if (coroutineResult.success && onBuiltAtlasesSuccess != null)
{
onBuiltAtlasesSuccess();
}
if (!coroutineResult.success && onBuiltAtlasesFail != null)
{
onBuiltAtlasesFail();
}
}
/// <summary>
/// Creates the atlases.
/// </summary>
/// <returns>
/// The atlases.
/// </returns>
/// <param name='progressInfo'>
/// Progress info is a delegate function that displays a progress dialog. Can be null
/// </param>
/// <param name='saveAtlasesAsAssets'>
/// if true atlases are saved as assets in the project folder. Othersise they are instances in memory
/// </param>
/// <param name='editorMethods'>
/// Texture format tracker. Contains editor functionality such as save assets. Can be null.
/// </param>
public MB_AtlasesAndRects[] CreateAtlases(ProgressUpdateDelegate progressInfo, bool saveAtlasesAsAssets = false, MB2_EditorMethodsInterface editorMethods = null)
{
MB_AtlasesAndRects[] mAndAs = null;
try
{
//mAndAs = _CreateAtlases(progressInfo, saveAtlasesAsAssets, editorMethods);
_coroutineResult = new CreateAtlasesCoroutineResult();
MB3_TextureCombiner.RunCorutineWithoutPause(CreateAtlasesCoroutine(progressInfo, _coroutineResult, saveAtlasesAsAssets, editorMethods, 1000f), 0);
if (_coroutineResult.success && textureBakeResults != null) {
mAndAs = this.OnCombinedTexturesCoroutineAtlasesAndRects;
}
}
catch (Exception e)
{
Debug.LogError(e);
}
finally
{
if (saveAtlasesAsAssets)
{ //Atlases were saved to project so we don't need these ones
if (mAndAs != null)
{
for (int j = 0; j < mAndAs.Length; j++)
{
MB_AtlasesAndRects mAndA = mAndAs[j];
if (mAndA != null && mAndA.atlases != null)
{
for (int i = 0; i < mAndA.atlases.Length; i++)
{
if (mAndA.atlases[i] != null)
{
if (editorMethods != null) editorMethods.Destroy(mAndA.atlases[i]);
else MB_Utility.Destroy(mAndA.atlases[i]);
}
}
}
}
}
}
}
return mAndAs;
}
void unpackMat2RectMap(MB2_TextureBakeResults tbr)
{
List<Material> ms = new List<Material>();
List<MB_MaterialAndUVRect> mss = new List<MB_MaterialAndUVRect>();
List<Rect> rs = new List<Rect>();
for (int i = 0; i < OnCombinedTexturesCoroutineAtlasesAndRects.Length; i++)
{
MB_AtlasesAndRects newMesh = this.OnCombinedTexturesCoroutineAtlasesAndRects[i];
List<MB_MaterialAndUVRect> map = newMesh.mat2rect_map;
if (map != null)
{
for (int j = 0; j < map.Count; j++)
{
mss.Add(map[j]);
ms.Add(map[j].material);
rs.Add(map[j].atlasRect);
}
}
}
tbr.version = MB2_TextureBakeResults.VERSION;
tbr.materialsAndUVRects = mss.ToArray();
}
public MB3_TextureCombiner CreateAndConfigureTextureCombiner()
{
MB3_TextureCombiner combiner = new MB3_TextureCombiner();
combiner.LOG_LEVEL = LOG_LEVEL;
combiner.atlasPadding = _atlasPadding;
combiner.maxAtlasSize = _maxAtlasSize;
combiner.maxAtlasHeightOverride = _maxAtlasHeightOverride;
combiner.maxAtlasWidthOverride = _maxAtlasWidthOverride;
combiner.useMaxAtlasHeightOverride = _useMaxAtlasHeightOverride;
combiner.useMaxAtlasWidthOverride = _useMaxAtlasWidthOverride;
combiner.customShaderPropNames = _customShaderProperties;
combiner.fixOutOfBoundsUVs = _fixOutOfBoundsUVs;
combiner.maxTilingBakeSize = _maxTilingBakeSize;
combiner.packingAlgorithm = _packingAlgorithm;
combiner.meshBakerTexturePackerForcePowerOfTwo = _meshBakerTexturePackerForcePowerOfTwo;
combiner.resizePowerOfTwoTextures = _resizePowerOfTwoTextures;
combiner.considerNonTextureProperties = _considerNonTextureProperties;
return combiner;
}
public static void ConfigureNewMaterialToMatchOld(Material newMat, Material original)
{
if (original == null)
{
Debug.LogWarning("Original material is null, could not copy properties to " + newMat + ". Setting shader to " + newMat.shader);
return;
}
newMat.shader = original.shader;
newMat.CopyPropertiesFromMaterial(original);
ShaderTextureProperty[] texPropertyNames = MB3_TextureCombinerPipeline.shaderTexPropertyNames;
for (int j = 0; j < texPropertyNames.Length; j++)
{
Vector2 scale = Vector2.one;
Vector2 offset = Vector2.zero;
if (newMat.HasProperty(texPropertyNames[j].name))
{
newMat.SetTextureOffset(texPropertyNames[j].name, offset);
newMat.SetTextureScale(texPropertyNames[j].name, scale);
}
}
}
string PrintSet(HashSet<Material> s)
{
StringBuilder sb = new StringBuilder();
foreach (Material m in s)
{
sb.Append(m + ",");
}
return sb.ToString();
}
bool _ValidateResultMaterials()
{
HashSet<Material> allMatsOnObjs = new HashSet<Material>();
for (int i = 0; i < objsToMesh.Count; i++)
{
if (objsToMesh[i] != null)
{
Material[] ms = MB_Utility.GetGOMaterials(objsToMesh[i]);
for (int j = 0; j < ms.Length; j++)
{
if (ms[j] != null) allMatsOnObjs.Add(ms[j]);
}
}
}
HashSet<Material> allMatsInMapping = new HashSet<Material>();
for (int i = 0; i < resultMaterials.Length; i++)
{
for (int j = i+1; j < resultMaterials.Length; j++)
{
if (resultMaterials[i].combinedMaterial == resultMaterials[j].combinedMaterial)
{
Debug.LogError(String.Format("Source To Combined Mapping: Submesh {0} and Submesh {1} use the same combined material. These should be different", i, j));
return false;
}
}
MB_MultiMaterial mm = resultMaterials[i];
if (mm.combinedMaterial == null)
{
Debug.LogError("Combined Material is null please create and assign a result material.");
return false;
}
Shader targShader = mm.combinedMaterial.shader;
for (int j = 0; j < mm.sourceMaterials.Count; j++)
{
if (mm.sourceMaterials[j] == null)
{
Debug.LogError("There are null entries in the list of Source Materials");
return false;
}
if (targShader != mm.sourceMaterials[j].shader)
{
Debug.LogWarning("Source material " + mm.sourceMaterials[j] + " does not use shader " + targShader + " it may not have the required textures. If not empty textures will be generated.");
}
if (allMatsInMapping.Contains(mm.sourceMaterials[j]))
{
Debug.LogError("A Material " + mm.sourceMaterials[j] + " appears more than once in the list of source materials in the source material to combined mapping. Each source material must be unique.");
return false;
}
allMatsInMapping.Add(mm.sourceMaterials[j]);
}
}
if (allMatsOnObjs.IsProperSubsetOf(allMatsInMapping))
{
allMatsInMapping.ExceptWith(allMatsOnObjs);
Debug.LogWarning("There are materials in the mapping that are not used on your source objects: " + PrintSet(allMatsInMapping));
}
if (resultMaterials != null && resultMaterials.Length > 0 && allMatsInMapping.IsProperSubsetOf(allMatsOnObjs))
{
allMatsOnObjs.ExceptWith(allMatsInMapping);
Debug.LogError("There are materials on the objects to combine that are not in the mapping: " + PrintSet(allMatsOnObjs));
return false;
}
return true;
}
}