ExecutionContext: Add stack timer

Author: adazem009

include/scratchcpp/dev/executioncontext.h

@@ -11,6 +11,7 @@ namespace libscratchcpp
 class Thread;
 class IEngine;
 class Promise;
+class IStackTimer;
 class ExecutionContextPrivate;
 
 /*! \brief The ExecutionContext represents the execution context of a target (can be a clone) with variables, lists, etc. */
@@ -27,6 +28,9 @@ class LIBSCRATCHCPP_EXPORT ExecutionContext
         std::shared_ptr<Promise> promise() const;
         void setPromise(std::shared_ptr<Promise> promise);
 
+        IStackTimer *stackTimer() const;
+        void setStackTimer(IStackTimer *newStackTimer);
+
     private:
         spimpl::unique_impl_ptr<ExecutionContextPrivate> impl;
 };

src/dev/engine/executioncontext.cpp

@@ -36,3 +36,15 @@ void ExecutionContext::setPromise(std::shared_ptr<Promise> promise)
 {
     impl->promise = promise;
 }
+
+/*! Returns the stack timer of this context. Can be used for wait blocks. */
+IStackTimer *ExecutionContext::stackTimer() const
+{
+    return impl->stackTimer;
+}
+
+/*! Sets a custom stack timer. */
+void ExecutionContext::setStackTimer(IStackTimer *newStackTimer)
+{
+    impl->stackTimer = newStackTimer;
+}

src/dev/engine/executioncontext_p.cpp

@@ -5,6 +5,8 @@
 using namespace libscratchcpp;
 
 ExecutionContextPrivate::ExecutionContextPrivate(Thread *thread) :
-    thread(thread)
+    thread(thread),
+    defaultStackTimer(std::make_unique<StackTimer>()),
+    stackTimer(defaultStackTimer.get())
 {
 }

src/dev/engine/executioncontext_p.h

@@ -4,11 +4,12 @@
 
 #include <memory>
 
+#include "../../engine/internal/stacktimer.h"
+
 namespace libscratchcpp
 {
 
 class Thread;
-class Target;
 class Promise;
 
 struct ExecutionContextPrivate
@@ -17,6 +18,8 @@ struct ExecutionContextPrivate
 
         Thread *thread = nullptr;
         std::shared_ptr<Promise> promise;
+        std::unique_ptr<IStackTimer> defaultStackTimer;
+        IStackTimer *stackTimer = nullptr;
 };
 
 } // namespace libscratchcpp

test/dev/executioncontext/executioncontext_test.cpp

@@ -3,6 +3,7 @@
 #include <scratchcpp/dev/promise.h>
 
 #include <enginemock.h>
+#include <stacktimermock.h>
 
 #include "../../common.h"
 
@@ -29,3 +30,13 @@ TEST(ExecutionContextTest, Promise)
     ctx.setPromise(nullptr);
     ASSERT_EQ(ctx.promise(), nullptr);
 }
+
+TEST(ExecutionContextTest, StackTimer)
+{
+    ExecutionContext ctx(nullptr);
+    ASSERT_TRUE(ctx.stackTimer());
+
+    StackTimerMock timer;
+    ctx.setStackTimer(&timer);
+    ASSERT_EQ(ctx.stackTimer(), &timer);
+}