language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-09-29 12:24:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement fake methods code generating

Browse source
This commit is contained in:
Shunsuke Shibayama 2022-09-11 21:36:02 +09:00
parent 670a16570a
commit f38bc88073
2 changed files with 30 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

29
compiler/erg_compiler/codegen.rs
View file

@ -201,6 +201,17 @@ fn is_python_global(name: &str) -> bool {
)
}
fn is_fake_method(class: &str, name: &str) -> bool {
matches!(
(class, name),
(
"Complex" | "Float" | "Ratio" | "Int" | "Nat" | "Bool",
"abs"
) | (_, "iter")
| (_, "map")
)
}
fn convert_to_python_attr(class: &str, uniq_obj_name: Option<&str>, name: Str) -> Str {
match (class, uniq_obj_name, &name[..]) {
("Array!", _, "push!") => Str::ever("append"),
@ -1028,11 +1039,14 @@ impl CodeGenerator {
fn emit_call_method(&mut self, obj: Expr, method_name: Identifier, args: Args) {
log!(info "entered {}", fn_name!());
if &method_name.inspect()[..] == "update!" {
return self.emit_call_update(obj, args);
}
let class = obj.ref_t().name(); // これは必ずmethodのあるクラスになっている
let uniq_obj_name = obj.__name__().map(Str::rc);
log!("{class} {method_name}");
if &method_name.inspect()[..] == "update!" {
return self.emit_call_update(obj, args);
} else if is_fake_method(&class, method_name.inspect()) {
return self.emit_call_fake_method(obj, method_name, args);
}
self.codegen_expr(obj);
self.emit_load_method_instr(&class, uniq_obj_name.as_ref().map(|s| &s[..]), method_name)
.unwrap_or_else(|err| {
@ -1109,6 +1123,15 @@ impl CodeGenerator {
self.store_acc(acc);
}
/// 1.abs() => abs(1)
fn emit_call_fake_method(&mut self, obj: Expr, mut method_name: Identifier, mut args: Args) {
log!(info "entered {}", fn_name!());
method_name.dot = None;
self.emit_load_name_instr(method_name).unwrap();
args.insert_pos(0, PosArg::new(obj));
self.emit_args(args, Name);
}
// assert takes 1 or 2 arguments (0: cond, 1: message)
fn emit_assert_instr(&mut self, mut args: Args) -> CompileResult<()> {
log!(info "entered {}", fn_name!());