comparison index.cpp @ 466:11fccb6a3bd5 api-inversion

Almost there!

audioDB::index_query_loop is now set, except for the lsh_in_core and
lsh_exact flags.  I think the plan is to zap lsh_in_core entirely, and
add some flags to the qid structure for indexed behaviour.

     PointPair p(trackID, qpos, spos);
     qstate->exact_evaluation_queue->push(p);
   }
 }
 
+// return -1 on error
 // return 0: if index does not exist
 // return nqv: if index exists
 int audioDB::index_query_loop(adb_t *adb, adb_query_spec_t *spec, adb_qstate_internal_t *qstate) {
   
   double *query = 0, *query_data = 0;

 
   adb_qcallback_t callback_data;
   callback_data.adb = adb;
   callback_data.qstate = qstate;
 
-  void (*add_point_func)(void*,Uns32T,Uns32T,float);
-
-  sequenceLength = spec->qid.sequence_length;
-  normalizedDistance = (spec->params.distance == ADB_DISTANCE_EUCLIDEAN_NORMED);
+  void (*add_point_func)(void *, uint32_t, uint32_t, float);
+
+  uint32_t sequence_length = spec->qid.sequence_length;
+  bool normalized = (spec->params.distance == ADB_DISTANCE_EUCLIDEAN_NORMED);
+  double radius = spec->refine.radius;
+  bool use_absolute_threshold = spec->refine.flags & ADB_REFINE_ABSOLUTE_THRESHOLD;
+  double absolute_threshold = spec->refine.absolute_threshold;
 
   // Set the point-reporter callback based on the value of lsh_exact
   if(lsh_exact) {
     qstate->exact_evaluation_queue = new std::priority_queue<PointPair>;
     add_point_func = &audiodb_index_add_point_exact;

 
   if(!audiodb_index_init_query(adb, spec, qstate, lsh_in_core)) {
     return 0;
   }
 
-  char *database = audiodb_index_get_name(adb->path, radius, sequenceLength);
+  char *database = audiodb_index_get_name(adb->path, radius, sequence_length);
   if(!database) {
-    error("failed to get index name", adb->path);
+    return -1;
   }
 
   if(audiodb_query_spec_qpointers(adb, spec, &query_data, &query, &qpointers)) {
-    error("failed to set up qpointers");
-  }
-
-  // query vector index
-  Uns32T Nq = (qpointers.nvectors>O2_MAXTRACKLEN?O2_MAXTRACKLEN:qpointers.nvectors) - sequenceLength + 1;
-  std::vector<std::vector<float> > *vv = audiodb_index_initialize_shingles(Nq, adb->header->dim, sequenceLength); // allocate memory to copy query vectors to shingles
+    delete [] database;
+    return -1;
+  }
+
+  uint32_t Nq = (qpointers.nvectors > O2_MAXTRACKLEN ? O2_MAXTRACKLEN : qpointers.nvectors) - sequence_length + 1;
+  std::vector<std::vector<float> > *vv = audiodb_index_initialize_shingles(Nq, adb->header->dim, sequence_length);
 
   // Construct shingles from query features  
-  for( Uns32T pointID = 0 ; pointID < Nq ; pointID++ ) 
-    audiodb_index_make_shingle(vv, pointID, query, dbH->dim, sequenceLength);
+  for(uint32_t pointID = 0; pointID < Nq; pointID++) 
+    audiodb_index_make_shingle(vv, pointID, query, adb->header->dim, sequence_length);
   
   // Normalize query vectors
-  int vcount = audiodb_index_norm_shingles(vv, qpointers.l2norm, qpointers.power, dbH->dim, sequenceLength, radius, normalizedDistance, use_absolute_threshold, absolute_threshold);
+  int vcount = audiodb_index_norm_shingles(vv, qpointers.l2norm, qpointers.power, adb->header->dim, sequence_length, radius, normalized, use_absolute_threshold, absolute_threshold);
   if(vcount == -1) {
     audiodb_index_delete_shingles(vv);
-    error("failed to norm shingles");
-  }
-  Uns32T numVecsAboveThreshold = vcount;
+    delete [] database;
+    return -1;
+  }
+  uint32_t numVecsAboveThreshold = vcount;
 
   // Nq contains number of inspected points in query file, 
   // numVecsAboveThreshold is number of points with power >= absolute_threshold
-  double* qpp = qpointers.power; // Keep original qpPtr for possible exact evaluation
-  if(usingQueryPoint && numVecsAboveThreshold){
-    if((qstate->lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2) || lsh_in_core)
-      qstate->lsh->retrieve_point((*vv)[0], queryPoint, add_point_func, &callback_data);
-    else
-      qstate->lsh->serial_retrieve_point(database, (*vv)[0], queryPoint, add_point_func, &callback_data);
-  }
-  else if(numVecsAboveThreshold)
-    for( Uns32T pointID = 0 ; pointID < Nq; pointID++ )
+  double *qpp = qpointers.power; // Keep original qpPtr for possible exact evaluation
+  if(!(spec->qid.flags & ADB_QUERY_ID_FLAG_EXHAUSTIVE) && numVecsAboveThreshold) {
+    if((qstate->lsh->get_lshHeader()->flags & O2_SERIAL_FILEFORMAT2) || lsh_in_core) {
+      qstate->lsh->retrieve_point((*vv)[0], spec->qid.sequence_start, add_point_func, &callback_data);
+    } else {
+      qstate->lsh->serial_retrieve_point(database, (*vv)[0], spec->qid.sequence_start, add_point_func, &callback_data);
+    }
+  } else if(numVecsAboveThreshold) {
+    for(uint32_t pointID = 0; pointID < Nq; pointID++) {
       if(!use_absolute_threshold || (use_absolute_threshold && (*qpp++ >= absolute_threshold))) {
-	if((qstate->lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2) || lsh_in_core) {
+	if((qstate->lsh->get_lshHeader()->flags & O2_SERIAL_FILEFORMAT2) || lsh_in_core) {
 	  qstate->lsh->retrieve_point((*vv)[pointID], pointID, add_point_func, &callback_data);
         } else {
 	  qstate->lsh->serial_retrieve_point(database, (*vv)[pointID], pointID, add_point_func, &callback_data);   
         }
       }
+    }
+  }
   audiodb_index_delete_shingles(vv);
 
-  if(lsh_exact)
-    // Perform exact distance computation on point pairs in exact_evaluation_queue
+  if(lsh_exact) {
+    // Perform exact distance computation on point pairs in
+    // exact_evaluation_queue
     audiodb_query_queue_loop(adb, spec, qstate, query, &qpointers);
+  }
   
  // Clean up
   if(query_data)
     delete[] query_data;
   if(qpointers.l2norm_data)