Merge pull request #1137 from Smit-create/complex1

Implement C backend for complex numbers

Author: Smit-create

integration_tests/CMakeLists.txt

@@ -235,7 +235,7 @@ RUN(NAME bindc_03   LABELS llvm
         EXTRAFILES bindc_03b.c)
 RUN(NAME test_generics_01    LABELS cpython llvm)
 RUN(NAME test_cmath          LABELS cpython llvm)
-RUN(NAME test_complex        LABELS cpython llvm)
+RUN(NAME test_complex        LABELS cpython llvm c)
 RUN(NAME test_max_min        LABELS cpython llvm)
 RUN(NAME test_global         LABELS cpython llvm)
 RUN(NAME test_integer_bitnot LABELS cpython llvm)

src/libasr/codegen/asr_to_c.cpp

@@ -590,16 +590,9 @@ R"(
 
     void visit_ComplexConstant(const ASR::ComplexConstant_t &x) {
         headers.insert("complex");
-
-        ASR::Complex_t *t = ASR::down_cast<ASR::Complex_t>(x.m_type);
-
         std::string re = std::to_string(x.m_re);
         std::string im = std::to_string(x.m_im);
-
-        std::string type_name = "float complex";
-        if (t->m_kind == 8) type_name = "double complex";
-
-        src = "(" + type_name + ") (" + re + ", " + im + ")";
+        src = "CMPLX(" + re + ", " + im + ")";
 
         last_expr_precedence = 2;
     }

src/libasr/codegen/asr_to_c_cpp.h

@@ -748,6 +748,10 @@ R"(#include <stdio.h>
         handle_UnaryMinus(x);
     }
 
+    void visit_ComplexUnaryMinus(const ASR::ComplexUnaryMinus_t &x) {
+        handle_UnaryMinus(x);
+    }
+
     template <typename T>
     void handle_UnaryMinus(const T &x) {
         self().visit_expr(*x.m_arg);
@@ -760,6 +764,26 @@ R"(#include <stdio.h>
         }
     }
 
+    void visit_ComplexRe(const ASR::ComplexRe_t &x) {
+        headers.insert("complex");
+        self().visit_expr(*x.m_arg);
+        if (is_c) {
+            src = "creal(" + src + ")";
+        } else {
+            src = src + ".real()";
+        }
+    }
+
+    void visit_ComplexIm(const ASR::ComplexIm_t &x) {
+        headers.insert("complex");
+        self().visit_expr(*x.m_arg);
+        if (is_c) {
+            src = "cimag(" + src + ")";
+        } else {
+            src = src + ".imag()";
+        }
+    }
+
     void visit_LogicalNot(const ASR::LogicalNot_t &x) {
         self().visit_expr(*x.m_arg);
         int expr_precedence = last_expr_precedence;