annotate query.cpp @ 455:93ce12fe2f76 api-inversion

Begin pushing adb_t *adb into index_query_loop and query_loop_points
author mas01cr
date Wed, 24 Dec 2008 10:57:23 +0000
parents f3b0ddc1ead0
children 913a95f06998
rev   line source
mas01cr@239 1 #include "audioDB.h"
mas01cr@239 2 #include "reporter.h"
mas01cr@239 3
mas01cr@422 4 #include "audioDB-internals.h"
mas01cr@422 5 #include "accumulators.h"
mas01cr@422 6
mas01cr@444 7 bool audiodb_powers_acceptable(adb_query_refine_t *r, double p1, double p2) {
mas01cr@425 8 if (r->flags & ADB_REFINE_ABSOLUTE_THRESHOLD) {
mas01cr@425 9 if ((p1 < r->absolute_threshold) || (p2 < r->absolute_threshold)) {
mas01cr@239 10 return false;
mas01cr@239 11 }
mas01cr@239 12 }
mas01cr@425 13 if (r->flags & ADB_REFINE_RELATIVE_THRESHOLD) {
mas01cr@425 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@239 21 void audioDB::query(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse) {
mas01cr@425 22
mas01cr@431 23 // init database tables and dbH first
mas01cr@431 24 if(query_from_key)
mas01cr@431 25 initTables(dbName);
mas01cr@431 26 else
mas01cr@431 27 initTables(dbName, inFile);
mas01cr@431 28
mas01cr@435 29 adb_query_spec_t qspec;
mas01cr@444 30 adb_datum_t datum = {0};
mas01cr@435 31
mas01cr@435 32 qspec.refine.flags = 0;
mas01cr@453 33 if(trackFile) {
mas01cr@453 34 qspec.refine.flags |= ADB_REFINE_INCLUDE_KEYLIST;
mas01cr@453 35 std::vector<const char *> v;
mas01cr@453 36 char *k = new char[MAXSTR];
mas01cr@453 37 trackFile->getline(k, MAXSTR);
mas01cr@453 38 while(!trackFile->eof()) {
mas01cr@453 39 v.push_back(k);
mas01cr@453 40 k = new char[MAXSTR];
mas01cr@453 41 trackFile->getline(k, MAXSTR);
mas01cr@453 42 }
mas01cr@453 43 delete [] k;
mas01cr@453 44 qspec.refine.include.nkeys = v.size();
mas01cr@453 45 qspec.refine.include.keys = new const char *[qspec.refine.include.nkeys];
mas01cr@453 46 for(unsigned int k = 0; k < qspec.refine.include.nkeys; k++) {
mas01cr@453 47 qspec.refine.include.keys[k] = v[k];
mas01cr@453 48 }
mas01cr@453 49 }
mas01cr@453 50 if(query_from_key) {
mas01cr@453 51 qspec.refine.flags |= ADB_REFINE_EXCLUDE_KEYLIST;
mas01cr@453 52 qspec.refine.exclude.nkeys = 1;
mas01cr@453 53 qspec.refine.exclude.keys = &key;
mas01cr@453 54 }
mas01cr@425 55 if(radius) {
mas01cr@435 56 qspec.refine.flags |= ADB_REFINE_RADIUS;
mas01cr@435 57 qspec.refine.radius = radius;
mas01cr@425 58 }
mas01cr@425 59 if(use_absolute_threshold) {
mas01cr@435 60 qspec.refine.flags |= ADB_REFINE_ABSOLUTE_THRESHOLD;
mas01cr@435 61 qspec.refine.absolute_threshold = absolute_threshold;
mas01cr@425 62 }
mas01cr@425 63 if(use_relative_threshold) {
mas01cr@435 64 qspec.refine.flags |= ADB_REFINE_RELATIVE_THRESHOLD;
mas01cr@435 65 qspec.refine.relative_threshold = relative_threshold;
mas01cr@425 66 }
mas01cr@425 67 if(usingTimes) {
mas01cr@435 68 qspec.refine.flags |= ADB_REFINE_DURATION_RATIO;
mas01cr@435 69 qspec.refine.duration_ratio = timesTol;
mas01cr@425 70 }
mas01cr@439 71 /* FIXME: not sure about this any more; maybe it belongs in
mas01cr@439 72 query_id? Or maybe we just don't need a flag for it? */
mas01cr@439 73 qspec.refine.hopsize = sequenceHop;
mas01cr@425 74 if(sequenceHop != 1) {
mas01cr@435 75 qspec.refine.flags |= ADB_REFINE_HOP_SIZE;
mas01cr@425 76 }
mas01cr@425 77
mas01cr@444 78 if(query_from_key) {
mas01cr@444 79 datum.key = key;
mas01cr@444 80 } else {
mas01cr@444 81 int fd;
mas01cr@444 82 struct stat st;
mas01cr@444 83
mas01cr@444 84 /* FIXME: around here there are all sorts of hideous leaks. */
mas01cr@444 85 fd = open(inFile, O_RDONLY);
mas01cr@444 86 if(fd < 0) {
mas01cr@444 87 error("failed to open feature file", inFile);
mas01cr@444 88 }
mas01cr@444 89 fstat(fd, &st);
mas01cr@444 90 read(fd, &datum.dim, sizeof(uint32_t));
mas01cr@444 91 datum.nvectors = (st.st_size - sizeof(uint32_t)) / (datum.dim * sizeof(double));
mas01cr@444 92 datum.data = (double *) malloc(st.st_size - sizeof(uint32_t));
mas01cr@444 93 read(fd, datum.data, st.st_size - sizeof(uint32_t));
mas01cr@444 94 close(fd);
mas01cr@444 95 if(usingPower) {
mas01cr@444 96 uint32_t one;
mas01cr@444 97 fd = open(powerFileName, O_RDONLY);
mas01cr@444 98 if(fd < 0) {
mas01cr@444 99 error("failed to open power file", powerFileName);
mas01cr@444 100 }
mas01cr@444 101 read(fd, &one, sizeof(uint32_t));
mas01cr@444 102 if(one != 1) {
mas01cr@444 103 error("malformed power file dimensionality", powerFileName);
mas01cr@444 104 }
mas01cr@444 105 datum.power = (double *) malloc(datum.nvectors * sizeof(double));
mas01cr@444 106 if(read(fd, datum.power, datum.nvectors * sizeof(double)) != (ssize_t) (datum.nvectors * sizeof(double))) {
mas01cr@444 107 error("malformed power file", powerFileName);
mas01cr@444 108 }
mas01cr@444 109 close(fd);
mas01cr@444 110 }
mas01cr@444 111 if(usingTimes) {
mas01cr@444 112 datum.times = (double *) malloc(2 * datum.nvectors * sizeof(double));
mas01cr@444 113 insertTimeStamps(datum.nvectors, timesFile, datum.times);
mas01cr@444 114 }
mas01cr@444 115 }
mas01cr@444 116
mas01cr@444 117 qspec.qid.datum = &datum;
mas01cr@435 118 qspec.qid.sequence_length = sequenceLength;
mas01cr@435 119 qspec.qid.flags = usingQueryPoint ? 0 : ADB_QUERY_ID_FLAG_EXHAUSTIVE;
mas01cr@435 120 qspec.qid.sequence_start = queryPoint;
mas01cr@435 121
mas01cr@431 122 switch(queryType) {
mas01cr@431 123 case O2_POINT_QUERY:
mas01cr@435 124 qspec.qid.sequence_length = 1;
mas01cr@435 125 qspec.params.accumulation = ADB_ACCUMULATION_DB;
mas01cr@435 126 qspec.params.distance = ADB_DISTANCE_DOT_PRODUCT;
mas01cr@435 127 qspec.params.npoints = pointNN;
mas01cr@435 128 qspec.params.ntracks = 0;
mas01cr@431 129 reporter = new pointQueryReporter< std::greater < NNresult > >(pointNN);
mas01cr@431 130 break;
mas01cr@431 131 case O2_TRACK_QUERY:
mas01cr@435 132 qspec.qid.sequence_length = 1;
mas01cr@435 133 qspec.params.accumulation = ADB_ACCUMULATION_PER_TRACK;
mas01cr@435 134 qspec.params.distance = ADB_DISTANCE_DOT_PRODUCT;
mas01cr@435 135 qspec.params.npoints = pointNN;
mas01cr@435 136 qspec.params.ntracks = trackNN;
mas01cr@431 137 reporter = new trackAveragingReporter< std::greater< NNresult > >(pointNN, trackNN, dbH->numFiles);
mas01cr@431 138 break;
mas01cr@431 139 case O2_SEQUENCE_QUERY:
mas01cr@431 140 case O2_N_SEQUENCE_QUERY:
mas01cr@435 141 qspec.params.accumulation = ADB_ACCUMULATION_PER_TRACK;
mas01cr@435 142 qspec.params.distance = no_unit_norming ? ADB_DISTANCE_EUCLIDEAN : ADB_DISTANCE_EUCLIDEAN_NORMED;
mas01cr@435 143 qspec.params.npoints = pointNN;
mas01cr@435 144 qspec.params.ntracks = trackNN;
mas01cr@431 145 switch(queryType) {
mas01cr@431 146 case O2_SEQUENCE_QUERY:
mas01cr@435 147 if(!(qspec.refine.flags & ADB_REFINE_RADIUS)) {
mas01cr@431 148 reporter = new trackAveragingReporter< std::less< NNresult > >(pointNN, trackNN, dbH->numFiles);
mas01cr@435 149 } else if (index_exists(adb->path, qspec.refine.radius, qspec.qid.sequence_length)) {
mas01cr@435 150 char* indexName = index_get_name(adb->path, qspec.refine.radius, qspec.qid.sequence_length);
mas01cr@431 151 lsh = index_allocate(indexName, false);
mas01cr@431 152 reporter = new trackSequenceQueryRadReporter(trackNN, index_to_trackID(lsh->get_maxp(), lsh_n_point_bits)+1);
mas01cr@431 153 delete[] indexName;
mas01cr@431 154 } else {
mas01cr@431 155 reporter = new trackSequenceQueryRadReporter(trackNN, dbH->numFiles);
mas01cr@431 156 }
mas01cr@431 157 break;
mas01cr@431 158 case O2_N_SEQUENCE_QUERY:
mas01cr@435 159 if(!(qspec.refine.flags & ADB_REFINE_RADIUS)) {
mas01cr@431 160 reporter = new trackSequenceQueryNNReporter< std::less < NNresult > >(pointNN, trackNN, dbH->numFiles);
mas01cr@435 161 } else if (index_exists(adb->path, qspec.refine.radius, qspec.qid.sequence_length)){
mas01cr@435 162 char* indexName = index_get_name(adb->path, qspec.refine.radius, qspec.qid.sequence_length);
mas01cr@431 163 lsh = index_allocate(indexName, false);
mas01cr@431 164 reporter = new trackSequenceQueryRadNNReporter(pointNN,trackNN, index_to_trackID(lsh->get_maxp(), lsh_n_point_bits)+1);
mas01cr@431 165 delete[] indexName;
mas01cr@431 166 } else {
mas01cr@431 167 reporter = new trackSequenceQueryRadNNReporter(pointNN,trackNN, dbH->numFiles);
mas01cr@431 168 }
mas01cr@431 169 break;
mas01cr@431 170 }
mas01cr@431 171 break;
mas01cr@431 172 case O2_ONE_TO_ONE_N_SEQUENCE_QUERY:
mas01cr@435 173 qspec.params.accumulation = ADB_ACCUMULATION_ONE_TO_ONE;
mas01cr@435 174 qspec.params.distance = ADB_DISTANCE_EUCLIDEAN_NORMED;
mas01cr@435 175 qspec.params.npoints = 0;
mas01cr@435 176 qspec.params.ntracks = 0;
mas01cr@431 177 break;
mas01cr@431 178 default:
mas01cr@431 179 error("unrecognized queryType");
mas01cr@431 180 }
mas01mc@292 181
mas01mc@292 182 // keyKeyPos requires dbH to be initialized
mas01cr@430 183 if(query_from_key && (!key || (query_from_key_index = audiodb_key_index(adb, key)) == (uint32_t) -1))
mas01cr@430 184 error("Query key not found", key);
mas01cr@431 185
mas01cr@435 186 switch(qspec.params.distance) {
mas01cr@431 187 case ADB_DISTANCE_DOT_PRODUCT:
mas01cr@435 188 switch(qspec.params.accumulation) {
mas01cr@431 189 case ADB_ACCUMULATION_DB:
mas01cr@435 190 accumulator = new DBAccumulator<adb_result_dist_gt>(qspec.params.npoints);
mas01cr@431 191 break;
mas01cr@431 192 case ADB_ACCUMULATION_PER_TRACK:
mas01cr@435 193 accumulator = new PerTrackAccumulator<adb_result_dist_gt>(qspec.params.npoints, qspec.params.ntracks);
mas01cr@431 194 break;
mas01cr@431 195 case ADB_ACCUMULATION_ONE_TO_ONE:
mas01cr@431 196 accumulator = new NearestAccumulator<adb_result_dist_gt>();
mas01cr@431 197 break;
mas01cr@431 198 default:
mas01cr@431 199 error("unknown accumulation");
mas01cr@239 200 }
mas01cr@239 201 break;
mas01cr@431 202 case ADB_DISTANCE_EUCLIDEAN_NORMED:
mas01cr@431 203 case ADB_DISTANCE_EUCLIDEAN:
mas01cr@435 204 switch(qspec.params.accumulation) {
mas01cr@431 205 case ADB_ACCUMULATION_DB:
mas01cr@435 206 accumulator = new DBAccumulator<adb_result_dist_lt>(qspec.params.npoints);
mas01cr@431 207 break;
mas01cr@431 208 case ADB_ACCUMULATION_PER_TRACK:
mas01cr@435 209 accumulator = new PerTrackAccumulator<adb_result_dist_lt>(qspec.params.npoints, qspec.params.ntracks);
mas01cr@431 210 break;
mas01cr@431 211 case ADB_ACCUMULATION_ONE_TO_ONE:
mas01cr@431 212 accumulator = new NearestAccumulator<adb_result_dist_lt>();
mas01cr@431 213 break;
mas01cr@431 214 default:
mas01cr@431 215 error("unknown accumulation");
mas01mc@263 216 }
mas01mc@263 217 break;
mas01cr@239 218 default:
mas01cr@431 219 error("unknown distance function");
mas01cr@431 220 }
mas01cr@431 221
mas01mc@292 222 // Test for index (again) here
mas01cr@435 223 if((qspec.refine.flags & ADB_REFINE_RADIUS) && index_exists(adb->path, qspec.refine.radius, qspec.qid.sequence_length)){
mas01cr@436 224 VERB_LOG(1, "Calling indexed query on database %s, radius=%f, sequence_length=%d\n", adb->path, qspec.refine.radius, qspec.qid.sequence_length);
mas01cr@455 225 index_query_loop(adb, &qspec);
mas01mc@329 226 }
mas01mc@329 227 else{
mas01mc@329 228 VERB_LOG(1, "Calling brute-force query on database %s\n", dbName);
mas01cr@453 229 if(query_loop(adb, &qspec)) {
mas01cr@452 230 error("query_loop failed");
mas01cr@452 231 }
mas01mc@329 232 }
mas01mc@292 233
mas01cr@423 234 adb_query_results_t *rs = accumulator->get_points();
mas01cr@423 235 for(unsigned int k = 0; k < rs->nresults; k++) {
mas01cr@423 236 adb_result_t r = rs->results[k];
mas01cr@430 237 reporter->add_point(audiodb_key_index(adb, r.key), r.qpos, r.ipos, r.dist);
mas01cr@423 238 }
mas01cr@423 239
mas01mc@292 240 reporter->report(fileTable, adbQueryResponse);
mas01cr@239 241 }
mas01cr@239 242
mas01cr@439 243 static void audiodb_initialize_arrays(adb_t *adb, adb_query_spec_t *spec, int track, unsigned int numVectors, double *query, double *data_buffer, double **D, double **DD) {
mas01cr@239 244 unsigned int j, k, l, w;
mas01cr@239 245 double *dp, *qp, *sp;
mas01cr@239 246
mas01cr@439 247 const unsigned HOP_SIZE = spec->refine.hopsize;
mas01cr@435 248 const unsigned wL = spec->qid.sequence_length;
mas01cr@239 249
mas01cr@239 250 for(j = 0; j < numVectors; j++) {
mas01cr@239 251 // Sum products matrix
mas01cr@433 252 D[j] = new double[(*adb->track_lengths)[track]];
mas01cr@239 253 assert(D[j]);
mas01cr@239 254 // Matched filter matrix
mas01cr@433 255 DD[j]=new double[(*adb->track_lengths)[track]];
mas01cr@239 256 assert(DD[j]);
mas01cr@239 257 }
mas01cr@239 258
mas01cr@239 259 // Dot product
mas01cr@239 260 for(j = 0; j < numVectors; j++)
mas01cr@433 261 for(k = 0; k < (*adb->track_lengths)[track]; k++){
mas01cr@439 262 qp = query + j * adb->header->dim;
mas01cr@439 263 sp = data_buffer + k * adb->header->dim;
mas01cr@239 264 DD[j][k] = 0.0; // Initialize matched filter array
mas01cr@239 265 dp = &D[j][k]; // point to correlation cell j,k
mas01cr@239 266 *dp = 0.0; // initialize correlation cell
mas01cr@439 267 l = adb->header->dim; // size of vectors
mas01cr@239 268 while(l--)
mas01cr@239 269 *dp += *qp++ * *sp++;
mas01cr@239 270 }
mas01cr@239 271
mas01cr@239 272 // Matched Filter
mas01cr@239 273 // HOP SIZE == 1
mas01cr@239 274 double* spd;
mas01cr@239 275 if(HOP_SIZE == 1) { // HOP_SIZE = shingleHop
mas01cr@239 276 for(w = 0; w < wL; w++) {
mas01cr@239 277 for(j = 0; j < numVectors - w; j++) {
mas01cr@239 278 sp = DD[j];
mas01cr@239 279 spd = D[j+w] + w;
mas01cr@433 280 k = (*adb->track_lengths)[track] - w;
mas01mc@292 281 while(k--)
mas01mc@292 282 *sp++ += *spd++;
mas01cr@239 283 }
mas01cr@239 284 }
mas01cr@239 285 } else { // HOP_SIZE != 1
mas01cr@239 286 for(w = 0; w < wL; w++) {
mas01cr@239 287 for(j = 0; j < numVectors - w; j += HOP_SIZE) {
mas01cr@239 288 sp = DD[j];
mas01cr@239 289 spd = D[j+w]+w;
mas01cr@433 290 for(k = 0; k < (*adb->track_lengths)[track] - w; k += HOP_SIZE) {
mas01cr@239 291 *sp += *spd;
mas01cr@239 292 sp += HOP_SIZE;
mas01cr@239 293 spd += HOP_SIZE;
mas01cr@239 294 }
mas01cr@239 295 }
mas01cr@239 296 }
mas01cr@239 297 }
mas01cr@239 298 }
mas01cr@239 299
mas01cr@433 300 static void audiodb_delete_arrays(int track, unsigned int numVectors, double **D, double **DD) {
mas01cr@239 301 if(D != NULL) {
mas01cr@239 302 for(unsigned int j = 0; j < numVectors; j++) {
mas01cr@239 303 delete[] D[j];
mas01cr@239 304 }
mas01cr@239 305 }
mas01cr@239 306 if(DD != NULL) {
mas01cr@239 307 for(unsigned int j = 0; j < numVectors; j++) {
mas01cr@239 308 delete[] DD[j];
mas01cr@239 309 }
mas01cr@239 310 }
mas01cr@239 311 }
mas01cr@239 312
mas01cr@433 313 int audiodb_read_data(adb_t *adb, int trkfid, int track, double **data_buffer_p, size_t *data_buffer_size_p) {
mas01cr@433 314 uint32_t track_length = (*adb->track_lengths)[track];
mas01cr@433 315 size_t track_size = track_length * sizeof(double) * adb->header->dim;
mas01cr@433 316 if (track_size > *data_buffer_size_p) {
mas01cr@239 317 if(*data_buffer_p) {
mas01cr@239 318 free(*data_buffer_p);
mas01cr@239 319 }
mas01cr@239 320 {
mas01cr@433 321 *data_buffer_size_p = track_size;
mas01cr@433 322 void *tmp = malloc(track_size);
mas01cr@239 323 if (tmp == NULL) {
mas01cr@433 324 goto error;
mas01cr@239 325 }
mas01cr@239 326 *data_buffer_p = (double *) tmp;
mas01cr@239 327 }
mas01cr@239 328 }
mas01cr@239 329
mas01cr@433 330 read_or_goto_error(trkfid, *data_buffer_p, track_size);
mas01cr@433 331 return 0;
mas01cr@433 332
mas01cr@433 333 error:
mas01cr@433 334 return 1;
mas01cr@239 335 }
mas01cr@239 336
mas01cr@405 337 void audioDB::insertTimeStamps(unsigned numVectors, std::ifstream *timesFile, double *timesdata) {
mas01cr@405 338 assert(usingTimes);
mas01cr@405 339
mas01cr@405 340 unsigned numtimes = 0;
mas01cr@405 341
mas01cr@405 342 if(!timesFile->is_open()) {
mas01cr@405 343 error("problem opening times file on timestamped database", timesFileName);
mas01cr@405 344 }
mas01cr@405 345
mas01cr@405 346 double timepoint, next;
mas01cr@405 347 *timesFile >> timepoint;
mas01cr@405 348 if (timesFile->eof()) {
mas01cr@405 349 error("no entries in times file", timesFileName);
mas01cr@405 350 }
mas01cr@405 351 numtimes++;
mas01cr@405 352 do {
mas01cr@405 353 *timesFile >> next;
mas01cr@405 354 if (timesFile->eof()) {
mas01cr@405 355 break;
mas01cr@405 356 }
mas01cr@405 357 numtimes++;
mas01cr@405 358 timesdata[0] = timepoint;
mas01cr@405 359 timepoint = (timesdata[1] = next);
mas01cr@405 360 timesdata += 2;
mas01cr@405 361 } while (numtimes < numVectors + 1);
mas01cr@405 362
mas01cr@405 363 if (numtimes < numVectors + 1) {
mas01cr@405 364 error("too few timepoints in times file", timesFileName);
mas01cr@405 365 }
mas01cr@405 366
mas01cr@405 367 *timesFile >> next;
mas01cr@405 368 if (!timesFile->eof()) {
mas01cr@405 369 error("too many timepoints in times file", timesFileName);
mas01cr@405 370 }
mas01cr@405 371 }
mas01cr@405 372
mas01cr@444 373 int audiodb_query_spec_qpointers(adb_t *adb, adb_query_spec_t *spec, double **vector_data, double **vector, adb_qpointers_internal_t *qpointers) {
mas01cr@443 374 adb_datum_t *datum;
mas01cr@443 375 adb_datum_t d = {0};
mas01cr@443 376 uint32_t nvectors;
mas01cr@443 377 uint32_t sequence_length;
mas01cr@443 378 uint32_t sequence_start;
mas01cr@443 379
mas01cr@443 380 datum = spec->qid.datum;
mas01cr@443 381 sequence_length = spec->qid.sequence_length;
mas01cr@443 382 sequence_start = spec->qid.sequence_start;
mas01cr@443 383
mas01cr@443 384 if(datum->data) {
mas01cr@443 385 if(datum->dim != adb->header->dim) {
mas01cr@443 386 return 1;
mas01cr@443 387 }
mas01cr@443 388 /* initialize d, and mark that nothing needs freeing later. */
mas01cr@443 389 d = *datum;
mas01cr@443 390 datum = &d;
mas01cr@443 391 } else if (datum->key) {
mas01cr@449 392 uint32_t track_id;
mas01cr@449 393 if((track_id = audiodb_key_index(adb, datum->key)) == (uint32_t) -1) {
mas01cr@443 394 return 1;
mas01cr@443 395 }
mas01cr@443 396 off_t track_offset = (*adb->track_offsets)[track_id];
mas01cr@443 397
mas01cr@443 398 if(adb->header->flags & O2_FLAG_LARGE_ADB) {
mas01cr@443 399 /* create a reference/insert, then use adb_insert_create_datum() */
mas01cr@443 400 adb_reference_t reference = {0};
mas01cr@443 401 char features[MAXSTR], power[MAXSTR], times[MAXSTR];
mas01cr@443 402 lseek(adb->fd, adb->header->dataOffset + track_id * O2_FILETABLE_ENTRY_SIZE, SEEK_SET);
mas01cr@443 403 /* FIXME: learn not to worry and love the bomb^Wbuffer overflow */
mas01cr@443 404 read(adb->fd, features, MAXSTR);
mas01cr@443 405 reference.features = features;
mas01cr@443 406 if(adb->header->flags & O2_FLAG_POWER) {
mas01cr@443 407 lseek(adb->fd, adb->header->powerTableOffset + track_id * O2_FILETABLE_ENTRY_SIZE, SEEK_SET);
mas01cr@443 408 read(adb->fd, power, MAXSTR);
mas01cr@443 409 reference.power = power;
mas01cr@443 410 }
mas01cr@443 411 if(adb->header->flags & O2_FLAG_TIMES) {
mas01cr@443 412 lseek(adb->fd, adb->header->timesTableOffset + track_id * O2_FILETABLE_ENTRY_SIZE, SEEK_SET);
mas01cr@443 413 read(adb->fd, times, MAXSTR);
mas01cr@443 414 reference.times = times;
mas01cr@443 415 }
mas01cr@443 416 audiodb_insert_create_datum(&reference, &d);
mas01cr@443 417 } else {
mas01cr@443 418 /* initialize from sources of data that we already have */
mas01cr@443 419 d.nvectors = (*adb->track_lengths)[track_id];
mas01cr@443 420 d.dim = adb->header->dim;
mas01cr@443 421 d.key = datum->key;
mas01cr@443 422 /* read out stuff from the database tables */
mas01cr@443 423 d.data = (double *) malloc(d.nvectors * d.dim * sizeof(double));
mas01cr@443 424 lseek(adb->fd, adb->header->dataOffset + track_offset, SEEK_SET);
mas01cr@443 425 read(adb->fd, d.data, d.nvectors * d.dim * sizeof(double));
mas01cr@443 426 if(adb->header->flags & O2_FLAG_POWER) {
mas01cr@443 427 d.power = (double *) malloc(d.nvectors * sizeof(double));
mas01cr@443 428 lseek(adb->fd, adb->header->powerTableOffset + track_offset / d.dim, SEEK_SET);
mas01cr@443 429 read(adb->fd, d.power, d.nvectors * sizeof(double));
mas01cr@443 430 }
mas01cr@443 431 if(adb->header->flags & O2_FLAG_TIMES) {
mas01cr@443 432 d.times = (double *) malloc(2 * d.nvectors * sizeof(double));
mas01cr@443 433 lseek(adb->fd, adb->header->timesTableOffset + track_offset / d.dim, SEEK_SET);
mas01cr@443 434 read(adb->fd, d.times, 2 * d.nvectors * sizeof(double));
mas01cr@443 435 }
mas01cr@443 436 }
mas01cr@443 437 } else {
mas01cr@443 438 return 1;
mas01cr@443 439 }
mas01cr@443 440
mas01cr@443 441 /* Now we have a full(ish) datum, compute all the qpointery stuff
mas01cr@443 442 that we care about (l2norm/power/mean duration). (This bit could
mas01cr@443 443 conceivably become a new function) */
mas01cr@443 444 nvectors = d.nvectors;
mas01cr@444 445 /* FIXME: check the overflow logic here */
mas01cr@444 446 if(sequence_start + sequence_length > nvectors) {
mas01cr@443 447 /* is there something to free? goto error */
mas01cr@443 448 return 1;
mas01cr@443 449 }
mas01cr@443 450
mas01cr@443 451 qpointers->nvectors = nvectors;
mas01cr@443 452
mas01cr@443 453 size_t vector_size = nvectors * sizeof(double) * d.dim;
mas01cr@445 454 *vector_data = new double[vector_size];
mas01cr@443 455 memcpy(*vector_data, d.data, vector_size);
mas01cr@443 456
mas01cr@445 457 qpointers->l2norm_data = new double[vector_size / d.dim];
mas01cr@443 458 audiodb_l2norm_buffer(*vector_data, d.dim, nvectors, qpointers->l2norm_data);
mas01cr@443 459 audiodb_sequence_sum(qpointers->l2norm_data, nvectors, sequence_length);
mas01cr@443 460 audiodb_sequence_sqrt(qpointers->l2norm_data, nvectors, sequence_length);
mas01cr@443 461
mas01cr@443 462 if(d.power) {
mas01cr@445 463 qpointers->power_data = new double[vector_size / d.dim];
mas01cr@443 464 memcpy(qpointers->power_data, d.power, vector_size / d.dim);
mas01cr@443 465 audiodb_sequence_sum(qpointers->power_data, nvectors, sequence_length);
mas01cr@443 466 audiodb_sequence_average(qpointers->power_data, nvectors, sequence_length);
mas01cr@443 467 }
mas01cr@443 468
mas01cr@443 469 if(d.times) {
mas01cr@445 470 qpointers->mean_duration = new double[1];
mas01cr@445 471 *qpointers->mean_duration = 0;
mas01cr@443 472 for(unsigned int k = 0; k < nvectors; k++) {
mas01cr@443 473 *qpointers->mean_duration += d.times[2*k+1] - d.times[2*k];
mas01cr@443 474 }
mas01cr@443 475 *qpointers->mean_duration /= nvectors;
mas01cr@443 476 }
mas01cr@443 477
mas01cr@443 478
mas01cr@443 479 /* Finally, set up the moving qpointers. */
mas01cr@443 480 if(spec->qid.flags & ADB_QUERY_ID_FLAG_EXHAUSTIVE) {
mas01cr@443 481 *vector = *vector_data;
mas01cr@443 482 qpointers->l2norm = qpointers->l2norm_data;
mas01cr@443 483 qpointers->power = qpointers->power_data;
mas01cr@443 484 } else {
mas01cr@443 485 *vector = *vector_data + spec->qid.sequence_start * d.dim;
mas01cr@443 486 qpointers->l2norm = qpointers->l2norm_data + spec->qid.sequence_start;
mas01cr@444 487 if(d.power) {
mas01cr@444 488 qpointers->power = qpointers->power_data + spec->qid.sequence_start;
mas01cr@444 489 }
mas01cr@444 490 /* FIXME: this is a little bit ugly. No, a lot ugly. But at the
mas01cr@444 491 * moment this is how query_loop() knows when to stop, so for
mas01cr@444 492 * now... */
mas01cr@444 493 qpointers->nvectors = sequence_length;
mas01cr@443 494 }
mas01cr@443 495
mas01cr@443 496 /* Clean up: free any bits of datum that we have ourselves
mas01cr@443 497 * allocated. */
mas01cr@443 498 if(datum != &d) {
mas01cr@443 499 audiodb_free_datum(&d);
mas01cr@443 500 }
mas01cr@444 501
mas01cr@444 502 return 0;
mas01cr@443 503 }
mas01cr@443 504
mas01cr@448 505 static int audiodb_set_up_dbpointers(adb_t *adb, adb_query_spec_t *spec, adb_qpointers_internal_t *dbpointers) {
mas01cr@438 506 uint32_t nvectors = adb->header->length / (adb->header->dim * sizeof(double));
mas01cr@435 507 uint32_t sequence_length = spec->qid.sequence_length;
mas01cr@438 508
mas01cr@437 509 bool using_power = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
mas01cr@437 510 bool using_times = spec->refine.flags & ADB_REFINE_DURATION_RATIO;
mas01cr@437 511 double *times_table = NULL;
mas01cr@435 512
mas01cr@239 513
mas01cr@438 514 dbpointers->nvectors = nvectors;
mas01cr@438 515 dbpointers->l2norm_data = new double[nvectors];
mas01cr@438 516
mas01cr@438 517 double *snpp = dbpointers->l2norm_data, *sppp = 0;
mas01cr@434 518 lseek(adb->fd, adb->header->l2normTableOffset, SEEK_SET);
mas01cr@438 519 read_or_goto_error(adb->fd, dbpointers->l2norm_data, nvectors * sizeof(double));
mas01cr@239 520
mas01cr@437 521 if (using_power) {
mas01cr@434 522 if (!(adb->header->flags & O2_FLAG_POWER)) {
mas01cr@434 523 goto error;
mas01cr@239 524 }
mas01cr@438 525 dbpointers->power_data = new double[nvectors];
mas01cr@438 526 sppp = dbpointers->power_data;
mas01cr@434 527 lseek(adb->fd, adb->header->powerTableOffset, SEEK_SET);
mas01cr@438 528 read_or_goto_error(adb->fd, dbpointers->power_data, nvectors * sizeof(double));
mas01cr@239 529 }
mas01cr@239 530
mas01cr@434 531 for(unsigned int i = 0; i < adb->header->numFiles; i++){
mas01cr@434 532 size_t track_length = (*adb->track_lengths)[i];
mas01cr@435 533 if(track_length >= sequence_length) {
mas01cr@435 534 audiodb_sequence_sum(snpp, track_length, sequence_length);
mas01cr@435 535 audiodb_sequence_sqrt(snpp, track_length, sequence_length);
mas01cr@437 536 if (using_power) {
mas01cr@435 537 audiodb_sequence_sum(sppp, track_length, sequence_length);
mas01cr@435 538 audiodb_sequence_average(sppp, track_length, sequence_length);
mas01cr@239 539 }
mas01cr@239 540 }
mas01cr@434 541 snpp += track_length;
mas01cr@437 542 if (using_power) {
mas01cr@434 543 sppp += track_length;
mas01cr@239 544 }
mas01cr@239 545 }
mas01cr@239 546
mas01cr@437 547 if (using_times) {
mas01cr@434 548 if(!(adb->header->flags & O2_FLAG_TIMES)) {
mas01cr@437 549 goto error;
mas01cr@239 550 }
mas01cr@239 551
mas01cr@438 552 dbpointers->mean_duration = new double[adb->header->numFiles];
mas01cr@239 553
mas01cr@438 554 times_table = (double *) malloc(2 * nvectors * sizeof(double));
mas01cr@437 555 if(!times_table) {
mas01cr@437 556 goto error;
mas01cr@437 557 }
mas01cr@437 558 lseek(adb->fd, adb->header->timesTableOffset, SEEK_SET);
mas01cr@438 559 read_or_goto_error(adb->fd, times_table, 2 * nvectors * sizeof(double));
mas01cr@434 560 for(unsigned int k = 0; k < adb->header->numFiles; k++) {
mas01cr@434 561 size_t track_length = (*adb->track_lengths)[k];
mas01cr@239 562 unsigned int j;
mas01cr@438 563 dbpointers->mean_duration[k] = 0.0;
mas01cr@434 564 for(j = 0; j < track_length; j++) {
mas01cr@438 565 dbpointers->mean_duration[k] += times_table[2*j+1] - times_table[2*j];
mas01cr@239 566 }
mas01cr@438 567 dbpointers->mean_duration[k] /= j;
mas01cr@239 568 }
mas01cr@437 569
mas01cr@437 570 free(times_table);
mas01cr@437 571 times_table = NULL;
mas01cr@239 572 }
mas01cr@239 573
mas01cr@438 574 dbpointers->l2norm = dbpointers->l2norm_data;
mas01cr@438 575 dbpointers->power = dbpointers->power_data;
mas01cr@434 576 return 0;
mas01cr@434 577
mas01cr@434 578 error:
mas01cr@438 579 if(dbpointers->l2norm_data) {
mas01cr@438 580 delete [] dbpointers->l2norm_data;
mas01cr@434 581 }
mas01cr@438 582 if(dbpointers->power_data) {
mas01cr@438 583 delete [] dbpointers->power_data;
mas01cr@434 584 }
mas01cr@438 585 if(dbpointers->mean_duration) {
mas01cr@438 586 delete [] dbpointers->mean_duration;
mas01cr@434 587 }
mas01cr@437 588 if(times_table) {
mas01cr@437 589 free(times_table);
mas01cr@437 590 }
mas01cr@434 591 return 1;
mas01cr@434 592
mas01cr@239 593 }
mas01cr@239 594
mas01mc@292 595 // query_points()
mas01mc@292 596 //
mas01mc@292 597 // using PointPairs held in the exact_evaluation_queue compute squared distance for each PointPair
mas01mc@292 598 // and insert result into the current reporter.
mas01mc@292 599 //
mas01mc@292 600 // Preconditions:
mas01mc@292 601 // A query inFile has been opened with setup_query(...) and query pointers initialized
mas01mc@292 602 // The database contains some points
mas01mc@292 603 // An exact_evaluation_queue has been allocated and populated
mas01mc@292 604 // A reporter has been allocated
mas01mc@292 605 //
mas01mc@292 606 // Postconditions:
mas01mc@292 607 // reporter contains the points and distances that meet the reporter constraints
mas01mc@292 608
mas01cr@455 609 void audioDB::query_loop_points(adb_t *adb, adb_query_spec_t *spec, double* query, adb_qpointers_internal_t *qpointers) {
mas01cr@438 610 adb_qpointers_internal_t dbpointers = {0};
mas01mc@292 611
mas01cr@436 612 uint32_t sequence_length = spec->qid.sequence_length;
mas01cr@437 613 bool power_refine = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
mas01cr@436 614
mas01cr@455 615 if(exact_evaluation_queue->size() == 0) {
mas01mc@292 616 return;
mas01cr@455 617 }
mas01mc@292 618
mas01cr@455 619 // Compute database info. FIXME: we more than likely don't need
mas01cr@455 620 // very much of the database so write a new function to build these
mas01cr@455 621 // values per-track or, even better, per-point
mas01cr@455 622 if(!(adb->header->flags & O2_FLAG_LARGE_ADB)) {
mas01cr@448 623 if(audiodb_set_up_dbpointers(adb, spec, &dbpointers)) {
mas01cr@455 624 error("failed to set up dbpointers");
mas01cr@434 625 }
mas01cr@455 626 }
mas01mc@292 627
mas01mc@292 628 // We are guaranteed that the order of points is sorted by:
mas01cr@455 629 // {trackID, spos, qpos} so we can be relatively efficient in
mas01cr@455 630 // initialization of track data. We assume that points usually
mas01cr@455 631 // don't overlap, so we will use exhaustive dot product evaluation
mas01cr@455 632 // instead of memoization of partial sums which is used for
mas01cr@455 633 // exhaustive brute-force evaluation from smaller databases in
mas01cr@455 634 // query_loop()
mas01mc@292 635 double dist;
mas01mc@292 636 size_t data_buffer_size = 0;
mas01mc@292 637 double *data_buffer = 0;
mas01mc@324 638 Uns32T trackOffset = 0;
mas01mc@324 639 Uns32T trackIndexOffset = 0;
mas01mc@292 640 Uns32T currentTrack = 0x80000000; // Initialize with a value outside of track index range
mas01mc@292 641 Uns32T npairs = exact_evaluation_queue->size();
mas01mc@292 642 while(npairs--){
mas01mc@292 643 PointPair pp = exact_evaluation_queue->top();
mas01mc@324 644 // Large ADB track data must be loaded here for sPower
mas01cr@455 645 if(adb->header->flags & O2_FLAG_LARGE_ADB) {
mas01mc@324 646 trackOffset=0;
mas01mc@324 647 trackIndexOffset=0;
mas01mc@292 648 if(currentTrack!=pp.trackID){
mas01mc@324 649 char* prefixedString = new char[O2_MAXFILESTR];
mas01mc@324 650 char* tmpStr = prefixedString;
mas01mc@324 651 // On currentTrack change, allocate and load track data
mas01mc@292 652 currentTrack=pp.trackID;
mas01cr@438 653 SAFE_DELETE_ARRAY(dbpointers.l2norm_data);
mas01cr@438 654 SAFE_DELETE_ARRAY(dbpointers.power_data);
mas01mc@324 655 if(infid>0)
mas01mc@324 656 close(infid);
mas01mc@324 657 // Open and check dimensions of feature file
mas01mc@324 658 strncpy(prefixedString, featureFileNameTable+pp.trackID*O2_FILETABLE_ENTRY_SIZE, O2_MAXFILESTR);
mas01mc@324 659 prefix_name((char ** const) &prefixedString, adb_feature_root);
mas01mc@324 660 if (prefixedString!=tmpStr)
mas01mc@324 661 delete[] tmpStr;
mas01cr@454 662 initInputFile(prefixedString);
mas01mc@324 663 // Load the feature vector data for current track into data_buffer
mas01cr@433 664 if(audiodb_read_data(adb, infid, pp.trackID, &data_buffer, &data_buffer_size))
mas01cr@433 665 error("failed to read data");
mas01mc@324 666 // Load power and calculate power and l2norm sequence sums
mas01cr@438 667 init_track_aux_data(pp.trackID, data_buffer, &dbpointers.l2norm_data, &dbpointers.l2norm, &dbpointers.power_data, &dbpointers.power);
mas01mc@292 668 }
mas01mc@324 669 }
mas01mc@324 670 else{
mas01mc@324 671 // These offsets are w.r.t. the entire database of feature vectors and auxillary variables
mas01cr@450 672 trackOffset = (*adb->track_offsets)[pp.trackID];
mas01cr@455 673 trackIndexOffset = trackOffset/(adb->header->dim * sizeof(double)); // num vectors offset
mas01mc@324 674 }
mas01mc@324 675 Uns32T qPos = usingQueryPoint?0:pp.qpos;// index for query point
mas01mc@324 676 Uns32T sPos = trackIndexOffset+pp.spos; // index into l2norm table
mas01mc@324 677 // Test power thresholds before computing distance
mas01cr@438 678 if( ( (!power_refine) || audiodb_powers_acceptable(&spec->refine, qpointers->power[qPos], dbpointers.power[sPos])) &&
mas01cr@451 679 ( qPos<qpointers->nvectors-sequence_length+1 && pp.spos<(*adb->track_lengths)[pp.trackID]-sequence_length+1 ) ){
mas01mc@324 680 // Non-large ADB track data is loaded inside power test for efficiency
mas01cr@455 681 if(!(adb->header->flags & O2_FLAG_LARGE_ADB) && (currentTrack!=pp.trackID) ){
mas01mc@324 682 // On currentTrack change, allocate and load track data
mas01mc@324 683 currentTrack=pp.trackID;
mas01cr@455 684 lseek(dbfid, adb->header->dataOffset + trackOffset, SEEK_SET);
mas01cr@433 685 if(audiodb_read_data(adb, dbfid, currentTrack, &data_buffer, &data_buffer_size))
mas01cr@433 686 error("failed to read data");
mas01mc@324 687 }
mas01mc@324 688 // Compute distance
mas01cr@455 689 dist = audiodb_dot_product(query + qPos*adb->header->dim, data_buffer + pp.spos*adb->header->dim, adb->header->dim*sequence_length);
mas01cr@437 690 double qn = qpointers->l2norm[qPos];
mas01cr@438 691 double sn = dbpointers.l2norm[sPos];
mas01cr@435 692 switch(spec->params.distance) {
mas01cr@431 693 case ADB_DISTANCE_EUCLIDEAN_NORMED:
mas01mc@324 694 dist = 2 - (2/(qn*sn))*dist;
mas01cr@431 695 break;
mas01cr@431 696 case ADB_DISTANCE_EUCLIDEAN:
mas01cr@431 697 dist = qn*qn + sn*sn - 2*dist;
mas01cr@431 698 break;
mas01cr@431 699 }
mas01cr@424 700 if((!radius) || dist <= (O2_LSH_EXACT_MULT*radius+O2_DISTANCE_TOLERANCE)) {
mas01cr@424 701 adb_result_t r;
mas01cr@453 702 r.key = (*adb->keys)[pp.trackID].c_str();
mas01cr@424 703 r.dist = dist;
mas01cr@424 704 r.qpos = pp.qpos;
mas01cr@424 705 r.ipos = pp.spos;
mas01cr@424 706 accumulator->add_point(&r);
mas01cr@424 707 }
mas01mc@292 708 }
mas01mc@292 709 exact_evaluation_queue->pop();
mas01mc@292 710 }
mas01mc@315 711 // Cleanup
mas01cr@438 712 SAFE_DELETE_ARRAY(dbpointers.l2norm_data);
mas01cr@438 713 SAFE_DELETE_ARRAY(dbpointers.power_data);
mas01cr@438 714 SAFE_DELETE_ARRAY(dbpointers.mean_duration);
mas01mc@292 715 }
mas01mc@292 716
mas01cr@453 717 int audioDB::query_loop(adb_t *adb, adb_query_spec_t *spec) {
mas01cr@239 718
mas01cr@239 719 double *query, *query_data;
mas01cr@438 720 adb_qpointers_internal_t qpointers = {0}, dbpointers = {0};
mas01cr@437 721
mas01cr@437 722 bool power_refine = spec->refine.flags & (ADB_REFINE_ABSOLUTE_THRESHOLD|ADB_REFINE_RELATIVE_THRESHOLD);
mas01cr@239 723
mas01cr@453 724 std::set<std::string> keys;
mas01cr@453 725 if(spec->refine.flags & ADB_REFINE_INCLUDE_KEYLIST) {
mas01cr@453 726 for(unsigned int k = 0; k < spec->refine.include.nkeys; k++) {
mas01cr@453 727 keys.insert(spec->refine.include.keys[k]);
mas01cr@453 728 }
mas01cr@453 729 } else {
mas01cr@453 730 for(unsigned int k = 0; k < adb->header->numFiles; k++) {
mas01cr@453 731 keys.insert((*adb->keys)[k]);
mas01cr@453 732 }
mas01cr@453 733 }
mas01cr@453 734 if(spec->refine.flags & ADB_REFINE_EXCLUDE_KEYLIST) {
mas01cr@453 735 for(unsigned int k = 0; k < spec->refine.exclude.nkeys; k++) {
mas01cr@453 736 keys.erase(spec->refine.exclude.keys[k]);
mas01cr@453 737 }
mas01cr@453 738 }
mas01cr@453 739
mas01cr@452 740 if(adb->header->flags & O2_FLAG_LARGE_ADB) {
mas01cr@452 741 /* FIXME: actually it would be nice to support this mode of
mas01cr@452 742 * operation, but for now... */
mas01cr@452 743 return 1;
mas01cr@452 744 }
mas01mc@324 745
mas01cr@444 746 if(audiodb_query_spec_qpointers(adb, spec, &query_data, &query, &qpointers)) {
mas01cr@452 747 return 1;
mas01cr@444 748 }
mas01cr@239 749
mas01cr@448 750 if(audiodb_set_up_dbpointers(adb, spec, &dbpointers)) {
mas01cr@452 751 return 1;
mas01cr@434 752 }
mas01cr@239 753
mas01cr@451 754 unsigned j,k,track,trackOffset=0, HOP_SIZE = spec->refine.hopsize;
mas01cr@435 755 unsigned wL = spec->qid.sequence_length;
mas01cr@239 756 double **D = 0; // Differences query and target
mas01cr@239 757 double **DD = 0; // Matched filter distance
mas01cr@239 758
mas01cr@437 759 D = new double*[qpointers.nvectors]; // pre-allocate
mas01cr@437 760 DD = new double*[qpointers.nvectors];
mas01cr@239 761
mas01cr@239 762 off_t trackIndexOffset;
mas01cr@239 763
mas01cr@239 764 // Track loop
mas01cr@239 765 size_t data_buffer_size = 0;
mas01cr@239 766 double *data_buffer = 0;
mas01cr@451 767 lseek(adb->fd, adb->header->dataOffset, SEEK_SET);
mas01cr@239 768
mas01cr@453 769 for(track = 0; track < adb->header->numFiles; track++) {
mas01cr@453 770 unsigned t = track;
mas01cr@239 771
mas01cr@453 772 while (keys.find((*adb->keys)[track]) == keys.end()) {
mas01cr@453 773 track++;
mas01cr@453 774 if(track == adb->header->numFiles) {
mas01cr@453 775 goto loop_finish;
mas01cr@239 776 }
mas01cr@239 777 }
mas01cr@453 778 trackOffset = (*adb->track_offsets)[track];
mas01cr@453 779 if(track != t) {
mas01cr@453 780 lseek(adb->fd, adb->header->dataOffset + trackOffset, SEEK_SET);
mas01mc@292 781 }
mas01cr@451 782 trackIndexOffset = trackOffset / (adb->header->dim * sizeof(double)); // dbpointers.nvectors offset
mas01cr@239 783
mas01cr@452 784 if(audiodb_read_data(adb, adb->fd, track, &data_buffer, &data_buffer_size)) {
mas01cr@452 785 return 1;
mas01cr@452 786 }
mas01cr@451 787 if(wL <= (*adb->track_lengths)[track]) { // test for short sequences
mas01cr@239 788
mas01cr@439 789 audiodb_initialize_arrays(adb, spec, track, qpointers.nvectors, query, data_buffer, D, DD);
mas01cr@239 790
mas01cr@437 791 if((!(spec->refine.flags & ADB_REFINE_DURATION_RATIO)) ||
mas01cr@438 792 fabs(dbpointers.mean_duration[track]-qpointers.mean_duration[0]) < qpointers.mean_duration[0]*spec->refine.duration_ratio) {
mas01cr@239 793
mas01cr@239 794 // Search for minimum distance by shingles (concatenated vectors)
mas01cr@437 795 for(j = 0; j <= qpointers.nvectors - wL; j += HOP_SIZE) {
mas01cr@451 796 for(k = 0; k <= (*adb->track_lengths)[track] - wL; k += HOP_SIZE) {
mas01cr@431 797 double thisDist = 0;
mas01cr@438 798 double qn = qpointers.l2norm[j];
mas01cr@438 799 double sn = dbpointers.l2norm[trackIndexOffset + k];
mas01cr@435 800 switch(spec->params.distance) {
mas01cr@431 801 case ADB_DISTANCE_EUCLIDEAN_NORMED:
mas01cr@438 802 thisDist = 2-(2/(qn*sn))*DD[j][k];
mas01cr@431 803 break;
mas01cr@431 804 case ADB_DISTANCE_EUCLIDEAN:
mas01cr@438 805 thisDist = qn*qn + sn*sn - 2*DD[j][k];
mas01cr@431 806 break;
mas01cr@431 807 case ADB_DISTANCE_DOT_PRODUCT:
mas01cr@431 808 thisDist = DD[j][k];
mas01cr@431 809 break;
mas01cr@431 810 }
mas01cr@239 811 // Power test
mas01cr@438 812 if ((!power_refine) || audiodb_powers_acceptable(&spec->refine, qpointers.power[j], dbpointers.power[trackIndexOffset + k])) {
mas01cr@239 813 // radius test
mas01cr@435 814 if((!(spec->refine.flags & ADB_REFINE_RADIUS)) ||
mas01cr@435 815 thisDist <= (spec->refine.radius+O2_DISTANCE_TOLERANCE)) {
mas01cr@423 816 adb_result_t r;
mas01cr@453 817 r.key = (*adb->keys)[track].c_str();
mas01cr@423 818 r.dist = thisDist;
mas01cr@451 819 if(spec->qid.flags & ADB_QUERY_ID_FLAG_EXHAUSTIVE) {
mas01cr@451 820 r.qpos = j;
mas01cr@451 821 } else {
mas01cr@451 822 r.qpos = spec->qid.sequence_start;
mas01cr@451 823 }
mas01cr@423 824 r.ipos = k;
mas01cr@423 825 accumulator->add_point(&r);
mas01cr@239 826 }
mas01cr@239 827 }
mas01cr@239 828 }
mas01cr@239 829 }
mas01cr@239 830 } // Duration match
mas01cr@437 831 audiodb_delete_arrays(track, qpointers.nvectors, D, DD);
mas01cr@239 832 }
mas01cr@239 833 }
mas01cr@239 834
mas01cr@453 835 loop_finish:
mas01cr@453 836
mas01cr@239 837 free(data_buffer);
mas01cr@239 838
mas01cr@239 839 // Clean up
mas01cr@239 840 if(query_data)
mas01cr@239 841 delete[] query_data;
mas01cr@437 842 if(qpointers.l2norm_data)
mas01cr@437 843 delete[] qpointers.l2norm_data;
mas01cr@437 844 if(qpointers.power_data)
mas01cr@437 845 delete[] qpointers.power_data;
mas01cr@437 846 if(qpointers.mean_duration)
mas01cr@437 847 delete[] qpointers.mean_duration;
mas01cr@438 848 if(dbpointers.power_data)
mas01cr@438 849 delete[] dbpointers.power_data;
mas01cr@438 850 if(dbpointers.l2norm_data)
mas01cr@438 851 delete[] dbpointers.l2norm_data;
mas01cr@239 852 if(D)
mas01cr@239 853 delete[] D;
mas01cr@239 854 if(DD)
mas01cr@239 855 delete[] DD;
mas01cr@438 856 if(dbpointers.mean_duration)
mas01cr@438 857 delete[] dbpointers.mean_duration;
mas01cr@452 858
mas01cr@452 859 return 0;
mas01cr@239 860 }