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

Update transpiler

Browse source
This commit is contained in:
Shunsuke Shibayama 2022-12-02 23:56:39 +09:00
parent b433a703dc
commit 5e29de9b53
3 changed files with 203 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

18
compiler/erg_compiler/codegen.rs
View file

@ -23,9 +23,7 @@ use erg_common::{
debug_power_assert, enum_unwrap, fn_name, fn_name_full, impl_stream_for_wrapper, log,
switch_unreachable,
};
use erg_parser::ast::ConstExpr;
use erg_parser::ast::DefId;
use erg_parser::ast::DefKind;
use erg_parser::ast::{DefId, DefKind};
use CommonOpcode::*;
use erg_parser::ast::PreDeclTypeSpec;
@ -36,7 +34,6 @@ use erg_parser::token::EQUAL;
use erg_parser::token::{Token, TokenKind};
use crate::compile::{AccessKind, Name, StoreLoadKind};
use crate::context::eval::type_from_token_kind;
use crate::error::CompileError;
use crate::hir::{
Accessor, Args, Array, AttrDef, BinOp, Block, Call, ClassDef, Def, DefBody, Expr, Identifier,
@ -1741,16 +1738,7 @@ impl PyCodeGenerator {
#[allow(clippy::single_match)]
match t_spec {
TypeSpec::Enum(enum_t) => {
let elems = enum_t
.deconstruct()
.0
.into_iter()
.map(|elem| {
let ConstExpr::Lit(lit) = elem.expr else { todo!() };
let t = type_from_token_kind(lit.token.kind);
ValueObj::from_str(t, lit.token.content).unwrap()
})
.collect::<Vec<_>>();
let elems = ValueObj::vec_from_const_args(enum_t);
self.emit_load_const(elems);
self.write_instr(CONTAINS_OP);
self.write_arg(0);
@ -2249,7 +2237,7 @@ impl PyCodeGenerator {
debug_assert_eq!(self.stack_len(), init_stack_len + 1);
}
fn get_root(acc: &Accessor) -> Identifier {
pub(crate) fn get_root(acc: &Accessor) -> Identifier {
match acc {
Accessor::Ident(ident) => ident.clone(),
Accessor::Attr(attr) => {