More changes. Still embryonic.

[u/mdw/catacomb] / genprimes.c

/* -*-c-*-
 *
 * $Id: genprimes.c,v 1.1 1999/11/19 13:19:37 mdw Exp $
 *
 * Generate prime number table
 *
 * (c) 1999 Straylight/Edgeware
 */

/*----- Licensing notice --------------------------------------------------* 
 *
 * This file is part of Catacomb.
 *
 * Catacomb is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Library General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 * 
 * Catacomb is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Library General Public License for more details.
 * 
 * You should have received a copy of the GNU Library General Public
 * License along with Catacomb; if not, write to the Free
 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 * MA 02111-1307, USA.
 */

/*----- Revision history --------------------------------------------------* 
 *
 * $Log: genprimes.c,v $
 * Revision 1.1  1999/11/19 13:19:37  mdw
 * Generate small primes table.
 *
 */

/*----- Header files ------------------------------------------------------*/

#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <mLib/darray.h>
#include <mLib/dstr.h>
#include <mLib/mdwopt.h>
#include <mLib/quis.h>
#include <mLib/report.h>

/*----- Data structures ---------------------------------------------------*/

DA_DECL(intv, int);

/*----- Main code ---------------------------------------------------------*/

int main(int argc, char *argv[])
{
  int p_max = 0, p_n = 0;
  char *type = "unsigned int";
  char *header = "ptab.h";
  char *source = "ptab.c";
  char *name = "ptab";
  intv p = DA_INIT;
  int i;

  ego(argv[0]);

  for (;;) {
    int i = getopt(argc, argv, "h:c:i:n:m:t:");
    if (i < 0)
      break;
    switch (i) {
      case 'h':
	header = optarg;
	break;
      case 'c':
	source = optarg;
	break;
      case 'i':
	name = optarg;
	break;
      case 'n':
	p_max = 0;
	p_n = atoi(optarg);
	break;
      case 'm':
	p_n = 0;
	p_max = atoi(optarg);
	break;
      case 't':
	type = optarg;
	break;
      default:
	pquis(stderr, "Usage: $ [-n nprimes] [-m maxprime] [-t type]\n");
	exit(EXIT_FAILURE);
    }
  }

  if (!p_max && !p_n)
    die(EXIT_FAILURE, "bad arguments to `-n' or `-m'");

  if (p_max >= 2)
    DA_PUSH(&p, 2);
  for (i = 3; (!p_max && !p_n) ||
	      (p_n && DA_LEN(&p) < p_n) ||
	      (p_max && i <= p_max);
       i += 2) {
    int j;
    for (j = 0; j < DA_LEN(&p); j++) {
      if (i % DA(&p)[j] == 0)
	goto composite;
    }
    DA_PUSH(&p, i);
  composite:;
  }

  {
    FILE *fp = fopen(header, "w");
    dstr d = DSTR_INIT;
    char *q;
    if (!fp)
      die(EXIT_FAILURE, "couldn't write `%s': %s", header, strerror(errno));
    for (q = header; *q; q++) {
      int ch = (unsigned char)*q;
      if (isalnum(ch))
	ch = toupper(ch);
      else
	ch = '_';
      DPUTC(&d, ch);
    }
    DPUTZ(&d);
    fprintf(fp, "\
/* -*-c-*-\n\
 *\n\
 * Table of small prime numbers [generated]\n\
 */\n\
\n\
#ifndef %s\n\
#define %s\n\
\n\
#define NPRIME %lu\n\
#define MAXPRIME %i\n\
\n\
extern %s %s[];\n\
\n\
#endif\n\
",
	    d.buf, d.buf,
	    (unsigned long)DA_LEN(&p),
	    DA(&p)[DA_LEN(&p) - 1],
	    type, name);
    dstr_destroy(&d);
    if (fclose(fp) == EOF) {
      remove(header);
      die(EXIT_FAILURE, "error writing `%s': %s", header, strerror(errno));
    }
  }

  {
    FILE *fp = fopen(source, "w");
    int i;
    if (!fp)
      die(EXIT_FAILURE, "couldn't write `%s': %s", source, strerror(errno));
    fprintf(fp, "\
/* -*-c-*-\n\
 *\n\
 * Table of small prime numbers [generated]\n\
 */\n\
\n\
#include \"%s\"\n\
\n\
%s %s[] = {",
	    header, type, name);
    for (i = 0; i < DA_LEN(&p); i++) {
      if (i % 8 == 0)
	fputs("\n  ", fp);
      fprintf(fp, "%5i, ", DA(&p)[i]);
    }
    fputs("\n\
};\n\
", fp);
    if (fclose(fp) == EOF) {
      remove(source);
      die(EXIT_FAILURE, "error writing `%s': %s", source, strerror(errno));
    }
  }

  return (0);
}

/*----- That's all, folks -------------------------------------------------*/