#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[] = {"iris.dat"};
    long          rec1 = 0;
    long          nrec = 150;
    long          nvar = 5;
    long          dblk = 150;
    double       *data = 0;
    long          nxvar = 4;
    long          xvar[] = {0,1,2,3};
    long          yvar = 4;
    long          nhid = 5;
    int           phi = 2;
    int           psi = 0;
    double        gain = 1.0;
    long          nit = 50;
    long          nit2 = 30;
    long          ninit = 5;
    long          iseed = 12;
    long         *val = 0;
    double       *vdata = 0;
    int           info = 0;
    double       *model = 0;
    double       *res = 0;
    double        objf = 0.0;
    FILE         *fp = 0;
    long          i, j, n;

#define DATA(I,J) data[(I)*nvar+J]

    /*
      Allocate memory.
    */
    if (!(data = (double *)malloc(dblk*nvar * sizeof(double)))) {
        printf(" Memory allocation failure.\n");
        return 2;
    }

    n = 7 + nxvar + 2 + nxvar * nhid + nhid;

    if (!(model = (double *)malloc(n * sizeof(double)))) {
        printf(" Memory allocation failure.\n");
        free(data);
        return 2;
    }

    if (!(res = (double *)malloc(nrec * sizeof(double)))) {
        printf(" Memory allocation failure.\n");
        free(data);
        free(model);
        return 2;
    }

    /*
      Read data values.
    */
    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);

    /*
      Scale data on (continuous) independent variables.
    */
    nagdmc_scale(rec1,nvar,nrec,dblk,data,nxvar,xvar,&info);

    if (handle_info("nagdmc_scale",info)) {
        free(data);
        free(model);
        free(res);
        return 2;
    }
    
    /*
      Set validation data.
    */
    if (!val)
        vdata = 0;
    else
        vdata = data + (rec1+nrec-val[0])*nvar;

    /*
      Fit MLP model.
    */
    nagdmc_mlp(rec1,nvar,nrec,dblk,data,nxvar,xvar,yvar,nhid,phi,psi,gain,
               &nit,nit2,ninit,iseed,model,val,vdata,res,&objf,&info);

    if (handle_info("nagdmc_mlp",info)) {
        free(data);
        free(model);
        free(res);
        return 2;
    }

    /*
      Print approximation results.
    */
    printf("\n Objective function: %8.4f\n",objf);
    printf ("\n No. iterations: %li\n",nit);
    
    printf("\n %12s  %8s\n","Fitted value","Residual");
    for (i=0; i<nrec; ++i)
        printf("   %8.4f    %8.4f\n",res[i],res[i]-data[(rec1+i)*nvar+yvar]);
    printf("\n");

    /*
      Return allocated memory to operating system.
    */
    free(data);
    free(model);
    free(res);

    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;
}