Fix LLVM for pointer to structs

Author: czgdp1807

integration_tests/CMakeLists.txt

@@ -313,6 +313,8 @@ RUN(NAME structs_08          LABELS cpython llvm c)
 RUN(NAME structs_09          LABELS cpython llvm c)
 RUN(NAME structs_10          LABELS cpython llvm c)
 RUN(NAME structs_12          LABELS cpython llvm c)
+RUN(NAME structs_13          LABELS llvm
+    EXTRAFILES structs_13b.c)
 RUN(NAME sizeof_01          LABELS llvm c
         EXTRAFILES sizeof_01b.c)
 RUN(NAME enum_01             LABELS cpython llvm c)

integration_tests/structs_13.py

@@ -0,0 +1,43 @@
+from ltypes import i32, i16, i64, CPtr, dataclass, ccall, Pointer, c_p_pointer
+
+@dataclass
+class A:
+    x: i32
+    y: i16
+
+@ccall
+def cmalloc(size: i64) -> CPtr:
+    pass
+
+@ccall
+def is_null(ptr: CPtr) -> i32:
+    pass
+
+def add_A_members(Ax: i32, Ay: i16) -> i32:
+    return Ax + i32(Ay)
+
+def test_A_member_passing():
+    array_cptr: CPtr = cmalloc(sizeof(A) * i64(10))
+    assert not bool(is_null(array_cptr)), "Failed to allocate array on memory"
+    array_ptr: Pointer[A[:]]
+    c_p_pointer(array_cptr, array_ptr)
+    i: i32; sum_A_members: i32
+    for i in range(10):
+        array_ptr[i] = A(i, i16(i + 1))
+
+    for i in range(5):
+        a: A = array_ptr[i]
+        sum_A_members = add_A_members(a.x, a.y)
+        assert a.x == i
+        assert a.y == i16(i + 1)
+        print(sum_A_members)
+        assert sum_A_members == 2*i + 1
+
+    for i in range(6, 10):
+        sum_A_members = add_A_members(array_ptr[i].x, array_ptr[i].y)
+        assert array_ptr[i].x == i
+        assert array_ptr[i].y == i16(i + 1)
+        print(sum_A_members)
+        assert sum_A_members == 2*i + 1
+
+test_A_member_passing()

integration_tests/structs_13b.c

@@ -0,0 +1,9 @@
+#include <stdlib.h>
+
+void* cmalloc(int64_t size) {
+    return malloc(size);
+}
+
+int32_t is_null(void* ptr) {
+    return ptr == NULL;
+}

integration_tests/structs_13b.h

@@ -0,0 +1,4 @@
+#include <inttypes.h>
+
+void* cmalloc(int64_t size);
+int32_t is_null(void* ptr);

src/libasr/codegen/asr_to_llvm.cpp

@@ -1685,8 +1685,9 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
         llvm::Value* array = nullptr;
         if( ASR::is_a<ASR::Var_t>(*x.m_v) ) {
             ASR::Variable_t *v = ASRUtils::EXPR2VAR(x.m_v);
-            if( ASR::is_a<ASR::Struct_t>(*v->m_type) ) {
-                ASR::Struct_t* der_type = ASR::down_cast<ASR::Struct_t>(v->m_type);
+            if( ASR::is_a<ASR::Struct_t>(*ASRUtils::get_contained_type(v->m_type)) ) {
+                ASR::Struct_t* der_type = ASR::down_cast<ASR::Struct_t>(
+                    ASRUtils::get_contained_type(v->m_type));
                 der_type_name = ASRUtils::symbol_name(ASRUtils::symbol_get_past_external(der_type->m_derived_type));
             }
             uint32_t v_h = get_hash((ASR::asr_t*)v);

src/libasr/pass/pass_utils.h

@@ -180,6 +180,12 @@ namespace LFortran {
                     }
                 }
 
+                void visit_DoLoop(const ASR::DoLoop_t& x) {
+                    self().visit_do_loop_head(x.m_head);
+                    ASR::DoLoop_t& xx = const_cast<ASR::DoLoop_t&>(x);
+                    transform_stmts(xx.m_body, xx.n_body);
+                }
+
         };
 
         template <class Struct>