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