mirror of
https://codeberg.org/scip/rpnc.git
synced 2025-12-17 12:31:04 +01:00
Compare commits
41 Commits
v2.0.1
...
intern/lin
| Author | SHA1 | Date | |
|---|---|---|---|
| c144658a53 | |||
| 6a2a501e48 | |||
|
|
e81be12b19 | ||
| 222dc3a734 | |||
| 49e01565b9 | |||
| e4a8af9b5b | |||
| ac9d08d6fc | |||
|
|
cb774b3b80 | ||
|
|
846b3e63fc | ||
|
|
5557ad5f99 | ||
|
|
d2db420837 | ||
|
|
b4f53d2dd6 | ||
| ec4d86f727 | |||
| 4c6caa7114 | |||
|
|
252e7eb8d9 | ||
|
|
416c163d94 | ||
| d93fbe33dc | |||
|
|
59241932e0 | ||
|
|
127483eea1 | ||
| 4846691c46 | |||
|
|
2f56761bf1 | ||
|
|
0782b0920b | ||
| 40c4cf0e45 | |||
|
|
b13fbc63e3 | ||
| bacbfcc517 | |||
| b91e024569 | |||
| a6f8a0fdbe | |||
| 7b656c492a | |||
| 7d0443ce4b | |||
|
|
a964a99f3d | ||
|
|
31a0ddd547 | ||
|
|
fa5f8dcb3b | ||
|
|
64e66e9d7b | ||
|
|
23a4d87514 | ||
|
|
2ce8cc7a7e | ||
|
|
e963a770a7 | ||
|
|
ad2d9d98d6 | ||
|
|
bb49cb7626 | ||
|
|
9441be35ef | ||
|
|
dac5c0967a | ||
|
|
b5430403fd |
31
.github/ISSUE_TEMPLATE/bug_report.md
vendored
Normal file
31
.github/ISSUE_TEMPLATE/bug_report.md
vendored
Normal file
@@ -0,0 +1,31 @@
|
||||
---
|
||||
name: Bug report
|
||||
about: Create a report to help us improve
|
||||
title: "[bug-report]"
|
||||
labels: bug
|
||||
assignees: TLINDEN
|
||||
|
||||
---
|
||||
|
||||
**Describtion**
|
||||
<!-- Please provide a clear and concise description of the issue: -->
|
||||
|
||||
|
||||
**Steps To Reproduce**
|
||||
<!-- Please detail the steps to reproduce the behavior: -->
|
||||
|
||||
|
||||
**Expected behavior**
|
||||
<!-- What do you expected to happen instead? -->
|
||||
|
||||
|
||||
**Version information**
|
||||
<!--
|
||||
Please provide as much version information as possible:
|
||||
- if you have just installed a binary, provide the output of: rpn -v
|
||||
- if you installed from source, provide the output of: make show-version
|
||||
- provide additional details: operating system and version and shell environment
|
||||
-->
|
||||
|
||||
|
||||
**Additional informations**
|
||||
23
.github/ISSUE_TEMPLATE/feature_request.md
vendored
Normal file
23
.github/ISSUE_TEMPLATE/feature_request.md
vendored
Normal file
@@ -0,0 +1,23 @@
|
||||
---
|
||||
name: Feature request
|
||||
about: Suggest a feature
|
||||
title: "[feature-request]"
|
||||
labels: feature-request
|
||||
assignees: TLINDEN
|
||||
|
||||
---
|
||||
|
||||
**Describtion**
|
||||
<!-- Please provide a clear and concise description of the feature you desire: -->
|
||||
|
||||
|
||||
|
||||
**Version information**
|
||||
<!--
|
||||
Just in case the feature is already present, please provide as
|
||||
much version information as possible:
|
||||
- if you have just installed a binary, provide the output of: rpn -v
|
||||
- if you installed from source, provide the output of: make show-version
|
||||
- provide additional details: operating system and version and shell environment
|
||||
-->
|
||||
|
||||
36
.github/workflows/ci.yaml
vendored
Normal file
36
.github/workflows/ci.yaml
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
name: build-and-test-rpn
|
||||
on: [push, pull_request]
|
||||
jobs:
|
||||
build:
|
||||
strategy:
|
||||
matrix:
|
||||
version: [1.21]
|
||||
os: [ubuntu-latest, windows-latest, macos-latest]
|
||||
name: Build
|
||||
runs-on: ${{ matrix.os }}
|
||||
steps:
|
||||
- name: Set up Go
|
||||
uses: actions/setup-go@v3
|
||||
with:
|
||||
go-version: ${{ matrix.version }}
|
||||
id: go
|
||||
|
||||
- name: checkout
|
||||
uses: actions/checkout@v3
|
||||
|
||||
- name: build
|
||||
run: go build
|
||||
|
||||
- name: test
|
||||
run: make test
|
||||
|
||||
golangci:
|
||||
name: lint
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/setup-go@v3
|
||||
with:
|
||||
go-version: 1.21
|
||||
- uses: actions/checkout@v3
|
||||
- name: golangci-lint
|
||||
uses: golangci/golangci-lint-action@v3
|
||||
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,3 +1,4 @@
|
||||
releases
|
||||
rpn
|
||||
rpn.1
|
||||
coverage.out
|
||||
|
||||
38
Makefile
38
Makefile
@@ -39,10 +39,6 @@ ifdef HAVE_POD
|
||||
echo "var manpage = \`" >> $*.go
|
||||
pod2text $*.pod >> $*.go
|
||||
echo "\`" >> $*.go
|
||||
|
||||
echo "var usage = \`" >> $*.go
|
||||
awk '/SYNOPS/{f=1;next} /DESCR/{f=0} f' $*.pod | sed 's/^ //' >> $*.go
|
||||
echo "\`" >> $*.go
|
||||
endif
|
||||
|
||||
buildlocal:
|
||||
@@ -55,15 +51,25 @@ install: buildlocal
|
||||
install -o $(UID) -g $(GID) -m 444 $(tool).1 $(PREFIX)/man/man1/
|
||||
|
||||
clean:
|
||||
rm -rf $(tool) coverage.out
|
||||
rm -rf $(tool) coverage.out testdata
|
||||
|
||||
test:
|
||||
go test -v ./...
|
||||
bash t/test.sh
|
||||
test: clean
|
||||
go test ./... $(ARGS)
|
||||
|
||||
testfuzzy: clean
|
||||
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:
|
||||
@echo "Call like this: ''make singletest TEST=TestPrepareColumns MOD=lib"
|
||||
go test -run $(TEST) github.com/tlinden/rpn/$(MOD)
|
||||
@echo "Call like this: make singletest TEST=TestPrepareColumns ARGS=-v"
|
||||
go test -run $(TEST) $(ARGS)
|
||||
|
||||
cover-report:
|
||||
go test ./... -cover -coverprofile=coverage.out
|
||||
@@ -77,3 +83,15 @@ buildall:
|
||||
|
||||
release: buildall
|
||||
gh release create v$(VERSION) --generate-notes releases/*
|
||||
|
||||
show-versions: buildlocal
|
||||
@echo "### rpn version:"
|
||||
@./rpn -v
|
||||
|
||||
@echo
|
||||
@echo "### go module versions:"
|
||||
@go list -m all
|
||||
|
||||
@echo
|
||||
@echo "### go version used for building:"
|
||||
@grep -m 1 go go.mod
|
||||
|
||||
27
README.md
27
README.md
@@ -1,4 +1,8 @@
|
||||
## Reverse Polish Notation Calculator for the commandline
|
||||
## Programmable command-line calculator using reverse polish notation
|
||||
|
||||
[](https://github.com/tlinden/rpnc/actions)
|
||||
[](https://github.com/tlinden/rpnc/blob/master/LICENSE)
|
||||
[](https://goreportcard.com/report/github.com/tlinden/rpnc)
|
||||
|
||||
This is a small commandline calculator which takes its input in
|
||||
[reverse polish notation](https://en.wikipedia.org/wiki/Reverse_Polish_notation)
|
||||
@@ -11,10 +15,18 @@ Features:
|
||||
- various stack manipulation commands
|
||||
- basic math operators
|
||||
- advanced math functions (not yet complete)
|
||||
- provides interactive repl
|
||||
- can be used on the commandline
|
||||
- can calculate data in batch mode (also from STDIN)
|
||||
- extensible with custom LUA functions
|
||||
- provides interactive repl
|
||||
- completion
|
||||
- history
|
||||
- comments (comment character is `#`)
|
||||
- variables
|
||||
|
||||
## Demo
|
||||
|
||||

|
||||
|
||||
## Working principle
|
||||
|
||||
@@ -65,6 +77,8 @@ DEBUG(012): push to stack: 200.00
|
||||
= 200
|
||||
```
|
||||
|
||||
For a very good explanation how reverse polish notation and the stack works [watch this video by Prof. Brailsford](https://youtu.be/7ha78yWRDlE?si=9MCp59jAAG8fXavP)
|
||||
|
||||
## Usage
|
||||
|
||||
Basically you enter numbers followed by an operator or a
|
||||
@@ -211,10 +225,15 @@ the `register()` function to register your functions to the
|
||||
calculator. This function takes these parameters:
|
||||
|
||||
- function name
|
||||
- number of arguments expected (1,2 or -1 allowed), -1 means batch
|
||||
mode
|
||||
- number of arguments expected (see below)
|
||||
- help text
|
||||
|
||||
Number of expected arguments can be:
|
||||
|
||||
- 0: expect 1 argument but do NOT modify the stack
|
||||
- 1-n: do a singular calculation
|
||||
- -1: batch mode work with all numbers on the stack
|
||||
|
||||
Please [refer to the lua language
|
||||
reference](https://www.lua.org/manual/5.4/) for more details about
|
||||
LUA.
|
||||
|
||||
507
calc.go
507
calc.go
@@ -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
|
||||
it under the terms of the GNU General Public License as published by
|
||||
@@ -21,8 +21,8 @@ import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"math"
|
||||
"os"
|
||||
"regexp"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
@@ -33,35 +33,43 @@ type Calc struct {
|
||||
debug bool
|
||||
batch bool
|
||||
stdin bool
|
||||
showstack bool
|
||||
intermediate bool
|
||||
notdone bool // set to true as long as there are items left in the eval loop
|
||||
precision int
|
||||
|
||||
stack *Stack
|
||||
history []string
|
||||
completer readline.AutoCompleter
|
||||
interpreter *Interpreter
|
||||
Space *regexp.Regexp
|
||||
Comment *regexp.Regexp
|
||||
Register *regexp.Regexp
|
||||
Constants []string
|
||||
LuaFunctions []string
|
||||
|
||||
Funcalls Funcalls
|
||||
BatchFuncalls Funcalls
|
||||
|
||||
// different kinds of commands, displays nicer in help output
|
||||
StackCommands Commands
|
||||
SettingsCommands Commands
|
||||
ShowCommands Commands
|
||||
Commands Commands
|
||||
|
||||
Vars map[string]float64
|
||||
}
|
||||
|
||||
// help for lua functions will be added dynamically
|
||||
const Help string = `Available commands:
|
||||
batch toggle batch mode
|
||||
debug toggle debug output
|
||||
dump display the stack contents
|
||||
clear clear the whole stack
|
||||
shift remove the last element of the stack
|
||||
history display calculation history
|
||||
help|? show this message
|
||||
quit|exit|c-d|c-c exit program
|
||||
|
||||
const Help string = `
|
||||
Operators:
|
||||
basic operators: + - x * / ^ (* is an alias of x)
|
||||
|
||||
Bitwise operators: and or xor < (left shift) > (right shift)
|
||||
|
||||
Percent functions:
|
||||
% percent
|
||||
%- substract percent
|
||||
%- subtract percent
|
||||
%+ add percent
|
||||
|
||||
Math functions (see https://pkg.go.dev/math):
|
||||
@@ -75,13 +83,18 @@ sum sum of all values (alias: +)
|
||||
max max of all values
|
||||
min min of all values
|
||||
mean mean of all values (alias: avg)
|
||||
median median of all values`
|
||||
median median of all values
|
||||
|
||||
Register variables:
|
||||
>NAME Put last stack element into variable NAME
|
||||
<NAME Retrieve variable NAME and put onto stack`
|
||||
|
||||
// commands, constants and operators, defined here to feed completion
|
||||
// and our mode switch in Eval() dynamically
|
||||
const (
|
||||
Commands string = `dump reverse debug undebug clear batch shift undo help history manual exit quit`
|
||||
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
|
||||
@@ -93,7 +106,6 @@ func GetCompleteCustomFunctions() func(string) []string {
|
||||
completions = append(completions, luafunc)
|
||||
}
|
||||
|
||||
completions = append(completions, strings.Split(Commands, " ")...)
|
||||
completions = append(completions, strings.Split(Constants, " ")...)
|
||||
|
||||
return completions
|
||||
@@ -112,38 +124,66 @@ func (c *Calc) GetCompleteCustomFuncalls() func(string) []string {
|
||||
completions = append(completions, function)
|
||||
}
|
||||
|
||||
return completions
|
||||
for command := range c.SettingsCommands {
|
||||
if len(command) > 1 {
|
||||
completions = append(completions, command)
|
||||
}
|
||||
}
|
||||
|
||||
for command := range c.ShowCommands {
|
||||
if len(command) > 1 {
|
||||
completions = append(completions, command)
|
||||
}
|
||||
}
|
||||
|
||||
for command := range c.StackCommands {
|
||||
if len(command) > 1 {
|
||||
completions = append(completions, command)
|
||||
}
|
||||
}
|
||||
|
||||
for command := range c.Commands {
|
||||
if len(command) > 1 {
|
||||
completions = append(completions, command)
|
||||
}
|
||||
}
|
||||
|
||||
return completions
|
||||
}
|
||||
}
|
||||
|
||||
func NewCalc() *Calc {
|
||||
c := Calc{stack: NewStack(), debug: false}
|
||||
calc := Calc{stack: NewStack(), debug: false, precision: Precision}
|
||||
|
||||
c.Funcalls = DefineFunctions()
|
||||
c.BatchFuncalls = DefineBatchFunctions()
|
||||
calc.Funcalls = DefineFunctions()
|
||||
calc.BatchFuncalls = DefineBatchFunctions()
|
||||
calc.Vars = map[string]float64{}
|
||||
|
||||
c.completer = readline.NewPrefixCompleter(
|
||||
calc.completer = readline.NewPrefixCompleter(
|
||||
// custom lua functions
|
||||
readline.PcItemDynamic(GetCompleteCustomFunctions()),
|
||||
readline.PcItemDynamic(c.GetCompleteCustomFuncalls()),
|
||||
readline.PcItemDynamic(calc.GetCompleteCustomFuncalls()),
|
||||
)
|
||||
|
||||
c.Space = regexp.MustCompile(`\s+`)
|
||||
calc.Space = regexp.MustCompile(`\s+`)
|
||||
calc.Comment = regexp.MustCompile(`#.*`) // ignore everything after #
|
||||
calc.Register = regexp.MustCompile(`^([<>])([A-Z][A-Z0-9]*)`)
|
||||
|
||||
// pre-calculate mode switching arrays
|
||||
c.Constants = strings.Split(Constants, " ")
|
||||
calc.Constants = strings.Split(Constants, " ")
|
||||
|
||||
calc.SetCommands()
|
||||
|
||||
return &calc
|
||||
}
|
||||
|
||||
// setup the interpreter, called from main(), import lua functions
|
||||
func (c *Calc) SetInt(interpreter *Interpreter) {
|
||||
c.interpreter = interpreter
|
||||
|
||||
for name := range LuaFuncs {
|
||||
c.LuaFunctions = append(c.LuaFunctions, name)
|
||||
}
|
||||
|
||||
return &c
|
||||
}
|
||||
|
||||
// setup the interpreter, called from main()
|
||||
func (c *Calc) SetInt(I *Interpreter) {
|
||||
c.interpreter = I
|
||||
}
|
||||
|
||||
func (c *Calc) ToggleDebug() {
|
||||
@@ -161,133 +201,176 @@ func (c *Calc) ToggleStdin() {
|
||||
c.stdin = !c.stdin
|
||||
}
|
||||
|
||||
func (c *Calc) Prompt() string {
|
||||
p := "\033[31m»\033[0m "
|
||||
b := ""
|
||||
if c.batch {
|
||||
b = "->batch"
|
||||
}
|
||||
d := ""
|
||||
v := ""
|
||||
if c.debug {
|
||||
d = "->debug"
|
||||
v = fmt.Sprintf("/rev%d", c.stack.rev)
|
||||
func (c *Calc) ToggleShow() {
|
||||
c.showstack = !c.showstack
|
||||
}
|
||||
|
||||
return fmt.Sprintf("rpn%s%s [%d%s]%s", b, d, c.stack.Len(), v, p)
|
||||
func (c *Calc) Prompt() string {
|
||||
prompt := "\033[31m»\033[0m "
|
||||
batch := ""
|
||||
|
||||
if c.batch {
|
||||
batch = "->batch"
|
||||
}
|
||||
|
||||
debug := ""
|
||||
revision := ""
|
||||
|
||||
if c.debug {
|
||||
debug = "->debug"
|
||||
revision = fmt.Sprintf("/rev%d", c.stack.rev)
|
||||
}
|
||||
|
||||
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]
|
||||
func (c *Calc) Eval(line string) {
|
||||
line = strings.TrimSpace(line)
|
||||
func (c *Calc) Eval(line string) error {
|
||||
// remove surrounding whitespace and comments, if any
|
||||
line = strings.TrimSpace(c.Comment.ReplaceAllString(line, ""))
|
||||
|
||||
if line == "" {
|
||||
return
|
||||
return nil
|
||||
}
|
||||
|
||||
for _, item := range c.Space.Split(line, -1) {
|
||||
items := c.Space.Split(line, -1)
|
||||
|
||||
for pos, item := range items {
|
||||
if pos+1 < len(items) {
|
||||
c.notdone = true
|
||||
} else {
|
||||
c.notdone = false
|
||||
}
|
||||
|
||||
if err := c.EvalItem(item); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if c.showstack && !c.stdin {
|
||||
dots := ""
|
||||
|
||||
if c.stack.Len() > ShowStackLen {
|
||||
dots = "... "
|
||||
}
|
||||
|
||||
last := c.stack.Last(ShowStackLen)
|
||||
|
||||
fmt.Printf("stack: %s%s\n", dots, list2str(last))
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Calc) EvalItem(item string) error {
|
||||
num, err := strconv.ParseFloat(item, 64)
|
||||
|
||||
if err == nil {
|
||||
c.stack.Backup()
|
||||
c.stack.Push(num)
|
||||
} else {
|
||||
/*
|
||||
if contains(c.MathFunctions, item) {
|
||||
// go builtin math function, if implemented
|
||||
c.mathfunc(item)
|
||||
continue
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if contains(c.BatchFunctions, item) {
|
||||
// math functions only supported in batch mode like max or mean
|
||||
c.batchfunc(item)
|
||||
continue
|
||||
// try hex
|
||||
var i int
|
||||
|
||||
_, err = fmt.Sscanf(item, "0x%x", &i)
|
||||
if err == nil {
|
||||
c.stack.Backup()
|
||||
c.stack.Push(float64(i))
|
||||
|
||||
return nil
|
||||
}
|
||||
*/
|
||||
|
||||
if contains(c.Constants, item) {
|
||||
// put the constant onto the stack
|
||||
c.stack.Backup()
|
||||
c.stack.Push(const2num(item))
|
||||
continue
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if _, ok := c.Funcalls[item]; ok {
|
||||
if exists(c.Funcalls, item) {
|
||||
if err := c.DoFuncall(item); err != nil {
|
||||
fmt.Println(err)
|
||||
} else {
|
||||
c.Result()
|
||||
}
|
||||
continue
|
||||
return Error(err.Error())
|
||||
}
|
||||
|
||||
if c.batch {
|
||||
if _, ok := c.BatchFuncalls[item]; ok {
|
||||
if err := c.DoFuncall(item); err != nil {
|
||||
fmt.Println(err)
|
||||
} else {
|
||||
c.Result()
|
||||
|
||||
return nil
|
||||
}
|
||||
continue
|
||||
|
||||
if exists(c.BatchFuncalls, item) {
|
||||
if !c.batch {
|
||||
return Error("only supported in batch mode")
|
||||
}
|
||||
} else {
|
||||
if _, ok := c.BatchFuncalls[item]; ok {
|
||||
fmt.Println("only supported in batch mode")
|
||||
continue
|
||||
|
||||
if err := c.DoFuncall(item); err != nil {
|
||||
return Error(err.Error())
|
||||
}
|
||||
|
||||
c.Result()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if contains(c.LuaFunctions, item) {
|
||||
// user provided custom lua functions
|
||||
c.luafunc(item)
|
||||
continue
|
||||
c.EvalLuaFunction(item)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
regmatches := c.Register.FindStringSubmatch(item)
|
||||
if len(regmatches) == 3 {
|
||||
switch regmatches[1] {
|
||||
case ">":
|
||||
c.PutVar(regmatches[2])
|
||||
case "<":
|
||||
c.GetVar(regmatches[2])
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// internal commands
|
||||
// FIXME: propagate errors
|
||||
if exists(c.Commands, item) {
|
||||
c.Commands[item].Func(c)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if exists(c.ShowCommands, item) {
|
||||
c.ShowCommands[item].Func(c)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if exists(c.StackCommands, item) {
|
||||
c.StackCommands[item].Func(c)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
if exists(c.SettingsCommands, item) {
|
||||
c.SettingsCommands[item].Func(c)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// management commands
|
||||
switch item {
|
||||
case "?":
|
||||
fallthrough
|
||||
case "help":
|
||||
fmt.Println(Help)
|
||||
fmt.Println("Lua functions:")
|
||||
for name, function := range LuaFuncs {
|
||||
fmt.Printf("%-20s %s\n", name, function.help)
|
||||
}
|
||||
case "dump":
|
||||
c.stack.Dump()
|
||||
case "debug":
|
||||
c.ToggleDebug()
|
||||
case "undebug":
|
||||
c.debug = false
|
||||
case "batch":
|
||||
c.ToggleBatch()
|
||||
case "clear":
|
||||
c.stack.Backup()
|
||||
c.stack.Clear()
|
||||
case "shift":
|
||||
c.stack.Backup()
|
||||
c.stack.Shift()
|
||||
case "reverse":
|
||||
c.stack.Backup()
|
||||
c.stack.Reverse()
|
||||
case "undo":
|
||||
c.stack.Restore()
|
||||
case "history":
|
||||
for _, entry := range c.history {
|
||||
fmt.Println(entry)
|
||||
}
|
||||
case "exit":
|
||||
fallthrough
|
||||
case "quit":
|
||||
os.Exit(0)
|
||||
case "manual":
|
||||
man()
|
||||
c.PrintHelp()
|
||||
|
||||
default:
|
||||
fmt.Println("unknown command or operator!")
|
||||
}
|
||||
}
|
||||
return Error("unknown command or operator")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Execute a math function, check if it is defined just in case
|
||||
@@ -300,10 +383,11 @@ func (c *Calc) DoFuncall(funcname string) error {
|
||||
}
|
||||
|
||||
if function == nil {
|
||||
panic("function not defined but in completion list")
|
||||
return Error("function not defined but in completion list")
|
||||
}
|
||||
|
||||
var args Numbers
|
||||
|
||||
batch := false
|
||||
|
||||
if function.Expectargs == -1 {
|
||||
@@ -325,13 +409,17 @@ func (c *Calc) DoFuncall(funcname string) error {
|
||||
// the actual lambda call, so to say. We provide a slice of
|
||||
// the requested size, fetched from the stack (but not popped
|
||||
// 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
|
||||
return R.Err
|
||||
return funcresult.Err
|
||||
}
|
||||
|
||||
// don't forget to backup!
|
||||
c.stack.Backup()
|
||||
|
||||
// "pop"
|
||||
if batch {
|
||||
// get rid of stack
|
||||
c.stack.Clear()
|
||||
@@ -341,10 +429,11 @@ func (c *Calc) DoFuncall(funcname string) error {
|
||||
}
|
||||
|
||||
// save result
|
||||
c.stack.Push(R.Res)
|
||||
c.stack.Push(funcresult.Res)
|
||||
|
||||
// thanks a lot
|
||||
c.SetHistory(funcname, args, R.Res)
|
||||
c.SetHistory(funcname, args, funcresult.Res)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -361,11 +450,25 @@ func (c *Calc) History(format string, args ...any) {
|
||||
|
||||
// print the result
|
||||
func (c *Calc) Result() float64 {
|
||||
// we only print the result if it's either a final result or
|
||||
// (if it is intermediate) if -i has been given
|
||||
if c.intermediate || !c.notdone {
|
||||
// only needed in repl
|
||||
if !c.stdin {
|
||||
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]
|
||||
}
|
||||
@@ -376,87 +479,139 @@ func (c *Calc) Debug(msg string) {
|
||||
}
|
||||
}
|
||||
|
||||
// do simple calculations
|
||||
func (c *Calc) simple(op byte) {
|
||||
c.stack.Backup()
|
||||
|
||||
for c.stack.Len() > 1 {
|
||||
b := c.stack.Pop()
|
||||
a := c.stack.Pop()
|
||||
var x float64
|
||||
|
||||
c.Debug(fmt.Sprintf("evaluating: %.2f %c %.2f", a, op, b))
|
||||
|
||||
switch op {
|
||||
case '+':
|
||||
x = a + b
|
||||
case '-':
|
||||
x = a - b
|
||||
case 'x':
|
||||
fallthrough // alias for *
|
||||
case '*':
|
||||
x = a * b
|
||||
case '/':
|
||||
if b == 0 {
|
||||
fmt.Println("error: division by null!")
|
||||
return
|
||||
}
|
||||
x = a / b
|
||||
case '^':
|
||||
x = math.Pow(a, b)
|
||||
default:
|
||||
panic("invalid operator!")
|
||||
}
|
||||
|
||||
c.stack.Push(x)
|
||||
|
||||
c.History("%f %c %f = %f", a, op, b, x)
|
||||
|
||||
if !c.batch {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
c.Result()
|
||||
}
|
||||
|
||||
func (c *Calc) luafunc(funcname string) {
|
||||
func (c *Calc) EvalLuaFunction(funcname string) {
|
||||
// called from calc loop
|
||||
var x float64
|
||||
var luaresult float64
|
||||
|
||||
var err error
|
||||
|
||||
switch c.interpreter.FuncNumArgs(funcname) {
|
||||
case 0:
|
||||
fallthrough
|
||||
case 1:
|
||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last())
|
||||
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last())
|
||||
case 2:
|
||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last(2))
|
||||
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.Last(2))
|
||||
case -1:
|
||||
x, err = c.interpreter.CallLuaFunc(funcname, c.stack.All())
|
||||
luaresult, err = c.interpreter.CallLuaFunc(funcname, c.stack.All())
|
||||
default:
|
||||
x, err = 0, errors.New("invalid number of argument requested")
|
||||
luaresult, err = 0, errors.New("invalid number of argument requested")
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
c.stack.Backup()
|
||||
|
||||
dopush := true
|
||||
|
||||
switch c.interpreter.FuncNumArgs(funcname) {
|
||||
case 0:
|
||||
a := c.stack.Last()
|
||||
|
||||
if len(a) == 1 {
|
||||
c.History("%s(%f) = %f", funcname, a, luaresult)
|
||||
}
|
||||
|
||||
dopush = false
|
||||
case 1:
|
||||
a := c.stack.Pop()
|
||||
c.History("%s(%f) = %f", funcname, a, x)
|
||||
c.History("%s(%f) = %f", funcname, a, luaresult)
|
||||
case 2:
|
||||
a := 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:
|
||||
c.stack.Clear()
|
||||
c.History("%s(*) = %f", funcname, x)
|
||||
c.History("%s(*) = %f", funcname, luaresult)
|
||||
}
|
||||
|
||||
c.stack.Push(x)
|
||||
if dopush {
|
||||
c.stack.Push(luaresult)
|
||||
}
|
||||
|
||||
c.Result()
|
||||
}
|
||||
|
||||
func (c *Calc) PutVar(name string) {
|
||||
last := c.stack.Last()
|
||||
|
||||
if len(last) == 1 {
|
||||
c.Debug(fmt.Sprintf("register %.2f in %s", last[0], name))
|
||||
c.Vars[name] = last[0]
|
||||
} else {
|
||||
fmt.Println("empty stack")
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Calc) GetVar(name string) {
|
||||
if exists(c.Vars, name) {
|
||||
c.Debug(fmt.Sprintf("retrieve %.2f from %s", c.Vars[name], name))
|
||||
c.stack.Backup()
|
||||
c.stack.Push(c.Vars[name])
|
||||
} else {
|
||||
fmt.Println("variable doesn't exist")
|
||||
}
|
||||
}
|
||||
|
||||
func sortcommands(hash Commands) []string {
|
||||
keys := make([]string, 0, len(hash))
|
||||
|
||||
for key := range hash {
|
||||
if len(key) > 1 {
|
||||
keys = append(keys, key)
|
||||
}
|
||||
}
|
||||
|
||||
sort.Strings(keys)
|
||||
|
||||
return keys
|
||||
}
|
||||
|
||||
func (c *Calc) PrintHelp() {
|
||||
fmt.Println("Available configuration commands:")
|
||||
|
||||
for _, name := range sortcommands(c.SettingsCommands) {
|
||||
fmt.Printf("%-20s %s\n", name, c.SettingsCommands[name].Help)
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("Available show commands:")
|
||||
|
||||
for _, name := range sortcommands(c.ShowCommands) {
|
||||
fmt.Printf("%-20s %s\n", name, c.ShowCommands[name].Help)
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("Available stack manipulation commands:")
|
||||
|
||||
for _, name := range sortcommands(c.StackCommands) {
|
||||
fmt.Printf("%-20s %s\n", name, c.StackCommands[name].Help)
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println("Other commands:")
|
||||
|
||||
for _, name := range sortcommands(c.Commands) {
|
||||
fmt.Printf("%-20s %s\n", name, c.Commands[name].Help)
|
||||
}
|
||||
|
||||
fmt.Println()
|
||||
|
||||
fmt.Println(Help)
|
||||
|
||||
// append lua functions, if any
|
||||
if len(LuaFuncs) > 0 {
|
||||
fmt.Println("Lua functions:")
|
||||
|
||||
for name, function := range LuaFuncs {
|
||||
fmt.Printf("%-20s %s\n", name, function.help)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
416
calc_test.go
Normal file
416
calc_test.go
Normal file
@@ -0,0 +1,416 @@
|
||||
/*
|
||||
Copyright © 2023 Thomas von Dein
|
||||
|
||||
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
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
lua "github.com/yuin/gopher-lua"
|
||||
)
|
||||
|
||||
func TestCommentsAndWhitespace(t *testing.T) {
|
||||
calc := NewCalc()
|
||||
|
||||
var tests = []struct {
|
||||
name string
|
||||
cmd []string
|
||||
exp float64 // last element of the stack
|
||||
}{
|
||||
{
|
||||
name: "whitespace prefix",
|
||||
cmd: []string{" 5"},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "whitespace postfix",
|
||||
cmd: []string{"5 "},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "whitespace both",
|
||||
cmd: []string{" 5 "},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "comment line w/ spaces",
|
||||
cmd: []string{"5", " # 19"},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "comment line w/o spaces",
|
||||
cmd: []string{"5", `#19`},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "inline comment w/ spaces",
|
||||
cmd: []string{"5 # 19"},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
name: "inline comment w/o spaces",
|
||||
cmd: []string{"5#19"},
|
||||
exp: 5.0,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
testname := fmt.Sprintf("%s .(expect %.2f)",
|
||||
test.name, test.exp)
|
||||
|
||||
t.Run(testname, func(t *testing.T) {
|
||||
for _, line := range test.cmd {
|
||||
if err := calc.Eval(line); err != nil {
|
||||
t.Errorf(err.Error())
|
||||
}
|
||||
}
|
||||
got := calc.stack.Last()
|
||||
|
||||
if len(got) > 0 {
|
||||
if got[0] != test.exp {
|
||||
t.Errorf("parsing failed:\n+++ got: %f\n--- want: %f",
|
||||
got, test.exp)
|
||||
}
|
||||
}
|
||||
|
||||
if calc.stack.Len() != 1 {
|
||||
t.Errorf("invalid stack size:\n+++ got: %d\n--- want: 1",
|
||||
calc.stack.Len())
|
||||
}
|
||||
})
|
||||
|
||||
calc.stack.Clear()
|
||||
}
|
||||
}
|
||||
|
||||
func TestCalc(t *testing.T) {
|
||||
calc := NewCalc()
|
||||
|
||||
var tests = []struct {
|
||||
name string
|
||||
cmd string
|
||||
exp float64
|
||||
batch bool
|
||||
}{
|
||||
// ops
|
||||
{
|
||||
name: "plus",
|
||||
cmd: `15 15 +`,
|
||||
exp: 30,
|
||||
},
|
||||
{
|
||||
name: "power",
|
||||
cmd: `4 2 ^`,
|
||||
exp: 16,
|
||||
},
|
||||
{
|
||||
name: "minus",
|
||||
cmd: `100 50 -`,
|
||||
exp: 50,
|
||||
},
|
||||
{
|
||||
name: "multi",
|
||||
cmd: `4 4 x`,
|
||||
exp: 16,
|
||||
},
|
||||
{
|
||||
name: "divide",
|
||||
cmd: `10 2 /`,
|
||||
exp: 5,
|
||||
},
|
||||
{
|
||||
name: "percent",
|
||||
cmd: `400 20 %`,
|
||||
exp: 80,
|
||||
},
|
||||
{
|
||||
name: "percent-minus",
|
||||
cmd: `400 20 %-`,
|
||||
exp: 320,
|
||||
},
|
||||
{
|
||||
name: "percent-plus",
|
||||
cmd: `400 20 %+`,
|
||||
exp: 480,
|
||||
},
|
||||
|
||||
// math tests
|
||||
{
|
||||
name: "mod",
|
||||
cmd: `9 2 mod`,
|
||||
exp: 1,
|
||||
},
|
||||
{
|
||||
name: "sqrt",
|
||||
cmd: `16 sqrt`,
|
||||
exp: 4,
|
||||
},
|
||||
{
|
||||
name: "ceil",
|
||||
cmd: `15.5 ceil`,
|
||||
exp: 16,
|
||||
},
|
||||
{
|
||||
name: "dim",
|
||||
cmd: `6 4 dim`,
|
||||
exp: 2,
|
||||
},
|
||||
|
||||
// constants tests
|
||||
{
|
||||
name: "pitimes2",
|
||||
cmd: `Pi 2 *`,
|
||||
exp: 6.283185307179586,
|
||||
},
|
||||
{
|
||||
name: "pi+sqrt2",
|
||||
cmd: `Pi Sqrt2 +`,
|
||||
exp: 4.555806215962888,
|
||||
},
|
||||
|
||||
// batch tests
|
||||
{
|
||||
name: "batch-sum",
|
||||
cmd: `2 2 2 2 sum`,
|
||||
exp: 8,
|
||||
batch: true,
|
||||
},
|
||||
{
|
||||
name: "batch-median",
|
||||
cmd: `1 2 3 4 5 median`,
|
||||
exp: 3,
|
||||
batch: true,
|
||||
},
|
||||
{
|
||||
name: "batch-mean",
|
||||
cmd: `2 2 8 2 2 mean`,
|
||||
exp: 3.2,
|
||||
batch: true,
|
||||
},
|
||||
{
|
||||
name: "batch-min",
|
||||
cmd: `1 2 3 4 5 min`,
|
||||
exp: 1,
|
||||
batch: true,
|
||||
},
|
||||
{
|
||||
name: "batch-max",
|
||||
cmd: `1 2 3 4 5 max`,
|
||||
exp: 5,
|
||||
batch: true,
|
||||
},
|
||||
|
||||
// stack tests
|
||||
{
|
||||
name: "use-vars",
|
||||
cmd: `10 >TEN clear 5 <TEN *`,
|
||||
exp: 50,
|
||||
},
|
||||
{
|
||||
name: "reverse",
|
||||
cmd: `100 500 reverse -`,
|
||||
exp: 400,
|
||||
},
|
||||
{
|
||||
name: "swap",
|
||||
cmd: `2 16 swap /`,
|
||||
exp: 8,
|
||||
},
|
||||
{
|
||||
name: "clear batch",
|
||||
cmd: "1 1 1 1 1 clear 1 1 sum",
|
||||
exp: 2,
|
||||
batch: true,
|
||||
},
|
||||
{
|
||||
name: "undo",
|
||||
cmd: `4 4 + undo *`,
|
||||
exp: 16,
|
||||
},
|
||||
|
||||
// bit tests
|
||||
{
|
||||
name: "bit and",
|
||||
cmd: `1 3 and`,
|
||||
exp: 1,
|
||||
},
|
||||
{
|
||||
name: "bit or",
|
||||
cmd: `1 3 or`,
|
||||
exp: 3,
|
||||
},
|
||||
{
|
||||
name: "bit xor",
|
||||
cmd: `1 3 xor`,
|
||||
exp: 2,
|
||||
},
|
||||
|
||||
// converters
|
||||
{
|
||||
name: "inch-to-cm",
|
||||
cmd: `111 inch-to-cm`,
|
||||
exp: 281.94,
|
||||
},
|
||||
{
|
||||
name: "gallons-to-liters",
|
||||
cmd: `111 gallons-to-liters`,
|
||||
exp: 420.135,
|
||||
},
|
||||
{
|
||||
name: "meters-to-yards",
|
||||
cmd: `111 meters-to-yards`,
|
||||
exp: 1.2139107611548556,
|
||||
},
|
||||
{
|
||||
name: "miles-to-kilometers",
|
||||
cmd: `111 miles-to-kilometers`,
|
||||
exp: 178.599,
|
||||
},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
testname := fmt.Sprintf("cmd-%s-expect-%.2f",
|
||||
test.name, test.exp)
|
||||
|
||||
t.Run(testname, func(t *testing.T) {
|
||||
calc.batch = test.batch
|
||||
if err := calc.Eval(test.cmd); err != nil {
|
||||
t.Errorf(err.Error())
|
||||
}
|
||||
got := calc.Result()
|
||||
calc.stack.Clear()
|
||||
if got != test.exp {
|
||||
t.Errorf("calc failed:\n+++ got: %f\n--- want: %f",
|
||||
got, test.exp)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestCalcLua(t *testing.T) {
|
||||
var tests = []struct {
|
||||
function string
|
||||
stack []float64
|
||||
exp float64
|
||||
}{
|
||||
{
|
||||
function: "lower",
|
||||
stack: []float64{5, 6},
|
||||
exp: 5.0,
|
||||
},
|
||||
{
|
||||
function: "parallelresistance",
|
||||
stack: []float64{100, 200, 300},
|
||||
exp: 54.54545454545455,
|
||||
},
|
||||
}
|
||||
|
||||
calc := NewCalc()
|
||||
|
||||
LuaInterpreter = lua.NewState(lua.Options{SkipOpenLibs: true})
|
||||
defer LuaInterpreter.Close()
|
||||
|
||||
luarunner := NewInterpreter("example.lua", false)
|
||||
luarunner.InitLua()
|
||||
calc.SetInt(luarunner)
|
||||
|
||||
for _, test := range tests {
|
||||
testname := fmt.Sprintf("lua-%s", test.function)
|
||||
|
||||
t.Run(testname, func(t *testing.T) {
|
||||
calc.stack.Clear()
|
||||
for _, item := range test.stack {
|
||||
calc.stack.Push(item)
|
||||
}
|
||||
|
||||
calc.EvalLuaFunction(test.function)
|
||||
|
||||
got := calc.stack.Last()
|
||||
|
||||
if calc.stack.Len() != 1 {
|
||||
t.Errorf("invalid stack size:\n+++ got: %d\n--- want: 1",
|
||||
calc.stack.Len())
|
||||
}
|
||||
|
||||
if got[0] != test.exp {
|
||||
t.Errorf("lua function %s failed:\n+++ got: %f\n--- want: %f",
|
||||
test.function, got, test.exp)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func FuzzEval(f *testing.F) {
|
||||
legal := []string{
|
||||
"dump",
|
||||
"showstack",
|
||||
"help",
|
||||
"Pi 31 *",
|
||||
"SqrtE Pi /",
|
||||
"55.5 yards-to-meters",
|
||||
"2 4 +",
|
||||
"7 8 batch sum",
|
||||
"7 8 %-",
|
||||
"7 8 clear",
|
||||
"7 8 /",
|
||||
"b",
|
||||
"#444",
|
||||
"<X",
|
||||
}
|
||||
|
||||
for _, item := range legal {
|
||||
f.Add(item)
|
||||
}
|
||||
|
||||
calc := NewCalc()
|
||||
|
||||
var hexnum int
|
||||
|
||||
f.Fuzz(func(t *testing.T, line string) {
|
||||
t.Logf("Stack:\n%v\n", calc.stack.All())
|
||||
if err := calc.EvalItem(line); err == nil {
|
||||
t.Logf("given: <%s>", line)
|
||||
// not corpus and empty?
|
||||
if !contains(legal, line) && len(line) > 0 {
|
||||
item := strings.TrimSpace(calc.Comment.ReplaceAllString(line, ""))
|
||||
_, hexerr := fmt.Sscanf(item, "0x%x", &hexnum)
|
||||
// no comment?
|
||||
if len(item) > 0 {
|
||||
// no known command or function?
|
||||
if _, err := strconv.ParseFloat(item, 64); err != nil {
|
||||
if !contains(calc.Constants, item) &&
|
||||
!exists(calc.Funcalls, item) &&
|
||||
!exists(calc.BatchFuncalls, item) &&
|
||||
!contains(calc.LuaFunctions, item) &&
|
||||
!exists(calc.Commands, item) &&
|
||||
!exists(calc.ShowCommands, item) &&
|
||||
!exists(calc.SettingsCommands, item) &&
|
||||
!exists(calc.StackCommands, item) &&
|
||||
!calc.Register.MatchString(item) &&
|
||||
item != "?" && item != "help" &&
|
||||
hexerr != nil {
|
||||
t.Errorf("Fuzzy input accepted: <%s>", line)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
346
command.go
Normal file
346
command.go
Normal file
@@ -0,0 +1,346 @@
|
||||
/*
|
||||
Copyright © 2023 Thomas von Dein
|
||||
|
||||
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
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/exec"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type CommandFunction func(*Calc)
|
||||
|
||||
type Command struct {
|
||||
Help string
|
||||
Func CommandFunction
|
||||
}
|
||||
|
||||
type Commands map[string]*Command
|
||||
|
||||
func NewCommand(help string, function CommandFunction) *Command {
|
||||
return &Command{
|
||||
Help: help,
|
||||
Func: function,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Calc) SetSettingsCommands() Commands {
|
||||
return Commands{
|
||||
// Toggles
|
||||
"debug": NewCommand(
|
||||
"toggle debugging",
|
||||
func(c *Calc) {
|
||||
c.ToggleDebug()
|
||||
},
|
||||
),
|
||||
|
||||
"nodebug": NewCommand(
|
||||
"disable debugging",
|
||||
func(c *Calc) {
|
||||
c.debug = false
|
||||
c.stack.debug = false
|
||||
},
|
||||
),
|
||||
|
||||
"batch": NewCommand(
|
||||
"toggle batch mode",
|
||||
func(c *Calc) {
|
||||
c.ToggleBatch()
|
||||
},
|
||||
),
|
||||
|
||||
"nobatch": NewCommand(
|
||||
"disable batch mode",
|
||||
func(c *Calc) {
|
||||
c.batch = false
|
||||
},
|
||||
),
|
||||
|
||||
"showstack": NewCommand(
|
||||
"toggle show last 5 items of the stack",
|
||||
func(c *Calc) {
|
||||
c.ToggleShow()
|
||||
},
|
||||
),
|
||||
|
||||
"noshowstack": NewCommand(
|
||||
"disable display of the stack",
|
||||
func(c *Calc) {
|
||||
c.showstack = false
|
||||
},
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Calc) SetShowCommands() Commands {
|
||||
return Commands{
|
||||
// Display commands
|
||||
"dump": NewCommand(
|
||||
"display the stack contents",
|
||||
func(c *Calc) {
|
||||
c.stack.Dump()
|
||||
},
|
||||
),
|
||||
|
||||
"history": NewCommand(
|
||||
"display calculation history",
|
||||
func(c *Calc) {
|
||||
for _, entry := range c.history {
|
||||
fmt.Println(entry)
|
||||
}
|
||||
},
|
||||
),
|
||||
|
||||
"vars": NewCommand(
|
||||
"show list of variables",
|
||||
func(c *Calc) {
|
||||
if len(c.Vars) > 0 {
|
||||
fmt.Printf("%-20s %s\n", "VARIABLE", "VALUE")
|
||||
for k, v := range c.Vars {
|
||||
fmt.Printf("%-20s -> %.2f\n", k, v)
|
||||
}
|
||||
} else {
|
||||
fmt.Println("no vars registered")
|
||||
}
|
||||
},
|
||||
),
|
||||
|
||||
"hex": NewCommand(
|
||||
"show last stack item in hex form (converted to int)",
|
||||
func(c *Calc) {
|
||||
if c.stack.Len() > 0 {
|
||||
fmt.Printf("0x%x\n", int(c.stack.Last()[0]))
|
||||
}
|
||||
},
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Calc) SetStackCommands() Commands {
|
||||
return Commands{
|
||||
"clear": NewCommand(
|
||||
"clear the whole stack",
|
||||
func(c *Calc) {
|
||||
c.stack.Backup()
|
||||
c.stack.Clear()
|
||||
},
|
||||
),
|
||||
|
||||
"shift": NewCommand(
|
||||
"remove the last element of the stack",
|
||||
func(c *Calc) {
|
||||
c.stack.Backup()
|
||||
c.stack.Shift()
|
||||
},
|
||||
),
|
||||
|
||||
"reverse": NewCommand(
|
||||
"reverse the stack elements",
|
||||
func(c *Calc) {
|
||||
c.stack.Backup()
|
||||
c.stack.Reverse()
|
||||
},
|
||||
),
|
||||
|
||||
"swap": NewCommand(
|
||||
"exchange the last two elements",
|
||||
CommandSwap,
|
||||
),
|
||||
|
||||
"undo": NewCommand(
|
||||
"undo last operation",
|
||||
func(c *Calc) {
|
||||
c.stack.Restore()
|
||||
},
|
||||
),
|
||||
|
||||
"dup": NewCommand(
|
||||
"duplicate last stack item",
|
||||
CommandDup,
|
||||
),
|
||||
|
||||
"edit": NewCommand(
|
||||
"edit the stack interactively",
|
||||
CommandEdit,
|
||||
),
|
||||
}
|
||||
}
|
||||
|
||||
// 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
|
||||
c.Commands = Commands{
|
||||
"exit": NewCommand(
|
||||
"exit program",
|
||||
func(c *Calc) {
|
||||
os.Exit(0)
|
||||
},
|
||||
),
|
||||
|
||||
"manual": NewCommand(
|
||||
"show manual",
|
||||
func(c *Calc) {
|
||||
man()
|
||||
},
|
||||
),
|
||||
}
|
||||
|
||||
// aliases
|
||||
c.Commands["quit"] = c.Commands["exit"]
|
||||
|
||||
c.SettingsCommands["d"] = c.SettingsCommands["debug"]
|
||||
c.SettingsCommands["b"] = c.SettingsCommands["batch"]
|
||||
c.SettingsCommands["s"] = c.SettingsCommands["showstack"]
|
||||
|
||||
c.ShowCommands["h"] = c.ShowCommands["history"]
|
||||
c.ShowCommands["p"] = c.ShowCommands["dump"]
|
||||
c.ShowCommands["v"] = c.ShowCommands["vars"]
|
||||
|
||||
c.StackCommands["c"] = c.StackCommands["clear"]
|
||||
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)
|
||||
}
|
||||
}
|
||||
7
demo/Makefile
Normal file
7
demo/Makefile
Normal file
@@ -0,0 +1,7 @@
|
||||
#
|
||||
|
||||
|
||||
all:
|
||||
asciinema rec --cols 80 --row 25 -c "env - PS1='> ' PATH=..:$PATH /bin/bash --norc --noprofile" --overwrite demo.cast
|
||||
agg demo.cast demo.gif
|
||||
|
||||
3
demo/demo.cast
Normal file
3
demo/demo.cast
Normal file
@@ -0,0 +1,3 @@
|
||||
{"version": 2, "width": 80, "height": 25, "timestamp": 1699358215, "env": {"SHELL": "/bin/bash", "TERM": "tmux-256color"}}
|
||||
[0.005628, "o", "> "]
|
||||
[0.941195, "o", "exit\r\n"]
|
||||
BIN
demo/demo.gif
Normal file
BIN
demo/demo.gif
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 613 KiB |
18
demo/sessions.txt
Normal file
18
demo/sessions.txt
Normal file
@@ -0,0 +1,18 @@
|
||||
# demo sessions, used to create an interactive rpn recording
|
||||
|
||||
# part 1: interactive
|
||||
|
||||
# put some numbers onto the stack
|
||||
2 4 6
|
||||
8
|
||||
10
|
||||
|
||||
# take a look at the stack
|
||||
dump
|
||||
|
||||
# add the last 2
|
||||
+
|
||||
|
||||
# multiply the result and the previous
|
||||
*
|
||||
|
||||
38
example.lua
Normal file
38
example.lua
Normal file
@@ -0,0 +1,38 @@
|
||||
-- simple function, return the lower number of the two operands
|
||||
function lower(a,b)
|
||||
if a < b then
|
||||
return a
|
||||
else
|
||||
return b
|
||||
end
|
||||
end
|
||||
|
||||
-- calculate parallel resistance. Batch function (registered with -1,
|
||||
-- see below). Takes a table as parameter.
|
||||
--
|
||||
-- Formula: 1/( (1/R1) + (1/R2) + ...)
|
||||
function parallelresistance(list)
|
||||
sumres = 0
|
||||
|
||||
for i, value in ipairs(list) do
|
||||
sumres = sumres + 1 / value
|
||||
end
|
||||
|
||||
return 1 / sumres
|
||||
end
|
||||
|
||||
-- converter example
|
||||
function inch2centimeter(inches)
|
||||
return inches * 2.54
|
||||
end
|
||||
|
||||
function init()
|
||||
-- expects 2 args
|
||||
register("lower", 2, "lower")
|
||||
|
||||
-- expects a list of all numbers on the stack, batch mode
|
||||
register("parallelresistance", -1, "parallel resistance")
|
||||
|
||||
-- expects 1 arg, but doesn't pop()
|
||||
register("inch2centimeter", 0)
|
||||
end
|
||||
348
funcs.go
348
funcs.go
@@ -22,14 +22,14 @@ import (
|
||||
"math"
|
||||
)
|
||||
|
||||
type R struct {
|
||||
type Result struct {
|
||||
Res float64
|
||||
Err error
|
||||
}
|
||||
|
||||
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
|
||||
// needs to specify how many numbers it expects and the actual go
|
||||
@@ -64,359 +64,450 @@ func NewFuncall(function Function, expectargs ...int) *Funcall {
|
||||
}
|
||||
|
||||
// Convenience function, create new result
|
||||
func NewR(n float64, e error) R {
|
||||
return R{Res: n, Err: e}
|
||||
func NewResult(n float64, e error) Result {
|
||||
return Result{Res: n, Err: e}
|
||||
}
|
||||
|
||||
// the actual functions, called once during initialization.
|
||||
func DefineFunctions() Funcalls {
|
||||
f := map[string]*Funcall{
|
||||
funcmap := map[string]*Funcall{
|
||||
// simple operators, they all expect 2 args
|
||||
"+": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(arg[0]+arg[1], nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]+arg[1], nil)
|
||||
},
|
||||
),
|
||||
|
||||
"-": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(arg[0]-arg[1], nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]-arg[1], nil)
|
||||
},
|
||||
),
|
||||
|
||||
"x": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(arg[0]*arg[1], nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]*arg[1], nil)
|
||||
},
|
||||
),
|
||||
|
||||
"/": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
func(arg Numbers) Result {
|
||||
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(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Pow(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Pow(arg[0], arg[1]), nil)
|
||||
},
|
||||
),
|
||||
|
||||
"%": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR((arg[0]/100)*arg[1], nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult((arg[0]/100)*arg[1], nil)
|
||||
},
|
||||
),
|
||||
|
||||
"%-": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(arg[0]-((arg[0]/100)*arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]-((arg[0]/100)*arg[1]), nil)
|
||||
},
|
||||
),
|
||||
|
||||
"%+": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(arg[0]+((arg[0]/100)*arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]+((arg[0]/100)*arg[1]), nil)
|
||||
},
|
||||
),
|
||||
|
||||
"mod": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Remainder(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Remainder(arg[0], arg[1]), nil)
|
||||
},
|
||||
),
|
||||
|
||||
"sqrt": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Sqrt(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Sqrt(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"abs": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Abs(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Abs(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"acos": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Acos(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Acos(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"acosh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Acosh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Acosh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"asin": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Asin(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Asin(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"asinh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Asinh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Asinh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"atan": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Atan(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Atan(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"atan2": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Atan2(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Atan2(arg[0], arg[1]), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"atanh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Atanh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Atanh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"cbrt": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Cbrt(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Cbrt(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"ceil": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Ceil(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Ceil(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"cos": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Cos(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Cos(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"cosh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Cosh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Cosh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"erf": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Erf(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Erf(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"erfc": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Erfc(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Erfc(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"erfcinv": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Erfcinv(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Erfcinv(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"erfinv": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Erfinv(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Erfinv(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"exp": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Exp(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Exp(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"exp2": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Exp2(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Exp2(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"expm1": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Expm1(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Expm1(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"floor": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Floor(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Floor(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"gamma": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Gamma(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Gamma(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"ilogb": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(float64(math.Ilogb(arg[0])), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(float64(math.Ilogb(arg[0])), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"j0": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.J0(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.J0(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"j1": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.J1(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.J1(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"log": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Log(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Log(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"log10": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Log10(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Log10(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"log1p": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Log1p(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Log1p(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"log2": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Log2(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Log2(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"logb": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Logb(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Logb(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"pow": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Pow(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Pow(arg[0], arg[1]), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"round": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Round(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Round(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"roundtoeven": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.RoundToEven(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.RoundToEven(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"sin": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Sin(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Sin(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"sinh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Sinh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Sinh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"tan": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Tan(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Tan(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"tanh": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Tanh(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Tanh(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"trunc": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Trunc(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Trunc(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"y0": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Y0(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Y0(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"y1": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Y1(arg[0]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Y1(arg[0]), nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"copysign": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Copysign(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Copysign(arg[0], arg[1]), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"dim": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Dim(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Dim(arg[0], arg[1]), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"hypot": NewFuncall(
|
||||
func(arg Numbers) R {
|
||||
return NewR(math.Hypot(arg[0], arg[1]), nil)
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(math.Hypot(arg[0], arg[1]), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
// converters of all kinds
|
||||
"cm-to-inch": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]/2.54, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"inch-to-cm": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]*2.54, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"gallons-to-liters": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]*3.785, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"liters-to-gallons": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]/3.785, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"yards-to-meters": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]*91.44, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"meters-to-yards": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]/91.44, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"miles-to-kilometers": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]*1.609, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"kilometers-to-miles": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(arg[0]/1.609, nil)
|
||||
},
|
||||
1),
|
||||
|
||||
"or": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(float64(int(arg[0])|int(arg[1])), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"and": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(float64(int(arg[0])&int(arg[1])), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"xor": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
return NewResult(float64(int(arg[0])^int(arg[1])), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
"<": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
// Shift by negative number provibited, so check it.
|
||||
// Note that we check against uint64 overflow as well here
|
||||
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
||||
return NewResult(0, errors.New("negative shift amount"))
|
||||
}
|
||||
|
||||
return NewResult(float64(int(arg[0])<<int(arg[1])), nil)
|
||||
},
|
||||
2),
|
||||
|
||||
">": NewFuncall(
|
||||
func(arg Numbers) Result {
|
||||
if arg[1] < 0 || uint64(arg[1]) > math.MaxInt64 {
|
||||
return NewResult(0, errors.New("negative shift amount"))
|
||||
}
|
||||
|
||||
return NewResult(float64(int(arg[0])>>int(arg[1])), nil)
|
||||
},
|
||||
2),
|
||||
}
|
||||
|
||||
// aliases
|
||||
f["*"] = f["x"]
|
||||
f["remainder"] = f["mod"]
|
||||
funcmap["*"] = funcmap["x"]
|
||||
funcmap["remainder"] = funcmap["mod"]
|
||||
|
||||
return f
|
||||
return funcmap
|
||||
}
|
||||
|
||||
func DefineBatchFunctions() Funcalls {
|
||||
f := map[string]*Funcall{
|
||||
funcmap := map[string]*Funcall{
|
||||
"median": NewFuncall(
|
||||
func(args Numbers) R {
|
||||
func(args Numbers) Result {
|
||||
middle := len(args) / 2
|
||||
return NewR(args[middle], nil)
|
||||
|
||||
return NewResult(args[middle], nil)
|
||||
},
|
||||
-1),
|
||||
|
||||
"mean": NewFuncall(
|
||||
func(args Numbers) R {
|
||||
func(args Numbers) Result {
|
||||
var sum float64
|
||||
for _, item := range args {
|
||||
sum += item
|
||||
}
|
||||
return NewR(sum/float64(len(args)), nil)
|
||||
|
||||
return NewResult(sum/float64(len(args)), nil)
|
||||
},
|
||||
-1),
|
||||
|
||||
"min": NewFuncall(
|
||||
func(args Numbers) R {
|
||||
func(args Numbers) Result {
|
||||
var min float64
|
||||
min, args = args[0], args[1:]
|
||||
for _, item := range args {
|
||||
@@ -424,12 +515,13 @@ func DefineBatchFunctions() Funcalls {
|
||||
min = item
|
||||
}
|
||||
}
|
||||
return NewR(min, nil)
|
||||
|
||||
return NewResult(min, nil)
|
||||
},
|
||||
-1),
|
||||
|
||||
"max": NewFuncall(
|
||||
func(args Numbers) R {
|
||||
func(args Numbers) Result {
|
||||
var max float64
|
||||
max, args = args[0], args[1:]
|
||||
for _, item := range args {
|
||||
@@ -437,24 +529,26 @@ func DefineBatchFunctions() Funcalls {
|
||||
max = item
|
||||
}
|
||||
}
|
||||
return NewR(max, nil)
|
||||
|
||||
return NewResult(max, nil)
|
||||
},
|
||||
-1),
|
||||
|
||||
"sum": NewFuncall(
|
||||
func(args Numbers) R {
|
||||
func(args Numbers) Result {
|
||||
var sum float64
|
||||
for _, item := range args {
|
||||
sum += item
|
||||
}
|
||||
return NewR(sum, nil)
|
||||
|
||||
return NewResult(sum, nil)
|
||||
},
|
||||
-1),
|
||||
}
|
||||
|
||||
// aliases
|
||||
f["+"] = f["sum"]
|
||||
f["avg"] = f["mean"]
|
||||
funcmap["+"] = funcmap["sum"]
|
||||
funcmap["avg"] = funcmap["mean"]
|
||||
|
||||
return f
|
||||
return funcmap
|
||||
}
|
||||
|
||||
4
go.mod
4
go.mod
@@ -4,7 +4,9 @@ go 1.20
|
||||
|
||||
require (
|
||||
github.com/chzyer/readline v1.5.1 // indirect
|
||||
github.com/rogpeppe/go-internal v1.11.0 // indirect
|
||||
github.com/spf13/pflag v1.0.5 // indirect
|
||||
github.com/yuin/gopher-lua v1.1.0 // indirect
|
||||
golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5 // indirect
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f // indirect
|
||||
golang.org/x/tools v0.1.12 // indirect
|
||||
)
|
||||
|
||||
6
go.sum
6
go.sum
@@ -2,9 +2,15 @@ github.com/chzyer/logex v1.2.1/go.mod h1:JLbx6lG2kDbNRFnfkgvh4eRJRPX1QCoOIWomwys
|
||||
github.com/chzyer/readline v1.5.1 h1:upd/6fQk4src78LMRzh5vItIt361/o4uq553V8B5sGI=
|
||||
github.com/chzyer/readline v1.5.1/go.mod h1:Eh+b79XXUwfKfcPLepksvw2tcLE/Ct21YObkaSkeBlk=
|
||||
github.com/chzyer/test v1.0.0/go.mod h1:2JlltgoNkt4TW/z9V/IzDdFaMTM2JPIi26O1pF38GC8=
|
||||
github.com/rogpeppe/go-internal v1.11.0 h1:cWPaGQEPrBb5/AsnsZesgZZ9yb1OQ+GOISoDNXVBh4M=
|
||||
github.com/rogpeppe/go-internal v1.11.0/go.mod h1:ddIwULY96R17DhadqLgMfk9H9tvdUzkipdSkR5nkCZA=
|
||||
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
||||
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||
github.com/yuin/gopher-lua v1.1.0 h1:BojcDhfyDWgU2f2TOzYK/g5p2gxMrku8oupLDqlnSqE=
|
||||
github.com/yuin/gopher-lua v1.1.0/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw=
|
||||
golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5 h1:y/woIyUBFbpQGKS0u1aHF/40WUDnek3fPOyD08H5Vng=
|
||||
golang.org/x/sys v0.0.0-20220310020820-b874c991c1a5/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f h1:v4INt8xihDGvnrfjMDVXGxw9wrfxYyCjk0KbXjhR55s=
|
||||
golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/tools v0.1.12 h1:VveCTK38A2rkS8ZqFY25HIDFscX5X9OoEhJd3quQmXU=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
|
||||
@@ -26,10 +26,11 @@ import (
|
||||
|
||||
type Interpreter struct {
|
||||
debug bool
|
||||
script string
|
||||
}
|
||||
|
||||
// LUA interpreter, instanciated in main()
|
||||
var L *lua.LState
|
||||
// LuaInterpreter is the lua interpreter, instantiated in main()
|
||||
var LuaInterpreter *lua.LState
|
||||
|
||||
// holds a user provided lua function
|
||||
type LuaFunction struct {
|
||||
@@ -38,12 +39,16 @@ type LuaFunction struct {
|
||||
numargs int
|
||||
}
|
||||
|
||||
// must be global since init() is being called from lua which doesn't
|
||||
// have access to the interpreter instance
|
||||
// LuaFuncs must be global since init() is being called from lua which
|
||||
// doesn't have access to the interpreter instance
|
||||
var LuaFuncs map[string]LuaFunction
|
||||
|
||||
func NewInterpreter(script string, debug bool) *Interpreter {
|
||||
return &Interpreter{debug: debug, script: script}
|
||||
}
|
||||
|
||||
// initialize the lua environment properly
|
||||
func InitLua(config string, debug bool) *Interpreter {
|
||||
func (i *Interpreter) InitLua() {
|
||||
// we only load a subset of lua Open modules and don't allow
|
||||
// net, system or io stuff
|
||||
for _, pair := range []struct {
|
||||
@@ -56,8 +61,8 @@ func InitLua(config string, debug bool) *Interpreter {
|
||||
{lua.DebugLibName, lua.OpenDebug},
|
||||
{lua.MathLibName, lua.OpenMath},
|
||||
} {
|
||||
if err := L.CallByParam(lua.P{
|
||||
Fn: L.NewFunction(pair.f),
|
||||
if err := LuaInterpreter.CallByParam(lua.P{
|
||||
Fn: LuaInterpreter.NewFunction(pair.f),
|
||||
NRet: 0,
|
||||
Protect: true,
|
||||
}, lua.LString(pair.n)); err != nil {
|
||||
@@ -66,26 +71,24 @@ func InitLua(config string, debug bool) *Interpreter {
|
||||
}
|
||||
|
||||
// load the lua config (which we expect to contain init() and math functions)
|
||||
if err := L.DoFile(config); err != nil {
|
||||
if err := LuaInterpreter.DoFile(i.script); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// instanciate
|
||||
// instantiate
|
||||
LuaFuncs = map[string]LuaFunction{}
|
||||
|
||||
// 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()
|
||||
if err := L.CallByParam(lua.P{
|
||||
Fn: L.GetGlobal("init"),
|
||||
if err := LuaInterpreter.CallByParam(lua.P{
|
||||
Fn: LuaInterpreter.GetGlobal("init"),
|
||||
NRet: 0,
|
||||
Protect: true,
|
||||
}); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
return &Interpreter{debug: debug}
|
||||
}
|
||||
|
||||
func (i *Interpreter) Debug(msg string) {
|
||||
@@ -105,54 +108,56 @@ func (i *Interpreter) FuncNumArgs(name string) int {
|
||||
// arguments. 1 uses the last item of the stack, 2 the last two and -1
|
||||
// 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
|
||||
// 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) {
|
||||
i.Debug(fmt.Sprintf("calling lua func %s() with %d args",
|
||||
funcname, LuaFuncs[funcname].numargs))
|
||||
|
||||
switch LuaFuncs[funcname].numargs {
|
||||
case 0:
|
||||
fallthrough
|
||||
case 1:
|
||||
// 1 arg variant
|
||||
if err := L.CallByParam(lua.P{
|
||||
Fn: L.GetGlobal(funcname),
|
||||
if err := LuaInterpreter.CallByParam(lua.P{
|
||||
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||
NRet: 1,
|
||||
Protect: true,
|
||||
}, lua.LNumber(items[0])); err != nil {
|
||||
fmt.Println(err)
|
||||
return 0, err
|
||||
return 0, fmt.Errorf("failed to exec lua func %s: %w", funcname, err)
|
||||
}
|
||||
case 2:
|
||||
// 2 arg variant
|
||||
if err := L.CallByParam(lua.P{
|
||||
Fn: L.GetGlobal(funcname),
|
||||
if err := LuaInterpreter.CallByParam(lua.P{
|
||||
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||
NRet: 1,
|
||||
Protect: true,
|
||||
}, 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:
|
||||
// batch variant, use lua table as array
|
||||
tb := L.NewTable()
|
||||
table := LuaInterpreter.NewTable()
|
||||
|
||||
// put the whole stack into it
|
||||
for _, item := range items {
|
||||
tb.Append(lua.LNumber(item))
|
||||
table.Append(lua.LNumber(item))
|
||||
}
|
||||
|
||||
if err := L.CallByParam(lua.P{
|
||||
Fn: L.GetGlobal(funcname),
|
||||
if err := LuaInterpreter.CallByParam(lua.P{
|
||||
Fn: LuaInterpreter.GetGlobal(funcname),
|
||||
NRet: 1,
|
||||
Protect: true,
|
||||
}, tb); err != nil {
|
||||
return 0, err
|
||||
}, table); err != nil {
|
||||
return 0, fmt.Errorf("failed to exec lua func %s: %w", funcname, err)
|
||||
}
|
||||
}
|
||||
|
||||
// get result and cast to float64
|
||||
if res, ok := L.Get(-1).(lua.LNumber); ok {
|
||||
L.Pop(1)
|
||||
if res, ok := LuaInterpreter.Get(-1).(lua.LNumber); ok {
|
||||
LuaInterpreter.Pop(1)
|
||||
|
||||
return float64(res), nil
|
||||
}
|
||||
|
||||
@@ -162,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
|
||||
// -1, it denotes the number of items from the stack requested by the
|
||||
// lua function. -1 means batch mode, that is all items
|
||||
func register(L *lua.LState) int {
|
||||
function := L.ToString(1)
|
||||
numargs := L.ToInt(2)
|
||||
help := L.ToString(3)
|
||||
func register(lstate *lua.LState) int {
|
||||
function := lstate.ToString(1)
|
||||
numargs := lstate.ToInt(2)
|
||||
help := lstate.ToString(3)
|
||||
|
||||
LuaFuncs[function] = LuaFunction{
|
||||
name: function,
|
||||
|
||||
85
main.go
85
main.go
@@ -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
|
||||
it under the terms of the GNU General Public License as published by
|
||||
@@ -30,7 +30,7 @@ import (
|
||||
lua "github.com/yuin/gopher-lua"
|
||||
)
|
||||
|
||||
const VERSION string = "2.0.1"
|
||||
const VERSION string = "2.1.0"
|
||||
|
||||
const Usage string = `This is rpn, a reverse polish notation calculator cli.
|
||||
|
||||
@@ -39,16 +39,24 @@ Usage: rpn [-bdvh] [<operator>]
|
||||
Options:
|
||||
-b, --batchmode enable batch mode
|
||||
-d, --debug enable debug mode
|
||||
-s, --stack show last 5 items of the stack (off by default)
|
||||
-i --intermediate print intermediate results
|
||||
-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
|
||||
-h, --help show help
|
||||
|
||||
When <operator> is given, batch mode ist automatically enabled. Use
|
||||
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() {
|
||||
os.Exit(Main())
|
||||
}
|
||||
|
||||
func Main() int {
|
||||
calc := NewCalc()
|
||||
|
||||
showversion := false
|
||||
@@ -58,23 +66,29 @@ func main() {
|
||||
configfile := ""
|
||||
|
||||
flag.BoolVarP(&calc.batch, "batchmode", "b", false, "batch mode")
|
||||
flag.BoolVarP(&calc.showstack, "show-stack", "s", false, "show stack")
|
||||
flag.BoolVarP(&calc.intermediate, "showin-termediate", "i", false,
|
||||
"show intermediate results")
|
||||
flag.BoolVarP(&enabledebug, "debug", "d", false, "debug mode")
|
||||
flag.BoolVarP(&showversion, "version", "v", false, "show version")
|
||||
flag.BoolVarP(&showhelp, "help", "h", false, "show usage")
|
||||
flag.BoolVarP(&showmanual, "manual", "m", false, "show manual")
|
||||
flag.StringVarP(&configfile, "config", "c",
|
||||
os.Getenv("HOME")+"/.rpn.lua", "config file (lua format)")
|
||||
flag.IntVarP(&calc.precision, "precision", "p", Precision, "floating point precision")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
if showversion {
|
||||
fmt.Printf("This is rpn version %s\n", VERSION)
|
||||
return
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
if showhelp {
|
||||
fmt.Println(Usage)
|
||||
return
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
if enabledebug {
|
||||
@@ -83,30 +97,43 @@ func main() {
|
||||
|
||||
if showmanual {
|
||||
man()
|
||||
os.Exit(0)
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
// the lua state object is global, instanciate it early
|
||||
L = lua.NewState(lua.Options{SkipOpenLibs: true})
|
||||
defer L.Close()
|
||||
// the lua state object is global, instantiate it early
|
||||
LuaInterpreter = lua.NewState(lua.Options{SkipOpenLibs: true})
|
||||
defer LuaInterpreter.Close()
|
||||
|
||||
// our config file is interpreted as lua code, only functions can
|
||||
// be defined, init() will be called by InitLua().
|
||||
if _, err := os.Stat(configfile); err == nil {
|
||||
I := InitLua(configfile, enabledebug)
|
||||
calc.SetInt(I)
|
||||
luarunner := NewInterpreter(configfile, enabledebug)
|
||||
luarunner.InitLua()
|
||||
calc.SetInt(luarunner)
|
||||
|
||||
if calc.debug {
|
||||
fmt.Println("loaded config")
|
||||
}
|
||||
} else if calc.debug {
|
||||
fmt.Println(err)
|
||||
}
|
||||
|
||||
if len(flag.Args()) > 1 {
|
||||
// commandline calc operation, no readline etc needed
|
||||
// called like rpn 2 2 +
|
||||
calc.stdin = true
|
||||
calc.Eval(strings.Join(flag.Args(), " "))
|
||||
return
|
||||
if err := calc.Eval(strings.Join(flag.Args(), " ")); err != nil {
|
||||
fmt.Println(err)
|
||||
|
||||
return 1
|
||||
}
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
// interactive mode, need readline
|
||||
rl, err := readline.NewEx(&readline.Config{
|
||||
reader, err := readline.NewEx(&readline.Config{
|
||||
Prompt: calc.Prompt(),
|
||||
HistoryFile: os.Getenv("HOME") + "/.rpn-history",
|
||||
HistoryLimit: 500,
|
||||
@@ -119,8 +146,8 @@ func main() {
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
defer rl.Close()
|
||||
rl.CaptureExitSignal()
|
||||
defer reader.Close()
|
||||
reader.CaptureExitSignal()
|
||||
|
||||
if inputIsStdin() {
|
||||
// commands are coming on stdin, however we will still enter
|
||||
@@ -130,13 +157,17 @@ func main() {
|
||||
|
||||
for {
|
||||
// primary program repl
|
||||
line, err := rl.Readline()
|
||||
line, err := reader.Readline()
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
|
||||
calc.Eval(line)
|
||||
rl.SetPrompt(calc.Prompt())
|
||||
err = calc.Eval(line)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
|
||||
reader.SetPrompt(calc.Prompt())
|
||||
}
|
||||
|
||||
if len(flag.Args()) > 0 {
|
||||
@@ -144,23 +175,31 @@ func main() {
|
||||
// echo 1 2 3 4 | rpn +
|
||||
// batch mode enabled automatically
|
||||
calc.batch = true
|
||||
calc.Eval(flag.Args()[0])
|
||||
if err = calc.Eval(flag.Args()[0]); err != nil {
|
||||
fmt.Println(err)
|
||||
|
||||
return 1
|
||||
}
|
||||
}
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
func inputIsStdin() bool {
|
||||
stat, _ := os.Stdin.Stat()
|
||||
|
||||
return (stat.Mode() & os.ModeCharDevice) == 0
|
||||
}
|
||||
|
||||
func man() {
|
||||
var buf bytes.Buffer
|
||||
|
||||
man := exec.Command("less", "-")
|
||||
|
||||
var b bytes.Buffer
|
||||
b.Write([]byte(manpage))
|
||||
buf.WriteString(manpage)
|
||||
|
||||
man.Stdout = os.Stdout
|
||||
man.Stdin = &b
|
||||
man.Stdin = &buf
|
||||
man.Stderr = os.Stderr
|
||||
|
||||
err := man.Run()
|
||||
|
||||
20
main_test.go
Normal file
20
main_test.go
Normal file
@@ -0,0 +1,20 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/rogpeppe/go-internal/testscript"
|
||||
)
|
||||
|
||||
func TestMain(m *testing.M) {
|
||||
os.Exit(testscript.RunMain(m, map[string]func() int{
|
||||
"testrpn": Main,
|
||||
}))
|
||||
}
|
||||
|
||||
func TestRpn(t *testing.T) {
|
||||
testscript.Run(t, testscript.Params{
|
||||
Dir: "t",
|
||||
})
|
||||
}
|
||||
163
rpn.go
163
rpn.go
@@ -2,7 +2,7 @@ package main
|
||||
|
||||
var manpage = `
|
||||
NAME
|
||||
rpn - Reverse Polish Notation Calculator for the commandline
|
||||
rpn - Programmable command-line calculator using reverse polish notation
|
||||
|
||||
SYNOPSIS
|
||||
Usage: rpn [-bdvh] [<operator>]
|
||||
@@ -10,6 +10,11 @@ SYNOPSIS
|
||||
Options:
|
||||
-b, --batchmode enable batch mode
|
||||
-d, --debug enable debug mode
|
||||
-s, --stack show last 5 items of the stack (off by default)
|
||||
-i --intermediate print intermediate results
|
||||
-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
|
||||
-h, --help show help
|
||||
|
||||
@@ -102,6 +107,9 @@ DESCRIPTION
|
||||
If the first parameter to rpn is a math operator or function, batch mode
|
||||
is enabled automatically, see last example.
|
||||
|
||||
You can enter integers, floating point numbers (positive or negative) or
|
||||
hex numbers (prefixed with 0x).
|
||||
|
||||
STACK MANIPULATION
|
||||
There are lots of stack manipulation commands provided. The most
|
||||
important one is undo which goes back to the stack before the last math
|
||||
@@ -111,7 +119,9 @@ DESCRIPTION
|
||||
switch or debug toggle command), then the backup stack is also being
|
||||
displayed.
|
||||
|
||||
The stack can be reversed using the reverse command.
|
||||
The stack can be reversed using the reverse command. However, sometimes
|
||||
only the last two values are in the wrong order. Use the swap command to
|
||||
exchange them.
|
||||
|
||||
You can use the shift command to remove the last number from the stack.
|
||||
|
||||
@@ -119,15 +129,23 @@ DESCRIPTION
|
||||
Basic operators:
|
||||
|
||||
+ add
|
||||
- substract
|
||||
- subtract
|
||||
/ divide
|
||||
x multiply (alias: *)
|
||||
^ power
|
||||
|
||||
Bitwise operators:
|
||||
|
||||
and bitwise and
|
||||
or bitwise or
|
||||
xor bitwise xor
|
||||
< left shift
|
||||
> right shift
|
||||
|
||||
Percent functions:
|
||||
|
||||
% percent
|
||||
%- substract percent
|
||||
%- subtract percent
|
||||
%+ add percent
|
||||
|
||||
Batch functions:
|
||||
@@ -145,8 +163,111 @@ DESCRIPTION
|
||||
log10 log1p log2 logb pow round roundtoeven sin sinh tan tanh trunc y0
|
||||
y1 copysign dim hypot
|
||||
|
||||
Conversion functions:
|
||||
|
||||
cm-to-inch
|
||||
inch-to-cm
|
||||
gallons-to-liters
|
||||
liters-to-gallons
|
||||
yards-to-meters
|
||||
meters-to-yards
|
||||
miles-to-kilometers
|
||||
kilometers-to-miles
|
||||
|
||||
Configuration Commands:
|
||||
|
||||
[no]batch toggle batch mode (nobatch turns it off)
|
||||
[no]debug toggle debug output (nodebug turns it off)
|
||||
[no]showstack show the last 5 items of the stack (noshowtack turns it off)
|
||||
|
||||
Show commands:
|
||||
|
||||
dump display the stack contents
|
||||
hex show last stack item in hex form (converted to int)
|
||||
history display calculation history
|
||||
vars show list of variables
|
||||
|
||||
Stack manipulation commands:
|
||||
|
||||
clear clear the whole stack
|
||||
shift remove the last element of the stack
|
||||
reverse reverse the stack elements
|
||||
swap exchange the last two stack elements
|
||||
dup duplicate last stack item
|
||||
undo undo last operation
|
||||
edit edit the stack interactively using vi or $EDITOR
|
||||
|
||||
Other commands:
|
||||
|
||||
help|? show this message
|
||||
manual show manual
|
||||
quit|exit|c-d|c-c exit program
|
||||
|
||||
Register variables:
|
||||
|
||||
>NAME Put last stack element into variable NAME
|
||||
<NAME Retrieve variable NAME and put onto stack
|
||||
|
||||
Refer to https://pkg.go.dev/math for details about those functions.
|
||||
|
||||
There are also a number of shortcuts for some commands available:
|
||||
|
||||
d debug
|
||||
b batch
|
||||
s showstack
|
||||
h history
|
||||
p dump (aka print)
|
||||
v vars
|
||||
c clear
|
||||
u undo
|
||||
|
||||
INTERACTIVE REPL
|
||||
While you can use rpn in the command-line, the best experience you'll
|
||||
have is the interactive repl (read eval print loop). Just execute "rpn"
|
||||
and you'll be there.
|
||||
|
||||
In interactive mode you can use TAB completion to complete commands,
|
||||
operators and functions. There's also a history, which allows you to
|
||||
repeat complicated calculations (as long as you've entered them in one
|
||||
line).
|
||||
|
||||
There are also a lot of key bindings, here are the most important ones:
|
||||
|
||||
ctrl-c + ctrl-d
|
||||
Exit interactive rpn
|
||||
|
||||
ctrl-z
|
||||
Send rpn to the backgound.
|
||||
|
||||
ctrl-a
|
||||
Beginning of line.
|
||||
|
||||
ctrl-e
|
||||
End of line.
|
||||
|
||||
ctrl-l
|
||||
Clear the screen.
|
||||
|
||||
ctrl-r
|
||||
Search through history.
|
||||
|
||||
COMMENTS
|
||||
Lines starting with "#" are being ignored as comments. You can also
|
||||
append comments to rpn input, e.g.:
|
||||
|
||||
# a comment
|
||||
123 # another comment
|
||||
|
||||
In this case only 123 will be added to the stack.
|
||||
|
||||
VARIABLES
|
||||
You can register the last item of the stack into a variable. Variable
|
||||
names must be all caps. Use the ">NAME" command to put a value into
|
||||
variable "NAME". Use "<NAME" to retrieve the value of variable "NAME"
|
||||
and put it onto the stack.
|
||||
|
||||
The command vars can be used to get a list of all variables.
|
||||
|
||||
EXTENDING RPN USING LUA
|
||||
You can use a lua script with lua functions to extend the calculator. By
|
||||
default the tool looks for "~/.rpn.lua". You can also specify a script
|
||||
@@ -173,8 +294,13 @@ EXTENDING RPN USING LUA
|
||||
|
||||
* function name
|
||||
|
||||
* number of arguments expected (1,2 or -1 allowed), -1 means batch
|
||||
mode.
|
||||
* number of arguments expected (see below)
|
||||
|
||||
Number of expected arguments can be:
|
||||
|
||||
- 0: expect 1 argument but do NOT modify the stack
|
||||
- 1-n: do a singular calculation
|
||||
- -1: batch mode work with all numbers on the stack
|
||||
|
||||
* help text
|
||||
|
||||
@@ -185,6 +311,15 @@ EXTENDING RPN USING LUA
|
||||
So you can't open files, execute other programs or open a connection to
|
||||
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
|
||||
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.
|
||||
@@ -204,7 +339,7 @@ LICENSE
|
||||
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:
|
||||
|
||||
@@ -222,17 +357,3 @@ AUTHORS
|
||||
Thomas von Dein tom AT vondein DOT org
|
||||
|
||||
`
|
||||
var usage = `
|
||||
|
||||
Usage: rpn [-bdvh] [<operator>]
|
||||
|
||||
Options:
|
||||
-b, --batchmode enable batch mode
|
||||
-d, --debug enable debug mode
|
||||
-v, --version show version
|
||||
-h, --help show help
|
||||
|
||||
When <operator> is given, batch mode ist automatically enabled. Use
|
||||
this only when working with stdin. E.g.: echo "2 3 4 5" | rpn +
|
||||
|
||||
`
|
||||
|
||||
165
rpn.pod
165
rpn.pod
@@ -1,6 +1,6 @@
|
||||
=head1 NAME
|
||||
|
||||
rpn - Reverse Polish Notation Calculator for the commandline
|
||||
rpn - Programmable command-line calculator using reverse polish notation
|
||||
|
||||
=head1 SYNOPSIS
|
||||
|
||||
@@ -9,6 +9,11 @@ rpn - Reverse Polish Notation Calculator for the commandline
|
||||
Options:
|
||||
-b, --batchmode enable batch mode
|
||||
-d, --debug enable debug mode
|
||||
-s, --stack show last 5 items of the stack (off by default)
|
||||
-i --intermediate print intermediate results
|
||||
-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
|
||||
-h, --help show help
|
||||
|
||||
@@ -106,6 +111,9 @@ Example of batch mode usage:
|
||||
If the first parameter to rpn is a math operator or function, batch
|
||||
mode is enabled automatically, see last example.
|
||||
|
||||
You can enter integers, floating point numbers (positive or negative)
|
||||
or hex numbers (prefixed with 0x).
|
||||
|
||||
=head2 STACK MANIPULATION
|
||||
|
||||
There are lots of stack manipulation commands provided. The most
|
||||
@@ -116,7 +124,9 @@ You can use B<dump> to display the stack. If debugging
|
||||
is enabled (C<-d> switch or B<debug> toggle command), then the backup
|
||||
stack is also being displayed.
|
||||
|
||||
The stack can be reversed using the B<reverse> command.
|
||||
The stack can be reversed using the B<reverse> command. However,
|
||||
sometimes only the last two values are in the wrong order. Use the
|
||||
B<swap> command to exchange them.
|
||||
|
||||
You can use the B<shift> command to remove the last number from the
|
||||
stack.
|
||||
@@ -126,15 +136,23 @@ stack.
|
||||
Basic operators:
|
||||
|
||||
+ add
|
||||
- substract
|
||||
- subtract
|
||||
/ divide
|
||||
x multiply (alias: *)
|
||||
^ power
|
||||
|
||||
Bitwise operators:
|
||||
|
||||
and bitwise and
|
||||
or bitwise or
|
||||
xor bitwise xor
|
||||
< left shift
|
||||
> right shift
|
||||
|
||||
Percent functions:
|
||||
|
||||
% percent
|
||||
%- substract percent
|
||||
%- subtract percent
|
||||
%+ add percent
|
||||
|
||||
Batch functions:
|
||||
@@ -152,8 +170,126 @@ Math functions:
|
||||
log10 log1p log2 logb pow round roundtoeven sin sinh tan tanh trunc y0
|
||||
y1 copysign dim hypot
|
||||
|
||||
Conversion functions:
|
||||
|
||||
cm-to-inch
|
||||
inch-to-cm
|
||||
gallons-to-liters
|
||||
liters-to-gallons
|
||||
yards-to-meters
|
||||
meters-to-yards
|
||||
miles-to-kilometers
|
||||
kilometers-to-miles
|
||||
|
||||
Configuration Commands:
|
||||
|
||||
[no]batch toggle batch mode (nobatch turns it off)
|
||||
[no]debug toggle debug output (nodebug turns it off)
|
||||
[no]showstack show the last 5 items of the stack (noshowtack turns it off)
|
||||
|
||||
Show commands:
|
||||
|
||||
dump display the stack contents
|
||||
hex show last stack item in hex form (converted to int)
|
||||
history display calculation history
|
||||
vars show list of variables
|
||||
|
||||
Stack manipulation commands:
|
||||
|
||||
clear clear the whole stack
|
||||
shift remove the last element of the stack
|
||||
reverse reverse the stack elements
|
||||
swap exchange the last two stack elements
|
||||
dup duplicate last stack item
|
||||
undo undo last operation
|
||||
edit edit the stack interactively using vi or $EDITOR
|
||||
|
||||
Other commands:
|
||||
|
||||
help|? show this message
|
||||
manual show manual
|
||||
quit|exit|c-d|c-c exit program
|
||||
|
||||
|
||||
Register variables:
|
||||
|
||||
>NAME Put last stack element into variable NAME
|
||||
<NAME Retrieve variable NAME and put onto stack
|
||||
|
||||
Refer to https://pkg.go.dev/math for details about those functions.
|
||||
|
||||
There are also a number of shortcuts for some commands available:
|
||||
|
||||
d debug
|
||||
b batch
|
||||
s showstack
|
||||
h history
|
||||
p dump (aka print)
|
||||
v vars
|
||||
c clear
|
||||
u undo
|
||||
|
||||
=head1 INTERACTIVE REPL
|
||||
|
||||
While you can use rpn in the command-line, the best experience you'll
|
||||
have is the interactive repl (read eval print loop). Just execute
|
||||
C<rpn> and you'll be there.
|
||||
|
||||
In interactive mode you can use TAB completion to complete commands,
|
||||
operators and functions. There's also a history, which allows you to
|
||||
repeat complicated calculations (as long as you've entered them in one
|
||||
line).
|
||||
|
||||
There are also a lot of key bindings, here are the most important
|
||||
ones:
|
||||
|
||||
=over
|
||||
|
||||
=item ctrl-c + ctrl-d
|
||||
|
||||
Exit interactive rpn
|
||||
|
||||
=item ctrl-z
|
||||
|
||||
Send rpn to the backgound.
|
||||
|
||||
=item ctrl-a
|
||||
|
||||
Beginning of line.
|
||||
|
||||
=item ctrl-e
|
||||
|
||||
End of line.
|
||||
|
||||
=item ctrl-l
|
||||
|
||||
Clear the screen.
|
||||
|
||||
=item ctrl-r
|
||||
|
||||
Search through history.
|
||||
|
||||
=back
|
||||
|
||||
=head1 COMMENTS
|
||||
|
||||
Lines starting with C<#> are being ignored as comments. You can also
|
||||
append comments to rpn input, e.g.:
|
||||
|
||||
# a comment
|
||||
123 # another comment
|
||||
|
||||
In this case only 123 will be added to the stack.
|
||||
|
||||
=head1 VARIABLES
|
||||
|
||||
You can register the last item of the stack into a variable. Variable
|
||||
names must be all caps. Use the ">NAME" command to put a value into
|
||||
variable "NAME". Use "<NAME" to retrieve the value of variable "NAME"
|
||||
and put it onto the stack.
|
||||
|
||||
The command B<vars> can be used to get a list of all variables.
|
||||
|
||||
=head1 EXTENDING RPN USING LUA
|
||||
|
||||
You can use a lua script with lua functions to extend the
|
||||
@@ -187,8 +323,13 @@ function name
|
||||
|
||||
=item *
|
||||
|
||||
number of arguments expected (1,2 or -1 allowed), -1 means batch
|
||||
mode.
|
||||
number of arguments expected (see below)
|
||||
|
||||
Number of expected arguments can be:
|
||||
|
||||
- 0: expect 1 argument but do NOT modify the stack
|
||||
- 1-n: do a singular calculation
|
||||
- -1: batch mode work with all numbers on the stack
|
||||
|
||||
=item *
|
||||
|
||||
@@ -203,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
|
||||
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
|
||||
|
||||
In interactive mode you can enter the B<help> command (or B<?>) to get
|
||||
@@ -225,7 +376,7 @@ L<https://github.com/TLINDEN/rpnc/issues>.
|
||||
|
||||
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:
|
||||
|
||||
|
||||
68
stack.go
68
stack.go
@@ -64,14 +64,14 @@ func (s *Stack) Bump() {
|
||||
}
|
||||
|
||||
// append an item to the stack
|
||||
func (s *Stack) Push(x float64) {
|
||||
func (s *Stack) Push(item float64) {
|
||||
s.mutex.Lock()
|
||||
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.linklist.PushBack(x)
|
||||
s.linklist.PushBack(item)
|
||||
}
|
||||
|
||||
// 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.Bump()
|
||||
|
||||
return val.(float64)
|
||||
}
|
||||
|
||||
@@ -115,20 +116,41 @@ func (s *Stack) Shift(num ...int) {
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Stack) Swap() {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
if s.linklist.Len() < 2 {
|
||||
return
|
||||
}
|
||||
|
||||
prevA := s.linklist.Back()
|
||||
s.linklist.Remove(prevA)
|
||||
|
||||
prevB := s.linklist.Back()
|
||||
s.linklist.Remove(prevB)
|
||||
|
||||
s.Debug(fmt.Sprintf("swapping %.2f with %.2f", prevB.Value, prevA.Value))
|
||||
|
||||
s.linklist.PushBack(prevA.Value)
|
||||
s.linklist.PushBack(prevB.Value)
|
||||
}
|
||||
|
||||
// Return the last num items from the stack w/o modifying it.
|
||||
func (s *Stack) Last(num ...int) []float64 {
|
||||
items := []float64{}
|
||||
i := s.Len()
|
||||
stacklen := s.Len()
|
||||
count := 1
|
||||
|
||||
if len(num) > 0 {
|
||||
count = num[0]
|
||||
}
|
||||
|
||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||
if i <= count {
|
||||
if stacklen <= count {
|
||||
items = append(items, e.Value.(float64))
|
||||
}
|
||||
i--
|
||||
stacklen--
|
||||
}
|
||||
|
||||
return items
|
||||
@@ -148,12 +170,14 @@ func (s *Stack) All() []float64 {
|
||||
// dump the stack to stdout, including backup if debug is enabled
|
||||
func (s *Stack) Dump() {
|
||||
fmt.Printf("Stack revision %d (%p):\n", s.rev, &s.linklist)
|
||||
|
||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||
fmt.Println(e.Value)
|
||||
}
|
||||
|
||||
if s.debug {
|
||||
fmt.Printf("Backup stack revision %d (%p):\n", s.backuprev, &s.backup)
|
||||
|
||||
for e := s.backup.Front(); e != nil; e = e.Next() {
|
||||
fmt.Println(e.Value)
|
||||
}
|
||||
@@ -175,31 +199,53 @@ func (s *Stack) Backup() {
|
||||
// make a backup, because the elements in list.List{} are pointers
|
||||
// and lead to unexpected results. The methid here works reliably
|
||||
// at least.
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
s.Debug(fmt.Sprintf("backing up %d items from rev %d",
|
||||
s.linklist.Len(), s.rev))
|
||||
|
||||
s.backup = list.List{}
|
||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||
s.backup.PushBack(e.Value)
|
||||
s.backup.PushBack(e.Value.(float64))
|
||||
}
|
||||
|
||||
s.backuprev = s.rev
|
||||
}
|
||||
|
||||
func (s *Stack) Restore() {
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
if s.rev == 0 {
|
||||
fmt.Println("error: stack is empty.")
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
s.Debug(fmt.Sprintf("restoring stack to revision %d", s.backuprev))
|
||||
|
||||
s.rev = s.backuprev
|
||||
s.linklist = s.backup
|
||||
|
||||
s.linklist = list.List{}
|
||||
for e := s.backup.Front(); e != nil; e = e.Next() {
|
||||
s.linklist.PushBack(e.Value.(float64))
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Stack) Reverse() {
|
||||
newstack := list.List{}
|
||||
s.mutex.Lock()
|
||||
defer s.mutex.Unlock()
|
||||
|
||||
items := []float64{}
|
||||
|
||||
for e := s.linklist.Front(); e != nil; e = e.Next() {
|
||||
newstack.PushFront(e.Value)
|
||||
tail := s.linklist.Back()
|
||||
items = append(items, tail.Value.(float64))
|
||||
s.linklist.Remove(tail)
|
||||
}
|
||||
|
||||
s.linklist = newstack
|
||||
for i := len(items) - 1; i >= 0; i-- {
|
||||
s.linklist.PushFront(items[i])
|
||||
}
|
||||
}
|
||||
|
||||
190
stack_test.go
Normal file
190
stack_test.go
Normal file
@@ -0,0 +1,190 @@
|
||||
/*
|
||||
Copyright © 2023 Thomas von Dein
|
||||
|
||||
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
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestPush(t *testing.T) {
|
||||
t.Run("push", func(t *testing.T) {
|
||||
s := NewStack()
|
||||
s.Push(5)
|
||||
|
||||
if s.linklist.Back().Value != 5.0 {
|
||||
t.Errorf("push failed:\n+++ got: %f\n--- want: %f",
|
||||
s.linklist.Back().Value, 5.0)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestPop(t *testing.T) {
|
||||
t.Run("pop", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Push(5)
|
||||
got := stack.Pop()
|
||||
|
||||
if got != 5.0 {
|
||||
t.Errorf("pop failed:\n+++ got: %f\n--- want: %f",
|
||||
got, 5.0)
|
||||
}
|
||||
|
||||
if stack.Len() != 0 {
|
||||
t.Errorf("stack not empty after pop()")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestPops(t *testing.T) {
|
||||
t.Run("pops", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Push(5)
|
||||
stack.Push(5)
|
||||
stack.Push(5)
|
||||
stack.Pop()
|
||||
|
||||
if stack.Len() != 2 {
|
||||
t.Errorf("stack len not correct after pop:\n+++ got: %d\n--- want: %d",
|
||||
stack.Len(), 2)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestShift(t *testing.T) {
|
||||
t.Run("shift", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Shift()
|
||||
|
||||
if stack.Len() != 0 {
|
||||
t.Errorf("stack not empty after shift()")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestClear(t *testing.T) {
|
||||
t.Run("clear", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Push(5)
|
||||
stack.Push(5)
|
||||
stack.Push(5)
|
||||
stack.Clear()
|
||||
|
||||
if stack.Len() != 0 {
|
||||
t.Errorf("stack not empty after clear()")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestLast(t *testing.T) {
|
||||
t.Run("last", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Push(5)
|
||||
got := stack.Last()
|
||||
|
||||
if len(got) != 1 {
|
||||
t.Errorf("last failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||
len(got), 1)
|
||||
}
|
||||
|
||||
if got[0] != 5.0 {
|
||||
t.Errorf("last failed:\n+++ got: %f\n--- want: %f",
|
||||
got, 5.0)
|
||||
}
|
||||
|
||||
if stack.Len() != 1 {
|
||||
t.Errorf("stack modified after last()")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestAll(t *testing.T) {
|
||||
t.Run("all", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
list := []float64{2, 4, 6, 8}
|
||||
|
||||
for _, item := range list {
|
||||
stack.Push(item)
|
||||
}
|
||||
|
||||
got := stack.All()
|
||||
|
||||
if len(got) != len(list) {
|
||||
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||
len(got), len(list))
|
||||
}
|
||||
|
||||
for i := 1; i < len(list); i++ {
|
||||
if got[i] != list[i] {
|
||||
t.Errorf("all failed (element %d):\n+++ got: %f\n--- want: %f",
|
||||
i, got[i], list[i])
|
||||
}
|
||||
}
|
||||
|
||||
if stack.Len() != len(list) {
|
||||
t.Errorf("stack modified after last()")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestBackupRestore(t *testing.T) {
|
||||
t.Run("shift", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
stack.Push(5)
|
||||
stack.Backup()
|
||||
stack.Clear()
|
||||
stack.Restore()
|
||||
|
||||
if stack.Len() != 1 {
|
||||
t.Errorf("stack not correctly restored()")
|
||||
}
|
||||
|
||||
value := stack.Pop()
|
||||
if value != 5.0 {
|
||||
t.Errorf("stack not identical to old revision:\n+++ got: %f\n--- want: %f",
|
||||
value, 5.0)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestReverse(t *testing.T) {
|
||||
t.Run("reverse", func(t *testing.T) {
|
||||
stack := NewStack()
|
||||
list := []float64{2, 4, 6}
|
||||
reverse := []float64{6, 4, 2}
|
||||
|
||||
for _, item := range list {
|
||||
stack.Push(item)
|
||||
}
|
||||
|
||||
stack.Reverse()
|
||||
|
||||
got := stack.All()
|
||||
|
||||
if len(got) != len(list) {
|
||||
t.Errorf("all failed:\n+++ got: %d elements\n--- want: %d elements",
|
||||
len(got), len(list))
|
||||
}
|
||||
|
||||
for i := 1; i < len(reverse); i++ {
|
||||
if got[i] != reverse[i] {
|
||||
t.Errorf("reverse failed (element %d):\n+++ got: %f\n--- want: %f",
|
||||
i, got[i], list[i])
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
2
t/cmdline-command.txtar
Normal file
2
t/cmdline-command.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn 1 2 dump
|
||||
stdout 'Stack revision 2 .0x'
|
||||
2
t/cmdline-invalidcommand.txtar
Normal file
2
t/cmdline-invalidcommand.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
! exec testrpn 1 2 dumb
|
||||
stdout 'unknown command or operator'
|
||||
2
t/cmdline-precision.txtar
Normal file
2
t/cmdline-precision.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn -p 4 2 3 /
|
||||
stdout '0.6667\n'
|
||||
2
t/cmdline-short-stack.txtar
Normal file
2
t/cmdline-short-stack.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
! exec testrpn 4 +
|
||||
stdout 'stack doesn''t provide enough arguments'
|
||||
2
t/cmdlinecalc-debug.txtar
Normal file
2
t/cmdlinecalc-debug.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn -d 44 55 *
|
||||
stdout 'push to stack: 2420.00\n'
|
||||
2
t/cmdlinecalc-divzero.txtar
Normal file
2
t/cmdlinecalc-divzero.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
! exec testrpn 100 50 50 - /
|
||||
stdout 'division by null'
|
||||
16
t/cmdlinecalc-lua.txtar
Normal file
16
t/cmdlinecalc-lua.txtar
Normal file
@@ -0,0 +1,16 @@
|
||||
exec testrpn -d -c test.lua 3 5 lower
|
||||
stdout '3\n'
|
||||
|
||||
-- test.lua --
|
||||
function lower(a,b)
|
||||
if a < b then
|
||||
return a
|
||||
else
|
||||
return b
|
||||
end
|
||||
end
|
||||
|
||||
function init()
|
||||
-- expects 2 args
|
||||
register("lower", 2, "lower")
|
||||
end
|
||||
2
t/cmdlinecalc.txtar
Normal file
2
t/cmdlinecalc.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn 44 55 *
|
||||
stdout '2420\n'
|
||||
2
t/getman.txtar
Normal file
2
t/getman.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn -m
|
||||
stdout 'This software is licensed under the GNU GENERAL PUBLIC LICENSE'
|
||||
2
t/getusage.txtar
Normal file
2
t/getusage.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn -h
|
||||
stdout 'This is rpn'
|
||||
2
t/getversion.txtar
Normal file
2
t/getversion.txtar
Normal file
@@ -0,0 +1,2 @@
|
||||
exec testrpn -v
|
||||
stdout 'This is rpn version'
|
||||
4
t/stdin-batch-cmd.txtar
Normal file
4
t/stdin-batch-cmd.txtar
Normal file
@@ -0,0 +1,4 @@
|
||||
exec echo 1 2 3 4 5 batch median
|
||||
stdin stdout
|
||||
exec testrpn
|
||||
[unix] stdout '3\n'
|
||||
4
t/stdin-batch.txtar
Normal file
4
t/stdin-batch.txtar
Normal file
@@ -0,0 +1,4 @@
|
||||
exec echo 1 2 3 4 5
|
||||
stdin stdout
|
||||
[unix] exec testrpn median
|
||||
[unix] stdout '3\n'
|
||||
4
t/stdin-calc.txtar
Normal file
4
t/stdin-calc.txtar
Normal file
@@ -0,0 +1,4 @@
|
||||
exec echo 10 10 +
|
||||
stdin stdout
|
||||
exec testrpn
|
||||
[unix] stdout '20\n'
|
||||
6
t/stdin-help.txtar
Normal file
6
t/stdin-help.txtar
Normal file
@@ -0,0 +1,6 @@
|
||||
stdin input.txt
|
||||
exec testrpn
|
||||
[unix] stdout 'Available configuration commands'
|
||||
|
||||
-- input.txt --
|
||||
?
|
||||
13
t/stdin-use-vars.txtar
Normal file
13
t/stdin-use-vars.txtar
Normal file
@@ -0,0 +1,13 @@
|
||||
stdin input.txt
|
||||
exec testrpn
|
||||
[unix] stdout '28\n'
|
||||
|
||||
-- input.txt --
|
||||
10
|
||||
10
|
||||
+
|
||||
>SUM
|
||||
clear
|
||||
8
|
||||
<SUM
|
||||
+
|
||||
4
t/stdint-explicit-batch.txtar
Normal file
4
t/stdint-explicit-batch.txtar
Normal file
@@ -0,0 +1,4 @@
|
||||
exec echo 1 2 3 4 5 median
|
||||
stdin stdout
|
||||
exec testrpn -b
|
||||
[unix] stdout '3\n'
|
||||
13
t/test.lua
Normal file
13
t/test.lua
Normal file
@@ -0,0 +1,13 @@
|
||||
-- simple function, return the lower number of the two operands
|
||||
function lower(a,b)
|
||||
if a < b then
|
||||
return a
|
||||
else
|
||||
return b
|
||||
end
|
||||
end
|
||||
|
||||
function init()
|
||||
-- expects 2 args
|
||||
register("lower", 2, "lower")
|
||||
end
|
||||
17
test.lua
17
test.lua
@@ -1,17 +0,0 @@
|
||||
function add(a,b)
|
||||
return a + b
|
||||
end
|
||||
|
||||
function test(a)
|
||||
return a
|
||||
end
|
||||
|
||||
function parallelresistance(a,b)
|
||||
return 1.0 / (a * b)
|
||||
end
|
||||
|
||||
function init()
|
||||
register("add", 2, "addition")
|
||||
register("test", 1, "test")
|
||||
register("parallelresistance", 2, "parallel resistance")
|
||||
end
|
||||
22
util.go
22
util.go
@@ -23,13 +23,23 @@ import (
|
||||
"strings"
|
||||
)
|
||||
|
||||
// find an item in a list
|
||||
func contains(s []string, e string) bool {
|
||||
for _, a := range s {
|
||||
if a == e {
|
||||
// find an item in a list, generic variant
|
||||
func contains[E comparable](s []E, v E) bool {
|
||||
for _, vs := range s {
|
||||
if v == vs {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
// look if a key in a map exists, generic variant
|
||||
func exists[K comparable, V any](m map[K]V, v K) bool {
|
||||
if _, ok := m[v]; ok {
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
@@ -63,3 +73,7 @@ func const2num(name string) float64 {
|
||||
func list2str(list Numbers) string {
|
||||
return strings.Trim(strings.Join(strings.Fields(fmt.Sprint(list)), " "), "[]")
|
||||
}
|
||||
|
||||
func Error(m string) error {
|
||||
return fmt.Errorf("Error: %s", m)
|
||||
}
|
||||
|
||||
32
util_test.go
Normal file
32
util_test.go
Normal file
@@ -0,0 +1,32 @@
|
||||
/*
|
||||
Copyright © 2023 Thomas von Dein
|
||||
|
||||
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
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestContains(t *testing.T) {
|
||||
list := []string{"a", "b", "c"}
|
||||
|
||||
t.Run("contains", func(t *testing.T) {
|
||||
if !contains(list, "a") {
|
||||
t.Errorf("a in [a,b,c] not found")
|
||||
}
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user