comparison query.cpp @ 214:3254b15472f3 refactoring

set_up_query()

Sets up the query data (from mmap()ed file).

   }
 
   read(dbfid, *data_buffer_p, trackTable[track] * sizeof(double) * dbH->dim);
 }
 
-void audioDB::trackSequenceQueryNN(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
-  
-  initTables(dbName, inFile);
-  
-  // For each input vector, find the closest pointNN matching output vectors and report
-  // we use stdout in this stub version
-  unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
-  double* query = (double*)(indata+sizeof(int));
-  double* queryCopy = 0;
-
-  if(!(dbH->flags & O2_FLAG_L2NORM) )
+void audioDB::set_up_query(double **qp, double **qnp, unsigned *nvp) {
+  *nvp = (statbuf.st_size - sizeof(int)) / (dbH->dim * sizeof(double));
+  
+  if(!(dbH->flags & O2_FLAG_L2NORM)) {
     error("Database must be L2 normed for sequence query","use -L2NORM");
-
-  if(numVectors<sequenceLength)
+  }
+
+  if(*nvp < sequenceLength) {
     error("Query shorter than requested sequence length", "maybe use -l");
+  }
   
   if(verbosity>1) {
     std::cerr << "performing norms ... "; std::cerr.flush();
   }
+
+  *qp = new double[*nvp * dbH->dim];
+  memcpy(*qp, indata+sizeof(int), *nvp * dbH->dim * sizeof(double));
+  *qnp = new double[*nvp];
+  unitNorm(*qp, dbH->dim, *nvp, *qnp);
+}
+
+void audioDB::trackSequenceQueryNN(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
+  
+  initTables(dbName, inFile);
+  
+  unsigned int numVectors;
+  double *query, *query_data;
+  double *qNorm, *sNorm;
+
+  set_up_query(&query, &qNorm, &numVectors);
+  query_data = query;
+
   unsigned dbVectors = dbH->length/(sizeof(double)*dbH->dim);
 
-  // Make a copy of the query
-  queryCopy = new double[numVectors*dbH->dim];
-  memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
-  qNorm = new double[numVectors];
   sNorm = new double[dbVectors];
-  assert(qNorm&&sNorm&&queryCopy&&sequenceLength);    
-  unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
-  query = queryCopy;
 
   // Make norm measurements relative to sequenceLength
   unsigned i,j;
 
   // Copy the L2 norm values to core to avoid disk random access later on

   }
 
   // Clean up
   if(trackOffsetTable)
     delete[] trackOffsetTable;
-  if(queryCopy)
-    delete[] queryCopy;
+  if(query_data)
+    delete[] query_data;
   if(qNorm)
     delete[] qNorm;
   if(sNorm)
     delete[] sNorm;
   if(qPower)

 
 void audioDB::trackSequenceQueryRad(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
   
   initTables(dbName, inFile);
   
-  // For each input vector, find the closest pointNN matching output vectors and report
-  // we use stdout in this stub version
-  unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
-  double* query = (double*)(indata+sizeof(int));
-  double* queryCopy = 0;
-
-  if(!(dbH->flags & O2_FLAG_L2NORM) )
-    error("Database must be L2 normed for sequence query","use -l2norm");
-  
-  if(verbosity>1) {
-    std::cerr << "performing norms ... "; std::cerr.flush();
-  }
+  unsigned int numVectors;
+  double *query, *query_data;
+  double *qNorm, *sNorm;
+
+  set_up_query(&query, &qNorm, &numVectors);
+  query_data = query;
+
   unsigned dbVectors = dbH->length/(sizeof(double)*dbH->dim);
 
-  // Make a copy of the query
-  queryCopy = new double[numVectors*dbH->dim];
-  memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
-  qNorm = new double[numVectors];
   sNorm = new double[dbVectors];
-  assert(qNorm&&sNorm&&queryCopy&&sequenceLength);    
-  unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
-  query = queryCopy;
 
   // Make norm measurements relative to sequenceLength
   unsigned i,j;
 
   // Copy the L2 norm values to core to avoid disk random access later on

   }
 
   // Clean up
   if(trackOffsetTable)
     delete[] trackOffsetTable;
-  if(queryCopy)
-    delete[] queryCopy;
+  if(query_data)
+    delete[] query_data;
   if(qNorm)
     delete[] qNorm;
   if(sNorm)
     delete[] sNorm;
   if(qPower)