LLVMCodeBuilder: Store loop list write instructions

Author: adazem009

src/engine/internal/llvm/llvmcodebuilder.cpp

@@ -1597,6 +1597,11 @@ void LLVMCodeBuilder::createListAppend(List *list, CompilerValue *item)
     if (m_listPtrs.find(list) == m_listPtrs.cend())
         m_listPtrs[list] = LLVMListPtr();
 
+    if (m_loopScope >= 0) {
+        auto scope = m_loopScopes[m_loopScope];
+        m_listPtrs[list].loopListWrites[scope].push_back(m_instructions.back());
+    }
+
     m_listInstructions.push_back(m_instructions.back());
 }
 
@@ -1609,6 +1614,11 @@ void LLVMCodeBuilder::createListInsert(List *list, CompilerValue *index, Compile
     if (m_listPtrs.find(list) == m_listPtrs.cend())
         m_listPtrs[list] = LLVMListPtr();
 
+    if (m_loopScope >= 0) {
+        auto scope = m_loopScopes[m_loopScope];
+        m_listPtrs[list].loopListWrites[scope].push_back(m_instructions.back());
+    }
+
     m_listInstructions.push_back(m_instructions.back());
 }
 
@@ -1621,6 +1631,11 @@ void LLVMCodeBuilder::createListReplace(List *list, CompilerValue *index, Compil
     if (m_listPtrs.find(list) == m_listPtrs.cend())
         m_listPtrs[list] = LLVMListPtr();
 
+    if (m_loopScope >= 0) {
+        auto scope = m_loopScopes[m_loopScope];
+        m_listPtrs[list].loopListWrites[scope].push_back(m_instructions.back());
+    }
+
     m_listInstructions.push_back(m_instructions.back());
 }
 

src/engine/internal/llvm/llvmlistptr.h

@@ -3,6 +3,7 @@
 #pragma once
 
 #include <scratchcpp/compiler.h>
+#include <unordered_map>
 
 namespace llvm
 {
@@ -14,6 +15,9 @@ class Value;
 namespace libscratchcpp
 {
 
+class LLVMLoopScope;
+class LLVMInstruction;
+
 struct LLVMListPtr
 {
         llvm::Value *ptr = nullptr;
@@ -22,6 +26,9 @@ struct LLVMListPtr
         llvm::Value *allocatedSizePtr = nullptr;
         llvm::Value *dataPtrDirty = nullptr;
         Compiler::StaticType type = Compiler::StaticType::Unknown;
+
+        // Used in build phase to check the type safety of lists in loops
+        std::unordered_map<std::shared_ptr<LLVMLoopScope>, std::vector<std::shared_ptr<LLVMInstruction>>> loopListWrites; // loop scope, write instructions
 };
 
 } // namespace libscratchcpp