LLVMCodeBuilder: Implement list contents

Author: adazem009

src/dev/engine/internal/llvm/llvmcodebuilder.cpp

@@ -623,6 +623,15 @@ std::shared_ptr<ExecutableCode> LLVMCodeBuilder::finalize()
                 break;
             }
 
+            case LLVMInstruction::Type::GetListContents: {
+                assert(step.args.size() == 0);
+                const LLVMListPtr &listPtr = m_listPtrs[step.workList];
+                llvm::Value *ptr = m_builder.CreateCall(resolve_list_to_string(), listPtr.ptr);
+                m_heap.push_back(ptr); // deallocate later
+                step.functionReturnReg->value = ptr;
+                break;
+            }
+
             case LLVMInstruction::Type::GetListItem: {
                 assert(step.args.size() == 1);
                 const auto &arg = step.args[0];
@@ -964,7 +973,10 @@ CompilerValue *LLVMCodeBuilder::addVariableValue(Variable *variable)
 
 CompilerValue *LLVMCodeBuilder::addListContents(List *list)
 {
-    return addConstValue(Value());
+    LLVMInstruction ins(LLVMInstruction::Type::GetListContents);
+    ins.workList = list;
+    m_listPtrs[list] = LLVMListPtr();
+    return createOp(ins, Compiler::StaticType::String);
 }
 
 CompilerValue *LLVMCodeBuilder::addListItem(List *list, CompilerValue *index)
@@ -2261,6 +2273,12 @@ llvm::FunctionCallee LLVMCodeBuilder::resolve_list_alloc_size_ptr()
     return resolveFunction("list_alloc_size_ptr", llvm::FunctionType::get(m_builder.getInt64Ty()->getPointerTo()->getPointerTo(), { listPtr }, false));
 }
 
+llvm::FunctionCallee LLVMCodeBuilder::resolve_list_to_string()
+{
+    llvm::Type *pointerType = llvm::PointerType::get(llvm::Type::getInt8Ty(m_ctx), 0);
+    return resolveFunction("list_to_string", llvm::FunctionType::get(pointerType, { pointerType }, false));
+}
+
 llvm::FunctionCallee LLVMCodeBuilder::resolve_strcasecmp()
 {
     llvm::Type *pointerType = llvm::PointerType::get(llvm::Type::getInt8Ty(m_ctx), 0);

src/dev/engine/internal/llvm/llvmcodebuilder.h

@@ -160,6 +160,7 @@ class LLVMCodeBuilder : public ICodeBuilder
         llvm::FunctionCallee resolve_list_data();
         llvm::FunctionCallee resolve_list_size_ptr();
         llvm::FunctionCallee resolve_list_alloc_size_ptr();
+        llvm::FunctionCallee resolve_list_to_string();
         llvm::FunctionCallee resolve_strcasecmp();
 
         Target *m_target = nullptr;

src/dev/engine/internal/llvm/llvminstruction.h

@@ -48,6 +48,7 @@ struct LLVMInstruction
             AppendToList,
             InsertToList,
             ListReplace,
+            GetListContents,
             GetListItem,
             GetListSize,
             GetListItemIndex,

test/dev/llvm/llvmcodebuilder_test.cpp

@@ -1801,6 +1801,54 @@ TEST_F(LLVMCodeBuilderTest, ListReplace)
     ASSERT_EQ(localList->toString(), "3 ipsum true hello world");
 }
 
+TEST_F(LLVMCodeBuilderTest, GetListContents)
+{
+    EngineMock engine;
+    Stage stage;
+    Sprite sprite;
+    sprite.setEngine(&engine);
+    EXPECT_CALL(engine, stage()).WillRepeatedly(Return(&stage));
+
+    std::unordered_map<List *, std::string> strings;
+
+    auto globalList = std::make_shared<List>("", "");
+    stage.addList(globalList);
+
+    auto localList = std::make_shared<List>("", "");
+    sprite.addList(localList);
+
+    globalList->append(1);
+    globalList->append(2);
+    globalList->append(3);
+
+    localList->append("Lorem");
+    localList->append("ipsum");
+    localList->append("dolor");
+    localList->append("sit");
+    strings[localList.get()] = localList->toString();
+
+    createBuilder(&sprite, true);
+
+    CompilerValue *v = m_builder->addListContents(globalList.get());
+    m_builder->addFunctionCall("test_print_string", Compiler::StaticType::Void, { Compiler::StaticType::String }, { v });
+
+    v = m_builder->addListContents(localList.get());
+    m_builder->addFunctionCall("test_print_string", Compiler::StaticType::Void, { Compiler::StaticType::String }, { v });
+
+    static const std::string expected =
+        "123\n"
+        "Lorem ipsum dolor sit\n";
+
+    auto code = m_builder->finalize();
+    auto ctx = code->createExecutionContext(&sprite);
+    testing::internal::CaptureStdout();
+    code->run(ctx.get());
+    ASSERT_EQ(testing::internal::GetCapturedStdout(), expected);
+
+    ASSERT_EQ(globalList->toString(), "123");
+    ASSERT_EQ(localList->toString(), "Lorem ipsum dolor sit");
+}
+
 TEST_F(LLVMCodeBuilderTest, GetListItem)
 {
     EngineMock engine;