Changeset 469 for vtcross/trunk/src/include/vtcross/cbr.h

Timestamp:

09/07/09 17:28:20 (15 years ago)

Author:

bhilburn

Message:

Moved over the necessary CBR functionality for DSA_CE using the new C++
classes. Note that CBR code is messy and needs to be cleaned up.

Files:

• vtcross/trunk/src/include/vtcross/cbr.h (modified) (7 diffs)

vtcross/trunk/src/include/vtcross/cbr.h

@@ -34 +34 @@
 #include <stdint.h>
 #include <string>
+
 #include <sqlite3.h>
 
@@ -49 +50 @@
 {
     public:
-        /* Default constructor and preferred constructor for the CBR class. */
+        /* Constructors for the CBR class. */
         CBR();
         CBR(char *_filename, char *_tablename, char *_cols[], uint32_t _len);
+        CBR(char *_filename, char *_tablename, char *_cols[], \
+                char *_primcols[], uint32_t _len, uint32_t  _primlen);
 
         /* Destructor for the CBR class. Note that this destructor will be
@@ -68 +71 @@
         virtual int32_t Search(char *_names[], int32_t *_ops, float *_vals, \
                 uint32_t _n, float *_retvals);
+        virtual int32_t SearchSum(char *_name, float *_retvals);
+        virtual int32_t SearchRand(char *_names[], int * _ops, float *_vals, unsigned int _n, \
+            float *_retvals);
 
         virtual int32_t Update(char *_where[], char *_set[], float *_wherevals, \
@@ -167 +173 @@
     // execute create database command
     // database handle
-    //_cbr->db = NULL;
+    //db = NULL;
     OpenDatabase();
 
@@ -193 +199 @@
     ExecuteCommand();
 }
+
+// create database
+CBR::CBR(char * _filename, char * _tablename, 
+        char * _cols[], char * _primcols[], unsigned int _len,
+        unsigned int _primlen)
+{
+    // create database
+
+    // copy filename
+    unsigned int i=0;
+    strcpy(filename, _filename);
+
+    // execute create database command
+    // database handle
+    //db = NULL;
+    OpenDatabase();
+
+    // create table
+
+    // copy tablename
+    strcpy(tablename, _tablename);
+
+    // number of columns in the table
+    numColumns = _len;
+
+    // generate command
+    strcpy(command, "CREATE TABLE ");
+    strcat(command, tablename);
+    strcat(command, "(");
+    for (i=0; i<numColumns; i++) {
+        strcat(command, _cols[i]);
+        strcat(command, " FLOAT");
+        strcat(command, ", ");
+    }
+   strcat(command, "PRIMARY KEY (");
+    for (i=0; i<_primlen; i++) {
+        strcat(command, _primcols[i]);
+        if (i != _primlen-1) // not last entry
+            strcat(command, ", ");
+    }
+    strcat(command, "));");
+
+    // execute create table command
+    ExecuteCommand();
+}
+
 
 
@@ -276 +328 @@
 }
 
+// cbr search for a sum
+int32_t 
+CBR::SearchSum(
+    char *_name,
+    float *_retvals)
+{   
+    int rc;
+    // generate command
+    strcpy(command, "select SUM( ");
+    strcat(command, tablename);
+    strcat(command, ".");
+    strcat(command, _name);
+    strcat(command, ") from ");
+    strcat(command, tablename);
+    strcat(command, ";");
+
+    //printf("search command: %s\n", command);
+
+    rc = ExecuteSearchCommand(_retvals);
+    /*printf("search result: ");
+    for (int i=0; i<numColumns; i++)
+        printf("%f, ",_retvals[i]);
+    printf("\n");
+    */
+    return rc;
+}
+
+
+// cbr search
+int32_t 
+CBR::SearchRand(
+    char *_names[],
+    int * _ops,
+    float *_vals,
+    unsigned int _n,
+    float *_retvals)
+{   
+        const char *ops_str[] = {"==", "!=", ">", ">=", "<", "<="};
+    int rc;
+    // generate command
+    strcpy(command, "select ");
+    strcat(command, tablename);
+    strcat(command, ".* from ");
+    strcat(command, tablename);
+    strcat(command, " where ");
+
+    unsigned int i;
+    char str_buffer[64];
+    for (i=0; i<_n; i++) {
+        // ensure valid ops value
+        if (_ops[i] < 0 || _ops[i] > 5) {
+            printf("error: cbr_search(), invalid ops id : %d\n", _ops[i]);
+            exit(1);
+        }
+
+        strcat(command, _names[i]);
+        strcat(command, ops_str[_ops[i]]);
+        sprintf(str_buffer, "%E", _vals[i]);
+        strcat(command, str_buffer);
+
+        if (i<_n-1)
+            strcat(command, " AND ");
+        else
+            strcat(command, " order by RAND();");
+    }
+
+    //printf("search command: %s\n", command);
+
+    rc = ExecuteSearchCommand(_retvals);
+    
+    /*printf("search result: ");
+    for (i=0; i<numColumns; i++)
+        printf("%f, ",_retvals[i]);
+    printf("\n");
+*/
+    return rc;
+}
+
+
+
 
 // update a row 
@@ -335 +467 @@
     strcat(command, " values(");
     for (i=0; i<_len; i++) {
-        // ???? how to fill the values if _cbr->numColumns != _len
+        // ???? how to fill the values if numColumns != _len
         // assume = in the following
         sprintf(command, "%s%f", command, _vals[i]);