mirror of
https://codeberg.org/scip/rpnc.git
synced 2025-12-17 12:31:04 +01:00
Compare commits
2 Commits
fix/floats
...
v2.1.0
| Author | SHA1 | Date | |
|---|---|---|---|
| 62188dda0c | |||
| 6a2a501e48 |
8
Makefile
8
Makefile
@@ -59,6 +59,14 @@ test: clean
|
|||||||
testfuzzy: clean
|
testfuzzy: clean
|
||||||
go test -fuzz ./... $(ARGS)
|
go test -fuzz ./... $(ARGS)
|
||||||
|
|
||||||
|
testlint: test lint
|
||||||
|
|
||||||
|
lint:
|
||||||
|
golangci-lint run
|
||||||
|
|
||||||
|
lint-full:
|
||||||
|
golangci-lint run --enable-all --exclude-use-default --disable exhaustivestruct,exhaustruct,depguard,interfacer,deadcode,golint,structcheck,scopelint,varcheck,ifshort,maligned,nosnakecase,godot,funlen,gofumpt,cyclop,noctx,gochecknoglobals,paralleltest,forbidigo,godox,dupword,forcetypeassert,goerr113,gomnd
|
||||||
|
|
||||||
singletest:
|
singletest:
|
||||||
@echo "Call like this: make singletest TEST=TestPrepareColumns ARGS=-v"
|
@echo "Call like this: make singletest TEST=TestPrepareColumns ARGS=-v"
|
||||||
go test -run $(TEST) $(ARGS)
|
go test -run $(TEST) $(ARGS)
|
||||||
|
|||||||
127
calc.go
127
calc.go
@@ -69,7 +69,7 @@ Bitwise operators: and or xor < (left shift) > (right shift)
|
|||||||
|
|
||||||
Percent functions:
|
Percent functions:
|
||||||
% percent
|
% percent
|
||||||
%- substract percent
|
%- subtract percent
|
||||||
%+ add percent
|
%+ add percent
|
||||||
|
|
||||||
Math functions (see https://pkg.go.dev/math):
|
Math functions (see https://pkg.go.dev/math):
|
||||||
@@ -92,9 +92,9 @@ Register variables:
|
|||||||
// commands, constants and operators, defined here to feed completion
|
// commands, constants and operators, defined here to feed completion
|
||||||
// and our mode switch in Eval() dynamically
|
// and our mode switch in Eval() dynamically
|
||||||
const (
|
const (
|
||||||
//Commands string = `dump reverse clear shift undo help history manual exit quit swap debug undebug nodebug batch nobatch showstack noshowstack vars`
|
|
||||||
Constants string = `Pi Phi Sqrt2 SqrtE SqrtPi SqrtPhi Ln2 Log2E Ln10 Log10E`
|
Constants string = `Pi Phi Sqrt2 SqrtE SqrtPi SqrtPhi Ln2 Log2E Ln10 Log10E`
|
||||||
Precision int = 2
|
Precision int = 2
|
||||||
|
ShowStackLen int = 5
|
||||||
)
|
)
|
||||||
|
|
||||||
// That way we can add custom functions to completion
|
// That way we can add custom functions to completion
|
||||||
@@ -150,37 +150,36 @@ func (c *Calc) GetCompleteCustomFuncalls() func(string) []string {
|
|||||||
|
|
||||||
return completions
|
return completions
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewCalc() *Calc {
|
func NewCalc() *Calc {
|
||||||
c := Calc{stack: NewStack(), debug: false, precision: Precision}
|
calc := Calc{stack: NewStack(), debug: false, precision: Precision}
|
||||||
|
|
||||||
c.Funcalls = DefineFunctions()
|
calc.Funcalls = DefineFunctions()
|
||||||
c.BatchFuncalls = DefineBatchFunctions()
|
calc.BatchFuncalls = DefineBatchFunctions()
|
||||||
c.Vars = map[string]float64{}
|
calc.Vars = map[string]float64{}
|
||||||
|
|
||||||
c.completer = readline.NewPrefixCompleter(
|
calc.completer = readline.NewPrefixCompleter(
|
||||||
// custom lua functions
|
// custom lua functions
|
||||||
readline.PcItemDynamic(GetCompleteCustomFunctions()),
|
readline.PcItemDynamic(GetCompleteCustomFunctions()),
|
||||||
readline.PcItemDynamic(c.GetCompleteCustomFuncalls()),
|
readline.PcItemDynamic(calc.GetCompleteCustomFuncalls()),
|
||||||
)
|
)
|
||||||
|
|
||||||
c.Space = regexp.MustCompile(`\s+`)
|
calc.Space = regexp.MustCompile(`\s+`)
|
||||||
c.Comment = regexp.MustCompile(`#.*`) // ignore everything after #
|
calc.Comment = regexp.MustCompile(`#.*`) // ignore everything after #
|
||||||
c.Register = regexp.MustCompile(`^([<>])([A-Z][A-Z0-9]*)`)
|
calc.Register = regexp.MustCompile(`^([<>])([A-Z][A-Z0-9]*)`)
|
||||||
|
|
||||||
// pre-calculate mode switching arrays
|
// pre-calculate mode switching arrays
|
||||||
c.Constants = strings.Split(Constants, " ")
|
calc.Constants = strings.Split(Constants, " ")
|
||||||
|
|
||||||
c.SetCommands()
|
calc.SetCommands()
|
||||||
|
|
||||||
return &c
|
return &calc
|
||||||
}
|
}
|
||||||
|
|
||||||
// setup the interpreter, called from main(), import lua functions
|
// setup the interpreter, called from main(), import lua functions
|
||||||
func (c *Calc) SetInt(I *Interpreter) {
|
func (c *Calc) SetInt(interpreter *Interpreter) {
|
||||||
c.interpreter = I
|
c.interpreter = interpreter
|
||||||
|
|
||||||
for name := range LuaFuncs {
|
for name := range LuaFuncs {
|
||||||
c.LuaFunctions = append(c.LuaFunctions, name)
|
c.LuaFunctions = append(c.LuaFunctions, name)
|
||||||
@@ -207,22 +206,22 @@ func (c *Calc) ToggleShow() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *Calc) Prompt() string {
|
func (c *Calc) Prompt() string {
|
||||||
p := "\033[31m»\033[0m "
|
prompt := "\033[31m»\033[0m "
|
||||||
b := ""
|
batch := ""
|
||||||
|
|
||||||
if c.batch {
|
if c.batch {
|
||||||
b = "->batch"
|
batch = "->batch"
|
||||||
}
|
}
|
||||||
|
|
||||||
d := ""
|
debug := ""
|
||||||
v := ""
|
revision := ""
|
||||||
|
|
||||||
if c.debug {
|
if c.debug {
|
||||||
d = "->debug"
|
debug = "->debug"
|
||||||
v = fmt.Sprintf("/rev%d", c.stack.rev)
|
revision = fmt.Sprintf("/rev%d", c.stack.rev)
|
||||||
}
|
}
|
||||||
|
|
||||||
return fmt.Sprintf("rpn%s%s [%d%s]%s", b, d, c.stack.Len(), v, p)
|
return fmt.Sprintf("rpn%s%s [%d%s]%s", batch, debug, c.stack.Len(), revision, prompt)
|
||||||
}
|
}
|
||||||
|
|
||||||
// the actual work horse, evaluate a line of calc command[s]
|
// the actual work horse, evaluate a line of calc command[s]
|
||||||
@@ -251,10 +250,12 @@ func (c *Calc) Eval(line string) error {
|
|||||||
if c.showstack && !c.stdin {
|
if c.showstack && !c.stdin {
|
||||||
dots := ""
|
dots := ""
|
||||||
|
|
||||||
if c.stack.Len() > 5 {
|
if c.stack.Len() > ShowStackLen {
|
||||||
dots = "... "
|
dots = "... "
|
||||||
}
|
}
|
||||||
last := c.stack.Last(5)
|
|
||||||
|
last := c.stack.Last(ShowStackLen)
|
||||||
|
|
||||||
fmt.Printf("stack: %s%s\n", dots, list2str(last))
|
fmt.Printf("stack: %s%s\n", dots, list2str(last))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -267,13 +268,18 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
if err == nil {
|
if err == nil {
|
||||||
c.stack.Backup()
|
c.stack.Backup()
|
||||||
c.stack.Push(num)
|
c.stack.Push(num)
|
||||||
} else {
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// try hex
|
// try hex
|
||||||
var i int
|
var i int
|
||||||
_, err := fmt.Sscanf(item, "0x%x", &i)
|
|
||||||
|
_, err = fmt.Sscanf(item, "0x%x", &i)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
c.stack.Backup()
|
c.stack.Backup()
|
||||||
c.stack.Push(float64(i))
|
c.stack.Push(float64(i))
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -281,15 +287,17 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
// put the constant onto the stack
|
// put the constant onto the stack
|
||||||
c.stack.Backup()
|
c.stack.Backup()
|
||||||
c.stack.Push(const2num(item))
|
c.stack.Push(const2num(item))
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if exists(c.Funcalls, item) {
|
if exists(c.Funcalls, item) {
|
||||||
if err := c.DoFuncall(item); err != nil {
|
if err := c.DoFuncall(item); err != nil {
|
||||||
return Error(err.Error())
|
return Error(err.Error())
|
||||||
} else {
|
|
||||||
c.Result()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
c.Result()
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -300,15 +308,17 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
|
|
||||||
if err := c.DoFuncall(item); err != nil {
|
if err := c.DoFuncall(item); err != nil {
|
||||||
return Error(err.Error())
|
return Error(err.Error())
|
||||||
} else {
|
|
||||||
c.Result()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
c.Result()
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if contains(c.LuaFunctions, item) {
|
if contains(c.LuaFunctions, item) {
|
||||||
// user provided custom lua functions
|
// user provided custom lua functions
|
||||||
c.EvalLuaFunction(item)
|
c.EvalLuaFunction(item)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -320,6 +330,7 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
case "<":
|
case "<":
|
||||||
c.GetVar(regmatches[2])
|
c.GetVar(regmatches[2])
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -327,21 +338,25 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
// FIXME: propagate errors
|
// FIXME: propagate errors
|
||||||
if exists(c.Commands, item) {
|
if exists(c.Commands, item) {
|
||||||
c.Commands[item].Func(c)
|
c.Commands[item].Func(c)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if exists(c.ShowCommands, item) {
|
if exists(c.ShowCommands, item) {
|
||||||
c.ShowCommands[item].Func(c)
|
c.ShowCommands[item].Func(c)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if exists(c.StackCommands, item) {
|
if exists(c.StackCommands, item) {
|
||||||
c.StackCommands[item].Func(c)
|
c.StackCommands[item].Func(c)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if exists(c.SettingsCommands, item) {
|
if exists(c.SettingsCommands, item) {
|
||||||
c.SettingsCommands[item].Func(c)
|
c.SettingsCommands[item].Func(c)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -354,7 +369,6 @@ func (c *Calc) EvalItem(item string) error {
|
|||||||
default:
|
default:
|
||||||
return Error("unknown command or operator")
|
return Error("unknown command or operator")
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -373,6 +387,7 @@ func (c *Calc) DoFuncall(funcname string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var args Numbers
|
var args Numbers
|
||||||
|
|
||||||
batch := false
|
batch := false
|
||||||
|
|
||||||
if function.Expectargs == -1 {
|
if function.Expectargs == -1 {
|
||||||
@@ -394,11 +409,11 @@ func (c *Calc) DoFuncall(funcname string) error {
|
|||||||
// the actual lambda call, so to say. We provide a slice of
|
// the actual lambda call, so to say. We provide a slice of
|
||||||
// the requested size, fetched from the stack (but not popped
|
// the requested size, fetched from the stack (but not popped
|
||||||
// yet!)
|
// yet!)
|
||||||
R := function.Func(args)
|
funcresult := function.Func(args)
|
||||||
|
|
||||||
if R.Err != nil {
|
if funcresult.Err != nil {
|
||||||
// leave the stack untouched in case of any error
|
// leave the stack untouched in case of any error
|
||||||
return R.Err
|
return funcresult.Err
|
||||||
}
|
}
|
||||||
|
|
||||||
// don't forget to backup!
|
// don't forget to backup!
|
||||||
@@ -414,10 +429,11 @@ func (c *Calc) DoFuncall(funcname string) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// save result
|
// save result
|
||||||
c.stack.Push(R.Res)
|
c.stack.Push(funcresult.Res)
|
||||||
|
|
||||||
// thanks a lot
|
// thanks a lot
|
||||||
c.SetHistory(funcname, args, R.Res)
|
c.SetHistory(funcname, args, funcresult.Res)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -465,24 +481,26 @@ func (c *Calc) Debug(msg string) {
|
|||||||
|
|
||||||
func (c *Calc) EvalLuaFunction(funcname string) {
|
func (c *Calc) EvalLuaFunction(funcname string) {
|
||||||
// called from calc loop
|
// called from calc loop
|
||||||
var x float64
|
var luaresult float64
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
switch c.interpreter.FuncNumArgs(funcname) {
|
switch c.interpreter.FuncNumArgs(funcname) {
|
||||||
case 0:
|
case 0:
|
||||||
fallthrough
|
fallthrough
|
||||||
case 1:
|
case 1:
|
||||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last())
|
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last())
|
||||||
case 2:
|
case 2:
|
||||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last(2))
|
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last(2))
|
||||||
case -1:
|
case -1:
|
||||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.All())
|
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.All())
|
||||||
default:
|
default:
|
||||||
x, err = 0, errors.New("invalid number of argument requested")
|
luaresult, err = 0, errors.New("invalid number of argument requested")
|
||||||
}
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -493,24 +511,26 @@ func (c *Calc) EvalLuaFunction(funcname string) {
|
|||||||
switch c.interpreter.FuncNumArgs(funcname) {
|
switch c.interpreter.FuncNumArgs(funcname) {
|
||||||
case 0:
|
case 0:
|
||||||
a := c.stack.Last()
|
a := c.stack.Last()
|
||||||
|
|
||||||
if len(a) == 1 {
|
if len(a) == 1 {
|
||||||
c.History("%s(%f) = %f", funcname, a, x)
|
c.History("%s(%f) = %f", funcname, a, luaresult)
|
||||||
}
|
}
|
||||||
|
|
||||||
dopush = false
|
dopush = false
|
||||||
case 1:
|
case 1:
|
||||||
a := c.stack.Pop()
|
a := c.stack.Pop()
|
||||||
c.History("%s(%f) = %f", funcname, a, x)
|
c.History("%s(%f) = %f", funcname, a, luaresult)
|
||||||
case 2:
|
case 2:
|
||||||
a := c.stack.Pop()
|
a := c.stack.Pop()
|
||||||
b := c.stack.Pop()
|
b := c.stack.Pop()
|
||||||
c.History("%s(%f,%f) = %f", funcname, a, b, x)
|
c.History("%s(%f,%f) = %f", funcname, a, b, luaresult)
|
||||||
case -1:
|
case -1:
|
||||||
c.stack.Clear()
|
c.stack.Clear()
|
||||||
c.History("%s(*) = %f", funcname, x)
|
c.History("%s(*) = %f", funcname, luaresult)
|
||||||
}
|
}
|
||||||
|
|
||||||
if dopush {
|
if dopush {
|
||||||
c.stack.Push(x)
|
c.stack.Push(luaresult)
|
||||||
}
|
}
|
||||||
|
|
||||||
c.Result()
|
c.Result()
|
||||||
@@ -553,27 +573,35 @@ func sortcommands(hash Commands) []string {
|
|||||||
|
|
||||||
func (c *Calc) PrintHelp() {
|
func (c *Calc) PrintHelp() {
|
||||||
fmt.Println("Available configuration commands:")
|
fmt.Println("Available configuration commands:")
|
||||||
|
|
||||||
for _, name := range sortcommands(c.SettingsCommands) {
|
for _, name := range sortcommands(c.SettingsCommands) {
|
||||||
fmt.Printf("%-20s %s\n", name, c.SettingsCommands[name].Help)
|
fmt.Printf("%-20s %s\n", name, c.SettingsCommands[name].Help)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println()
|
fmt.Println()
|
||||||
|
|
||||||
fmt.Println("Available show commands:")
|
fmt.Println("Available show commands:")
|
||||||
|
|
||||||
for _, name := range sortcommands(c.ShowCommands) {
|
for _, name := range sortcommands(c.ShowCommands) {
|
||||||
fmt.Printf("%-20s %s\n", name, c.ShowCommands[name].Help)
|
fmt.Printf("%-20s %s\n", name, c.ShowCommands[name].Help)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println()
|
fmt.Println()
|
||||||
|
|
||||||
fmt.Println("Available stack manipulation commands:")
|
fmt.Println("Available stack manipulation commands:")
|
||||||
|
|
||||||
for _, name := range sortcommands(c.StackCommands) {
|
for _, name := range sortcommands(c.StackCommands) {
|
||||||
fmt.Printf("%-20s %s\n", name, c.StackCommands[name].Help)
|
fmt.Printf("%-20s %s\n", name, c.StackCommands[name].Help)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println()
|
fmt.Println()
|
||||||
|
|
||||||
fmt.Println("Other commands:")
|
fmt.Println("Other commands:")
|
||||||
|
|
||||||
for _, name := range sortcommands(c.Commands) {
|
for _, name := range sortcommands(c.Commands) {
|
||||||
fmt.Printf("%-20s %s\n", name, c.Commands[name].Help)
|
fmt.Printf("%-20s %s\n", name, c.Commands[name].Help)
|
||||||
}
|
}
|
||||||
|
|
||||||
fmt.Println()
|
fmt.Println()
|
||||||
|
|
||||||
fmt.Println(Help)
|
fmt.Println(Help)
|
||||||
@@ -581,6 +609,7 @@ func (c *Calc) PrintHelp() {
|
|||||||
// append lua functions, if any
|
// append lua functions, if any
|
||||||
if len(LuaFuncs) > 0 {
|
if len(LuaFuncs) > 0 {
|
||||||
fmt.Println("Lua functions:")
|
fmt.Println("Lua functions:")
|
||||||
|
|
||||||
for name, function := range LuaFuncs {
|
for name, function := range LuaFuncs {
|
||||||
fmt.Printf("%-20s %s\n", name, function.help)
|
fmt.Printf("%-20s %s\n", name, function.help)
|
||||||
}
|
}
|
||||||
|
|||||||
45
calc_test.go
45
calc_test.go
@@ -71,12 +71,12 @@ func TestCommentsAndWhitespace(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, test := range tests {
|
||||||
testname := fmt.Sprintf("%s .(expect %.2f)",
|
testname := fmt.Sprintf("%s .(expect %.2f)",
|
||||||
tt.name, tt.exp)
|
test.name, test.exp)
|
||||||
|
|
||||||
t.Run(testname, func(t *testing.T) {
|
t.Run(testname, func(t *testing.T) {
|
||||||
for _, line := range tt.cmd {
|
for _, line := range test.cmd {
|
||||||
if err := calc.Eval(line); err != nil {
|
if err := calc.Eval(line); err != nil {
|
||||||
t.Errorf(err.Error())
|
t.Errorf(err.Error())
|
||||||
}
|
}
|
||||||
@@ -84,9 +84,9 @@ func TestCommentsAndWhitespace(t *testing.T) {
|
|||||||
got := calc.stack.Last()
|
got := calc.stack.Last()
|
||||||
|
|
||||||
if len(got) > 0 {
|
if len(got) > 0 {
|
||||||
if got[0] != tt.exp {
|
if got[0] != test.exp {
|
||||||
t.Errorf("parsing failed:\n+++ got: %f\n--- want: %f",
|
t.Errorf("parsing failed:\n+++ got: %f\n--- want: %f",
|
||||||
got, tt.exp)
|
got, test.exp)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -94,7 +94,6 @@ func TestCommentsAndWhitespace(t *testing.T) {
|
|||||||
t.Errorf("invalid stack size:\n+++ got: %d\n--- want: 1",
|
t.Errorf("invalid stack size:\n+++ got: %d\n--- want: 1",
|
||||||
calc.stack.Len())
|
calc.stack.Len())
|
||||||
}
|
}
|
||||||
|
|
||||||
})
|
})
|
||||||
|
|
||||||
calc.stack.Clear()
|
calc.stack.Clear()
|
||||||
@@ -286,20 +285,20 @@ func TestCalc(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, test := range tests {
|
||||||
testname := fmt.Sprintf("cmd-%s-expect-%.2f",
|
testname := fmt.Sprintf("cmd-%s-expect-%.2f",
|
||||||
tt.name, tt.exp)
|
test.name, test.exp)
|
||||||
|
|
||||||
t.Run(testname, func(t *testing.T) {
|
t.Run(testname, func(t *testing.T) {
|
||||||
calc.batch = tt.batch
|
calc.batch = test.batch
|
||||||
if err := calc.Eval(tt.cmd); err != nil {
|
if err := calc.Eval(test.cmd); err != nil {
|
||||||
t.Errorf(err.Error())
|
t.Errorf(err.Error())
|
||||||
}
|
}
|
||||||
got := calc.Result()
|
got := calc.Result()
|
||||||
calc.stack.Clear()
|
calc.stack.Clear()
|
||||||
if got != tt.exp {
|
if got != test.exp {
|
||||||
t.Errorf("calc failed:\n+++ got: %f\n--- want: %f",
|
t.Errorf("calc failed:\n+++ got: %f\n--- want: %f",
|
||||||
got, tt.exp)
|
got, test.exp)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -324,23 +323,24 @@ func TestCalcLua(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
calc := NewCalc()
|
calc := NewCalc()
|
||||||
L = lua.NewState(lua.Options{SkipOpenLibs: true})
|
|
||||||
defer L.Close()
|
LuaInterpreter = lua.NewState(lua.Options{SkipOpenLibs: true})
|
||||||
|
defer LuaInterpreter.Close()
|
||||||
|
|
||||||
luarunner := NewInterpreter("example.lua", false)
|
luarunner := NewInterpreter("example.lua", false)
|
||||||
luarunner.InitLua()
|
luarunner.InitLua()
|
||||||
calc.SetInt(luarunner)
|
calc.SetInt(luarunner)
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, test := range tests {
|
||||||
testname := fmt.Sprintf("lua-%s", tt.function)
|
testname := fmt.Sprintf("lua-%s", test.function)
|
||||||
|
|
||||||
t.Run(testname, func(t *testing.T) {
|
t.Run(testname, func(t *testing.T) {
|
||||||
calc.stack.Clear()
|
calc.stack.Clear()
|
||||||
for _, item := range tt.stack {
|
for _, item := range test.stack {
|
||||||
calc.stack.Push(item)
|
calc.stack.Push(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
calc.EvalLuaFunction(tt.function)
|
calc.EvalLuaFunction(test.function)
|
||||||
|
|
||||||
got := calc.stack.Last()
|
got := calc.stack.Last()
|
||||||
|
|
||||||
@@ -349,9 +349,9 @@ func TestCalcLua(t *testing.T) {
|
|||||||
calc.stack.Len())
|
calc.stack.Len())
|
||||||
}
|
}
|
||||||
|
|
||||||
if got[0] != tt.exp {
|
if got[0] != test.exp {
|
||||||
t.Errorf("lua function %s failed:\n+++ got: %f\n--- want: %f",
|
t.Errorf("lua function %s failed:\n+++ got: %f\n--- want: %f",
|
||||||
tt.function, got, tt.exp)
|
test.function, got, test.exp)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -380,7 +380,8 @@ func FuzzEval(f *testing.F) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
calc := NewCalc()
|
calc := NewCalc()
|
||||||
var i int
|
|
||||||
|
var hexnum int
|
||||||
|
|
||||||
f.Fuzz(func(t *testing.T, line string) {
|
f.Fuzz(func(t *testing.T, line string) {
|
||||||
t.Logf("Stack:\n%v\n", calc.stack.All())
|
t.Logf("Stack:\n%v\n", calc.stack.All())
|
||||||
@@ -389,7 +390,7 @@ func FuzzEval(f *testing.F) {
|
|||||||
// not corpus and empty?
|
// not corpus and empty?
|
||||||
if !contains(legal, line) && len(line) > 0 {
|
if !contains(legal, line) && len(line) > 0 {
|
||||||
item := strings.TrimSpace(calc.Comment.ReplaceAllString(line, ""))
|
item := strings.TrimSpace(calc.Comment.ReplaceAllString(line, ""))
|
||||||
_, hexerr := fmt.Sscanf(item, "0x%x", &i)
|
_, hexerr := fmt.Sscanf(item, "0x%x", &hexnum)
|
||||||
// no comment?
|
// no comment?
|
||||||
if len(item) > 0 {
|
if len(item) > 0 {
|
||||||
// no known command or function?
|
// no known command or function?
|
||||||
|
|||||||
255
command.go
255
command.go
@@ -42,9 +42,8 @@ func NewCommand(help string, function CommandFunction) *Command {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// define all management (that is: non calculation) commands
|
func (c *Calc) SetSettingsCommands() Commands {
|
||||||
func (c *Calc) SetCommands() {
|
return Commands{
|
||||||
c.SettingsCommands = Commands{
|
|
||||||
// Toggles
|
// Toggles
|
||||||
"debug": NewCommand(
|
"debug": NewCommand(
|
||||||
"toggle debugging",
|
"toggle debugging",
|
||||||
@@ -89,8 +88,10 @@ func (c *Calc) SetCommands() {
|
|||||||
},
|
},
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
c.ShowCommands = Commands{
|
func (c *Calc) SetShowCommands() Commands {
|
||||||
|
return Commands{
|
||||||
// Display commands
|
// Display commands
|
||||||
"dump": NewCommand(
|
"dump": NewCommand(
|
||||||
"display the stack contents",
|
"display the stack contents",
|
||||||
@@ -131,8 +132,10 @@ func (c *Calc) SetCommands() {
|
|||||||
},
|
},
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
c.StackCommands = Commands{
|
func (c *Calc) SetStackCommands() Commands {
|
||||||
|
return Commands{
|
||||||
"clear": NewCommand(
|
"clear": NewCommand(
|
||||||
"clear the whole stack",
|
"clear the whole stack",
|
||||||
func(c *Calc) {
|
func(c *Calc) {
|
||||||
@@ -159,14 +162,7 @@ func (c *Calc) SetCommands() {
|
|||||||
|
|
||||||
"swap": NewCommand(
|
"swap": NewCommand(
|
||||||
"exchange the last two elements",
|
"exchange the last two elements",
|
||||||
func(c *Calc) {
|
CommandSwap,
|
||||||
if c.stack.Len() < 2 {
|
|
||||||
fmt.Println("stack too small, can't swap")
|
|
||||||
} else {
|
|
||||||
c.stack.Backup()
|
|
||||||
c.stack.Swap()
|
|
||||||
}
|
|
||||||
},
|
|
||||||
),
|
),
|
||||||
|
|
||||||
"undo": NewCommand(
|
"undo": NewCommand(
|
||||||
@@ -178,113 +174,21 @@ func (c *Calc) SetCommands() {
|
|||||||
|
|
||||||
"dup": NewCommand(
|
"dup": NewCommand(
|
||||||
"duplicate last stack item",
|
"duplicate last stack item",
|
||||||
func(c *Calc) {
|
CommandDup,
|
||||||
item := c.stack.Last()
|
|
||||||
if len(item) == 1 {
|
|
||||||
c.stack.Backup()
|
|
||||||
c.stack.Push(item[0])
|
|
||||||
} else {
|
|
||||||
fmt.Println("stack empty")
|
|
||||||
}
|
|
||||||
},
|
|
||||||
),
|
),
|
||||||
|
|
||||||
"edit": NewCommand(
|
"edit": NewCommand(
|
||||||
"edit the stack interactively",
|
"edit the stack interactively",
|
||||||
func(c *Calc) {
|
CommandEdit,
|
||||||
if c.stack.Len() == 0 {
|
|
||||||
fmt.Println("empty stack")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
c.stack.Backup()
|
|
||||||
|
|
||||||
// put the stack contents into a tmp file
|
|
||||||
tmp, err := os.CreateTemp("", "stack")
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer os.Remove(tmp.Name())
|
|
||||||
|
|
||||||
comment := `# add or remove numbers as you wish.
|
|
||||||
# each number must be on its own line.
|
|
||||||
# numbers must be floating point formatted.
|
|
||||||
`
|
|
||||||
_, err = tmp.WriteString(comment)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, item := range c.stack.All() {
|
|
||||||
_, err = fmt.Fprintf(tmp, "%f\n", item)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tmp.Close()
|
|
||||||
|
|
||||||
// determine which editor to use
|
|
||||||
editor := "vi"
|
|
||||||
enveditor, present := os.LookupEnv("EDITOR")
|
|
||||||
if present {
|
|
||||||
if editor != "" {
|
|
||||||
if _, err := os.Stat(editor); err == nil {
|
|
||||||
editor = enveditor
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// execute editor with our tmp file containing current stack
|
|
||||||
cmd := exec.Command(editor, tmp.Name())
|
|
||||||
|
|
||||||
cmd.Stdin = os.Stdin
|
|
||||||
cmd.Stdout = os.Stdout
|
|
||||||
cmd.Stderr = os.Stderr
|
|
||||||
|
|
||||||
err = cmd.Run()
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println("could not run editor command: ", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// read the file back in
|
|
||||||
modified, err := os.Open(tmp.Name())
|
|
||||||
if err != nil {
|
|
||||||
fmt.Println("Error opening file:", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer modified.Close()
|
|
||||||
|
|
||||||
// reset the stack
|
|
||||||
c.stack.Clear()
|
|
||||||
|
|
||||||
// and put the new contents (if legit) back onto the stack
|
|
||||||
scanner := bufio.NewScanner(modified)
|
|
||||||
for scanner.Scan() {
|
|
||||||
line := strings.TrimSpace(c.Comment.ReplaceAllString(scanner.Text(), ""))
|
|
||||||
if line == "" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
num, err := strconv.ParseFloat(line, 64)
|
|
||||||
if err != nil {
|
|
||||||
fmt.Printf("%s is not a floating point number!\n", line)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
c.stack.Push(num)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := scanner.Err(); err != nil {
|
|
||||||
fmt.Println("Error reading from file:", err)
|
|
||||||
}
|
|
||||||
},
|
|
||||||
),
|
),
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// define all management (that is: non calculation) commands
|
||||||
|
func (c *Calc) SetCommands() {
|
||||||
|
c.SettingsCommands = c.SetSettingsCommands()
|
||||||
|
c.ShowCommands = c.SetShowCommands()
|
||||||
|
c.StackCommands = c.SetStackCommands()
|
||||||
|
|
||||||
// general commands
|
// general commands
|
||||||
c.Commands = Commands{
|
c.Commands = Commands{
|
||||||
@@ -317,3 +221,126 @@ func (c *Calc) SetCommands() {
|
|||||||
c.StackCommands["c"] = c.StackCommands["clear"]
|
c.StackCommands["c"] = c.StackCommands["clear"]
|
||||||
c.StackCommands["u"] = c.StackCommands["undo"]
|
c.StackCommands["u"] = c.StackCommands["undo"]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// added to the command map:
|
||||||
|
func CommandSwap(c *Calc) {
|
||||||
|
if c.stack.Len() < 2 {
|
||||||
|
fmt.Println("stack too small, can't swap")
|
||||||
|
} else {
|
||||||
|
c.stack.Backup()
|
||||||
|
c.stack.Swap()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func CommandDup(c *Calc) {
|
||||||
|
item := c.stack.Last()
|
||||||
|
if len(item) == 1 {
|
||||||
|
c.stack.Backup()
|
||||||
|
c.stack.Push(item[0])
|
||||||
|
} else {
|
||||||
|
fmt.Println("stack empty")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func CommandEdit(calc *Calc) {
|
||||||
|
if calc.stack.Len() == 0 {
|
||||||
|
fmt.Println("empty stack")
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
calc.stack.Backup()
|
||||||
|
|
||||||
|
// put the stack contents into a tmp file
|
||||||
|
tmp, err := os.CreateTemp("", "stack")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer os.Remove(tmp.Name())
|
||||||
|
|
||||||
|
comment := `# add or remove numbers as you wish.
|
||||||
|
# each number must be on its own line.
|
||||||
|
# numbers must be floating point formatted.
|
||||||
|
`
|
||||||
|
_, err = tmp.WriteString(comment)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, item := range calc.stack.All() {
|
||||||
|
_, err = fmt.Fprintf(tmp, "%f\n", item)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tmp.Close()
|
||||||
|
|
||||||
|
// determine which editor to use
|
||||||
|
editor := "vi"
|
||||||
|
|
||||||
|
enveditor, present := os.LookupEnv("EDITOR")
|
||||||
|
if present {
|
||||||
|
if editor != "" {
|
||||||
|
if _, err := os.Stat(editor); err == nil {
|
||||||
|
editor = enveditor
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// execute editor with our tmp file containing current stack
|
||||||
|
cmd := exec.Command(editor, tmp.Name())
|
||||||
|
|
||||||
|
cmd.Stdin = os.Stdin
|
||||||
|
cmd.Stdout = os.Stdout
|
||||||
|
cmd.Stderr = os.Stderr
|
||||||
|
|
||||||
|
err = cmd.Run()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("could not run editor command: ", err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// read the file back in
|
||||||
|
modified, err := os.Open(tmp.Name())
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("Error opening file:", err)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer modified.Close()
|
||||||
|
|
||||||
|
// reset the stack
|
||||||
|
calc.stack.Clear()
|
||||||
|
|
||||||
|
// and put the new contents (if legit) back onto the stack
|
||||||
|
scanner := bufio.NewScanner(modified)
|
||||||
|
for scanner.Scan() {
|
||||||
|
line := strings.TrimSpace(calc.Comment.ReplaceAllString(scanner.Text(), ""))
|
||||||
|
if line == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
num, err := strconv.ParseFloat(line, 64)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("%s is not a floating point number!\n", line)
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
calc.stack.Push(num)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
fmt.Println("Error reading from file:", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
319
funcs.go
319
funcs.go
@@ -22,14 +22,14 @@ import (
|
|||||||
"math"
|
"math"
|
||||||
)
|
)
|
||||||
|
|
||||||
type R struct {
|
type Result struct {
|
||||||
Res float64
|
Res float64
|
||||||
Err error
|
Err error
|
||||||
}
|
}
|
||||||
|
|
||||||
type Numbers []float64
|
type Numbers []float64
|
||||||
|
|
||||||
type Function func(Numbers) R
|
type Function func(Numbers) Result
|
||||||
|
|
||||||
// every function we are able to call must be of type Funcall, which
|
// every function we are able to call must be of type Funcall, which
|
||||||
// needs to specify how many numbers it expects and the actual go
|
// needs to specify how many numbers it expects and the actual go
|
||||||
@@ -64,446 +64,450 @@ func NewFuncall(function Function, expectargs ...int) *Funcall {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Convenience function, create new result
|
// Convenience function, create new result
|
||||||
func NewR(n float64, e error) R {
|
func NewResult(n float64, e error) Result {
|
||||||
return R{Res: n, Err: e}
|
return Result{Res: n, Err: e}
|
||||||
}
|
}
|
||||||
|
|
||||||
// the actual functions, called once during initialization.
|
// the actual functions, called once during initialization.
|
||||||
func DefineFunctions() Funcalls {
|
func DefineFunctions() Funcalls {
|
||||||
f := map[string]*Funcall{
|
funcmap := map[string]*Funcall{
|
||||||
// simple operators, they all expect 2 args
|
// simple operators, they all expect 2 args
|
||||||
"+": NewFuncall(
|
"+": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]+arg[1], nil)
|
return NewResult(arg[0]+arg[1], nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"-": NewFuncall(
|
"-": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]-arg[1], nil)
|
return NewResult(arg[0]-arg[1], nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"x": NewFuncall(
|
"x": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]*arg[1], nil)
|
return NewResult(arg[0]*arg[1], nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"/": NewFuncall(
|
"/": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
if arg[1] == 0 {
|
if arg[1] == 0 {
|
||||||
return NewR(0, errors.New("division by null"))
|
return NewResult(0, errors.New("division by null"))
|
||||||
}
|
}
|
||||||
|
|
||||||
return NewR(arg[0]/arg[1], nil)
|
return NewResult(arg[0]/arg[1], nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"^": NewFuncall(
|
"^": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Pow(arg[0], arg[1]), nil)
|
return NewResult(math.Pow(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"%": NewFuncall(
|
"%": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR((arg[0]/100)*arg[1], nil)
|
return NewResult((arg[0]/100)*arg[1], nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"%-": NewFuncall(
|
"%-": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]-((arg[0]/100)*arg[1]), nil)
|
return NewResult(arg[0]-((arg[0]/100)*arg[1]), nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"%+": NewFuncall(
|
"%+": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]+((arg[0]/100)*arg[1]), nil)
|
return NewResult(arg[0]+((arg[0]/100)*arg[1]), nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"mod": NewFuncall(
|
"mod": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Remainder(arg[0], arg[1]), nil)
|
return NewResult(math.Remainder(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
|
|
||||||
"sqrt": NewFuncall(
|
"sqrt": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Sqrt(arg[0]), nil)
|
return NewResult(math.Sqrt(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"abs": NewFuncall(
|
"abs": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Abs(arg[0]), nil)
|
return NewResult(math.Abs(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"acos": NewFuncall(
|
"acos": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Acos(arg[0]), nil)
|
return NewResult(math.Acos(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"acosh": NewFuncall(
|
"acosh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Acosh(arg[0]), nil)
|
return NewResult(math.Acosh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"asin": NewFuncall(
|
"asin": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Asin(arg[0]), nil)
|
return NewResult(math.Asin(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"asinh": NewFuncall(
|
"asinh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Asinh(arg[0]), nil)
|
return NewResult(math.Asinh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"atan": NewFuncall(
|
"atan": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Atan(arg[0]), nil)
|
return NewResult(math.Atan(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"atan2": NewFuncall(
|
"atan2": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Atan2(arg[0], arg[1]), nil)
|
return NewResult(math.Atan2(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"atanh": NewFuncall(
|
"atanh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Atanh(arg[0]), nil)
|
return NewResult(math.Atanh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"cbrt": NewFuncall(
|
"cbrt": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Cbrt(arg[0]), nil)
|
return NewResult(math.Cbrt(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"ceil": NewFuncall(
|
"ceil": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Ceil(arg[0]), nil)
|
return NewResult(math.Ceil(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"cos": NewFuncall(
|
"cos": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Cos(arg[0]), nil)
|
return NewResult(math.Cos(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"cosh": NewFuncall(
|
"cosh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Cosh(arg[0]), nil)
|
return NewResult(math.Cosh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"erf": NewFuncall(
|
"erf": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Erf(arg[0]), nil)
|
return NewResult(math.Erf(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"erfc": NewFuncall(
|
"erfc": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Erfc(arg[0]), nil)
|
return NewResult(math.Erfc(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"erfcinv": NewFuncall(
|
"erfcinv": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Erfcinv(arg[0]), nil)
|
return NewResult(math.Erfcinv(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"erfinv": NewFuncall(
|
"erfinv": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Erfinv(arg[0]), nil)
|
return NewResult(math.Erfinv(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"exp": NewFuncall(
|
"exp": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Exp(arg[0]), nil)
|
return NewResult(math.Exp(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"exp2": NewFuncall(
|
"exp2": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Exp2(arg[0]), nil)
|
return NewResult(math.Exp2(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"expm1": NewFuncall(
|
"expm1": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Expm1(arg[0]), nil)
|
return NewResult(math.Expm1(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"floor": NewFuncall(
|
"floor": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Floor(arg[0]), nil)
|
return NewResult(math.Floor(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"gamma": NewFuncall(
|
"gamma": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Gamma(arg[0]), nil)
|
return NewResult(math.Gamma(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"ilogb": NewFuncall(
|
"ilogb": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(float64(math.Ilogb(arg[0])), nil)
|
return NewResult(float64(math.Ilogb(arg[0])), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"j0": NewFuncall(
|
"j0": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.J0(arg[0]), nil)
|
return NewResult(math.J0(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"j1": NewFuncall(
|
"j1": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.J1(arg[0]), nil)
|
return NewResult(math.J1(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"log": NewFuncall(
|
"log": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Log(arg[0]), nil)
|
return NewResult(math.Log(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"log10": NewFuncall(
|
"log10": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Log10(arg[0]), nil)
|
return NewResult(math.Log10(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"log1p": NewFuncall(
|
"log1p": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Log1p(arg[0]), nil)
|
return NewResult(math.Log1p(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"log2": NewFuncall(
|
"log2": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Log2(arg[0]), nil)
|
return NewResult(math.Log2(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"logb": NewFuncall(
|
"logb": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Logb(arg[0]), nil)
|
return NewResult(math.Logb(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"pow": NewFuncall(
|
"pow": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Pow(arg[0], arg[1]), nil)
|
return NewResult(math.Pow(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"round": NewFuncall(
|
"round": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Round(arg[0]), nil)
|
return NewResult(math.Round(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"roundtoeven": NewFuncall(
|
"roundtoeven": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.RoundToEven(arg[0]), nil)
|
return NewResult(math.RoundToEven(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"sin": NewFuncall(
|
"sin": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Sin(arg[0]), nil)
|
return NewResult(math.Sin(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"sinh": NewFuncall(
|
"sinh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Sinh(arg[0]), nil)
|
return NewResult(math.Sinh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"tan": NewFuncall(
|
"tan": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Tan(arg[0]), nil)
|
return NewResult(math.Tan(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"tanh": NewFuncall(
|
"tanh": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Tanh(arg[0]), nil)
|
return NewResult(math.Tanh(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"trunc": NewFuncall(
|
"trunc": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Trunc(arg[0]), nil)
|
return NewResult(math.Trunc(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"y0": NewFuncall(
|
"y0": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Y0(arg[0]), nil)
|
return NewResult(math.Y0(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"y1": NewFuncall(
|
"y1": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Y1(arg[0]), nil)
|
return NewResult(math.Y1(arg[0]), nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"copysign": NewFuncall(
|
"copysign": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Copysign(arg[0], arg[1]), nil)
|
return NewResult(math.Copysign(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"dim": NewFuncall(
|
"dim": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Dim(arg[0], arg[1]), nil)
|
return NewResult(math.Dim(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"hypot": NewFuncall(
|
"hypot": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(math.Hypot(arg[0], arg[1]), nil)
|
return NewResult(math.Hypot(arg[0], arg[1]), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
// converters of all kinds
|
// converters of all kinds
|
||||||
"cm-to-inch": NewFuncall(
|
"cm-to-inch": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]/2.54, nil)
|
return NewResult(arg[0]/2.54, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"inch-to-cm": NewFuncall(
|
"inch-to-cm": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]*2.54, nil)
|
return NewResult(arg[0]*2.54, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"gallons-to-liters": NewFuncall(
|
"gallons-to-liters": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]*3.785, nil)
|
return NewResult(arg[0]*3.785, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"liters-to-gallons": NewFuncall(
|
"liters-to-gallons": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]/3.785, nil)
|
return NewResult(arg[0]/3.785, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"yards-to-meters": NewFuncall(
|
"yards-to-meters": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]*91.44, nil)
|
return NewResult(arg[0]*91.44, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"meters-to-yards": NewFuncall(
|
"meters-to-yards": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]/91.44, nil)
|
return NewResult(arg[0]/91.44, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"miles-to-kilometers": NewFuncall(
|
"miles-to-kilometers": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]*1.609, nil)
|
return NewResult(arg[0]*1.609, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"kilometers-to-miles": NewFuncall(
|
"kilometers-to-miles": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(arg[0]/1.609, nil)
|
return NewResult(arg[0]/1.609, nil)
|
||||||
},
|
},
|
||||||
1),
|
1),
|
||||||
|
|
||||||
"or": NewFuncall(
|
"or": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(float64(int(arg[0])|int(arg[1])), nil)
|
return NewResult(float64(int(arg[0])|int(arg[1])), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"and": NewFuncall(
|
"and": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(float64(int(arg[0])&int(arg[1])), nil)
|
return NewResult(float64(int(arg[0])&int(arg[1])), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"xor": NewFuncall(
|
"xor": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
return NewR(float64(int(arg[0])^int(arg[1])), nil)
|
return NewResult(float64(int(arg[0])^int(arg[1])), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
"<": NewFuncall(
|
"<": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
// Shift by negative number provibited, so check it.
|
// Shift by negative number provibited, so check it.
|
||||||
// Note that we check agains uint64 overflow as well here
|
// Note that we check against uint64 overflow as well here
|
||||||
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
||||||
return NewR(0, errors.New("negative shift amount"))
|
return NewResult(0, errors.New("negative shift amount"))
|
||||||
}
|
}
|
||||||
return NewR(float64(int(arg[0])<<int(arg[1])), nil)
|
|
||||||
|
return NewResult(float64(int(arg[0])<<int(arg[1])), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
|
|
||||||
">": NewFuncall(
|
">": NewFuncall(
|
||||||
func(arg Numbers) R {
|
func(arg Numbers) Result {
|
||||||
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
||||||
return NewR(0, errors.New("negative shift amount"))
|
return NewResult(0, errors.New("negative shift amount"))
|
||||||
}
|
}
|
||||||
return NewR(float64(int(arg[0])>>int(arg[1])), nil)
|
|
||||||
|
return NewResult(float64(int(arg[0])>>int(arg[1])), nil)
|
||||||
},
|
},
|
||||||
2),
|
2),
|
||||||
}
|
}
|
||||||
|
|
||||||
// aliases
|
// aliases
|
||||||
f["*"] = f["x"]
|
funcmap["*"] = funcmap["x"]
|
||||||
f["remainder"] = f["mod"]
|
funcmap["remainder"] = funcmap["mod"]
|
||||||
|
|
||||||
return f
|
return funcmap
|
||||||
}
|
}
|
||||||
|
|
||||||
func DefineBatchFunctions() Funcalls {
|
func DefineBatchFunctions() Funcalls {
|
||||||
f := map[string]*Funcall{
|
funcmap := map[string]*Funcall{
|
||||||
"median": NewFuncall(
|
"median": NewFuncall(
|
||||||
func(args Numbers) R {
|
func(args Numbers) Result {
|
||||||
middle := len(args) / 2
|
middle := len(args) / 2
|
||||||
return NewR(args[middle], nil)
|
|
||||||
|
return NewResult(args[middle], nil)
|
||||||
},
|
},
|
||||||
-1),
|
-1),
|
||||||
|
|
||||||
"mean": NewFuncall(
|
"mean": NewFuncall(
|
||||||
func(args Numbers) R {
|
func(args Numbers) Result {
|
||||||
var sum float64
|
var sum float64
|
||||||
for _, item := range args {
|
for _, item := range args {
|
||||||
sum += item
|
sum += item
|
||||||
}
|
}
|
||||||
return NewR(sum/float64(len(args)), nil)
|
|
||||||
|
return NewResult(sum/float64(len(args)), nil)
|
||||||
},
|
},
|
||||||
-1),
|
-1),
|
||||||
|
|
||||||
"min": NewFuncall(
|
"min": NewFuncall(
|
||||||
func(args Numbers) R {
|
func(args Numbers) Result {
|
||||||
var min float64
|
var min float64
|
||||||
min, args = args[0], args[1:]
|
min, args = args[0], args[1:]
|
||||||
for _, item := range args {
|
for _, item := range args {
|
||||||
@@ -511,12 +515,13 @@ func DefineBatchFunctions() Funcalls {
|
|||||||
min = item
|
min = item
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return NewR(min, nil)
|
|
||||||
|
return NewResult(min, nil)
|
||||||
},
|
},
|
||||||
-1),
|
-1),
|
||||||
|
|
||||||
"max": NewFuncall(
|
"max": NewFuncall(
|
||||||
func(args Numbers) R {
|
func(args Numbers) Result {
|
||||||
var max float64
|
var max float64
|
||||||
max, args = args[0], args[1:]
|
max, args = args[0], args[1:]
|
||||||
for _, item := range args {
|
for _, item := range args {
|
||||||
@@ -524,24 +529,26 @@ func DefineBatchFunctions() Funcalls {
|
|||||||
max = item
|
max = item
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return NewR(max, nil)
|
|
||||||
|
return NewResult(max, nil)
|
||||||
},
|
},
|
||||||
-1),
|
-1),
|
||||||
|
|
||||||
"sum": NewFuncall(
|
"sum": NewFuncall(
|
||||||
func(args Numbers) R {
|
func(args Numbers) Result {
|
||||||
var sum float64
|
var sum float64
|
||||||
for _, item := range args {
|
for _, item := range args {
|
||||||
sum += item
|
sum += item
|
||||||
}
|
}
|
||||||
return NewR(sum, nil)
|
|
||||||
|
return NewResult(sum, nil)
|
||||||
},
|
},
|
||||||
-1),
|
-1),
|
||||||
}
|
}
|
||||||
|
|
||||||
// aliases
|
// aliases
|
||||||
f["+"] = f["sum"]
|
funcmap["+"] = funcmap["sum"]
|
||||||
f["avg"] = f["mean"]
|
funcmap["avg"] = funcmap["mean"]
|
||||||
|
|
||||||
return f
|
return funcmap
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -29,8 +29,8 @@ type Interpreter struct {
|
|||||||
script string
|
script string
|
||||||
}
|
}
|
||||||
|
|
||||||
// LUA interpreter, instanciated in main()
|
// LuaInterpreter is the lua interpreter, instantiated in main()
|
||||||
var L *lua.LState
|
var LuaInterpreter *lua.LState
|
||||||
|
|
||||||
// holds a user provided lua function
|
// holds a user provided lua function
|
||||||
type LuaFunction struct {
|
type LuaFunction struct {
|
||||||
@@ -39,8 +39,8 @@ type LuaFunction struct {
|
|||||||
numargs int
|
numargs int
|
||||||
}
|
}
|
||||||
|
|
||||||
// must be global since init() is being called from lua which doesn't
|
// LuaFuncs must be global since init() is being called from lua which
|
||||||
// have access to the interpreter instance
|
// doesn't have access to the interpreter instance
|
||||||
var LuaFuncs map[string]LuaFunction
|
var LuaFuncs map[string]LuaFunction
|
||||||
|
|
||||||
func NewInterpreter(script string, debug bool) *Interpreter {
|
func NewInterpreter(script string, debug bool) *Interpreter {
|
||||||
@@ -61,8 +61,8 @@ func (i *Interpreter) InitLua() {
|
|||||||
{lua.DebugLibName, lua.OpenDebug},
|
{lua.DebugLibName, lua.OpenDebug},
|
||||||
{lua.MathLibName, lua.OpenMath},
|
{lua.MathLibName, lua.OpenMath},
|
||||||
} {
|
} {
|
||||||
if err := L.CallByParam(lua.P{
|
if err := LuaInterpreter.CallByParam(lua.P{
|
||||||
Fn: L.NewFunction(pair.f),
|
Fn: LuaInterpreter.NewFunction(pair.f),
|
||||||
NRet: 0,
|
NRet: 0,
|
||||||
Protect: true,
|
Protect: true,
|
||||||
}, lua.LString(pair.n)); err != nil {
|
}, lua.LString(pair.n)); err != nil {
|
||||||
@@ -71,19 +71,19 @@ func (i *Interpreter) InitLua() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// load the lua config (which we expect to contain init() and math functions)
|
// load the lua config (which we expect to contain init() and math functions)
|
||||||
if err := L.DoFile(i.script); err != nil {
|
if err := LuaInterpreter.DoFile(i.script); err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// instanciate
|
// instantiate
|
||||||
LuaFuncs = map[string]LuaFunction{}
|
LuaFuncs = map[string]LuaFunction{}
|
||||||
|
|
||||||
// that way the user can call register(...) from lua inside init()
|
// that way the user can call register(...) from lua inside init()
|
||||||
L.SetGlobal("register", L.NewFunction(register))
|
LuaInterpreter.SetGlobal("register", LuaInterpreter.NewFunction(register))
|
||||||
|
|
||||||
// actually call init()
|
// actually call init()
|
||||||
if err := L.CallByParam(lua.P{
|
if err := LuaInterpreter.CallByParam(lua.P{
|
||||||
Fn: L.GetGlobal("init"),
|
Fn: LuaInterpreter.GetGlobal("init"),
|
||||||
NRet: 0,
|
NRet: 0,
|
||||||
Protect: true,
|
Protect: true,
|
||||||
}); err != nil {
|
}); err != nil {
|
||||||
@@ -108,9 +108,9 @@ func (i *Interpreter) FuncNumArgs(name string) int {
|
|||||||
// arguments. 1 uses the last item of the stack, 2 the last two and -1
|
// arguments. 1 uses the last item of the stack, 2 the last two and -1
|
||||||
// all items (which translates to batch mode)
|
// all items (which translates to batch mode)
|
||||||
//
|
//
|
||||||
// The items array will be provded by calc.Eval(), these are
|
// The items array will be provided by calc.Eval(), these are
|
||||||
// non-popped stack items. So the items will only removed from the
|
// non-popped stack items. So the items will only removed from the
|
||||||
// stack when the lua function execution is successfull.
|
// stack when the lua function execution is successful.
|
||||||
func (i *Interpreter) CallLuaFunc(funcname string, items []float64) (float64, error) {
|
func (i *Interpreter) CallLuaFunc(funcname string, items []float64) (float64, error) {
|
||||||
i.Debug(fmt.Sprintf("calling lua func %s() with %d args",
|
i.Debug(fmt.Sprintf("calling lua func %s() with %d args",
|
||||||
funcname, LuaFuncs[funcname].numargs))
|
funcname, LuaFuncs[funcname].numargs))
|
||||||
@@ -120,44 +120,44 @@ func (i *Interpreter) CallLuaFunc(funcname string, items []float64) (float64, er
|
|||||||
fallthrough
|
fallthrough
|
||||||
case 1:
|
case 1:
|
||||||
// 1 arg variant
|
// 1 arg variant
|
||||||
if err := L.CallByParam(lua.P{
|
if err := LuaInterpreter.CallByParam(lua.P{
|
||||||
Fn: L.GetGlobal(funcname),
|
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||||
NRet: 1,
|
NRet: 1,
|
||||||
Protect: true,
|
Protect: true,
|
||||||
}, lua.LNumber(items[0])); err != nil {
|
}, lua.LNumber(items[0])); err != nil {
|
||||||
fmt.Println(err)
|
return 0, fmt.Errorf("failed to exec lua func %s: %w", funcname, err)
|
||||||
return 0, err
|
|
||||||
}
|
}
|
||||||
case 2:
|
case 2:
|
||||||
// 2 arg variant
|
// 2 arg variant
|
||||||
if err := L.CallByParam(lua.P{
|
if err := LuaInterpreter.CallByParam(lua.P{
|
||||||
Fn: L.GetGlobal(funcname),
|
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||||
NRet: 1,
|
NRet: 1,
|
||||||
Protect: true,
|
Protect: true,
|
||||||
}, lua.LNumber(items[0]), lua.LNumber(items[1])); err != nil {
|
}, lua.LNumber(items[0]), lua.LNumber(items[1])); err != nil {
|
||||||
return 0, err
|
return 0, fmt.Errorf("failed to exec lua func %s: %w", funcname, err)
|
||||||
}
|
}
|
||||||
case -1:
|
case -1:
|
||||||
// batch variant, use lua table as array
|
// batch variant, use lua table as array
|
||||||
tb := L.NewTable()
|
table := LuaInterpreter.NewTable()
|
||||||
|
|
||||||
// put the whole stack into it
|
// put the whole stack into it
|
||||||
for _, item := range items {
|
for _, item := range items {
|
||||||
tb.Append(lua.LNumber(item))
|
table.Append(lua.LNumber(item))
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := L.CallByParam(lua.P{
|
if err := LuaInterpreter.CallByParam(lua.P{
|
||||||
Fn: L.GetGlobal(funcname),
|
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||||
NRet: 1,
|
NRet: 1,
|
||||||
Protect: true,
|
Protect: true,
|
||||||
}, tb); err != nil {
|
}, table); err != nil {
|
||||||
return 0, err
|
return 0, fmt.Errorf("failed to exec lua func %s: %w", funcname, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// get result and cast to float64
|
// get result and cast to float64
|
||||||
if res, ok := L.Get(-1).(lua.LNumber); ok {
|
if res, ok := LuaInterpreter.Get(-1).(lua.LNumber); ok {
|
||||||
L.Pop(1)
|
LuaInterpreter.Pop(1)
|
||||||
|
|
||||||
return float64(res), nil
|
return float64(res), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -167,10 +167,10 @@ func (i *Interpreter) CallLuaFunc(funcname string, items []float64) (float64, er
|
|||||||
// called from lua to register a math function numargs may be 1, 2 or
|
// called from lua to register a math function numargs may be 1, 2 or
|
||||||
// -1, it denotes the number of items from the stack requested by the
|
// -1, it denotes the number of items from the stack requested by the
|
||||||
// lua function. -1 means batch mode, that is all items
|
// lua function. -1 means batch mode, that is all items
|
||||||
func register(L *lua.LState) int {
|
func register(lstate *lua.LState) int {
|
||||||
function := L.ToString(1)
|
function := lstate.ToString(1)
|
||||||
numargs := L.ToInt(2)
|
numargs := lstate.ToInt(2)
|
||||||
help := L.ToString(3)
|
help := lstate.ToString(3)
|
||||||
|
|
||||||
LuaFuncs[function] = LuaFunction{
|
LuaFuncs[function] = LuaFunction{
|
||||||
name: function,
|
name: function,
|
||||||
|
|||||||
35
main.go
35
main.go
@@ -81,11 +81,13 @@ func Main() int {
|
|||||||
|
|
||||||
if showversion {
|
if showversion {
|
||||||
fmt.Printf("This is rpn version %s\n", VERSION)
|
fmt.Printf("This is rpn version %s\n", VERSION)
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
if showhelp {
|
if showhelp {
|
||||||
fmt.Println(Usage)
|
fmt.Println(Usage)
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -95,12 +97,13 @@ func Main() int {
|
|||||||
|
|
||||||
if showmanual {
|
if showmanual {
|
||||||
man()
|
man()
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
// the lua state object is global, instanciate it early
|
// the lua state object is global, instantiate it early
|
||||||
L = lua.NewState(lua.Options{SkipOpenLibs: true})
|
LuaInterpreter = lua.NewState(lua.Options{SkipOpenLibs: true})
|
||||||
defer L.Close()
|
defer LuaInterpreter.Close()
|
||||||
|
|
||||||
// our config file is interpreted as lua code, only functions can
|
// our config file is interpreted as lua code, only functions can
|
||||||
// be defined, init() will be called by InitLua().
|
// be defined, init() will be called by InitLua().
|
||||||
@@ -108,14 +111,13 @@ func Main() int {
|
|||||||
luarunner := NewInterpreter(configfile, enabledebug)
|
luarunner := NewInterpreter(configfile, enabledebug)
|
||||||
luarunner.InitLua()
|
luarunner.InitLua()
|
||||||
calc.SetInt(luarunner)
|
calc.SetInt(luarunner)
|
||||||
|
|
||||||
if calc.debug {
|
if calc.debug {
|
||||||
fmt.Println("loaded config")
|
fmt.Println("loaded config")
|
||||||
}
|
}
|
||||||
} else {
|
} else if calc.debug {
|
||||||
if calc.debug {
|
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if len(flag.Args()) > 1 {
|
if len(flag.Args()) > 1 {
|
||||||
// commandline calc operation, no readline etc needed
|
// commandline calc operation, no readline etc needed
|
||||||
@@ -123,6 +125,7 @@ func Main() int {
|
|||||||
calc.stdin = true
|
calc.stdin = true
|
||||||
if err := calc.Eval(strings.Join(flag.Args(), " ")); err != nil {
|
if err := calc.Eval(strings.Join(flag.Args(), " ")); err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
|
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -130,7 +133,7 @@ func Main() int {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// interactive mode, need readline
|
// interactive mode, need readline
|
||||||
rl, err := readline.NewEx(&readline.Config{
|
reader, err := readline.NewEx(&readline.Config{
|
||||||
Prompt: calc.Prompt(),
|
Prompt: calc.Prompt(),
|
||||||
HistoryFile: os.Getenv("HOME") + "/.rpn-history",
|
HistoryFile: os.Getenv("HOME") + "/.rpn-history",
|
||||||
HistoryLimit: 500,
|
HistoryLimit: 500,
|
||||||
@@ -143,8 +146,8 @@ func Main() int {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
defer rl.Close()
|
defer reader.Close()
|
||||||
rl.CaptureExitSignal()
|
reader.CaptureExitSignal()
|
||||||
|
|
||||||
if inputIsStdin() {
|
if inputIsStdin() {
|
||||||
// commands are coming on stdin, however we will still enter
|
// commands are coming on stdin, however we will still enter
|
||||||
@@ -154,7 +157,7 @@ func Main() int {
|
|||||||
|
|
||||||
for {
|
for {
|
||||||
// primary program repl
|
// primary program repl
|
||||||
line, err := rl.Readline()
|
line, err := reader.Readline()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
@@ -163,7 +166,8 @@ func Main() int {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
}
|
}
|
||||||
rl.SetPrompt(calc.Prompt())
|
|
||||||
|
reader.SetPrompt(calc.Prompt())
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(flag.Args()) > 0 {
|
if len(flag.Args()) > 0 {
|
||||||
@@ -173,6 +177,7 @@ func Main() int {
|
|||||||
calc.batch = true
|
calc.batch = true
|
||||||
if err = calc.Eval(flag.Args()[0]); err != nil {
|
if err = calc.Eval(flag.Args()[0]); err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
|
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -182,17 +187,19 @@ func Main() int {
|
|||||||
|
|
||||||
func inputIsStdin() bool {
|
func inputIsStdin() bool {
|
||||||
stat, _ := os.Stdin.Stat()
|
stat, _ := os.Stdin.Stat()
|
||||||
|
|
||||||
return (stat.Mode() & os.ModeCharDevice) == 0
|
return (stat.Mode() & os.ModeCharDevice) == 0
|
||||||
}
|
}
|
||||||
|
|
||||||
func man() {
|
func man() {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
|
||||||
man := exec.Command("less", "-")
|
man := exec.Command("less", "-")
|
||||||
|
|
||||||
var b bytes.Buffer
|
buf.WriteString(manpage)
|
||||||
b.Write([]byte(manpage))
|
|
||||||
|
|
||||||
man.Stdout = os.Stdout
|
man.Stdout = os.Stdout
|
||||||
man.Stdin = &b
|
man.Stdin = &buf
|
||||||
man.Stderr = os.Stderr
|
man.Stderr = os.Stderr
|
||||||
|
|
||||||
err := man.Run()
|
err := man.Run()
|
||||||
|
|||||||
4
rpn.go
4
rpn.go
@@ -129,7 +129,7 @@ DESCRIPTION
|
|||||||
Basic operators:
|
Basic operators:
|
||||||
|
|
||||||
+ add
|
+ add
|
||||||
- substract
|
- subtract
|
||||||
/ divide
|
/ divide
|
||||||
x multiply (alias: *)
|
x multiply (alias: *)
|
||||||
^ power
|
^ power
|
||||||
@@ -145,7 +145,7 @@ DESCRIPTION
|
|||||||
Percent functions:
|
Percent functions:
|
||||||
|
|
||||||
% percent
|
% percent
|
||||||
%- substract percent
|
%- subtract percent
|
||||||
%+ add percent
|
%+ add percent
|
||||||
|
|
||||||
Batch functions:
|
Batch functions:
|
||||||
|
|||||||
4
rpn.pod
4
rpn.pod
@@ -136,7 +136,7 @@ stack.
|
|||||||
Basic operators:
|
Basic operators:
|
||||||
|
|
||||||
+ add
|
+ add
|
||||||
- substract
|
- subtract
|
||||||
/ divide
|
/ divide
|
||||||
x multiply (alias: *)
|
x multiply (alias: *)
|
||||||
^ power
|
^ power
|
||||||
@@ -152,7 +152,7 @@ Bitwise operators:
|
|||||||
Percent functions:
|
Percent functions:
|
||||||
|
|
||||||
% percent
|
% percent
|
||||||
%- substract percent
|
%- subtract percent
|
||||||
%+ add percent
|
%+ add percent
|
||||||
|
|
||||||
Batch functions:
|
Batch functions:
|
||||||
|
|||||||
31
stack.go
31
stack.go
@@ -64,14 +64,14 @@ func (s *Stack) Bump() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// append an item to the stack
|
// append an item to the stack
|
||||||
func (s *Stack) Push(x float64) {
|
func (s *Stack) Push(item float64) {
|
||||||
s.mutex.Lock()
|
s.mutex.Lock()
|
||||||
defer s.mutex.Unlock()
|
defer s.mutex.Unlock()
|
||||||
|
|
||||||
s.Debug(fmt.Sprintf(" push to stack: %.2f", x))
|
s.Debug(fmt.Sprintf(" push to stack: %.2f", item))
|
||||||
|
|
||||||
s.Bump()
|
s.Bump()
|
||||||
s.linklist.PushBack(x)
|
s.linklist.PushBack(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
// remove and return an item from the stack
|
// remove and return an item from the stack
|
||||||
@@ -90,6 +90,7 @@ func (s *Stack) Pop() float64 {
|
|||||||
s.Debug(fmt.Sprintf(" remove from stack: %.2f", val))
|
s.Debug(fmt.Sprintf(" remove from stack: %.2f", val))
|
||||||
|
|
||||||
s.Bump()
|
s.Bump()
|
||||||
|
|
||||||
return val.(float64)
|
return val.(float64)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -123,32 +124,33 @@ func (s *Stack) Swap() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
a := s.linklist.Back()
|
prevA := s.linklist.Back()
|
||||||
s.linklist.Remove(a)
|
s.linklist.Remove(prevA)
|
||||||
|
|
||||||
b := s.linklist.Back()
|
prevB := s.linklist.Back()
|
||||||
s.linklist.Remove(b)
|
s.linklist.Remove(prevB)
|
||||||
|
|
||||||
s.Debug(fmt.Sprintf("swapping %.2f with %.2f", b.Value, a.Value))
|
s.Debug(fmt.Sprintf("swapping %.2f with %.2f", prevB.Value, prevA.Value))
|
||||||
|
|
||||||
s.linklist.PushBack(a.Value)
|
s.linklist.PushBack(prevA.Value)
|
||||||
s.linklist.PushBack(b.Value)
|
s.linklist.PushBack(prevB.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return the last num items from the stack w/o modifying it.
|
// Return the last num items from the stack w/o modifying it.
|
||||||
func (s *Stack) Last(num ...int) []float64 {
|
func (s *Stack) Last(num ...int) []float64 {
|
||||||
items := []float64{}
|
items := []float64{}
|
||||||
i := s.Len()
|
stacklen := s.Len()
|
||||||
count := 1
|
count := 1
|
||||||
|
|
||||||
if len(num) > 0 {
|
if len(num) > 0 {
|
||||||
count = num[0]
|
count = num[0]
|
||||||
}
|
}
|
||||||
|
|
||||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||||
if i <= count {
|
if stacklen <= count {
|
||||||
items = append(items, e.Value.(float64))
|
items = append(items, e.Value.(float64))
|
||||||
}
|
}
|
||||||
i--
|
stacklen--
|
||||||
}
|
}
|
||||||
|
|
||||||
return items
|
return items
|
||||||
@@ -168,12 +170,14 @@ func (s *Stack) All() []float64 {
|
|||||||
// dump the stack to stdout, including backup if debug is enabled
|
// dump the stack to stdout, including backup if debug is enabled
|
||||||
func (s *Stack) Dump() {
|
func (s *Stack) Dump() {
|
||||||
fmt.Printf("Stack revision %d (%p):\n", s.rev, &s.linklist)
|
fmt.Printf("Stack revision %d (%p):\n", s.rev, &s.linklist)
|
||||||
|
|
||||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||||
fmt.Println(e.Value)
|
fmt.Println(e.Value)
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.debug {
|
if s.debug {
|
||||||
fmt.Printf("Backup stack revision %d (%p):\n", s.backuprev, &s.backup)
|
fmt.Printf("Backup stack revision %d (%p):\n", s.backuprev, &s.backup)
|
||||||
|
|
||||||
for e := s.backup.Front(); e != nil; e = e.Next() {
|
for e := s.backup.Front(); e != nil; e = e.Next() {
|
||||||
fmt.Println(e.Value)
|
fmt.Println(e.Value)
|
||||||
}
|
}
|
||||||
@@ -215,6 +219,7 @@ func (s *Stack) Restore() {
|
|||||||
|
|
||||||
if s.rev == 0 {
|
if s.rev == 0 {
|
||||||
fmt.Println("error: stack is empty.")
|
fmt.Println("error: stack is empty.")
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -35,16 +35,16 @@ func TestPush(t *testing.T) {
|
|||||||
|
|
||||||
func TestPop(t *testing.T) {
|
func TestPop(t *testing.T) {
|
||||||
t.Run("pop", func(t *testing.T) {
|
t.Run("pop", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
got := s.Pop()
|
got := stack.Pop()
|
||||||
|
|
||||||
if got != 5.0 {
|
if got != 5.0 {
|
||||||
t.Errorf("pop failed:\n+++ got: %f\n--- want: %f",
|
t.Errorf("pop failed:\n+++ got: %f\n--- want: %f",
|
||||||
got, 5.0)
|
got, 5.0)
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.Len() != 0 {
|
if stack.Len() != 0 {
|
||||||
t.Errorf("stack not empty after pop()")
|
t.Errorf("stack not empty after pop()")
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -52,25 +52,25 @@ func TestPop(t *testing.T) {
|
|||||||
|
|
||||||
func TestPops(t *testing.T) {
|
func TestPops(t *testing.T) {
|
||||||
t.Run("pops", func(t *testing.T) {
|
t.Run("pops", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Pop()
|
stack.Pop()
|
||||||
|
|
||||||
if s.Len() != 2 {
|
if stack.Len() != 2 {
|
||||||
t.Errorf("stack len not correct after pop:\n+++ got: %d\n--- want: %d",
|
t.Errorf("stack len not correct after pop:\n+++ got: %d\n--- want: %d",
|
||||||
s.Len(), 2)
|
stack.Len(), 2)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestShift(t *testing.T) {
|
func TestShift(t *testing.T) {
|
||||||
t.Run("shift", func(t *testing.T) {
|
t.Run("shift", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Shift()
|
stack.Shift()
|
||||||
|
|
||||||
if s.Len() != 0 {
|
if stack.Len() != 0 {
|
||||||
t.Errorf("stack not empty after shift()")
|
t.Errorf("stack not empty after shift()")
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -78,13 +78,13 @@ func TestShift(t *testing.T) {
|
|||||||
|
|
||||||
func TestClear(t *testing.T) {
|
func TestClear(t *testing.T) {
|
||||||
t.Run("clear", func(t *testing.T) {
|
t.Run("clear", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Clear()
|
stack.Clear()
|
||||||
|
|
||||||
if s.Len() != 0 {
|
if stack.Len() != 0 {
|
||||||
t.Errorf("stack not empty after clear()")
|
t.Errorf("stack not empty after clear()")
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -92,9 +92,9 @@ func TestClear(t *testing.T) {
|
|||||||
|
|
||||||
func TestLast(t *testing.T) {
|
func TestLast(t *testing.T) {
|
||||||
t.Run("last", func(t *testing.T) {
|
t.Run("last", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
got := s.Last()
|
got := stack.Last()
|
||||||
|
|
||||||
if len(got) != 1 {
|
if len(got) != 1 {
|
||||||
t.Errorf("last failed:\n+++ got: %d elements\n--- want: %d elements",
|
t.Errorf("last failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||||
@@ -106,7 +106,7 @@ func TestLast(t *testing.T) {
|
|||||||
got, 5.0)
|
got, 5.0)
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.Len() != 1 {
|
if stack.Len() != 1 {
|
||||||
t.Errorf("stack modified after last()")
|
t.Errorf("stack modified after last()")
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -114,14 +114,14 @@ func TestLast(t *testing.T) {
|
|||||||
|
|
||||||
func TestAll(t *testing.T) {
|
func TestAll(t *testing.T) {
|
||||||
t.Run("all", func(t *testing.T) {
|
t.Run("all", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
list := []float64{2, 4, 6, 8}
|
list := []float64{2, 4, 6, 8}
|
||||||
|
|
||||||
for _, item := range list {
|
for _, item := range list {
|
||||||
s.Push(item)
|
stack.Push(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
got := s.All()
|
got := stack.All()
|
||||||
|
|
||||||
if len(got) != len(list) {
|
if len(got) != len(list) {
|
||||||
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||||
@@ -135,7 +135,7 @@ func TestAll(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.Len() != len(list) {
|
if stack.Len() != len(list) {
|
||||||
t.Errorf("stack modified after last()")
|
t.Errorf("stack modified after last()")
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
@@ -143,37 +143,37 @@ func TestAll(t *testing.T) {
|
|||||||
|
|
||||||
func TestBackupRestore(t *testing.T) {
|
func TestBackupRestore(t *testing.T) {
|
||||||
t.Run("shift", func(t *testing.T) {
|
t.Run("shift", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
s.Push(5)
|
stack.Push(5)
|
||||||
s.Backup()
|
stack.Backup()
|
||||||
s.Clear()
|
stack.Clear()
|
||||||
s.Restore()
|
stack.Restore()
|
||||||
|
|
||||||
if s.Len() != 1 {
|
if stack.Len() != 1 {
|
||||||
t.Errorf("stack not correctly restored()")
|
t.Errorf("stack not correctly restored()")
|
||||||
}
|
}
|
||||||
|
|
||||||
a := s.Pop()
|
value := stack.Pop()
|
||||||
if a != 5.0 {
|
if value != 5.0 {
|
||||||
t.Errorf("stack not identical to old revision:\n+++ got: %f\n--- want: %f",
|
t.Errorf("stack not identical to old revision:\n+++ got: %f\n--- want: %f",
|
||||||
a, 5.0)
|
value, 5.0)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestReverse(t *testing.T) {
|
func TestReverse(t *testing.T) {
|
||||||
t.Run("reverse", func(t *testing.T) {
|
t.Run("reverse", func(t *testing.T) {
|
||||||
s := NewStack()
|
stack := NewStack()
|
||||||
list := []float64{2, 4, 6}
|
list := []float64{2, 4, 6}
|
||||||
reverse := []float64{6, 4, 2}
|
reverse := []float64{6, 4, 2}
|
||||||
|
|
||||||
for _, item := range list {
|
for _, item := range list {
|
||||||
s.Push(item)
|
stack.Push(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
s.Reverse()
|
stack.Reverse()
|
||||||
|
|
||||||
got := s.All()
|
got := stack.All()
|
||||||
|
|
||||||
if len(got) != len(list) {
|
if len(got) != len(list) {
|
||||||
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||||
|
|||||||
4
util.go
4
util.go
@@ -30,6 +30,7 @@ func contains[E comparable](s []E, v E) bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -38,6 +39,7 @@ func exists[K comparable, V any](m map[K]V, v K) bool {
|
|||||||
if _, ok := m[v]; ok {
|
if _, ok := m[v]; ok {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -73,5 +75,5 @@ func list2str(list Numbers) string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func Error(m string) error {
|
func Error(m string) error {
|
||||||
return fmt.Errorf("Error: %s!", m)
|
return fmt.Errorf("Error: %s", m)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user