libbb/human_readable.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * June 30, 2001                 Manuel Novoa III
   4  *
   5  * All-integer version (hey, not everyone has floating point) of
   6  * make_human_readable_str, modified from similar code I had written
   7  * for busybox several months ago.
   8  *
   9  * Notes:
  10  *   1) I'm using an unsigned long long to hold the product size * block_size,
  11  *      as df (which calls this routine) could request a representation of a
  12  *      partition size in bytes > max of unsigned long.  If long longs aren't
  13  *      available, it would be possible to do what's needed using polynomial
  14  *      representations (say, powers of 1024) and manipulating coefficients.
  15  *      The base ten "bytes" output could be handled similarly.
  16  *
  17  *   2) This routine always outputs a decimal point and a tenths digit when
  18  *      display_unit != 0.  Hence, it isn't uncommon for the returned string
  19  *      to have a length of 5 or 6.
  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  * Licensed under GPLv2, see file LICENSE in this tarball for details.
  29  */
  30
  31 #include "libbb.h"
  32
  33 const char* FAST_FUNC make_human_readable_str(unsigned long long size,
  34         unsigned long block_size, unsigned long display_unit)
  35 {
  36         /* The code will adjust for additional (appended) units */
  37         static const char unit_chars[] ALIGN1 = {
  38                 '\0', 'K', 'M', 'G', 'T', 'P', 'E'
  39         };
  40         static const char fmt[] ALIGN1 = "%llu";
  41         static const char fmt_tenths[] ALIGN1 = "%llu.%d%c";
  42
  43         static char str[21] ALIGN1;  /* Sufficient for 64 bit unsigned integers */
  44
  45         unsigned long long val;
  46         int frac;
  47         const char *u;
  48         const char *f;
  49         smallint no_tenths;
  50
  51         if (size == 0)
  52                 return "0";
  53
  54         /* If block_size is 0 then do not print tenths */
  55         no_tenths = 0;
  56         if (block_size == 0) {
  57                 no_tenths = 1;
  58                 block_size = 1;
  59         }
  60
  61         u = unit_chars;
  62         val = size * block_size;
  63         f = fmt;
  64         frac = 0;
  65
  66         if (display_unit) {
  67                 val += display_unit/2;  /* Deal with rounding */
  68                 val /= display_unit;    /* Don't combine with the line above!!! */
  69                 /* will just print it as ulonglong (below) */
  70         } else {
  71                 while ((val >= 1024)
  72                  && (u < unit_chars + sizeof(unit_chars) - 1)
  73                 ) {
  74                         f = fmt_tenths;
  75                         u++;
  76                         frac = (((int)(val % 1024)) * 10 + 1024/2) / 1024;
  77                         val /= 1024;
  78                 }
  79                 if (frac >= 10) {               /* We need to round up here. */
  80                         ++val;
  81                         frac = 0;
  82                 }
  83 #if 1
  84                 /* Sample code to omit decimal point and tenths digit. */
  85                 if (no_tenths) {
  86                         if (frac >= 5) {
  87                                 ++val;
  88                         }
  89                         f = "%llu%*c" /* fmt_no_tenths */;
  90                         frac = 1;
  91                 }
  92 #endif
  93         }
  94
  95         /* If f==fmt then 'frac' and 'u' are ignored. */
  96         snprintf(str, sizeof(str), f, val, frac, *u);
  97
  98         return str;
  99 }