Add NodeRegistry.

Author: kovacsv

Sources/NodeUIEngine/NUIE_NodeRegistry.cpp

@@ -0,0 +1,129 @@
+#include "NUIE_NodeRegistry.hpp"
+
+namespace NUIE
+{
+
+NodeRegistry::NodeData::NodeData (NodeRegistryId id, const std::wstring& name, const CreatorFunction& creator) :
+	id (id),
+	name (name),
+	creator (creator)
+{
+}
+
+NodeRegistry::GroupData::GroupData (const std::wstring& name) :
+	name (name),
+	nodes (nodes)
+{
+}
+
+void NodeRegistry::GroupData::AddNode (NodeRegistryId id)
+{
+	nodes.push_back (id);
+}
+
+const std::wstring& NodeRegistry::GroupData::GetName () const
+{
+	return name;
+}
+
+const std::vector<NodeRegistryId>& NodeRegistry::GroupData::GetNodes () const
+{
+	return nodes;
+}
+
+NodeRegistryId NodeRegistry::NodeData::GetNodeId () const
+{
+	return id;
+}
+
+const std::wstring& NodeRegistry::NodeData::GetNodeName  () const
+{
+	return name;
+}
+
+const CreatorFunction & NodeRegistry::NodeData::GetCreatorFunction () const
+{
+	return creator;
+}
+
+NUIE::UINodePtr NodeRegistry::NodeData::CreateNode (const NUIE::Point& position) const
+{
+	return creator (position);
+}
+
+NodeRegistry::NodeRegistry () :
+	nextNodeId (0)
+{
+
+}
+
+void NodeRegistry::RegisterNode (const std::wstring& group, const std::wstring& name, const CreatorFunction& creator)
+{
+	auto foundGroup = nameToGroupData.find (group);
+	if (foundGroup == nameToGroupData.end ()) {
+		GroupData groupData (group);
+		nameToGroupData.insert ({ group, groupRegistry.size () });
+		groupRegistry.push_back (groupData);
+	}
+
+	NodeRegistryId id = nextNodeId++;
+	NodeData nodeData = {
+		id,
+		name,
+		creator
+	};
+
+	idToNodeData.insert ({ id, nodeRegistry.size () });
+	nameToNodeData.insert ({ name, nodeRegistry.size () });
+	groupRegistry[nameToGroupData[group]].AddNode (nodeRegistry.size ());
+	nodeRegistry.push_back (nodeData);
+}
+
+bool NodeRegistry::Contains (size_t id) const
+{
+	return GetNodeData (id) != nullptr;
+}
+
+bool NodeRegistry::Contains (const std::wstring& name) const
+{
+	return GetNodeData (name) != nullptr;
+}
+
+const NodeRegistry::NodeData* NodeRegistry::GetNodeData (size_t id) const
+{
+	auto found = idToNodeData.find (id);
+	if (found == idToNodeData.end ()) {
+		return nullptr;
+	}
+	return &nodeRegistry.at (found->second);
+}
+
+const NodeRegistry::NodeData* NodeRegistry::GetNodeData (const std::wstring& name) const
+{
+	auto found = nameToNodeData.find (name);
+	if (found == nameToNodeData.end ()) {
+		return nullptr;
+	}
+	return &nodeRegistry.at (found->second);
+}
+
+void NodeRegistry::EnumerateGroups (const std::function<void (const std::wstring&)>& processor) const
+{
+	for (const GroupData& group : groupRegistry) {
+		processor (group.GetName ());
+	}
+}
+
+void NodeRegistry::EnumerateGroupNodes (const std::wstring& groupName, const std::function<void (const NodeData&)>& processor) const
+{
+	auto found = nameToGroupData.find (groupName);
+	if (found == nameToGroupData.end ()) {
+		return;
+	}
+	const GroupData& groupData = groupRegistry[found->second];
+	for (NodeRegistryId nodeId : groupData.GetNodes ()) {
+		processor (nodeRegistry[nodeId]);
+	}
+}
+
+}

Sources/NodeUIEngine/NUIE_NodeRegistry.hpp

@@ -0,0 +1,69 @@
+#ifndef NUIE_NODEREGISTRY_HPP
+#define NUIE_NODEREGISTRY_HPP
+
+#include "NUIE_UINode.hpp"
+
+namespace NUIE
+{
+
+using NodeRegistryId = size_t;
+using CreatorFunction = std::function<NUIE::UINodePtr (const NUIE::Point& position)>;
+
+class NodeRegistry
+{
+public:
+	class NodeData
+	{
+	public:
+		NodeData (NodeRegistryId id, const std::wstring& name, const CreatorFunction& creator);
+
+		NodeRegistryId			GetNodeId () const;
+		const std::wstring&		GetNodeName  () const;
+		const CreatorFunction&	GetCreatorFunction () const;
+		NUIE::UINodePtr			CreateNode (const NUIE::Point& position) const;
+
+	private:
+		NodeRegistryId		id;
+		std::wstring		name;
+		CreatorFunction		creator;
+	};
+
+	class GroupData
+	{
+	public:
+		GroupData (const std::wstring& name);
+
+		void								AddNode (NodeRegistryId id);
+		const std::wstring&					GetName () const;
+		const std::vector<NodeRegistryId>&	GetNodes () const;
+
+	private:
+		std::wstring					name;
+		std::vector<NodeRegistryId>		nodes;
+	};
+
+	NodeRegistry ();
+
+	void				RegisterNode (const std::wstring& group, const std::wstring& name, const CreatorFunction& creator);
+
+	bool				Contains (size_t id) const;
+	bool				Contains (const std::wstring& name) const;
+
+	const NodeData*		GetNodeData (size_t id) const;
+	const NodeData*		GetNodeData (const std::wstring& name) const;
+
+	void				EnumerateGroups (const std::function<void (const std::wstring&)>& processor) const;
+	void				EnumerateGroupNodes (const std::wstring& groupName, const std::function<void (const NodeData&)>& processor) const;
+
+private:
+	std::vector<GroupData>						groupRegistry;
+	std::vector<NodeData>						nodeRegistry;
+	std::unordered_map<NodeRegistryId, size_t>	idToNodeData;
+	std::unordered_map<std::wstring, size_t>	nameToNodeData;
+	std::unordered_map<std::wstring, size_t>	nameToGroupData;
+	NodeRegistryId								nextNodeId;
+};
+
+}
+
+#endif