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