mpir/printf/doprnt.c
2009-02-12 11:23:26 +00:00

624 lines
20 KiB
C

/* __gmp_doprnt -- printf style formatted output.
THE FUNCTIONS IN THIS FILE ARE FOR INTERNAL USE ONLY. THEY'RE ALMOST
CERTAIN TO BE SUBJECT TO INCOMPATIBLE CHANGES OR DISAPPEAR COMPLETELY IN
FUTURE GNU MP RELEASES.
Copyright 2001, 2002, 2003 Free Software Foundation, Inc.
This file is part of the GNU MP Library.
The GNU MP Library is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or (at your
option) any later version.
The GNU MP Library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
License for more details.
You should have received a copy of the GNU Lesser General Public License
along with the GNU MP Library; see the file COPYING.LIB. If not, write to
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
MA 02110-1301, USA. */
#define _GNU_SOURCE /* for DECIMAL_POINT in glibc langinfo.h */
#include "config.h"
#if HAVE_STDARG
#include <stdarg.h>
#else
#include <varargs.h>
#endif
#include <ctype.h> /* for isdigit */
#include <stddef.h> /* for ptrdiff_t */
#include <string.h>
#include <stdio.h> /* for NULL */
#include <stdlib.h>
#if HAVE_INTTYPES_H
# include <inttypes.h> /* for intmax_t */
#else
# if HAVE_STDINT_H
# include <stdint.h>
# endif
#endif
#if HAVE_LANGINFO_H
#include <langinfo.h> /* for nl_langinfo */
#endif
#if HAVE_LOCALE_H
#include <locale.h> /* for localeconv */
#endif
#if HAVE_SYS_TYPES_H
#include <sys/types.h> /* for quad_t */
#endif
#include "mpir.h"
#include "gmp-impl.h"
/* change this to "#define TRACE(x) x" for diagnostics */
#define TRACE(x)
/* Should be portable, but in any case this is only used under some ASSERTs. */
#define va_equal(x, y) \
(memcmp (&(x), &(y), sizeof(va_list)) == 0)
/* printf is convenient because it allows various types to be printed in one
fairly compact call, so having gmp_printf support the standard types as
well as the gmp ones is important. This ends up meaning all the standard
parsing must be duplicated, to get a new routine recognising the gmp
extras.
With the currently favoured handling of mpz etc as Z, Q and F type
markers, it's not possible to use glibc register_printf_function since
that only accepts new conversion characters, not new types. If Z was a
conversion there'd be no way to specify hex, decimal or octal, or
similarly with F no way to specify fixed point or scientific format.
It seems wisest to pass conversions %f, %e and %g of float, double and
long double over to the standard printf. It'd be hard to be sure of
getting the right handling for NaNs, rounding, etc. Integer conversions
%d etc and string conversions %s on the other hand could be easily enough
handled within gmp_doprnt, but if floats are going to libc then it's just
as easy to send all non-gmp types there.
"Z" was a type marker for size_t in old glibc, but there seems no need to
provide access to that now "z" is standard.
In GMP 4.1.1 we documented "ll" and "L" as being equivalent, but in C99
in fact "ll" is just for long long and "L" just for long double.
Apparentely GLIBC allows "L" for long long though. This doesn't affect
us as such, since both are passed through to the C library. To be
consistent with what we said before, the two are treated equivalently
here, and it's left to the C library to do what it thinks with them.
Possibilities:
"b" might be nice for binary output, and could even be supported for the
standard C types too if desired.
POSIX style "%n$" parameter numbering would be possible, but would need
to be handled completely within gmp_doprnt, since the numbering will be
all different once the format string it cut into pieces.
Some options for mpq formatting would be good. Perhaps a non-zero
precision field could give a width for the denominator and mean always
put a "/". A form "n+p/q" might interesting too, though perhaps that's
better left to applications.
Right now there's no way for an application to know whether types like
intmax_t are supported here. If configure is doing its job and the same
compiler is used for gmp as for the application then there shouldn't be
any problem, but perhaps mpir.h should have some preprocessor symbols to
say what libgmp can do. */
/* If a gmp format is the very first thing or there are two gmp formats with
nothing in between then we'll reach here with this_fmt == last_fmt and we
can do nothing in that case.
last_ap is always replaced after a FLUSH, so it doesn't matter if va_list
is a call-by-reference and the funs->format routine modifies it. */
#define FLUSH() \
do { \
if (this_fmt == last_fmt) \
{ \
TRACE (printf ("nothing to flush\n")); \
ASSERT (va_equal (this_ap, last_ap)); \
} \
else \
{ \
ASSERT (*this_fmt == '%'); \
*this_fmt = '\0'; \
TRACE (printf ("flush \"%s\"\n", last_fmt)); \
DOPRNT_FORMAT (last_fmt, last_ap); \
} \
} while (0)
/* Parse up the given format string and do the appropriate output using the
given "funs" routines. The data parameter is passed through to those
routines. */
int
__gmp_doprnt (const struct doprnt_funs_t *funs, void *data,
const char *orig_fmt, va_list orig_ap)
{
va_list ap, this_ap, last_ap;
size_t alloc_fmt_size;
char *fmt, *alloc_fmt, *last_fmt, *this_fmt, *gmp_str;
int retval = 0;
int type, fchar, *value, seen_precision;
struct doprnt_params_t param;
TRACE (printf ("gmp_doprnt \"%s\"\n", orig_fmt));
/* Don't modify orig_ap, if va_list is actually an array and hence call by
reference. It could be argued that it'd be more efficient to leave the
caller to make a copy if it cared, but doing so here is going to be a
very small part of the total work, and we may as well keep applications
out of trouble. */
va_copy (ap, orig_ap);
/* The format string is chopped up into pieces to be passed to
funs->format. Unfortunately that means it has to be copied so each
piece can be null-terminated. We're not going to be very fast here, so
use __gmp_allocate_func rather than TMP_ALLOC, to avoid overflowing the
stack if a long output string is given. */
alloc_fmt_size = strlen (orig_fmt) + 1;
#if _LONG_LONG_LIMB
/* for a long long limb we change %Mx to %llx, so could need an extra 1
char for every 3 existing */
alloc_fmt_size += alloc_fmt_size / 3;
#endif
alloc_fmt = __GMP_ALLOCATE_FUNC_TYPE (alloc_fmt_size, char);
fmt = alloc_fmt;
strcpy (fmt, orig_fmt);
/* last_fmt and last_ap are just after the last output, and hence where
the next output will begin, when that's done */
last_fmt = fmt;
va_copy (last_ap, ap);
for (;;)
{
TRACE (printf ("next: \"%s\"\n", fmt));
fmt = strchr (fmt, '%');
if (fmt == NULL)
break;
/* this_fmt and this_ap are the current '%' sequence being considered */
this_fmt = fmt;
va_copy (this_ap, ap);
fmt++; /* skip the '%' */
TRACE (printf ("considering\n");
printf (" last: \"%s\"\n", last_fmt);
printf (" this: \"%s\"\n", this_fmt));
type = '\0';
value = &param.width;
param.base = 10;
param.conv = 0;
param.expfmt = "e%c%02d";
param.exptimes4 = 0;
param.fill = ' ';
param.justify = DOPRNT_JUSTIFY_RIGHT;
param.prec = 6;
param.showbase = DOPRNT_SHOWBASE_NO;
param.showpoint = 0;
param.showtrailing = 1;
param.sign = '\0';
param.width = 0;
seen_precision = 0;
/* This loop parses a single % sequence. "break" from the switch
means continue with this %, "goto next" means the conversion
character has been seen and a new % should be sought. */
for (;;)
{
fchar = *fmt++;
if (fchar == '\0')
break;
switch (fchar) {
case 'a':
/* %a behaves like %e, but defaults to all significant digits,
and there's no leading zeros on the exponent (which is in
fact bit-based) */
param.base = 16;
param.expfmt = "p%c%d";
goto conv_a;
case 'A':
param.base = -16;
param.expfmt = "P%c%d";
conv_a:
param.conv = DOPRNT_CONV_SCIENTIFIC;
param.exptimes4 = 1;
if (! seen_precision)
param.prec = -1; /* default to all digits */
param.showbase = DOPRNT_SHOWBASE_YES;
param.showtrailing = 1;
goto floating_a;
case 'c':
/* Let's assume wchar_t will be promoted to "int" in the call,
the same as char will be. */
(void) va_arg (ap, int);
goto next;
case 'd':
case 'i':
case 'u':
integer:
TRACE (printf ("integer, base=%d\n", param.base));
if (! seen_precision)
param.prec = -1;
switch (type) {
case 'j':
/* Let's assume uintmax_t is the same size as intmax_t. */
#if HAVE_INTMAX_T
(void) va_arg (ap, intmax_t);
#else
ASSERT_FAIL (intmax_t not available);
#endif
break;
case 'l':
(void) va_arg (ap, long);
break;
case 'L':
#if HAVE_LONG_LONG
(void) va_arg (ap, long long);
#else
ASSERT_FAIL (long long not available);
#endif
break;
case 'N':
{
mp_ptr xp;
mp_size_t xsize, abs_xsize;
mpz_t z;
FLUSH ();
xp = va_arg (ap, mp_ptr);
PTR(z) = xp;
xsize = (int) va_arg (ap, mp_size_t);
abs_xsize = ABS (xsize);
MPN_NORMALIZE (xp, abs_xsize);
SIZ(z) = (xsize >= 0 ? abs_xsize : -abs_xsize);
ASSERT_CODE (ALLOC(z) = abs_xsize);
gmp_str = mpz_get_str (NULL, param.base, z);
goto gmp_integer;
}
/* break; */
case 'q':
/* quad_t is probably the same as long long, but let's treat
it separately just to be sure. Also let's assume u_quad_t
will be the same size as quad_t. */
#if HAVE_QUAD_T
(void) va_arg (ap, quad_t);
#else
ASSERT_FAIL (quad_t not available);
#endif
break;
case 'Q':
FLUSH ();
gmp_str = mpq_get_str (NULL, param.base, va_arg(ap, mpq_srcptr));
goto gmp_integer;
case 't':
#if HAVE_PTRDIFF_T
(void) va_arg (ap, ptrdiff_t);
#else
ASSERT_FAIL (ptrdiff_t not available);
#endif
break;
case 'z':
(void) va_arg (ap, size_t);
break;
case 'Z':
{
int ret;
FLUSH ();
gmp_str = mpz_get_str (NULL, param.base,
va_arg (ap, mpz_srcptr));
gmp_integer:
ret = __gmp_doprnt_integer (funs, data, &param, gmp_str);
(*__gmp_free_func) (gmp_str, strlen(gmp_str)+1);
DOPRNT_ACCUMULATE (ret);
va_copy (last_ap, ap);
last_fmt = fmt;
}
break;
default:
/* default is an "int", and this includes h=short and hh=char
since they're promoted to int in a function call */
(void) va_arg (ap, int);
break;
}
goto next;
case 'E':
param.base = -10;
param.expfmt = "E%c%02d";
/*FALLTHRU*/
case 'e':
param.conv = DOPRNT_CONV_SCIENTIFIC;
floating:
if (param.showbase == DOPRNT_SHOWBASE_NONZERO)
{
/* # in %e, %f and %g */
param.showpoint = 1;
param.showtrailing = 1;
}
floating_a:
switch (type) {
case 'F':
FLUSH ();
DOPRNT_ACCUMULATE (__gmp_doprnt_mpf (funs, data, &param,
GMP_DECIMAL_POINT,
va_arg (ap, mpf_srcptr)));
va_copy (last_ap, ap);
last_fmt = fmt;
break;
case 'L':
#if HAVE_LONG_DOUBLE
(void) va_arg (ap, long double);
#else
ASSERT_FAIL (long double not available);
#endif
break;
default:
(void) va_arg (ap, double);
break;
}
goto next;
case 'f':
param.conv = DOPRNT_CONV_FIXED;
goto floating;
case 'F': /* mpf_t */
case 'j': /* intmax_t */
case 'L': /* long long */
case 'N': /* mpn */
case 'q': /* quad_t */
case 'Q': /* mpq_t */
case 't': /* ptrdiff_t */
case 'z': /* size_t */
case 'Z': /* mpz_t */
set_type:
type = fchar;
break;
case 'G':
param.base = -10;
param.expfmt = "E%c%02d";
/*FALLTHRU*/
case 'g':
param.conv = DOPRNT_CONV_GENERAL;
param.showtrailing = 0;
goto floating;
case 'h':
if (type != 'h')
goto set_type;
type = 'H'; /* internal code for "hh" */
break;
case 'l':
if (type != 'l')
goto set_type;
type = 'L'; /* "ll" means "L" */
break;
case 'm':
/* glibc strerror(errno), no argument */
goto next;
case 'M': /* mp_limb_t */
/* mung format string to l or ll and let plain printf handle it */
#if _LONG_LONG_LIMB
memmove (fmt+1, fmt, strlen (fmt)+1);
fmt[-1] = 'l';
fmt[0] = 'l';
fmt++;
type = 'L';
#else
fmt[-1] = 'l';
type = 'l';
#endif
break;
case 'n':
{
void *p;
FLUSH ();
p = va_arg (ap, void *);
switch (type) {
case '\0': * (int *) p = retval; break;
case 'F': mpf_set_si ((mpf_ptr) p, (long) retval); break;
case 'H': * (char *) p = retval; break;
case 'h': * (short *) p = retval; break;
#if HAVE_INTMAX_T
case 'j': * (intmax_t *) p = retval; break;
#else
case 'j': ASSERT_FAIL (intmax_t not available); break;
#endif
case 'l': * (long *) p = retval; break;
#if HAVE_QUAD_T && HAVE_LONG_LONG
case 'q':
ASSERT_ALWAYS (sizeof (quad_t) == sizeof (long long));
/*FALLTHRU*/
#else
case 'q': ASSERT_FAIL (quad_t not available); break;
#endif
#if HAVE_LONG_LONG
case 'L': * (long long *) p = retval; break;
#else
case 'L': ASSERT_FAIL (long long not available); break;
#endif
case 'N':
{
mp_size_t n;
n = va_arg (ap, mp_size_t);
n = ABS (n);
if (n != 0)
{
* (mp_ptr) p = retval;
MPN_ZERO ((mp_ptr) p + 1, n - 1);
}
}
break;
case 'Q': mpq_set_si ((mpq_ptr) p, (long) retval, 1L); break;
#if HAVE_PTRDIFF_T
case 't': * (ptrdiff_t *) p = retval; break;
#else
case 't': ASSERT_FAIL (ptrdiff_t not available); break;
#endif
case 'z': * (size_t *) p = retval; break;
case 'Z': mpz_set_si ((mpz_ptr) p, (long) retval); break;
}
}
va_copy (last_ap, ap);
last_fmt = fmt;
goto next;
case 'o':
param.base = 8;
goto integer;
case 'p':
case 's':
/* "void *" will be good enough for "char *" or "wchar_t *", no
need for separate code. */
(void) va_arg (ap, const void *);
goto next;
case 'x':
param.base = 16;
goto integer;
case 'X':
param.base = -16;
goto integer;
case '%':
goto next;
case '#':
param.showbase = DOPRNT_SHOWBASE_NONZERO;
break;
case '\'':
/* glibc digit grouping, just pass it through, no support for it
on gmp types */
break;
case '+':
case ' ':
param.sign = fchar;
break;
case '-':
param.justify = DOPRNT_JUSTIFY_LEFT;
break;
case '.':
seen_precision = 1;
param.prec = -1; /* "." alone means all necessary digits */
value = &param.prec;
break;
case '*':
{
int n = va_arg (ap, int);
if (value == &param.width)
{
/* negative width means left justify */
if (n < 0)
{
param.justify = DOPRNT_JUSTIFY_LEFT;
n = -n;
}
param.width = n;
}
else
{
/* don't allow negative precision */
param.prec = MAX (0, n);
}
}
break;
case '0':
if (value == &param.width)
{
/* in width field, set fill */
param.fill = '0';
/* for right justify, put the fill after any minus sign */
if (param.justify == DOPRNT_JUSTIFY_RIGHT)
param.justify = DOPRNT_JUSTIFY_INTERNAL;
}
else
{
/* in precision field, set value */
*value = 0;
}
break;
case '1': case '2': case '3': case '4': case '5':
case '6': case '7': case '8': case '9':
/* process all digits to form a value */
{
int n = 0;
do {
n = n * 10 + (fchar-'0');
fchar = *fmt++;
} while (isascii (fchar) && isdigit (fchar));
fmt--; /* unget the non-digit */
*value = n;
}
break;
default:
/* something invalid */
ASSERT (0);
goto next;
}
}
next:
/* Stop parsing the current "%" format, look for a new one. */
;
}
TRACE (printf ("remainder: \"%s\"\n", last_fmt));
if (*last_fmt != '\0')
DOPRNT_FORMAT (last_fmt, last_ap);
if (funs->final != NULL)
if ((*funs->final) (data) == -1)
goto error;
done:
(*__gmp_free_func) (alloc_fmt, alloc_fmt_size);
return retval;
error:
retval = -1;
goto done;
}