Fio 2.1.14
[fio.git] / lib / num2str.c
1 #include <stdlib.h>
2 #include <stdio.h>
3 #include <string.h>
4
5 #include "../fio.h"
6
7 #define ARRAY_LENGTH(arr)       sizeof(arr) / sizeof((arr)[0])
8
9 /*
10  * Cheesy number->string conversion, complete with carry rounding error.
11  */
12 char *num2str(unsigned long num, int maxlen, int base, int pow2, int unit_base)
13 {
14         const char *postfix[] = { "", "K", "M", "G", "P", "E" };
15         const char *byte_postfix[] = { "", "B", "bit" };
16         const unsigned int thousand[] = { 1000, 1024 };
17         unsigned int modulo, decimals;
18         int byte_post_index = 0, post_index, carry = 0;
19         char tmp[32];
20         char *buf;
21
22         buf = malloc(128);
23
24         for (post_index = 0; base > 1; post_index++)
25                 base /= thousand[!!pow2];
26
27         switch (unit_base) {
28         case 1:
29                 byte_post_index = 2;
30                 num *= 8;
31                 break;
32         case 8:
33                 byte_post_index = 1;
34                 break;
35         }
36
37         modulo = -1U;
38         while (post_index < sizeof(postfix)) {
39                 sprintf(tmp, "%lu", num);
40                 if (strlen(tmp) <= maxlen)
41                         break;
42
43                 modulo = num % thousand[!!pow2];
44                 num /= thousand[!!pow2];
45                 carry = modulo >= thousand[!!pow2] / 2;
46                 post_index++;
47         }
48
49         if (modulo == -1U) {
50 done:
51                 if (post_index >= ARRAY_LENGTH(postfix))
52                         post_index = 0;
53
54                 sprintf(buf, "%lu%s%s", num, postfix[post_index],
55                         byte_postfix[byte_post_index]);
56                 return buf;
57         }
58
59         sprintf(tmp, "%lu", num);
60         decimals = maxlen - strlen(tmp);
61         if (decimals <= 1) {
62                 if (carry)
63                         num++;
64                 goto done;
65         }
66
67         do {
68                 sprintf(tmp, "%u", modulo);
69                 if (strlen(tmp) <= decimals - 1)
70                         break;
71
72                 modulo = (modulo + 9) / 10;
73         } while (1);
74
75         sprintf(buf, "%lu.%u%s%s", num, modulo, postfix[post_index],
76                 byte_postfix[byte_post_index]);
77         return buf;
78 }