scheduler.h

Go to the documentation of this file.

 
#ifndef NSBACI_SERVICES_RUNTIME_SCHEDULER_H
#define NSBACI_SERVICES_RUNTIME_SCHEDULER_H
 
#include <optional>
#include <unordered_map>
#include <vector>
 
#include "thread.h"

namespace nsbaci::services::runtime {

class Scheduler {
 public:
  Scheduler() = default;
  virtual ~Scheduler() = default;

  virtual Thread* pickNext() = 0;

  virtual void addThread(Thread thread) = 0;

  virtual void blockCurrent() = 0;

  virtual void blockOnSemaphore(uint32_t semaphoreAddr) = 0;

  virtual size_t unblockSemaphore(uint32_t semaphoreAddr) = 0;

  virtual void unblock(nsbaci::types::ThreadID threadId) = 0;

  virtual void yield() = 0;

  virtual void terminateCurrent() = 0;

  virtual bool hasThreads() const = 0;

  virtual Thread* current() = 0;

  virtual void clear() = 0;

  virtual void unblockIO() = 0;

  virtual const std::vector<Thread>& getThreads() const = 0;

  virtual void blockOnCoend(int32_t expectedThreads) = 0;

  virtual void checkCoendUnblock() = 0;
 
 protected:
  std::vector<Thread> threads;         
  std::vector<size_t> readyQueue;      
  std::vector<size_t> blockedQueue;    
  std::vector<size_t> ioQueue;         
  std::optional<size_t> runningIndex;  

  std::unordered_map<uint32_t, std::vector<size_t>> semaphoreQueues;

  std::vector<std::pair<size_t, int32_t>> coendQueue;
};
 
}  // namespace nsbaci::services::runtime
 
#endif  // NSBACI_SERVICES_RUNTIME_SCHEDULER_H