root/vtcross/branches/trnewman/CR_engines/CBR/src/main_cognitive_engine.c @ 79

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
//#include <iostream>
#include <stdlib.h>
#include <string.h>

#include "cbr.h"

//using namespace std;

struct CE_Info {
    int numUtilities;
    int numParameters;
    int numObservables;
};

struct Utility {
    char name[50];
    char units[50];
    char goal[50];
   /* string name;
    string units;
    string goal;*/
    float target;
    float value;
};

struct Affect {
    struct Utility u; 
    //string relation;
    char relation[50];
};      

struct Parameter {
    /*string name;
    string units;*/
    char name[50];
    char units[50];
    float min;
    int numAffects; 
    struct Affect affection_list[10];
    float max;
    float step;
    float value;
};      
        
struct Observable {
    //string name;
    char name[50];
    struct Affect affection_list[10];
    int numAffects;
    float value;
};

// error handling
void error(char *msg)
{
    perror(msg);
    exit(0);
}

void ReadMessage(int socket,char * buffer) {
    int i,n;

    n = recv(socket,buffer,256,MSG_PEEK);
    for(i=0;i<256;i++){
            if(strcmp(&buffer[i],"\0") == 0) break;
    }
    n = recv(socket,buffer,i+1,0);
    if (n < 0) 
         error("ERROR reading from socket");
//    printf("ReadMessage:%s %d\n",buffer,n);
}


int GetUtility(int sockfd, struct Utility uList[], struct CE_Info *ce_info)
{
    char buffer[256];
    int i;

    // read unitilities
    // numUtilities
    bzero(buffer,256);
    ReadMessage(sockfd,buffer);
    ce_info->numUtilities = atoi(buffer);
    //printf("number of utilities: %d\n", ce_info->numUtilities);
        
    for (i = 0; i < ce_info->numUtilities; i++){
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
            //printf("Name: %s\n", buffer);
        strcpy(uList[i].name, buffer);
                 
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
            //printf("Units: %s\n", buffer);
        strcpy(uList[i].units, buffer);

        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
        //printf("Goal: %s\n", buffer);
        strcpy(uList[i].goal, buffer);
        
            bzero(buffer,256);
            ReadMessage(sockfd,buffer);
        //printf("Target: %s\n", buffer);
        uList[i].target = atof(buffer);
    }
    return 1;
}


int GetParameter(int sockfd, struct Parameter pList[], struct CE_Info *ce_info)
{
    char buffer[256];
    int i, j;

    // numParameters
    bzero(buffer,256);
    ReadMessage(sockfd,buffer);
    ce_info->numParameters = atoi(buffer);
    //printf("number of parameters: %d\n", ce_info->numParameters);
        
    for (i = 0; i < ce_info->numParameters; i++){
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
            //printf("Name: %s\n", buffer);
        strcpy(pList[i].name, buffer);
                 
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
            //printf("Units: %s\n", buffer);
        strcpy(pList[i].units, buffer);

        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
        //printf("Min: %s\n", buffer);
        pList[i].min = atof(buffer);
        
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
        //printf("Max: %s\n", buffer);
        pList[i].max = atof(buffer);
        
            bzero(buffer,256);
            ReadMessage(sockfd,buffer);
        //printf("Step: %s\n", buffer);
        pList[i].step = atof(buffer);
    
        // numAffects
        bzero(buffer,256);
        ReadMessage(sockfd,buffer);
        pList[i].numAffects = atoi(buffer);
        //printf("number of affects: %d\n", pList[i].numAffects);
        
        for (j = 0; j < pList[i].numAffects; j++){
            bzero(buffer,256);
                ReadMessage(sockfd,buffer);
                //printf("Utility name: %s\n", buffer);
            strcpy(pList[i].affection_list[j].u.name, buffer);
                 
            bzero(buffer,256);
                ReadMessage(sockfd,buffer);
                //printf("Relation: %s\n", buffer);
            strcpy(pList[i].affection_list[j].relation, buffer);
        }
    }
    return 1;
}


int GetObservable(int sockfd, struct Observable oList[], struct CE_Info *ce_info)
{
    char buffer[256];
    int i,j;

    // numParameters
    bzero(buffer,256);
    ReadMessage(sockfd,buffer);
    ce_info->numObservables = atoi(buffer);
    //printf("number of observables: %d\n", ce_info->numObservables);
        
    for (i = 0; i < ce_info->numObservables; i++){
        bzero(buffer,256);
            ReadMessage(sockfd,buffer);
            //printf("Name: %s\n", buffer);
        strcpy(oList[i].name, buffer);
                 
        // numAffects
        bzero(buffer,256);
        ReadMessage(sockfd,buffer);
        oList[i].numAffects = atoi(buffer);
        //printf("number of affects: %d\n", oList[i].numAffects);
        
        for (j = 0; j < oList[i].numAffects; j++){
            bzero(buffer,256);
                ReadMessage(sockfd,buffer);
                //printf("Utility name: %s\n", buffer);
            strcpy(oList[i].affection_list[j].u.name, buffer);
                 
            bzero(buffer,256);
                ReadMessage(sockfd,buffer);
                //printf("Relation: %s\n", buffer);
            strcpy(oList[i].affection_list[j].relation, buffer);
        }
    }
    return 1;
}


// setup client socket connection
int ClientSocket(int argc, char** argv)
{
    int sockfd;
    int portno;
    struct sockaddr_in serv_addr;
    struct hostent *server;
    
    // setup client socket connection
    if (argc < 3) {
       fprintf(stderr,"usage: %s hostname port\n", argv[0]);
       exit(0);
    }
    // server name
    server = gethostbyname(argv[1]);
    if (server == NULL) {
        fprintf(stderr,"ERROR, no such host\n");
        exit(0);
    }
    // port no.
    portno = atoi(argv[2]);
    // socket file descriptor
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) 
        error("ERROR opening socket");
    // initiate struct socketaddr_in
    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(portno);
    bcopy((char *)server->h_addr, 
         (char *)&serv_addr.sin_addr.s_addr,
         server->h_length);
    // cast sockaddr_in to sockaddr
    if (connect(sockfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) < 0) 
        error("ERROR connecting");

    return sockfd;
}

    
int GetXMLFromServer(int sockfd, struct Utility uList[], struct Parameter pList[], 
        struct Observable oList[], struct CE_Info *ce_info)
{
    // get utilities
    GetUtility(sockfd, uList, ce_info);
    
    // get parameters
    GetParameter(sockfd, pList, ce_info);

    // read obervables
    GetObservable(sockfd, oList, ce_info);

    printf("xml reception done\n");
    
    return 1;
}


int SendMessage(int socketfd, char *message) 
{
    int n;
     
    strcat(message, "\0000");
    // Write message back to client
    //n = write(socketfd,message,(strlen(message)+1));
    n = send(socketfd,message,(strlen(message)+1),0);
    if (n<0)
        error("Error sending to server\n");
    else if(n == 0)
        printf("Server closed the socket.\n");
    
    return n;
}


void print_current_config(struct Utility uList[], 
        struct Parameter pList[], 
        struct Observable oList[], 
        struct CE_Info * ce_info) 
{    
    int i = 0;
    int j = 0;

    // utilities
    for(i = 0; i < ce_info->numUtilities ; i++) {
        printf("Utility: %s\n", uList[i].name);
        printf("    Units: %s\n", uList[i].units);
        printf("    Goal: %s\n", uList[i].goal);
        printf("    Target: %f\n", uList[i].target);
        /*cout << "Utility:  " << uList[i].name << endl;
        cout << "     Units:  " << uList[i].units << endl;
        cout << "     Goal:   " << uList[i].goal << endl;
        cout << "     Target: " << uList[i].target << endl;*/
    }
                       
    // parameters                        
    for(i = 0; i < ce_info->numParameters; i++) {
        printf("Paramter: %s\n", pList[i].name);
        printf("    Units: %s\n", pList[i].units);
        printf("    Min: %f\n", pList[i].min);
        printf("    Max: %f\n", pList[i].max);
        printf("    Step: %f\n", pList[i].step);
        /*cout << "Parameter:  " << pList[i].name << endl;
        cout << "       Units:   " << pList[i].units << endl;
        cout << "       Min:     " << pList[i].min << endl;
        cout << "       Max:     " << pList[i].max << endl;
        cout << "       Step:    " << pList[i].step << endl;*/
        for(j = 0; j < pList[i].numAffects; j++) {
            printf("        Affect: %s -> %s\n", pList[i].affection_list[j].u.name, pList[i].affection_list[j].relation);
            //cout << "       Affect: " << pList[i].affection_list[j].u.name << " -> " << pList[i].affection_list[j].relation << endl;
        }
    }

    // observables
    for(i = 0; i < ce_info->numObservables; i++) {
        printf("Observable: %s\n", oList[i].name);
        //cout << "Observable:  " << oList[i].name << endl;
        for(j = 0; j < oList[i].numAffects; j++) {
            printf("        Affect: %s -> %s\n", oList[i].affection_list[j].u.name, oList[i].affection_list[j].relation);
            //cout << "       Affect: " << oList[i].affection_list[j].u.name << " -> " << oList[i].affection_list[j].relation << endl;
        }
    }
}

int RunCBREngine(struct Utility uList[], struct Parameter pList[],
        struct Observable oList[], struct CE_Info *ce_info, 
        int num_cols)
{
    
    int i;

    // get column names
    //char *cols[50];
    char **cols;
    cols = (char **)malloc(sizeof(char)*num_cols);
    int j = 0;
    for (i = 0; i < ce_info->numUtilities; i++){
        cols[j] = malloc(strlen(uList[i].name)+1);
        cols[j] = uList[i].name;
        j++;
    }
    for (i = 0; i < ce_info->numParameters; i++){
        cols[j] = malloc(strlen(pList[i].name)+1);
        cols[j] = pList[i].name;
        j++;
    }
    for (i = 0; i < ce_info->numObservables; i++){
        cols[j] = malloc(strlen(oList[i].name)+1);
        cols[j] = oList[i].name;
        j++;
    }
    cols[j] = malloc(strlen("utility")+1);
    cols[j] = "utility";
    
    // sample table entry
    float vals[num_cols];
    vals[0] = 10e3f;    // throughput
    vals[1] = 1;        // spectral_efficiency
    vals[2] = 1e-3f;    // log10_ber
    vals[3] = 1;        // mod_scheme
    vals[4] = -3.0f;    // tx_power
    vals[5] = 10.0f;   // SNR
    vals[6] = 0.762;    // utility
    
    printf("number of columns: %d\n", num_cols);
    printf("column names:");
    for (i = 0; i<num_cols; i++){
        printf(" %s",cols[i]);
    }
    printf("\n");

    int rc;

    // create cbr database/table
    //printf("create cbr database\n");
    char filename[] = {"ex1"}, tablename[] = {"data"};
    cbr mycbr = cbr_create(filename, tablename, cols, num_cols);
    //cbr mycbr = cbr_create("ex1", "data", cols, num_cols);
    
    // add row here
    //printf("add row to cbr table\n");
    rc = cbr_add_row(mycbr, cols, vals, num_cols);
    
    // print
    printf("cbr table\n");
    cbr_print(mycbr);
    
    // simple search: find entry where...
    //   log10_ber < 1e-2 and SNR > 5
    char * search_names[] = {"log10_ber", "SNR"};
    int search_ops[] = {LT, GT};
    float search_vals[] = {1e-2f, 5.f};
    float retvals[num_cols];
    rc = cbr_search(mycbr, search_names, search_ops, 
            search_vals, 2, retvals);

    j = 0;
    for (i = 0; i < ce_info->numUtilities; i++){
        uList[i].value = retvals[j];
        j++;
    }
    for (i = 0; i < ce_info->numParameters; i++){
        pList[i].value = retvals[j];
        j++;
    }
    for (i = 0; i < ce_info->numObservables; i++){
        oList[i].value = retvals[j];
        j++;
    }

    // clean up database
    //printf("clean cbr database\n");
    cbr_free(mycbr);
    
    return 1;
}

// main client socket
int main(int argc, char *argv[])
{
    struct Utility uList[10];
    struct Parameter pList[10];
    struct Observable oList[10];
    struct CE_Info ce_info;

    // setup client socket connection
    int sockfd;
    sockfd = ClientSocket(argc, argv);
    
    // get xml info from server
    GetXMLFromServer(sockfd, uList, pList, oList, &ce_info);
    print_current_config(uList, pList, oList, &ce_info); 

    // cbr operation
    unsigned int num_cols;
    // get number of columns
    num_cols = ce_info.numUtilities + ce_info.numParameters;
    num_cols = num_cols + ce_info.numObservables;
    num_cols = num_cols + 1;
    RunCBREngine(uList, pList, oList, &ce_info, num_cols);    
    // print out results
    int i;
    printf("search result: ");
    for (i=0;i<ce_info.numUtilities; i++)
        printf("%f, ",uList[i].value);
    for (i=0;i<ce_info.numParameters; i++)
        printf("%f, ",pList[i].value);
    for (i=0;i<ce_info.numObservables; i++)
        printf("%f, ",oList[i].value);
    printf("\n");
    
    // send results back to server
    char counter[50];
    char var[50];
    // utility
    sprintf(counter, "%d", ce_info.numUtilities);
    SendMessage(sockfd, counter);
    // send utilities
    for (i = 0; i < ce_info.numUtilities; i++){
        sprintf(var, "%f", uList[i].value);
        SendMessage(sockfd, var);
    }

    // parameter
    sprintf(counter, "%d", ce_info.numParameters);
    SendMessage(sockfd, counter);
    // send parameters
    for (i = 0; i < ce_info.numParameters; i++){
        sprintf(var, "%f", pList[i].value);
        SendMessage(sockfd, var);
    }

    // observable
    sprintf(counter, "%d", ce_info.numObservables);
    SendMessage(sockfd, counter);
    // send utilities
    for (i = 0; i < ce_info.numObservables; i++){
        sprintf(var, "%f", oList[i].value);
        SendMessage(sockfd, var);
    }


  return 0;
}