cannam@127: /* Re-use libbench2 and the test program, but override bench_main so that
cannam@127:    we can have different command-line syntax. */
cannam@127: #include "my-getopt.h"
cannam@127: #include "bench.h"
cannam@127: 
cannam@127: #include <stdio.h>
cannam@127: #include <stdlib.h>
cannam@127: #include <ctype.h>
cannam@127: #include <fftw3.h>
cannam@127: #include <string.h>
cannam@127: #include <time.h>
cannam@127: 
cannam@127: #if defined(HAVE_THREADS) || defined(HAVE_OPENMP)
cannam@127: #  define HAVE_SMP
cannam@127:    extern int threads_ok;
cannam@127: #endif
cannam@127: 
cannam@127: #define CONCAT(prefix, name) prefix ## name
cannam@127: #if defined(BENCHFFT_SINGLE)
cannam@127: #define FFTW(x) CONCAT(fftwf_, x)
cannam@127: #elif defined(BENCHFFT_LDOUBLE)
cannam@127: #define FFTW(x) CONCAT(fftwl_, x)
cannam@127: #elif defined(BENCHFFT_QUAD)
cannam@127: #define FFTW(x) CONCAT(fftwq_, x)
cannam@127: #else
cannam@127: #define FFTW(x) CONCAT(fftw_, x)
cannam@127: #endif
cannam@127: 
cannam@127: /* from bench.c: */
cannam@127: extern unsigned the_flags;
cannam@127: extern int usewisdom;
cannam@127: extern int nthreads;
cannam@127: 
cannam@127: /* dummy routines to replace those in hook.c */
cannam@127: void install_hook(void) {}
cannam@127: void uninstall_hook(void) {}
cannam@127: 
cannam@127: int verbose;
cannam@127: 
cannam@127: static void do_problem(bench_problem *p)
cannam@127: {
cannam@127:      if (verbose)
cannam@127: 	  printf("Planning transform: %s\n", p->pstring);
cannam@127:      /* BENCH_ASSERT(can_do(p)); */
cannam@127:      problem_alloc(p);
cannam@127:      setup(p);
cannam@127:      done(p);
cannam@127: }
cannam@127: 
cannam@127: static void add_problem(const char *pstring,
cannam@127: 			bench_problem ***p, int *ip, int *np)
cannam@127: {
cannam@127:      if (*ip >= *np) {
cannam@127: 	  *np = *np * 2 + 1;
cannam@127: 	  *p = (bench_problem **) realloc(*p, sizeof(bench_problem *) * *np);
cannam@127:      }
cannam@127:      (*p)[(*ip)++] = problem_parse(pstring);
cannam@127: }
cannam@127: 
cannam@127: static int sz(const bench_problem *p)
cannam@127: {
cannam@127:      return tensor_sz(p->sz) * tensor_sz(p->vecsz);
cannam@127: }
cannam@127: 
cannam@127: static int prob_size_cmp(const void *p1_, const void *p2_)
cannam@127: {
cannam@127:      const bench_problem * const *p1 = (const bench_problem * const *) p1_;
cannam@127:      const bench_problem * const *p2 = (const bench_problem * const *) p2_;
cannam@127:      return (sz(*p1) - sz(*p2));
cannam@127: }
cannam@127: 
cannam@127: static struct my_option options[] =
cannam@127: {
cannam@127:   {"help", NOARG, 'h'},
cannam@127:   {"version", NOARG, 'V'},
cannam@127:   {"verbose", NOARG, 'v'},
cannam@127: 
cannam@127:   {"canonical", NOARG, 'c'},
cannam@127:   {"time-limit", REQARG, 't'},
cannam@127: 
cannam@127:   {"output-file", REQARG, 'o'},
cannam@127: 
cannam@127:   {"impatient", NOARG, 'i'},
cannam@127:   {"measure", NOARG, 'm'},
cannam@127:   {"estimate", NOARG, 'e'},
cannam@127:   {"exhaustive", NOARG, 'x'},
cannam@127: 
cannam@127:   {"no-system-wisdom", NOARG, 'n'},
cannam@127:   {"wisdom-file", REQARG, 'w'},
cannam@127: 
cannam@127: #ifdef HAVE_SMP
cannam@127:   {"threads", REQARG, 'T'},
cannam@127: #endif
cannam@127: 
cannam@127:   /* options to restrict configuration to rdft-only, etcetera? */
cannam@127:   
cannam@127:   {0, NOARG, 0}
cannam@127: };
cannam@127: 
cannam@127: static void help(FILE *f, const char *program_name)
cannam@127: {
cannam@127:      fprintf(
cannam@127: 	  f, 
cannam@127: 	  "Usage: %s [options] [sizes]\n"
cannam@127: "    Create wisdom (pre-planned/optimized transforms) for specified sizes,\n"
cannam@127: "    writing wisdom to stdout (or to a file, using -o).\n"
cannam@127: 	  "\nOptions:\n"
cannam@127:  "                   -h, --help: print this help\n"
cannam@127:  "                -V, --version: print version/copyright info\n"
cannam@127:  "                -v, --verbose: verbose output\n"
cannam@127:  "              -c, --canonical: plan/optimize canonical set of sizes\n"
cannam@127:  "     -t <h>, --time-limit=<h>: time limit in hours (default: 0, no limit)\n"
cannam@127:  "  -o FILE, --output-file=FILE: output to FILE instead of stdout\n"
cannam@127:  "                -m, --measure: plan in MEASURE mode (PATIENT is default)\n"
cannam@127:  "               -e, --estimate: plan in ESTIMATE mode (not recommended)\n"
cannam@127:  "             -x, --exhaustive: plan in EXHAUSTIVE mode (may be slow)\n"
cannam@127:  "       -n, --no-system-wisdom: don't read /etc/fftw/ system wisdom file\n"
cannam@127:  "  -w FILE, --wisdom-file=FILE: read wisdom from FILE (stdin if -)\n"
cannam@127: #ifdef HAVE_SMP
cannam@127:  "            -T N, --threads=N: plan with N threads\n"
cannam@127: #endif
cannam@127: 	  "\nSize syntax: <type><inplace><direction><geometry>\n"
cannam@127:  "      <type> = c/r/k for complex/real(r2c,c2r)/r2r\n" 
cannam@127:  "   <inplace> = i/o for in/out-of place\n"
cannam@127:  " <direction> = f/b for forward/backward, omitted for k transforms\n"
cannam@127:  "  <geometry> = <n1>[x<n2>[x...]], e.g. 10x12x14\n"
cannam@127:  "               -- for k transforms, after each dimension is a <kind>:\n"
cannam@127:  "                     <kind> = f/b/h/e00/e01/e10/e11/o00/o01/o10/o11\n"
cannam@127:  "                              for R2HC/HC2R/DHT/REDFT00/.../RODFT11\n"
cannam@127: 	  , program_name);
cannam@127: }
cannam@127: 
cannam@127: /* powers of two and ten up to 2^20, for now */
cannam@127: static char canonical_sizes[][32] = {
cannam@127:      "1", "2", "4", "8", "16", "32", "64", "128", "256", "512", "1024",
cannam@127:      "2048", "4096", "8192", "16384", "32768", "65536", "131072",
cannam@127:      "262144", "524288", "1048576",
cannam@127: 
cannam@127:      "10", "100", "1000", "10000", "100000", "1000000",
cannam@127: 
cannam@127:      "2x2", "4x4", "8x8", "10x10", "16x16", "32x32", "64x64", "100x100",
cannam@127:      "128x128", "256x256", "512x512", "1000x1000", "1024x1024",
cannam@127: 
cannam@127:      "2x2x2", "4x4x4", "8x8x8", "10x10x10", "16x16x16", "32x32x32",
cannam@127:      "64x64x64", "100x100x100"
cannam@127: };
cannam@127: 
cannam@127: #define NELEM(array)(sizeof(array) / sizeof((array)[0]))
cannam@127: 
cannam@127: int bench_main(int argc, char *argv[])
cannam@127: {
cannam@127:      int c;
cannam@127:      unsigned i;
cannam@127:      int impatient = 0;
cannam@127:      int system_wisdom = 1;
cannam@127:      int canonical = 0;
cannam@127:      double hours = 0;
cannam@127:      FILE *output_file;
cannam@127:      char *output_fname = 0;
cannam@127:      bench_problem **problems = 0;
cannam@127:      int nproblems = 0, iproblem = 0;
cannam@127:      time_t begin;
cannam@127: 
cannam@127:      verbose = 0;
cannam@127:      usewisdom = 0;
cannam@127: 
cannam@127:      bench_srand(1);
cannam@127: #ifdef HAVE_SMP
cannam@127:      /* do not configure FFTW with threads, unless the
cannam@127: 	user requests -T */
cannam@127:      threads_ok = 0;
cannam@127: #endif
cannam@127: 
cannam@127:      while ((c = my_getopt(argc, argv, options)) != -1) {
cannam@127: 	  switch (c) {
cannam@127: 	      case 'h':
cannam@127: 		   help(stdout, argv[0]);
cannam@127: 		   exit(EXIT_SUCCESS);
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'V':
cannam@127: 		   printf("fftw-wisdom tool for FFTW version " VERSION ".\n");
cannam@127: 		   printf(
cannam@127: "\n"
cannam@127: "Copyright (c) 2003, 2007-14 Matteo Frigo\n"
cannam@127: "Copyright (c) 2003, 2007-14 Massachusetts Institute of Technology\n"
cannam@127: "\n"
cannam@127: "This program is free software; you can redistribute it and/or modify\n"
cannam@127: "it under the terms of the GNU General Public License as published by\n"
cannam@127: "the Free Software Foundation; either version 2 of the License, or\n"
cannam@127: "(at your option) any later version.\n"
cannam@127: "\n"
cannam@127: "This program is distributed in the hope that it will be useful,\n"
cannam@127: "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
cannam@127: "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
cannam@127: "GNU General Public License for more details.\n"
cannam@127: "\n"
cannam@127: "You should have received a copy of the GNU General Public License\n"
cannam@127: "along with this program; if not, write to the Free Software\n"
cannam@127: "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA\n"
cannam@127: 			);
cannam@127: 		   exit(EXIT_SUCCESS);
cannam@127: 		   break;
cannam@127: 		   
cannam@127: 	      case 'v':
cannam@127: 		   verbose = 1;
cannam@127: 		   break;
cannam@127: 		   
cannam@127: 	      case 'c':
cannam@127: 		   canonical = 1;
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 't':
cannam@127: 		   hours = atof(my_optarg);
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'o':
cannam@127: 		   if (output_fname)
cannam@127: 			bench_free(output_fname);
cannam@127: 		   
cannam@127: 		   if (!strcmp(my_optarg, "-"))
cannam@127: 			output_fname = 0;
cannam@127: 		   else {
cannam@127: 			output_fname = (char *) bench_malloc(sizeof(char) *
cannam@127: 						    (strlen(my_optarg) + 1));
cannam@127: 			strcpy(output_fname, my_optarg);
cannam@127: 		   }
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'm':
cannam@127: 	      case 'i':
cannam@127: 		   impatient = 1;
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'e':
cannam@127: 		   the_flags |= FFTW_ESTIMATE;
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'x':
cannam@127: 		   the_flags |= FFTW_EXHAUSTIVE;
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'n':
cannam@127: 		   system_wisdom = 0;
cannam@127: 		   break;
cannam@127: 
cannam@127: 	      case 'w': {
cannam@127: 		   FILE *w = stdin;
cannam@127: 		   if (strcmp(my_optarg, "-") && !(w = fopen(my_optarg, "r"))) {
cannam@127: 			fprintf(stderr,
cannam@127: 				"fftw-wisdom: error opening \"%s\": ", my_optarg);
cannam@127: 			perror("");
cannam@127: 			exit(EXIT_FAILURE);
cannam@127: 		   }
cannam@127: 		   if (!FFTW(import_wisdom_from_file)(w)) {
cannam@127: 			fprintf(stderr, "fftw_wisdom: error reading wisdom "
cannam@127: 				"from \"%s\"\n", my_optarg);
cannam@127: 			exit(EXIT_FAILURE);
cannam@127: 		   }
cannam@127: 		   if (w != stdin)
cannam@127: 			fclose(w);
cannam@127: 		   break;
cannam@127: 	      }
cannam@127: 
cannam@127: #ifdef HAVE_SMP
cannam@127: 	      case 'T':
cannam@127: 		   nthreads = atoi(my_optarg);
cannam@127: 		   if (nthreads < 1) nthreads = 1;
cannam@127: 		   threads_ok = 1;
cannam@127: 		   BENCH_ASSERT(FFTW(init_threads)());
cannam@127: 		   break;
cannam@127: #endif
cannam@127: 
cannam@127: 	      case '?':
cannam@127: 		   /* `my_getopt' already printed an error message. */
cannam@127: 		   cleanup();
cannam@127: 		   return EXIT_FAILURE;
cannam@127: 
cannam@127: 	      default:
cannam@127: 		   abort ();
cannam@127: 	  }
cannam@127:      }
cannam@127: 
cannam@127:      if (!impatient)
cannam@127: 	  the_flags |= FFTW_PATIENT;
cannam@127: 
cannam@127:      if (system_wisdom)
cannam@127: 	  if (!FFTW(import_system_wisdom)() && verbose)
cannam@127: 	       fprintf(stderr, "fftw-wisdom: system-wisdom import failed\n");
cannam@127: 
cannam@127:      if (canonical) {
cannam@127: 	  for (i = 0; i < NELEM(canonical_sizes); ++i) {
cannam@127: 	       unsigned j;
cannam@127: 	       char types[][8] = { 
cannam@127: 		    "cof", "cob", "cif", "cib", "rof", "rob", "rif", "rib"
cannam@127: 	       };
cannam@127: 	       
cannam@127: 	       for (j = 0; j < NELEM(types); ++j) {
cannam@127: 		    char ps[64];
cannam@127: 		    if (!strchr(canonical_sizes[i],'x')
cannam@127: 			|| !strchr(types[j],'o')) {
cannam@127: #ifdef HAVE_SNPRINTF
cannam@127: 			 snprintf(ps, sizeof(ps), "%s%s", types[j], canonical_sizes[i]);
cannam@127: #else
cannam@127: 			 sprintf(ps, "%s%s", types[j], canonical_sizes[i]);
cannam@127: #endif
cannam@127: 			 add_problem(ps, &problems, &iproblem, &nproblems);
cannam@127: 		    }
cannam@127: 	       }
cannam@127: 	  }
cannam@127:      }
cannam@127: 
cannam@127:      while (my_optind < argc) {
cannam@127: 	  if (!strcmp(argv[my_optind], "-")) {
cannam@127: 	       char s[1025];
cannam@127: 	       while (1 == fscanf(stdin, "%1024s", s))
cannam@127: 		    add_problem(s, &problems, &iproblem, &nproblems);
cannam@127: 	  }
cannam@127: 	  else
cannam@127: 	       add_problem(argv[my_optind], &problems, &iproblem, &nproblems);
cannam@127: 	  ++my_optind;
cannam@127:      }
cannam@127: 
cannam@127:      nproblems = iproblem;
cannam@127:      qsort(problems, nproblems, sizeof(bench_problem *), prob_size_cmp);
cannam@127: 
cannam@127:      if (!output_fname)
cannam@127: 	  output_file = stdout;
cannam@127:      else
cannam@127: 	  if (!(output_file = fopen(output_fname, "w"))) {
cannam@127: 	       fprintf(stderr,
cannam@127: 		       "fftw-wisdom: error creating \"%s\"", output_fname);
cannam@127: 	       perror("");
cannam@127: 	       exit(EXIT_FAILURE);
cannam@127: 	  }
cannam@127: 
cannam@127:      begin = time((time_t*)0);
cannam@127:      for (iproblem = 0; iproblem < nproblems; ++iproblem) {
cannam@127: 	  if (hours <= 0
cannam@127: 	      || hours > (time((time_t*)0) - begin) / 3600.0)
cannam@127: 	       do_problem(problems[iproblem]);
cannam@127: 	  problem_destroy(problems[iproblem]);
cannam@127: 	  
cannam@127:      }
cannam@127:      free(problems);
cannam@127:      
cannam@127:      if (verbose && hours > 0
cannam@127: 	 && hours < (time((time_t*)0) - begin) / 3600.0)
cannam@127: 	  fprintf(stderr, "EXCEEDED TIME LIMIT OF %g HOURS.\n", hours);
cannam@127: 
cannam@127:      FFTW(export_wisdom_to_file)(output_file);
cannam@127:      if (output_file != stdout)
cannam@127: 	  fclose(output_file);
cannam@127:      if (output_fname)
cannam@127: 	  bench_free(output_fname);
cannam@127: 
cannam@127:      cleanup();
cannam@127: 
cannam@127:      return EXIT_SUCCESS;
cannam@127: }