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