#include <nagdmc.h>

/*
  handle_info() prints to screen information based on the value of the info
  parameter.
*/
int
handle_info(const char func[], int info);

int
main(void) {
	const char    file[] = {"continuous_full.dat"};
    long          rec1 = 0;
    long          nvar = 8;
    long          nrec = 200;
    long          dblk = 200;
    double       *data = 0;
    long          nxvar = 8;
    long          xvar[] = {0,1,2,3,4,5,6,7};
    double       *wt = 0;
    long          c = 3;
    double        alpha = 0.05;
    int           method = 1;
    double        mval = -999.0;
    int           opt = 0;
    long          maxit = 6;
    double        tol = 1.0e-6;
    double       *indwt = 0;
    int           info = 0;
    FILE         *fp = 0;
    long          i, j;

    /*
      Allocate memory for data and load data values.
    */
    if (!(data = (double *)malloc(dblk*nvar * sizeof(double)))) {
        printf(" Memory allocation failure.\n\n");
        return 2;
    }
    
    if ((fp = fopen(file,"r")) == 0) {
        printf(" Data file named %s was not found.\n\n",file);
        return 2;
    }
    
    for (i=0; i<dblk; ++i) {
        for (j=0; j<nvar; ++j) 
            fscanf(fp,"%lf ",&data[i*nvar+j]);
    }

    fclose(fp);

    /*
      Allocate memory for indicative weights.
    */
    if (!(indwt = (double *)malloc(nrec * sizeof(double)))) {
        printf (" Memory allocation failure.\n\n");
        free(data);
        return 2;
    }
    
	/*
      Call BACON function.
    */
	nagdmc_bacon(rec1,nvar,nrec,dblk,data,nxvar,xvar,wt,c,alpha,method,
                 opt,mval,maxit,tol,indwt,&info);

    if (handle_info("nagdmc_bacon",info)) {
        free(data);
        free(indwt);
        return 2;
    }

    /*
      Print the results.
    */
    for (i=0; i<nrec; i++) {
        if (indwt[i] == 0.0)
            printf(" Record %ld suspected outlier \n",i);
        else if (indwt[i] == -1.0)
            printf(" Record %ld contains missing values \n",i);
    }

    free(data);
    free(indwt);

    return 0;
}

int
handle_info(const char func[], int info) {
    if (info == -999)
    {
        printf(" Invalid licence, please contact NAG.\n\n");
        return 2;
    }
    else if (info > 0)
    {
        printf(" Error code %i from %s.\n\n",info,func);
        return 1;
    }
    else if (info < 0)
        printf (" Information code %i from %s.\n\n",info,func);

    return 0;
}