SDK: Add try_cast for cObj derived types

praydog · Aug 18, 2022 · a0f7eb5 · a0f7eb5
1 parent 23852f7
commit a0f7eb5
Show file tree

Hide file tree

Showing 7 changed files with 58 additions and 0 deletions.
diff --git a/shared/sdk/Behavior.hpp b/shared/sdk/Behavior.hpp
@@ -15,6 +15,11 @@ namespace sdk {
 class Entity;
 
 class Behavior : public sdk::Obj {
+public:
+    static constexpr std::string_view class_name() {
+        return "class Behavior";
+    }
+
 public:
     virtual void pad4() = 0;
     virtual void pad5() = 0;

diff --git a/shared/sdk/BehaviorAppBase.hpp b/shared/sdk/BehaviorAppBase.hpp
@@ -8,6 +8,11 @@
 
 namespace sdk {
 class BehaviorAppBase : public sdk::Behavior {
+public:
+    static constexpr std::string_view class_name() {
+        return "class BehaviorAppBase";
+    }
+
 public:
     __forceinline ::regenny::BehaviorAppBase* regenny() const {
         return (::regenny::BehaviorAppBase*)this;

diff --git a/shared/sdk/EmBase.hpp b/shared/sdk/EmBase.hpp
@@ -6,6 +6,11 @@
 
 namespace sdk {
 class EmBase : public sdk::BehaviorAppBase {
+public:
+    static constexpr std::string_view class_name() {
+        return "class EmBase";
+    }
+
 public:
     OBJECT_SCRIPT_FUNCTION(EmBase, addExpDirect, void) // base + 0x574040
     OBJECT_SCRIPT_FUNCTION(EmBase, callEnemy, void) // base + 0x537010

diff --git a/shared/sdk/Model.hpp b/shared/sdk/Model.hpp
@@ -7,6 +7,11 @@
 namespace sdk {
 // cModel
 class Model : public sdk::Parts {
+public:
+    static constexpr std::string_view class_name() {
+        return "class cModel";
+    }
+
 public:
     virtual ~Model() = 0;
     virtual void model1(...) = 0;

diff --git a/shared/sdk/NPC.hpp b/shared/sdk/NPC.hpp
@@ -6,6 +6,11 @@
 
 namespace sdk {
 class NPC : public BehaviorAppBase {
+public:
+    static constexpr std::string_view class_name() {
+        return "class NPC";
+    }
+
 public:
     OBJECT_SCRIPT_FUNCTION(NPC, changeSetTypeFollow, void) // base + 0x7008f0
     OBJECT_SCRIPT_FUNCTION(NPC, changeSetTypeIdle, void) // base + 0x7009f0

diff --git a/shared/sdk/Obj.hpp b/shared/sdk/Obj.hpp
@@ -10,6 +10,11 @@
 
 namespace sdk {
 class Obj : public sdk::Model {
+public:
+    static constexpr std::string_view class_name() {
+        return "class cObj";
+    }
+
 public:
     // Vtable starts at index 3
     virtual NierRTTI2* get_rtti() = 0;
@@ -27,6 +32,21 @@ class Obj : public sdk::Model {
         return utility::rtti::derives_from(this, name);
     }
 
+    template<typename T>
+    bool is() const {
+        return utility::rtti::derives_from(this, T::class_name());
+    }
+
+    template<typename T>
+    T* try_cast() {
+        return this->is<T>() ? static_cast<T*>(this) : nullptr;
+    }
+
+    template<typename T>
+    const T* try_cast() const {
+        return this->is<const T>() ? static_cast<const T*>(this) : nullptr;
+    }
+
     bool is_pl0000() const {
         return is("class Pl0000");
     }
@@ -39,13 +59,22 @@ class Obj : public sdk::Model {
         return is("class BehaviorAppBase");
     }
 
+    bool is_em_base() const {
+        return is("class EmBase");
+    }
+
+    bool is_npc() const {
+        return is("class NPC");
+    }
+
     auto& obj_flags() {
         return regenny()->obj_flags;
     }
 
     auto& model_index() {
         return regenny()->model_index;
     }
+
 private:
     uint8_t detail_obj_data[sizeof(::regenny::Obj) - sizeof(Model)];
 };

diff --git a/shared/sdk/Pl0000.hpp b/shared/sdk/Pl0000.hpp
@@ -14,6 +14,10 @@
 namespace sdk {
 class Pl0000 : public sdk::BehaviorAppBase {
 public:
+    static constexpr std::string_view class_name() {
+        return "class Pl0000";
+    }
+
     enum EButtonIndex {
         INDEX_ATTACK_LIGHT = 0,
         INDEX_ATTACK_HEAVY = 1,