diff options
| author | Miki Tebeka <miki.tebeka@gmail.com> | 2015-03-01 05:57:02 +0200 |
|---|---|---|
| committer | Miki Tebeka <miki.tebeka@gmail.com> | 2015-03-01 05:57:02 +0200 |
| commit | 978af8a768be2ed4b0be249a2e87744d0ba8307d (patch) | |
| tree | f10a81f4e874716898af755d9c5b21b85fc606c6 /go/src/step2_eval | |
| parent | 1218ce98a40ef243824fed0efce7160a10fe5f36 (diff) | |
| download | mal-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.go | 211 |
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) + } } |
