annotate query.cpp @ 543:6afeb2c76957 multiprobeLSH

Fixed non-NULL memory initializaton error. We require NULL fields for correct free() behaviour.
author mas01mc
date Sun, 08 Feb 2009 05:06:01 +0000
parents 52d82badc544
children bf89c80ec4cc
rev   line source
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 }