From: mdw <mdw>
Date: Fri, 19 Nov 1999 13:19:37 +0000 (+0000)
Subject: Generate small primes table.
X-Git-Url: https://git.distorted.org.uk/u/mdw/catacomb/commitdiff_plain/8ecfe0dc4e8d52324a9f839f02e3ca2fad417c51

Generate small primes table.
---

diff --git a/genprimes.c b/genprimes.c
new file mode 100644
index 0000000..4474110
--- /dev/null
+++ b/genprimes.c
@@ -0,0 +1,193 @@
+/* -*-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 -------------------------------------------------*/