You're breathtaking!

Source/Engine/Threading/MainThreadTask.h

@@ -0,0 +1,153 @@
+// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
+
+#pragma once
+
+#include "Task.h"
+#include "Engine/Core/Types/String.h"
+
+// Invokes a target method on a main thread (using task or directly if already on main thread)
+// Example: INVOKE_ON_MAIN_THREAD(Collector, Collector::SyncData, this);
+#define INVOKE_ON_MAIN_THREAD(targetType, targetMethod, targetObject) \
+    if (IsInMainThread()) \
+	{ \
+		targetObject->targetMethod(); \
+	} else { \
+		Function<void()> action; \
+		action.Bind<targetType, &targetMethod>(targetObject); \
+		Task::StartNew(New<MainThreadActionTask>(action))->Wait(); \
+	}
+
+/// <summary>
+/// General purpose task executed on Main Thread in the beginning of the next frame.
+/// </summary>
+/// <seealso cref="Task" />
+class FLAXENGINE_API MainThreadTask : public Task
+{
+    friend class Engine;
+
+protected:
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MainThreadTask"/> class.
+    /// </summary>
+    MainThreadTask()
+        : Task()
+    {
+    }
+
+private:
+
+    /// <summary>
+    /// Runs all main thread tasks. Called only by the Engine class.
+    /// </summary>
+    static void RunAll();
+
+public:
+
+    // [Task]
+    String ToString() const override
+    {
+        return String::Format(TEXT("Main Thread Task ({0})"), ::ToString(GetState()));
+    }
+
+    bool HasReference(Object* obj) const override
+    {
+        return false;
+    }
+
+protected:
+
+    // [Task]
+    void Enqueue() override;
+};
+
+/// <summary>
+/// General purpose task executing custom action using Main Thread in the beginning of the next frame.
+/// </summary>
+/// <seealso cref="MainThreadTask" />
+/// <seealso cref="Task" />
+class FLAXENGINE_API MainThreadActionTask : public MainThreadTask
+{
+protected:
+
+    Function<void()> _action1;
+    Function<bool()> _action2;
+    Object* _target;
+
+public:
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MainThreadActionTask"/> class.
+    /// </summary>
+    /// <param name="action">The action.</param>
+    /// <param name="target">The target object.</param>
+    MainThreadActionTask(Function<void()>& action, Object* target = nullptr)
+        : MainThreadTask()
+        , _action1(action)
+        , _target(target)
+    {
+    }
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MainThreadActionTask"/> class.
+    /// </summary>
+    /// <param name="action">The action.</param>
+    /// <param name="target">The target object.</param>
+    MainThreadActionTask(Function<void()>::Signature action, Object* target = nullptr)
+        : MainThreadTask()
+        , _action1(action)
+        , _target(target)
+    {
+    }
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MainThreadActionTask"/> class.
+    /// </summary>
+    /// <param name="action">The action.</param>
+    /// <param name="target">The target object.</param>
+    MainThreadActionTask(Function<bool()>& action, Object* target = nullptr)
+        : MainThreadTask()
+        , _action2(action)
+        , _target(target)
+    {
+    }
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="MainThreadActionTask"/> class.
+    /// </summary>
+    /// <param name="action">The action.</param>
+    /// <param name="target">The target object.</param>
+    MainThreadActionTask(Function<bool()>::Signature action, Object* target = nullptr)
+        : MainThreadTask()
+        , _action2(action)
+        , _target(target)
+    {
+    }
+
+protected:
+
+    // [MainThreadTask]
+    bool Run() override
+    {
+        if (_action1.IsBinded())
+        {
+            _action1();
+            return false;
+        }
+
+        if (_action2.IsBinded())
+        {
+            return _action2();
+        }
+
+        return true;
+    }
+
+public:
+
+    // [MainThreadTask]
+    bool HasReference(Object* obj) const override
+    {
+        return obj == _target;
+    }
+};