===================================================================
RCS file: /home/cvs/OpenXM/src/hgm/mh/src/wmain.c,v
retrieving revision 1.2
retrieving revision 1.18
diff -u -p -r1.2 -r1.18
--- OpenXM/src/hgm/mh/src/wmain.c	2013/02/19 08:03:14	1.2
+++ OpenXM/src/hgm/mh/src/wmain.c	2014/03/14 02:21:40	1.18
@@ -1,17 +1,22 @@
 /*
-$OpenXM$
-License: LGPL
- */
+  $OpenXM: OpenXM/src/hgm/mh/src/wmain.c,v 1.17 2014/03/12 07:50:37 takayama Exp $
+  License: LGPL
+*/
 #include <stdio.h>
 #include <stdlib.h>
 #include <math.h>
 #include <string.h>
+#include "sfile.h"
+#include "mh.h"
 #define SMAX 4096
-#define inci(i) { i++; if (i >= argc) { fprintf(stderr,"Option argument is not given.\n"); return(-1); }}
+#define inci(i) { i++; if (i >= argc) { fprintf(stderr,"Option argument is not given.\n"); return(NULL); }}
+int MH_deallocate=0;
+
 extern char *MH_Gfname;
 extern char *MH_Dfname;
 
 /* global variables. They are set in setParam() */
+int MH_byFile=1;
 int MH_RANK;
 int MH_M;
 
@@ -28,85 +33,114 @@ int MH_RawName = 0;
 static int Testrank=0;
 extern int MH_Verbose;
 
+
 extern int MH_P95;  /* 95 % points */
 int mh_gopen_file(void);
-double mh_rkmain(double x0,double y0[],double xn);
 static int setParamTest(void);
 static int setParamDefault(void);
 static int setParam(char *fname);
 static int showParam(void);
+static int next(struct SFILE *fp,char *s, char *msg);
 
 /* #define DEBUG */
 #ifdef DEBUG
 char *MH_Dfname; char *MH_Gfname; double MH_Hg;
 int mh_gopen_file(void) { }
-double mh_rkmain(double x0,double y0[],double xn) { }
+struct MH_RESULT mh_rkmain(double x0,double y0[],double xn) { }
 #endif
 
-void *mh_malloc(int s) {
-  void *p;
-  p = (void*)malloc(s);
-  if (p == NULL) {
-	fprintf(stderr,"No memory.\n"); mh_exit(-1);
-  }
-  return(p);
+void mh_freeWorkArea(void) {
+  extern int MH_deallocate;
+  MH_deallocate=1; /* switch to deallocation mode. */
+  mh_main(0,NULL); 
+  setParam(NULL);  
+  mh_rkmain(0.0, NULL, 0.0);
+  mh_rf(0.0, NULL, 0, NULL, 0);
+  MH_deallocate=0; /* switch to the normal mode. */
 }
-mh_free(void *p) {
-  free(p);
-  return(0);
-}
-static mypower(int x,int n) {
+static int mypower(int x,int n) {
   int a,i;
   a = 1;
   for (i=0; i<n; i++) a = a*x;
   return(a);
 }
-#ifdef STANDALONE
+#ifdef STANDALONE2
 main(int argc,char *argv[]) {
-  return mh_main(argc,char *argv[]);
+  int strategy=STRATEGY_DEFAULT;
+  double err[2]={-1.0,-1.0};
+  int i;
+  for (i=1; i<argc; i++) {
+    if (strcmp(argv[i],"--strategy")==0) {
+      i++; sscanf(argv[i],"%d",&strategy);
+    }else if (strcmp(argv[i],"--abserr")==0) {
+      i++; sscanf(argv[i],"%lg",&(err[0]));
+    }else if (strcmp(argv[i],"--relerr")==0) {
+      i++; sscanf(argv[i],"%lg",&(err[1]));
+    }else ;
+  }
+  mh_set_strategy(strategy,err);
+  mh_exit(MH_RESET_EXIT); /* standalone mode */
+  /*  mh_main(argc,argv);
+      mh_freeWorkArea(); */
+  mh_main(argc,argv);
+  /* showParam(); */
+  return(0);
 }
 #endif
-mh_main(int argc,char *argv[]) {
-  double *y0;
+struct MH_RESULT *mh_main(int argc,char *argv[]) {
+  static double *y0=NULL;
   double x0,xn;
   double ef;
   int i,rank;
+  struct MH_RESULT *rp=NULL;
+  extern int MH_deallocate;
+  extern int MH_byFile;
+  MH_byFile=1;
+  if (MH_deallocate) { if (y0) mh_free(y0); return(rp); }
   setParam(NULL); MH_Gfname = MH_Dfname = NULL; MH_Verbose=1;
   for (i=1; i<argc; i++) {
-	if (strcmp(argv[i],"--idata")==0) {
-	  inci(i);
-	  setParam(argv[i]); MH_Verbose=0;
-	}else if (strcmp(argv[i],"--gnuplotf")==0) {
-	  inci(i);
-	  MH_Gfname = (char *)mh_malloc(SMAX);
-	  strncpy(MH_Gfname,argv[i],SMAX-1);
-	}else if (strcmp(argv[i],"--dataf")==0) {
-	  inci(i);
-	  MH_Dfname = (char *)mh_malloc(SMAX);
-	  strncpy(MH_Dfname,argv[i],SMAX-1);
-	}else if (strcmp(argv[i],"--xmax")==0) {
+    if (strcmp(argv[i],"--idata")==0) {
       inci(i);
-	  sscanf(argv[i],"%lf",&Xng);
-	}else if (strcmp(argv[i],"--step")==0) {
+      setParam(argv[i]); MH_Verbose=0;
+    }else if (strcmp(argv[i],"--gnuplotf")==0) {
       inci(i);
-	  sscanf(argv[i],"%lg",&MH_Hg);
-	}else if (strcmp(argv[i],"--help")==0) {
-	  mh_usage(); return(0);
-	}else if (strcmp(argv[i],"--raw")==0) {
-	  MH_RawName = 1;
-	}else if (strcmp(argv[i],"--test")==0) {
-	  inci(i);
-	  sscanf(argv[i],"%d",&Testrank);
-	  setParamTest();
-	}else if (strcmp(argv[i],"--95")==0) {
-	  MH_P95=1;
-	}else if (strcmp(argv[i],"--verbose")==0) {
-	  MH_Verbose=1;
-	}else {
-	  fprintf(stderr,"Unknown option %s\n",argv[i]);
-	  mh_usage();
-	  return(-1);
-	}
+      MH_Gfname = (char *)mh_malloc(SMAX);
+      strcpy(MH_Gfname,argv[i]);
+    }else if (strcmp(argv[i],"--dataf")==0) {
+      inci(i);
+      MH_Dfname = (char *)mh_malloc(SMAX);
+      strcpy(MH_Dfname,argv[i]);
+    }else if (strcmp(argv[i],"--xmax")==0) {
+      inci(i);
+      sscanf(argv[i],"%lf",&Xng);
+    }else if (strcmp(argv[i],"--step")==0) {
+      inci(i);
+      sscanf(argv[i],"%lg",&MH_Hg);
+    }else if (strcmp(argv[i],"--help")==0) {
+      mh_usage(); return(rp);
+    }else if (strcmp(argv[i],"--raw")==0) {
+      MH_RawName = 1;
+    }else if (strcmp(argv[i],"--test")==0) {
+      inci(i);
+      sscanf(argv[i],"%d",&Testrank);
+      setParamTest();
+    }else if (strcmp(argv[i],"--95")==0) {
+      MH_P95=1;
+    }else if (strcmp(argv[i],"--verbose")==0) {
+      MH_Verbose=1;
+    }else if (strcmp(argv[i],"--bystring")==0) {
+      MH_byFile = 0;
+	}else if (strcmp(argv[i],"--strategy")==0) {
+	  i++; /* ignore */
+	}else if (strcmp(argv[i],"--abserr")==0) {
+	  i++; /* ignore */
+	}else if (strcmp(argv[i],"--relerr")==0) {
+	  i++; /* ignore */
+    }else {
+      fprintf(stderr,"Unknown option %s\n",argv[i]);
+      mh_usage();
+      return(rp);
+    }
   }
   if (MH_Verbose) showParam();
   x0 = MH_X0g;
@@ -117,40 +151,48 @@ mh_main(int argc,char *argv[]) {
   for (i=0; i<rank; i++) y0[i] = ef*Iv[i];
   mh_gopen_file();
   if (MH_Verbose) {for (i=0; i<rank; i++) printf("%lf\n",y0[i]); }
-  mh_rkmain(x0,y0,xn);
+  rp = (struct MH_RESULT*) mh_malloc(sizeof(struct MH_RESULT));
+  rp->message = NULL;
+  *rp=mh_rkmain(x0,y0,xn);
+  return(rp);
 }
 
-mh_usage() {
+int mh_usage() {
   fprintf(stderr,"Usages:\n");
-  fprintf(stderr,"w-m [--idata input_data_file --gnuplotf gnuplot_file_name\n");
+  fprintf(stderr,"hgm_w-n [--idata input_data_file --gnuplotf gnuplot_file_name\n");
   fprintf(stderr," --dataf output_data_file --raw --xmax xmax --test m --step h]\n");
   fprintf(stderr,"[ --95 --verbose] \n");
+  fprintf(stderr,"[ --strategy s --abserr ae --relerr re] \n");
+  fprintf(stderr,"s:0 rk, s:1 adaptive, s:2 adaptive&multiply, see rk.c for the default value of ae and re.\n");
   fprintf(stderr,"--raw does not add data parameters to the output_data_file.\n");
-  fprintf(stderr,"\nThe command w-m [options] evaluates Pr({y | y<xmax}), which is the cumulative distribution function of the largest root of the m by m Wishart matrix with n degrees of freedom and the covariantce matrix sigma.\n");
+  fprintf(stderr,"\nThe command hgm_w-n [options] evaluates Pr({y | y<xmax}), which is the cumulative distribution function of the largest root of the m by m Wishart matrix with n degrees of freedom and the covariantce matrix sigma.\n");
   fprintf(stderr,"All the eigenvalues of sigma must be simple.\n");
   fprintf(stderr,"Parameters are specified by the input_data_file.\n");
-  fprintf(stderr,"The format of the input_data_file.\n");
-  fprintf(stderr," MH_Mg: m, MH_Beta: beta=sigma^(-1)/2 (diagonized), MH_Ng: n, Iv: initial values at MH_X0g*MH_Beta (see our paper how to order them), \n");
+  fprintf(stderr,"The format of the input_data_file, which should be generated by the command hgm_jack-n.\n");
+  fprintf(stderr," MH_Mg: m, MH_Beta: beta=sigma^(-1)/2 (diagonized), MH_Ng: n, MH_X0g: starting value of x,\n");
+  fprintf(stderr," Iv: initial values at MH_X0g*MH_Beta (see our paper how to order them), \n");
   fprintf(stderr," Ef: a scalar factor to the initial value. It may set to 1.\n");
-  fprintf(stderr," MH_Hg: h (step size), MH_X0g: starting value of x, Xng: terminating value of x\n");
+  fprintf(stderr," MH_Hg: h (step size),\n");
   fprintf(stderr," MH_Dp: output data is stored in every MH_Dp steps when output_data_file is specified.\n");
-  fprintf(stderr," --95: output the 95% point. --verbose: verbose mode.\n");
+  fprintf(stderr," Xng: terminating value of x.\n");
+  fprintf(stderr," --95: output the 95%% point. --verbose: verbose mode.\n");
   fprintf(stderr," The line started with %% is a comment line.\n");
-  fprintf(stderr," An example format of the input_data_file can be obtained by executing w-2 with no option.\n");
+  fprintf(stderr," An example format of the input_data_file can be obtained by executing hgm_jack-n with no option.\n");
+  fprintf(stderr,"When --idata option is used, this command is quiet. Use --verbose option if you want to see some messages.\n");
   fprintf(stderr,"\nExamples:\n");
-  fprintf(stderr,"[1] ./w-2 \n");
-  fprintf(stderr,"[2] ./w-2 --xmax 20\n");
-  fprintf(stderr,"[3] ./w-6 --test 6\n");
-  fprintf(stderr,"   A test run of w-6.\n");
-  fprintf(stderr,"   The number m and mm must agree for  ./w-m --test mm.\n");
-  fprintf(stderr,"[4] ./w-5 --idata tmp-inm-m5* --gnuplot tmp-graph\n");
-  fprintf(stderr,"  tmp-inm-m5* is a sample input data distributed with this file.\n");
-  fprintf(stderr,"  tmp-graph-gp.txt is an input file of the gnuplot\n");
-  fprintf(stderr,"  It can be executed with the load command in the gnuplot.\n");
-  fprintf(stderr,"  tmp-graph is the table of x and the values of Pr({y | y<x}).\n");
+  fprintf(stderr,"[1] ./hgm_w-n \n");
+  fprintf(stderr,"[2] ./hgm_w-n --xmax 20\n");
+  fprintf(stderr,"[3] ./hgm_w-n --test 6\n");
+  fprintf(stderr,"   A test run in Mg=6.\n");
+  fprintf(stderr,"[4] ./hgm_jack-n --idata Testdata/tmp-idata3.txt --degree 15 >t.txt\n");
+  fprintf(stderr,"    ./hgm_w-n --idata t.txt --gnuplotf test-g\n");
+  fprintf(stderr,"    gnuplot -persist <test-g-gp.txt\n");
+  fprintf(stderr,"  tmp-idata3.txt is a sample input data distributed with this file.\n");
+  fprintf(stderr,"  test-g-gp.txt is an input file of the gnuplot\n");
+  fprintf(stderr,"  test-g is the table of x and the values of Pr({y | y<x}).\n");
 }
 
-static setParamTest() {
+static int setParamTest() {
   int rank;
   int i;
   extern int Testrank;
@@ -170,7 +212,7 @@ static setParamTest() {
   MH_Dp = 1;
   Xng = 10.0;
 }
-static setParamDefault() {
+static int setParamDefault() {
   int rank;
   MH_M=MH_Mg = 2 ;
   MH_RANK=rank = mypower(2,MH_Mg);
@@ -189,26 +231,34 @@ static setParamDefault() {
   Xng = 10.0;
 }
 
-next(FILE *fp,char *s,char *msg) {
+static int next(struct SFILE *sfp,char *s,char *msg) {
   s[0] = '%';
   while (s[0] == '%') {
-	if (!fgets(s,SMAX,fp)) {
-	  fprintf(stderr,"Data format error at %s\n",msg);
-	  mh_exit(-1);
-	}
-	if (s[0] != '%') return(0);
+    if (!mh_fgets(s,SMAX,sfp)) {
+      fprintf(stderr,"Data format error at %s\n",msg);
+      mh_exit(-1);
+    }
+    if (s[0] != '%') return(0);
   }
 }
-static setParam(char *fname) {
+static int setParam(char *fname) {
   int rank;
   char s[SMAX];
-  FILE *fp;
+  struct SFILE *fp;
   int i;
+  extern int MH_deallocate;
+  extern int MH_byFile;
+  if (MH_deallocate) {
+    if (MH_Beta) mh_free(MH_Beta);
+    if (MH_Ng) mh_free(MH_Ng);
+    if (Iv) mh_free(Iv);
+    return(0);
+  }
   if (fname == NULL) return(setParamDefault());
 
-  if ((fp=fopen(fname,"r")) == NULL) {
-	fprintf(stderr,"File %s is not found.\n",fname);
-	mh_exit(-1);
+  if ((fp=mh_fopen(fname,"r",MH_byFile)) == NULL) {
+    fprintf(stderr,"File %s is not found.\n",fname);
+    mh_exit(-1);
   }
   next(fp,s,"MH_Mg(m)");
   sscanf(s,"%d",&MH_Mg); MH_M=MH_Mg;
@@ -217,7 +267,7 @@ static setParam(char *fname) {
   MH_Beta = (double *)mh_malloc(sizeof(double)*MH_Mg);
   for (i=0; i<MH_Mg; i++) {
     next(fp,s,"MH_Beta");
-	sscanf(s,"%lf",&(MH_Beta[i]));
+    sscanf(s,"%lf",&(MH_Beta[i]));
   }
 
   MH_Ng = (double *)mh_malloc(sizeof(double));
@@ -229,8 +279,8 @@ static setParam(char *fname) {
   
   Iv = (double *)mh_malloc(sizeof(double)*rank);
   for (i=0; i<rank; i++) {
-	next(fp,s,"Iv(initial values)");
-	sscanf(s,"%lg",&(Iv[i]));
+    next(fp,s,"Iv(initial values)");
+    sscanf(s,"%lg",&(Iv[i]));
   }
 
   next(fp,s,"Ef(exponential factor)");
@@ -244,31 +294,29 @@ static setParam(char *fname) {
 
   next(fp,s,"Xng (the last point, cf. --xmax)");
   sscanf(s,"%lf",&Xng);
-  fclose(fp);
+  mh_fclose(fp);
 }
 
-showParam() {
+int showParam() {
   int rank,i;
+  extern int MH_strategy;
+  extern double MH_abserr;
+  extern double MH_relerr;
   rank = mypower(2,MH_Mg);
   printf("%%MH_Mg=\n%d\n",MH_Mg);
   for (i=0; i<MH_Mg; i++) {
-	printf("%%MH_Beta[%d]=\n%lf\n",i,MH_Beta[i]);
+    printf("%%MH_Beta[%d]=\n%lf\n",i,MH_Beta[i]);
   }
   printf("%%MH_Ng=\n%lf\n",*MH_Ng);
   printf("%%MH_X0g=\n%lf\n",MH_X0g);
   for (i=0; i<rank; i++) {
-	printf("%%Iv[%d]=\n%lg\n",i,Iv[i]);
+    printf("%%Iv[%d]=\n%lg\n",i,Iv[i]);
   }
   printf("%%Ef=\n%lf\n",Ef);
   printf("%%MH_Hg=\n%lf\n",MH_Hg);
   printf("%%MH_Dp=\n%d\n",MH_Dp);
   printf("%%Xng=\n%lf\n",Xng);
+  printf("%%strategy=%d\n",MH_strategy);
+  printf("%%abserr=%lg, relerr=%lg\n",MH_abserr,MH_relerr);
 }
 
-mh_exit(int n) {
-#ifdef STANDALONE
-  exit(n);
-#else
-  fprintf(stderr,"Fatal error mh_exit(%d) in mh-w-n.\n",n);
-#endif  
-}