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[1]={{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("testfeature", 2, (double[4]){0, 1, 1, 0}, 4);
|
mas01cr@498
|
13 maketestfile("testpower", 1, (double[2]) {-0.5, -1}, 2);
|
mas01ik@355
|
14
|
mas01cr@498
|
15 batch[0].features="testfeature";
|
mas01cr@498
|
16 batch[0].power="testpower";
|
mas01cr@498
|
17 if(!audiodb_batchinsert(adb, batch, 1))
|
mas01cr@498
|
18 return 1;
|
mas01ik@355
|
19
|
mas01cr@498
|
20 if(audiodb_power(adb))
|
mas01cr@498
|
21 return 1;
|
mas01ik@355
|
22
|
mas01cr@498
|
23 batch[0].power=NULL;
|
mas01cr@498
|
24 if(!audiodb_batchinsert(adb, batch, 1))
|
mas01cr@498
|
25 return 1;
|
mas01ik@355
|
26
|
mas01cr@498
|
27 batch[0].power="testpower";
|
mas01cr@498
|
28 if(audiodb_batchinsert(adb, batch, 1))
|
mas01cr@498
|
29 return 1;
|
mas01ik@355
|
30
|
mas01cr@498
|
31 if(audiodb_l2norm(adb))
|
mas01cr@498
|
32 return 1;
|
mas01ik@355
|
33
|
mas01cr@498
|
34 adb_datum_t query = {1, 2, "testquery", (double[2]) {0, 0.5}};
|
mas01cr@498
|
35 adb_query_id_t qid = {0};
|
mas01cr@498
|
36 qid.datum = &query;
|
mas01cr@498
|
37 qid.sequence_length = 1;
|
mas01cr@498
|
38 qid.sequence_start = 0;
|
mas01cr@498
|
39 adb_query_parameters_t parms =
|
mas01cr@498
|
40 {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
|
mas01cr@498
|
41 adb_query_refine_t refine = {0};
|
mas01ik@355
|
42
|
mas01cr@498
|
43 adb_query_spec_t spec;
|
mas01cr@498
|
44 spec.qid = qid;
|
mas01cr@498
|
45 spec.params = parms;
|
mas01cr@498
|
46 spec.refine = refine;
|
mas01ik@355
|
47
|
mas01cr@498
|
48 adb_query_results_t *results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
49 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
50 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
51 result_present_or_fail(results, "testfeature", 2, 0, 1);
|
mas01cr@498
|
52 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
53
|
mas01cr@498
|
54 spec.params.npoints = 1;
|
mas01cr@498
|
55 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
56 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
57 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
58 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
59
|
mas01cr@498
|
60 spec.qid.datum->data = (double [2]) {0.5, 0};
|
mas01cr@498
|
61 spec.params.npoints = 10;
|
mas01cr@498
|
62 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
63 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
64 result_present_or_fail(results, "testfeature", 0, 0, 1);
|
mas01cr@498
|
65 result_present_or_fail(results, "testfeature", 2, 0, 0);
|
mas01cr@498
|
66 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
67
|
mas01cr@498
|
68 spec.params.npoints = 1;
|
mas01cr@498
|
69 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
70 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
71 result_present_or_fail(results, "testfeature", 0, 0, 1);
|
mas01cr@498
|
72 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@498
|
73
|
mas01cr@498
|
74 /* queries with associated power data */
|
mas01cr@498
|
75 spec.qid.datum->data = (double [2]) {0, 0.5};
|
mas01cr@498
|
76 spec.qid.datum->power = (double [1]) {-0.5};
|
mas01cr@498
|
77 spec.params.npoints = 10;
|
mas01cr@498
|
78 spec.refine.flags = ADB_REFINE_ABSOLUTE_THRESHOLD;
|
mas01cr@498
|
79 spec.refine.absolute_threshold = -1.4;
|
mas01cr@498
|
80 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
81 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
82 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
83 result_present_or_fail(results, "testfeature", 2, 0, 1);
|
mas01cr@498
|
84 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
85
|
mas01cr@498
|
86 spec.refine.absolute_threshold = -0.6;
|
mas01cr@498
|
87 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
88 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
89 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
90 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
91
|
mas01cr@498
|
92 spec.refine.absolute_threshold = -0.2;
|
mas01cr@498
|
93 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
94 if(!results || results->nresults != 0) return 1;
|
mas01cr@498
|
95 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
96
|
mas01cr@498
|
97 spec.refine.absolute_threshold = 0;
|
mas01cr@498
|
98 spec.refine.flags = ADB_REFINE_RELATIVE_THRESHOLD;
|
mas01cr@498
|
99 spec.refine.relative_threshold = 1;
|
mas01cr@498
|
100 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
101 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
102 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
103 result_present_or_fail(results, "testfeature", 2, 0, 1);
|
mas01cr@498
|
104 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
105
|
mas01cr@498
|
106 spec.refine.relative_threshold = 0.2;
|
mas01cr@498
|
107 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
108 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
109 result_present_or_fail(results, "testfeature", 0, 0, 0);
|
mas01cr@498
|
110 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
111
|
mas01cr@498
|
112 spec.qid.datum->data = (double [2]) {0.5, 0};
|
mas01cr@498
|
113 spec.refine.flags = ADB_REFINE_ABSOLUTE_THRESHOLD;
|
mas01cr@498
|
114 spec.refine.absolute_threshold = -1.4;
|
mas01cr@498
|
115 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
116 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
117 result_present_or_fail(results, "testfeature", 0, 0, 1);
|
mas01cr@498
|
118 result_present_or_fail(results, "testfeature", 2, 0, 0);
|
mas01cr@498
|
119 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
120
|
mas01cr@498
|
121 spec.refine.absolute_threshold = -0.6;
|
mas01cr@498
|
122 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
123 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
124 result_present_or_fail(results, "testfeature", 2, 0, 0);
|
mas01cr@498
|
125 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
126
|
mas01cr@498
|
127 spec.refine.absolute_threshold = -0.2;
|
mas01cr@498
|
128 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
129 if(!results || results->nresults != 0) return 1;
|
mas01cr@498
|
130 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
131
|
mas01cr@498
|
132 spec.refine.absolute_threshold = 0;
|
mas01cr@498
|
133 spec.refine.flags = ADB_REFINE_RELATIVE_THRESHOLD;
|
mas01cr@498
|
134 spec.refine.relative_threshold = 1;
|
mas01cr@498
|
135 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
136 if(!results || results->nresults != 2) return 1;
|
mas01cr@498
|
137 result_present_or_fail(results, "testfeature", 0, 0, 1);
|
mas01cr@498
|
138 result_present_or_fail(results, "testfeature", 2, 0, 0);
|
mas01cr@498
|
139 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
140
|
mas01cr@498
|
141 spec.refine.relative_threshold = 0.2;
|
mas01cr@498
|
142 results = audiodb_query_spec(adb, &spec);
|
mas01cr@498
|
143 if(!results || results->nresults != 1) return 1;
|
mas01cr@498
|
144 result_present_or_fail(results, "testfeature", 2, 0, 0);
|
mas01cr@498
|
145 audiodb_query_free_results(adb, &spec, results);
|
mas01ik@355
|
146
|
mas01cr@498
|
147 audiodb_close(adb);
|
mas01ik@355
|
148
|
mas01cr@498
|
149 return 104;
|
mas01ik@355
|
150 }
|