1/* Print size value using units for orders of magnitude.
2 Copyright (C) 1997-2017 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, see
19 <http://www.gnu.org/licenses/>. */
20
21#include <ctype.h>
22#include <ieee754.h>
23#include <math.h>
24#include <printf.h>
25#include <libioP.h>
26
27
28/* This defines make it possible to use the same code for GNU C library and
29 the GNU I/O library. */
30#define PUT(f, s, n) _IO_sputn (f, s, n)
31#define PAD(f, c, n) (wide ? _IO_wpadn (f, c, n) : _IO_padn (f, c, n))
32/* We use this file GNU C library and GNU I/O library. So make
33 names equal. */
34#undef putc
35#define putc(c, f) (wide \
36 ? (int)_IO_putwc_unlocked (c, f) : _IO_putc_unlocked (c, f))
37#define size_t _IO_size_t
38#define FILE _IO_FILE
39
40/* Macros for doing the actual output. */
41
42#define outchar(ch) \
43 do \
44 { \
45 const int outc = (ch); \
46 if (putc (outc, fp) == EOF) \
47 return -1; \
48 ++done; \
49 } while (0)
50
51#define PRINT(ptr, wptr, len) \
52 do \
53 { \
54 size_t outlen = (len); \
55 if (len > 20) \
56 { \
57 if (PUT (fp, wide ? (const char *) wptr : ptr, outlen) != outlen) \
58 return -1; \
59 ptr += outlen; \
60 done += outlen; \
61 } \
62 else \
63 { \
64 if (wide) \
65 while (outlen-- > 0) \
66 outchar (*wptr++); \
67 else \
68 while (outlen-- > 0) \
69 outchar (*ptr++); \
70 } \
71 } while (0)
72
73#define PADN(ch, len) \
74 do \
75 { \
76 if (PAD (fp, ch, len) != len) \
77 return -1; \
78 done += len; \
79 } \
80 while (0)
81
82/* Prototype for helper functions. */
83extern int __printf_fp (FILE *fp, const struct printf_info *info,
84 const void *const *args);
85
86
87int
88__printf_size (FILE *fp, const struct printf_info *info,
89 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 long double ldbl;
107 }
108 fpnum;
109 const void *ptr = &fpnum;
110
111 int is_neg = 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 /* Fetch the argument value. */
122#ifndef __NO_LONG_DOUBLE_MATH
123 if (info->is_long_double && sizeof (long double) > sizeof (double))
124 {
125 fpnum.ldbl = *(const long double *) args[0];
126
127 /* Check for special values: not a number or infinity. */
128 if (isnan (fpnum.ldbl))
129 {
130 special = "nan";
131 wspecial = L"nan";
132 // is_neg = 0; Already zero
133 }
134 else if (isinf (fpnum.ldbl))
135 {
136 is_neg = signbit (fpnum.ldbl);
137 special = "inf";
138 wspecial = L"inf";
139 }
140 else
141 while (fpnum.ldbl >= divisor && tag[1] != '\0')
142 {
143 fpnum.ldbl /= divisor;
144 ++tag;
145 }
146 }
147 else
148#endif /* no long double */
149 {
150 fpnum.dbl.d = *(const double *) args[0];
151
152 /* Check for special values: not a number or infinity. */
153 if (isnan (fpnum.dbl.d))
154 {
155 special = "nan";
156 wspecial = L"nan";
157 // is_neg = 0; Already zero
158 }
159 else if (isinf (fpnum.dbl.d))
160 {
161 is_neg = signbit (fpnum.dbl.d);
162 special = "inf";
163 wspecial = L"inf";
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 (is_neg || info->showsign || info->space)
178 --width;
179 width -= 3;
180
181 if (!info->left && width > 0)
182 PADN (' ', width);
183
184 if (is_neg)
185 outchar ('-');
186 else if (info->showsign)
187 outchar ('+');
188 else if (info->space)
189 outchar (' ');
190
191 PRINT (special, wspecial, 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 = *info;
202 fp_info.spec = 'f';
203 fp_info.prec = info->prec < 0 ? 3 : info->prec;
204 fp_info.wide = wide;
205
206 if (fp_info.left && fp_info.pad == L' ')
207 {
208 /* We must do the padding ourself since the unit character must
209 be placed before the padding spaces. */
210 fp_info.width = 0;
211
212 done = __printf_fp (fp, &fp_info, &ptr);
213 if (done > 0)
214 {
215 outchar (*tag);
216 if (info->width > done)
217 PADN (' ', info->width - done);
218 }
219 }
220 else
221 {
222 /* We can let __printf_fp do all the printing and just add our
223 unit character afterwards. */
224 fp_info.width = info->width - 1;
225
226 done = __printf_fp (fp, &fp_info, &ptr);
227 if (done > 0)
228 outchar (*tag);
229 }
230
231 return done;
232}
233ldbl_strong_alias (__printf_size, printf_size);
234
235/* This is the function used by `vfprintf' to determine number and
236 type of the arguments. */
237int
238printf_size_info (const struct printf_info *info, size_t n, int *argtypes)
239{
240 /* We need only one double or long double argument. */
241 if (n >= 1)
242 argtypes[0] = PA_DOUBLE | (info->is_long_double ? PA_FLAG_LONG_DOUBLE : 0);
243
244 return 1;
245}
246