mas01cr@487
|
1 #include "audioDB_API.h"
|
mas01cr@487
|
2 #include "test_utils_lib.h"
|
mas01cr@413
|
3
|
mas01cr@470
|
4 int result_position(adb_query_results_t *r, const char *key, float dist, uint32_t qpos, uint32_t ipos) {
|
mas01cr@470
|
5 for(uint32_t k = 0; k < r->nresults; k++) {
|
mas01cr@470
|
6 adb_result_t result = r->results[k];
|
mas01cr@470
|
7 if((dist == result.dist) && (qpos == result.qpos) &&
|
mas01cr@470
|
8 (ipos == result.ipos) && !(strcmp(key, result.key))) {
|
mas01cr@470
|
9 return k;
|
mas01cr@470
|
10 }
|
mas01cr@470
|
11 }
|
mas01cr@470
|
12 return -1;
|
mas01cr@470
|
13 }
|
mas01cr@470
|
14
|
mas01cr@413
|
15 int main(int argc, char *argv[]) {
|
mas01cr@413
|
16 char *dbname = "testdb";
|
mas01cr@413
|
17 adb_t *adb;
|
mas01cr@413
|
18 adb_insert_t *batch = 0;
|
mas01cr@413
|
19 adb_status_t status;
|
mas01cr@413
|
20 adb_query_t query = {0};
|
mas01cr@413
|
21 adb_queryresult_t result;
|
mas01cr@413
|
22
|
mas01cr@413
|
23 clean_remove_db(dbname);
|
mas01cr@413
|
24 adb = audiodb_create("testdb", 0, 0, 0);
|
mas01cr@413
|
25 if(!adb) {
|
mas01cr@413
|
26 return 1;
|
mas01cr@413
|
27 }
|
mas01cr@413
|
28
|
mas01cr@413
|
29 maketestfile("testfeature01", (int[1]) {2}, (double[4]) {0,1,1,0}, 4);
|
mas01cr@413
|
30 maketestfile("testfeature10", (int[1]) {2}, (double[4]) {1,0,0,1}, 4);
|
mas01cr@413
|
31
|
mas01cr@413
|
32 batch = (adb_insert_t *) calloc(6, sizeof(adb_insert_t));
|
mas01cr@413
|
33 if(!batch) {
|
mas01cr@413
|
34 return 1;
|
mas01cr@413
|
35 }
|
mas01cr@413
|
36 batch[0].features = "testfeature01";
|
mas01cr@413
|
37 batch[1].features = "testfeature01";
|
mas01cr@413
|
38 batch[2].features = "testfeature10";
|
mas01cr@413
|
39 batch[3].features = "testfeature10";
|
mas01cr@413
|
40 batch[4].features = "testfeature01";
|
mas01cr@413
|
41 batch[5].features = "testfeature10";
|
mas01cr@413
|
42
|
mas01cr@413
|
43 audiodb_batchinsert(adb, batch, 6);
|
mas01cr@470
|
44 free(batch);
|
mas01cr@413
|
45
|
mas01cr@413
|
46 if(audiodb_status(adb, &status)) {
|
mas01cr@413
|
47 return 1;
|
mas01cr@413
|
48 }
|
mas01cr@413
|
49 if(status.numFiles != 2) {
|
mas01cr@413
|
50 return 1;
|
mas01cr@413
|
51 }
|
mas01cr@413
|
52
|
mas01cr@413
|
53 if(audiodb_l2norm(adb)) {
|
mas01cr@413
|
54 return 1;
|
mas01cr@413
|
55 }
|
mas01cr@413
|
56
|
mas01cr@470
|
57 adb_datum_t datum = {1, 2, NULL, (double [2]){0, 0.5}, NULL, NULL};
|
mas01cr@470
|
58 adb_query_id_t qid = {0};
|
mas01cr@470
|
59 qid.datum = &datum;
|
mas01cr@470
|
60 qid.sequence_length = 1;
|
mas01cr@470
|
61 adb_query_parameters_t parms =
|
mas01cr@470
|
62 {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
|
mas01cr@470
|
63 adb_query_refine_t refine = {0};
|
mas01cr@470
|
64 refine.hopsize = 1;
|
mas01cr@470
|
65
|
mas01cr@470
|
66 /* FIXME: structs / pointers */
|
mas01cr@470
|
67 adb_query_spec_t spec;
|
mas01cr@470
|
68 spec.qid = qid;
|
mas01cr@470
|
69 spec.params = parms;
|
mas01cr@470
|
70 spec.refine = refine;
|
mas01cr@470
|
71
|
mas01cr@470
|
72 adb_query_results_t *results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
73
|
mas01cr@470
|
74 if(results->nresults != 4) return 1;
|
mas01cr@470
|
75 if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
76 if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
77 if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
78 if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
79 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
80
|
mas01cr@470
|
81 spec.params.npoints = 2;
|
mas01cr@470
|
82 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
83
|
mas01cr@470
|
84 if(results->nresults != 4) return 1;
|
mas01cr@470
|
85 if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
86 if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
87 if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
88 if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
89 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
90
|
mas01cr@470
|
91 spec.params.npoints = 5;
|
mas01cr@470
|
92 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
93
|
mas01cr@470
|
94 if(results->nresults != 4) return 1;
|
mas01cr@470
|
95 if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
96 if(result_position(results, "testfeature01", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
97 if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
98 if(result_position(results, "testfeature10", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
99 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
100
|
mas01cr@470
|
101 spec.params.npoints = 1;
|
mas01cr@470
|
102 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
103
|
mas01cr@470
|
104 if(results->nresults != 2) return 1;
|
mas01cr@470
|
105 if(result_position(results, "testfeature01", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
106 if(result_position(results, "testfeature10", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
107
|
mas01cr@470
|
108 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
109
|
mas01cr@470
|
110 spec.qid.datum->data = (double [2]) {0.5, 0};
|
mas01cr@470
|
111 spec.params.npoints = 10;
|
mas01cr@470
|
112 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
113
|
mas01cr@470
|
114 if(results->nresults != 4) return 1;
|
mas01cr@470
|
115 if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
116 if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
117 if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
118 if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
119 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
120
|
mas01cr@470
|
121 spec.params.npoints = 2;
|
mas01cr@470
|
122 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
123
|
mas01cr@470
|
124 if(results->nresults != 4) return 1;
|
mas01cr@470
|
125 if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
126 if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
127 if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
128 if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
129 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
130
|
mas01cr@470
|
131 spec.params.npoints = 5;
|
mas01cr@470
|
132 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
133
|
mas01cr@470
|
134 if(results->nresults != 4) return 1;
|
mas01cr@470
|
135 if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
136 if(result_position(results, "testfeature01", 2, 0, 0) < 0) return 1;
|
mas01cr@470
|
137 if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
138 if(result_position(results, "testfeature10", 2, 0, 1) < 0) return 1;
|
mas01cr@470
|
139 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
140
|
mas01cr@470
|
141 spec.params.npoints = 1;
|
mas01cr@470
|
142 results = audiodb_query_spec(adb, &spec);
|
mas01cr@470
|
143
|
mas01cr@470
|
144 if(results->nresults != 2) return 1;
|
mas01cr@470
|
145 if(result_position(results, "testfeature01", 0, 0, 1) < 0) return 1;
|
mas01cr@470
|
146 if(result_position(results, "testfeature10", 0, 0, 0) < 0) return 1;
|
mas01cr@470
|
147 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@470
|
148
|
mas01cr@470
|
149 audiodb_close(adb);
|
mas01cr@470
|
150
|
mas01cr@413
|
151 return 104;
|
mas01cr@413
|
152 }
|
mas01cr@413
|
153
|