mirror of
https://github.com/kovidgoyal/kitty.git
synced 2026-02-01 11:34:59 +01:00
105 lines
2.0 KiB
Go
105 lines
2.0 KiB
Go
package humanize
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"golang.org/x/exp/constraints"
|
|
)
|
|
|
|
// IEC Sizes.
|
|
// kibis of bits
|
|
const (
|
|
Byte = 1 << (iota * 10)
|
|
KiByte
|
|
MiByte
|
|
GiByte
|
|
TiByte
|
|
PiByte
|
|
EiByte
|
|
)
|
|
|
|
// SI Sizes.
|
|
const (
|
|
IByte = 1
|
|
KByte = IByte * 1000
|
|
MByte = KByte * 1000
|
|
GByte = MByte * 1000
|
|
TByte = GByte * 1000
|
|
PByte = TByte * 1000
|
|
EByte = PByte * 1000
|
|
)
|
|
|
|
func logn(n, b float64) float64 {
|
|
return math.Log(n) / math.Log(b)
|
|
}
|
|
|
|
func humanize_bytes(s uint64, base float64, sizes []string, sep string) string {
|
|
if s < 10 {
|
|
return fmt.Sprintf("%d%sB", s, sep)
|
|
}
|
|
e := math.Floor(logn(float64(s), base))
|
|
suffix := sizes[int(e)]
|
|
val := math.Floor(float64(s)/math.Pow(base, e)*10+0.5) / 10
|
|
f := "%.0f%s%s"
|
|
if val < 10 {
|
|
f = "%.1f%s%s"
|
|
}
|
|
return fmt.Sprintf(f, val, sep, suffix)
|
|
}
|
|
|
|
// Bytes produces a human readable representation of an SI size.
|
|
// Bytes(82854982) -> 83 MB
|
|
func Bytes(s uint64) string {
|
|
return Size(s, SizeOptions{})
|
|
}
|
|
|
|
// IBytes produces a human readable representation of an IEC size.
|
|
// IBytes(82854982) -> 79 MiB
|
|
func IBytes(s uint64) string {
|
|
return Size(s, SizeOptions{Base: 1024})
|
|
}
|
|
|
|
type SizeOptions struct {
|
|
Separator string
|
|
Base int
|
|
}
|
|
|
|
func Size[T constraints.Integer | constraints.Float](s T, opts ...SizeOptions) string {
|
|
var o SizeOptions
|
|
prefix := ""
|
|
if len(opts) == 0 {
|
|
o = SizeOptions{}
|
|
} else {
|
|
o = opts[0]
|
|
}
|
|
if s < 0 {
|
|
prefix = "-"
|
|
}
|
|
if o.Separator == "" {
|
|
o.Separator = " "
|
|
}
|
|
if o.Base == 0 {
|
|
o.Base = 1000
|
|
}
|
|
var sizes []string
|
|
switch o.Base {
|
|
default:
|
|
sizes = []string{"B", "kB", "MB", "GB", "TB", "PB", "EB"}
|
|
case 1024:
|
|
sizes = []string{"B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB"}
|
|
}
|
|
return prefix + humanize_bytes(uint64(s), float64(o.Base), sizes, o.Separator)
|
|
}
|
|
|
|
func FormatNumber[T constraints.Float](n T, max_num_of_decimals ...int) string {
|
|
prec := 2
|
|
if len(max_num_of_decimals) > 0 {
|
|
prec = max_num_of_decimals[0]
|
|
}
|
|
ans := strconv.FormatFloat(float64(n), 'f', prec, 64)
|
|
return strings.TrimRight(strings.TrimRight(ans, "0"), ".")
|
|
}
|