VariableBlocks: Add compile functions

Author: adazem009

src/blocks/variableblocks.cpp

@@ -1,12 +1,15 @@
 // SPDX-License-Identifier: Apache-2.0
 
 #include "variableblocks.h"
+#include "../engine/compiler.h"
 
 using namespace libscratchcpp;
 
 VariableBlocks::VariableBlocks()
 {
     // Blocks
+    addCompileFunction("data_setvariableto", &VariableBlocks::compileSetVariable);
+    addCompileFunction("data_changevariableby", &VariableBlocks::compileChangeVariableBy);
     addBlock("data_setvariableto", &VariableBlocks::setVariable);
     addBlock("data_changevariableby", &VariableBlocks::changeVariableBy);
 
@@ -22,6 +25,18 @@ std::string VariableBlocks::name() const
     return "Variables";
 }
 
+void VariableBlocks::compileSetVariable(Compiler *compiler)
+{
+    compiler->addInput(VALUE);
+    compiler->addInstruction(vm::OP_SET_VAR, { compiler->variableIndex(compiler->field(VARIABLE)->valuePtr()) });
+}
+
+void VariableBlocks::compileChangeVariableBy(Compiler *compiler)
+{
+    compiler->addInput(VALUE);
+    compiler->addInstruction(vm::OP_CHANGE_VAR, { compiler->variableIndex(compiler->field(VARIABLE)->valuePtr()) });
+}
+
 Value VariableBlocks::setVariable(const BlockArgs &args)
 {
     auto variable = std::static_pointer_cast<Variable>(args.field(VARIABLE)->valuePtr());

src/blocks/variableblocks.h

@@ -25,6 +25,9 @@ class LIBSCRATCHCPP_EXPORT VariableBlocks : public IBlockSection
 
         std::string name() const override;
 
+        static void compileSetVariable(Compiler *compiler);
+        static void compileChangeVariableBy(Compiler *compiler);
+
         static Value setVariable(const BlockArgs &args);
         static Value changeVariableBy(const BlockArgs &args);
 };