1fb83c551a
Use BLI_uniquename_cb to check and fix naming conflicts, and keep track of the potentially modified node names in a map. Reserve the material name, so that later the output node can be renamed to it without conflicts. Also make sure there are no conflicts with names auto generated by MaterialX, by always naming nodes ourselves. Pull Request: https://projects.blender.org/blender/blender/pulls/133234
94 lines
2.9 KiB
C++
94 lines
2.9 KiB
C++
/* SPDX-FileCopyrightText: 2011-2022 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#include <MaterialXFormat/XmlIo.h>
|
|
|
|
#include "node_parser.h"
|
|
|
|
#include "DEG_depsgraph.hh"
|
|
|
|
#include "DNA_material_types.h"
|
|
|
|
#include "NOD_shader.h"
|
|
|
|
#include "material.h"
|
|
|
|
namespace blender::nodes::materialx {
|
|
|
|
class DefaultMaterialNodeParser : public NodeParser {
|
|
public:
|
|
using NodeParser::NodeParser;
|
|
|
|
NodeItem compute() override
|
|
{
|
|
const Material *material = graph_.material;
|
|
NodeItem surface = create_node(
|
|
"standard_surface",
|
|
NodeItem::Type::SurfaceShader,
|
|
{{"base", val(1.0f)},
|
|
{"base_color", val(MaterialX::Color3(material->r, material->g, material->b))},
|
|
{"diffuse_roughness", val(material->roughness)},
|
|
{"specular", val(material->spec)},
|
|
{"metalness", val(material->metallic)}});
|
|
|
|
NodeItem res = create_node(
|
|
"surfacematerial", NodeItem::Type::Material, {{"surfaceshader", surface}});
|
|
return res;
|
|
}
|
|
|
|
NodeItem compute_error()
|
|
{
|
|
NodeItem surface = create_node("standard_surface",
|
|
NodeItem::Type::SurfaceShader,
|
|
{{"base_color", val(MaterialX::Color3(1.0f, 0.0f, 1.0f))}});
|
|
NodeItem res = create_node(
|
|
"surfacematerial", NodeItem::Type::Material, {{"surfaceshader", surface}});
|
|
return res;
|
|
}
|
|
};
|
|
|
|
MaterialX::DocumentPtr export_to_materialx(Depsgraph *depsgraph,
|
|
Material *material,
|
|
const ExportParams &export_params)
|
|
{
|
|
CLOG_INFO(LOG_MATERIALX_SHADER, 0, "Material: %s", material->id.name);
|
|
|
|
MaterialX::DocumentPtr doc = MaterialX::createDocument();
|
|
NodeItem output_item;
|
|
|
|
NodeGraph graph(depsgraph, material, export_params, doc);
|
|
|
|
if (material->use_nodes) {
|
|
material->nodetree->ensure_topology_cache();
|
|
bNode *output_node = ntreeShaderOutputNode(material->nodetree, SHD_OUTPUT_ALL);
|
|
if (output_node && output_node->typeinfo->materialx_fn) {
|
|
NodeParserData data = {graph, NodeItem::Type::Material, nullptr, graph.empty_node()};
|
|
output_node->typeinfo->materialx_fn(&data, output_node, nullptr);
|
|
output_item = data.result;
|
|
}
|
|
else {
|
|
output_item = DefaultMaterialNodeParser(
|
|
graph, nullptr, nullptr, NodeItem::Type::Material, nullptr)
|
|
.compute_error();
|
|
}
|
|
}
|
|
else {
|
|
output_item = DefaultMaterialNodeParser(
|
|
graph, nullptr, nullptr, NodeItem::Type::Material, nullptr)
|
|
.compute();
|
|
}
|
|
|
|
/* This node is expected to have a specific name to link up to USD. */
|
|
graph.set_output_node_name(output_item);
|
|
|
|
CLOG_INFO(LOG_MATERIALX_SHADER,
|
|
1,
|
|
"Material: %s\n%s",
|
|
material->id.name,
|
|
MaterialX::writeToXmlString(doc).c_str());
|
|
return doc;
|
|
}
|
|
|
|
} // namespace blender::nodes::materialx
|