replace lir find call with 2 lir::store

Author: ValKmjolnir

godel-script/godel-frontend/src/ir/inst_combine.cpp

@@ -367,4 +367,69 @@ void inst_elimination_worker::copy(souffle_rule_impl* impl) {
     delete impl_blk;
 }
 
+void replace_find_call::visit_block(lir::block* node) {
+    bool has_find_call = false;
+    for (auto i : node->get_content()) {
+        if (i->get_kind() != lir::inst_kind::inst_call) {
+            continue;
+        }
+        auto call = reinterpret_cast<lir::call*>(i);
+        if (call->get_func_kind() == lir::call::kind::find &&
+            call->get_function_name() == "find") {
+            has_find_call = true;
+            break;
+        }
+    }
+
+    if (has_find_call) {
+        std::vector<lir::inst*> new_content;
+        for (auto i : node->get_content()) {
+            if (i->get_kind() != lir::inst_kind::inst_call) {
+                new_content.push_back(i);
+                continue;
+            }
+
+            auto call = reinterpret_cast<lir::call*>(i);
+            if (call->get_func_kind() != lir::call::kind::find ||
+                call->get_function_name() != "find") {
+                new_content.push_back(i);
+                continue;
+            }
+
+            auto dst = call->get_return();
+            auto arg0 = call->get_arguments()[0];
+            auto arg1 = call->get_arguments()[1];
+            auto new_block = new lir::block(call->get_location());
+            new_block->set_use_comma();
+            new_content.push_back(new_block);
+
+            new_block->add_new_content(new lir::store(arg0, dst, call->get_location()));
+            new_block->add_new_content(new lir::store(arg1, arg0, call->get_location()));
+
+            delete i;
+        }
+        node->get_mutable_content().swap(new_content);
+    } else {
+        for (auto i : node->get_content()) {
+            i->accept(this);
+        }
+    }
+}
+
+bool replace_find_call::run() {
+    for (auto impl : ctx->rule_impls) {
+        impl->get_block()->accept(this);
+    }
+    for (auto impl : ctx->database_get_table) {
+        impl->get_block()->accept(this);
+    }
+    for (auto impl : ctx->schema_get_field) {
+        impl->get_block()->accept(this);
+    }
+    for (auto impl : ctx->schema_data_constraint_impls) {
+        impl->get_block()->accept(this);
+    }
+    return true;
+}
+
 }

godel-script/godel-frontend/src/ir/inst_combine.h

@@ -118,4 +118,16 @@ class inst_elimination_worker: public lir::inst_visitor {
     }
 };
 
+class replace_find_call: public pass {
+private:
+    void visit_block(lir::block*) override;
+
+public:
+    replace_find_call(ir_context& c): pass(pass_kind::ps_replace_find_call, c) {}
+    const char* get_name() const override {
+        return "[Transform] Replace Find Call";
+    }
+    bool run() override;
+};
+
 }

godel-script/godel-frontend/src/ir/pass.h

@@ -13,6 +13,7 @@ enum class pass_kind {
     ps_remove_unused,
     ps_remove_unused_type,
     ps_inst_combine,
+    ps_replace_find_call,
     ps_flatten_nested_block,
     ps_aggregator_inline_remark,
     ps_ungrounded_check,

godel-script/godel-frontend/src/ir/pass_manager.cpp

@@ -29,6 +29,7 @@ void pass_manager::run(ir_context& ctx, const cli::configure& conf) {
         ordered_pass_list.push_back(new unused_type_alias_remove_pass(ctx));
     }
     if (!conf.count(cli::option::cli_disable_inst_combine)) {
+        ordered_pass_list.push_back(new replace_find_call(ctx));
         ordered_pass_list.push_back(new inst_combine_pass(ctx));
     }
     ordered_pass_list.push_back(new flatten_nested_block(ctx));