2 Commits

13 changed files with 667 additions and 541 deletions

View File

@@ -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)

144
calc.go
View File

@@ -1,5 +1,5 @@
/* /*
Copyright © 2023 Thomas von Dein Copyright © 2023-2024 Thomas von Dein
This program is free software: you can redistribute it and/or modify This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by it under the terms of the GNU General Public License as published by
@@ -20,6 +20,7 @@ package main
import ( import (
"errors" "errors"
"fmt" "fmt"
"math"
"regexp" "regexp"
"sort" "sort"
"strconv" "strconv"
@@ -35,6 +36,8 @@ type Calc struct {
showstack bool showstack bool
intermediate bool intermediate bool
notdone bool // set to true as long as there are items left in the eval loop notdone bool // set to true as long as there are items left in the eval loop
precision int
stack *Stack stack *Stack
history []string history []string
completer readline.AutoCompleter completer readline.AutoCompleter
@@ -66,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):
@@ -89,8 +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
ShowStackLen int = 5
) )
// That way we can add custom functions to completion // That way we can add custom functions to completion
@@ -146,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} 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)
@@ -203,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]
@@ -247,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))
} }
@@ -263,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
} }
@@ -277,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
} }
@@ -296,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
} }
@@ -316,6 +330,7 @@ func (c *Calc) EvalItem(item string) error {
case "<": case "<":
c.GetVar(regmatches[2]) c.GetVar(regmatches[2])
} }
return nil return nil
} }
@@ -323,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
} }
@@ -350,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
} }
@@ -369,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 {
@@ -390,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!
@@ -410,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
} }
@@ -438,7 +458,16 @@ func (c *Calc) Result() float64 {
fmt.Print("= ") fmt.Print("= ")
} }
fmt.Println(c.stack.Last()[0]) result := c.stack.Last()[0]
truncated := math.Trunc(result)
precision := c.precision
if result == truncated {
precision = 0
}
format := fmt.Sprintf("%%.%df\n", precision)
fmt.Printf(format, result)
} }
return c.stack.Last()[0] return c.stack.Last()[0]
@@ -452,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
} }
@@ -480,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()
@@ -540,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)
@@ -568,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)
} }

View File

@@ -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?

View File

@@ -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
View File

@@ -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
} }

View File

@@ -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,

43
main.go
View File

@@ -1,5 +1,5 @@
/* /*
Copyright © 2023 Thomas von Dein Copyright © 2023-2024 Thomas von Dein
This program is free software: you can redistribute it and/or modify This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by it under the terms of the GNU General Public License as published by
@@ -30,7 +30,7 @@ import (
lua "github.com/yuin/gopher-lua" lua "github.com/yuin/gopher-lua"
) )
const VERSION string = "2.0.13" const VERSION string = "2.1.0"
const Usage string = `This is rpn, a reverse polish notation calculator cli. const Usage string = `This is rpn, a reverse polish notation calculator cli.
@@ -43,13 +43,14 @@ Options:
-i --intermediate print intermediate results -i --intermediate print intermediate results
-m, --manual show manual -m, --manual show manual
-c, --config <file> load <file> containing LUA code -c, --config <file> load <file> containing LUA code
-p, --precision <int> floating point number precision (default 2)
-v, --version show version -v, --version show version
-h, --help show help -h, --help show help
When <operator> is given, batch mode ist automatically enabled. Use When <operator> is given, batch mode ist automatically enabled. Use
this only when working with stdin. E.g.: echo "2 3 4 5" | rpn + this only when working with stdin. E.g.: echo "2 3 4 5" | rpn +
Copyright (c) 2023 T.v.Dein` Copyright (c) 2023-2024 T.v.Dein`
func main() { func main() {
os.Exit(Main()) os.Exit(Main())
@@ -74,16 +75,19 @@ func Main() int {
flag.BoolVarP(&showmanual, "manual", "m", false, "show manual") flag.BoolVarP(&showmanual, "manual", "m", false, "show manual")
flag.StringVarP(&configfile, "config", "c", flag.StringVarP(&configfile, "config", "c",
os.Getenv("HOME")+"/.rpn.lua", "config file (lua format)") os.Getenv("HOME")+"/.rpn.lua", "config file (lua format)")
flag.IntVarP(&calc.precision, "precision", "p", Precision, "floating point precision")
flag.Parse() flag.Parse()
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
} }
@@ -93,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().
@@ -106,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
@@ -121,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
} }
@@ -128,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,
@@ -141,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
@@ -152,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
} }
@@ -161,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 {
@@ -171,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
} }
} }
@@ -180,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()

17
rpn.go
View File

@@ -13,6 +13,8 @@ SYNOPSIS
-s, --stack show last 5 items of the stack (off by default) -s, --stack show last 5 items of the stack (off by default)
-i --intermediate print intermediate results -i --intermediate print intermediate results
-m, --manual show manual -m, --manual show manual
-c, --config <file> load <file> containing LUA code
-p, --precision <int> floating point number precision (default 2)
-v, --version show version -v, --version show version
-h, --help show help -h, --help show help
@@ -127,7 +129,7 @@ DESCRIPTION
Basic operators: Basic operators:
+ add + add
- substract - subtract
/ divide / divide
x multiply (alias: *) x multiply (alias: *)
^ power ^ power
@@ -143,7 +145,7 @@ DESCRIPTION
Percent functions: Percent functions:
% percent % percent
%- substract percent %- subtract percent
%+ add percent %+ add percent
Batch functions: Batch functions:
@@ -309,6 +311,15 @@ EXTENDING RPN USING LUA
So you can't open files, execute other programs or open a connection to So you can't open files, execute other programs or open a connection to
the outside! the outside!
CONFIGURATION
rpn can be configured via command line flags (see usage above). Most of
the flags are also available as interactive commands, such as "--batch"
has the same effect as the batch command.
The floating point number precision option "-p, --precision" however is
not available as interactive command, it MUST be configured on the
command line, if needed. The default precision is 2.
GETTING HELP GETTING HELP
In interactive mode you can enter the help command (or ?) to get a short In interactive mode you can enter the help command (or ?) to get a short
help along with a list of all supported operators and functions. help along with a list of all supported operators and functions.
@@ -328,7 +339,7 @@ LICENSE
This software is licensed under the GNU GENERAL PUBLIC LICENSE version This software is licensed under the GNU GENERAL PUBLIC LICENSE version
3. 3.
Copyright (c) 2023 by Thomas von Dein Copyright (c) 2023-2024 by Thomas von Dein
This software uses the following GO modules: This software uses the following GO modules:

18
rpn.pod
View File

@@ -12,6 +12,8 @@ rpn - Programmable command-line calculator using reverse polish notation
-s, --stack show last 5 items of the stack (off by default) -s, --stack show last 5 items of the stack (off by default)
-i --intermediate print intermediate results -i --intermediate print intermediate results
-m, --manual show manual -m, --manual show manual
-c, --config <file> load <file> containing LUA code
-p, --precision <int> floating point number precision (default 2)
-v, --version show version -v, --version show version
-h, --help show help -h, --help show help
@@ -134,7 +136,7 @@ stack.
Basic operators: Basic operators:
+ add + add
- substract - subtract
/ divide / divide
x multiply (alias: *) x multiply (alias: *)
^ power ^ power
@@ -150,7 +152,7 @@ Bitwise operators:
Percent functions: Percent functions:
% percent % percent
%- substract percent %- subtract percent
%+ add percent %+ add percent
Batch functions: Batch functions:
@@ -342,6 +344,16 @@ B<Please note, that io, networking and system stuff is not allowed
though. So you can't open files, execute other programs or open a though. So you can't open files, execute other programs or open a
connection to the outside!> connection to the outside!>
=head1 CONFIGURATION
B<rpn> can be configured via command line flags (see usage
above). Most of the flags are also available as interactive commands,
such as C<--batch> has the same effect as the B<batch> command.
The floating point number precision option C<-p, --precision> however
is not available as interactive command, it MUST be configured on the
command line, if needed. The default precision is 2.
=head1 GETTING HELP =head1 GETTING HELP
In interactive mode you can enter the B<help> command (or B<?>) to get In interactive mode you can enter the B<help> command (or B<?>) to get
@@ -364,7 +376,7 @@ L<https://github.com/TLINDEN/rpnc/issues>.
This software is licensed under the GNU GENERAL PUBLIC LICENSE version 3. This software is licensed under the GNU GENERAL PUBLIC LICENSE version 3.
Copyright (c) 2023 by Thomas von Dein Copyright (c) 2023-2024 by Thomas von Dein
This software uses the following GO modules: This software uses the following GO modules:

View File

@@ -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
} }

View File

@@ -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",

View File

@@ -0,0 +1,2 @@
exec testrpn -p 4 2 3 /
stdout '0.6667\n'

View File

@@ -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)
} }