Add function to check whether task is ready (i.e., scheduled)

d-torrance · Nov 21, 2024 · 507e642 · 507e642
1 parent 0d91bf4
commit 507e642
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/M2/Macaulay2/d/pthread.d b/M2/Macaulay2/d/pthread.d
@@ -12,6 +12,7 @@ taskCreate(f:function(TaskCellBody):null,tb:TaskCellBody) ::=  Ccode(taskPointer
 
 export taskDone(tp:taskPointer) ::= Ccode(int, "taskDone(",tp,")")==1;
 export taskStarted(tp:taskPointer) ::=Ccode(int, "taskStarted(",tp,")")==1;
+export taskReady(tp:taskPointer) ::= Ccode(int, "taskReady(", tp, ")") == 1;
 pushTask(tp:taskPointer) ::=Ccode(void, "pushTask(",tp,")");
 taskResult(tp:taskPointer) ::=Ccode(voidPointer, "taskResult(",tp,")");
 export taskKeepRunning(tp:taskPointer) ::= Ccode(int, "taskKeepRunning(",tp,")")==1;

diff --git a/M2/Macaulay2/system/supervisor.cpp b/M2/Macaulay2/system/supervisor.cpp
@@ -141,6 +141,13 @@ extern "C" {
 	// synchronizes-with the task's thread, avoids data races
     return task->m_Started;
   }
+  int taskReady(struct ThreadTask *task)
+  {
+    if (!task)
+      return false;
+    lock_guard<pthreadMutex> lock(task->m_Mutex);
+    return task->m_ReadyToRun;
+  }
   void* taskResult(struct ThreadTask* task)
   {
     lock_guard<pthreadMutex> lock(task->m_Mutex);

diff --git a/M2/Macaulay2/system/supervisorinterface.h b/M2/Macaulay2/system/supervisorinterface.h
@@ -44,6 +44,10 @@ extern "C" {
      Returns 1 if the task has been started, 0 otherwise
   **/
   extern int taskStarted(struct ThreadTask* task);
+  /**
+     Returns 1 if the task is ready to run, 0 otherwise
+  **/
+  extern int taskReady(struct ThreadTask* task);
   /**
      Returns the taskResult if finished, NULL otherwise
   **/