|
mas01cr@509
|
1 extern "C" {
|
|
mas01cr@509
|
2 #include "audioDB_API.h"
|
|
mas01cr@509
|
3 }
|
|
mas01cr@498
|
4 #include "audioDB-internals.h"
|
|
mas01cr@498
|
5 #include "accumulators.h"
|
|
mas01cr@239
|
6
|
|
mas01cr@498
|
7 bool audiodb_powers_acceptable(const adb_query_refine_t *r, double p1, double p2) {
|
|
mas01cr@498
|
8 if (r->flags & ADB_REFINE_ABSOLUTE_THRESHOLD) {
|
|
mas01cr@498
|
9 if ((p1 < r->absolute_threshold) || (p2 < r->absolute_threshold)) {
|
|
mas01cr@239
|
10 return false;
|
|
mas01cr@239
|
11 }
|
|
mas01cr@239
|
12 }
|
|
mas01cr@498
|
13 if (r->flags & ADB_REFINE_RELATIVE_THRESHOLD) {
|
|
mas01cr@498
|
14 if (fabs(p1-p2) > fabs(r->relative_threshold)) {
|
|
mas01cr@239
|
15 return false;
|
|
mas01cr@239
|
16 }
|
|
mas01cr@239
|
17 }
|
|
mas01cr@239
|
18 return true;
|
|
mas01cr@239
|
19 }
|
|
mas01cr@239
|
20
|
|
mas01cr@498
|
21 adb_query_results_t *audiodb_query_spec(adb_t *adb, const adb_query_spec_t *qspec) {
|
|
mas01cr@498
|
22 adb_qstate_internal_t qstate = {0};
|
|
mas01cr@498
|
23 qstate.allowed_keys = new std::set<std::string>;
|
|
mas01cr@498
|
24 adb_query_results_t *results;
|
|
mas01cr@498
|
25 if(qspec->refine.flags & ADB_REFINE_INCLUDE_KEYLIST) {
|
|
mas01cr@498
|
26 for(unsigned int k = 0; k < qspec->refine.include.nkeys; k++) {
|
|
mas01cr@498
|
27 qstate.allowed_keys->insert(qspec->refine.include.keys[k]);
|
|
mas01cr@498
|
28 }
|
|
mas01cr@498
|
29 } else {
|
|
mas01cr@498
|
30 for(unsigned int k = 0; k < adb->header->numFiles; k++) {
|
|
mas01cr@498
|
31 qstate.allowed_keys->insert((*adb->keys)[k]);
|
|
mas01cr@498
|
32 }
|
|
mas01cr@498
|
33 }
|
|
mas01cr@498
|
34 if(qspec->refine.flags & ADB_REFINE_EXCLUDE_KEYLIST) {
|
|
mas01cr@498
|
35 for(unsigned int k = 0; k < qspec->refine.exclude.nkeys; k++) {
|
|
mas01cr@498
|
36 qstate.allowed_keys->erase(qspec->refine.exclude.keys[k]);
|
|
mas01cr@498
|
37 }
|
|
mas01cr@498
|
38 }
|
|
mas01mc@292
|
39
|
|
mas01cr@498
|
40 switch(qspec->params.distance) {
|
|
mas01cr@498
|
41 case ADB_DISTANCE_DOT_PRODUCT:
|
|
mas01cr@498
|
42 switch(qspec->params.accumulation) {
|
|
mas01cr@498
|
43 case ADB_ACCUMULATION_DB:
|
|
mas01cr@498
|
44 qstate.accumulator = new DBAccumulator<adb_result_dist_gt>(qspec->params.npoints);
|
|
mas01cr@498
|
45 break;
|
|
mas01cr@498
|
46 case ADB_ACCUMULATION_PER_TRACK:
|
|
mas01cr@498
|
47 qstate.accumulator = new PerTrackAccumulator<adb_result_dist_gt>(qspec->params.npoints, qspec->params.ntracks);
|
|
mas01cr@498
|
48 break;
|
|
mas01cr@498
|
49 case ADB_ACCUMULATION_ONE_TO_ONE:
|
|
mas01cr@498
|
50 qstate.accumulator = new NearestAccumulator<adb_result_dist_gt>();
|
|
mas01cr@498
|
51 break;
|
|
mas01cr@498
|
52 default:
|
|
mas01cr@498
|
53 goto error;
|
|
mas01cr@239
|
54 }
|
|
mas01cr@239
|
55 break;
|
|
mas01cr@498
|
56 case ADB_DISTANCE_EUCLIDEAN_NORMED:
|
|
mas01cr@498
|
57 case ADB_DISTANCE_EUCLIDEAN:
|
|
mas01cr@498
|
58 switch(qspec->params.accumulation) {
|
|
mas01cr@498
|
59 case ADB_ACCUMULATION_DB:
|
|
mas01cr@498
|
60 qstate.accumulator = new DBAccumulator<adb_result_dist_lt>(qspec->params.npoints);
|
|
mas01cr@498
|
61 break;
|
|
mas01cr@498
|
62 case ADB_ACCUMULATION_PER_TRACK:
|
|
mas01cr@498
|
63 qstate.accumulator = new PerTrackAccumulator<adb_result_dist_lt>(qspec->params.npoints, qspec->params.ntracks);
|
|
mas01cr@498
|
64 break;
|
|
mas01cr@498
|
65 case ADB_ACCUMULATION_ONE_TO_ONE:
|
|
mas01cr@498
|
66 qstate.accumulator = new NearestAccumulator<adb_result_dist_lt>();
|
|
mas01cr@498
|
67 break;
|
|
mas01cr@498
|
68 default:
|
|
mas01cr@498
|
69 goto error;
|
|
mas01mc@263
|
70 }
|
|
mas01mc@263
|
71 break;
|
|
mas01cr@239
|
72 default:
|
|
mas01cr@498
|
73 goto error;
|
|
mas01mc@329
|
74 }
|
|
mas01cr@498
|
75
|
|
mas01cr@498
|
76 if((qspec->refine.flags & ADB_REFINE_RADIUS) && audiodb_index_exists(adb->path, qspec->refine.radius, qspec->qid.sequence_length)) {
|
|
mas01cr@498
|
77 if(audiodb_index_query_loop(adb, qspec, &qstate) < 0) {
|
|
mas01cr@498
|
78 goto error;
|
|
mas01cr@498
|
79 }
|
|
mas01cr@498
|
80 } else {
|
|
mas01cr@498
|
81 if(audiodb_query_loop(adb, qspec, &qstate)) {
|
|
mas01cr@498
|
82 goto error;
|
|
mas01cr@498
|
83 }
|
|
mas01mc@329
|
84 }
|
|
mas01mc@292
|
85
|
|
mas01cr@498
|
86 results = qstate.accumulator->get_points();
|
|
mas01cr@498
|
87
|
|
mas01cr@498
|
88 delete qstate.accumulator;
|
|
mas01cr@498
|
89 delete qstate.allowed_keys;
|
|
mas01cr@498
|
90
|
|
mas01cr@498
|
91 return results;
|
|
mas01cr@498
|
92
|
|
mas01cr@498
|
93 error:
|
|
mas01cr@498
|
94 if(qstate.accumulator)
|
|
mas01cr@498
|
95 delete qstate.accumulator;
|
|
mas01cr@498
|
96 if(qstate.allowed_keys)
|
|
mas01cr@498
|
97 delete qstate.allowed_keys;
|
|
mas01cr@498
|
98 return NULL;
|
|
mas01cr@239
|
99 }
|
|
mas01cr@239
|
100
|
|
mas01cr@498
|
101 int audiodb_query_free_results(adb_t *adb, const adb_query_spec_t *spec, adb_query_results_t *rs) {
|
|
mas01cr@498
|
102 free(rs->results);
|
|
mas01cr@498
|
103 free(rs);
|
|
mas01cr@498
|
104 return 0;
|
|
mas01cr@239
|
105 }
|
|
mas01cr@239
|
106
|
|
mas01cr@498
|
107 static void audiodb_initialize_arrays(adb_t *adb, const adb_query_spec_t *spec, int track, unsigned int numVectors, double *query, double *data_buffer, double **D, double **DD) {
|
|
mas01cr@239
|
108 unsigned int j, k, l, w;
|
|
mas01cr@239
|
109 double *dp, *qp, *sp;
|
|
mas01cr@239
|
110
|
|
mas01cr@498
|
111 const unsigned HOP_SIZE = spec->refine.hopsize;
|
|
mas01cr@498
|
112 const unsigned wL = spec->qid.sequence_length;
|
|
mas01cr@239
|
113
|
|
mas01cr@239
|
114 for(j = 0; j < numVectors; j++) {
|
|
mas01cr@239
|
115 // Sum products matrix
|
|
mas01cr@498
|
116 D[j] = new double[(*adb->track_lengths)[track]];
|
|
mas01cr@239
|
117 assert(D[j]);
|
|
mas01cr@239
|
118 // Matched filter matrix
|
|
mas01cr@498
|
119 DD[j]=new double[(*adb->track_lengths)[track]];
|
|
mas01cr@239
|
120 assert(DD[j]);
|
|
mas01cr@239
|
121 }
|
|
mas01cr@239
|
122
|
|
mas01cr@239
|
123 // Dot product
|
|
mas01cr@239
|
124 for(j = 0; j < numVectors; j++)
|
|
mas01cr@498
|
125 for(k = 0; k < (*adb->track_lengths)[track]; k++){
|
|
mas01cr@498
|
126 qp = query + j * adb->header->dim;
|
|
mas01cr@498
|
127 sp = data_buffer + k * adb->header->dim;
|
|
mas01cr@239
|
128 DD[j][k] = 0.0; // Initialize matched filter array
|
|
mas01cr@239
|
129 dp = &D[j][k]; // point to correlation cell j,k
|
|
mas01cr@239
|
130 *dp = 0.0; // initialize correlation cell
|
|
mas01cr@498
|
131 l = adb->header->dim; // size of vectors
|
|
mas01cr@239
|
132 while(l--)
|
|
mas01cr@239
|
133 *dp += *qp++ * *sp++;
|
|
mas01cr@239
|
134 }
|
|
mas01cr@239
|
135
|
|
mas01cr@239
|
136 // Matched Filter
|
|
mas01cr@239
|
137 // HOP SIZE == 1
|
|
mas01cr@239
|
138 double* spd;
|
|
mas01cr@239
|
139 if(HOP_SIZE == 1) { // HOP_SIZE = shingleHop
|
|
mas01cr@239
|
140 for(w = 0; w < wL; w++) {
|
|
mas01cr@239
|
141 for(j = 0; j < numVectors - w; j++) {
|
|
mas01cr@239
|
142 sp = DD[j];
|
|
mas01cr@239
|
143 spd = D[j+w] + w;
|
|
mas01cr@498
|
144 k = (*adb->track_lengths)[track] - w;
|
|
mas01mc@292
|
145 while(k--)
|
|
mas01mc@292
|
146 *sp++ += *spd++;
|
|
mas01cr@239
|
147 }
|
|
mas01cr@239
|
148 }
|
|
mas01cr@239
|
149 } else { // HOP_SIZE != 1
|
|
mas01cr@239
|
150 for(w = 0; w < wL; w++) {
|
|
mas01cr@239
|
151 for(j = 0; j < numVectors - w; j += HOP_SIZE) {
|
|
mas01cr@239
|
152 sp = DD[j];
|
|
mas01cr@239
|
153 spd = D[j+w]+w;
|
|
mas01cr@498
|
154 for(k = 0; k < (*adb->track_lengths)[track] - w; k += HOP_SIZE) {
|
|
mas01cr@239
|
155 *sp += *spd;
|
|
mas01cr@239
|
156 sp += HOP_SIZE;
|
|
mas01cr@239
|
157 spd += HOP_SIZE;
|
|
mas01cr@239
|
158 }
|
|
mas01cr@239
|
159 }
|
|
mas01cr@239
|
160 }
|
|
mas01cr@239
|
161 }
|
|
mas01cr@239
|
162 }
|
|
mas01cr@239
|
163
|
|
mas01cr@498
|
164 static void audiodb_delete_arrays(int track, unsigned int numVectors, double **D, double **DD) {
|
|
mas01cr@239
|
165 if(D != NULL) {
|
|
mas01cr@239
|
166 for(unsigned int j = 0; j < numVectors; j++) {
|
|
mas01cr@239
|
167 delete[] D[j];
|
|
mas01cr@239
|
168 }
|
|
mas01cr@239
|
169 }
|
|
mas01cr@239
|
170 if(DD != NULL) {
|
|
mas01cr@239
|
171 for(unsigned int j = 0; j < numVectors; j++) {
|
|
mas01cr@239
|
172 delete[] DD[j];
|
|
mas01cr@239
|
173 }
|
|
mas01cr@239
|
174 }
|
|
mas01cr@239
|
175 }
|
|
mas01cr@239
|
176
|
|
mas01cr@498
|
177 int audiodb_read_data(adb_t *adb, int trkfid, int track, double **data_buffer_p, size_t *data_buffer_size_p) {
|
|
mas01cr@498
|
178 uint32_t track_length = (*adb->track_lengths)[track];
|
|
mas01cr@498
|
179 size_t track_size = track_length * sizeof(double) * adb->header->dim;
|
|
mas01cr@498
|
180 if (track_size > *data_buffer_size_p) {
|
|
mas01cr@239
|
181 if(*data_buffer_p) {
|
|
mas01cr@239
|
182 free(*data_buffer_p);
|
|
mas01cr@239
|
183 }
|
|
mas01cr@239
|
184 {
|
|
mas01cr@498
|
185 *data_buffer_size_p = track_size;
|
|
mas01cr@498
|
186 void *tmp = malloc(track_size);
|
|
mas01cr@239
|
187 if (tmp == NULL) {
|
|
mas01cr@498
|
188 goto error;
|
|
mas01cr@239
|
189 }
|
|
mas01cr@239
|
190 *data_buffer_p = (double *) tmp;
|
|
mas01cr@239
|
191 }
|
|
mas01cr@239
|
192 }
|
|
mas01cr@239
|
193
|
|
mas01cr@498
|
194 read_or_goto_error(trkfid, *data_buffer_p, track_size);
|
|
mas01cr@498
|
195 return 0;
|
|
mas01cr@498
|
196
|
|
mas01cr@498
|
197 error:
|
|
mas01cr@498
|
198 return 1;
|
|
mas01cr@239
|
199 }
|
|
mas01cr@239
|
200
|
|
mas01mc@541
|
201 int audiodb_track_id_datum(adb_t *adb, uint32_t track_id, adb_datum_t *d, off_t vector_offset=0, size_t num_vectors=0, adb_fd_cache_t* cache=0){
|
|
mas01cr@498
|
202 off_t track_offset = (*adb->track_offsets)[track_id];
|
|
mas01mc@541
|
203
|
|
mas01cr@509
|
204 if(adb->header->flags & ADB_HEADER_FLAG_REFERENCES) {
|
|
mas01cr@498
|
205 /* create a reference/insert, then use adb_insert_create_datum() */
|
|
mas01mc@541
|
206 adb_reference_t *reference = NULL;
|
|
mas01mc@541
|
207 if(! (cache && cache->reference) ){
|
|
mas01mc@541
|
208 reference = (adb_reference_t *) malloc(sizeof(adb_reference_t));
|
|
mas01mc@541
|
209 reference->features = (char*) malloc(ADB_MAXSTR*sizeof(char));
|
|
mas01mc@541
|
210 if(adb->header->flags & ADB_HEADER_FLAG_POWER)
|
|
mas01mc@541
|
211 reference->power = (char*) malloc(ADB_MAXSTR*sizeof(char));
|
|
mas01mc@543
|
212 else
|
|
mas01mc@543
|
213 reference->power = NULL;
|
|
mas01mc@541
|
214 if(adb->header->flags & ADB_HEADER_FLAG_TIMES)
|
|
mas01mc@541
|
215 reference->times = (char*)malloc(ADB_MAXSTR*sizeof(char));
|
|
mas01mc@543
|
216 else
|
|
mas01mc@543
|
217 reference->times = NULL;
|
|
mas01mc@541
|
218 if(cache)
|
|
mas01mc@541
|
219 cache->reference = reference;
|
|
mas01cr@498
|
220 }
|
|
mas01mc@541
|
221 else
|
|
mas01mc@541
|
222 reference = cache->reference;
|
|
mas01mc@541
|
223
|
|
mas01mc@541
|
224 if(! (cache && cache->track_id==track_id) ){
|
|
mas01mc@541
|
225 if(cache)
|
|
mas01mc@541
|
226 cache->track_id = track_id;
|
|
mas01mc@541
|
227 lseek(adb->fd, adb->header->dataOffset + track_id * ADB_FILETABLE_ENTRY_SIZE, SEEK_SET);
|
|
mas01mc@541
|
228 read_or_goto_error(adb->fd, (void *)reference->features, ADB_MAXSTR);
|
|
mas01mc@541
|
229 if(adb->header->flags & ADB_HEADER_FLAG_POWER) {
|
|
mas01mc@541
|
230 lseek(adb->fd, adb->header->powerTableOffset + track_id * ADB_FILETABLE_ENTRY_SIZE, SEEK_SET);
|
|
mas01mc@541
|
231 read_or_goto_error(adb->fd, (void *)reference->power, ADB_MAXSTR);
|
|
mas01mc@541
|
232 }
|
|
mas01mc@541
|
233 if(adb->header->flags & ADB_HEADER_FLAG_TIMES) {
|
|
mas01mc@541
|
234 lseek(adb->fd, adb->header->timesTableOffset + track_id * ADB_FILETABLE_ENTRY_SIZE, SEEK_SET);
|
|
mas01mc@541
|
235 read_or_goto_error(adb->fd, (void *)reference->times, ADB_MAXSTR);
|
|
mas01mc@541
|
236 }
|
|
mas01cr@498
|
237 }
|
|
mas01mc@541
|
238
|
|
mas01mc@541
|
239 int retval = audiodb_insert_create_datum(reference, d, vector_offset*adb->header->dim*sizeof(double), num_vectors*adb->header->dim*sizeof(double), cache);
|
|
mas01mc@541
|
240 if(!cache){
|
|
mas01mc@541
|
241 audiodb_free_datum_reference(reference);
|
|
mas01mc@541
|
242 free(reference);
|
|
mas01mc@541
|
243 }
|
|
mas01mc@541
|
244 return retval;
|
|
mas01cr@498
|
245 } else {
|
|
mas01cr@498
|
246 /* initialize from sources of data that we already have */
|
|
mas01mc@539
|
247 if(num_vectors)
|
|
mas01mc@539
|
248 d->nvectors = num_vectors;
|
|
mas01mc@539
|
249 else
|
|
mas01mc@539
|
250 d->nvectors = (*adb->track_lengths)[track_id];
|
|
mas01cr@498
|
251 d->dim = adb->header->dim;
|
|
mas01cr@498
|
252 d->key = (*adb->keys)[track_id].c_str();
|
|
mas01cr@498
|
253 /* read out stuff from the database tables */
|
|
mas01cr@498
|
254 d->data = (double *) malloc(d->nvectors * d->dim * sizeof(double));
|
|
mas01mc@539
|
255 lseek(adb->fd, adb->header->dataOffset + track_offset + vector_offset*d->dim*sizeof(double), SEEK_SET);
|
|
mas01cr@498
|
256 read_or_goto_error(adb->fd, d->data, d->nvectors * d->dim * sizeof(double));
|
|
mas01cr@509
|
257 if(adb->header->flags & ADB_HEADER_FLAG_POWER) {
|
|
mas01cr@498
|
258 d->power = (double *) malloc(d->nvectors * sizeof(double));
|
|
mas01mc@539
|
259 lseek(adb->fd, adb->header->powerTableOffset + track_offset / d->dim + vector_offset*sizeof(double), SEEK_SET);
|
|
mas01cr@498
|
260 read_or_goto_error(adb->fd, d->power, d->nvectors * sizeof(double));
|
|
mas01cr@498
|
261 }
|
|
mas01cr@509
|
262 if(adb->header->flags & ADB_HEADER_FLAG_TIMES) {
|
|
mas01cr@498
|
263 d->times = (double *) malloc(2 * d->nvectors * sizeof(double));
|
|
mas01mc@539
|
264 lseek(adb->fd, adb->header->timesTableOffset + track_offset / d->dim + 2 * vector_offset*sizeof(double), SEEK_SET);
|
|
mas01cr@498
|
265 read_or_goto_error(adb->fd, d->times, 2 * d->nvectors * sizeof(double));
|
|
mas01cr@498
|
266 }
|
|
mas01cr@498
|
267 return 0;
|
|
mas01cr@498
|
268 }
|
|
mas01cr@498
|
269 error:
|
|
mas01cr@498
|
270 audiodb_free_datum(d);
|
|
mas01cr@498
|
271 return 1;
|
|
mas01cr@498
|
272 }
|
|
mas01mc@292
|
273
|
|
mas01cr@498
|
274 int audiodb_datum_qpointers(adb_datum_t *d, uint32_t sequence_length, double **vector_data, double **vector, adb_qpointers_internal_t *qpointers) {
|
|
mas01cr@498
|
275 uint32_t nvectors = d->nvectors;
|
|
mas01cr@498
|
276
|
|
mas01cr@498
|
277 qpointers->nvectors = nvectors;
|
|
mas01cr@498
|
278
|
|
mas01cr@498
|
279 size_t vector_size = nvectors * sizeof(double) * d->dim;
|
|
mas01cr@498
|
280 *vector_data = new double[vector_size];
|
|
mas01cr@498
|
281 memcpy(*vector_data, d->data, vector_size);
|
|
mas01cr@498
|
282
|
|
mas01cr@498
|
283 qpointers->l2norm_data = new double[vector_size / d->dim];
|
|
mas01cr@498
|
284 audiodb_l2norm_buffer(*vector_data, d->dim, nvectors, qpointers->l2norm_data);
|
|
mas01cr@498
|
285 audiodb_sequence_sum(qpointers->l2norm_data, nvectors, sequence_length);
|
|
mas01cr@498
|
286 audiodb_sequence_sqrt(qpointers->l2norm_data, nvectors, sequence_length);
|
|
mas01cr@498
|
287
|
|
mas01cr@498
|
288 if(d->power) {
|
|
mas01cr@498
|
289 qpointers->power_data = new double[vector_size / d->dim];
|
|
mas01cr@498
|
290 memcpy(qpointers->power_data, d->power, vector_size / d->dim);
|
|
mas01cr@498
|
291 audiodb_sequence_sum(qpointers->power_data, nvectors, sequence_length);
|
|
mas01cr@498
|
292 audiodb_sequence_average(qpointers->power_data, nvectors, sequence_length);
|
|
mas01cr@239
|
293 }
|
|
mas01cr@239
|
294
|
|
mas01cr@498
|
295 if(d->times) {
|
|
mas01cr@498
|
296 qpointers->mean_duration = new double[1];
|
|
mas01cr@498
|
297 *qpointers->mean_duration = 0;
|
|
mas01cr@498
|
298 for(unsigned int k = 0; k < nvectors; k++) {
|
|
mas01cr@498
|
299 *qpointers->mean_duration += d->times[2*k+1] - d->times[2*k];
|
|
mas01cr@239
|
300 }
|
|
mas01cr@498
|
301 *qpointers->mean_duration /= nvectors;
|
|
mas01cr@239
|
302 }
|
|
mas01cr@239
|
303
|
|
mas01cr@498
|
304 *vector = *vector_data;
|
|
mas01cr@498
|
305 qpointers->l2norm = qpointers->l2norm_data;
|
|
mas01cr@498
|
306 qpointers->power = qpointers->power_data;
|
|
mas01cr@498
|
307 return 0;
|
|
mas01cr@498
|
308 }
|
|
mas01cr@498
|
309
|
|
mas01mc@528
|
310 int audiodb_datum_qpointers_partial(adb_datum_t *d, uint32_t sequence_length, double **vector_data,
|
|
mas01mc@528
|
311 double **vector, adb_qpointers_internal_t *qpointers,
|
|
mas01mc@528
|
312 adb_qstate_internal_t *qstate){
|
|
mas01mc@528
|
313 uint32_t nvectors = d->nvectors;
|
|
mas01mc@528
|
314 qpointers->nvectors = nvectors;
|
|
mas01mc@528
|
315
|
|
mas01mc@539
|
316 PointPair pp = (*qstate->exact_evaluation_queue).top();
|
|
mas01mc@529
|
317 #ifdef _LSH_DEBUG_
|
|
mas01mc@539
|
318 cout << "tid=" << pp.trackID << " qpos=" << pp.qpos << " spos=" << pp.spos << endl;
|
|
mas01mc@539
|
319 cout.flush();
|
|
mas01mc@529
|
320 #endif
|
|
mas01mc@539
|
321
|
|
mas01mc@539
|
322 if(d->power) {
|
|
mas01mc@539
|
323 //memcpy(qpointers->power_data, d->power, seq_len_dbl);
|
|
mas01mc@539
|
324 audiodb_sequence_sum(d->power, sequence_length, sequence_length);
|
|
mas01mc@539
|
325 audiodb_sequence_average(d->power, sequence_length, sequence_length);
|
|
mas01mc@528
|
326 }
|
|
mas01mc@539
|
327
|
|
mas01mc@528
|
328 if(d->times) {
|
|
mas01mc@528
|
329 qpointers->mean_duration = new double[1];
|
|
mas01mc@528
|
330 *qpointers->mean_duration = 0;
|
|
mas01mc@528
|
331 for(unsigned int k = 0; k < nvectors; k++) {
|
|
mas01mc@528
|
332 *qpointers->mean_duration += d->times[2*k+1] - d->times[2*k];
|
|
mas01mc@528
|
333 }
|
|
mas01mc@528
|
334 *qpointers->mean_duration /= nvectors;
|
|
mas01mc@528
|
335 }
|
|
mas01mc@539
|
336
|
|
mas01mc@531
|
337 *vector = d->data;
|
|
mas01mc@531
|
338 *vector_data = d->data;
|
|
mas01mc@531
|
339 qpointers->l2norm = 0 ;
|
|
mas01mc@539
|
340 qpointers->power = d->power;
|
|
mas01mc@528
|
341 return 0;
|
|
mas01mc@528
|
342 }
|
|
mas01mc@528
|
343
|
|
mas01cr@498
|
344 int audiodb_query_spec_qpointers(adb_t *adb, const adb_query_spec_t *spec, double **vector_data, double **vector, adb_qpointers_internal_t *qpointers) {
|
|
mas01cr@498
|
345 adb_datum_t *datum;
|
|
mas01cr@498
|
346 adb_datum_t d = {0};
|
|
mas01cr@498
|
347 uint32_t sequence_length;
|
|
mas01cr@498
|
348 uint32_t sequence_start;
|
|
mas01cr@498
|
349
|
|
mas01cr@498
|
350 datum = spec->qid.datum;
|
|
mas01cr@498
|
351 sequence_length = spec->qid.sequence_length;
|
|
mas01cr@498
|
352 sequence_start = spec->qid.sequence_start;
|
|
mas01cr@498
|
353
|
|
mas01cr@498
|
354 if(datum->data) {
|
|
mas01cr@498
|
355 if(datum->dim != adb->header->dim) {
|
|
mas01cr@498
|
356 return 1;
|
|
mas01cr@239
|
357 }
|
|
mas01cr@498
|
358 /* initialize d, and mark that nothing needs freeing later. */
|
|
mas01cr@498
|
359 d = *datum;
|
|
mas01cr@498
|
360 datum = &d;
|
|
mas01cr@498
|
361 } else if (datum->key) {
|
|
mas01cr@498
|
362 uint32_t track_id;
|
|
mas01cr@498
|
363 if((track_id = audiodb_key_index(adb, datum->key)) == (uint32_t) -1) {
|
|
mas01cr@498
|
364 return 1;
|
|
mas01cr@498
|
365 }
|
|
mas01cr@498
|
366 audiodb_track_id_datum(adb, track_id, &d);
|
|
mas01cr@498
|
367 } else {
|
|
mas01cr@498
|
368 return 1;
|
|
mas01cr@239
|
369 }
|
|
mas01cr@239
|
370
|
|
mas01cr@498
|
371 /* FIXME: check the overflow logic here */
|
|
mas01cr@498
|
372 if(sequence_start + sequence_length > d.nvectors) {
|
|
mas01cr@498
|
373 if(datum != &d) {
|
|
mas01cr@498
|
374 audiodb_free_datum(&d);
|
|
mas01cr@498
|
375 }
|
|
mas01cr@498
|
376 return 1;
|
|
mas01cr@498
|
377 }
|
|
mas01cr@239
|
378
|
|
mas01cr@498
|
379 audiodb_datum_qpointers(&d, sequence_length, vector_data, vector, qpointers);
|
|
mas01cr@498
|
380
|
|
mas01cr@498
|
381 /* Finally, if applicable, set up the moving qpointers. */
|
|
mas01cr@498
|
382 if(spec->qid.flags & ADB_QID_FLAG_EXHAUSTIVE) {
|
|
mas01cr@498
|
383 /* the qpointers are already at the start, and so correct. */
|
|
mas01cr@498
|
384 } else {
|
|
mas01cr@498
|
385 /* adjust the qpointers to point to the correct place in the sequence */
|
|
mas01cr@498
|
386 *vector = *vector_data + spec->qid.sequence_start * d.dim;
|
|
mas01cr@498
|
387 qpointers->l2norm = qpointers->l2norm_data + spec->qid.sequence_start;
|
|
mas01cr@498
|
388 if(d.power) {
|
|
mas01cr@498
|
389 qpointers->power = qpointers->power_data + spec->qid.sequence_start;
|
|
mas01cr@239
|
390 }
|
|
mas01cr@498
|
391 qpointers->nvectors = sequence_length;
|
|
mas01cr@239
|
392 }
|
|
mas01cr@498
|
393
|
|
mas01cr@498
|
394 /* Clean up: free any bits of datum that we have ourselves
|
|
mas01cr@498
|
395 * allocated. */
|
|
mas01cr@498
|
396 if(datum != &d) {
|
|
mas01cr@498
|
397 audiodb_free_datum(&d);
|
|
mas01cr@498
|
398 }
|
|
mas01cr@498
|
399
|
|
mas01cr@498
|
400 return 0;
|
|
mas01cr@239
|
401 }
|
|
mas01cr@239
|
402
|
|
mas01cr@498
|
403 static int audiodb_set_up_dbpointers(adb_t *adb, const adb_query_spec_t *spec, adb_qpointers_internal_t *dbpointers) {
|
|
mas01cr@498
|
404 uint32_t nvectors = adb->header->length / (adb->header->dim * sizeof(double));
|
|
mas01cr@498
|
405 uint32_t sequence_length = spec->qid.sequence_length;
|
|
mas01mc@292
|
406
|
|
mas01cr@498
|
407 bool using_power = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
|
|
mas01cr@498
|
408 bool using_times = spec->refine.flags & ADB_REFINE_DURATION_RATIO;
|
|
mas01cr@498
|
409 double *times_table = NULL;
|
|
mas01cr@498
|
410
|
|
mas01cr@498
|
411
|
|
mas01cr@498
|
412 dbpointers->nvectors = nvectors;
|
|
mas01cr@498
|
413 dbpointers->l2norm_data = new double[nvectors];
|
|
mas01cr@498
|
414
|
|
mas01cr@498
|
415 double *snpp = dbpointers->l2norm_data, *sppp = 0;
|
|
mas01cr@498
|
416 lseek(adb->fd, adb->header->l2normTableOffset, SEEK_SET);
|
|
mas01cr@498
|
417 read_or_goto_error(adb->fd, dbpointers->l2norm_data, nvectors * sizeof(double));
|
|
mas01cr@498
|
418
|
|
mas01cr@498
|
419 if (using_power) {
|
|
mas01cr@509
|
420 if (!(adb->header->flags & ADB_HEADER_FLAG_POWER)) {
|
|
mas01cr@498
|
421 goto error;
|
|
mas01cr@498
|
422 }
|
|
mas01cr@498
|
423 dbpointers->power_data = new double[nvectors];
|
|
mas01cr@498
|
424 sppp = dbpointers->power_data;
|
|
mas01cr@498
|
425 lseek(adb->fd, adb->header->powerTableOffset, SEEK_SET);
|
|
mas01cr@498
|
426 read_or_goto_error(adb->fd, dbpointers->power_data, nvectors * sizeof(double));
|
|
mas01mc@292
|
427 }
|
|
mas01mc@292
|
428
|
|
mas01cr@498
|
429 for(unsigned int i = 0; i < adb->header->numFiles; i++){
|
|
mas01cr@498
|
430 size_t track_length = (*adb->track_lengths)[i];
|
|
mas01cr@498
|
431 if(track_length >= sequence_length) {
|
|
mas01cr@498
|
432 audiodb_sequence_sum(snpp, track_length, sequence_length);
|
|
mas01cr@498
|
433 audiodb_sequence_sqrt(snpp, track_length, sequence_length);
|
|
mas01cr@498
|
434 if (using_power) {
|
|
mas01cr@498
|
435 audiodb_sequence_sum(sppp, track_length, sequence_length);
|
|
mas01cr@498
|
436 audiodb_sequence_average(sppp, track_length, sequence_length);
|
|
mas01cr@498
|
437 }
|
|
mas01mc@324
|
438 }
|
|
mas01cr@498
|
439 snpp += track_length;
|
|
mas01cr@498
|
440 if (using_power) {
|
|
mas01cr@498
|
441 sppp += track_length;
|
|
mas01mc@324
|
442 }
|
|
mas01mc@292
|
443 }
|
|
mas01mc@292
|
444
|
|
mas01cr@498
|
445 if (using_times) {
|
|
mas01cr@509
|
446 if(!(adb->header->flags & ADB_HEADER_FLAG_TIMES)) {
|
|
mas01cr@498
|
447 goto error;
|
|
mas01cr@498
|
448 }
|
|
mas01mc@292
|
449
|
|
mas01cr@498
|
450 dbpointers->mean_duration = new double[adb->header->numFiles];
|
|
mas01cr@498
|
451
|
|
mas01cr@498
|
452 times_table = (double *) malloc(2 * nvectors * sizeof(double));
|
|
mas01cr@498
|
453 if(!times_table) {
|
|
mas01cr@498
|
454 goto error;
|
|
mas01cr@498
|
455 }
|
|
mas01cr@498
|
456 lseek(adb->fd, adb->header->timesTableOffset, SEEK_SET);
|
|
mas01cr@498
|
457 read_or_goto_error(adb->fd, times_table, 2 * nvectors * sizeof(double));
|
|
mas01cr@498
|
458 for(unsigned int k = 0; k < adb->header->numFiles; k++) {
|
|
mas01cr@498
|
459 size_t track_length = (*adb->track_lengths)[k];
|
|
mas01cr@498
|
460 unsigned int j;
|
|
mas01cr@498
|
461 dbpointers->mean_duration[k] = 0.0;
|
|
mas01cr@498
|
462 for(j = 0; j < track_length; j++) {
|
|
mas01cr@498
|
463 dbpointers->mean_duration[k] += times_table[2*j+1] - times_table[2*j];
|
|
mas01mc@292
|
464 }
|
|
mas01cr@498
|
465 dbpointers->mean_duration[k] /= j;
|
|
mas01mc@292
|
466 }
|
|
mas01cr@498
|
467
|
|
mas01cr@498
|
468 free(times_table);
|
|
mas01cr@498
|
469 times_table = NULL;
|
|
mas01mc@292
|
470 }
|
|
mas01cr@498
|
471
|
|
mas01cr@498
|
472 dbpointers->l2norm = dbpointers->l2norm_data;
|
|
mas01cr@498
|
473 dbpointers->power = dbpointers->power_data;
|
|
mas01cr@498
|
474 return 0;
|
|
mas01cr@498
|
475
|
|
mas01cr@498
|
476 error:
|
|
mas01cr@498
|
477 if(dbpointers->l2norm_data) {
|
|
mas01cr@498
|
478 delete [] dbpointers->l2norm_data;
|
|
mas01cr@498
|
479 }
|
|
mas01cr@498
|
480 if(dbpointers->power_data) {
|
|
mas01cr@498
|
481 delete [] dbpointers->power_data;
|
|
mas01cr@498
|
482 }
|
|
mas01cr@498
|
483 if(dbpointers->mean_duration) {
|
|
mas01cr@498
|
484 delete [] dbpointers->mean_duration;
|
|
mas01cr@498
|
485 }
|
|
mas01cr@498
|
486 if(times_table) {
|
|
mas01cr@498
|
487 free(times_table);
|
|
mas01cr@498
|
488 }
|
|
mas01cr@498
|
489 return 1;
|
|
mas01cr@498
|
490
|
|
mas01mc@292
|
491 }
|
|
mas01mc@292
|
492
|
|
mas01cr@498
|
493 int audiodb_query_queue_loop(adb_t *adb, const adb_query_spec_t *spec, adb_qstate_internal_t *qstate, double *query, adb_qpointers_internal_t *qpointers) {
|
|
mas01cr@498
|
494 adb_qpointers_internal_t dbpointers = {0};
|
|
mas01mc@292
|
495
|
|
mas01cr@498
|
496 uint32_t sequence_length = spec->qid.sequence_length;
|
|
mas01cr@498
|
497 bool power_refine = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
|
|
mas01cr@239
|
498
|
|
mas01cr@498
|
499 if(qstate->exact_evaluation_queue->size() == 0) {
|
|
mas01cr@498
|
500 return 0;
|
|
mas01cr@239
|
501 }
|
|
mas01cr@239
|
502
|
|
mas01cr@498
|
503 /* We are guaranteed that the order of points is sorted by:
|
|
mas01cr@498
|
504 * {trackID, spos, qpos} so we can be relatively efficient in
|
|
mas01cr@498
|
505 * initialization of track data. We assume that points usually
|
|
mas01cr@498
|
506 * don't overlap, so we will use exhaustive dot product evaluation
|
|
mas01cr@498
|
507 * (instead of memoization of partial sums, as in query_loop()).
|
|
mas01cr@498
|
508 */
|
|
mas01cr@498
|
509 double dist;
|
|
mas01cr@498
|
510 double *dbdata = 0, *dbdata_pointer;
|
|
mas01cr@498
|
511 Uns32T npairs = qstate->exact_evaluation_queue->size();
|
|
mas01mc@541
|
512 Uns32T currentTrack = qstate->exact_evaluation_queue->top().trackID+1; // i.e. not first track
|
|
mas01mc@527
|
513 #ifdef _LSH_DEBUG_
|
|
mas01mc@527
|
514 cout << "Num vector pairs to evaluate: " << npairs << "..." << endl;
|
|
mas01mc@527
|
515 cout.flush();
|
|
mas01mc@527
|
516 #endif
|
|
mas01mc@531
|
517 adb_datum_t d = {0};
|
|
mas01mc@541
|
518 adb_fd_cache_t c = {0};
|
|
mas01mc@541
|
519 c.track_id = currentTrack;
|
|
mas01cr@498
|
520 while(npairs--) {
|
|
mas01cr@498
|
521 PointPair pp = qstate->exact_evaluation_queue->top();
|
|
mas01mc@541
|
522 if(pp.trackID != currentTrack){
|
|
mas01mc@541
|
523 audiodb_free_datum(&d);
|
|
mas01mc@541
|
524 currentTrack = pp.trackID;
|
|
mas01mc@541
|
525 }
|
|
mas01mc@539
|
526 maybe_delete_array(dbpointers.mean_duration);
|
|
mas01mc@541
|
527 if(audiodb_track_id_datum(adb, pp.trackID, &d, pp.spos, sequence_length, &c)) {
|
|
mas01mc@539
|
528 delete qstate->exact_evaluation_queue;
|
|
mas01mc@539
|
529 delete qstate->set;
|
|
mas01mc@541
|
530 audiodb_free_datum(&d);
|
|
mas01mc@541
|
531 audiodb_free_datum_cache(&c);
|
|
mas01mc@539
|
532 return 1;
|
|
mas01mc@539
|
533 }
|
|
mas01mc@539
|
534
|
|
mas01mc@539
|
535 if(audiodb_datum_qpointers_partial(&d, sequence_length, &dbdata, &dbdata_pointer, &dbpointers, qstate)) {
|
|
mas01mc@539
|
536 delete qstate->exact_evaluation_queue;
|
|
mas01mc@539
|
537 delete qstate->set;
|
|
mas01mc@531
|
538 audiodb_free_datum(&d);
|
|
mas01mc@541
|
539 audiodb_free_datum_cache(&c);
|
|
mas01mc@539
|
540 return 1;
|
|
mas01mc@539
|
541 }
|
|
mas01mc@531
|
542
|
|
mas01cr@498
|
543 Uns32T qPos = (spec->qid.flags & ADB_QID_FLAG_EXHAUSTIVE) ? pp.qpos : 0;
|
|
mas01cr@498
|
544 // Test power thresholds before computing distance
|
|
mas01mc@539
|
545 if( ( (!power_refine) || audiodb_powers_acceptable(&spec->refine, qpointers->power[qPos], dbpointers.power[0])) &&
|
|
mas01mc@539
|
546 ( qPos<qpointers->nvectors-sequence_length+1 && pp.spos<(*adb->track_lengths)[pp.trackID]-sequence_length+1 ) ){
|
|
mas01mc@531
|
547 // Compute distance
|
|
mas01mc@539
|
548 dist = audiodb_dot_product(query + qPos*adb->header->dim, dbdata, adb->header->dim*sequence_length);
|
|
mas01mc@531
|
549 double qn = audiodb_dot_product(query + qPos*adb->header->dim, query + qPos*adb->header->dim, adb->header->dim*sequence_length);
|
|
mas01mc@539
|
550 double sn = audiodb_dot_product(dbdata, dbdata, adb->header->dim*sequence_length);
|
|
mas01mc@531
|
551 qn = sqrt(qn);
|
|
mas01mc@531
|
552 sn = sqrt(sn);
|
|
mas01cr@498
|
553 switch(spec->params.distance) {
|
|
mas01cr@498
|
554 case ADB_DISTANCE_EUCLIDEAN_NORMED:
|
|
mas01cr@498
|
555 dist = 2 - (2/(qn*sn))*dist;
|
|
mas01cr@498
|
556 break;
|
|
mas01cr@498
|
557 case ADB_DISTANCE_EUCLIDEAN:
|
|
mas01cr@498
|
558 dist = qn*qn + sn*sn - 2*dist;
|
|
mas01cr@498
|
559 break;
|
|
mas01cr@498
|
560 }
|
|
mas01cr@498
|
561 if((!(spec->refine.flags & ADB_REFINE_RADIUS)) ||
|
|
mas01cr@509
|
562 dist <= (spec->refine.radius + ADB_DISTANCE_TOLERANCE)) {
|
|
mas01cr@498
|
563 adb_result_t r;
|
|
mas01cr@498
|
564 r.key = (*adb->keys)[pp.trackID].c_str();
|
|
mas01cr@498
|
565 r.dist = dist;
|
|
mas01cr@498
|
566 r.qpos = pp.qpos;
|
|
mas01cr@498
|
567 r.ipos = pp.spos;
|
|
mas01cr@498
|
568 qstate->accumulator->add_point(&r);
|
|
mas01cr@239
|
569 }
|
|
mas01cr@239
|
570 }
|
|
mas01cr@498
|
571 qstate->exact_evaluation_queue->pop();
|
|
mas01mc@292
|
572 }
|
|
mas01mc@474
|
573
|
|
mas01mc@315
|
574 // Cleanup
|
|
mas01mc@541
|
575 audiodb_free_datum(&d);
|
|
mas01mc@541
|
576 audiodb_free_datum_cache(&c);
|
|
mas01cr@509
|
577 maybe_delete_array(dbpointers.mean_duration);
|
|
mas01cr@498
|
578 delete qstate->exact_evaluation_queue;
|
|
mas01mc@529
|
579 delete qstate->set;
|
|
mas01cr@498
|
580 return 0;
|
|
mas01mc@292
|
581 }
|
|
mas01mc@292
|
582
|
|
mas01cr@498
|
583 int audiodb_query_loop(adb_t *adb, const adb_query_spec_t *spec, adb_qstate_internal_t *qstate) {
|
|
mas01cr@498
|
584
|
|
mas01cr@498
|
585 double *query, *query_data;
|
|
mas01cr@498
|
586 adb_qpointers_internal_t qpointers = {0}, dbpointers = {0};
|
|
mas01mc@292
|
587
|
|
mas01cr@498
|
588 bool power_refine = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
|
|
mas01cr@239
|
589
|
|
mas01cr@509
|
590 if(adb->header->flags & ADB_HEADER_FLAG_REFERENCES) {
|
|
mas01cr@498
|
591 /* FIXME: actually it would be nice to support this mode of
|
|
mas01cr@498
|
592 * operation, but for now... */
|
|
mas01cr@498
|
593 return 1;
|
|
mas01cr@498
|
594 }
|
|
mas01mc@324
|
595
|
|
mas01cr@498
|
596 if(audiodb_query_spec_qpointers(adb, spec, &query_data, &query, &qpointers)) {
|
|
mas01cr@498
|
597 return 1;
|
|
mas01cr@498
|
598 }
|
|
mas01cr@239
|
599
|
|
mas01cr@498
|
600 if(audiodb_set_up_dbpointers(adb, spec, &dbpointers)) {
|
|
mas01cr@498
|
601 return 1;
|
|
mas01cr@498
|
602 }
|
|
mas01cr@239
|
603
|
|
mas01cr@498
|
604 unsigned j,k,track,trackOffset=0, HOP_SIZE = spec->refine.hopsize;
|
|
mas01cr@498
|
605 unsigned wL = spec->qid.sequence_length;
|
|
mas01cr@239
|
606 double **D = 0; // Differences query and target
|
|
mas01cr@239
|
607 double **DD = 0; // Matched filter distance
|
|
mas01cr@239
|
608
|
|
mas01cr@498
|
609 D = new double*[qpointers.nvectors]; // pre-allocate
|
|
mas01cr@498
|
610 DD = new double*[qpointers.nvectors];
|
|
mas01cr@239
|
611
|
|
mas01cr@239
|
612 off_t trackIndexOffset;
|
|
mas01cr@239
|
613
|
|
mas01cr@239
|
614 // Track loop
|
|
mas01cr@239
|
615 size_t data_buffer_size = 0;
|
|
mas01cr@239
|
616 double *data_buffer = 0;
|
|
mas01cr@498
|
617 lseek(adb->fd, adb->header->dataOffset, SEEK_SET);
|
|
mas01cr@239
|
618
|
|
mas01cr@498
|
619 std::set<std::string>::iterator keys_end = qstate->allowed_keys->end();
|
|
mas01cr@498
|
620 for(track = 0; track < adb->header->numFiles; track++) {
|
|
mas01cr@498
|
621 unsigned t = track;
|
|
mas01cr@498
|
622
|
|
mas01cr@498
|
623 while (qstate->allowed_keys->find((*adb->keys)[track]) == keys_end) {
|
|
mas01cr@498
|
624 track++;
|
|
mas01cr@498
|
625 if(track == adb->header->numFiles) {
|
|
mas01cr@498
|
626 goto loop_finish;
|
|
mas01cr@239
|
627 }
|
|
mas01cr@239
|
628 }
|
|
mas01cr@498
|
629 trackOffset = (*adb->track_offsets)[track];
|
|
mas01cr@498
|
630 if(track != t) {
|
|
mas01cr@498
|
631 lseek(adb->fd, adb->header->dataOffset + trackOffset, SEEK_SET);
|
|
mas01cr@498
|
632 }
|
|
mas01cr@498
|
633 trackIndexOffset = trackOffset / (adb->header->dim * sizeof(double)); // dbpointers.nvectors offset
|
|
mas01cr@239
|
634
|
|
mas01cr@498
|
635 if(audiodb_read_data(adb, adb->fd, track, &data_buffer, &data_buffer_size)) {
|
|
mas01cr@498
|
636 return 1;
|
|
mas01mc@292
|
637 }
|
|
mas01cr@498
|
638 if(wL <= (*adb->track_lengths)[track]) { // test for short sequences
|
|
mas01cr@498
|
639
|
|
mas01cr@498
|
640 audiodb_initialize_arrays(adb, spec, track, qpointers.nvectors, query, data_buffer, D, DD);
|
|
mas01mc@292
|
641
|
|
mas01cr@498
|
642 if((!(spec->refine.flags & ADB_REFINE_DURATION_RATIO)) ||
|
|
mas01cr@498
|
643 fabs(dbpointers.mean_duration[track]-qpointers.mean_duration[0]) < qpointers.mean_duration[0]*spec->refine.duration_ratio) {
|
|
mas01cr@239
|
644
|
|
mas01cr@239
|
645 // Search for minimum distance by shingles (concatenated vectors)
|
|
mas01cr@498
|
646 for(j = 0; j <= qpointers.nvectors - wL; j += HOP_SIZE) {
|
|
mas01cr@498
|
647 for(k = 0; k <= (*adb->track_lengths)[track] - wL; k += HOP_SIZE) {
|
|
mas01cr@498
|
648 double thisDist = 0;
|
|
mas01cr@498
|
649 double qn = qpointers.l2norm[j];
|
|
mas01cr@498
|
650 double sn = dbpointers.l2norm[trackIndexOffset + k];
|
|
mas01cr@498
|
651 switch(spec->params.distance) {
|
|
mas01cr@498
|
652 case ADB_DISTANCE_EUCLIDEAN_NORMED:
|
|
mas01cr@498
|
653 thisDist = 2-(2/(qn*sn))*DD[j][k];
|
|
mas01cr@498
|
654 break;
|
|
mas01cr@498
|
655 case ADB_DISTANCE_EUCLIDEAN:
|
|
mas01cr@498
|
656 thisDist = qn*qn + sn*sn - 2*DD[j][k];
|
|
mas01cr@498
|
657 break;
|
|
mas01cr@498
|
658 case ADB_DISTANCE_DOT_PRODUCT:
|
|
mas01cr@498
|
659 thisDist = DD[j][k];
|
|
mas01cr@498
|
660 break;
|
|
mas01cr@498
|
661 }
|
|
mas01cr@239
|
662 // Power test
|
|
mas01cr@498
|
663 if ((!power_refine) || audiodb_powers_acceptable(&spec->refine, qpointers.power[j], dbpointers.power[trackIndexOffset + k])) {
|
|
mas01cr@239
|
664 // radius test
|
|
mas01cr@498
|
665 if((!(spec->refine.flags & ADB_REFINE_RADIUS)) ||
|
|
mas01cr@509
|
666 thisDist <= (spec->refine.radius + ADB_DISTANCE_TOLERANCE)) {
|
|
mas01cr@498
|
667 adb_result_t r;
|
|
mas01cr@498
|
668 r.key = (*adb->keys)[track].c_str();
|
|
mas01cr@498
|
669 r.dist = thisDist;
|
|
mas01cr@498
|
670 if(spec->qid.flags & ADB_QID_FLAG_EXHAUSTIVE) {
|
|
mas01cr@498
|
671 r.qpos = j;
|
|
mas01cr@498
|
672 } else {
|
|
mas01cr@498
|
673 r.qpos = spec->qid.sequence_start;
|
|
mas01cr@498
|
674 }
|
|
mas01cr@498
|
675 r.ipos = k;
|
|
mas01cr@498
|
676 qstate->accumulator->add_point(&r);
|
|
mas01cr@239
|
677 }
|
|
mas01cr@239
|
678 }
|
|
mas01cr@239
|
679 }
|
|
mas01cr@239
|
680 }
|
|
mas01cr@239
|
681 } // Duration match
|
|
mas01cr@498
|
682 audiodb_delete_arrays(track, qpointers.nvectors, D, DD);
|
|
mas01cr@239
|
683 }
|
|
mas01cr@239
|
684 }
|
|
mas01cr@239
|
685
|
|
mas01cr@498
|
686 loop_finish:
|
|
mas01cr@498
|
687
|
|
mas01cr@239
|
688 free(data_buffer);
|
|
mas01cr@239
|
689
|
|
mas01cr@239
|
690 // Clean up
|
|
mas01cr@239
|
691 if(query_data)
|
|
mas01cr@239
|
692 delete[] query_data;
|
|
mas01cr@498
|
693 if(qpointers.l2norm_data)
|
|
mas01cr@498
|
694 delete[] qpointers.l2norm_data;
|
|
mas01cr@498
|
695 if(qpointers.power_data)
|
|
mas01cr@498
|
696 delete[] qpointers.power_data;
|
|
mas01cr@498
|
697 if(qpointers.mean_duration)
|
|
mas01cr@498
|
698 delete[] qpointers.mean_duration;
|
|
mas01cr@498
|
699 if(dbpointers.power_data)
|
|
mas01cr@498
|
700 delete[] dbpointers.power_data;
|
|
mas01cr@498
|
701 if(dbpointers.l2norm_data)
|
|
mas01cr@498
|
702 delete[] dbpointers.l2norm_data;
|
|
mas01cr@239
|
703 if(D)
|
|
mas01cr@239
|
704 delete[] D;
|
|
mas01cr@239
|
705 if(DD)
|
|
mas01cr@239
|
706 delete[] DD;
|
|
mas01cr@498
|
707 if(dbpointers.mean_duration)
|
|
mas01cr@498
|
708 delete[] dbpointers.mean_duration;
|
|
mas01cr@498
|
709
|
|
mas01cr@498
|
710 return 0;
|
|
mas01cr@239
|
711 }
|