File: bytes.go

package info (click to toggle)
kitty 0.42.1-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 28,564 kB
  • sloc: ansic: 82,787; python: 55,191; objc: 5,122; sh: 1,295; xml: 364; makefile: 143; javascript: 78
file content (104 lines) | stat: -rw-r--r-- 2,055 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
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"), ".")
}