Files
Sourcetrail/src/lib/utility/scheduling/TaskGroupSelector.cpp
2019-11-18 17:07:20 +01:00

62 lines
1.2 KiB
C++

#include "TaskGroupSelector.h"
TaskGroupSelector::TaskGroupSelector(): m_taskIndex(0) {}
void TaskGroupSelector::addTask(std::shared_ptr<Task> task)
{
m_taskRunners.push_back(std::make_shared<TaskRunner>(task));
}
void TaskGroupSelector::doEnter(std::shared_ptr<Blackboard> blackboard)
{
m_taskIndex = 0;
}
Task::TaskState TaskGroupSelector::doUpdate(std::shared_ptr<Blackboard> blackboard)
{
if (m_taskIndex >= int(m_taskRunners.size()))
{
return STATE_FAILURE;
}
TaskState state = m_taskRunners[m_taskIndex]->update(blackboard);
if (state == STATE_FAILURE)
{
m_taskIndex++;
if (m_taskIndex >= int(m_taskRunners.size()))
{
return STATE_FAILURE;
}
}
else if (state == STATE_SUCCESS)
{
return STATE_SUCCESS;
}
else if (state == STATE_HOLD)
{
return STATE_HOLD;
}
return STATE_RUNNING;
}
void TaskGroupSelector::doExit(std::shared_ptr<Blackboard> blackboard) {}
void TaskGroupSelector::doReset(std::shared_ptr<Blackboard> blackboard)
{
for (size_t i = 0; i < m_taskRunners.size(); i++)
{
m_taskRunners[i]->reset();
}
}
void TaskGroupSelector::doTerminate()
{
for (size_t i = 0; i < m_taskRunners.size(); i++)
{
m_taskRunners[i]->terminate();
}
}