exa/src/format.rs

static METRIC_PREFIXES: &'static [&'static str] = &[
    "", "K", "M", "G", "T", "P", "E", "Z", "Y"
];

static IEC_PREFIXES: &'static [&'static str] = &[
    "", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi", "Yi"
];

fn format_bytes<'s>(mut amount: f64, kilo: f64, prefixes: &[&'s str]) -> (String, &'s str) {
    let mut prefix = 0;
    while amount >= kilo {
        amount /= kilo;
        prefix += 1;
    }

    if amount < 10.0 && prefix != 0 {
        (format!("{:.1}", amount), prefixes[prefix])
    }
    else {
        (format!("{:.0}", amount), prefixes[prefix])
    }
}

#[allow(non_snake_case)]
pub fn format_IEC_bytes<'s>(amount: u64) -> (String, &'s str) {
    format_bytes(amount as f64, 1024.0, IEC_PREFIXES)
}

pub fn format_metric_bytes<'s>(amount: u64) -> (String, &'s str) {
    format_bytes(amount as f64, 1000.0, METRIC_PREFIXES)
}

#[test]
fn test_0() {
    let kk = format_metric_bytes(0);
    assert!(kk == ("0".to_string(), ""));
}

#[test]
fn test_999() {
    let kk = format_metric_bytes(999);
    assert!(kk == ("999".to_string(), ""));
}

#[test]
fn test_1000() {
    let kk = format_metric_bytes(1000);
    assert!(kk == ("1.0".to_string(), "K"));
}

#[test]
fn test_1030() {
    let kk = format_metric_bytes(1030);
    assert!(kk == ("1.0".to_string(), "K"));
}

#[test]
fn test_1100() {
    let kk = format_metric_bytes(1100);
    assert!(kk == ("1.1".to_string(), "K"));
}

#[test]
fn test_1111() {
    let kk = format_metric_bytes(1111);
    assert!(kk == ("1.1".to_string(), "K"));
}

#[test]
fn test_104857() {
    let kk = format_IEC_bytes(126456);
    assert!(kk == ("123".to_string(), "Ki"));
}

#[test]
fn test_1048576() {
    let kk = format_IEC_bytes(1048576);
    assert!(kk == ("1.0".to_string(), "Mi"));
}