#include "c.h"

static char rcsid[] = "$Id: output.nw,v 2.12 1997/06/03 23:23:20 drh Exp $";

static char *outs(const char *str, FILE *f, char *bp) {
        if (f)
                fputs(str, f);
        else
                while (*bp = *str++)
                        bp++;
        return bp;
}

static char *outd(long n, FILE *f, char *bp) {
        unsigned long m;
        char buf[25], *s = buf + sizeof buf;

        *--s = '\0';
        if (n < 0)
                m = -n;
        else
                m = n;
        do
                *--s = m%10 + '0';
        while ((m /= 10) != 0);
        if (n < 0)
                *--s = '-';
        return outs(s, f, bp);
}

static char *outu(unsigned long n, int base, FILE *f, char *bp) {
        char buf[25], *s = buf + sizeof buf;

        *--s = '\0';
        do
                *--s = "0123456789abcdef"[n%base];
        while ((n /= base) != 0);
        return outs(s, f, bp);
}
void print(const char *fmt, ...) {
        va_list ap;

        va_start(ap, fmt);
        vfprint(stdout, NULL, fmt, ap);
        va_end(ap);
}
/* fprint - formatted output to  f */
void fprint(FILE *f, const char *fmt, ...) {
        va_list ap;

        va_start(ap, fmt);
        vfprint(f, NULL, fmt, ap);
        va_end(ap);
}

/* stringf - formatted output to a saved string */
char *stringf(const char *fmt, ...) {
        char buf[1024];
        va_list ap;

        va_start(ap, fmt);
        vfprint(NULL, buf, fmt, ap);
        va_end(ap);
        return string(buf);
}

/* vfprint - formatted output to f or string bp */
void vfprint(FILE *f, char *bp, const char *fmt, va_list ap) {
        for (; *fmt; fmt++)
                if (*fmt == '%')
                        switch (*++fmt) {
                        case 'd': bp = outd(va_arg(ap, int), f, bp); break;
                        case 'D': bp = outd(va_arg(ap, long), f, bp); break;
                        case 'U': bp = outu(va_arg(ap, unsigned long), 10, f, bp); break;
                        case 'u': bp = outu(va_arg(ap, unsigned), 10, f, bp); break;
                        case 'o': bp = outu(va_arg(ap, unsigned), 8, f, bp); break;
                        case 'X': bp = outu(va_arg(ap, unsigned long), 16, f, bp); break;
                        case 'x': bp = outu(va_arg(ap, unsigned), 16, f, bp); break;
                        case 'f': case 'e':
                        case 'g': {
				          static char format[] = "%f";
				          char buf[128];
				          format[1] = *fmt;
				          sprintf(buf, format, va_arg(ap, double));
				          outs(buf, f, bp);
				  }
; break;
                        case 's': bp = outs(va_arg(ap, char *), f, bp); break;
                        case 'p': {
                                void *p = va_arg(ap, void *);
                                if (p)
                                        bp = outs("0x", f, bp);
                                bp = outu((unsigned long)p, 16, f, bp);
                                break;
                                  }
                        case 'c': if (f) fputc(va_arg(ap, int), f); else *bp++ = va_arg(ap, int); break;
                        case 'S': { char *s = va_arg(ap, char *);
				    int n = va_arg(ap, int);
				    if (s)
				            for ( ; n-- > 0; s++)
				                    if (f) putc(*s, f); else *bp++ = *s;
 } break;
                        case 'k': { int t = va_arg(ap, int);
				    static char *tokens[] = {
#define xx(a,b,c,d,e,f,g) g,
#define yy(a,b,c,d,e,f,g) g,
#include "token.h"
				    };
				    assert(tokens[t&0177]);
				    bp = outs(tokens[t&0177], f, bp);
 } break;
                        case 't': { Type ty = va_arg(ap, Type);
				    assert(f);
				    outtype(ty ? ty : voidtype, f);
 } break;
                        case 'w': { Coordinate *p = va_arg(ap, Coordinate *);
				    if (p->file && *p->file) {
				            bp = outs(p->file, f, bp);
				            bp = outs(":", f, bp);
				    }
				    bp = outd(p->y, f, bp);
 } break;
                        case 'I': { int n = va_arg(ap, int);
				    while (--n >= 0)
				            if (f) putc(' ', f); else *bp++ = ' ';
 } break;
                        default:  if (f) putc(*fmt, f); else *bp++ = *fmt; break;
                        }
                else if (f)
                        putc(*fmt, f);
                else
                        *bp++ = *fmt;
        if (!f)
                *bp = '\0';
}
