Compiler: Add methods for comparison operators

Author: adazem009

include/scratchcpp/dev/compiler.h

@@ -53,6 +53,10 @@ class LIBSCRATCHCPP_EXPORT Compiler
         void createMul();
         void createDiv();
 
+        void createCmpEQ();
+        void createCmpGT();
+        void createCmpLT();
+
         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

@@ -126,6 +126,24 @@ void Compiler::createDiv()
     impl->builder->createDiv();
 }
 
+/*! Creates an equality comparison instruction using the last 2 values. */
+void Compiler::createCmpEQ()
+{
+    impl->builder->createCmpEQ();
+}
+
+/*! Creates a greater than comparison instruction using the last 2 values. */
+void Compiler::createCmpGT()
+{
+    impl->builder->createCmpGT();
+}
+
+/*! Creates a lower than comparison instruction using the last 2 values. */
+void Compiler::createCmpLT()
+{
+    impl->builder->createCmpLT();
+}
+
 /*! Jumps to the given if substack. */
 void Compiler::moveToIf(std::shared_ptr<Block> substack)
 {

test/dev/compiler/compiler_test.cpp

@@ -271,6 +271,45 @@ TEST_F(CompilerTest, CreateDiv)
     compile(compiler, block);
 }
 
+TEST_F(CompilerTest, CreateCmpEQ)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createCmpEQ);
+        compiler->createCmpEQ();
+    });
+
+    compile(compiler, block);
+}
+
+TEST_F(CompilerTest, CreateCmpGT)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createCmpGT);
+        compiler->createCmpGT();
+    });
+
+    compile(compiler, block);
+}
+
+TEST_F(CompilerTest, CreateCmpLT)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("", "");
+
+    block->setCompileFunction([](Compiler *compiler) {
+        EXPECT_CALL(*m_builder, createCmpLT);
+        compiler->createCmpLT();
+    });
+
+    compile(compiler, block);
+}
+
 TEST_F(CompilerTest, MoveToIf)
 {
     Compiler compiler(&m_engine, &m_target);