mirror/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
Code Issues Projects Releases Wiki Activity

cgen: sum type: it variable

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

116
vlib/v/gen/cgen.v
View File

@ -702,54 +702,7 @@ fn (g mut Gen) expr(node ast.Expr) {
g.write(it.val.str())
}
ast.MatchExpr {
// println('match expr typ=$it.expr_type')
// 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('}')
}
g.match_expr(it)
}
ast.MapInit {
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) {
name := node.name.replace('.', '__')
if name.starts_with('C__') {

11
vlib/v/gen/tests/3.c
View File

@ -41,16 +41,15 @@ void println(string s) {
void handle_expr(Expr e) {
Expr tmp1 = e;
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) {
println(tos3("integer"));
IntegerLiteral* it = (IntegerLiteral*)tmp3.obj; // ST it
println(tos3("integer"));
}
else {
println(tos3("else"));
println(tos3("else"));
}
;
}