154 lines
3.8 KiB
C++
154 lines
3.8 KiB
C++
// Copyright (c) 2012-2021 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;
|
|
}
|
|
};
|