1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

cgen: sum type: it variable

This commit is contained in:
Alexander Medvednikov 2020-03-16 03:56:38 +01:00
parent 37504c487a
commit ac0b235d3a
2 changed files with 73 additions and 54 deletions

View File

@ -702,54 +702,7 @@ fn (g mut Gen) expr(node ast.Expr) {
g.write(it.val.str()) g.write(it.val.str())
} }
ast.MatchExpr { ast.MatchExpr {
// println('match expr typ=$it.expr_type') g.match_expr(it)
// TODO
if it.expr_type == 0 {
g.writeln('// match 0')
return
}
type_sym := g.table.get_type_symbol(it.expr_type)
mut tmp := ''
if type_sym.kind != .void {
tmp = g.new_tmp_var()
}
styp := g.typ(it.expr_type)
g.write('$styp $tmp = ')
g.expr(it.cond)
g.writeln(';') // $it.blocks.len')
for j, branch in it.branches {
if j == it.branches.len - 1 {
// last block is an `else{}`
g.writeln('else {')
}
else {
if j > 0 {
g.write('else ')
}
g.write('if (')
for i, expr in branch.exprs {
if it.is_sum_type {
g.write('${tmp}.typ == ')
}
else if type_sym.kind == .string {
g.write('string_eq($tmp, ')
}
else {
g.write('$tmp == ')
}
g.expr(expr)
if type_sym.kind == .string {
g.write(')')
}
if i < branch.exprs.len - 1 {
g.write(' || ')
}
}
g.writeln(') {')
}
g.stmts(branch.stmts)
g.writeln('}')
}
} }
ast.MapInit { ast.MapInit {
key_typ_sym := g.table.get_type_symbol(it.key_type) key_typ_sym := g.table.get_type_symbol(it.key_type)
@ -1006,6 +959,73 @@ fn (g mut Gen) infix_expr(node ast.InfixExpr) {
} }
} }
fn (g mut Gen) match_expr(node ast.MatchExpr) {
// println('match expr typ=$it.expr_type')
// TODO
if node.expr_type == 0 {
g.writeln('// match 0')
return
}
type_sym := g.table.get_type_symbol(node.expr_type)
mut tmp := ''
if type_sym.kind != .void {
tmp = g.new_tmp_var()
}
styp := g.typ(node.expr_type)
g.write('$styp $tmp = ')
g.expr(node.cond)
g.writeln(';') // $it.blocks.len')
// mut sum_type_str = ''
for j, branch in node.branches {
if j == node.branches.len - 1 {
// last block is an `else{}`
g.writeln('else {')
}
else {
if j > 0 {
g.write('else ')
}
g.write('if (')
for i, expr in branch.exprs {
if node.is_sum_type {
g.write('${tmp}.typ == ')
// sum_type_str
}
else if type_sym.kind == .string {
g.write('string_eq($tmp, ')
}
else {
g.write('$tmp == ')
}
g.expr(expr)
if type_sym.kind == .string {
g.write(')')
}
if i < branch.exprs.len - 1 {
g.write(' || ')
}
}
g.writeln(') {')
}
if node.is_sum_type && branch.exprs.len > 0 {
// The first node in expr is an ast.Type
// Use it to generate `it` variable.
fe := branch.exprs[0]
match fe {
ast.Type {
it_type := g.typ(it.typ)
g.writeln('$it_type* it = ($it_type*)tmp3.obj; // ST it')
}
else {
verror('match sum type')
}
}
}
g.stmts(branch.stmts)
g.writeln('}')
}
}
fn (g mut Gen) ident(node ast.Ident) { fn (g mut Gen) ident(node ast.Ident) {
name := node.name.replace('.', '__') name := node.name.replace('.', '__')
if name.starts_with('C__') { if name.starts_with('C__') {

View File

@ -41,16 +41,15 @@ void println(string s) {
void handle_expr(Expr e) { void handle_expr(Expr e) {
Expr tmp1 = e; Expr tmp1 = e;
if (tmp1.typ == _type_idx_IfExpr) { if (tmp1.typ == _type_idx_IfExpr) {
println(tos3("if")); IfExpr* it = (IfExpr*)tmp3.obj; // ST it
println(tos3("if"));
} }
else if (tmp1.typ == _type_idx_IntegerLiteral) { else if (tmp1.typ == _type_idx_IntegerLiteral) {
println(tos3("integer")); IntegerLiteral* it = (IntegerLiteral*)tmp3.obj; // ST it
println(tos3("integer"));
} }
else { else {
println(tos3("else")); println(tos3("else"));
} }
; ;
} }