wip

Author: czgdp1807

integration_tests/CMakeLists.txt

@@ -150,7 +150,7 @@ RUN(NAME test_str_03         LABELS cpython llvm c)
 RUN(NAME test_list_01        LABELS cpython llvm c)
 RUN(NAME test_list_02        LABELS cpython llvm c)
 RUN(NAME test_list_03        LABELS cpython llvm c)
-RUN(NAME test_list_04        LABELS cpython llvm)
+RUN(NAME test_list_04        LABELS cpython llvm c)
 RUN(NAME test_list_05        LABELS cpython llvm)
 RUN(NAME test_list_06        LABELS cpython llvm)
 RUN(NAME test_list_07        LABELS cpython llvm)

src/libasr/codegen/asr_to_c_cpp.h

@@ -127,6 +127,8 @@ class CList {
             resize_if_needed(list_struct_type, list_type_code, list_element_type);
             list_append(list_struct_type, list_type_code, list_element_type, list_type->m_type);
             list_insert(list_struct_type, list_type_code, list_element_type, list_type->m_type);
+            list_find_item_position(list_struct_type, list_type_code, list_element_type, list_type->m_type);
+            list_remove(list_struct_type, list_type_code, list_element_type, list_type->m_type);
             return list_struct_type;
         }
 
@@ -154,6 +156,15 @@ class CList {
             return typecode2listfuncs[list_type_code]["list_resize"];
         }
 
+        std::string get_list_remove_func(ASR::List_t* list_type) {
+            std::string list_type_code = ASRUtils::get_type_code(list_type->m_type, true);
+            return typecode2listfuncs[list_type_code]["list_remove"];
+        }
+
+        std::string get_list_find_item_position_function(std::string list_type_code) {
+            return typecode2listfuncs[list_type_code]["list_find_item"];
+        }
+
         std::string get_generated_code() {
             return generated_code;
         }
@@ -274,6 +285,51 @@ class CList {
             generated_code += indent + "}\n\n";
         }
 
+        void list_find_item_position(std::string list_struct_type,
+            std::string list_type_code, std::string list_element_type,
+            ASR::ttype_t* /*m_type*/) {
+            std::string indent(indentation_level * indentation_spaces, ' ');
+            std::string tab(indentation_spaces, ' ');
+            std::string list_find_item_pos_func = global_scope->get_unique_name("list_find_item_" + list_type_code);
+            typecode2listfuncs[list_type_code]["list_find_item"] = list_find_item_pos_func;
+            std::string signature = "int " + list_find_item_pos_func + "("
+                                + list_struct_type + "* x, "
+                                + list_element_type + " element)";
+            list_func_decls += signature + ";\n";
+            generated_code += indent + signature + " {\n";
+            generated_code += indent + tab + "int el_pos = 0;\n";
+            generated_code += indent + tab + "while(x->current_end_point > el_pos) {;\n";
+            generated_code += indent + tab + tab + "if (x->data[el_pos] == element) return el_pos;\n";
+            generated_code += indent + tab + tab + "el_pos++;\n";
+            generated_code += indent + tab + "}\n";
+            generated_code += indent + tab + "return -1;\n";
+            generated_code += indent + "}\n\n";
+        }
+
+        void list_remove(std::string list_struct_type,
+            std::string list_type_code, std::string list_element_type,
+            ASR::ttype_t* /*m_type*/) {
+            std::string indent(indentation_level * indentation_spaces, ' ');
+            std::string tab(indentation_spaces, ' ');
+            std::string list_remove_func = global_scope->get_unique_name("list_remove_" + list_type_code);
+            typecode2listfuncs[list_type_code]["list_remove"] = list_remove_func;
+            std::string signature = "void " + list_remove_func + "("
+                                + list_struct_type + "* x, "
+                                + list_element_type + " element)";
+            list_func_decls += signature + ";\n";
+            generated_code += indent + signature + " {\n";
+            std::string find_item_pos_func = get_list_find_item_position_function(list_type_code);
+            generated_code += indent + tab + "int el_pos = " + find_item_pos_func + "(x, element);\n";
+            generated_code += indent + tab + "while(x->current_end_point > el_pos) {;\n";
+            generated_code += indent + tab + tab + "int tmp = el_pos + 1;\n";
+            generated_code += indent + tab + tab + "x->data[el_pos] = x->data[tmp];\n";
+            generated_code += indent + tab + tab + "el_pos = tmp;\n";
+            generated_code += indent + tab + "}\n";
+
+            generated_code += indent + tab + "x->current_end_point -= 1;\n";
+            generated_code += indent + "}\n\n";
+        }
+
         ~CList() {
             typecode2listtype.clear();
             generated_code.clear();
@@ -883,6 +939,18 @@ R"(#include <stdio.h>
         src = indent + list_insert_func + "(&" + list_var + ", " + pos + ", " + element + ");\n";
     }
 
+    void visit_ListRemove(const ASR::ListRemove_t& x) {
+        ASR::ttype_t* t_ttype = ASRUtils::expr_type(x.m_a);
+        ASR::List_t* t = ASR::down_cast<ASR::List_t>(t_ttype);
+        std::string list_remove_func = list_api->get_list_remove_func(t);
+        self().visit_expr(*x.m_a);
+        std::string list_var = std::move(src);
+        self().visit_expr(*x.m_ele);
+        std::string element = std::move(src);
+        std::string indent(indentation_level * indentation_spaces, ' ');
+        src = indent + list_remove_func + "(&" + list_var + ", " + element + ");\n";
+    }
+
     void visit_ListLen(const ASR::ListLen_t& x) {
         if( x.m_value ) {
             self().visit_expr(*x.m_value);