Merge pull request #1653 from harshsingh-24/dict-funcs

Initial Support for Global Dictionaries

Author: certik

integration_tests/CMakeLists.txt

@@ -443,3 +443,4 @@ RUN(NAME global_syms_02    LABELS cpython llvm c)
 RUN(NAME global_syms_03_b  LABELS cpython llvm c)
 RUN(NAME global_syms_03_c  LABELS cpython llvm c)
 RUN(NAME global_syms_04    LABELS cpython llvm c wasm wasm_x64)
+RUN(NAME global_syms_05    LABELS cpython llvm c)

integration_tests/global_syms_05.py

@@ -0,0 +1,21 @@
+from lpython import i32
+
+# GLOBAL Dictionary
+x: dict[i32, i32]
+x = {0 : 0,  1: 1000, 2: 2000, 3 : 3000}
+
+x[4] = 4000
+assert len(x) == 5
+x[5] = 5000
+assert x[2] == 2000
+i: i32
+for i in range(len(x)):
+    assert x[i] == i * 1000
+
+# Copy of Dictionary
+tmp: dict[i32, i32]
+tmp = x
+tmp[6] = 6000
+assert len(tmp) == 7
+assert tmp[6] == 6000
+assert tmp[1] == 1000

src/libasr/codegen/asr_to_llvm.cpp

@@ -2304,6 +2304,14 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
                 llvm::ConstantStruct::get(list_type,
                 llvm::Constant::getNullValue(list_type)));
             llvm_symtab[h] = ptr;
+        } else if(x.m_type->type == ASR::ttypeType::Dict) {
+            llvm::StructType* dict_type = static_cast<llvm::StructType*>(
+                get_type_from_ttype_t_util(x.m_type));
+            llvm::Constant *ptr = module->getOrInsertGlobal(x.m_name, dict_type);
+            module->getNamedGlobal(x.m_name)->setInitializer(
+                llvm::ConstantStruct::get(dict_type,
+                llvm::Constant::getNullValue(dict_type)));
+            llvm_symtab[h] = ptr;
         } else if (x.m_type->type == ASR::ttypeType::TypeParameter) {
             // Ignore type variables
         } else {