ksys/act: Add BaseProcCreateTaskSelector

src/KingSystem/ActorSystem/CMakeLists.txt

@@ -54,6 +54,8 @@ target_sources(uking PRIVATE
   actBaseProc.h
   actBaseProcCreateTask.cpp
   actBaseProcCreateTask.h
+  actBaseProcCreateTaskSelector.cpp
+  actBaseProcCreateTaskSelector.h
   actBaseProcDeleter.cpp
   actBaseProcDeleter.h
   actBaseProcHandle.cpp

src/KingSystem/ActorSystem/actBaseProcCreateTask.h

@@ -74,6 +74,8 @@ protected:
     void prepareImpl_(util::TaskRequest* req) override;
 
 private:
+    friend class BaseProcCreateTaskSelector;
+
     bool onTaskDelegateInvoked(void* arg);
     void doPrepare(const BaseProcCreateTaskData* data);
 

src/KingSystem/ActorSystem/actBaseProcCreateTaskSelector.cpp

@@ -0,0 +1,45 @@
+#include "KingSystem/ActorSystem/actBaseProcCreateTaskSelector.h"
+#include <limits>
+#include "KingSystem/ActorSystem/actBaseProcCreateTask.h"
+#include "KingSystem/Map/mapObject.h"
+
+namespace ksys::act {
+
+SEAD_SINGLETON_DISPOSER_IMPL(BaseProcCreateTaskSelector)
+
+util::Task* BaseProcCreateTaskSelector::selectTask(const util::TaskSelectionContext& context) {
+    util::Task* min_task = nullptr;
+    util::Task* lane1_task = nullptr;
+    auto min = std::numeric_limits<float>::max();
+
+    for (auto& it : context) {
+        auto* task = sead::DynamicCast<BaseProcCreateTask>(&it);
+        if (!task)
+            return &it;
+
+        if (task->getLaneId() == 2)
+            return &it;
+
+        if (task->mMapObject && task->mMapObject->getFlags0().isOff(map::Object::Flag0::_4))
+            return &it;
+
+        if (task->mDistanceToLoadSphere >= 0.0 && min > task->mDistanceToLoadSphere)
+            min_task = &it;
+
+        if (task->mDistanceToLoadSphere >= 0.0 && min > task->mDistanceToLoadSphere)
+            min = task->mDistanceToLoadSphere;
+
+        if (task->getLaneId() == 1)
+            lane1_task = &it;
+    }
+
+    if (min_task)
+        return min_task;
+
+    if (lane1_task)
+        return lane1_task;
+
+    return std::addressof(*context.begin());
+}
+
+}  // namespace ksys::act

src/KingSystem/ActorSystem/actBaseProcCreateTaskSelector.h

@@ -0,0 +1,25 @@
+#pragma once
+
+#include <heap/seadDisposer.h>
+#include <prim/seadDelegate.h>
+#include "KingSystem/Utils/Thread/TaskQueueBase.h"
+#include "KingSystem/Utils/Types.h"
+
+namespace ksys::act {
+
+class BaseProcCreateTaskSelector {
+    SEAD_SINGLETON_DISPOSER(BaseProcCreateTaskSelector)
+    BaseProcCreateTaskSelector() = default;
+
+public:
+    auto& getDelegate() { return mDelegate; }
+
+private:
+    util::Task* selectTask(const util::TaskSelectionContext& context);
+
+    util::TaskSelectionDelegateT<BaseProcCreateTaskSelector> mDelegate{
+        this, &BaseProcCreateTaskSelector::selectTask};
+};
+KSYS_CHECK_SIZE_NX150(BaseProcCreateTaskSelector, 0x40);
+
+}  // namespace ksys::act

src/KingSystem/Utils/Thread/TaskQueueBase.h

@@ -26,6 +26,8 @@ struct TaskSelectionContext {
 };
 
 using TaskSelectionDelegate = sead::IDelegate1R<const TaskSelectionContext&, Task*>;
+template <typename T>
+using TaskSelectionDelegateT = sead::Delegate1R<T, const TaskSelectionContext&, Task*>;
 
 class TaskQueueBase {
     SEAD_RTTI_BASE(TaskQueueBase)