ExecutionContext: Add random number generator

Author: adazem009

include/scratchcpp/dev/executioncontext.h

@@ -12,6 +12,7 @@ class Thread;
 class IEngine;
 class Promise;
 class IStackTimer;
+class IRandomGenerator;
 class ExecutionContextPrivate;
 
 /*! \brief The ExecutionContext represents the execution context of a target (can be a clone) with variables, lists, etc. */
@@ -31,6 +32,9 @@ class LIBSCRATCHCPP_EXPORT ExecutionContext
         IStackTimer *stackTimer() const;
         void setStackTimer(IStackTimer *newStackTimer);
 
+        IRandomGenerator *rng() const;
+        void setRng(IRandomGenerator *newRng);
+
     private:
         spimpl::unique_impl_ptr<ExecutionContextPrivate> impl;
 };

src/dev/engine/executioncontext.cpp

@@ -48,3 +48,15 @@ void ExecutionContext::setStackTimer(IStackTimer *newStackTimer)
 {
     impl->stackTimer = newStackTimer;
 }
+
+/*! Returns the random number generator of this context. */
+IRandomGenerator *ExecutionContext::rng() const
+{
+    return impl->rng;
+}
+
+/*! Sets a custom random number generator. */
+void ExecutionContext::setRng(IRandomGenerator *newRng)
+{
+    impl->rng = newRng;
+}

src/dev/engine/executioncontext_p.cpp

@@ -1,12 +1,14 @@
 // SPDX-License-Identifier: Apache-2.0
 
 #include "executioncontext_p.h"
+#include "../../engine/internal/randomgenerator.h"
 
 using namespace libscratchcpp;
 
 ExecutionContextPrivate::ExecutionContextPrivate(Thread *thread) :
     thread(thread),
     defaultStackTimer(std::make_unique<StackTimer>()),
-    stackTimer(defaultStackTimer.get())
+    stackTimer(defaultStackTimer.get()),
+    rng(RandomGenerator::instance().get())
 {
 }

src/dev/engine/executioncontext_p.h

@@ -11,6 +11,7 @@ namespace libscratchcpp
 
 class Thread;
 class Promise;
+class IRandomGenerator;
 
 struct ExecutionContextPrivate
 {
@@ -20,6 +21,7 @@ struct ExecutionContextPrivate
         std::shared_ptr<Promise> promise;
         std::unique_ptr<IStackTimer> defaultStackTimer;
         IStackTimer *stackTimer = nullptr;
+        IRandomGenerator *rng = nullptr;
 };
 
 } // namespace libscratchcpp

test/dev/executioncontext/executioncontext_test.cpp

@@ -4,6 +4,7 @@
 
 #include <enginemock.h>
 #include <stacktimermock.h>
+#include <randomgeneratormock.h>
 
 #include "../../common.h"
 
@@ -40,3 +41,13 @@ TEST(ExecutionContextTest, StackTimer)
     ctx.setStackTimer(&timer);
     ASSERT_EQ(ctx.stackTimer(), &timer);
 }
+
+TEST(ExecutionContextTest, Rng)
+{
+    ExecutionContext ctx(nullptr);
+    ASSERT_TRUE(ctx.rng());
+
+    RandomGeneratorMock rng;
+    ctx.setRng(&rng);
+    ASSERT_EQ(ctx.rng(), &rng);
+}