mas01cr@487: #include "audioDB_API.h"
mas01cr@487: #include "test_utils_lib.h"
mas01cr@413: 
mas01cr@470: int result_position(adb_query_results_t *r, const char *key, float dist, uint32_t qpos, uint32_t ipos) {
mas01cr@470:   for(uint32_t k = 0; k < r->nresults; k++) {
mas01cr@470:     adb_result_t result = r->results[k];
mas01cr@470:     if((dist == result.dist) && (qpos == result.qpos) &&
mas01cr@470:        (ipos == result.ipos) && !(strcmp(key, result.key))) {
mas01cr@470:       return k;
mas01cr@470:     }
mas01cr@470:   }
mas01cr@470:   return -1;
mas01cr@470: }
mas01cr@470: 
mas01cr@413: int main(int argc, char *argv[]) {
mas01cr@413:   char *dbname = "testdb";
mas01cr@413:   adb_t *adb;
mas01cr@413:   adb_insert_t *batch = 0;
mas01cr@413:   adb_status_t status;
mas01cr@413:   adb_query_t query = {0};
mas01cr@413:   adb_queryresult_t result;
mas01cr@413: 
mas01cr@413:   clean_remove_db(dbname);
mas01cr@413:   adb = audiodb_create("testdb", 0, 0, 0);
mas01cr@413:   if(!adb) {
mas01cr@413:     return 1;
mas01cr@413:   }
mas01cr@413: 
mas01cr@413:   maketestfile("testfeature01", (int[1]) {2}, (double[4]) {0,1,1,0}, 4);
mas01cr@413:   maketestfile("testfeature10", (int[1]) {2}, (double[4]) {1,0,0,1}, 4);
mas01cr@413: 
mas01cr@413:   batch = (adb_insert_t *) calloc(6, sizeof(adb_insert_t));
mas01cr@413:   if(!batch) {
mas01cr@413:     return 1;
mas01cr@413:   }
mas01cr@413:   batch[0].features = "testfeature01";
mas01cr@413:   batch[1].features = "testfeature01";
mas01cr@413:   batch[2].features = "testfeature10";
mas01cr@413:   batch[3].features = "testfeature10";
mas01cr@413:   batch[4].features = "testfeature01";
mas01cr@413:   batch[5].features = "testfeature10";
mas01cr@413:   
mas01cr@413:   audiodb_batchinsert(adb, batch, 6);
mas01cr@470:   free(batch);
mas01cr@413: 
mas01cr@413:   if(audiodb_status(adb, &status)) {
mas01cr@413:     return 1;
mas01cr@413:   }
mas01cr@413:   if(status.numFiles != 2) {
mas01cr@413:     return 1;
mas01cr@413:   }
mas01cr@413: 
mas01cr@413:   if(audiodb_l2norm(adb)) {
mas01cr@413:     return 1;
mas01cr@413:   }
mas01cr@413: 
mas01cr@470:   adb_datum_t datum = {1, 2, NULL, (double [2]){0, 0.5}, NULL, NULL};
mas01cr@470:   adb_query_id_t qid = {0};
mas01cr@470:   qid.datum = &datum;
mas01cr@470:   qid.sequence_length = 1;
mas01cr@470:   adb_query_parameters_t parms = 
mas01cr@470:     {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
mas01cr@470:   adb_query_refine_t refine = {0};
mas01cr@470:   refine.hopsize = 1;
mas01cr@470: 
mas01cr@470:   /* FIXME: structs / pointers */
mas01cr@470:   adb_query_spec_t spec;
mas01cr@470:   spec.qid = qid;
mas01cr@470:   spec.params = parms;
mas01cr@470:   spec.refine = refine;
mas01cr@470: 
mas01cr@470:   adb_query_results_t *results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 2;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 5;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 1;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 2) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
mas01cr@470: 
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.qid.datum->data = (double [2]) {0.5, 0};
mas01cr@470:   spec.params.npoints = 10;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 2;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 5;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 4) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   spec.params.npoints = 1;
mas01cr@470:   results = audiodb_query_spec(adb, &spec);
mas01cr@470: 
mas01cr@470:   if(results->nresults != 2) return 1;
mas01cr@470:   if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
mas01cr@470:   if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
mas01cr@470:   audiodb_query_free_results(adb, &spec, results);
mas01cr@470: 
mas01cr@470:   audiodb_close(adb);
mas01cr@470: 
mas01cr@413:   return 104;
mas01cr@413: }
mas01cr@413: