Allow adding enum options

psiberx · Nov 7, 2024 · ca8e2c4 · ca8e2c4
1 parent 972b3bf
commit ca8e2c4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/scripts/Reflection/ReflectionEnum.reds b/scripts/Reflection/ReflectionEnum.reds
@@ -1,6 +1,7 @@
 public native class ReflectionEnum extends ReflectionType {
     public native func GetConstants() -> array<ref<ReflectionConst>>
     public native func IsNative() -> Bool
+    public native func AddConstant(name: CName, value: Int64)
 }
 
 public native class ReflectionConst {

diff --git a/src/App/Reflection/ReflectionEnum.hpp b/src/App/Reflection/ReflectionEnum.hpp
@@ -61,6 +61,18 @@ struct ReflectionEnum : ReflectionType
         return !m_enum->flags.isScripted;
     }
 
+    void AddConstant(Red::CName aName, int64_t aValue) const
+    {
+        for (uint32_t i = 0; i != m_enum->valueList.size; ++i)
+        {
+            if (aValue == m_enum->valueList.entries[i] || aName == m_enum->hashList.entries[i])
+                return;
+        }
+
+        m_enum->valueList.PushBack(aValue);
+        m_enum->hashList.PushBack(aName);
+    }
+
     Red::CEnum* m_enum;
 
     RTTI_IMPL_TYPEINFO(App::ReflectionEnum);
@@ -77,4 +89,5 @@ RTTI_DEFINE_CLASS(App::ReflectionEnum, {
     RTTI_PARENT(App::ReflectionType);
     RTTI_METHOD(GetConstants);
     RTTI_METHOD(IsNative);
+    RTTI_METHOD(AddConstant);
 });