a6108f69f2
With chains of nodes with the same name. Pull Request: https://projects.blender.org/blender/blender/pulls/145800
70 lines
1.8 KiB
C++
70 lines
1.8 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_set.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, StringRef child_name);
|
|
|
|
NodeItem empty_node() const;
|
|
NodeItem get_node(StringRef name) const;
|
|
NodeItem get_output(StringRef name) const;
|
|
NodeItem get_input(StringRef name) const;
|
|
|
|
std::string unique_node_name(const bNode *node,
|
|
StringRef socket_out_name,
|
|
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_;
|
|
Set<std::string> used_node_names_;
|
|
std::string node_name_prefix_;
|
|
};
|
|
|
|
} // namespace blender::nodes::materialx
|