Skip to content
Snippets Groups Projects
Commit 1e4532cd authored by Axel Farrugia's avatar Axel Farrugia
Browse files

Fixed UpdateNodeNamex

parent acd012c0
No related branches found
No related tags found
No related merge requests found
......@@ -406,16 +406,19 @@ public:
addChild(toOtherNode, mNodeRegistry.at(fromOutNodeName), fromTensor, toTensor);
}
inline void updateNodeName(NodePtr nodeToRename, const std::string& newName){
const std::string& oldName = nodeToRename->name();
AIDGE_ASSERT(mNodeRegistry.find(newName) != mNodeRegistry.end(), "Name {} is already used in graph {}.", newName, name());
if (nodeToRename->name() != ""){ // Case node already had a name
AIDGE_ASSERT(mNodeRegistry.find(oldName) != mNodeRegistry.end(), "No node named {} in graph {}, the graph may be corrupted !", oldName, name());
mNodeRegistry[newName] = mNodeRegistry[oldName];
mNodeRegistry.erase(oldName);
}else{ // Case node did not had a name
mNodeRegistry[newName] = nodeToRename;
inline void updateNodeName(const std::shared_ptr<Node>& node, const std::string& newName){
if (!newName.empty()) {
auto itNew = mNodeRegistry.insert(std::make_pair(newName, node));
if (!itNew.second) {
Log::notice("Replacing existing node name in graph node name registry: {}", newName);
(itNew.first)->second = node;
}
}
if (!node->name().empty()) {
const auto it = mNodeRegistry.find(node->name());
AIDGE_ASSERT(it != mNodeRegistry.end(), "No node named {} in graph {}, the graph may be corrupted !", node->name(), name());
mNodeRegistry.erase(it);
}
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment