DebugSolverTasks.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "Chaos/Framework/DebugSubstep.h"
#include "Templates/Function.h"
#include "Chaos/Declares.h"
 
#if CHAOS_DEBUG_SUBSTEP
#include "Async/AsyncWork.h"
 
namespace Chaos
{
class FPhysicsSolverBase;
    class FDebugSolverTask final : public FNonAbandonableTask
    {
        friend class FAsyncTask<FDebugSolverTask>;
 
    public:
        FDebugSolverTask(TFunction<void()> InStepFunction, FDebugSubstep& InDebugSubstep);
        ~FDebugSolverTask() {}
 
        void DoWork();
 
    private:
        FDebugSolverTask(const FDebugSolverTask&) = delete;
        FDebugSolverTask& operator=(const FDebugSolverTask&) = delete;
 
        FORCEINLINE TStatId GetStatId() const
        {
            RETURN_QUICK_DECLARE_CYCLE_STAT(FDebugSolverTask, STATGROUP_ThreadPoolAsyncTasks);
        }
 
    private:
        TFunction<void()> StepFunction;
        FDebugSubstep& DebugSubstep;
    };
 
    class FDebugSolverTasks final
    {
    public:
        FDebugSolverTasks() : SolverToTaskMap() {}
        ~FDebugSolverTasks() { Shutdown(); }
 
        void Add(FPhysicsSolverBase* Solver);
 
        void Remove(FPhysicsSolverBase* Solver);
 
        void DebugStep(FPhysicsSolverBase* Solver, TFunction<void()> StepFunction);
 
        void Shutdown();
 
    private:
        TMap<FPhysicsSolverBase*, FAsyncTask<FDebugSolverTask>*> SolverToTaskMap;
    };
}  // namespace Chaos
 
#else  // #if CHAOS_DEBUG_SUBSTEP
 
namespace Chaos
{
class FPhysicsSolverBase;
 
    class FDebugSolverTasks final
    {
    public:
        FDebugSolverTasks() {}
        ~FDebugSolverTasks() {}
 
        void Add(FPhysicsSolverBase* /*Solver*/) {}
        void Remove(FPhysicsSolverBase* /*Solver*/) {}
 
        FORCEINLINE void DebugStep(FPhysicsSolverBase* /*Solver*/, TFunction<void()> StepFunction) { StepFunction(); }
 
        void Shutdown() {}
    };
}  // namespace Chaos
 
#endif  // #if CHAOS_DEBUG_SUBSTEP #else