annotate libtests/0032/prog1.c @ 506:33913e11348a

Makefile fixup.

Whoops.  audioDB.h currently pulls in soapH.h and cmdline.h, and because
we haven't finished the unintertwingling of the library from everything
else we still need to depend on those files even for LIBOBJS.

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

int main(int argc, char **argv) {
  adb_t *adb;
  const char *keys[2];

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

  adb_datum_t datum1 = {1, 2, "testfeature01", (double[2]) {0, 1}};
  adb_datum_t datum2 = {1, 2, "testfeature10", (double[2]) {1, 0}};
  if(audiodb_insert_datum(adb, &datum1))
    return 1;
  if(audiodb_insert_datum(adb, &datum2))
    return 1;

  adb_datum_t query = {1, 2, "testquery", (double[2]) {0, 0.5}};

  adb_query_id_t qid = {0};
  qid.datum = &query;
  qid.sequence_length = 1;
  qid.sequence_start = 0;
  adb_query_parameters_t parms =
    {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_DOT_PRODUCT, 10, 10};
  adb_query_refine_t refine = {0};
  refine.hopsize = 1;

  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 != 2) return 1;
  result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
  result_present_or_fail(results, "testfeature10", 0, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.flags = ADB_REFINE_INCLUDE_KEYLIST;
  spec.refine.include.nkeys = 0;
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 0) return 1;
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.include.nkeys = 1;
  spec.refine.include.keys = keys;
  spec.refine.include.keys[0] = "testfeature01";
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 1) return 1;
  result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.include.keys[0] = "testfeature10";
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 1) return 1;
  result_present_or_fail(results, "testfeature10", 0, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.include.nkeys = 2;
  spec.refine.include.keys[0] = "testfeature01";
  spec.refine.include.keys[1] = "testfeature10";
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 2) return 1;
  result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
  result_present_or_fail(results, "testfeature10", 0, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.include.nkeys = 0;
  spec.refine.include.keys = NULL;
  spec.refine.flags = ADB_REFINE_EXCLUDE_KEYLIST;
  spec.refine.exclude.nkeys = 1;
  spec.refine.exclude.keys = keys;
  spec.refine.exclude.keys[0] = "testfeature10";
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 1) return 1;
  result_present_or_fail(results, "testfeature01", 0.5, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  spec.refine.exclude.keys[0] = "testfeature01";
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 1) return 1;
  result_present_or_fail(results, "testfeature10", 0, 0, 0);
  audiodb_query_free_results(adb, &spec, results);

  audiodb_close(adb);

  return 104;
}