You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.2 KiB
Go
112 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
)
|
|
|
|
type cmp_bin_i func(int64, int64) bool
|
|
type cmp_bin_f func(float64, float64) bool
|
|
|
|
func cmp_left(vals []interface{}, fni cmp_bin_i, fnf cmp_bin_f) (bool, error) {
|
|
if len(vals) < 2 {
|
|
return false, errors.New("expected at least 2 arguments")
|
|
}
|
|
|
|
var lasti int64
|
|
var lastf float64
|
|
var floating bool
|
|
|
|
switch v := vals[0].(type) {
|
|
case float64:
|
|
floating = true
|
|
lastf = v
|
|
case int64:
|
|
lasti = v
|
|
default:
|
|
return false, fmt.Errorf("gt is not defined for %v", reflect.TypeOf(v))
|
|
}
|
|
|
|
for _, raw := range vals[1:] {
|
|
switch v := raw.(type) {
|
|
case float64:
|
|
if !floating {
|
|
floating = true
|
|
lastf = float64(lasti)
|
|
}
|
|
if !fnf(lastf, v) {
|
|
return false, nil
|
|
}
|
|
lastf = v
|
|
case int64:
|
|
if floating {
|
|
f := float64(v)
|
|
if !fnf(lastf, f) {
|
|
return false, nil
|
|
}
|
|
lastf = f
|
|
} else {
|
|
if !fni(lasti, v) {
|
|
return false, nil
|
|
}
|
|
lasti = v
|
|
}
|
|
default:
|
|
return false, errors.New("ooga booga")
|
|
}
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
var gt = builtin{
|
|
name: ">",
|
|
variadic: true,
|
|
fn: func(vals []interface{}) (interface{}, error) {
|
|
fni := func(x, y int64) bool { return x > y }
|
|
fnf := func(x, y float64) bool { return x > y }
|
|
return cmp_left(vals, fni, fnf)
|
|
},
|
|
}
|
|
|
|
var gte = builtin{
|
|
name: ">=",
|
|
variadic: true,
|
|
fn: func(vals []interface{}) (interface{}, error) {
|
|
fni := func(x, y int64) bool { return x >= y }
|
|
fnf := func(x, y float64) bool { return x >= y }
|
|
return cmp_left(vals, fni, fnf)
|
|
},
|
|
}
|
|
|
|
var lt = builtin{
|
|
name: "<",
|
|
variadic: true,
|
|
fn: func(vals []interface{}) (interface{}, error) {
|
|
fni := func(x, y int64) bool { return x < y }
|
|
fnf := func(x, y float64) bool { return x < y }
|
|
return cmp_left(vals, fni, fnf)
|
|
},
|
|
}
|
|
|
|
var lte = builtin{
|
|
name: "<=",
|
|
variadic: true,
|
|
fn: func(vals []interface{}) (interface{}, error) {
|
|
fni := func(x, y int64) bool { return x <= y }
|
|
fnf := func(x, y float64) bool { return x <= y }
|
|
return cmp_left(vals, fni, fnf)
|
|
},
|
|
}
|
|
|
|
var equals = builtin{
|
|
name: "=",
|
|
variadic: true,
|
|
fn: func(vals []interface{}) (interface{}, error) {
|
|
fni := func(x, y int64) bool { return x == y }
|
|
fnf := func(x, y float64) bool { return x == y }
|
|
return cmp_left(vals, fni, fnf)
|
|
},
|
|
}
|