<- Back to shtanton's homepage
aboutsummaryrefslogtreecommitdiff
path: root/subex/arithmetic.go
blob: 5e0eb4453f0dc365279b472e58395b629e023dfe (plain)
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package subex

import (
	"main/walk"
	"errors"
)

func binopAdd(values []walk.Value) ([]walk.Value, error) {
	if len(values) != 2 {
		return nil, errors.New("Tried to sum a weird number of values")
	}

	lhs, lhsIsNumber := values[0].(walk.NumberValue)
	if !lhsIsNumber {
		return nil, errors.New("Tried to sum a lhs that is not a number")
	}

	rhs, rhsIsNumber := values[1].(walk.NumberValue)
	if !rhsIsNumber {
		return nil, errors.New("Tried to sum a rhs that is not a number")
	}

	return []walk.Value{walk.NumberValue(float64(lhs) + float64(rhs))}, nil
}

func binopMultiply(values []walk.Value) ([]walk.Value, error) {
	if len(values) != 2 {
		return nil, errors.New("Tried to multiply a weird number of values")
	}

	lhs, lhsIsNumber := values[0].(walk.NumberValue)
	if !lhsIsNumber {
		return nil, errors.New("Tried to multiply a lhs that is not a number")
	}

	rhs, rhsIsNumber := values[1].(walk.NumberValue)
	if !rhsIsNumber {
		return nil, errors.New("Tried to multiply a rhs that is not a number")
	}

	return []walk.Value{walk.NumberValue(float64(lhs) * float64(rhs))}, nil
}

func binopDivide(values []walk.Value) ([]walk.Value, error) {
	if len(values) != 2 {
		return nil, errors.New("Tried to divide a weird number of values")
	}

	lhs, lhsIsNumber := values[0].(walk.NumberValue)
	if !lhsIsNumber {
		return nil, errors.New("Tried to divide a lhs that is not a number")
	}

	rhs, rhsIsNumber := values[1].(walk.NumberValue)
	if !rhsIsNumber {
		return nil, errors.New("Tried to divide a rhs that is not a number")
	}

	return []walk.Value{walk.NumberValue(float64(lhs) / float64(rhs))}, nil
}

func arithmeticSum(values []walk.Value) ([]walk.Value, error) {
	var total float64 = 0
	for _, value := range values {
		n, isNumber := value.(walk.NumberValue)
		if !isNumber {
			return nil, errors.New("Tried to sum non-number value")
		}
		total += float64(n)
	}

	return []walk.Value{walk.NumberValue(total)}, nil
}

func arithmeticProduct(values []walk.Value) ([]walk.Value, error) {
	var product float64 = 0
	for _, value := range values {
		n, isNumber := value.(walk.NumberValue)
		if !isNumber {
			return nil, errors.New("Tried to sum non-number value")
		}
		product *= float64(n)
	}

	return []walk.Value{walk.NumberValue(product)}, nil
}