mas01cr@498
|
1 #include "audioDB_API.h"
|
mas01cr@498
|
2 #include "test_utils_lib.h"
|
mas01ik@355
|
3
|
mas01cr@498
|
4 int main(int argc, char **argv) {
|
mas01cr@498
|
5 adb_t *adb;
|
mas01cr@498
|
6 adb_insert_t batch[2]={{0}, {0}};
|
mas01ik@355
|
7
|
mas01cr@498
|
8 clean_remove_db(TESTDB);
|
mas01cr@498
|
9 if(!(adb = audiodb_create(TESTDB, 0, 0, 0)))
|
mas01cr@498
|
10 return 1;
|
mas01ik@355
|
11
|
mas01cr@498
|
12 maketestfile("testfeature01", 2, (double[2]) {0, 1}, 2);
|
mas01cr@498
|
13 maketestfile("testfeature10", 2, (double[2]) {1, 0}, 2);
|
mas01ik@355
|
14
|
mas01cr@498
|
15 batch[0].features="testfeature01";
|
mas01cr@498
|
16 batch[1].features="testfeature10";
|
mas01cr@498
|
17 if(audiodb_batchinsert(adb, batch, 2))
|
mas01cr@498
|
18 return 1;
|
mas01cr@498
|
19 if(audiodb_l2norm(adb))
|
mas01cr@498
|
20 return 1;
|
mas01ik@355
|
21
|
mas01cr@498
|
22 adb_datum_t query = {2, 2, "testquery", (double[4]) {0, 0.5, 0.5, 0}};
|
mas01cr@498
|
23 adb_query_id_t qid = {0};
|
mas01cr@498
|
24 qid.datum = &query;
|
mas01cr@498
|
25 qid.sequence_length = 1;
|
mas01cr@498
|
26 qid.flags = ADB_QID_FLAG_EXHAUSTIVE;
|
mas01cr@498
|
27 adb_query_parameters_t parms =
|
mas01cr@498
|
28 {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
|
mas01cr@498
|
29 adb_query_refine_t refine = {0};
|
mas01ik@355
|
30
|
mas01cr@498
|
31 adb_query_spec_t spec;
|
mas01cr@498
|
32 spec.qid = qid;
|
mas01cr@498
|
33 spec.params = parms;
|
mas01cr@498
|
34 spec.refine = refine;
|
mas01cr@498
|
35
|
mas01cr@498
|
36 adb_query_results_t *results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
37 if(!results || results->nresults != 4) return 1;
|
mas01cr@498
|
38 result_present_or_fail(results, "testfeature01", 0, 0, 0);
|
mas01cr@498
|
39 result_present_or_fail(results, "testfeature01", 2, 1, 0);
|
mas01cr@498
|
40 result_present_or_fail(results, "testfeature10", 2, 0, 0);
|
mas01cr@498
|
41 result_present_or_fail(results, "testfeature10", 0, 1, 0);
|
mas01cr@498
|
42 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
43
|
mas01cr@498
|
44 spec.params.npoints = 1;
|
mas01cr@498
|
45 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
46 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
47 result_present_or_fail(results, "testfeature01", 0, 0, 0);
|
mas01cr@498
|
48 result_present_or_fail(results, "testfeature10", 0, 1, 0);
|
mas01cr@498
|
49 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
50
|
mas01cr@498
|
51 audiodb_close(adb);
|
mas01ik@355
|
52
|
mas01cr@498
|
53 return 104;
|
mas01ik@355
|
54 }
|