From d3a9d7565b18404b2042c9dc025cc690250cb50c Mon Sep 17 00:00:00 2001
From: rtk0c <mail.tianyig@gmail.com>
Date: Sun, 13 Jun 2021 17:16:46 -0700
Subject: Move UI_Workflows.cpp > WorkflowCategory to be a part of
 WorkflowNode's RTTI

(cherry picked from commit f45084562cfc95392b4dfb968dd215c25be3fc67)
---
 core/src/Model/Workflow/Workflow.hpp | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'core/src/Model/Workflow/Workflow.hpp')

diff --git a/core/src/Model/Workflow/Workflow.hpp b/core/src/Model/Workflow/Workflow.hpp
index 9c809bf..cba0d72 100644
--- a/core/src/Model/Workflow/Workflow.hpp
+++ b/core/src/Model/Workflow/Workflow.hpp
@@ -7,6 +7,7 @@
 
 #include <imgui_node_editor.h>
 #include <cstddef>
+#include <span>
 #include <cstdint>
 #include <filesystem>
 #include <functional>
@@ -72,6 +73,19 @@ public:
 		KindCount = InvalidKind,
 	};
 
+	enum Category
+	{
+		CG_Numeric,
+		CG_Text,
+		CG_Document,
+		CG_UserInput,
+		CG_SystemInput,
+		CG_Output,
+
+		InvalidCategory,
+		CategoryCount = InvalidCategory,
+	};
+
 	struct InputPin
 	{
 		uint32_t Connection = WorkflowConnection::kInvalidId;
@@ -108,7 +122,10 @@ protected:
 
 public:
 	static const char* FormatKind(Kind kind);
+	static const char* FormatCategory(Category category);
 	static const char* FormatType(Type type);
+	static Category QueryCategory(Kind kind);
+	static std::span<const Kind> QueryCategoryMembers(Category category);
 	static std::unique_ptr<WorkflowNode> CreateByKind(Kind kind);
 
 	static bool IsInstance(const WorkflowNode* node);
-- 
cgit v1.2.3-70-g09d2