Store more info about used procedures

Author: adazem009

src/engine/internal/llvm/instructions/procedures.cpp

@@ -44,7 +44,7 @@ LLVMInstruction *Procedures::buildCallProcedure(LLVMInstruction *ins)
     llvm::FunctionType *type = m_utils.scriptFunctionType(ins->procedurePrototype);
     std::vector<llvm::Value *> args;
 
-    m_utils.compilerCtx()->addUsedProcedure(name);
+    m_utils.compilerCtx()->addUsedProcedure(ins->procedurePrototype, name, type);
 
     llvm::FunctionType *funcType = m_utils.scriptFunctionType(nullptr);
     int passArgCount = funcType->getNumParams();

src/engine/internal/llvm/llvmcompilercontext.cpp

@@ -65,9 +65,9 @@ const std::unordered_map<function_id_t, LLVMExecutableCode *> &LLVMCompilerConte
     return m_codeMap;
 }
 
-void LLVMCompilerContext::addUsedProcedure(const std::string &functionName)
+void LLVMCompilerContext::addUsedProcedure(BlockPrototype *prototype, const std::string &functionName, llvm::FunctionType *functionType)
 {
-    m_usedProcedures.insert(functionName);
+    m_usedProcedures[prototype] = { functionName, functionType };
 }
 
 function_id_t LLVMCompilerContext::getNextFunctionId()

src/engine/internal/llvm/llvmcompilercontext.h

@@ -8,14 +8,15 @@
 
 #include <scratchcpp/compilercontext.h>
 
-#include <unordered_set>
+#include <unordered_map>
 
 namespace libscratchcpp
 {
 
 class ExecutionContext;
 class ValueData;
 class List;
+class BlockPrototype;
 class LLVMExecutableCode;
 
 // NOTE: Change this in LLVMTypes as well
@@ -34,7 +35,7 @@ class LLVMCompilerContext : public CompilerContext
         void addCode(LLVMExecutableCode *code);
         const std::unordered_map<function_id_t, LLVMExecutableCode *> &codeMap() const;
 
-        void addUsedProcedure(const std::string &functionName);
+        void addUsedProcedure(BlockPrototype *prototype, const std::string &functionName, llvm::FunctionType *functionType);
 
         function_id_t getNextFunctionId();
 
@@ -94,7 +95,7 @@ class LLVMCompilerContext : public CompilerContext
         llvm::StructType *m_stringPtrType = nullptr;
         llvm::Type *m_functionIdType = nullptr;
 
-        std::unordered_set<std::string> m_usedProcedures;
+        std::unordered_map<BlockPrototype *, std::pair<std::string, llvm::FunctionType *>> m_usedProcedures; // function name, function type
 };
 
 } // namespace libscratchcpp