[BACK]Return to wmain.c CVS log [TXT][DIR] Up to [local] / OpenXM / src / hgm / mh / src

Annotation of OpenXM/src/hgm/mh/src/wmain.c, Revision 1.15

1.1       takayama    1: /*
1.15    ! takayama    2:   $OpenXM: OpenXM/src/hgm/mh/src/wmain.c,v 1.14 2014/02/25 05:25:39 takayama Exp $
1.10      takayama    3:   License: LGPL
                      4: */
1.1       takayama    5: #include <stdio.h>
                      6: #include <stdlib.h>
                      7: #include <math.h>
                      8: #include <string.h>
1.4       takayama    9: #include "sfile.h"
1.11      takayama   10: #include "mh.h"
1.1       takayama   11: #define SMAX 4096
1.4       takayama   12: #define inci(i) { i++; if (i >= argc) { fprintf(stderr,"Option argument is not given.\n"); return(NULL); }}
1.3       takayama   13: int MH_deallocate=0;
                     14:
1.1       takayama   15: extern char *MH_Gfname;
                     16: extern char *MH_Dfname;
                     17:
                     18: /* global variables. They are set in setParam() */
1.4       takayama   19: int MH_byFile=1;
1.1       takayama   20: int MH_RANK;
                     21: int MH_M;
                     22:
                     23: int MH_Mg;  /* m */
                     24: double *MH_Beta; /* beta[0], ..., beta[m-1] */
1.2       takayama   25: double *MH_Ng;   /* freedom n.  c=(m+1)/2+n/2; Note that it is a pointer */
                     26: double MH_X0g;   /* initial point */
                     27: static double *Iv;   /* Initial values of mhg sorted by mhbase() in rd.rr at beta*x0 */
                     28: static double Ef;   /* exponential factor at beta*x0 */
                     29: extern double MH_Hg;   /* step size of rk defined in rk.c */
                     30: int MH_Dp;      /* Data sampling period */
                     31: static double Xng=0.0;   /* the last point */
                     32: int MH_RawName = 0;
                     33: static int Testrank=0;
                     34: extern int MH_Verbose;
                     35:
                     36: extern int MH_P95;  /* 95 % points */
                     37: int mh_gopen_file(void);
                     38: static int setParamTest(void);
                     39: static int setParamDefault(void);
                     40: static int setParam(char *fname);
                     41: static int showParam(void);
1.6       takayama   42: static int next(struct SFILE *fp,char *s, char *msg);
1.1       takayama   43:
                     44: /* #define DEBUG */
                     45: #ifdef DEBUG
1.2       takayama   46: char *MH_Dfname; char *MH_Gfname; double MH_Hg;
                     47: int mh_gopen_file(void) { }
1.4       takayama   48: struct MH_RESULT mh_rkmain(double x0,double y0[],double xn) { }
1.1       takayama   49: #endif
                     50:
1.3       takayama   51: void mh_freeWorkArea(void) {
                     52:   extern int MH_deallocate;
                     53:   MH_deallocate=1; /* switch to deallocation mode. */
                     54:   mh_main(0,NULL);
                     55:   setParam(NULL);
                     56:   mh_rkmain(0.0, NULL, 0.0);
                     57:   mh_rf(0.0, NULL, 0, NULL, 0);
                     58:   MH_deallocate=0; /* switch to the normal mode. */
                     59: }
1.8       takayama   60: static int mypower(int x,int n) {
1.1       takayama   61:   int a,i;
                     62:   a = 1;
                     63:   for (i=0; i<n; i++) a = a*x;
                     64:   return(a);
                     65: }
1.9       takayama   66: #ifdef STANDALONE2
1.1       takayama   67: main(int argc,char *argv[]) {
1.12      takayama   68:   int strategy=STRATEGY_DEFAULT;
                     69:   double err[2]={-1.0,-1.0};
                     70:   int i;
                     71:   for (i=1; i<argc; i++) {
                     72:     if (strcmp(argv[i],"--strategy")==0) {
                     73:       i++; sscanf(argv[i],"%d",&strategy);
                     74:     }else if (strcmp(argv[i],"--abserr")==0) {
                     75:       i++; sscanf(argv[i],"%lg",&(err[0]));
                     76:     }else if (strcmp(argv[i],"--relerr")==0) {
                     77:       i++; sscanf(argv[i],"%lg",&(err[1]));
                     78:     }else ;
                     79:   }
                     80:   mh_set_strategy(strategy,err);
1.6       takayama   81:   mh_exit(MH_RESET_EXIT); /* standalone mode */
1.10      takayama   82:   /*  mh_main(argc,argv);
                     83:       mh_freeWorkArea(); */
1.3       takayama   84:   mh_main(argc,argv);
1.12      takayama   85:   /* showParam(); */
1.2       takayama   86: }
                     87: #endif
1.4       takayama   88: struct MH_RESULT *mh_main(int argc,char *argv[]) {
1.13      takayama   89:   static double *y0=NULL;
1.1       takayama   90:   double x0,xn;
                     91:   double ef;
                     92:   int i,rank;
1.4       takayama   93:   struct MH_RESULT *rp=NULL;
1.3       takayama   94:   extern int MH_deallocate;
1.4       takayama   95:   extern int MH_byFile;
                     96:   MH_byFile=1;
                     97:   if (MH_deallocate) { if (y0) mh_free(y0); return(rp); }
1.2       takayama   98:   setParam(NULL); MH_Gfname = MH_Dfname = NULL; MH_Verbose=1;
1.1       takayama   99:   for (i=1; i<argc; i++) {
1.10      takayama  100:     if (strcmp(argv[i],"--idata")==0) {
1.1       takayama  101:       inci(i);
1.10      takayama  102:       setParam(argv[i]); MH_Verbose=0;
                    103:     }else if (strcmp(argv[i],"--gnuplotf")==0) {
1.1       takayama  104:       inci(i);
1.10      takayama  105:       MH_Gfname = (char *)mh_malloc(SMAX);
                    106:       strcpy(MH_Gfname,argv[i]);
                    107:     }else if (strcmp(argv[i],"--dataf")==0) {
                    108:       inci(i);
                    109:       MH_Dfname = (char *)mh_malloc(SMAX);
                    110:       strcpy(MH_Dfname,argv[i]);
                    111:     }else if (strcmp(argv[i],"--xmax")==0) {
                    112:       inci(i);
                    113:       sscanf(argv[i],"%lf",&Xng);
                    114:     }else if (strcmp(argv[i],"--step")==0) {
                    115:       inci(i);
                    116:       sscanf(argv[i],"%lg",&MH_Hg);
                    117:     }else if (strcmp(argv[i],"--help")==0) {
                    118:       mh_usage(); return(rp);
                    119:     }else if (strcmp(argv[i],"--raw")==0) {
                    120:       MH_RawName = 1;
                    121:     }else if (strcmp(argv[i],"--test")==0) {
                    122:       inci(i);
                    123:       sscanf(argv[i],"%d",&Testrank);
                    124:       setParamTest();
                    125:     }else if (strcmp(argv[i],"--95")==0) {
                    126:       MH_P95=1;
                    127:     }else if (strcmp(argv[i],"--verbose")==0) {
                    128:       MH_Verbose=1;
                    129:     }else if (strcmp(argv[i],"--bystring")==0) {
                    130:       MH_byFile = 0;
1.12      takayama  131:        }else if (strcmp(argv[i],"--strategy")==0) {
                    132:          i++; /* ignore */
                    133:        }else if (strcmp(argv[i],"--abserr")==0) {
                    134:          i++; /* ignore */
                    135:        }else if (strcmp(argv[i],"--relerr")==0) {
                    136:          i++; /* ignore */
1.10      takayama  137:     }else {
                    138:       fprintf(stderr,"Unknown option %s\n",argv[i]);
                    139:       mh_usage();
                    140:       return(rp);
                    141:     }
1.1       takayama  142:   }
1.2       takayama  143:   if (MH_Verbose) showParam();
                    144:   x0 = MH_X0g;
1.1       takayama  145:   xn = Xng;
                    146:   ef = Ef;
                    147:   rank = mypower(2,MH_Mg);
1.2       takayama  148:   y0 = (double *) mh_malloc(sizeof(double)*rank);
1.1       takayama  149:   for (i=0; i<rank; i++) y0[i] = ef*Iv[i];
1.2       takayama  150:   mh_gopen_file();
                    151:   if (MH_Verbose) {for (i=0; i<rank; i++) printf("%lf\n",y0[i]); }
1.4       takayama  152:   rp = (struct MH_RESULT*) mh_malloc(sizeof(struct MH_RESULT));
                    153:   *rp=mh_rkmain(x0,y0,xn);
                    154:   return(rp);
1.1       takayama  155: }
                    156:
1.8       takayama  157: int mh_usage() {
1.1       takayama  158:   fprintf(stderr,"Usages:\n");
1.14      takayama  159:   fprintf(stderr,"hgm_w-n [--idata input_data_file --gnuplotf gnuplot_file_name\n");
1.1       takayama  160:   fprintf(stderr," --dataf output_data_file --raw --xmax xmax --test m --step h]\n");
                    161:   fprintf(stderr,"[ --95 --verbose] \n");
                    162:   fprintf(stderr,"--raw does not add data parameters to the output_data_file.\n");
1.14      takayama  163:   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");
1.1       takayama  164:   fprintf(stderr,"All the eigenvalues of sigma must be simple.\n");
                    165:   fprintf(stderr,"Parameters are specified by the input_data_file.\n");
1.14      takayama  166:   fprintf(stderr,"The format of the input_data_file, which should be generated by the command hgm_jack-n.\n");
                    167:   fprintf(stderr," MH_Mg: m, MH_Beta: beta=sigma^(-1)/2 (diagonized), MH_Ng: n, MH_X0g: starting value of x,\n");
                    168:   fprintf(stderr," Iv: initial values at MH_X0g*MH_Beta (see our paper how to order them), \n");
1.1       takayama  169:   fprintf(stderr," Ef: a scalar factor to the initial value. It may set to 1.\n");
1.14      takayama  170:   fprintf(stderr," MH_Hg: h (step size),\n");
1.2       takayama  171:   fprintf(stderr," MH_Dp: output data is stored in every MH_Dp steps when output_data_file is specified.\n");
1.14      takayama  172:   fprintf(stderr," Xng: terminating value of x.\n");
1.7       takayama  173:   fprintf(stderr," --95: output the 95%% point. --verbose: verbose mode.\n");
1.1       takayama  174:   fprintf(stderr," The line started with %% is a comment line.\n");
1.15    ! takayama  175:   fprintf(stderr," An example format of the input_data_file can be obtained by executing hgm_jack-n with no option.\n");
        !           176:   fprintf(stderr,"When --idata option is used, this command is quiet. Use --verbose option if you want to see some messages.\n");
1.1       takayama  177:   fprintf(stderr,"\nExamples:\n");
1.14      takayama  178:   fprintf(stderr,"[1] ./hgm_w-n \n");
                    179:   fprintf(stderr,"[2] ./hgm_w-n --xmax 20\n");
                    180:   fprintf(stderr,"[3] ./hgm_w-n --test 6\n");
                    181:   fprintf(stderr,"   A test run in Mg=6.\n");
                    182:   fprintf(stderr,"[4] ./hgm_jack-n --idata Testdata/tmp-idata3.txt --degree 15 >t.txt\n");
                    183:   fprintf(stderr,"    ./hgm_w-n --idata t.txt --gnuplotf test-g\n");
                    184:   fprintf(stderr,"    gnuplot -persist <test-g-gp.txt\n");
                    185:   fprintf(stderr,"  tmp-idata3.txt is a sample input data distributed with this file.\n");
                    186:   fprintf(stderr,"  test-g-gp.txt is an input file of the gnuplot\n");
                    187:   fprintf(stderr,"  test-g is the table of x and the values of Pr({y | y<x}).\n");
1.1       takayama  188: }
                    189:
1.8       takayama  190: static int setParamTest() {
1.1       takayama  191:   int rank;
                    192:   int i;
                    193:   extern int Testrank;
1.2       takayama  194:   extern int MH_Verbose;
                    195:   MH_Verbose=1;
1.1       takayama  196:   MH_M= MH_Mg = Testrank ;
                    197:   MH_RANK = rank = mypower(2,MH_Mg);
1.2       takayama  198:   MH_Beta = (double *)mh_malloc(sizeof(double)*MH_Mg);
1.1       takayama  199:   for (i=0; i<MH_Mg; i++) MH_Beta[i] = 1.0+0.1*i;
1.2       takayama  200:   MH_Ng = (double *)mh_malloc(sizeof(double)); *MH_Ng = 3.0;
                    201:   Iv = (double *)mh_malloc(sizeof(double)*rank);
1.1       takayama  202:   for (i=0; i<rank; i++) Iv[i] = 0;
                    203:   Iv[0] = 0.001;
                    204:   Ef = 1;
1.2       takayama  205:   MH_X0g = 0.3;
                    206:   MH_Hg = 0.001;
                    207:   MH_Dp = 1;
1.1       takayama  208:   Xng = 10.0;
                    209: }
1.8       takayama  210: static int setParamDefault() {
1.1       takayama  211:   int rank;
                    212:   MH_M=MH_Mg = 2 ;
                    213:   MH_RANK=rank = mypower(2,MH_Mg);
1.2       takayama  214:   MH_Beta = (double *)mh_malloc(sizeof(double)*MH_Mg);
1.1       takayama  215:   MH_Beta[0] = 1.0; MH_Beta[1] = 2.0;
1.2       takayama  216:   MH_Ng = (double *)mh_malloc(sizeof(double)); *MH_Ng = 3.0;
                    217:   Iv = (double *)mh_malloc(sizeof(double)*rank);
1.1       takayama  218:   Iv[0] = 1.58693;
                    219:   Iv[1] = 0.811369;
                    220:   Iv[2] = 0.846874;
                    221:   Iv[3] = 0.413438;
                    222:   Ef = 0.01034957388338225707;
1.2       takayama  223:   MH_X0g = 0.3;
                    224:   MH_Hg = 0.001;
                    225:   MH_Dp = 1;
1.1       takayama  226:   Xng = 10.0;
                    227: }
                    228:
1.8       takayama  229: static int next(struct SFILE *sfp,char *s,char *msg) {
1.1       takayama  230:   s[0] = '%';
                    231:   while (s[0] == '%') {
1.10      takayama  232:     if (!mh_fgets(s,SMAX,sfp)) {
                    233:       fprintf(stderr,"Data format error at %s\n",msg);
                    234:       mh_exit(-1);
                    235:     }
                    236:     if (s[0] != '%') return(0);
1.1       takayama  237:   }
                    238: }
1.8       takayama  239: static int setParam(char *fname) {
1.1       takayama  240:   int rank;
                    241:   char s[SMAX];
1.4       takayama  242:   struct SFILE *fp;
1.1       takayama  243:   int i;
1.3       takayama  244:   extern int MH_deallocate;
1.4       takayama  245:   extern int MH_byFile;
1.3       takayama  246:   if (MH_deallocate) {
1.10      takayama  247:     if (MH_Beta) mh_free(MH_Beta);
                    248:     if (MH_Ng) mh_free(MH_Ng);
                    249:     if (Iv) mh_free(Iv);
                    250:     return(0);
1.3       takayama  251:   }
1.1       takayama  252:   if (fname == NULL) return(setParamDefault());
                    253:
1.4       takayama  254:   if ((fp=mh_fopen(fname,"r",MH_byFile)) == NULL) {
1.10      takayama  255:     fprintf(stderr,"File %s is not found.\n",fname);
                    256:     mh_exit(-1);
1.1       takayama  257:   }
                    258:   next(fp,s,"MH_Mg(m)");
                    259:   sscanf(s,"%d",&MH_Mg); MH_M=MH_Mg;
                    260:   MH_RANK=rank = mypower(2,MH_Mg);
                    261:
1.2       takayama  262:   MH_Beta = (double *)mh_malloc(sizeof(double)*MH_Mg);
1.1       takayama  263:   for (i=0; i<MH_Mg; i++) {
                    264:     next(fp,s,"MH_Beta");
1.10      takayama  265:     sscanf(s,"%lf",&(MH_Beta[i]));
1.1       takayama  266:   }
                    267:
1.2       takayama  268:   MH_Ng = (double *)mh_malloc(sizeof(double));
                    269:   next(fp,s,"MH_Ng(freedom parameter n)");
                    270:   sscanf(s,"%lf",MH_Ng);
1.1       takayama  271:
1.2       takayama  272:   next(fp,s,"MH_X0g(initial point)");
                    273:   sscanf(s,"%lf",&MH_X0g);
1.1       takayama  274:
1.2       takayama  275:   Iv = (double *)mh_malloc(sizeof(double)*rank);
1.1       takayama  276:   for (i=0; i<rank; i++) {
1.10      takayama  277:     next(fp,s,"Iv(initial values)");
                    278:     sscanf(s,"%lg",&(Iv[i]));
1.1       takayama  279:   }
                    280:
                    281:   next(fp,s,"Ef(exponential factor)");
                    282:   sscanf(s,"%lg",&Ef);
                    283:
1.2       takayama  284:   next(fp,s,"MH_Hg (step size of rk)");
                    285:   sscanf(s,"%lg",&MH_Hg);
1.1       takayama  286:
1.2       takayama  287:   next(fp,s,"MH_Dp (data sampling period)");
                    288:   sscanf(s,"%d",&MH_Dp);
1.1       takayama  289:
                    290:   next(fp,s,"Xng (the last point, cf. --xmax)");
                    291:   sscanf(s,"%lf",&Xng);
1.4       takayama  292:   mh_fclose(fp);
1.1       takayama  293: }
                    294:
1.8       takayama  295: int showParam() {
1.1       takayama  296:   int rank,i;
1.12      takayama  297:   extern int MH_strategy;
                    298:   extern double MH_abserr;
                    299:   extern double MH_relerr;
1.1       takayama  300:   rank = mypower(2,MH_Mg);
                    301:   printf("%%MH_Mg=\n%d\n",MH_Mg);
                    302:   for (i=0; i<MH_Mg; i++) {
1.10      takayama  303:     printf("%%MH_Beta[%d]=\n%lf\n",i,MH_Beta[i]);
1.1       takayama  304:   }
1.2       takayama  305:   printf("%%MH_Ng=\n%lf\n",*MH_Ng);
                    306:   printf("%%MH_X0g=\n%lf\n",MH_X0g);
1.1       takayama  307:   for (i=0; i<rank; i++) {
1.10      takayama  308:     printf("%%Iv[%d]=\n%lg\n",i,Iv[i]);
1.1       takayama  309:   }
                    310:   printf("%%Ef=\n%lf\n",Ef);
1.2       takayama  311:   printf("%%MH_Hg=\n%lf\n",MH_Hg);
                    312:   printf("%%MH_Dp=\n%d\n",MH_Dp);
1.1       takayama  313:   printf("%%Xng=\n%lf\n",Xng);
1.12      takayama  314:   printf("%%strategy=%d\n",MH_strategy);
                    315:   printf("%%abserr=%lg, relerr=%lg\n",MH_abserr,MH_relerr);
1.1       takayama  316: }
                    317:

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>