mas01cr@678: #include "audioDB_API.h"
mas01cr@678: #include "test_utils_lib.h"
mas01cr@678: 
mas01cr@678: int main(int argc, char **argv) {
mas01cr@678:   adb_t *adb;
mas01cr@678: 
mas01cr@678:   clean_remove_db(TESTDB);
mas01cr@678:   if(!(adb = audiodb_create(TESTDB, 0, 0, 0)))
mas01cr@678:     return 1;
mas01cr@678: 
mas01cr@678:   adb_datum_t datum1 = {1, 2, "testfeature1", (double[2]) {0, 1}};
mas01cr@678:   adb_datum_t datum3 = {3, 2, "testfeature3", (double[6]) {1, 0, 0, 1, 1, 0}};
mas01cr@678: 
mas01cr@678:   if(audiodb_insert_datum(adb, &datum1))
mas01cr@678:     return 1;
mas01cr@678:   if(audiodb_insert_datum(adb, &datum3))
mas01cr@678:     return 1;
mas01cr@678:   if(audiodb_l2norm(adb))
mas01cr@678:     return 1;
mas01cr@678: 
mas01cr@678:   adb_datum_t query = {2, 2, "testquery", (double[4]) {0, 1, 1, 0}};
mas01cr@678:   adb_query_id_t qid = {0};
mas01cr@678:   qid.datum = &query;
mas01cr@678:   qid.sequence_length = 1;
mas01cr@678:   qid.flags |= ADB_QID_FLAG_EXHAUSTIVE;
mas01cr@678: 
mas01cr@678:   adb_query_parameters_t parms =
mas01cr@678:     {ADB_ACCUMULATION_DB, ADB_DISTANCE_EUCLIDEAN, 10};
mas01cr@678:   adb_query_refine_t refine = {0};
mas01cr@678:   refine.qhopsize = 1;
mas01cr@678:   refine.ihopsize = 1;
mas01cr@678:   refine.flags |= ADB_REFINE_HOP_SIZE;
mas01cr@678: 
mas01cr@678:   adb_query_spec_t spec;
mas01cr@678:   spec.qid = qid;
mas01cr@678:   spec.params = parms;
mas01cr@678:   spec.refine = refine;
mas01cr@678: 
mas01cr@678:   adb_query_results_t *results = audiodb_query_spec(adb, &spec);
mas01cr@678:   if(!results || results->nresults != 8) return 1;
mas01cr@678:   result_present_or_fail(results, "testfeature1", 0, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 0, 1);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 1, 2);
mas01cr@678:   result_present_or_fail(results, "testfeature1", 2, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 1, 1);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 2);
mas01cr@678:   
mas01cr@678:   audiodb_query_free_results(adb, &spec, results);
mas01cr@678: 
mas01cr@678:   spec.refine.qhopsize = 2;
mas01cr@678:   results = audiodb_query_spec(adb, &spec);
mas01cr@678:   if(!results || results->nresults != 4) return 1;
mas01cr@678:   result_present_or_fail(results, "testfeature1", 0, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 0, 1);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 2);
mas01cr@678: 
mas01cr@678:   audiodb_query_free_results(adb, &spec, results);
mas01cr@678: 
mas01cr@678:   spec.refine.qhopsize = 1;
mas01cr@678:   spec.refine.ihopsize = 2;
mas01cr@678:   results = audiodb_query_spec(adb, &spec);
mas01cr@678:   if(!results || results->nresults != 6) return 1;
mas01cr@678:   result_present_or_fail(results, "testfeature1", 0, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 1, 2);
mas01cr@678:   result_present_or_fail(results, "testfeature1", 2, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 2);
mas01cr@678: 
mas01cr@678:   audiodb_query_free_results(adb, &spec, results);
mas01cr@678: 
mas01cr@678:   spec.refine.ihopsize = 3;
mas01cr@678:   results = audiodb_query_spec(adb, &spec);
mas01cr@678:   if(!results || results->nresults != 4) return 1;
mas01cr@678:   result_present_or_fail(results, "testfeature1", 0, 0, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 0, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature1", 2, 1, 0);
mas01cr@678:   result_present_or_fail(results, "testfeature3", 2, 0, 0);
mas01cr@678: 
mas01cr@678:   audiodb_query_free_results(adb, &spec, results);
mas01cr@678: 
mas01cr@678:   audiodb_close(adb);
mas01cr@678: 
mas01cr@678:   return 104;
mas01cr@678: }