Compare commits
4 Commits
9dc6e1443d
...
8b88beae88
Author | SHA1 | Date | |
---|---|---|---|
8b88beae88 | |||
860e63b4be | |||
079535ce6b | |||
10dbda7986 |
13
f.go
13
f.go
|
@ -9,7 +9,6 @@ import (
|
||||||
"math"
|
"math"
|
||||||
"os"
|
"os"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -89,7 +88,7 @@ const HELP_MSG = " Operation | Variable assignment | Special\n" +
|
||||||
" exit\n"
|
" exit\n"
|
||||||
const NORMAL90CONFIDENCE = 1.6448536269514727
|
const NORMAL90CONFIDENCE = 1.6448536269514727
|
||||||
const INIT_DIST Scalar = Scalar(1)
|
const INIT_DIST Scalar = Scalar(1)
|
||||||
const N_SAMPLES = 100_000
|
const N_SAMPLES = 1_000_000
|
||||||
|
|
||||||
/* Printers */
|
/* Printers */
|
||||||
func prettyPrintDist(dist Dist) {
|
func prettyPrintDist(dist Dist) {
|
||||||
|
@ -305,7 +304,7 @@ func parseWordsIntoOpAndDist(words []string, vars map[string]Dist) (string, Dist
|
||||||
return parseWordsErr("Operator must have operand; can't operate on nothing")
|
return parseWordsErr("Operator must have operand; can't operate on nothing")
|
||||||
case 1:
|
case 1:
|
||||||
var_word, var_word_exists := vars[words[0]]
|
var_word, var_word_exists := vars[words[0]]
|
||||||
single_float, err1 := strconv.ParseFloat(words[0], 64) // abstract this away to search for K/M/B/T/etc.
|
single_float, err1 := pretty.ParseFloat(words[0]) // abstract this away to search for K/M/B/T/etc.
|
||||||
switch {
|
switch {
|
||||||
case var_word_exists:
|
case var_word_exists:
|
||||||
dist = var_word
|
dist = var_word
|
||||||
|
@ -315,16 +314,16 @@ func parseWordsIntoOpAndDist(words []string, vars map[string]Dist) (string, Dist
|
||||||
return parseWordsErr("Trying to operate on a scalar, but scalar is neither a float nor an assigned variable")
|
return parseWordsErr("Trying to operate on a scalar, but scalar is neither a float nor an assigned variable")
|
||||||
}
|
}
|
||||||
case 2:
|
case 2:
|
||||||
new_low, err1 := strconv.ParseFloat(words[0], 64)
|
new_low, err1 := pretty.ParseFloat(words[0])
|
||||||
new_high, err2 := strconv.ParseFloat(words[1], 64)
|
new_high, err2 := pretty.ParseFloat(words[1])
|
||||||
if err1 != nil || err2 != nil {
|
if err1 != nil || err2 != nil {
|
||||||
return parseWordsErr("Trying to operate by a distribution, but distribution is not specified as two floats")
|
return parseWordsErr("Trying to operate by a distribution, but distribution is not specified as two floats")
|
||||||
}
|
}
|
||||||
dist = Lognormal{low: new_low, high: new_high}
|
dist = Lognormal{low: new_low, high: new_high}
|
||||||
case 3:
|
case 3:
|
||||||
if words[0] == "beta" || words[0] == "b" {
|
if words[0] == "beta" || words[0] == "b" {
|
||||||
a, err1 := strconv.ParseFloat(words[1], 64)
|
a, err1 := pretty.ParseFloat(words[1])
|
||||||
b, err2 := strconv.ParseFloat(words[2], 64)
|
b, err2 := pretty.ParseFloat(words[2])
|
||||||
if err1 != nil || err2 != nil {
|
if err1 != nil || err2 != nil {
|
||||||
return parseWordsErr("Trying to specify a beta distribution? Try beta 1 2")
|
return parseWordsErr("Trying to specify a beta distribution? Try beta 1 2")
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,8 +1,10 @@
|
||||||
package pretty
|
package pretty
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"math"
|
"math"
|
||||||
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
func PrettyPrintInt(n int) {
|
func PrettyPrintInt(n int) {
|
||||||
|
@ -49,3 +51,37 @@ func PrettyPrint2Floats(low float64, high float64) {
|
||||||
fmt.Printf(" ")
|
fmt.Printf(" ")
|
||||||
PrettyPrintFloat(high)
|
PrettyPrintFloat(high)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func multiplyOrPassThroughError(a float64, b float64, err error) (float64, error) {
|
||||||
|
if err != nil {
|
||||||
|
return b, err
|
||||||
|
} else {
|
||||||
|
return a * b, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseFloat(word string) (float64, error) {
|
||||||
|
// l = len(word) // assuming no UTF stuff
|
||||||
|
switch len(word) {
|
||||||
|
case 0:
|
||||||
|
return 0, errors.New("String to be parsed into float must not be the empty string")
|
||||||
|
case 1:
|
||||||
|
return strconv.ParseFloat(word, 64)
|
||||||
|
}
|
||||||
|
|
||||||
|
n := len(word) - 1
|
||||||
|
f, err := strconv.ParseFloat(word[:n], 64)
|
||||||
|
switch word[n] {
|
||||||
|
case 'K':
|
||||||
|
return multiplyOrPassThroughError(1_000, f, err)
|
||||||
|
case 'M':
|
||||||
|
return multiplyOrPassThroughError(1_000_000, f, err)
|
||||||
|
case 'B':
|
||||||
|
return multiplyOrPassThroughError(1_000_000_000, f, err)
|
||||||
|
case 'T':
|
||||||
|
return multiplyOrPassThroughError(1_000_000_000_000, f, err)
|
||||||
|
default:
|
||||||
|
return strconv.ParseFloat(word, 64)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user