ControlBlocks: Add wait block

Author: adazem009

src/blocks/controlblocks.cpp

@@ -2,6 +2,7 @@
 
 #include "controlblocks.h"
 #include "../engine/compiler.h"
+#include <cassert>
 
 namespace libscratchcpp
 {
@@ -15,12 +16,14 @@ ControlBlocks::ControlBlocks()
     addCompileFunction("control_if", &compileIfStatement);
     addCompileFunction("control_if_else", &compileIfElseStatement);
     addCompileFunction("control_stop", &compileStop);
+    addCompileFunction("control_wait", &compileWait);
 
     // Inputs
     addInput("SUBSTACK", SUBSTACK);
     addInput("SUBSTACK2", SUBSTACK2);
     addInput("TIMES", TIMES);
     addInput("CONDITION", CONDITION);
+    addInput("DURATION", DURATION);
 
     // Fields
     addField("STOP_OPTION", STOP_OPTION);
@@ -122,6 +125,13 @@ void ControlBlocks::compileStop(Compiler *compiler)
     }
 }
 
+void ControlBlocks::compileWait(Compiler *compiler)
+{
+    compiler->addInput(DURATION);
+    compiler->addFunctionCall(&startWait);
+    compiler->addFunctionCall(&wait);
+}
+
 unsigned int ControlBlocks::stopAll(VirtualMachine *vm)
 {
     vm->engine()->stop();
@@ -134,4 +144,24 @@ unsigned int ControlBlocks::stopOtherScriptsInSprite(VirtualMachine *vm)
     return 0;
 }
 
+unsigned int ControlBlocks::startWait(VirtualMachine *vm)
+{
+    auto currentTime = std::chrono::steady_clock::now();
+    assert(m_startTimeMap.count(vm) == 0);
+    m_timeMap[vm] = { currentTime, vm->getInput(0, 1)->toDouble() * 1000 };
+    return 1;
+}
+
+unsigned int ControlBlocks::wait(VirtualMachine *vm)
+{
+    auto currentTime = std::chrono::steady_clock::now();
+    assert(m_startTimeMap.count(vm) == 1);
+    if (std::chrono::duration_cast<std::chrono::milliseconds>(currentTime - m_timeMap[vm].first).count() >= m_timeMap[vm].second) {
+        m_timeMap.erase(vm);
+        vm->stop(true, true, false);
+    } else
+        vm->stop(true, true, true);
+    return 0;
+}
+
 } // namespace libscratchcpp

src/blocks/controlblocks.h

@@ -16,7 +16,8 @@ class LIBSCRATCHCPP_EXPORT ControlBlocks : public IBlockSection
             SUBSTACK,
             SUBSTACK2,
             TIMES,
-            CONDITION
+            CONDITION,
+            DURATION
         };
 
         enum Fields
@@ -42,9 +43,15 @@ class LIBSCRATCHCPP_EXPORT ControlBlocks : public IBlockSection
         static void compileIfStatement(Compiler *compiler);
         static void compileIfElseStatement(Compiler *compiler);
         static void compileStop(Compiler *compiler);
+        static void compileWait(Compiler *compiler);
 
+    private:
         static unsigned int stopAll(VirtualMachine *vm);
         static unsigned int stopOtherScriptsInSprite(VirtualMachine *vm);
+        static unsigned int startWait(VirtualMachine *vm);
+        static unsigned int wait(VirtualMachine *vm);
+
+        static inline std::unordered_map<VirtualMachine *, std::pair<std::chrono::steady_clock::time_point, int>> m_timeMap;
 };
 
 } // namespace libscratchcpp