X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=main.c;h=da9e5fa089503f9c3dfdbb44507316824f113fbd;hb=f4a5716e75bb34407a452675ef662e8d8c95c67f;hp=6f0df725949c9b26fccb613768f74c283bc9e591;hpb=0169095a9c345a65ee9e091df63a0e52d3910d1d;p=matchsticks-search.git

diff --git a/main.c b/main.c
index 6f0df72..da9e5fa 100644
--- a/main.c
+++ b/main.c
@@ -27,6 +27,7 @@
 #include <stdlib.h>
 #include <string.h>
 #include <assert.h>
+#include <unistd.h>
 #include <stdbool.h>
 #include <inttypes.h>
 
@@ -88,6 +89,8 @@ static AdjWord *best_adjmatrix;
 
 static unsigned printcounter;
 
+static int ncpus = 1;
+
 static AdjWord *xalloc_adjmatrix(void) {
   return xmalloc(sizeof(*adjmatrix)*n);
 }
@@ -350,12 +353,7 @@ static void iterate(void) {
   }
 }
 
-int main(int argc, char **argv) {
-  assert(argc==3);
-  n = atoi(argv[1]);
-  m = atoi(argv[2]);
-  prep();
-  iterate();
+static void report(void) {
   fprintf(stderr, "\n");
   if (best_prob) {
     double min = glp_get_obj_val(best_prob);
@@ -383,5 +381,25 @@ int main(int argc, char **argv) {
     }
   }
   if (ferror(stdout) || fclose(stdout)) { perror("stdout"); exit(-1); }
+}
+ 
+int main(int argc, char **argv) {
+  int opt;
+  while ((opt = getopt(argc,argv,"j:")) >= 0) {
+    switch (opt) {
+    case 'j': ncpus = atoi(optarg); break;
+    case '+': assert(!"bad option");
+    default: abort();
+    }
+  }
+  argc -= optind-1;
+  argv += optind-1;
+  assert(argc==3);
+  n = atoi(argv[1]);
+  m = atoi(argv[2]);
+
+  prep();
+  iterate();
+  report();
   return 0;
 }