Compiler: Add addLoopIndex() method

Author: adazem009

include/scratchcpp/dev/compiler.h

@@ -51,6 +51,7 @@ class LIBSCRATCHCPP_EXPORT Compiler
         CompilerValue *addTargetFunctionCall(const std::string &functionName, StaticType returnType = StaticType::Void, const ArgTypes &argTypes = {}, const Args &args = {});
         CompilerValue *addFunctionCallWithCtx(const std::string &functionName, StaticType returnType = StaticType::Void, const ArgTypes &argTypes = {}, const Args &args = {});
         CompilerConstant *addConstValue(const Value &value);
+        CompilerValue *addLoopIndex();
         CompilerValue *addVariableValue(Variable *variable);
         CompilerValue *addListContents(List *list);
         CompilerValue *addListItem(List *list, CompilerValue *index);

src/dev/engine/compiler.cpp

@@ -115,6 +115,12 @@ CompilerConstant *Compiler::addConstValue(const Value &value)
     return static_cast<CompilerConstant *>(impl->builder->addConstValue(value));
 }
 
+/*! Adds the index of the current repeat loop to the compiled code. */
+CompilerValue *Compiler::addLoopIndex()
+{
+    return impl->builder->addLoopIndex();
+}
+
 /*! Adds the value of the given variable to the code. */
 CompilerValue *Compiler::addVariableValue(Variable *variable)
 {

test/dev/compiler/compiler_test.cpp

@@ -161,6 +161,25 @@ TEST_F(CompilerTest, AddConstValue)
     compile(compiler, block);
 }
 
+TEST_F(CompilerTest, AddLoopIndex)
+{
+    Compiler compiler(&m_engine, &m_target);
+    auto block = std::make_shared<Block>("a", "");
+    block->setCompileFunction([](Compiler *compiler) -> CompilerValue * {
+        CompilerValue ret(Compiler::StaticType::Unknown);
+
+        EXPECT_CALL(*m_builder, addLoopIndex()).WillOnce(Return(&ret));
+        EXPECT_EQ(compiler->addLoopIndex(), &ret);
+
+        EXPECT_CALL(*m_builder, addLoopIndex()).WillOnce(Return(nullptr));
+        EXPECT_EQ(compiler->addLoopIndex(), nullptr);
+
+        return nullptr;
+    });
+
+    compile(compiler, block);
+}
+
 TEST_F(CompilerTest, AddVariableValue)
 {
     Compiler compiler(&m_engine, &m_target);