mas01cr@498: #include "audioDB_API.h"
mas01cr@498: #include "test_utils_lib.h"
mas01ik@355: 
mas01cr@498: int main(int argc, char **argv) {
mas01cr@498:   adb_t *adb;
mas01cr@498:   adb_insert_t batch[1]={{0}};
mas01ik@355: 
mas01cr@498:   clean_remove_db(TESTDB);
mas01cr@498:   if(!(adb = audiodb_create(TESTDB,0,0,0)))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   maketestfile("testfeature", 2, (double[4]){0, 1, 1, 0}, 4);
mas01cr@498:   maketestfile("testpower", 1, (double[2]) {-0.5, -1}, 2);
mas01ik@355: 
mas01cr@498:   batch[0].features="testfeature";
mas01cr@498:   batch[0].power="testpower";
mas01cr@498:   if(!audiodb_batchinsert(adb, batch, 1))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   if(audiodb_power(adb))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   batch[0].power=NULL;
mas01cr@498:   if(!audiodb_batchinsert(adb, batch, 1))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   batch[0].power="testpower";
mas01cr@498:   if(audiodb_batchinsert(adb, batch, 1))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   if(audiodb_l2norm(adb))
mas01cr@498:     return 1;
mas01ik@355: 
mas01cr@498:   adb_datum_t query = {1, 2, "testquery", (double[2]) {0, 0.5}};
mas01cr@498:   adb_query_id_t qid = {0};
mas01cr@498:   qid.datum = &query;
mas01cr@498:   qid.sequence_length = 1;
mas01cr@498:   qid.sequence_start = 0;
mas01cr@498:   adb_query_parameters_t parms = 
mas01cr@498:     {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
mas01cr@498:   adb_query_refine_t refine = {0};
mas01ik@355: 
mas01cr@498:   adb_query_spec_t spec;
mas01cr@498:   spec.qid = qid;
mas01cr@498:   spec.params = parms;
mas01cr@498:   spec.refine = refine;
mas01ik@355: 
mas01cr@498:   adb_query_results_t *results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 1);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.params.npoints = 1;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.qid.datum->data = (double [2]) {0.5, 0};
mas01cr@498:   spec.params.npoints = 10;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 1);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.params.npoints = 1;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 1);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01cr@498:   
mas01cr@498:   /* queries with associated power data */
mas01cr@498:   spec.qid.datum->data = (double [2]) {0, 0.5};
mas01cr@498:   spec.qid.datum->power = (double [1]) {-0.5};
mas01cr@498:   spec.params.npoints = 10;
mas01cr@498:   spec.refine.flags = ADB_REFINE_ABSOLUTE_THRESHOLD;
mas01cr@498:   spec.refine.absolute_threshold = -1.4;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 1);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = -0.6;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = -0.2;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 0) return 1;
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = 0;
mas01cr@498:   spec.refine.flags = ADB_REFINE_RELATIVE_THRESHOLD;
mas01cr@498:   spec.refine.relative_threshold = 1;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 1);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.relative_threshold = 0.2;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.qid.datum->data = (double [2]) {0.5, 0};
mas01cr@498:   spec.refine.flags = ADB_REFINE_ABSOLUTE_THRESHOLD;
mas01cr@498:   spec.refine.absolute_threshold = -1.4;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 1);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = -0.6;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = -0.2;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 0) return 1;
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.absolute_threshold = 0;
mas01cr@498:   spec.refine.flags = ADB_REFINE_RELATIVE_THRESHOLD;
mas01cr@498:   spec.refine.relative_threshold = 1;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 2) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 0, 0, 1);
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   spec.refine.relative_threshold = 0.2;
mas01cr@498:   results = audiodb_query_spec(adb, &spec);
mas01cr@498:   if(!results || results->nresults != 1) return 1;
mas01cr@498:   result_present_or_fail(results, "testfeature", 2, 0, 0);
mas01cr@498:   audiodb_query_free_results(adb, &spec, results);
mas01ik@355: 
mas01cr@498:   audiodb_close(adb);
mas01ik@355: 
mas01cr@498:   return 104;
mas01ik@355: }