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

Add: emit expr of set

Browse source
This commit is contained in:
GreasySlug 2022-10-05 13:15:50 +09:00
parent bb96853cbe
commit bdbfcbd846
1 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
compiler/erg_compiler/codegen.rs
View file

@ -217,6 +217,7 @@ fn is_fake_method(class: &str, name: &str) -> bool {
fn convert_to_python_attr(class: &str, uniq_obj_name: Option<&str>, name: Str) -> Str { fn convert_to_python_attr(class: &str, uniq_obj_name: Option<&str>, name: Str) -> Str {
match (class, uniq_obj_name, &name[..]) { match (class, uniq_obj_name, &name[..]) {
("Array!", _, "push!") => Str::ever("append"), ("Array!", _, "push!") => Str::ever("append"),
("Set!", _, "add") => Str::ever("add"),
("Complex" | "Float" | "Ratio" | "Int" | "Nat" | "Bool", _, "Real") => Str::ever("real"), ("Complex" | "Float" | "Ratio" | "Int" | "Nat" | "Bool", _, "Real") => Str::ever("real"),
("Complex" | "Float" | "Ratio" | "Int" | "Nat" | "Bool", _, "Imag") => Str::ever("imag"), ("Complex" | "Float" | "Ratio" | "Int" | "Nat" | "Bool", _, "Imag") => Str::ever("imag"),
("File!", _, "read!") => Str::ever("read"), ("File!", _, "read!") => Str::ever("read"),
@ -273,6 +274,7 @@ fn convert_to_python_name(name: Str) -> Str {
"Str" | "Str!" => Str::ever("str"), "Str" | "Str!" => Str::ever("str"),
"Bool" | "Bool!" => Str::ever("bool"), "Bool" | "Bool!" => Str::ever("bool"),
"Array" | "Array!" => Str::ever("list"), "Array" | "Array!" => Str::ever("list"),
"Set" | "Set!" => Str::ever("set"),
_ => name, _ => name,
} }
} }
@ -1692,6 +1694,22 @@ impl CodeGenerator {
} }
} }
}, },
Expr::Set(set) => match set {
crate::hir::Set::Normal(mut set) => {
let len = set.elems.len();
while let Some(arg) = set.elems.try_remove_pos(0) {
self.emit_expr(arg.expr);
}
self.write_instr(BUILD_SET);
self.write_arg(len as u8);
if len == 0 {
self.stack_inc();
} else {
self.stack_dec_n(len - 1);
}
}
crate::hir::Set::WithLength(_) => todo!(),
},
Expr::Record(rec) => self.emit_record(rec), Expr::Record(rec) => self.emit_record(rec),
Expr::Code(code) => { Expr::Code(code) => {
let code = self.emit_block(code, None, vec![]); let code = self.emit_block(code, None, vec![]);