comparison query.cpp @ 529:e532666226bc multiprobeLSH

Fixed a problem with inserting into exact_evaluation_queue, duplicate pairs were being inserted when querying multiple hashtables. That is now fixed by adding a unique constraint using set<PointPair> inserts and testing. This should speed things up quite a bit (like, 10-fold with default num tables).

   *vector_data = new double[vector_size];
   qpointers->l2norm_data = new double[vector_size / d->dim];
   if(d->power)
     qpointers->power_data = new double[vector_size / d->dim];
 
-  uint32_t seq_len_dbl = sequence_length*d->dim*sizeof(double);
+  uint32_t seq_len_dim_dbl = sequence_length*d->dim*sizeof(double);
+  uint32_t seq_len_dbl = sequence_length*sizeof(double);
   PointPair pp = ppairs.top();
   uint32_t tid = pp.trackID;
 
   while( !ppairs.empty() && pp.trackID==tid){
     uint32_t spos = pp.spos;
     uint32_t spos_dim = spos*d->dim;
-
-    memcpy(*vector_data+spos_dim, d->data+spos_dim, seq_len_dbl); 
-
-    audiodb_l2norm_buffer(*vector_data+spos_dim, d->dim, 1, qpointers->l2norm_data+spos);
+#ifdef _LSH_DEBUG_
+    cout << "tid=" << pp.trackID << " qpos=" << pp.qpos << " spos=" << pp.spos << endl;
+    cout.flush();
+#endif
+    memcpy((void*)(*vector_data+spos_dim), (void*)(d->data+spos_dim), seq_len_dim_dbl); 
+
+    audiodb_l2norm_buffer(*vector_data+spos_dim, d->dim, sequence_length, qpointers->l2norm_data+spos);
     audiodb_sequence_sum(qpointers->l2norm_data+spos, sequence_length, sequence_length);
     audiodb_sequence_sqrt(qpointers->l2norm_data+spos, sequence_length, sequence_length);
 
     if(d->power) {
       memcpy(qpointers->power_data+spos, d->power+spos, seq_len_dbl);

       maybe_delete_array(dbpointers.mean_duration);
       currentTrack = pp.trackID;
       adb_datum_t d = {0};
       if(audiodb_track_id_datum(adb, pp.trackID, &d)) {
         delete qstate->exact_evaluation_queue;
+        delete qstate->set;
         return 1;
       }
-      if(audiodb_datum_qpointers_partial(&d, sequence_length, &dbdata, &dbdata_pointer, &dbpointers, qstate)) {
+      //      if(audiodb_datum_qpointers_partial(&d, sequence_length, &dbdata, &dbdata_pointer, &dbpointers, qstate)) {
+      if(audiodb_datum_qpointers(&d, sequence_length, &dbdata, &dbdata_pointer, &dbpointers)) {
         delete qstate->exact_evaluation_queue;
+        delete qstate->set;
         audiodb_free_datum(&d);
         return 1;
       }
       audiodb_free_datum(&d);
     }

   maybe_delete_array(dbdata);
   maybe_delete_array(dbpointers.l2norm_data);
   maybe_delete_array(dbpointers.power_data);
   maybe_delete_array(dbpointers.mean_duration);
   delete qstate->exact_evaluation_queue;
+  delete qstate->set;
   return 0;
 }
 
 int audiodb_query_loop(adb_t *adb, const adb_query_spec_t *spec, adb_qstate_internal_t *qstate) {