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
|
/// The Units system used for sizes
#[derive(Debug, Clone, Copy, PartialEq)]
pub enum Units {
Si, // Units according to the SI system, based on multiples of 1000
Binary, // Old binary based units, based on multiples of 1024
Bytes, // Just the raw byte counts, with commas separating thousands
}
impl Default for Units {
fn default() -> Self {
Self::Si
}
}
static PREFIXES: &[char] = &['K', 'M', 'G', 'T', 'P'];
impl Units {
pub fn fmt(self, size: u64) -> String {
match self {
Self::Si => file_size::fit_4(size),
Self::Binary => {
if size < 10_000 {
size.to_string()
} else {
let size = size as f64;
let i: u32 = (size.ln() / 1024f64.ln()) as u32;
let idx = i as usize - 1;
if idx >= PREFIXES.len() {
"huge".to_string()
} else {
let v = size / (1024u64.pow(i) as f64);
if v >= 10f64 {
format!("{:.0}{}i", v.round(), PREFIXES[idx])
} else {
format!("{:.1}{}i", v, PREFIXES[idx])
}
}
}
}
Self::Bytes => {
let mut rev: Vec<char> = Vec::new();
for (i, c) in size.to_string().chars().rev().enumerate() {
if i>0 && i%3==0 {
rev.push(',');
}
rev.push(c);
}
rev.drain(..).rev().collect()
}
}
}
}
#[test]
fn test_fmt_binary() {
fn check(v: u64, s: &str) {
assert_eq!(&Units::Binary.fmt(v), s);
}
check(0, "0");
check(1, "1");
check(456, "456");
check(1456, "1456");
check(9_999, "9999");
check(10_000, "9.8Ki");
check(12_345, "12Ki");
check(123_456, "121Ki");
check(1_000_000_000, "954Mi");
check(1_073_741_824, "1.0Gi");
check(1_234_567_890, "1.1Gi");
}
#[test]
fn test_fmt_bytes() {
fn check(v: u64, s: &str) {
assert_eq!(&Units::Bytes.fmt(v), s);
}
check(0, "0");
check(1, "1");
check(456, "456");
check(1456, "1,456");
check(9_999, "9,999");
check(10_000, "10,000");
check(12_345, "12,345");
check(123_456, "123,456");
check(1_234_567, "1,234,567");
check(1_000_000_000, "1,000,000,000");
check(1_234_567_890, "1,234,567,890");
}
|
