stdio-common/printf_size.c

   1 /* Print size value using units for orders of magnitude.
   2    Copyright (C) 1997 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@cygnus.com>, 1997.
   5    Based on a proposal by Larry McVoy <lm@sgi.com>.
   6
   7    The GNU C Library is free software; you can redistribute it and/or
   8    modify it under the terms of the GNU Library General Public License as
   9    published by the Free Software Foundation; either version 2 of the
  10    License, or (at your option) any later version.
  11
  12    The GNU C Library is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15    Library General Public License for more details.
  16
  17    You should have received a copy of the GNU Library General Public
  18    License along with the GNU C Library; see the file COPYING.LIB.  If not,
  19    write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  20    Boston, MA 02111-1307, USA.  */
  21
  22 #include <ctype.h>
  23 #include <ieee754.h>
  24 #include <math.h>
  25 #include <printf.h>
  26 #ifdef USE_IN_LIBIO
  27 #  include <libioP.h>
  28 #else
  29 #  include <stdio.h>
  30 #endif
  31
  32
  33 /* This defines make it possible to use the same code for GNU C library and
  34    the GNU I/O library.  */
  35 #ifdef USE_IN_LIBIO
  36 #  define PUT(f, s, n) _IO_sputn (f, s, n)
  37 #  define PAD(f, c, n) _IO_padn (f, c, n)
  38 /* We use this file GNU C library and GNU I/O library.  So make
  39    names equal.  */
  40 #  undef putc
  41 #  define putc(c, f) _IO_putc_unlocked (c, f)
  42 #  define size_t     _IO_size_t
  43 #  define FILE       _IO_FILE
  44 #else   /* ! USE_IN_LIBIO */
  45 #  define PUT(f, s, n) fwrite (s, 1, n, f)
  46 #  define PAD(f, c, n) __printf_pad (f, c, n)
  47 ssize_t __printf_pad __P ((FILE *, char pad, int n)); /* In vfprintf.c.  */
  48 #endif  /* USE_IN_LIBIO */
  49 \f
  50 /* Macros for doing the actual output.  */
  51
  52 #define outchar(ch)                                                           \
  53   do                                                                          \
  54     {                                                                         \
  55       register const int outc = (ch);                                         \
  56       if (putc (outc, fp) == EOF)                                             \
  57         return -1;                                                            \
  58       ++done;                                                                 \
  59     } while (0)
  60
  61 #define PRINT(ptr, len)                                                       \
  62   do                                                                          \
  63     {                                                                         \
  64       register size_t outlen = (len);                                         \
  65       if (len > 20)                                                           \
  66         {                                                                     \
  67           if (PUT (fp, ptr, outlen) != outlen)                                \
  68             return -1;                                                        \
  69           ptr += outlen;                                                      \
  70           done += outlen;                                                     \
  71         }                                                                     \
  72       else                                                                    \
  73         {                                                                     \
  74           while (outlen-- > 0)                                                \
  75             outchar (*ptr++);                                                 \
  76         }                                                                     \
  77     } while (0)
  78
  79 #define PADN(ch, len)                                                         \
  80   do                                                                          \
  81     {                                                                         \
  82       if (PAD (fp, ch, len) != len)                                           \
  83         return -1;                                                            \
  84       done += len;                                                            \
  85     }                                                                         \
  86   while (0)
  87 \f
  88 /* Prototype for helper functions.  */
  89 extern int __printf_fp (FILE *fp, const struct printf_info *info,
  90                         const void *const *args);
  91
  92 \f
  93 int
  94 printf_size (FILE *fp, const struct printf_info *info, const void *const *args)
  95 {
  96   /* Units for the both formats.  */
  97 #define BINARY_UNITS    " kmgtpezy"
  98 #define DECIMAL_UNITS   " KMGTPEZY"
  99   static const char units[2][sizeof (BINARY_UNITS)] =
 100   {
 101     BINARY_UNITS,       /* For binary format.  */
 102     DECIMAL_UNITS       /* For decimal format.  */
 103   };
 104   const char *tag = units[isupper (info->spec) != 0];
 105   int divisor = isupper (info->spec) ? 1000 : 1024;
 106
 107   /* The floating-point value to output.  */
 108   union
 109     {
 110       union ieee754_double dbl;
 111       union ieee854_long_double ldbl;
 112     }
 113   fpnum;
 114   const void *ptr = &fpnum;
 115
 116   int negative = 0;
 117
 118   /* "NaN" or "Inf" for the special cases.  */
 119   const char *special = NULL;
 120
 121   struct printf_info fp_info;
 122   int done = 0;
 123
 124
 125   /* Fetch the argument value.  */
 126   if (info->is_long_double && sizeof (long double) > sizeof (double))
 127     {
 128       fpnum.ldbl.d = *(const long double *) args[0];
 129
 130       /* Check for special values: not a number or infinity.  */
 131       if (__isnanl (fpnum.ldbl.d))
 132         {
 133           special = "nan";
 134           negative = 0;
 135         }
 136       else if (__isinfl (fpnum.ldbl.d))
 137         {
 138           special = "inf";
 139
 140           negative = fpnum.ldbl.d < 0;
 141         }
 142       else
 143         while (fpnum.ldbl.d >= divisor && tag[1] != '\0')
 144           {
 145             fpnum.ldbl.d /= divisor;
 146             ++tag;
 147           }
 148     }
 149   else
 150     {
 151       fpnum.dbl.d = *(const double *) args[0];
 152
 153       /* Check for special values: not a number or infinity.  */
 154       if (__isnan (fpnum.dbl.d))
 155         {
 156           special = "nan";
 157           negative = 0;
 158         }
 159       else if (__isinf (fpnum.dbl.d))
 160         {
 161           special = "inf";
 162
 163           negative = fpnum.dbl.d < 0;
 164         }
 165       else
 166         while (fpnum.dbl.d >= divisor && tag[1] != '\0')
 167           {
 168             fpnum.dbl.d /= divisor;
 169             ++tag;
 170           }
 171     }
 172
 173   if (special)
 174     {
 175       int width = info->prec > info->width ? info->prec : info->width;
 176
 177       if (negative || info->showsign || info->space)
 178         --width;
 179       width -= 3;
 180
 181       if (!info->left && width > 0)
 182         PADN (' ', width);
 183
 184       if (negative)
 185         outchar ('-');
 186       else if (info->showsign)
 187         outchar ('+');
 188       else if (info->space)
 189         outchar (' ');
 190
 191       PRINT (special, 3);
 192
 193       if (info->left && width > 0)
 194         PADN (' ', width);
 195
 196       return done;
 197     }
 198
 199   /* Prepare to print the number.  We want to use `__printf_fp' so we
 200      have to prepare a `printf_info' structure.  */
 201   fp_info.spec = 'f';
 202   fp_info.prec = info->prec < 0 ? 3 : info->prec;
 203   fp_info.is_long_double = info->is_long_double;
 204   fp_info.is_short = info->is_short;
 205   fp_info.is_long = info->is_long;
 206   fp_info.alt = info->alt;
 207   fp_info.space = info->space;
 208   fp_info.left = info->left;
 209   fp_info.showsign = info->showsign;
 210   fp_info.group = info->group;
 211   fp_info.extra = info->extra;
 212   fp_info.pad = info->pad;
 213
 214   if (fp_info.left && fp_info.pad == L' ')
 215     {
 216       /* We must do the padding ourself since the unit character must
 217          be placed before the padding spaces.  */
 218       fp_info.width = 0;
 219
 220       done = __printf_fp (fp, &fp_info, &ptr);
 221       if (done > 0)
 222         {
 223           outchar (*tag);
 224           if (info->width > done)
 225             PADN (' ', info->width - done);
 226         }
 227     }
 228   else
 229     {
 230       /* We can let __printf_fp do all the printing and just add our
 231          unit character afterwards.  */
 232       fp_info.width = info->width - 1;
 233
 234       done = __printf_fp (fp, &fp_info, &ptr);
 235       if (done > 0)
 236         outchar (*tag);
 237     }
 238
 239   return done;
 240 }
 241 \f
 242 /* This is the function used by `vfprintf' to determine number and
 243    type of the arguments.  */
 244 int
 245 printf_size_info (const struct printf_info *info, size_t n, int *argtypes)
 246 {
 247   /* We need only one double or long double argument.  */
 248   if (n >= 1)
 249     argtypes[0] = PA_DOUBLE | (info->is_long_double ? PA_FLAG_LONG_DOUBLE : 0);
 250
 251   return 1;
 252 }