File: if.go

package info (click to toggle)
mumax3 3.10-9
  • links: PTS, VCS
  • area: contrib
  • in suites: trixie
  • size: 7,596 kB
  • sloc: makefile: 181; ansic: 155; sh: 77
file content (44 lines) | stat: -rw-r--r-- 700 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package script

import (
	"go/ast"
)

// if statement
type ifStmt struct {
	cond, body, else_ Expr
	void
}

func (b *ifStmt) Eval() interface{} {
	if b.cond.Eval().(bool) {
		b.body.Eval()
	} else {
		if b.else_ != nil {
			b.else_.Eval()
		}
	}
	return nil // void
}

func (w *World) compileIfStmt(n *ast.IfStmt) *ifStmt {
	w.EnterScope()
	defer w.ExitScope()

	stmt := &ifStmt{
		cond: typeConv(n.Cond.Pos(), w.compileExpr(n.Cond), bool_t),
		body: w.compileBlockStmt_noScope(n.Body)}
	if n.Else != nil {
		stmt.else_ = w.compileStmt(n.Else)
	}

	return stmt
}

func (e *ifStmt) Child() []Expr {
	child := []Expr{e.cond, e.body, e.else_}
	if e.else_ == nil {
		child = child[:2]
	}
	return child
}