mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 21:27:52 -04:00
40 lines
1.5 KiB
C++
40 lines
1.5 KiB
C++
#include "anonymous_function_def.h"
|
|
#include "common/goos/PrettyPrinter.h"
|
|
#include "decompiler/Function/Function.h"
|
|
#include "decompiler/ObjectFile/LinkedObjectFile.h"
|
|
#include "decompiler/analysis/final_output.h"
|
|
|
|
namespace decompiler {
|
|
int insert_anonymous_functions(Form* top_level_form,
|
|
FormPool& pool,
|
|
const Function& function,
|
|
const DecompilerTypeSystem&) {
|
|
int replaced = 0;
|
|
top_level_form->apply_form([&](Form* f) {
|
|
auto atom = form_as_atom(f);
|
|
if (atom && atom->is_static_addr()) {
|
|
auto lab = function.ir2.env.file->labels.at(atom->label());
|
|
auto& env = function.ir2.env;
|
|
auto label_kv = env.label_types().find(lab.name);
|
|
if (label_kv != env.label_types().end()) {
|
|
if (label_kv->second.type_name == "_lambda_") {
|
|
auto& file = env.file;
|
|
auto other_func = file->try_get_function_at_label(atom->label());
|
|
if (other_func) {
|
|
std::vector<goos::Object> inline_body;
|
|
other_func->ir2.top_form->inline_forms(inline_body, other_func->ir2.env);
|
|
auto result = pretty_print::build_list(
|
|
"lambda", get_arg_list_for_function(*other_func, other_func->ir2.env));
|
|
pretty_print::append(result, pretty_print::build_list(inline_body));
|
|
|
|
f->clear();
|
|
f->push_back(pool.alloc_element<LambdaDefinitionElement>(result));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
});
|
|
return replaced;
|
|
}
|
|
} // namespace decompiler
|