mas01cr@678
|
1 #include "audioDB_API.h"
|
mas01cr@678
|
2 #include "test_utils_lib.h"
|
mas01cr@678
|
3
|
mas01cr@678
|
4 int main(int argc, char **argv) {
|
mas01cr@678
|
5 adb_t *adb;
|
mas01cr@678
|
6
|
mas01cr@678
|
7 clean_remove_db(TESTDB);
|
mas01cr@678
|
8 if(!(adb = audiodb_create(TESTDB, 0, 0, 0)))
|
mas01cr@678
|
9 return 1;
|
mas01cr@678
|
10
|
mas01cr@678
|
11 adb_datum_t datum1 = {1, 2, "testfeature1", (double[2]) {0, 1}};
|
mas01cr@678
|
12 adb_datum_t datum3 = {3, 2, "testfeature3", (double[6]) {1, 0, 0, 1, 1, 0}};
|
mas01cr@678
|
13
|
mas01cr@678
|
14 if(audiodb_insert_datum(adb, &datum1))
|
mas01cr@678
|
15 return 1;
|
mas01cr@678
|
16 if(audiodb_insert_datum(adb, &datum3))
|
mas01cr@678
|
17 return 1;
|
mas01cr@678
|
18 if(audiodb_l2norm(adb))
|
mas01cr@678
|
19 return 1;
|
mas01cr@678
|
20
|
mas01cr@678
|
21 adb_datum_t query = {2, 2, "testquery", (double[4]) {0, 1, 1, 0}};
|
mas01cr@678
|
22 adb_query_id_t qid = {0};
|
mas01cr@678
|
23 qid.datum = &query;
|
mas01cr@678
|
24 qid.sequence_length = 1;
|
mas01cr@678
|
25 qid.flags |= ADB_QID_FLAG_EXHAUSTIVE;
|
mas01cr@678
|
26
|
mas01cr@678
|
27 adb_query_parameters_t parms =
|
mas01cr@678
|
28 {ADB_ACCUMULATION_DB, ADB_DISTANCE_EUCLIDEAN, 10};
|
mas01cr@678
|
29 adb_query_refine_t refine = {0};
|
mas01cr@678
|
30 refine.qhopsize = 1;
|
mas01cr@678
|
31 refine.ihopsize = 1;
|
mas01cr@678
|
32 refine.flags |= ADB_REFINE_HOP_SIZE;
|
mas01cr@678
|
33
|
mas01cr@678
|
34 adb_query_spec_t spec;
|
mas01cr@678
|
35 spec.qid = qid;
|
mas01cr@678
|
36 spec.params = parms;
|
mas01cr@678
|
37 spec.refine = refine;
|
mas01cr@678
|
38
|
mas01cr@678
|
39 adb_query_results_t *results = audiodb_query_spec(adb, &spec);
|
mas01cr@678
|
40 if(!results || results->nresults != 8) return 1;
|
mas01cr@678
|
41 result_present_or_fail(results, "testfeature1", 0, 0, 0);
|
mas01cr@678
|
42 result_present_or_fail(results, "testfeature3", 0, 0, 1);
|
mas01cr@678
|
43 result_present_or_fail(results, "testfeature3", 0, 1, 0);
|
mas01cr@678
|
44 result_present_or_fail(results, "testfeature3", 0, 1, 2);
|
mas01cr@678
|
45 result_present_or_fail(results, "testfeature1", 2, 1, 0);
|
mas01cr@678
|
46 result_present_or_fail(results, "testfeature3", 2, 1, 1);
|
mas01cr@678
|
47 result_present_or_fail(results, "testfeature3", 2, 0, 0);
|
mas01cr@678
|
48 result_present_or_fail(results, "testfeature3", 2, 0, 2);
|
mas01cr@678
|
49
|
mas01cr@678
|
50 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@678
|
51
|
mas01cr@678
|
52 spec.refine.qhopsize = 2;
|
mas01cr@678
|
53 results = audiodb_query_spec(adb, &spec);
|
mas01cr@678
|
54 if(!results || results->nresults != 4) return 1;
|
mas01cr@678
|
55 result_present_or_fail(results, "testfeature1", 0, 0, 0);
|
mas01cr@678
|
56 result_present_or_fail(results, "testfeature3", 0, 0, 1);
|
mas01cr@678
|
57 result_present_or_fail(results, "testfeature3", 2, 0, 0);
|
mas01cr@678
|
58 result_present_or_fail(results, "testfeature3", 2, 0, 2);
|
mas01cr@678
|
59
|
mas01cr@678
|
60 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@678
|
61
|
mas01cr@678
|
62 spec.refine.qhopsize = 1;
|
mas01cr@678
|
63 spec.refine.ihopsize = 2;
|
mas01cr@678
|
64 results = audiodb_query_spec(adb, &spec);
|
mas01cr@678
|
65 if(!results || results->nresults != 6) return 1;
|
mas01cr@678
|
66 result_present_or_fail(results, "testfeature1", 0, 0, 0);
|
mas01cr@678
|
67 result_present_or_fail(results, "testfeature3", 0, 1, 0);
|
mas01cr@678
|
68 result_present_or_fail(results, "testfeature3", 0, 1, 2);
|
mas01cr@678
|
69 result_present_or_fail(results, "testfeature1", 2, 1, 0);
|
mas01cr@678
|
70 result_present_or_fail(results, "testfeature3", 2, 0, 0);
|
mas01cr@678
|
71 result_present_or_fail(results, "testfeature3", 2, 0, 2);
|
mas01cr@678
|
72
|
mas01cr@678
|
73 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@678
|
74
|
mas01cr@678
|
75 spec.refine.ihopsize = 3;
|
mas01cr@678
|
76 results = audiodb_query_spec(adb, &spec);
|
mas01cr@678
|
77 if(!results || results->nresults != 4) return 1;
|
mas01cr@678
|
78 result_present_or_fail(results, "testfeature1", 0, 0, 0);
|
mas01cr@678
|
79 result_present_or_fail(results, "testfeature3", 0, 1, 0);
|
mas01cr@678
|
80 result_present_or_fail(results, "testfeature1", 2, 1, 0);
|
mas01cr@678
|
81 result_present_or_fail(results, "testfeature3", 2, 0, 0);
|
mas01cr@678
|
82
|
mas01cr@678
|
83 audiodb_query_free_results(adb, &spec, results);
|
mas01cr@678
|
84
|
mas01cr@678
|
85 audiodb_close(adb);
|
mas01cr@678
|
86
|
mas01cr@678
|
87 return 104;
|
mas01cr@678
|
88 }
|