UI Editor: Enhance GameObject system and panel functionality

ui_editor/src/Core/GameObject.h

@@ -14,45 +14,10 @@ namespace UI {
 using EntityID = uint64_t;
 constexpr EntityID INVALID_ENTITY_ID = 0;
 
-class Component {
-public:
-    virtual ~Component() = default;
-    virtual std::string GetName() const = 0;
-    
-    virtual void Awake() {}
-    virtual void Start() {}
-    virtual void Update(float deltaTime) {}
-    virtual void OnDestroy() {}
-    
-    class Entity* GetEntity() const { return m_entity; }
-    bool IsEnabled() const { return m_enabled; }
-    void SetEnabled(bool enabled) { m_enabled = enabled; }
-    
-protected:
-    class Entity* m_entity = nullptr;
-    bool m_enabled = true;
-    
-    friend class Entity;
-};
+class Component;
+class TransformComponent;
 
-class TransformComponent : public Component {
-public:
-    float position[3] = {0.0f, 0.0f, 0.0f};
-    float rotation[3] = {0.0f, 0.0f, 0.0f};
-    float scale[3] = {1.0f, 1.0f, 1.0f};
-    
-    std::string GetName() const override { return "Transform"; }
-};
-
-class MeshRendererComponent : public Component {
-public:
-    std::string materialName = "Default-Material";
-    std::string meshName = "";
-    
-    std::string GetName() const override { return "Mesh Renderer"; }
-};
-
-class Entity {
+class GameObject {
 public:
     EntityID id = INVALID_ENTITY_ID;
     std::string name;
@@ -60,16 +25,16 @@ public:
     std::vector<EntityID> children;
     std::vector<std::unique_ptr<Component>> components;
     bool selected = false;
-    
+
     template<typename T, typename... Args>
     T* AddComponent(Args&&... args) {
         auto comp = std::make_unique<T>(std::forward<Args>(args)...);
-        comp->m_entity = this;
+        comp->m_gameObject = this;
         T* ptr = comp.get();
         components.push_back(std::move(comp));
         return ptr;
     }
-    
+
     template<typename T>
     T* GetComponent() {
         for (auto& comp : components) {
@@ -79,7 +44,7 @@ public:
         }
         return nullptr;
     }
-    
+
     template<typename T>
     std::vector<T*> GetComponents() {
         std::vector<T*> result;
@@ -90,6 +55,48 @@ public:
         }
         return result;
     }
+
+    TransformComponent* GetTransform() {
+        return GetComponent<TransformComponent>();
+    }
+};
+
+class Component {
+public:
+    virtual ~Component() = default;
+    virtual std::string GetName() const = 0;
+
+    virtual void Awake() {}
+    virtual void Start() {}
+    virtual void Update(float deltaTime) {}
+    virtual void OnDestroy() {}
+
+    GameObject* GetGameObject() const { return m_gameObject; }
+    bool IsEnabled() const { return m_enabled; }
+    void SetEnabled(bool enabled) { m_enabled = enabled; }
+
+protected:
+    GameObject* m_gameObject = nullptr;
+    bool m_enabled = true;
+
+    friend class GameObject;
+};
+
+class TransformComponent : public Component {
+public:
+    float position[3] = {0.0f, 0.0f, 0.0f};
+    float rotation[3] = {0.0f, 0.0f, 0.0f};
+    float scale[3] = {1.0f, 1.0f, 1.0f};
+
+    std::string GetName() const override { return "Transform"; }
+};
+
+class MeshRendererComponent : public Component {
+public:
+    std::string materialName = "Default-Material";
+    std::string meshName = "";
+
+    std::string GetName() const override { return "Mesh Renderer"; }
 };
 
 using ComponentInspectorFn = std::function<void(Component*)>;
@@ -105,7 +112,7 @@ public:
         static ComponentRegistry instance;
         return instance;
     }
-    
+
     template<typename T>
     void RegisterComponent(const std::string& name, ComponentInspectorFn inspectorFn) {
         m_inspectors[name] = {name, inspectorFn};
@@ -113,7 +120,7 @@ public:
             return std::make_unique<T>();
         };
     }
-    
+
     ComponentInspectorInfo* GetInspector(const std::string& name) {
         auto it = m_inspectors.find(name);
         if (it != m_inspectors.end()) {
@@ -121,7 +128,7 @@ public:
         }
         return nullptr;
     }
-    
+
 private:
     ComponentRegistry() = default;
     std::unordered_map<std::string, ComponentInspectorInfo> m_inspectors;