Files
FlaxEngine/Source/Engine/Threading/MainThreadTask.h
2021-06-14 16:10:08 +02:00

121 lines
3.2 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;
private:
static void RunAll(float dt);
public:
/// <summary>
/// The initial time delay (in seconds) before task execution. Use 0 to skip this feature.
/// </summary>
float InitialDelay = 0.0f;
public:
// [Task]
String ToString() const override;
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;
public:
// [MainThreadTask]
bool HasReference(Object* obj) const override;
};