cannam@127: /* cannam@127: * Copyright (c) 2003, 2007-14 Matteo Frigo cannam@127: * Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology cannam@127: * cannam@127: * This program is free software; you can redistribute it and/or modify cannam@127: * it under the terms of the GNU General Public License as published by cannam@127: * the Free Software Foundation; either version 2 of the License, or cannam@127: * (at your option) any later version. cannam@127: * cannam@127: * This program is distributed in the hope that it will be useful, cannam@127: * but WITHOUT ANY WARRANTY; without even the implied warranty of cannam@127: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the cannam@127: * GNU General Public License for more details. cannam@127: * cannam@127: * You should have received a copy of the GNU General Public License cannam@127: * along with this program; if not, write to the Free Software cannam@127: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA cannam@127: * cannam@127: */ cannam@127: cannam@127: cannam@127: #include "ifftw.h" cannam@127: #include cannam@127: #include cannam@127: #include cannam@127: cannam@127: #define BSZ 64 cannam@127: cannam@127: static void myputs(printer *p, const char *s) cannam@127: { cannam@127: char c; cannam@127: while ((c = *s++)) cannam@127: p->putchr(p, c); cannam@127: } cannam@127: cannam@127: static void newline(printer *p) cannam@127: { cannam@127: int i; cannam@127: cannam@127: p->putchr(p, '\n'); cannam@127: for (i = 0; i < p->indent; ++i) cannam@127: p->putchr(p, ' '); cannam@127: } cannam@127: cannam@127: static const char *digits = "0123456789abcdef"; cannam@127: cannam@127: static void putint(printer *p, INT i) cannam@127: { cannam@127: char buf[BSZ]; cannam@127: char *f = buf; cannam@127: cannam@127: if (i < 0) { cannam@127: p->putchr(p, '-'); cannam@127: i = -i; cannam@127: } cannam@127: cannam@127: do { cannam@127: *f++ = digits[i % 10]; cannam@127: i /= 10; cannam@127: } while (i); cannam@127: cannam@127: do { cannam@127: p->putchr(p, *--f); cannam@127: } while (f != buf); cannam@127: } cannam@127: cannam@127: static void putulong(printer *p, unsigned long i, unsigned base, int width) cannam@127: { cannam@127: char buf[BSZ]; cannam@127: char *f = buf; cannam@127: cannam@127: do { cannam@127: *f++ = digits[i % base]; cannam@127: i /= base; cannam@127: } while (i); cannam@127: cannam@127: while (width > f - buf) { cannam@127: p->putchr(p, '0'); cannam@127: --width; cannam@127: } cannam@127: cannam@127: do { cannam@127: p->putchr(p, *--f); cannam@127: } while (f != buf); cannam@127: } cannam@127: cannam@127: static void vprint(printer *p, const char *format, va_list ap) cannam@127: { cannam@127: const char *s = format; cannam@127: char c; cannam@127: INT ival; cannam@127: cannam@127: while ((c = *s++)) { cannam@127: switch (c) { cannam@127: case '%': cannam@127: switch ((c = *s++)) { cannam@127: case 'M': { cannam@127: /* md5 value */ cannam@127: md5uint x = va_arg(ap, md5uint); cannam@127: putulong(p, (unsigned long)(0xffffffffUL & x), cannam@127: 16u, 8); cannam@127: break; cannam@127: } cannam@127: case 'c': { cannam@127: int x = va_arg(ap, int); cannam@127: p->putchr(p, (char)x); cannam@127: break; cannam@127: } cannam@127: case 's': { cannam@127: char *x = va_arg(ap, char *); cannam@127: if (x) cannam@127: myputs(p, x); cannam@127: else cannam@127: goto putnull; cannam@127: break; cannam@127: } cannam@127: case 'd': { cannam@127: int x = va_arg(ap, int); cannam@127: ival = (INT)x; cannam@127: goto putival; cannam@127: } cannam@127: case 'D': { cannam@127: ival = va_arg(ap, INT); cannam@127: goto putival; cannam@127: } cannam@127: case 'v': { cannam@127: /* print optional vector length */ cannam@127: ival = va_arg(ap, INT); cannam@127: if (ival > 1) { cannam@127: myputs(p, "-x"); cannam@127: goto putival; cannam@127: } cannam@127: break; cannam@127: } cannam@127: case 'o': { cannam@127: /* integer option. Usage: %oNAME= */ cannam@127: ival = va_arg(ap, INT); cannam@127: if (ival) cannam@127: p->putchr(p, '/'); cannam@127: while ((c = *s++) != '=') cannam@127: if (ival) cannam@127: p->putchr(p, c); cannam@127: if (ival) { cannam@127: p->putchr(p, '='); cannam@127: goto putival; cannam@127: } cannam@127: break; cannam@127: } cannam@127: case 'u': { cannam@127: unsigned x = va_arg(ap, unsigned); cannam@127: putulong(p, (unsigned long)x, 10u, 0); cannam@127: break; cannam@127: } cannam@127: case 'x': { cannam@127: unsigned x = va_arg(ap, unsigned); cannam@127: putulong(p, (unsigned long)x, 16u, 0); cannam@127: break; cannam@127: } cannam@127: case '(': { cannam@127: /* newline, augment indent level */ cannam@127: p->indent += p->indent_incr; cannam@127: newline(p); cannam@127: break; cannam@127: } cannam@127: case ')': { cannam@127: /* decrement indent level */ cannam@127: p->indent -= p->indent_incr; cannam@127: break; cannam@127: } cannam@127: case 'p': { /* note difference from C's %p */ cannam@127: /* print plan */ cannam@127: plan *x = va_arg(ap, plan *); cannam@127: if (x) cannam@127: x->adt->print(x, p); cannam@127: else cannam@127: goto putnull; cannam@127: break; cannam@127: } cannam@127: case 'P': { cannam@127: /* print problem */ cannam@127: problem *x = va_arg(ap, problem *); cannam@127: if (x) cannam@127: x->adt->print(x, p); cannam@127: else cannam@127: goto putnull; cannam@127: break; cannam@127: } cannam@127: case 'T': { cannam@127: /* print tensor */ cannam@127: tensor *x = va_arg(ap, tensor *); cannam@127: if (x) cannam@127: X(tensor_print)(x, p); cannam@127: else cannam@127: goto putnull; cannam@127: break; cannam@127: } cannam@127: default: cannam@127: A(0 /* unknown format */); cannam@127: break; cannam@127: cannam@127: putnull: cannam@127: myputs(p, "(null)"); cannam@127: break; cannam@127: cannam@127: putival: cannam@127: putint(p, ival); cannam@127: break; cannam@127: } cannam@127: break; cannam@127: default: cannam@127: p->putchr(p, c); cannam@127: break; cannam@127: } cannam@127: } cannam@127: } cannam@127: cannam@127: static void print(printer *p, const char *format, ...) cannam@127: { cannam@127: va_list ap; cannam@127: va_start(ap, format); cannam@127: vprint(p, format, ap); cannam@127: va_end(ap); cannam@127: } cannam@127: cannam@127: printer *X(mkprinter)(size_t size, cannam@127: void (*putchr)(printer *p, char c), cannam@127: void (*cleanup)(printer *p)) cannam@127: { cannam@127: printer *s = (printer *)MALLOC(size, OTHER); cannam@127: s->print = print; cannam@127: s->vprint = vprint; cannam@127: s->putchr = putchr; cannam@127: s->cleanup = cleanup; cannam@127: s->indent = 0; cannam@127: s->indent_incr = 2; cannam@127: return s; cannam@127: } cannam@127: cannam@127: void X(printer_destroy)(printer *p) cannam@127: { cannam@127: if (p->cleanup) cannam@127: p->cleanup(p); cannam@127: X(ifree)(p); cannam@127: }