view liszt.cpp @ 659:536cfa209e7f

Sampling enhancements

Allow specification of a key with -k for doing sampling between a single
datum and the rest of the database.  (The implementation is slightly
icky, as mentioned in the previous commit message; hopefully this
situation will not remain for too long).

While we're at it, allow -r/--resultlength as a synonym for --nsamples
(if --nsamples is not given) in order better to reflect the analogy
between querying and sampling.

extern "C" {
#include "audioDB_API.h"
}
#include "audioDB-internals.h"

adb_liszt_results_t *audiodb_liszt(adb_t *adb) {
  uint32_t nfiles = adb->header->numFiles;
  adb_liszt_results_t *results;
  results = (adb_liszt_results_t *) calloc(sizeof(adb_liszt_results_t),1);
  results->nresults = nfiles;
  if(nfiles > 0) {
    results->entries = (adb_track_entry_t *) malloc(nfiles * sizeof(adb_track_entry_t));
  }
  for(uint32_t k = 0; k < nfiles; k++) {
    results->entries[k].nvectors = (*adb->track_lengths)[k];
    results->entries[k].key = audiodb_index_key(adb, k);
  }
  return results;
}

int audiodb_liszt_free_results(adb_t *adb, adb_liszt_results_t *results) {
  free(results->entries);
  free(results);
  return 0;
}