AuroraRuntime/Source/Async/GroupState.hpp

/***
    Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: GroupState.hpp
    Date: 2021-11-1
    Author: Reece
***/
#pragma once
#include "ThreadState.hpp"

namespace Aurora::Async
{
    struct ThreadPool;
    struct GroupWorkQueue
    {
        AuThreadPrimitives::Mutex mutex;
        AuUInt32 uItems {};
        AuList<WorkEntry_t> sortedWork[AuAsync::kEWorkPrioCount];

        bool IsEmpty();
        bool IsEmpty(ThreadPool *pPool, AuWorkerId_t id);

        void AddWorkEntry(ThreadState *pState, WorkEntry_t entry);

        void Dequeue(AuList<WorkEntry_t> &queue, int maxPopCount, AuAsync::ThreadId_t idd);
    };

    struct GroupState
    {
        ThreadGroup_t group;

        GroupWorkQueue workQueue;

        AuThreadPrimitives::Mutex workersMutex;
        AuBST<ThreadId_t, AuSPtr<ThreadState>> workers;
        AuWPtr<ThreadState> wpWorkers[32];

        bool Init();

        void BroadCast();

        void AddWorker(ThreadId_t id, AuSPtr<ThreadState> pState);

        AuSPtr<ThreadState> GetThreadByIndex(ThreadId_t uIndex);

        bool inline IsSysThread()
        {
            return group == 0;
        }
    };
}