aboutsummaryrefslogtreecommitdiff
path: root/go/src/step2_eval
diff options
context:
space:
mode:
authorMiki Tebeka <miki.tebeka@gmail.com>2015-03-01 05:57:02 +0200
committerMiki Tebeka <miki.tebeka@gmail.com>2015-03-01 05:57:02 +0200
commit978af8a768be2ed4b0be249a2e87744d0ba8307d (patch)
treef10a81f4e874716898af755d9c5b21b85fc606c6 /go/src/step2_eval
parent1218ce98a40ef243824fed0efce7160a10fe5f36 (diff)
downloadmal-978af8a768be2ed4b0be249a2e87744d0ba8307d.tar.gz
mal-978af8a768be2ed4b0be249a2e87744d0ba8307d.zip
running "gofmt" on the code
Diffstat (limited to 'go/src/step2_eval')
-rw-r--r--go/src/step2_eval/step2_eval.go211
1 files changed, 117 insertions, 94 deletions
diff --git a/go/src/step2_eval/step2_eval.go b/go/src/step2_eval/step2_eval.go
index 7a9a33f..246c4cf 100644
--- a/go/src/step2_eval/step2_eval.go
+++ b/go/src/step2_eval/step2_eval.go
@@ -1,127 +1,150 @@
package main
import (
- "fmt"
- "strings"
- "errors"
+ "errors"
+ "fmt"
+ "strings"
)
import (
- "readline"
- . "types"
- "reader"
- "printer"
+ "printer"
+ "reader"
+ "readline"
+ . "types"
)
// read
func READ(str string) (MalType, error) {
- return reader.Read_str(str)
+ return reader.Read_str(str)
}
// eval
func eval_ast(ast MalType, env map[string]MalType) (MalType, error) {
- //fmt.Printf("eval_ast: %#v\n", ast)
- if Symbol_Q(ast) {
- k := ast.(Symbol).Val
- exp, ok := env[k]
- if !ok { return nil, errors.New("'" + k + "' not found") }
- return exp, nil
- } else if List_Q(ast) {
- lst := []MalType{}
- for _, a := range ast.(List).Val {
- exp, e := EVAL(a, env)
- if e != nil { return nil, e }
- lst = append(lst, exp)
- }
- return List{lst,nil}, nil
- } else if Vector_Q(ast) {
- lst := []MalType{}
- for _, a := range ast.(Vector).Val {
- exp, e := EVAL(a, env)
- if e != nil { return nil, e }
- lst = append(lst, exp)
- }
- return Vector{lst,nil}, nil
- } else if HashMap_Q(ast) {
- m := ast.(HashMap)
- new_hm := HashMap{map[string]MalType{},nil}
- for k, v := range m.Val {
- ke, e1 := EVAL(k, env)
- if e1 != nil { return nil, e1 }
- if _, ok := ke.(string); !ok {
- return nil, errors.New("non string hash-map key")
- }
- kv, e2 := EVAL(v, env)
- if e2 != nil { return nil, e2 }
- new_hm.Val[ke.(string)] = kv
- }
- return new_hm, nil
- } else {
- return ast, nil
- }
+ //fmt.Printf("eval_ast: %#v\n", ast)
+ if Symbol_Q(ast) {
+ k := ast.(Symbol).Val
+ exp, ok := env[k]
+ if !ok {
+ return nil, errors.New("'" + k + "' not found")
+ }
+ return exp, nil
+ } else if List_Q(ast) {
+ lst := []MalType{}
+ for _, a := range ast.(List).Val {
+ exp, e := EVAL(a, env)
+ if e != nil {
+ return nil, e
+ }
+ lst = append(lst, exp)
+ }
+ return List{lst, nil}, nil
+ } else if Vector_Q(ast) {
+ lst := []MalType{}
+ for _, a := range ast.(Vector).Val {
+ exp, e := EVAL(a, env)
+ if e != nil {
+ return nil, e
+ }
+ lst = append(lst, exp)
+ }
+ return Vector{lst, nil}, nil
+ } else if HashMap_Q(ast) {
+ m := ast.(HashMap)
+ new_hm := HashMap{map[string]MalType{}, nil}
+ for k, v := range m.Val {
+ ke, e1 := EVAL(k, env)
+ if e1 != nil {
+ return nil, e1
+ }
+ if _, ok := ke.(string); !ok {
+ return nil, errors.New("non string hash-map key")
+ }
+ kv, e2 := EVAL(v, env)
+ if e2 != nil {
+ return nil, e2
+ }
+ new_hm.Val[ke.(string)] = kv
+ }
+ return new_hm, nil
+ } else {
+ return ast, nil
+ }
}
func EVAL(ast MalType, env map[string]MalType) (MalType, error) {
- //fmt.Printf("EVAL: %v\n", printer.Pr_str(ast, true))
- switch ast.(type) {
- case List: // continue
- default: return eval_ast(ast, env)
- }
+ //fmt.Printf("EVAL: %v\n", printer.Pr_str(ast, true))
+ switch ast.(type) {
+ case List: // continue
+ default:
+ return eval_ast(ast, env)
+ }
- // apply list
- el, e := eval_ast(ast, env)
- if e != nil { return nil, e }
- f, ok := el.(List).Val[0].(func([]MalType)(MalType, error))
- if !ok { return nil, errors.New("attempt to call non-function") }
- return f(el.(List).Val[1:])
+ // apply list
+ el, e := eval_ast(ast, env)
+ if e != nil {
+ return nil, e
+ }
+ f, ok := el.(List).Val[0].(func([]MalType) (MalType, error))
+ if !ok {
+ return nil, errors.New("attempt to call non-function")
+ }
+ return f(el.(List).Val[1:])
}
// print
func PRINT(exp MalType) (string, error) {
- return printer.Pr_str(exp, true), nil
+ return printer.Pr_str(exp, true), nil
}
var repl_env = map[string]MalType{
- "+": func(a []MalType) (MalType, error) {
- return a[0].(int) + a[1].(int), nil
- },
- "-": func(a []MalType) (MalType, error) {
- return a[0].(int) - a[1].(int), nil
- },
- "*": func(a []MalType) (MalType, error) {
- return a[0].(int) * a[1].(int), nil
- },
- "/": func(a []MalType) (MalType, error) {
- return a[0].(int) / a[1].(int), nil
- },
+ "+": func(a []MalType) (MalType, error) {
+ return a[0].(int) + a[1].(int), nil
+ },
+ "-": func(a []MalType) (MalType, error) {
+ return a[0].(int) - a[1].(int), nil
+ },
+ "*": func(a []MalType) (MalType, error) {
+ return a[0].(int) * a[1].(int), nil
+ },
+ "/": func(a []MalType) (MalType, error) {
+ return a[0].(int) / a[1].(int), nil
+ },
}
// repl
func rep(str string) (MalType, error) {
- var exp MalType
- var res string
- var e error
- if exp, e = READ(str); e != nil { return nil, e }
- if exp, e = EVAL(exp, repl_env); e != nil { return nil, e }
- if res, e = PRINT(exp); e != nil { return nil, e }
- return res, nil
+ var exp MalType
+ var res string
+ var e error
+ if exp, e = READ(str); e != nil {
+ return nil, e
+ }
+ if exp, e = EVAL(exp, repl_env); e != nil {
+ return nil, e
+ }
+ if res, e = PRINT(exp); e != nil {
+ return nil, e
+ }
+ return res, nil
}
func main() {
- // repl loop
- for {
- text, err := readline.Readline("user> ")
- text = strings.TrimRight(text, "\n");
- if (err != nil) {
- return
- }
- var out MalType
- var e error
- if out, e = rep(text); e != nil {
- if e.Error() == "<empty line>" { continue }
- fmt.Printf("Error: %v\n", e)
- continue
- }
- fmt.Printf("%v\n", out)
- }
+ // repl loop
+ for {
+ text, err := readline.Readline("user> ")
+ text = strings.TrimRight(text, "\n")
+ if err != nil {
+ return
+ }
+ var out MalType
+ var e error
+ if out, e = rep(text); e != nil {
+ if e.Error() == "<empty line>" {
+ continue
+ }
+ fmt.Printf("Error: %v\n", e)
+ continue
+ }
+ fmt.Printf("%v\n", out)
+ }
}