src/common/fmt.rs

   1 /*
   2  * Copyright (c) 2017 Tilman Sauerbeck (tilman at code-monkey de)
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining
   5  * a copy of this software and associated documentation files (the
   6  * "Software"), to deal in the Software without restriction, including
   7  * without limitation the rights to use, copy, modify, merge, publish,
   8  * distribute, sublicense, and/or sell copies of the Software, and to
   9  * permit persons to whom the Software is furnished to do so, subject to
  10  * the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be
  13  * included in all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  19  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  20  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  21  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 pub fn fmt_u32(s: &mut [u8], n: u32) -> usize {
  25     let mut len = 1usize;
  26     let mut tmp = n;
  27
  28     // Count the number of bytes needed.
  29     while tmp > 9 {
  30         tmp /= 10;
  31         len += 1;
  32     }
  33
  34     if len <= s.len() {
  35         let mut tmp = n;
  36         let mut offset = len;
  37         let mut len2 = len + 1;
  38
  39         while len2 > 1 {
  40             len2 -= 1;
  41
  42             offset -= 1;
  43             s[offset] = b'0' + (tmp % 10) as u8;
  44
  45             tmp /= 10;
  46         }
  47     }
  48
  49     len
  50 }
  51
  52 pub fn fmt_u32_pad(s: &mut [u8], n: u32, pad: usize, pad_char: u8) -> usize {
  53     let mut len = 1usize;
  54     let mut tmp = n;
  55
  56     // Count the number of bytes needed.
  57     while tmp > 9 {
  58         tmp /= 10;
  59         len += 1;
  60     }
  61
  62     if s.len() == 0 {
  63         return if len > pad {
  64             len
  65         } else {
  66             pad
  67         };
  68     }
  69
  70     let mut offset = 0;
  71
  72     while len < pad {
  73         s[offset] = pad_char;
  74         offset += 1;
  75         len += 1;
  76     }
  77
  78     fmt_u32(&mut s[offset..], n);
  79
  80     len
  81 }
  82
  83 fn to_hex(c: u8) -> u8 {
  84     if c >= 10 {
  85         c - 10 + b'a'
  86     } else {
  87         c + b'0'
  88     }
  89 }
  90
  91 pub fn fmt_x32(s: &mut [u8], n: u32) -> usize {
  92     let len = 8;
  93     let mut tmp = n;
  94
  95     if s.len() >= len {
  96         for i in (0..8).rev() {
  97             s[i] = to_hex((tmp & 0xf) as u8);
  98             tmp >>= 4;
  99         }
 100     }
 101
 102     len
 103 }
 104
 105 pub fn fmt_human(s: &mut [u8], _u: u32) -> usize {
 106     let suffix;
 107     let mut u = _u;
 108
 109     if u < (1 << 10) {
 110         suffix = None;
 111     } else if u < (1 << 20) {
 112         u >>= 10;
 113         suffix = Some(b'K');
 114     } else {
 115         u >>= 20;
 116         suffix = Some(b'M');
 117     }
 118
 119     let mut len = fmt_u32(s, u);
 120
 121     if let Some(suffix_c) = suffix {
 122         if len < s.len() {
 123             s[len] = suffix_c;
 124         }
 125
 126         len += 1;
 127     }
 128
 129     len
 130 }