mirror of https://github.com/bcicen/ctop.git
75 lines
1.3 KiB
Go
75 lines
1.3 KiB
Go
package cwidgets
|
|
|
|
import (
|
|
"strconv"
|
|
)
|
|
|
|
const (
|
|
// byte ratio constants
|
|
_ = iota
|
|
kib float64 = 1 << (10 * iota)
|
|
mib
|
|
gib
|
|
tib
|
|
pib
|
|
)
|
|
|
|
var (
|
|
units = []float64{
|
|
1,
|
|
kib,
|
|
mib,
|
|
gib,
|
|
tib,
|
|
pib,
|
|
}
|
|
|
|
// short, full unit labels
|
|
labels = [][2]string{
|
|
[2]string{"B", "B"},
|
|
[2]string{"K", "KiB"},
|
|
[2]string{"M", "MiB"},
|
|
[2]string{"G", "GiB"},
|
|
[2]string{"T", "TiB"},
|
|
[2]string{"P", "PiB"},
|
|
}
|
|
)
|
|
|
|
// convenience methods
|
|
func ByteFormat(n int) string { return byteFormat(float64(n), false) }
|
|
func ByteFormatShort(n int) string { return byteFormat(float64(n), true) }
|
|
func ByteFormat64(n int64) string { return byteFormat(float64(n), false) }
|
|
func ByteFormat64Short(n int64) string { return byteFormat(float64(n), true) }
|
|
|
|
func byteFormat(n float64, short bool) string {
|
|
i := len(units) - 1
|
|
|
|
for i > 0 {
|
|
if n >= units[i] {
|
|
n /= units[i]
|
|
break
|
|
}
|
|
i--
|
|
}
|
|
|
|
if short {
|
|
return unpadFloat(n, 0) + labels[i][0]
|
|
}
|
|
return unpadFloat(n, 2) + labels[i][1]
|
|
}
|
|
|
|
func unpadFloat(f float64, maxp int) string {
|
|
return strconv.FormatFloat(f, 'f', getPrecision(f, maxp), 64)
|
|
}
|
|
|
|
func getPrecision(f float64, maxp int) int {
|
|
frac := int((f - float64(int(f))) * 100)
|
|
if frac == 0 || maxp == 0 {
|
|
return 0
|
|
}
|
|
if frac%10 == 0 || maxp < 2 {
|
|
return 1
|
|
}
|
|
return maxp
|
|
}
|