comparison query.cpp @ 434:7af140bf8a0a api-inversion

adb_t-ize most of audioDB::set_up_db.

All of this "adding code" nonsense (mostly because of the conversion
from C++ non-local exits using audioDB::error into lame C "must return
and check integer error code") has to have a payoff sometime.  That day
will come.  That day will come.

 
 
 // FIXME: this is not the right name; we're not actually setting up
 // the database, but copying various bits of it out of mmap()ed tables
 // in order to reduce seeks.
-void audioDB::set_up_db(double **snp, double **vsnp, double **spp, double **vspp, double **mddp, unsigned int *dvp) {
-  *dvp = dbH->length / (dbH->dim * sizeof(double));
+int audioDB::set_up_db(adb_t *adb, double **snp, double **vsnp, double **spp, double **vspp, double **mddp, unsigned int *dvp) {
+  *dvp = adb->header->length / (adb->header->dim * sizeof(double));
   *snp = new double[*dvp];
 
   double *snpp = *snp, *sppp = 0;
-  memcpy(*snp, l2normTable, *dvp * sizeof(double));
+  lseek(adb->fd, adb->header->l2normTableOffset, SEEK_SET);
+  read_or_goto_error(adb->fd, *snp, *dvp * sizeof(double));
 
   if (usingPower) {
-    if (!(dbH->flags & O2_FLAG_POWER)) {
-      error("database not power-enabled", dbName);
+    if (!(adb->header->flags & O2_FLAG_POWER)) {
+      goto error;
     }
     *spp = new double[*dvp];
     sppp = *spp;
-    memcpy(*spp, powerTable, *dvp * sizeof(double));
-  }
-
-  for(unsigned int i = 0; i < dbH->numFiles; i++){
-    if(trackTable[i] >= sequenceLength) {
-      audiodb_sequence_sum(snpp, trackTable[i], sequenceLength);
-      audiodb_sequence_sqrt(snpp, trackTable[i], sequenceLength);
-
+    lseek(adb->fd, adb->header->powerTableOffset, SEEK_SET);
+    read_or_goto_error(adb->fd, *spp, *dvp * sizeof(double));
+  }
+
+  for(unsigned int i = 0; i < adb->header->numFiles; i++){
+    size_t track_length = (*adb->track_lengths)[i];
+    if(track_length >= sequenceLength) {
+      audiodb_sequence_sum(snpp, track_length, sequenceLength);
+      audiodb_sequence_sqrt(snpp, track_length, sequenceLength);
       if (usingPower) {
-	audiodb_sequence_sum(sppp, trackTable[i], sequenceLength);
-        audiodb_sequence_average(sppp, trackTable[i], sequenceLength);
-      }
-    }
-    snpp += trackTable[i];
+	audiodb_sequence_sum(sppp, track_length, sequenceLength);
+        audiodb_sequence_average(sppp, track_length, sequenceLength);
+      }
+    }
+    snpp += track_length;
     if (usingPower) {
-      sppp += trackTable[i];
+      sppp += track_length;
     }
   }
 
   if (usingTimes) {
-    if(!(dbH->flags & O2_FLAG_TIMES)) {
+    if(!(adb->header->flags & O2_FLAG_TIMES)) {
       error("query timestamps provided for non-timed database", dbName);
     }
 
-    *mddp = new double[dbH->numFiles];
-
-    for(unsigned int k = 0; k < dbH->numFiles; k++) {
+    *mddp = new double[adb->header->numFiles];
+
+    for(unsigned int k = 0; k < adb->header->numFiles; k++) {
+      size_t track_length = (*adb->track_lengths)[k];
       unsigned int j;
       (*mddp)[k] = 0.0;
-      for(j = 0; j < trackTable[k]; j++) {
+      for(j = 0; j < track_length; j++) {
 	(*mddp)[k] += timesTable[2*j+1] - timesTable[2*j];
       }
       (*mddp)[k] /= j;
     }
   }
 
   *vsnp = *snp;
   *vspp = *spp;
+  return 0;
+
+ error:
+  if(*snp) {
+    delete [] *snp;
+  }
+  if(*spp) {
+    delete [] *spp;
+  }
+  if(*mddp) {
+    delete [] *mddp;
+  }
+  return 1;
+  
 }
 
 // query_points()
 //
 // using PointPairs held in the exact_evaluation_queue compute squared distance for each PointPair

 
   // Compute database info
   // FIXME: we more than likely don't need very much of the database
   // so make a new method to build these values per-track or, even better, per-point
   if( !( dbH->flags & O2_FLAG_LARGE_ADB) )
-    set_up_db(&sNorm, &snPtr, &sPower, &spPtr, &meanDBdur, &dbVectors);
+    if(set_up_db(adb, &sNorm, &snPtr, &sPower, &spPtr, &meanDBdur, &dbVectors)) {
+      error("failed to set up db");
+    }
 
   VERB_LOG(1, "matching points...");
 
   // We are guaranteed that the order of points is sorted by:
   // trackID, spos, qpos

 
   unsigned int dbVectors;
   double *sNorm, *snPtr, *sPower = 0, *spPtr = 0;
   double *meanDBdur = 0;
 
-  set_up_db(&sNorm, &snPtr, &sPower, &spPtr, &meanDBdur, &dbVectors);
+  if(set_up_db(adb, &sNorm, &snPtr, &sPower, &spPtr, &meanDBdur, &dbVectors)) {
+    error("failed to set up db");
+  }
 
   VERB_LOG(1, "matching tracks...");
   
   unsigned j,k,track,trackOffset=0, HOP_SIZE=sequenceHop, wL=sequenceLength;
   double **D = 0;    // Differences query and target