All-integer version (but it does use an unsigned long long) which fixes
[oweals/busybox.git] / libbb / human_readable.c
1 /*
2  * June 30, 2001                 Manuel Novoa III
3  *
4  * All-integer version (hey, not everyone has floating point) of
5  * make_human_readable_str, modified from similar code I had written
6  * for busybox several months ago.
7  *
8  * Notes:
9  *   1) I'm using an unsigned long long to hold the product size * block_size,
10  *      as df (which calls this routine) could request a representation of a
11  *      partition size in bytes > max of unsigned long.  If long longs aren't
12  *      available, it would be possible to do what's needed using polynomial
13  *      representations (say, powers of 1024) and manipulating coefficients.
14  *      The base ten "bytes" output could be handled similarly.
15  *
16  *   2) The output of "ls -sh" can be misaligned because this routine always
17  *      outputs a decimal point and a tenths digit when display_unit != 0.
18  *      Hence, it isn't uncommon for the returned string to have a length
19  *      of 5 or 6 instead of <= 4 (as assumed).
20  *
21  *      It might be nice to add a flag to indicate no decimal digits in
22  *      that case.  This could be either an additional parameter, or a
23  *      special value of display_unit.  Such a flag would also be nice for du.
24  *
25  *      Some code to omit the decimal point and tenths digit is sketched out
26  *      and "#if 0"'d below.
27  */
28
29 #include <stdio.h>
30 #include "libbb.h"
31
32 const char *make_human_readable_str(unsigned long size, 
33                                                                         unsigned long block_size,
34                                                                         unsigned long display_unit)
35 {
36         /* The code will adjust for additional (appended) units. */
37         static const char zero_and_units[] = { '0', 0, 'k', 'M', 'G', 'T' };
38         static const char fmt[] = "%Lu";
39         static const char fmt_tenths[] = "%Lu.%d%c";
40
41         static char str[21];            /* Sufficient for 64 bit unsigned integers. */
42         
43         unsigned long long val;
44         int frac;
45         const char *u;
46         const char *f;
47
48         u = zero_and_units;
49         f = fmt;
50         frac = 0;
51
52         val = ((unsigned long long) size) * block_size;
53         if (val == 0) {
54                 return u;
55         }
56
57         if (display_unit) {
58                 val += display_unit/2;  /* Deal with rounding. */
59                 val /= display_unit;    /* Don't combine with the line above!!! */
60         } else {
61                 ++u;
62                 while ((val >= KILOBYTE)
63                            && (u < zero_and_units + sizeof(zero_and_units) - 1)) {
64                         f = fmt_tenths;
65                         ++u;
66                         frac = ((((int)(val % KILOBYTE)) * 10) + (KILOBYTE/2)) / KILOBYTE;
67                         val /= KILOBYTE;
68                 }
69                 if (frac >= 10) {               /* We need to round up here. */
70                         ++val;
71                         frac = 0;
72                 }
73 #if 0
74                 /* Sample code to omit decimal point and tenths digit. */
75                 if ( /* no_tenths */ 1 ) {
76                         if ( frac >= 5 ) {
77                                 ++val;
78                         }
79                         f = "%Lu%*c" /* fmt_no_tenths */ ;
80                         frac = 1;
81                 }
82 #endif
83         }
84
85         /* If f==fmt then 'frac' and 'u' are ignored. */
86         snprintf(str, sizeof(str), f, val, frac, *u);
87
88         return str;
89 }