mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
ast/cgen/parser: combine IfExpr branches into single node
This commit is contained in:
parent
c993489fe9
commit
a331abf675
@ -336,16 +336,21 @@ mut:
|
|||||||
|
|
||||||
pub struct IfExpr {
|
pub struct IfExpr {
|
||||||
pub:
|
pub:
|
||||||
tok_kind token.Kind
|
tok_kind token.Kind
|
||||||
cond Expr
|
branches []IfBranch
|
||||||
stmts []Stmt
|
left Expr // `a` in `a := if ...`
|
||||||
else_stmts []Stmt
|
pos token.Position
|
||||||
left Expr // `a` in `a := if ...`
|
|
||||||
pos token.Position
|
|
||||||
mut:
|
mut:
|
||||||
is_expr bool
|
is_expr bool
|
||||||
typ table.Type
|
typ table.Type
|
||||||
has_else bool
|
has_else bool
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct IfBranch {
|
||||||
|
pub:
|
||||||
|
cond Expr
|
||||||
|
stmts []Stmt
|
||||||
|
pos token.Position
|
||||||
}
|
}
|
||||||
|
|
||||||
pub struct MatchExpr {
|
pub struct MatchExpr {
|
||||||
|
@ -909,32 +909,34 @@ pub fn (c mut Checker) if_expr(node mut ast.IfExpr) table.Type {
|
|||||||
// println('$c.file.path $node.pos.line_nr IF: checker exp type = ' + sym.name)
|
// println('$c.file.path $node.pos.line_nr IF: checker exp type = ' + sym.name)
|
||||||
node.is_expr = true
|
node.is_expr = true
|
||||||
}
|
}
|
||||||
typ := c.expr(node.cond)
|
|
||||||
node.typ = table.void_type
|
node.typ = table.void_type
|
||||||
// node.typ = typ
|
for i, branch in node.branches {
|
||||||
typ_sym := c.table.get_type_symbol(typ)
|
typ := c.expr(branch.cond)
|
||||||
// if typ_sym.kind != .bool {
|
if i < node.branches.len-1 || !node.has_else {
|
||||||
if table.type_idx(typ) != table.bool_type_idx {
|
typ_sym := c.table.get_type_symbol(typ)
|
||||||
c.error('non-bool (`$typ_sym.name`) used as if condition', node.pos)
|
// if typ_sym.kind != .bool {
|
||||||
}
|
if table.type_idx(typ) != table.bool_type_idx {
|
||||||
c.stmts(node.stmts)
|
c.error('non-bool (`$typ_sym.name`) used as if condition', node.pos)
|
||||||
if node.else_stmts.len > 0 {
|
|
||||||
c.stmts(node.else_stmts)
|
|
||||||
}
|
|
||||||
if node.stmts.len > 0 {
|
|
||||||
match node.stmts[node.stmts.len - 1] {
|
|
||||||
ast.ExprStmt {
|
|
||||||
// type_sym := p.table.get_type_symbol(it.typ)
|
|
||||||
// p.warn('if expr ret $type_sym.name')
|
|
||||||
t := c.expr(it.expr)
|
|
||||||
node.typ = t
|
|
||||||
return t
|
|
||||||
}
|
}
|
||||||
else {}
|
}
|
||||||
|
c.stmts(branch.stmts)
|
||||||
}
|
}
|
||||||
|
if node.has_else && node.is_expr {
|
||||||
|
last_branch := node.branches[node.branches.len-1]
|
||||||
|
if last_branch.stmts.len > 0 {
|
||||||
|
match last_branch.stmts[last_branch.stmts.len - 1] {
|
||||||
|
ast.ExprStmt {
|
||||||
|
// type_sym := p.table.get_type_symbol(it.typ)
|
||||||
|
// p.warn('if expr ret $type_sym.name')
|
||||||
|
t := c.expr(it.expr)
|
||||||
|
node.typ = t
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
else {}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return typ
|
return table.bool_type
|
||||||
// return table.void_type
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn (c mut Checker) postfix_expr(node ast.PostfixExpr) table.Type {
|
pub fn (c mut Checker) postfix_expr(node ast.PostfixExpr) table.Type {
|
||||||
|
@ -378,38 +378,38 @@ fn (f mut Fmt) expr(node ast.Expr) {
|
|||||||
f.write(it.val)
|
f.write(it.val)
|
||||||
}
|
}
|
||||||
ast.IfExpr {
|
ast.IfExpr {
|
||||||
single_line := it.stmts.len == 1 && it.else_stmts.len == 1 && it.typ != table.void_type
|
single_line := it.branches.len == 2 && it.has_else //
|
||||||
|
&& it.branches[0].stmts.len == 1 && it.branches[1].stmts.len == 1
|
||||||
f.single_line_if = single_line
|
f.single_line_if = single_line
|
||||||
f.write('if ')
|
for i, branch in it.branches {
|
||||||
f.expr(it.cond)
|
if i == 0 {
|
||||||
if single_line {
|
f.write('if ')
|
||||||
f.write(' { ')
|
f.expr(branch.cond)
|
||||||
}
|
f.write(' {')
|
||||||
else {
|
}
|
||||||
f.writeln(' {')
|
else if i < it.branches.len-1 || !it.has_else {
|
||||||
}
|
f.write('} else if ')
|
||||||
f.stmts(it.stmts)
|
f.expr(branch.cond)
|
||||||
if single_line {
|
f.write(' {')
|
||||||
f.write(' ')
|
}
|
||||||
}
|
else if i == it.branches.len-1 && it.has_else {
|
||||||
f.write('}')
|
f.write('} else {')
|
||||||
if it.has_else {
|
}
|
||||||
f.write(' else ')
|
|
||||||
}
|
|
||||||
else if it.else_stmts.len > 0 {
|
|
||||||
f.write(' else {')
|
|
||||||
if single_line {
|
if single_line {
|
||||||
f.write(' ')
|
f.write(' ')
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
f.writeln('')
|
f.writeln('')
|
||||||
}
|
}
|
||||||
f.stmts(it.else_stmts)
|
f.stmts(branch.stmts)
|
||||||
if single_line {
|
if single_line {
|
||||||
f.write(' ')
|
f.write(' ')
|
||||||
}
|
}
|
||||||
f.write('}')
|
|
||||||
}
|
}
|
||||||
|
// if !single_line {
|
||||||
|
// f.writeln('')
|
||||||
|
// }
|
||||||
|
f.write('}')
|
||||||
f.single_line_if = false
|
f.single_line_if = false
|
||||||
}
|
}
|
||||||
ast.Ident {
|
ast.Ident {
|
||||||
|
@ -1176,64 +1176,68 @@ fn (g mut Gen) if_expr(node ast.IfExpr) {
|
|||||||
}
|
}
|
||||||
// one line ?:
|
// one line ?:
|
||||||
// TODO clean this up once `is` is supported
|
// TODO clean this up once `is` is supported
|
||||||
if node.is_expr && node.stmts.len == 1 && node.else_stmts.len == 1 && type_sym.kind != .void {
|
// TODO: make sure only one stmt in eac branch
|
||||||
cond := node.cond
|
if node.is_expr && node.branches.len >= 2 && node.has_else && type_sym.kind != .void {
|
||||||
stmt1 := node.stmts[0]
|
g.inside_ternary = true
|
||||||
else_stmt1 := node.else_stmts[0]
|
for i, branch in node.branches {
|
||||||
match stmt1 {
|
if i > 0 {
|
||||||
ast.ExprStmt {
|
|
||||||
g.inside_ternary = true
|
|
||||||
g.expr(cond)
|
|
||||||
g.write(' ? ')
|
|
||||||
expr_stmt := stmt1 as ast.ExprStmt
|
|
||||||
g.expr(expr_stmt.expr)
|
|
||||||
g.write(' : ')
|
g.write(' : ')
|
||||||
g.stmt(else_stmt1)
|
|
||||||
}
|
}
|
||||||
else {}
|
if i < node.branches.len-1 || !node.has_else {
|
||||||
}
|
g.expr(branch.cond)
|
||||||
|
g.write(' ? ')
|
||||||
|
}
|
||||||
|
g.stmts(branch.stmts)
|
||||||
|
}
|
||||||
|
g.inside_ternary = false
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
mut is_guard := false
|
|
||||||
guard_ok := g.new_tmp_var()
|
guard_ok := g.new_tmp_var()
|
||||||
match node.cond {
|
mut is_guard := false
|
||||||
ast.IfGuardExpr {
|
for i, branch in node.branches {
|
||||||
is_guard = true
|
if i == 0 {
|
||||||
g.writeln('bool $guard_ok;')
|
match branch.cond {
|
||||||
g.write('{ /* if guard */ ${g.typ(it.expr_type)} $it.var_name = ')
|
ast.IfGuardExpr {
|
||||||
g.expr(it.expr)
|
is_guard = true
|
||||||
g.writeln(';')
|
g.writeln('bool $guard_ok;')
|
||||||
g.writeln('if (($guard_ok = ${it.var_name}.ok)) {')
|
g.write('{ /* if guard */ ${g.typ(it.expr_type)} $it.var_name = ')
|
||||||
|
g.expr(it.expr)
|
||||||
|
g.writeln(';')
|
||||||
|
g.writeln('if (($guard_ok = ${it.var_name}.ok)) {')
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
g.write('if (')
|
||||||
|
g.expr(branch.cond)
|
||||||
|
g.writeln(') {')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
else {
|
else if i < node.branches.len-1 || !node.has_else {
|
||||||
g.inside_ternary = false
|
g.writeln('} else if (')
|
||||||
g.write('if (')
|
g.expr(branch.cond)
|
||||||
g.expr(node.cond)
|
g.write(') {')
|
||||||
g.writeln(') {')
|
}
|
||||||
|
else if i == node.branches.len-1 && node.has_else {
|
||||||
|
if is_guard {
|
||||||
|
g.writeln('} if (!$guard_ok) { /* else */')
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
g.writeln('} else {')
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
for i, stmt in node.stmts {
|
|
||||||
// Assign ret value
|
// Assign ret value
|
||||||
if i == node.stmts.len - 1 && type_sym.kind != .void {}
|
// if i == node.stmts.len - 1 && type_sym.kind != .void {}
|
||||||
// g.writeln('$tmp =')
|
// g.writeln('$tmp =')
|
||||||
g.stmt(stmt)
|
//
|
||||||
|
g.stmts(branch.stmts)
|
||||||
|
// g.writeln('')
|
||||||
}
|
}
|
||||||
if is_guard {
|
if is_guard {
|
||||||
g.write('}')
|
g.write('}')
|
||||||
}
|
}
|
||||||
g.writeln('}')
|
g.writeln('}')
|
||||||
if node.else_stmts.len > 0 {
|
|
||||||
if is_guard {
|
|
||||||
g.writeln('if (!$guard_ok) { /* else */')
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
g.writeln('else { ')
|
|
||||||
}
|
|
||||||
g.stmts(node.else_stmts)
|
|
||||||
g.writeln('}')
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
g.inside_ternary = false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (g mut Gen) index_expr(node ast.IndexExpr) {
|
fn (g mut Gen) index_expr(node ast.IndexExpr) {
|
||||||
|
@ -34,6 +34,12 @@ pub fn (g mut JsGen) writeln(s string) {
|
|||||||
g.out.writeln(s)
|
g.out.writeln(s)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn (g mut JsGen) stmts(stmts []ast.Stmt) {
|
||||||
|
for stmt in stmts {
|
||||||
|
g.stmt(stmt)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn (g mut JsGen) stmt(node ast.Stmt) {
|
fn (g mut JsGen) stmt(node ast.Stmt) {
|
||||||
match node {
|
match node {
|
||||||
ast.FnDecl {
|
ast.FnDecl {
|
||||||
@ -163,13 +169,23 @@ fn (g mut JsGen) expr(node ast.Expr) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
ast.IfExpr {
|
ast.IfExpr {
|
||||||
g.write('if (')
|
for i, branch in it.branches {
|
||||||
g.expr(it.cond)
|
if i == 0 {
|
||||||
g.writeln(') {')
|
g.write('if (')
|
||||||
for stmt in it.stmts {
|
g.expr(branch.cond)
|
||||||
g.stmt(stmt)
|
g.writeln(') {')
|
||||||
|
}
|
||||||
|
else if i < it.branches.len-1 || !it.has_else {
|
||||||
|
g.write('else if (')
|
||||||
|
g.expr(branch.cond)
|
||||||
|
g.writeln(') {')
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
g.write('else {')
|
||||||
|
}
|
||||||
|
g.stmts(branch.stmts)
|
||||||
|
g.writeln('}')
|
||||||
}
|
}
|
||||||
g.writeln('}')
|
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
println(term.red('jsgen.expr(): bad node'))
|
println(term.red('jsgen.expr(): bad node'))
|
||||||
|
@ -52,7 +52,7 @@ void function2() {
|
|||||||
}
|
}
|
||||||
if (false) {
|
if (false) {
|
||||||
foo(1);
|
foo(1);
|
||||||
} else {
|
} else {
|
||||||
puts(tos3("else"));
|
puts(tos3("else"));
|
||||||
foo(100);
|
foo(100);
|
||||||
}
|
}
|
||||||
|
@ -1175,82 +1175,70 @@ fn (p mut Parser) for_statement() ast.Stmt {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (p mut Parser) if_expr() ast.Expr {
|
fn (p mut Parser) if_expr() ast.IfExpr {
|
||||||
p.inside_if = true
|
p.inside_if = true
|
||||||
// defer {
|
|
||||||
// }
|
|
||||||
mut node := ast.Expr{}
|
|
||||||
p.check(.key_if)
|
|
||||||
pos := p.tok.position()
|
pos := p.tok.position()
|
||||||
// `if x := opt() {`
|
mut branches := []ast.IfBranch
|
||||||
mut cond := ast.Expr{}
|
|
||||||
mut is_or := false
|
|
||||||
if p.peek_tok.kind == .decl_assign {
|
|
||||||
is_or = true
|
|
||||||
p.open_scope()
|
|
||||||
var_name := p.check_name()
|
|
||||||
p.check(.decl_assign)
|
|
||||||
expr := p.expr(0)
|
|
||||||
p.scope.register_var(ast.Var{
|
|
||||||
name: var_name
|
|
||||||
expr: expr
|
|
||||||
})
|
|
||||||
cond = ast.IfGuardExpr{
|
|
||||||
var_name: var_name
|
|
||||||
expr: expr
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
cond = p.expr(0)
|
|
||||||
}
|
|
||||||
p.inside_if = false
|
|
||||||
mut has_else := false
|
mut has_else := false
|
||||||
stmts := p.parse_block()
|
for p.tok.kind in [.key_if, .key_else] {
|
||||||
mut else_stmts := []ast.Stmt
|
branch_pos := p.tok.position()
|
||||||
if p.tok.kind == .key_else {
|
|
||||||
p.check(.key_else)
|
|
||||||
if p.tok.kind == .key_if {
|
if p.tok.kind == .key_if {
|
||||||
// The next if block is handled by next if_expr()
|
p.check(.key_if)
|
||||||
has_else = true
|
|
||||||
}
|
}
|
||||||
// p.if_expr()
|
|
||||||
else {
|
else {
|
||||||
else_stmts = p.parse_block()
|
p.check(.key_else)
|
||||||
|
if p.tok.kind == .key_if {
|
||||||
|
p.check(.key_if)
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
has_else = true
|
||||||
|
branches << ast.IfBranch{
|
||||||
|
stmts: p.parse_block()
|
||||||
|
pos: branch_pos
|
||||||
|
}
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mut cond := ast.Expr{}
|
||||||
|
mut is_or := false
|
||||||
|
// `if x := opt() {`
|
||||||
|
if p.peek_tok.kind == .decl_assign {
|
||||||
|
is_or = true
|
||||||
|
p.open_scope()
|
||||||
|
var_name := p.check_name()
|
||||||
|
p.check(.decl_assign)
|
||||||
|
expr := p.expr(0)
|
||||||
|
p.scope.register_var(ast.Var{
|
||||||
|
name: var_name
|
||||||
|
expr: expr
|
||||||
|
})
|
||||||
|
cond = ast.IfGuardExpr{
|
||||||
|
var_name: var_name
|
||||||
|
expr: expr
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
cond = p.expr(0)
|
||||||
|
}
|
||||||
|
p.inside_if = false
|
||||||
|
stmts := p.parse_block()
|
||||||
|
if is_or {
|
||||||
|
p.close_scope()
|
||||||
|
}
|
||||||
|
branches << ast.IfBranch{
|
||||||
|
cond: cond
|
||||||
|
stmts: stmts
|
||||||
|
pos: branch_pos
|
||||||
|
}
|
||||||
|
if p.tok.kind != .key_else {
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if is_or {
|
return ast.IfExpr{
|
||||||
p.close_scope()
|
branches: branches
|
||||||
}
|
|
||||||
// mut typ := table.void_type
|
|
||||||
// mut left := ast.Expr{}
|
|
||||||
// If the last statement is an expression, return its type
|
|
||||||
/*
|
|
||||||
if stmts.len > 0 {
|
|
||||||
match stmts[stmts.len - 1] {
|
|
||||||
ast.ExprStmt {
|
|
||||||
type_sym := p.table.get_type_symbol(it.typ)
|
|
||||||
p.warn('if expr ret $type_sym.name')
|
|
||||||
typ = it.typ
|
|
||||||
// return node,it.ti
|
|
||||||
// left =
|
|
||||||
}
|
|
||||||
else {}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
node = ast.IfExpr{
|
|
||||||
cond: cond
|
|
||||||
stmts: stmts
|
|
||||||
else_stmts: else_stmts
|
|
||||||
// typ: typ
|
|
||||||
|
|
||||||
pos: pos
|
pos: pos
|
||||||
has_else: has_else
|
has_else: has_else
|
||||||
// left: left
|
|
||||||
|
|
||||||
}
|
}
|
||||||
return node
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (p mut Parser) string_expr() ast.Expr {
|
fn (p mut Parser) string_expr() ast.Expr {
|
||||||
|
Loading…
Reference in New Issue
Block a user