public static Texture2D RenderAndExtractImageToTexture(HEU_SessionBase session, HAPI_MaterialInfo materialInfo, HAPI_ParmId textureParmID, string textureName, string assetCacheFolderPath)
{
//Debug.LogFormat("Rendering texture {0} with name {1} for material {2} at path {3}", textureParmID, textureName, materialInfo.nodeId, assetCacheFolderPath);
Texture2D texture = null;
// First we get Houdini to render the texture to an image buffer, then query the buffer over HAPI
// Next we convert to PNG, and write out to file in our Assets directory
// The reason for querying as a buffer is to workaround a bug with ExtractHoudiniImageToTextureFile
// Note: intentionly ignoring any errors as sometimes there aren't any textures
if (session.RenderTextureToImage(materialInfo.nodeId, textureParmID, false))
{
texture = HEU_MaterialFactory.ExtractHoudiniImageToTextureRaw(session, materialInfo, "C A");
if(texture != null)
{
texture.name = textureName;
// Get the Textures folder in the assetCacheFolderPath. Make sure it exists.
assetCacheFolderPath = HEU_AssetDatabase.AppendTexturesPathToAssetFolder(assetCacheFolderPath);
HEU_AssetDatabase.CreatePathWithFolders(assetCacheFolderPath);
// We are defaulting to PNG here if no extension already set. This forces it to use PNG format below.
if (!textureName.EndsWith(".png") && !textureName.EndsWith(".jpg"))
{
textureName = textureName + ".png";
}
string textureFileName = HEU_Platform.BuildPath(assetCacheFolderPath, string.Format("{0}", textureName));
byte[] encodedBytes;
if(textureName.EndsWith(".jpg"))
{
encodedBytes = texture.EncodeToJPG();
}
else // Use PNG otherwise
{
encodedBytes = texture.EncodeToPNG();
}
HEU_Platform.WriteBytes(textureFileName, encodedBytes);
// Re-import for project to recognize the new texture file
HEU_AssetDatabase.ImportAsset(textureFileName, HEU_AssetDatabase.HEU_ImportAssetOptions.Default);
// Load the new texture file
texture = HEU_AssetDatabase.LoadAssetAtPath(textureFileName, typeof(Texture2D)) as Texture2D;
}
//texture = HEU_MaterialFactory.ExtractHoudiniImageToTextureFile(session, materialInfo, "C A", assetCacheFolderPath);
}
return texture;
}
private static Texture2D ExtractHoudiniImageToTextureRaw(HEU_SessionBase session, HAPI_MaterialInfo materialInfo, string imagePlanes)
{
Texture2D textureResult = null;
HAPI_ImageInfo imageInfo = new HAPI_ImageInfo();
if (!session.GetImageInfo(materialInfo.nodeId, ref imageInfo))
{
return textureResult;
}
imageInfo.dataFormat = HAPI_ImageDataFormat.HAPI_IMAGE_DATA_INT8;
imageInfo.interleaved = true;
imageInfo.packing = HAPI_ImagePacking.HAPI_IMAGE_PACKING_RGBA;
imageInfo.gamma = HEU_PluginSettings.ImageGamma;
session.SetImageInfo(materialInfo.nodeId, ref imageInfo);
// Extract image to buffer
byte[] imageData = null;
if (!session.ExtractImageToMemory(materialInfo.nodeId, HEU_Defines.HAPI_RAW_FORMAT_NAME, imagePlanes, out imageData))
{
return textureResult;
}
int colorDataSize = imageInfo.xRes * imageInfo.yRes;
if (colorDataSize * 4 != imageData.Length)
{
Debug.LogErrorFormat("Extracted image size does not match expected image info size."
+ " Try using non-raw format for texture extraction.");
return textureResult;
}
Color32[] colorData = new Color32[colorDataSize];
for (int i = 0; i < colorDataSize; ++i)
{
colorData[i].r = imageData[i * 4 + 0];
colorData[i].g = imageData[i * 4 + 1];
colorData[i].b = imageData[i * 4 + 2];
colorData[i].a = imageData[i * 4 + 3];
}
textureResult = new Texture2D(imageInfo.xRes, imageInfo.yRes, TextureFormat.ARGB32, false);
textureResult.SetPixels32(colorData);
textureResult.Apply();
return textureResult;
}
public static Texture2D ExtractHoudiniImageToTextureFile(HEU_SessionBase session, HAPI_MaterialInfo materialInfo, string imagePlanes, string assetCacheFolderPath)
{
Texture2D textureResult = null;
// Get the Textures folder in the assetCacheFolderPath. Make sure it exists.
assetCacheFolderPath = HEU_AssetDatabase.AppendTexturesPathToAssetFolder(assetCacheFolderPath);
HEU_AssetDatabase.CreatePathWithFolders(assetCacheFolderPath);
// Need to pass in full path to Houdini to write out the file
assetCacheFolderPath = HEU_AssetDatabase.GetAssetFullPath(assetCacheFolderPath);
if (assetCacheFolderPath == null)
{
return textureResult;
}
HAPI_ImageInfo imageInfo = new HAPI_ImageInfo();
if (!session.GetImageInfo(materialInfo.nodeId, ref imageInfo))
{
return textureResult;
}
// This will return null if the current imageInfo file format is supported by Unity, otherwise
// returns a Unity supported file format.
string desiredFileFormatName = HEU_MaterialData.GetSupportedFileFormat(session, ref imageInfo);
// Extract image to file
string writtenFilePath = null;
if (!session.ExtractImageToFile(materialInfo.nodeId, desiredFileFormatName, imagePlanes, assetCacheFolderPath, out writtenFilePath))
{
return textureResult;
}
HEU_AssetDatabase.SaveAndRefreshDatabase();
// Convert full path back to relative in order to work with AssetDatabase
string assetRelativePath = HEU_AssetDatabase.GetAssetRelativePath(writtenFilePath);
// Re-import to refresh the project
HEU_AssetDatabase.ImportAsset(assetRelativePath, HEU_AssetDatabase.HEU_ImportAssetOptions.Default);
textureResult = HEU_AssetDatabase.LoadAssetAtPath(assetRelativePath, typeof(Texture2D)) as Texture2D;
//Debug.LogFormat("Loaded texture to file {0} with format {1}", writtenFilePath, textureResult != null ? textureResult.format.ToString() : "none");
return textureResult;
}
public static HEU_MaterialData CreateHoudiniMaterialData(HEU_SessionBase session, HAPI_NodeId assetID, HAPI_NodeId materialID, HAPI_NodeId geoID, HAPI_PartId partID, List<HEU_MaterialData> materialCache, string assetCacheFolderPath)
{
string materialName = "";
if (materialID == HEU_Defines.HEU_INVALID_NODE_ID)
{
return GetOrCreateDefaultMaterialInCache(session, geoID, partID, false, materialCache, assetCacheFolderPath);
}
else
{
materialName = HEU_SessionManager.GetUniqueMaterialShopName(assetID, materialID);
}
HEU_MaterialData materialData = ScriptableObject.CreateInstance<HEU_MaterialData>();
materialData._materialSource = HEU_MaterialData.Source.HOUDINI;
materialData._materialKey = materialID;
materialData._material = HEU_MaterialFactory.CreateNewHoudiniStandardMaterial(assetCacheFolderPath, materialName, true);
materialData._material.name = materialName;
//Debug.LogFormat("New Material ID: {0} - {1}", materialID, materialName);
if (materialID != HEU_Defines.HEU_INVALID_NODE_ID)
{
// Get material info from Houdini to populate the Unity material values
HAPI_MaterialInfo materialInfo = new HAPI_MaterialInfo();
if (session.GetMaterialInfo(materialID, ref materialInfo))
{
if (materialInfo.exists)
{
materialData.UpdateMaterialFromHoudini(materialInfo, assetCacheFolderPath);
}
}
}
//Debug.LogFormat("Created new material with id={0} and name={1}", materialID, materialName);
materialCache.Add(materialData);
return materialData;
}
// GEOMETRY GETTERS -------------------------------------------------------------------------------------------
public static string GetUniqueMaterialShopName(HAPI_NodeId assetID, HAPI_NodeId materialID)
{
HEU_SessionBase sessionBase = GetOrCreateDefaultSession();
if (sessionBase != null)
{
HAPI_AssetInfo assetInfo = new HAPI_AssetInfo();
if (!sessionBase.GetAssetInfo(assetID, ref assetInfo))
{
return "";
}
HAPI_MaterialInfo materialInfo = new HAPI_MaterialInfo();
if (!sessionBase.GetMaterialInfo(materialID, ref materialInfo))
{
return "";
}
HAPI_NodeInfo assetNodeInfo = new HAPI_NodeInfo();
if (!sessionBase.GetNodeInfo(assetID, ref assetNodeInfo))
{
return "";
}
HAPI_NodeInfo materialNodeInfo = new HAPI_NodeInfo();
if (!sessionBase.GetNodeInfo(materialInfo.nodeId, ref materialNodeInfo))
{
return "";
}
string assetNodeName = HEU_SessionManager.GetString(assetNodeInfo.internalNodePathSH, sessionBase);
string materialNodeName = HEU_SessionManager.GetString(materialNodeInfo.internalNodePathSH, sessionBase);
if (assetNodeName.Length > 0 && materialNodeName.Length > 0)
{
// Remove assetNodeName from materialNodeName. Extra position is for separator.
string materialName = materialNodeName.Substring(assetNodeName.Length + 1);
return materialName.Replace("/", "_");
}
}
return "";
}
/// <summary>
/// For this object's _material, we update the shader attributes and
/// fetch the textures from Houdini.
/// </summary>
/// <param name="materialInfo">This material's info from Houdini</param>
/// <param name="assetCacheFolderPath">Path to asset's cache folder</param>
public bool UpdateMaterialFromHoudini(HAPI_MaterialInfo materialInfo, string assetCacheFolderPath)
{
if (_material == null)
{
return false;
}
HEU_SessionBase session = HEU_SessionManager.GetOrCreateDefaultSession();
HAPI_NodeInfo nodeInfo = new HAPI_NodeInfo();
if (!session.GetNodeInfo(materialInfo.nodeId, ref nodeInfo))
{
return false;
}
// Get all parameters of this material
HAPI_ParmInfo[] parmInfos = new HAPI_ParmInfo[nodeInfo.parmCount];
if (!HEU_GeneralUtility.GetArray1Arg(materialInfo.nodeId, session.GetParams, parmInfos, 0, nodeInfo.parmCount))
{
return false;
}
// Assign transparency shader or non-transparent.
bool isTransparent = IsTransparentMaterial(session, materialInfo.nodeId, parmInfos);
if (isTransparent)
{
_material.shader = HEU_MaterialFactory.FindPluginShader(HEU_PluginSettings.DefaultTransparentShader);
}
else
{
_material.shader = HEU_MaterialFactory.FindPluginShader(HEU_PluginSettings.DefaultStandardShader);
}
if (HEU_PluginSettings.UseLegacyShaders)
{
return UseLegacyShaders(materialInfo, assetCacheFolderPath, session, nodeInfo, parmInfos);
}
// Diffuse texture - render & extract
int diffuseMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_TEX1_ATTR, HEU_Defines.MAT_OGL_TEX1_ATTR_ENABLED);
if (diffuseMapParmIndex < 0)
{
diffuseMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_BASECOLOR_ATTR, HEU_Defines.MAT_BASECOLOR_ATTR_ENABLED);
}
if (diffuseMapParmIndex >= 0 && diffuseMapParmIndex < parmInfos.Length)
{
string diffuseTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[diffuseMapParmIndex]);
_material.mainTexture = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[diffuseMapParmIndex].id, diffuseTextureFileName, assetCacheFolderPath, false);
}
Color diffuseColor;
if (!HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_DIFF_ATTR, Color.white, out diffuseColor))
{
HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_DIFF_ATTR, Color.white, out diffuseColor);
}
float alpha;
GetMaterialAlpha(session, materialInfo.nodeId, parmInfos, 1f, out alpha);
if (isTransparent)
{
int opacityMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_OPACITY_MAP_ATTR, HEU_Defines.MAT_OGL_OPACITY_MAP_ATTR_ENABLED);
if (opacityMapParmIndex < 0)
{
opacityMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OPACITY_MAP_ATTR, HEU_Defines.MAT_OPACITY_MAP_ATTR_ENABLED);
}
if (opacityMapParmIndex >= 0 && opacityMapParmIndex < parmInfos.Length)
{
string opacityTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[opacityMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_OPACITY_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[opacityMapParmIndex].id, opacityTextureFileName, assetCacheFolderPath, false));
}
}
diffuseColor.a = alpha;
_material.SetColor(HEU_Defines.UNITY_SHADER_COLOR, diffuseColor);
if (HEU_PluginSettings.UseSpecularShader)
{
Color specular;
Color defaultSpecular = new Color(0.2f, 0.2f, 0.2f, 1);
if (!HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_SPEC_ATTR, defaultSpecular, out specular))
{
HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_SPEC_ATTR, defaultSpecular, out specular);
}
_material.SetColor(HEU_Defines.UNITY_SHADER_SPEC_COLOR, specular);
int specMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_SPEC_MAP_ATTR, HEU_Defines.MAT_OGL_SPEC_MAP_ATTR_ENABLED);
if (specMapParmIndex < 0)
{
specMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_SPEC_MAP_ATTR, HEU_Defines.MAT_SPEC_MAP_ATTR_ENABLED);
}
if (specMapParmIndex >= 0 && specMapParmIndex < parmInfos.Length)
{
string specTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[specMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_SPEC_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[specMapParmIndex].id, specTextureFileName, assetCacheFolderPath, false));
}
}
else
{
float metallic = 0;
if (!HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_METALLIC_ATTR, 0f, out metallic))
{
HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_METALLIC_ATTR, 0f, out metallic);
}
_material.SetFloat(HEU_Defines.UNITY_SHADER_METALLIC, metallic);
int metallicMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_METALLIC_MAP_ATTR, HEU_Defines.MAT_OGL_METALLIC_MAP_ATTR_ENABLED);
if (metallicMapParmIndex < 0)
{
metallicMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_METALLIC_MAP_ATTR, HEU_Defines.MAT_METALLIC_MAP_ATTR_ENABLED);
}
if (metallicMapParmIndex >= 0 && metallicMapParmIndex < parmInfos.Length)
{
string metallicTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[metallicMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_METALLIC_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[metallicMapParmIndex].id, metallicTextureFileName, assetCacheFolderPath, false));
}
}
// Normal map - render & extract texture
int normalMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_NORMAL_ATTR, HEU_Defines.MAT_NORMAL_ATTR_ENABLED);
if (normalMapParmIndex < 0)
{
normalMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_NORMAL_ATTR, "");
}
if (normalMapParmIndex >= 0 && normalMapParmIndex < parmInfos.Length)
{
string normalTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[normalMapParmIndex]);
Texture2D normalMap = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[normalMapParmIndex].id, normalTextureFileName, assetCacheFolderPath, true);
if (normalMap != null)
{
_material.SetTexture(HEU_Defines.UNITY_SHADER_BUMP_MAP, normalMap);
}
}
// Emission
Color emission;
Color defaultEmission = new Color(0, 0, 0, 0);
if (!HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_EMISSIVE_ATTR, defaultEmission, out emission))
{
HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_EMISSIVE_ATTR, defaultEmission, out emission);
}
_material.SetColor(HEU_Defines.UNITY_SHADER_EMISSION_COLOR, emission);
int emissionMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_EMISSIVE_MAP_ATTR, HEU_Defines.MAT_OGL_EMISSIVE_MAP_ATTR_ENABLED);
if (emissionMapParmIndex < 0)
{
emissionMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_EMISSIVE_MAP_ATTR, HEU_Defines.MAT_EMISSIVE_MAP_ATTR_ENABLED);
}
if (emissionMapParmIndex >= 0 && emissionMapParmIndex < parmInfos.Length)
{
string emissionTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[emissionMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_EMISSION_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[emissionMapParmIndex].id, emissionTextureFileName, assetCacheFolderPath, false));
}
// Smoothness (need to invert roughness!)
float roughness;
float defaultRoughness = 0.5f;
if (!HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_ROUGH_ATTR, defaultRoughness, out roughness))
{
HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_ROUGH_ATTR, defaultRoughness, out roughness);
}
// Clamp shininess to non-zero as results in very hard shadows. Unity's UI does not allow zero either.
_material.SetFloat(HEU_Defines.UNITY_SHADER_SMOOTHNESS, Mathf.Max(0.03f, 1.0f - roughness));
int roughMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_ROUGH_MAP_ATTR, HEU_Defines.MAT_OGL_ROUGH_MAP_ATTR_ENABLED);
if (roughMapParmIndex < 0)
{
roughMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_ROUGH_MAP_ATTR, HEU_Defines.MAT_ROUGH_MAP_ATTR_ENABLED);
}
if (roughMapParmIndex >= 0 && roughMapParmIndex < parmInfos.Length)
{
string roughTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[roughMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_SMOOTHNESS_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[roughMapParmIndex].id, roughTextureFileName, assetCacheFolderPath, false, invertTexture: true));
}
// Occlusion (only has ogl map)
int occlusionMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_OCCLUSION_MAP_ATTR, HEU_Defines.MAT_OGL_ROUGH_MAP_ATTR_ENABLED);
if (occlusionMapParmIndex >= 0 && occlusionMapParmIndex < parmInfos.Length)
{
string occlusionTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[occlusionMapParmIndex]);
_material.SetTexture(HEU_Defines.UNITY_SHADER_OCCLUSION_MAP, HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[occlusionMapParmIndex].id, occlusionTextureFileName, assetCacheFolderPath, false));
}
return true;
}
private bool UseLegacyShaders(HAPI_MaterialInfo materialInfo, string assetCacheFolderPath, HEU_SessionBase session, HAPI_NodeInfo nodeInfo, HAPI_ParmInfo[] parmInfos)
{
// Diffuse texture - render & extract
int diffuseMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_TEX1_ATTR, HEU_Defines.MAT_OGL_TEX1_ATTR_ENABLED);
if (diffuseMapParmIndex < 0)
{
diffuseMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_BASECOLOR_ATTR, HEU_Defines.MAT_BASECOLOR_ATTR_ENABLED);
if (diffuseMapParmIndex < 0)
{
diffuseMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_MAP_ATTR, "");
}
}
if (diffuseMapParmIndex >= 0 && diffuseMapParmIndex < parmInfos.Length)
{
string diffuseTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[diffuseMapParmIndex]);
_material.mainTexture = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[diffuseMapParmIndex].id, diffuseTextureFileName, assetCacheFolderPath, false);
}
// Normal map - render & extract texture
int normalMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_NORMAL_ATTR, HEU_Defines.MAT_NORMAL_ATTR_ENABLED);
if (normalMapParmIndex < 0)
{
normalMapParmIndex = HEU_ParameterUtility.FindTextureParamByNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_NORMAL_ATTR, "");
}
if (normalMapParmIndex >= 0 && normalMapParmIndex < parmInfos.Length)
{
string normalTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[normalMapParmIndex]);
Texture2D normalMap = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[normalMapParmIndex].id, normalTextureFileName, assetCacheFolderPath, true);
if (normalMap != null)
{
_material.SetTexture(HEU_Defines.UNITY_SHADER_BUMP_MAP, normalMap);
}
}
// Assign shader properties
// Clamp shininess to non-zero as results in very hard shadows. Unity's UI does not allow zero either.
float shininess;
if (!HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_ROUGH_ATTR, 0f, out shininess))
{
HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_ROUGH_ATTR, 0f, out shininess);
}
_material.SetFloat(HEU_Defines.UNITY_SHADER_SHININESS, Mathf.Max(0.03f, 1.0f - shininess));
Color diffuseColor;
if (!HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_DIFF_ATTR, Color.white, out diffuseColor))
{
HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_DIFF_ATTR, Color.white, out diffuseColor);
}
float alpha;
if (!HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_ALPHA_ATTR, 1f, out alpha))
{
HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_ALPHA_ATTR, 1f, out alpha);
}
diffuseColor.a = alpha;
_material.SetColor(HEU_Defines.UNITY_SHADER_COLOR, diffuseColor);
Color specular;
if (!HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_SPEC_ATTR, Color.black, out specular))
{
HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_SPEC_ATTR, Color.black, out specular);
}
_material.SetColor(HEU_Defines.UNITY_SHADER_SPEC_COLOR, specular);
return true;
}
/// <summary>
/// For this object's _material, we update the shader attributes and
/// fetch the textures from Houdini.
/// </summary>
/// <param name="materialInfo">This material's info from Houdini</param>
/// <param name="assetCacheFolderPath">Path to asset's cache folder</param>
public void UpdateMaterialFromHoudini(HAPI_MaterialInfo materialInfo, string assetCacheFolderPath)
{
HEU_SessionBase session = HEU_SessionManager.GetOrCreateDefaultSession();
HAPI_NodeInfo nodeInfo = new HAPI_NodeInfo();
if (!session.GetNodeInfo(materialInfo.nodeId, ref nodeInfo))
{
return;
}
// Get all parameters of this material
HAPI_ParmInfo[] parmInfos = new HAPI_ParmInfo[nodeInfo.parmCount];
if (!HEU_GeneralUtility.GetArray1Arg(materialInfo.nodeId, session.GetParams, parmInfos, 0, nodeInfo.parmCount))
{
return;
}
// Assign transparency shader or non-transparent.
if (IsTransparentMaterial(session, materialInfo.nodeId, parmInfos))
{
_material.shader = HEU_MaterialFactory.FindPluginShader(HEU_PluginSettings.DefaultTransparentShader);
}
else
{
_material.shader = HEU_MaterialFactory.FindPluginShader(HEU_PluginSettings.DefaultStandardShader);
}
// Diffuse texture - render & extract
int diffuseMapParmIndex = HEU_ParameterUtility.GetParameterIndexFromNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_TEX1_ATTR);
if (diffuseMapParmIndex < 0)
{
diffuseMapParmIndex = HEU_ParameterUtility.GetParameterIndexFromNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_BASECOLOR_ATTR);
if (diffuseMapParmIndex < 0)
{
diffuseMapParmIndex = HEU_ParameterUtility.GetParameterIndexFromNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_MAP_ATTR);
}
}
if (diffuseMapParmIndex >= 0 && diffuseMapParmIndex < parmInfos.Length)
{
string diffuseTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[diffuseMapParmIndex]);
_material.mainTexture = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[diffuseMapParmIndex].id, diffuseTextureFileName, assetCacheFolderPath);
}
// Normal map - render & extract texture
int normalMapParmIndex = HEU_ParameterUtility.GetParameterIndexFromNameOrTag(session, nodeInfo.id, parmInfos, HEU_Defines.MAT_OGL_NORMAL_ATTR);
if (normalMapParmIndex >= 0 && normalMapParmIndex < parmInfos.Length)
{
string normalTextureFileName = GetTextureFileNameFromMaterialParam(session, materialInfo.nodeId, parmInfos[normalMapParmIndex]);
Texture2D normalMap = HEU_MaterialFactory.RenderAndExtractImageToTexture(session, materialInfo, parmInfos[normalMapParmIndex].id, normalTextureFileName, assetCacheFolderPath);
if (normalMap != null)
{
_material.SetTexture(HEU_Defines.UNITY_SHADER_BUMP_MAP, normalMap);
}
}
// Assign shader properties
// Clamp shininess to non-zero as results in very hard shadows. Unity's UI does not allow zero either.
float shininess = HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_ROUGH_ATTR, 0f);
_material.SetFloat(HEU_Defines.UNITY_SHADER_SHININESS, Mathf.Max(0.03f, 1.0f - shininess));
Color diffuseColor = HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_DIFF_ATTR, Color.white);
diffuseColor.a = HEU_ParameterUtility.GetParameterFloatValue(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_ALPHA_ATTR, 1f);
_material.SetColor(HEU_Defines.UNITY_SHADER_COLOR, diffuseColor);
Color specular = HEU_ParameterUtility.GetParameterColor3Value(session, materialInfo.nodeId, parmInfos, HEU_Defines.MAT_OGL_SPEC_ATTR, Color.black);
_material.SetColor(HEU_Defines.UNITY_SHADER_SPECCOLOR, specular);
}
/// <summary>
/// Get the material info for the given material node ID.
/// </summary>
/// <param name="materialNodeID">The material node ID</param>
/// <param name="materialInfo">Material info to populate</param>
/// <returns>True if successfully returned the material info</returns>
public virtual bool GetMaterialInfo(HAPI_NodeId materialNodeID, ref HAPI_MaterialInfo materialInfo, bool bLogError = true)
{
return false;
}
// MATERIALS --------------------------------------------------------------------------------------------------
/// <summary>
/// Get the Material on the specified Part
/// </summary>
/// <param name="nodeID">The node ID</param>
/// <param name="partID">The part ID</param>
/// <param name="materialInfo">A valid material info to populate</param>
/// <returns>True if successfully queried the material info</returns>
public virtual bool GetMaterialOnPart(HAPI_NodeId nodeID, HAPI_PartId partID, ref HAPI_MaterialInfo materialInfo)
{
return false;
}
private static Texture2D ExtractHoudiniImageToTexturePNGJPEG(HEU_SessionBase session, HAPI_MaterialInfo materialInfo, string imagePlanes)
{
Texture2D textureResult = null;
HAPI_ImageInfo imageInfo = new HAPI_ImageInfo();
if (!session.GetImageInfo(materialInfo.nodeId, ref imageInfo))
{
return textureResult;
}
// This will return null if the current imageInfo file format is supported by Unity, otherwise
// returns a Unity supported file format.
string desiredFileFormatName = HEU_MaterialData.GetSupportedFileFormat(session, ref imageInfo);
imageInfo.gamma = HEU_PluginSettings.ImageGamma;
session.SetImageInfo(materialInfo.nodeId, ref imageInfo);
// Download the image into memory buffer
byte[] imageData = null;
if (!session.ExtractImageToMemory(materialInfo.nodeId, desiredFileFormatName, imagePlanes, out imageData))
{
return textureResult;
}
// Upload to Unity
textureResult = new Texture2D(1, 1);
textureResult.LoadImage(imageData);
return textureResult;
}
/// <summary>
/// Get the material info for the given material node ID.
/// </summary>
/// <param name="materialNodeID">The material node ID</param>
/// <param name="materialInfo">Material info to populate</param>
/// <returns>True if successfully returned the material info</returns>
public virtual bool GetMaterialInfo(HAPI_NodeId materialNodeID, ref HAPI_MaterialInfo materialInfo)
{
return false;
}
