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
68 lines
1.7 KiB
C++
68 lines
1.7 KiB
C++
/* SPDX-FileCopyrightText: 2011-2025 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
|
|
#include <MaterialXCore/Document.h>
|
|
|
|
#include "material.h"
|
|
#include "node_item.h"
|
|
|
|
#include "BLI_map.hh"
|
|
#include "BLI_string_ref.hh"
|
|
|
|
struct bNode;
|
|
struct Material;
|
|
struct Depsgraph;
|
|
|
|
namespace blender::nodes::materialx {
|
|
|
|
/*
|
|
* Wrapper around MaterialX graph.
|
|
*/
|
|
struct NodeGraph {
|
|
public:
|
|
const Depsgraph *depsgraph = nullptr;
|
|
const Material *material = nullptr;
|
|
const ExportParams &export_params;
|
|
|
|
NodeGraph(const Depsgraph *depsgraph,
|
|
const Material *material,
|
|
const ExportParams &export_params,
|
|
const MaterialX::DocumentPtr &document);
|
|
NodeGraph(const NodeGraph &parent, const StringRef child_name);
|
|
|
|
NodeItem empty_node() const;
|
|
NodeItem get_node(const StringRef name) const;
|
|
NodeItem get_output(const StringRef name) const;
|
|
NodeItem get_input(const StringRef name) const;
|
|
|
|
std::string unique_node_name(const bNode *node,
|
|
const StringRef socket_out_name,
|
|
const NodeItem::Type to_type);
|
|
void set_output_node_name(const NodeItem &item) const;
|
|
|
|
static std::string unique_anonymous_node_name(MaterialX::GraphElement *graph_element);
|
|
|
|
protected:
|
|
struct NodeKey {
|
|
const bNode *node;
|
|
std::string socket_name;
|
|
NodeItem::Type to_type;
|
|
MaterialX::GraphElement *graph_element;
|
|
|
|
uint64_t hash() const;
|
|
bool operator==(const NodeKey &other) const;
|
|
};
|
|
|
|
MaterialX::GraphElement *graph_element_ = nullptr;
|
|
Map<NodeKey, const std::string> root_key_to_name_map_;
|
|
Map<NodeKey, const std::string> &key_to_name_map_;
|
|
std::string node_name_prefix_;
|
|
};
|
|
|
|
} // namespace blender::nodes::materialx
|