#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 feature = {2, 2, "testfeature", (double[4]) {0, 1, 1, 0}};
  if(audiodb_insert_datum(adb, &feature))
    return 1;
  audiodb_l2norm(adb);

  adb_datum_t query = {1, 2, "testquery", (double[2]) {0, 0.5}};
  adb_query_id_t qid = {0};
  qid.datum = &query;
  qid.sequence_length = 16;
  qid.sequence_start = 0;
  adb_query_parameters_t parms = 
    {ADB_ACCUMULATION_PER_TRACK, ADB_DISTANCE_EUCLIDEAN_NORMED, 10, 10};
  adb_query_refine_t refine = {0};

  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) return 1;

  spec.params.npoints = 1;
  results = audiodb_query_spec(adb, &spec);
  if(results) return 1;

  spec.qid.datum->data = (double [2]) {0.5, 0};
  spec.params.npoints = 10;
  results = audiodb_query_spec(adb, &spec);
  if(results) return 1;
  
  spec.params.npoints = 1;
  results = audiodb_query_spec(adb, &spec);
  if(results) return 1;

  /* the above tests mirror those in the audioDB command-line test
   * suite.  We can test for additional bad input cases too: */

  spec.qid.sequence_start = 1;
  spec.qid.sequence_length = 1;
  results = audiodb_query_spec(adb, &spec);
  if(results) return 1;

  /* and just sanity check that we haven't broken everything */
  spec.qid.sequence_start = 0;
  spec.params.npoints = 2;
  results = audiodb_query_spec(adb, &spec);
  if(!results || results->nresults != 2) return 1;
  audiodb_query_free_results(adb, &spec, results);

  audiodb_close(adb);

  return 104;
}
