TaskSchedulerTestTasks.cpp

#include "TaskSchedulerTestTasks.h"

#include <sofa/simulation/TaskScheduler.h>

using sofa::simulation::Task;

namespace sofa
{

    
    Task::MemoryAlloc FibonacciTask::run()
    {
        if (_N < 2)
        {
            *_sum = _N;
            return MemoryAlloc::Stack;
        }
        
        simulation::CpuTask::Status status;
        
        int64_t x, y;
        
        simulation::TaskScheduler* scheduler = simulation::TaskScheduler::getInstance();
        
        FibonacciTask task0(_N - 1, &x, &status);
        FibonacciTask task1(_N - 2, &y, &status);
        
        scheduler->addTask(&task0);
        scheduler->addTask(&task1);
        scheduler->workUntilDone(&status);
        
        // Do the sum
        *_sum = x + y;
        
        return MemoryAlloc::Stack;
    }
    
    
    
    Task::MemoryAlloc IntSumTask::run()
    {
        const int64_t count = _last - _first;
        if (count < 1)
        {
            *_sum = _first;
            return MemoryAlloc::Stack;
        }
        
        const int64_t mid = _first + (count / 2);
        
        simulation::CpuTask::Status status;
        
        int64_t x, y;
        
        simulation::TaskScheduler* scheduler = simulation::TaskScheduler::getInstance();
        
        IntSumTask task0(_first, mid, &x, &status);
        IntSumTask task1(mid+1, _last, &y, &status);
        
        scheduler->addTask(&task0);
        scheduler->addTask(&task1);
        scheduler->workUntilDone(&status);
        
        // Do the sum
        *_sum = x + y;
        
        
        return MemoryAlloc::Stack;
    }


    Task::MemoryAlloc SleepTask::run()
    {
        using namespace std::chrono_literals;
        std::this_thread::sleep_for(20ms);
        m_isTaskDone = true;
        return Task::Stack;
    }

    Task::MemoryAlloc ThreadIdTask::run()
    {
        m_isTaskDone = true;
        using namespace std::chrono_literals;
        std::this_thread::sleep_for(20ms); //add sleep period to simulate the worker thread is busy
        *m_threadId = std::this_thread::get_id();
        return Task::Stack;
    }
} // namespace sofa