cannam@95: /*
cannam@95:  * Copyright (c) 2001 Matteo Frigo
cannam@95:  * Copyright (c) 2001 Massachusetts Institute of Technology
cannam@95:  *
cannam@95:  * This program is free software; you can redistribute it and/or modify
cannam@95:  * it under the terms of the GNU General Public License as published by
cannam@95:  * the Free Software Foundation; either version 2 of the License, or
cannam@95:  * (at your option) any later version.
cannam@95:  *
cannam@95:  * This program is distributed in the hope that it will be useful,
cannam@95:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@95:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
cannam@95:  * GNU General Public License for more details.
cannam@95:  *
cannam@95:  * You should have received a copy of the GNU General Public License
cannam@95:  * along with this program; if not, write to the Free Software
cannam@95:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
cannam@95:  *
cannam@95:  */
cannam@95: 
cannam@95: 
cannam@95: #include "config.h"
cannam@95: #include "bench.h"
cannam@95: #include <stdio.h>
cannam@95: #include <stdlib.h>
cannam@95: #include <string.h>
cannam@95: #include <ctype.h>
cannam@95: 
cannam@95: int always_pad_real = 0; /* by default, only pad in-place case */
cannam@95: 
cannam@95: typedef enum {
cannam@95:      SAME, PADDED, HALFISH
cannam@95: } n_transform;
cannam@95: 
cannam@95: /* funny transformations for last dimension of PROBLEM_REAL */
cannam@95: static int transform_n(int n, n_transform nt)
cannam@95: {
cannam@95:      switch (nt) {
cannam@95: 	 case SAME: return n;
cannam@95: 	 case PADDED: return 2*(n/2+1);
cannam@95: 	 case HALFISH: return (n/2+1);
cannam@95: 	 default: BENCH_ASSERT(0); return 0;
cannam@95:      }
cannam@95: }
cannam@95: 
cannam@95: /* do what I mean */
cannam@95: static bench_tensor *dwim(bench_tensor *t, bench_iodim **last_iodim,
cannam@95: 			  n_transform nti, n_transform nto,
cannam@95: 			  bench_iodim *dt)
cannam@95: {
cannam@95:      int i;
cannam@95:      bench_iodim *d, *d1;
cannam@95: 
cannam@95:      if (!FINITE_RNK(t->rnk) || t->rnk < 1)
cannam@95: 	  return t;
cannam@95: 
cannam@95:      i = t->rnk;
cannam@95:      d1 = *last_iodim;
cannam@95: 
cannam@95:      while (--i >= 0) {
cannam@95: 	  d = t->dims + i;
cannam@95: 	  if (!d->is) 
cannam@95: 	       d->is = d1->is * transform_n(d1->n, d1==dt ? nti : SAME); 
cannam@95: 	  if (!d->os) 
cannam@95: 	       d->os = d1->os * transform_n(d1->n, d1==dt ? nto : SAME); 
cannam@95: 	  d1 = d;
cannam@95:      }
cannam@95: 
cannam@95:      *last_iodim = d1;
cannam@95:      return t;
cannam@95: }
cannam@95: 
cannam@95: static void transpose_tensor(bench_tensor *t)
cannam@95: {
cannam@95:      if (!FINITE_RNK(t->rnk) || t->rnk < 2)
cannam@95:           return;
cannam@95: 
cannam@95:      t->dims[0].os = t->dims[1].os;
cannam@95:      t->dims[1].os = t->dims[0].os * t->dims[0].n;
cannam@95: }
cannam@95: 
cannam@95: static const char *parseint(const char *s, int *n)
cannam@95: {
cannam@95:      int sign = 1;
cannam@95: 
cannam@95:      *n = 0;
cannam@95: 
cannam@95:      if (*s == '-') { 
cannam@95: 	  sign = -1;
cannam@95: 	  ++s;
cannam@95:      } else if (*s == '+') { 
cannam@95: 	  sign = +1; 
cannam@95: 	  ++s; 
cannam@95:      }
cannam@95: 
cannam@95:      BENCH_ASSERT(isdigit(*s));
cannam@95:      while (isdigit(*s)) {
cannam@95: 	  *n = *n * 10 + (*s - '0');
cannam@95: 	  ++s;
cannam@95:      }
cannam@95:      
cannam@95:      *n *= sign;
cannam@95: 
cannam@95:      if (*s == 'k' || *s == 'K') {
cannam@95: 	  *n *= 1024;
cannam@95: 	  ++s;
cannam@95:      }
cannam@95: 
cannam@95:      if (*s == 'm' || *s == 'M') {
cannam@95: 	  *n *= 1024 * 1024;
cannam@95: 	  ++s;
cannam@95:      }
cannam@95: 
cannam@95:      return s;
cannam@95: }
cannam@95: 
cannam@95: struct dimlist { bench_iodim car; r2r_kind_t k; struct dimlist *cdr; };
cannam@95: 
cannam@95: static const char *parsetensor(const char *s, bench_tensor **tp,
cannam@95: 			       r2r_kind_t **k)
cannam@95: {
cannam@95:      struct dimlist *l = 0, *m;
cannam@95:      bench_tensor *t;
cannam@95:      int rnk = 0;
cannam@95: 
cannam@95:  L1:
cannam@95:      m = (struct dimlist *)bench_malloc(sizeof(struct dimlist));
cannam@95:      /* nconc onto l */
cannam@95:      m->cdr = l; l = m;
cannam@95:      ++rnk; 
cannam@95: 
cannam@95:      s = parseint(s, &m->car.n);
cannam@95: 
cannam@95:      if (*s == ':') {
cannam@95: 	  /* read input stride */
cannam@95: 	  ++s;
cannam@95: 	  s = parseint(s, &m->car.is);
cannam@95: 	  if (*s == ':') {
cannam@95: 	       /* read output stride */
cannam@95: 	       ++s;
cannam@95: 	       s = parseint(s, &m->car.os);
cannam@95: 	  } else {
cannam@95: 	       /* default */
cannam@95: 	       m->car.os = m->car.is;
cannam@95: 	  }
cannam@95:      } else {
cannam@95: 	  m->car.is = 0;
cannam@95: 	  m->car.os = 0;
cannam@95:      }
cannam@95: 
cannam@95:      if (*s == 'f' || *s == 'F') {
cannam@95: 	  m->k = R2R_R2HC;
cannam@95: 	  ++s;
cannam@95:      }
cannam@95:      else if (*s == 'b' || *s == 'B') {
cannam@95: 	  m->k = R2R_HC2R;
cannam@95: 	  ++s;
cannam@95:      }
cannam@95:      else if (*s == 'h' || *s == 'H') {
cannam@95: 	  m->k = R2R_DHT;
cannam@95: 	  ++s;
cannam@95:      }
cannam@95:      else if (*s == 'e' || *s == 'E' || *s == 'o' || *s == 'O') {
cannam@95: 	  char c = *(s++);
cannam@95: 	  int ab;
cannam@95: 
cannam@95: 	  s = parseint(s, &ab);
cannam@95: 
cannam@95: 	  if (c == 'e' || c == 'E') {
cannam@95: 	       if (ab == 0)
cannam@95: 		    m->k = R2R_REDFT00;
cannam@95: 	       else if (ab == 1)
cannam@95: 		    m->k = R2R_REDFT01;
cannam@95: 	       else if (ab == 10)
cannam@95: 		    m->k = R2R_REDFT10;
cannam@95: 	       else if (ab == 11)
cannam@95: 		    m->k = R2R_REDFT11;
cannam@95: 	       else
cannam@95: 		    BENCH_ASSERT(0);
cannam@95: 	  }
cannam@95: 	  else {
cannam@95: 	       if (ab == 0)
cannam@95: 		    m->k = R2R_RODFT00;
cannam@95: 	       else if (ab == 1)
cannam@95: 		    m->k = R2R_RODFT01;
cannam@95: 	       else if (ab == 10)
cannam@95: 		    m->k = R2R_RODFT10;
cannam@95: 	       else if (ab == 11)
cannam@95: 		    m->k = R2R_RODFT11;
cannam@95: 	       else
cannam@95: 		    BENCH_ASSERT(0);
cannam@95: 	  }
cannam@95:      }
cannam@95:      else
cannam@95: 	  m->k = R2R_R2HC;
cannam@95: 
cannam@95:      if (*s == 'x' || *s == 'X') {
cannam@95: 	  ++s;
cannam@95: 	  goto L1;
cannam@95:      }
cannam@95:      
cannam@95:      /* now we have a dimlist.  Build bench_tensor, etc. */
cannam@95: 
cannam@95:      if (k && rnk > 0) {
cannam@95: 	  int i;
cannam@95: 	  *k = (r2r_kind_t *) bench_malloc(sizeof(r2r_kind_t) * rnk);
cannam@95: 	  for (m = l, i = rnk - 1; i >= 0; --i, m = m->cdr) {
cannam@95: 	       BENCH_ASSERT(m);
cannam@95: 	       (*k)[i] = m->k;
cannam@95: 	  }
cannam@95:      }
cannam@95: 
cannam@95:      t = mktensor(rnk);
cannam@95:      while (--rnk >= 0) {
cannam@95: 	  bench_iodim *d = t->dims + rnk;
cannam@95: 	  BENCH_ASSERT(l);
cannam@95: 	  m = l; l = m->cdr;
cannam@95: 	  d->n = m->car.n;
cannam@95: 	  d->is = m->car.is;
cannam@95: 	  d->os = m->car.os;
cannam@95: 	  bench_free(m);
cannam@95:      }
cannam@95: 
cannam@95:      *tp = t;
cannam@95:      return s;
cannam@95: }
cannam@95: 
cannam@95: /* parse a problem description, return a problem */
cannam@95: bench_problem *problem_parse(const char *s)
cannam@95: {
cannam@95:      bench_problem *p;
cannam@95:      bench_iodim last_iodim0 = {1,1,1}, *last_iodim = &last_iodim0;
cannam@95:      bench_iodim *sz_last_iodim;
cannam@95:      bench_tensor *sz;
cannam@95:      n_transform nti = SAME, nto = SAME;
cannam@95:      int transpose = 0;
cannam@95: 
cannam@95:      p = (bench_problem *) bench_malloc(sizeof(bench_problem));
cannam@95:      p->kind = PROBLEM_COMPLEX;
cannam@95:      p->k = 0;
cannam@95:      p->sign = -1;
cannam@95:      p->in = p->out = 0;
cannam@95:      p->inphys = p->outphys = 0;
cannam@95:      p->iphyssz = p->ophyssz = 0;
cannam@95:      p->in_place = 0;
cannam@95:      p->destroy_input = 0;
cannam@95:      p->split = 0;
cannam@95:      p->userinfo = 0;
cannam@95:      p->scrambled_in = p->scrambled_out = 0;
cannam@95:      p->sz = p->vecsz = 0;
cannam@95:      p->ini = p->outi = 0;
cannam@95:      p->pstring = (char *) bench_malloc(sizeof(char) * (strlen(s) + 1));
cannam@95:      strcpy(p->pstring, s);
cannam@95: 
cannam@95:  L1:
cannam@95:      switch (tolower(*s)) {
cannam@95: 	 case 'i': p->in_place = 1; ++s; goto L1;
cannam@95: 	 case 'o': p->in_place = 0; ++s; goto L1;
cannam@95: 	 case 'd': p->destroy_input = 1; ++s; goto L1;
cannam@95: 	 case '/': p->split = 1; ++s; goto L1;
cannam@95: 	 case 'f': 
cannam@95: 	 case '-': p->sign = -1; ++s; goto L1;
cannam@95: 	 case 'b': 
cannam@95: 	 case '+': p->sign = 1; ++s; goto L1;
cannam@95: 	 case 'r': p->kind = PROBLEM_REAL; ++s; goto L1;
cannam@95: 	 case 'c': p->kind = PROBLEM_COMPLEX; ++s; goto L1;
cannam@95: 	 case 'k': p->kind = PROBLEM_R2R; ++s; goto L1;
cannam@95: 	 case 't': transpose = 1; ++s; goto L1;
cannam@95: 	      
cannam@95: 	 /* hack for MPI: */
cannam@95: 	 case '[': p->scrambled_in = 1; ++s; goto L1;
cannam@95: 	 case ']': p->scrambled_out = 1; ++s; goto L1;
cannam@95: 
cannam@95: 	 default : ;
cannam@95:      }
cannam@95: 
cannam@95:      s = parsetensor(s, &sz, p->kind == PROBLEM_R2R ? &p->k : 0);
cannam@95: 
cannam@95:      if (p->kind == PROBLEM_REAL) {
cannam@95: 	  if (p->sign < 0) {
cannam@95: 	       nti = p->in_place || always_pad_real ? PADDED : SAME;
cannam@95: 	       nto = HALFISH;
cannam@95: 	  }
cannam@95: 	  else {
cannam@95: 	       nti = HALFISH;
cannam@95: 	       nto = p->in_place || always_pad_real ? PADDED : SAME;
cannam@95: 	  }
cannam@95:      }
cannam@95: 
cannam@95:      sz_last_iodim = sz->dims + sz->rnk - 1;
cannam@95:      if (*s == '*') { /* "external" vector */
cannam@95: 	  ++s;
cannam@95: 	  p->sz = dwim(sz, &last_iodim, nti, nto, sz_last_iodim);
cannam@95: 	  s = parsetensor(s, &sz, 0);
cannam@95: 	  p->vecsz = dwim(sz, &last_iodim, nti, nto, sz_last_iodim);
cannam@95:      } else if (*s == 'v' || *s == 'V') { /* "internal" vector */
cannam@95: 	  bench_tensor *vecsz;
cannam@95: 	  ++s;
cannam@95: 	  s = parsetensor(s, &vecsz, 0);
cannam@95: 	  p->vecsz = dwim(vecsz, &last_iodim, nti, nto, sz_last_iodim);
cannam@95: 	  p->sz = dwim(sz, &last_iodim, nti, nto, sz_last_iodim);
cannam@95:      } else {
cannam@95: 	  p->sz = dwim(sz, &last_iodim, nti, nto, sz_last_iodim);
cannam@95: 	  p->vecsz = mktensor(0);
cannam@95:      }
cannam@95: 
cannam@95:      if (transpose) {
cannam@95: 	  transpose_tensor(p->sz);
cannam@95: 	  transpose_tensor(p->vecsz);
cannam@95:      }
cannam@95: 
cannam@95:      if (!p->in_place)
cannam@95: 	  p->out = ((bench_real *) p->in) + (1 << 20);  /* whatever */
cannam@95: 
cannam@95:      BENCH_ASSERT(p->sz && p->vecsz);
cannam@95:      BENCH_ASSERT(!*s);
cannam@95:      return p;
cannam@95: }
cannam@95: 
cannam@95: void problem_destroy(bench_problem *p)
cannam@95: {
cannam@95:      BENCH_ASSERT(p);
cannam@95:      problem_free(p);
cannam@95:      bench_free0(p->k);
cannam@95:      bench_free0(p->pstring);
cannam@95:      bench_free(p);
cannam@95: }
cannam@95: