VVMTaskGroup.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#if WITH_VERSE_VM || defined(__INTELLISENSE__)
 
#include "AutoRTFM.h"
#include "Containers/Set.h"
#include "VerseVM/VVMCell.h"
#include "VerseVM/VVMContext.h"
#include "VerseVM/VVMMutableArray.h"
#include "VerseVM/VVMTask.h"
#include "VerseVM/VVMWeakBarrier.h"
 
namespace Verse
{
 
struct VTaskGroup : VCell
{
    DECLARE_DERIVED_VCPPCLASSINFO(COREUOBJECT_API, VCell);
    COREUOBJECT_API static TGlobalTrivialEmergentTypePtr<&StaticCppClassInfo> GlobalTrivialEmergentType;
 
    AUTORTFM_OPEN static VTaskGroup& New(FAllocationContext Context)
    {
        // We use `DestructorAndCensusSpace` because of ActiveTasks. Its elements are weak and must participate in census, and TSet destruction must occur.
        // VTaskGroup::New is safe to call from the closed; if an abort occurs, the allocated VTaskGroup can be cleaned up by garbage collection.
        return *new (Context.Allocate(FHeap::DestructorAndCensusSpace, sizeof(VTaskGroup))) VTaskGroup(Context);
    }
 
    AUTORTFM_OPEN COREUOBJECT_API bool HasActiveTasks() const;
 
    AUTORTFM_OPEN COREUOBJECT_API int64 GetNumActive() const;
 
    AUTORTFM_DISABLE COREUOBJECT_API bool AddTask(FAllocationContext, VTask& Task);
 
    AUTORTFM_DISABLE COREUOBJECT_API bool RemoveTask(FAllocationContext, VTask& Task);
 
    /*AUTORTFM_DISABLE*/ COREUOBJECT_API bool AddTaskTransactionally(FAllocationContext, VTask& Task);
 
    /*AUTORTFM_DISABLE*/ COREUOBJECT_API bool RemoveTaskTransactionally(FAllocationContext, VTask& Task);
 
    AUTORTFM_DISABLE COREUOBJECT_API void TerminateAll(FAllocationContext Context);
 
private:
    AUTORTFM_OPEN VTaskGroup(FAllocationContext Context)
        : VCell(Context, &GlobalTrivialEmergentType.Get(Context))
    {
    }
 
    AUTORTFM_DISABLE VTask* GetAnyTask();
 
    AUTORTFM_DISABLE void ConductCensusImpl();
 
    TSet<TWeakBarrier<VTask>> ActiveTasks;
};
 
} // namespace Verse
 
#endif // WITH_VERSE_VM