//
//
//

#include <stdio.h>
#include "cbr.h"

int main(int argc, char **argv) {
    
    unsigned int num_cols = 7; 
    unsigned int i;

    /*char * cols[] = {
        "BER", "throughput",
        "mod_scheme", "tx_power",
        "noise_power", "path_loss",
        "utility"
    };*/
    
    const char * xmlfilename;
    if (argc == 1){
        printf("wrong! please specify xml file as parameter.\n");
        return 0;
    } else {
        xmlfilename = argv[1];
    }

    ezxml_t f2 = ezxml_parse_file(xmlfilename), engine2;
    ezxml_t utilities, utility;
    ezxml_t parameters, parameter, affect;
    ezxml_t observables, observable;
    const char *enginename, *enginefilename;
    const char *utilityname[10], *utilityunits[10], *utilitygoal[10], *utilitytarget[10];
    const char *parametername[10], *parameterunits[10], *parametermin[10], *parametermax[10], *parameterstep[10];
    const char *affectutility_p[10], *affectrelationship_p[10];
    const char *observablename[10];
    const char *affectutility_o[10], *affectrelationship_o[10];

    unsigned int i_u;
    unsigned int i_p, i_pa;
    unsigned int i_o, i_oa;

    for (engine2 = ezxml_child(f2, "engine"); engine2; engine2 = engine2->next){
        // get engine name
        enginename = ezxml_attr(engine2, "name");

        // get engine file name
        enginefilename = ezxml_attr(engine2, "filename");
        //printf("engine name: %s, engine file name: %s\n", enginename, enginefilename);
        
        // get utilities
        utilities = ezxml_child(engine2, "utilities");
        i_u = 0;    // index for utilities
        //printf("utilities: ");
        for (utility = ezxml_child(utilities, "utility"); utility; utility = utility->next){
            utilityname[i_u] = ezxml_attr(utility, "name");
            utilityunits[i_u] = ezxml_attr(utility, "units");
            utilitygoal[i_u] = ezxml_attr(utility, "goal");
            utilitytarget[i_u] = ezxml_attr(utility, "target");
            //printf(" %s", utilityname[i_u]);
            i_u++;
        }
        //printf("\n");
        //printf("number of utilities: %d\n", i_u);

        // get parameters
        parameters = ezxml_child(engine2, "parameters");
        i_p = 0;    // index for parameters
        i_pa = 0;   // index for affect
        //printf("parameters: ");
        for (parameter = ezxml_child(parameters, "parameter"); parameter; parameter = parameter->next){
            parametername[i_p] = ezxml_attr(parameter, "name");
            parameterunits[i_p] = ezxml_attr(parameter, "units");
            parametermin[i_p] = ezxml_attr(parameter, "min");
            parametermax[i_p] = ezxml_attr(parameter, "max");
            parameterstep[i_p] = ezxml_attr(parameter, "step");
            //printf(" %s", parametername[i_p]);
            i_p++;
            for (affect = ezxml_child(parameter, "affect"); affect; affect = affect->next){
                affectutility_p[i_pa] = ezxml_attr(affect, "utility");
                affectrelationship_p[i_pa] = ezxml_attr(affect, "relationship");
                i_pa++;
            }
        }
        //printf("\n");
        //printf("number of parameters: %d\n", i_p);
        //printf("number of affects: %d\n", i_pa);

        // get observables
        observables = ezxml_child(engine2, "observables");
        i_o = 0;    // index for observables
        i_oa = 0;   // index for affect
        //printf("observables: ");
        for (observable = ezxml_child(observables, "observable"); observable; observable = observable->next){
            observablename[i_o] = ezxml_attr(observable, "name");
            //printf(" %s", observablename[i_o]);
            i_o++;
            for (affect = ezxml_child(observable, "affect"); affect; affect = affect->next){
                affectutility_o[i_oa] = ezxml_attr(affect, "utility");
                affectrelationship_o[i_oa] = ezxml_attr(affect, "relationship");
                i_oa++;
            }
        }
        //printf("\n");
        //printf("number of observables: %d\n", i_o);
        //printf("number of affects: %d\n", i_oa);
    }

    free(f2);
    
    char * cols[num_cols];
    for (i = 0; i < i_u; i++){
        cols[i] = utilityname[i];
    }
    for (; i < i_u+i_p; i++){
        cols[i] = parametername[i-i_u];
    }
    for (; i < i_u+i_p+i_o; i++){
        cols[i] = observablename[i-i_u-i_p];
    }
   
    
    printf("column names:");
    for (i = 0; i<num_cols; i++){
        printf(" %s",cols[i]);
    }
    printf("\n");

    float vals[num_cols];
    vals[0] = 1e-3f;    // BER
    vals[1] = 10e3f;    // throughput
    vals[2] = 1;        // mod_scheme
    vals[3] = -3.0f;    // tx_power
    vals[4] = -50.0f;   // noise_power
    vals[5] = 125.0f;   // path_loss
    vals[6] = 0.762;    // utility
    int rc;

    // create cbr database/table
    cbr mycbr = cbr_create("ex1", "data", cols, num_cols);

    // add row here
    rc = cbr_add_row(mycbr, cols, vals, num_cols);

    // print
    cbr_print(mycbr);

    // simple search: find entry where...
    //   BER < 1e-2 and path_loss > 100
    char * search_names[] = {"BER", "path_loss"};
    int search_ops[] = {LT, GT};
    float search_vals[] = {1e-2f, 100.f};
    float retvals[num_cols];
    rc = cbr_search(mycbr, search_names, search_ops, search_vals, 2, retvals);

    // clean up
    cbr_free(mycbr);

    printf("done.\n");
    return 0;
}