X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ian/git?p=moebius2.git;a=blobdiff_plain;f=minimise.c;h=bf5b096c89339acdee0428147f503c2f94bfcc41;hp=d7cc86dcd1571fa4e9c375dbc2f47f0eb0d29ee0;hb=cb6eb4221e7a5ac6963a1e9c32d0aa25061be353;hpb=bc3b5b8813e120b28b3148d8c7412fa9937f6cb1

diff --git a/minimise.c b/minimise.c
index d7cc86d..bf5b096 100644
--- a/minimise.c
+++ b/minimise.c
@@ -37,14 +37,16 @@
 #include "half.h"
 #include "minimise.h"
 
-const char *input_file, *output_file;
-char *output_file_tmp;
+const char *input_file, *best_file;
+char *best_file_tmp;
+long long evaluations;
+double stop_epsilon= 1e-20;
 
 static void printing_init(void);
 
 static gsl_multimin_fminimizer *minimiser;
-
-static const double stop_epsilon= 1e-6;
+static const char *final_file;
+static char *final_file_tmp;
 
 static double minfunc_f(const gsl_vector *x, void *params) {
   struct Vertices vs;
@@ -55,6 +57,11 @@ static double minfunc_f(const gsl_vector *x, void *params) {
   return compute_energy(&vs);
 }
 
+static void badusage(void) {
+  fputs("usage: minimise <input> [-i<intermediate>] -o<output\n",stderr);
+  exit(8);
+}
+
 int main(int argc, const char *const *argv) {
   gsl_multimin_function multimin_function;
   double size;
@@ -62,17 +69,28 @@ int main(int argc, const char *const *argv) {
   double initial_half[DIM], step_size[DIM];
   FILE *initial_f;
   gsl_vector initial_gsl, step_size_gsl;
-  int r, i;
-
-  if (argc!=3 || argv[1][0]=='-' || strncmp(argv[2],"-o",2))
-    { fputs("usage: minimise <input> -o<output\n",stderr); exit(8); }
+  int r, i, size_check_counter= 0;
+
+  if (argc==3) {
+    input_file= argv[1];
+    if (strncmp(argv[2],"-o",2)) badusage();   best_file= argv[2]+2;
+    final_file= 0;
+  } else if (argc==4) {
+    input_file= argv[1];
+    if (strncmp(argv[2],"-i",2)) badusage();   best_file= argv[2]+2;
+    if (strncmp(argv[3],"-o",2)) badusage();   final_file= argv[3]+2;
+  } else {
+    badusage();
+  }
+  if (argv[1][0]=='-') badusage();
 
-  input_file= argv[1];
-  output_file= argv[2]+2;
-  if (asprintf(&output_file_tmp,"%s.new",output_file) <= 0) diee("asprintf");
+  if (asprintf(&best_file_tmp,"%s.new",best_file) <= 0) diee("asprintf");
+  if (final_file)
+    if (asprintf(&final_file_tmp,"%s.new",final_file) <= 0) diee("asprintf");
 
   graph_layout_prepare();
   printing_init();
+  energy_init();
 
   printf("X=%d=0x%x  Y=%d=0x%x  DIM=%d\n",X,X,Y,Y,DIM);
   
@@ -108,16 +126,26 @@ int main(int argc, const char *const *argv) {
   for (;;) {
     GA( gsl_multimin_fminimizer_iterate(minimiser) );
 
-    size= gsl_multimin_fminimizer_size(minimiser);
-    r= gsl_multimin_test_size(size, stop_epsilon);
+    if (!(size_check_counter++ % DIM)) {
+      size= gsl_multimin_fminimizer_size(minimiser);
+      r= gsl_multimin_test_size(size, stop_epsilon);
+
+      if (printing_check(pr_size,215))
+	printf("r=%2d, size %# e\n", r, size);
+      flushoutput();
 
-    if (printing_check(pr_size))
-      printf("%*s size %# e, r=%d\n", 135,"", size, r);
-    flushoutput();
+      if (r==GSL_SUCCESS) break;
+      assert(r==GSL_CONTINUE);
+    }
+  }
 
-    if (r==GSL_SUCCESS) break;
-    assert(r==GSL_CONTINUE);
+  if (final_file) {
+    if (unlink(final_file_tmp) && errno != ENOENT) diee("unlink final .tmp");
+    if (link(best_file,final_file_tmp)) diee("link final .tmp");
+    if (rename(final_file_tmp,final_file)) diee("install final");
   }
+  printf("FINISHED (%lld evaluations)\n",evaluations);
+  
   return 0;
 }
 
@@ -126,7 +154,7 @@ int main(int argc, const char *const *argv) {
 static volatile unsigned print_todo;
 static sigset_t print_alarmset;
 
-int printing_check(enum printing_instance which) {
+int printing_check(enum printing_instance which, int indent) {
   static int skipped[pr__max];
   
   unsigned bits= 1u << which;
@@ -141,9 +169,10 @@ int printing_check(enum printing_instance which) {
   print_todo &= ~bits;
   sigprocmask(SIG_UNBLOCK,&print_alarmset,0);
 
+  printf("%*s",indent,"");
   sk= skipped[which];
   if (sk) printf("[%4d] ",sk);
-  else printf("       ");
+  else printf("      ");
   skipped[which]= 0;
 
   return 1;