#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[] = {"trans.dat"};
    long          rec1 = 0;
    long          nrec = 10;
    long          dblk = 10;
    long          ni = 5;
    long          nit[10];
    long          data[50];
    double        minsupp = 10.0;
    double        minconf = 80.0;
    long          maxnir = 5;
    long          maxnr = 20;
    long          nr = 0;
    long          rule[(5+1)*20]; 
    double        stat[2*20];
    int           supptype = 1;
    int           info = 0;

    nagdmc_assoc_data(dblk,nit,data,file,&info);
    if (handle_info("nagdmc_assoc_data",info))
        return 2;

    nagdmc_assoc(rec1,nrec,dblk,ni,nit,data,supptype,minsupp,minconf,maxnir,
                 maxnr,&nr,rule,stat,&info);
    if (handle_info("nagdmc_assoc",info))
        return 2;

    nagdmc_assoc_print(nr,maxnir,rule,stat,&info);
    if (handle_info("nagdmc_assoc_print",info))
        return 2;
        
    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;
}