move yielding into Thread base class + make user fn a pure virtual fn and require user to derive from Thread

1 files changed, 11 insertions, 7 deletions

diff --git a/lib/matcha.h b/lib/matcha.h
index 905107f..1506081 100644
--- a/lib/matcha.h
+++ b/lib/matcha.h
@@ -1,14 +1,18 @@
 struct Thread {
     Thread(const Thread&) = delete;
     Thread& operator=(const Thread&) = delete;
-    Thread(void (*fn)());
+    Thread();
 
-    static void entry(void* obj);
+    virtual void threadFn() = 0;
+
+    // use from executor
+    void yield_to() const;
+
+  protected:
+    // use in thread
+    void yield();
 
-    // private:
+  private:
+    static void entry(void* obj);
     void* mStackPtr;
-    void (*mUserFn)();
 };
-
-void yield_to(const Thread& t);
-void yield_from(Thread& t);