Changeset 596f987b for src/libcfa/concurrency/threads

Timestamp:

Dec 15, 2016, 12:40:01 PM (8 years ago)

Author:

Thierry Delisle <tdelisle@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

66f8528

Parents:

12d44bb

Message:

Some optimization and clean-up of coroutines

File:

• src/libcfa/concurrency/threads (modified) (2 diffs)

src/libcfa/concurrency/threads

@@ -18 +18 @@
 #define __THREADS_H__
 
+#include "assert"       //
 #include "invoke.h"
 
-void ?{}(coStack_t* this);
-
-void ?{}(coroutine* this);
-
+//-----------------------------------------------------------------------------
+// Coroutine trait
+// Anything that implements this trait can be resumed.
+// Anything that is resumed is a coroutine.
 trait is_coroutine(dtype T) {
       void co_main(T* this);
@@ -29 +30 @@
 };
 
-void suspend(void);
+//-----------------------------------------------------------------------------
+// Ctors and dtors
+void ?{}(coStack_t* this);
+void ?{}(coroutine* this);
+void ^?{}(coStack_t* this);
+void ^?{}(coroutine* this);
+
+//-----------------------------------------------------------------------------
+// Public coroutine API
+static inline void suspend();
 
 forall(dtype T | is_coroutine(T))
-void resume(T* cor);
+static inline void resume(T* cor);
 
 forall(dtype T | is_coroutine(T))
 void prime(T* cor);
+
+//-----------------------------------------------------------------------------
+// PRIVATE exposed because of inline
+
+// Start coroutine routines
+extern "C" {
+      forall(dtype T | is_coroutine(T))
+      void CtxInvokeCoroutine(T* this);
+
+      forall(dtype T | is_coroutine(T))
+      void CtxStart(T* this, void (*invoke)(T*));
+}
+
+// Get current coroutine
+extern coroutine* current_coroutine; //PRIVATE, never use directly
+static inline coroutine* this_coroutine(void) {
+        return current_coroutine;
+}
+
+// Private wrappers for context switch and stack creation
+extern void corCxtSw(coroutine* src, coroutine* dst);
+extern void create_stack( coStack_t* this, unsigned int storageSize );
+
+// Suspend implementation inlined for performance
+static inline void suspend() {
+      coroutine* src = this_coroutine();                // optimization
+
+        assertf( src->last != 0, 
+                "Attempt to suspend coroutine %.256s (%p) that has never been resumed.\n"
+                "Possible cause is a suspend executed in a member called by a coroutine user rather than by the coroutine main.",
+                src->name, src );
+        assertf( src->last->notHalted, 
+                "Attempt by coroutine %.256s (%p) to suspend back to terminated coroutine %.256s (%p).\n"
+                "Possible cause is terminated coroutine's main routine has already returned.",
+                src->name, src, src->last->name, src->last );
+
+        corCxtSw( src, src->last );
+}
+
+// Resume implementation inlined for performance
+forall(dtype T | is_coroutine(T))
+static inline void resume(T* cor) {
+        coroutine* src = this_coroutine();              // optimization
+        coroutine* dst = get_coroutine(cor);
+
+      if( unlikely(!dst->stack.base) ) {
+                create_stack(&dst->stack, dst->stack.size);
+                CtxStart(cor, CtxInvokeCoroutine);
+        }
+
+      // not resuming self ?
+        if ( src != dst ) {
+                assertf( dst->notHalted , 
+                        "Attempt by coroutine %.256s (%p) to resume terminated coroutine %.256s (%p).\n"
+                        "Possible cause is terminated coroutine's main routine has already returned.",
+                        src->name, src, dst->name, dst );
+
+            // set last resumer
+                dst->last = src;
+        } // if
+
+      // always done for performance testing
+        corCxtSw( src, dst );
+}
 
 #endif //__THREADS_H__