view libtests/0049/prog1.c @ 693:b1723ae7675e

begin work on sampling API

This is motivated by the need to be able to sample with arbitrary feature data
(e.g. from a feature file) against a database, for the JNMR "collections" paper
revisions or possible ISMIR paper revisions.  That bit doesn't work yet, but
the C-ified version of the current functionality (sample db x db and
sample key x db) works to the level of anecdotal tests.

The general approach is to mirror the _query_spec() API, where a whole heap
of knobs and twiddles are available to the user.  Unlike in the _query_spec()
API, not quite all of the knobs make sense (and even fewer are actually
implemented), but the basic idea is the same.

I pity the poor chump who will have to document all this.

#include "audioDB_API.h"
#include "test_utils_lib.h"

int main(int argc, char **argv) {
  adb_t *adb;

  clean_remove_db(TESTDB);
  if(!(adb = audiodb_create(TESTDB, 0, 0, 0)))
    return 1;

  adb_datum_t datum1 = {1, 2, "testfeature1", (double[2]) {0, 1}};
  adb_datum_t datum3 = {3, 2, "testfeature3", (double[6]) {1, 0, 0, 1, 1, 0}};

  if(audiodb_insert_datum(adb, &datum1))
    return 1;
  if(audiodb_insert_datum(adb, &datum3))
    return 1;
  if(audiodb_l2norm(adb))
    return 1;

  adb_datum_t query = {2, 2, "testquery", (double[4]) {0, 1, 1, 0}};
  adb_query_id_t qid = {0};
  qid.datum = &query;
  qid.sequence_length = 1;
  qid.flags |= ADB_QID_FLAG_EXHAUSTIVE;

  adb_query_parameters_t parms =
    {ADB_ACCUMULATION_DB, ADB_DISTANCE_EUCLIDEAN, 10};
  adb_query_refine_t refine = {0};
  refine.qhopsize = 1;
  refine.ihopsize = 1;
  refine.flags |= ADB_REFINE_HOP_SIZE;

  adb_query_spec_t spec;
  spec.qid = qid;
  spec.params = parms;
  spec.refine = refine;

  adb_query_results_t *results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 8) return 1;
  result_present_or_fail(results, "testfeature1", 0, 0, 0);
  result_present_or_fail(results, "testfeature3", 0, 0, 1);
  result_present_or_fail(results, "testfeature3", 0, 1, 0);
  result_present_or_fail(results, "testfeature3", 0, 1, 2);
  result_present_or_fail(results, "testfeature1", 2, 1, 0);
  result_present_or_fail(results, "testfeature3", 2, 1, 1);
  result_present_or_fail(results, "testfeature3", 2, 0, 0);
  result_present_or_fail(results, "testfeature3", 2, 0, 2);
  
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.qhopsize = 2;
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 4) return 1;
  result_present_or_fail(results, "testfeature1", 0, 0, 0);
  result_present_or_fail(results, "testfeature3", 0, 0, 1);
  result_present_or_fail(results, "testfeature3", 2, 0, 0);
  result_present_or_fail(results, "testfeature3", 2, 0, 2);

  audiodb_query_free_results(adb, &spec, results);

  spec.refine.qhopsize = 1;
  spec.refine.ihopsize = 2;
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 6) return 1;
  result_present_or_fail(results, "testfeature1", 0, 0, 0);
  result_present_or_fail(results, "testfeature3", 0, 1, 0);
  result_present_or_fail(results, "testfeature3", 0, 1, 2);
  result_present_or_fail(results, "testfeature1", 2, 1, 0);
  result_present_or_fail(results, "testfeature3", 2, 0, 0);
  result_present_or_fail(results, "testfeature3", 2, 0, 2);

  audiodb_query_free_results(adb, &spec, results);

  spec.refine.ihopsize = 3;
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 4) return 1;
  result_present_or_fail(results, "testfeature1", 0, 0, 0);
  result_present_or_fail(results, "testfeature3", 0, 1, 0);
  result_present_or_fail(results, "testfeature1", 2, 1, 0);
  result_present_or_fail(results, "testfeature3", 2, 0, 0);

  audiodb_query_free_results(adb, &spec, results);

  audiodb_close(adb);

  return 104;
}