Compiler: Add methods for add, sub, mul and div operators

Author: adazem009

include/scratchcpp/dev/compiler.h

@@ -47,6 +47,11 @@ class LIBSCRATCHCPP_EXPORT Compiler
         void addListContents(List *list);
         void addInput(const std::string &name);
 
+        void createAdd();
+        void createSub();
+        void createMul();
+        void createDiv();
+
         void moveToIf(std::shared_ptr<Block> substack);
         void moveToIfElse(std::shared_ptr<Block> substack1, std::shared_ptr<Block> substack2);
         void moveToRepeatLoop(std::shared_ptr<Block> substack);

src/dev/engine/compiler.cpp

@@ -102,6 +102,30 @@ void Compiler::addInput(const std::string &name)
     addInput(impl->block->inputAt(impl->block->findInput(name)).get());
 }
 
+/*! Creates an add instruction from the last 2 values. */
+void Compiler::createAdd()
+{
+    impl->builder->createAdd();
+}
+
+/*! Creates a subtract instruction from the last 2 values. */
+void Compiler::createSub()
+{
+    impl->builder->createSub();
+}
+
+/*! Creates a multiply instruction from the last 2 values. */
+void Compiler::createMul()
+{
+    impl->builder->createMul();
+}
+
+/*! Creates a divide instruction from the last 2 values. */
+void Compiler::createDiv()
+{
+    impl->builder->createDiv();
+}
+
 /*! Jumps to the given if substack. */
 void Compiler::moveToIf(std::shared_ptr<Block> substack)
 {

test/dev/compiler/compiler_test.cpp

@@ -219,6 +219,58 @@ TEST_F(CompilerTest, AddInput)
     compile(compiler, block);
 }
 
+TEST_F(CompilerTest, CreateAdd)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createAdd);
+        compiler->createAdd();
+    });
+
+    compile(compiler, block);
+}
+
+TEST_F(CompilerTest, CreateSub)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createSub);
+        compiler->createSub();
+    });
+
+    compile(compiler, block);
+}
+
+TEST_F(CompilerTest, CreateMul)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createMul);
+        compiler->createMul();
+    });
+
+    compile(compiler, block);
+}
+
+TEST_F(CompilerTest, CreateDiv)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createDiv);
+        compiler->createDiv();
+    });
+
+    compile(compiler, block);
+}
+
 TEST_F(CompilerTest, MoveToIf)
 {
     Compiler compiler(&m_engine, &m_target);