From f1059f98169bffa12f501f8bfac1eff710f7efad Mon Sep 17 00:00:00 2001
From: Yoshi Askharoun <jjask7@gmail.com>
Date: Sat, 10 Feb 2024 19:35:14 -0600
Subject: [PATCH] Add MarkupType overload to MarkupTypeSchema.Build

---
 UIX/Microsoft/Iris/Markup/MarkupTypeSchema.cs | 21 ++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/UIX/Microsoft/Iris/Markup/MarkupTypeSchema.cs b/UIX/Microsoft/Iris/Markup/MarkupTypeSchema.cs
index 042c652..3b79b25 100644
--- a/UIX/Microsoft/Iris/Markup/MarkupTypeSchema.cs
+++ b/UIX/Microsoft/Iris/Markup/MarkupTypeSchema.cs
@@ -52,7 +52,26 @@ namespace Microsoft.Iris.Markup
                 return new EffectClassTypeSchema(owner, name);
             if (markupTypeDefinition == DataTypeSchema.Type)
                 return new MarkupDataTypeSchema(owner, name);
-            return markupTypeDefinition == DataQuerySchema.Type ? new MarkupDataQuerySchema(owner, name) : null;
+            if (markupTypeDefinition == DataQuerySchema.Type)
+                return new MarkupDataQuerySchema(owner, name);
+            
+            return null;
+        }
+
+        public static MarkupTypeSchema Build(MarkupType markupType, MarkupLoadResult owner, string name)
+        {
+            if (markupType == MarkupType.Class)
+                return new ClassTypeSchema(owner, name);
+            if (markupType == MarkupType.UI)
+                return new UIClassTypeSchema(owner, name);
+            if (markupType == MarkupType.Effect)
+                return new EffectClassTypeSchema(owner, name);
+            if (markupType == MarkupType.DataType)
+                return new MarkupDataTypeSchema(owner, name);
+            if (markupType == MarkupType.DataQuery)
+                return new MarkupDataQuerySchema(owner, name);
+            
+            return null;
         }
 
         public MarkupTypeSchema(MarkupLoadResult owner, string name)