Update gen_pass.py from LFortran

certik · certik · commit eedc16592eec · 2023-07-06T08:43:03.000-06:00
diff --git a/src/libasr/gen_pass.py b/src/libasr/gen_pass.py
@@ -1,62 +1,44 @@
 passes = [
-        # name, optional prefix (pass_PREFIX_name())
-        ("arr_slice", "replace"),
-        ("array_op", "replace"),
-        ("class_constructor", "replace"),
-        ("dead_code_removal", ""),
-        ("div_to_mul", "replace"),
-        ("do_loops", "replace"),
-        ("flip_sign", "replace"),
-        ("fma", "replace"),
-        ("for_all", "replace"),
-        ("global_stmts", "wrap"),
-        ("global_stmts_program", "wrap"),
-        ("global_symbols", "wrap"),
-        ("implied_do_loops", "replace"),
-        ("init_expr", "replace"),
-        ("inline_function_calls", ""),
-        ("instantiate_template", ""),
-        ("intrinsic_function", "replace"),
-        ("loop_unroll", ""),
-        ("loop_vectorise", ""),
-        ("nested_vars", ""),
-        ("param_to_const", "replace"),
-        ("pass_array_by_data", ""),
-        ("pass_compare", ""),
-        ("pass_list_expr", ""),
-        ("print_arr", "replace"),
-        ("print_list_tuple", "replace"),
-        ("print_struct_type", "replace"),
-        ("select_case", "replace"),
-        ("sign_from_value", "replace"),
-        ("subroutine_from_function", "create"),
-        ("transform_optional_argument_functions", ""),
-        ("unused_functions", ""),
-        ("update_array_dim_intrinsic_calls", ""),
-        ("where", "replace"),
+        "replace_arr_slice",
+        "replace_array_op",
+        "replace_class_constructor",
+        "dead_code_removal",
+        "replace_div_to_mul",
+        "replace_do_loops",
+        "replace_flip_sign",
+        "replace_fma",
+        "replace_for_all",
+        "wrap_global_stmts",
+        "wrap_global_stmts_program",
+        "wrap_global_symbols",
+        "replace_implied_do_loops",
+        "replace_init_expr",
+        "inline_function_calls",
+        "replace_intrinsic_function",
+        "loop_unroll",
+        "loop_vectorise",
+        "nested_vars",
+        "replace_param_to_const",
+        "array_by_data",
+        "compare",
+        "list_expr",
+        "replace_print_arr",
+        "replace_print_list_tuple",
+        "replace_print_struct_type",
+        "replace_select_case",
+        "replace_sign_from_value",
+        "create_subroutine_from_function",
+        "transform_optional_argument_functions",
+        "unused_functions",
+        "update_array_dim_intrinsic_calls",
+        "replace_where",
 ]
 
 
 
-for pass_name, prefix in passes:
-    print(f"Processing: {pass_name}")
-    name = pass_name
-    if name.startswith("pass"):
-        name = name[5:]
+for name in passes:
+    print(f"Processing: {name}")
     name_up = name.upper()
-    if prefix != "":
-        prefix += "_"
-    arguments = r"""Allocator &al, ASR::TranslationUnit_t &unit,
-                                const PassOptions &pass_options"""
-    return_type = "void"
-    if pass_name == "instantiate_template":
-        arguments = r"""Allocator &al,
-        std::map<std::string, ASR::ttype_t*> subs, std::map<std::string, ASR::symbol_t*> rt_subs,
-        SymbolTable *current_scope, std::string new_func_name, ASR::symbol_t *sym"""
-        return_type = "ASR::symbol_t*"
-
-
-
     header = rf"""#ifndef LIBASR_PASS_{name_up}_H
 #define LIBASR_PASS_{name_up}_H
 
@@ -65,12 +47,13 @@
 
 namespace LCompilers {{
 
-    {return_type} pass_{prefix}{name}({arguments});
+    void pass_{name}(Allocator &al, ASR::TranslationUnit_t &unit,
+                                const PassOptions &pass_options);
 
 }} // namespace LCompilers
 
 #endif // LIBASR_PASS_{name_up}_H
 """
-    header_filename = f"pass/{pass_name}.h"
+    header_filename = f"pass/{name}.h"
     f = open(header_filename, "w")
     f.write(header)
