Engine: Add broadcast methods

Author: adazem009

src/engine/engine.cpp

@@ -76,9 +76,11 @@ void Engine::compile()
             if (block->topLevel()) {
                 auto section = blockSection(block->opcode());
                 if (section) {
+                    auto vm = std::make_shared<VirtualMachine>(target.get(), this);
+                    m_scripts[block] = vm;
+
                     compiler.compile(block);
 
-                    auto vm = std::make_shared<VirtualMachine>(target.get(), this);
                     vm->setFunctions(m_functions);
                     vm->setConstValues(compiler.constValues());
                     vm->setVariables(compiler.variablePtrs());
@@ -88,7 +90,6 @@ void Engine::compile()
                         auto b = block->inputAt(block->findInput("custom_block"))->valueBlock();
                         procedureBytecodeMap[b->mutationPrototype()->procCode()] = vm->bytecode();
                     }
-                    m_scripts[block] = vm;
                 } else
                     std::cout << "warning: unsupported top level block: " << block->opcode() << std::endl;
             }
@@ -179,6 +180,24 @@ void Engine::startScript(std::shared_ptr<Block> topLevelBlock, std::shared_ptr<T
     }
 }
 
+/*! Starts the script of the broadcast with the given index. */
+void libscratchcpp::Engine::broadcast(unsigned int index)
+{
+    const std::vector<VirtualMachine *> scripts = m_broadcastMap[index];
+    for (auto vm : scripts) {
+        auto it = std::find(m_runningScripts.begin(), m_runningScripts.end(), vm);
+        if (it != m_runningScripts.end()) {
+            // Reset the script if it's already running
+            auto i = it - m_runningScripts.begin();
+            m_scriptPositions[i] = m_runningScripts[i]->bytecode();
+        } else {
+            m_runningScripts.push_back(vm);
+            m_scriptPositions.push_back(vm->bytecode());
+            scriptCount++;
+        }
+    }
+}
+
 /*! Stops the given script. */
 void Engine::stopScript(VirtualMachine *vm)
 {

src/engine/engine.h

@@ -35,6 +35,7 @@ class LIBSCRATCHCPP_EXPORT Engine
         void start();
         void stop();
         void startScript(std::shared_ptr<Block> topLevelBlock, std::shared_ptr<Target> target);
+        void broadcast(unsigned int index);
         void stopScript(VirtualMachine *vm);
         void stopTarget(Target *target, VirtualMachine *exceptScript);
         void run();
@@ -54,6 +55,8 @@ class LIBSCRATCHCPP_EXPORT Engine
         int findBroadcast(const std::string &broadcastName) const;
         int findBroadcastById(const std::string &broadcastId) const;
 
+        void addBroadcastScript(std::shared_ptr<Block> whenReceivedBlock, std::shared_ptr<Broadcast> broadcast);
+
         std::vector<std::shared_ptr<Target>> targets() const;
         void setTargets(const std::vector<std::shared_ptr<Target>> &newTargets);
         Target *targetAt(int index) const;
@@ -71,6 +74,7 @@ class LIBSCRATCHCPP_EXPORT Engine
         std::vector<std::shared_ptr<IBlockSection>> m_sections;
         std::vector<std::shared_ptr<Target>> m_targets;
         std::vector<std::shared_ptr<Broadcast>> m_broadcasts;
+        std::unordered_map<unsigned int, std::vector<VirtualMachine *>> m_broadcastMap;
         std::vector<std::string> m_extensions;
         std::vector<VirtualMachine *> m_runningScripts;
         std::vector<unsigned int *> m_scriptPositions;