#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);

/*
  imputed_values() is the function described in 'Explanatory Code' in 
  nagdmc_impute_dist.pdf.
*/
void 
imputed_values(long nvar, double data[], long nrepl, long indexes[]);

int
main(void) {
	const char    file[] = {"categorical.dat"};
    long          rec1 = 0;
    long          nvar = 7;
    long          nrec = 100;
    long          dblk = 100;
	double       *data = 0;
	long          ncat[] = {2,2,2,3,5,10,11};
	long          maxcat = 11;
	long          cat[] = {1, 2,-1,-1,-1,-1,-1,-1,-1,-1,-1,
                           1, 2,-1,-1,-1,-1,-1,-1,-1,-1,-1,
                           1, 2,-1,-1,-1,-1,-1,-1,-1,-1,-1,
                           1, 2, 3,-1,-1,-1,-1,-1,-1,-1,-1,
                           1, 2, 3, 4, 5,-1,-1,-1,-1,-1,-1,
                           0, 1, 2, 3, 4, 5, 6, 7, 8, 9,-1,
                           0, 1, 2, 3, 4, 5, 6, 7, 8, 9,10};
    double       *wts = 0;
	double        mval = -999.0;
	int           stype[] = {0,-1,-1};
	int           randsel = 0;
	long          ndonors = 5;
	long          matchall = -1;
	int           reuse = 1;
	double        minwt = 0.0;
	double       *R = 0;
	int           dtype[] = {1,1,1,1,1,1,1};
	double       *udist = 0;
	long          nrepl = 0;
    long          nfail = 0;
    long          noimp = 0;
	int           info = 0;
    long         *indexes = 0;
    FILE         *fp = 0;
    long          i, j;

    /*
      Allocate memory and load data values.
    */
    if (!(data = (double *)malloc(dblk*nvar * sizeof(double)))) {
        printf(" Memory allocation failure.\n\n");
        return 2;
    }
    
    if (maxcat > 0) {
        if (!(udist = (double *)malloc(nvar*maxcat*maxcat * sizeof(double)))) {
            printf(" Memory allocation error.\n\n");
            free(data);
            return 2;
        }
    }

    /*
      Read data values.
    */
    if ((fp = fopen(file,"r")) == 0) {
        printf(" Data file named %s was not found.\n\n",file);
        free(data);
        if (udist)
            free(udist);
        return 2;
    }
    
    for (i=0; i<dblk; ++i) {
        for (j=0; j<nvar; ++j) 
            fscanf(fp,"%lf ",&data[i*nvar+j]);
    }

    fclose(fp);

	/*
      Run imputation algorithm.
    */
    indexes = nagdmc_impute_dist(rec1,nvar,nrec,dblk,data,ncat,maxcat,cat,wts,
                                 mval,stype,randsel,ndonors,matchall,reuse,minwt,
                                 R,dtype,udist,&nrepl,&nfail,&noimp,&info);

    if (handle_info("nagdmc_impute_dist",info)) {
        free(data);
        if (udist)
            free(udist);
        nagdmc_free_impute(indexes);
        return 2;
    }

    /*
      Explanatory code from function document.
    */
    imputed_values(nvar,data,nrepl,indexes);

    /*
      Return allocated memory to operating system.
    */
	free(data);
	if (udist)
        free(udist);
    nagdmc_free_impute(indexes);

    return 0;
}

void
imputed_values(long nvar, double data[], long nrepl, long indexes[]) {
    long i, j, k, l;

#define MISSING_ROW(I) indexes[I]
#define MISSING_COL(I) indexes[I+nrepl]
#define DONOR_REC(I) indexes[I+2*nrepl]
#define DATA(I,J) data[(I)*nvar+J]

    printf("\n\tRow \tCol \tValue \tDonor Record\n");
    for (i=0; i<nrepl; ++i) {
        j = MISSING_ROW(i);
        k = MISSING_COL(i);
        l = DONOR_REC(i);
        printf("\t%-4li\t%-4li\t%-6.3f\t%-li\n",j,k,DATA(j,k),l);
    }
}

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