mas01cr@487
|
1 #include "audioDB_API.h"
|
mas01cr@487
|
2 #include "test_utils_lib.h"
|
mas01ik@355
|
3
|
mas01cr@494
|
4 int main(int argc, char **argv) {
|
mas01cr@494
|
5 adb_t *adb;
|
mas01cr@494
|
6 const char *keys[2];
|
mas01ik@355
|
7
|
mas01cr@494
|
8 clean_remove_db(TESTDB);
|
mas01cr@494
|
9 if(!(adb = audiodb_create(TESTDB, 0, 0, 0)))
|
mas01cr@494
|
10 return 1;
|
mas01cr@494
|
11 if(audiodb_l2norm(adb))
|
mas01cr@494
|
12 return 1;
|
mas01ik@355
|
13
|
mas01cr@494
|
14 adb_datum_t datum1 = {1, 2, "testfeature01", (double[2]) {0, 1}};
|
mas01cr@494
|
15 adb_datum_t datum2 = {1, 2, "testfeature10", (double[2]) {1, 0}};
|
mas01cr@494
|
16 if(audiodb_insert_datum(adb, &datum1))
|
mas01cr@494
|
17 return 1;
|
mas01cr@494
|
18 if(audiodb_insert_datum(adb, &datum2))
|
mas01cr@494
|
19 return 1;
|
mas01ik@355
|
20
|
mas01cr@494
|
21 adb_datum_t query = {1, 2, "testquery", (double[2]) {0, 0.5}};
|
mas01ik@355
|
22
|
mas01cr@494
|
23 adb_query_id_t qid = {0};
|
mas01cr@494
|
24 qid.datum = &query;
|
mas01cr@494
|
25 qid.sequence_length = 1;
|
mas01cr@494
|
26 qid.sequence_start = 0;
|
mas01cr@494
|
27 adb_query_parameters_t parms =
|
mas01cr@494
|
28 {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_DOT_PRODUCT, 10, 10};
|
mas01cr@494
|
29 adb_query_refine_t refine = {0};
|
mas01cr@494
|
30 refine.hopsize = 1;
|
mas01ik@355
|
31
|
mas01cr@494
|
32 adb_query_spec_t spec;
|
mas01cr@494
|
33 spec.qid = qid;
|
mas01cr@494
|
34 spec.params = parms;
|
mas01cr@494
|
35 spec.refine = refine;
|
mas01ik@355
|
36
|
mas01cr@494
|
37 adb_query_results_t *results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
38 if(!results || results->nresults != 2) return 1;
|
mas01cr@494
|
39 result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
|
mas01cr@494
|
40 result_present_or_fail(results, "testfeature10", 0, 0, 0);
|
mas01cr@494
|
41 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
42
|
mas01cr@494
|
43 spec.refine.flags = ADB_REFINE_INCLUDE_KEYLIST;
|
mas01cr@494
|
44 spec.refine.include.nkeys = 0;
|
mas01cr@494
|
45 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
46 if(!results || results->nresults != 0) return 1;
|
mas01cr@494
|
47 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
48
|
mas01cr@494
|
49 spec.refine.include.nkeys = 1;
|
mas01cr@494
|
50 spec.refine.include.keys = keys;
|
mas01cr@494
|
51 spec.refine.include.keys[0] = "testfeature01";
|
mas01cr@494
|
52 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
53 if(!results || results->nresults != 1) return 1;
|
mas01cr@494
|
54 result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
|
mas01cr@494
|
55 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
56
|
mas01cr@494
|
57 spec.refine.include.keys[0] = "testfeature10";
|
mas01cr@494
|
58 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
59 if(!results || results->nresults != 1) return 1;
|
mas01cr@494
|
60 result_present_or_fail(results, "testfeature10", 0, 0, 0);
|
mas01cr@494
|
61 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
62
|
mas01cr@494
|
63 spec.refine.include.nkeys = 2;
|
mas01cr@494
|
64 spec.refine.include.keys[0] = "testfeature01";
|
mas01cr@494
|
65 spec.refine.include.keys[1] = "testfeature10";
|
mas01cr@494
|
66 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
67 if(!results || results->nresults != 2) return 1;
|
mas01cr@494
|
68 result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
|
mas01cr@494
|
69 result_present_or_fail(results, "testfeature10", 0, 0, 0);
|
mas01cr@494
|
70 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
71
|
mas01cr@494
|
72 spec.refine.include.nkeys = 0;
|
mas01cr@494
|
73 spec.refine.include.keys = NULL;
|
mas01cr@494
|
74 spec.refine.flags = ADB_REFINE_EXCLUDE_KEYLIST;
|
mas01cr@494
|
75 spec.refine.exclude.nkeys = 1;
|
mas01cr@494
|
76 spec.refine.exclude.keys = keys;
|
mas01cr@494
|
77 spec.refine.exclude.keys[0] = "testfeature10";
|
mas01cr@494
|
78 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
79 if(!results || results->nresults != 1) return 1;
|
mas01cr@494
|
80 result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
|
mas01cr@494
|
81 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
82
|
mas01cr@494
|
83 spec.refine.exclude.keys[0] = "testfeature01";
|
mas01cr@494
|
84 results = audiodb_query_spec(adb, &spec);
|
mas01cr@494
|
85 if(!results || results->nresults != 1) return 1;
|
mas01cr@494
|
86 result_present_or_fail(results, "testfeature10", 0, 0, 0);
|
mas01cr@494
|
87 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
88
|
mas01cr@497
|
89 audiodb_close(adb);
|
mas01cr@497
|
90
|
mas01cr@494
|
91 return 104;
|
mas01ik@355
|
92 }
|