Adds the reciprocal operator

1 files changed, 34 insertions, 0 deletions

diff --git a/subex/arithmetic.go b/subex/arithmetic.go
index ff30a58..4404a1c 100644
--- a/subex/arithmetic.go
+++ b/subex/arithmetic.go
@@ -118,3 +118,37 @@ func negateValues(atoms []walk.Atom) ([]walk.Atom, error) {
 	}
 	return negatedNumbers, nil
 }
+
+// If all are castable to numbers, takes reciprocals of all and returns them
+// Else errors
+func reciprocalValues(atoms []walk.Atom) ([]walk.Atom, error) {
+	var reciprocals []walk.Atom
+	values, err := walk.MemoryCompound(atoms)
+	if err != nil {
+		return nil, err
+	}
+	for _, value := range values {
+		switch v := value.(type) {
+			case walk.ValueNull:
+				return nil, errors.New("Tried to take reciprocal of null")
+			case walk.ValueBool:
+				if bool(v) {
+					reciprocals = append(reciprocals, walk.ValueNumber(1))
+				} else {
+					return nil, errors.New("Tried to take reciprocal of false")
+				}
+			case walk.ValueNumber:
+				reciprocals = append(reciprocals, walk.ValueNumber(1 / v))
+			case walk.ValueString:
+				num, err := strconv.ParseFloat(string(v), 64)
+				if err == nil {
+					reciprocals = append(reciprocals, walk.ValueNumber(1 / num))
+				} else {
+					return nil, errors.New("Tried to take reciprocal of non-castable string")
+				}
+			default:
+				return nil, errors.New("Tried to take reciprocal of non-number")
+		}
+	}
+	return reciprocals, nil
+}