annotate audioDB.cpp @ 177:c32bf13c3978 no-big-mmap

forWrite audioDB flag (rather than function argument) so that we can tell when it comes to munmap() how much we need to. ALIGN_PAGE_FOO() takes one argument not two.
author mas01cr
date Thu, 15 Nov 2007 14:58:16 +0000
parents 8193dbd66e34
children 7bbe5d48a7ef
rev   line source
mas01cr@0 1 #include "audioDB.h"
mas01cr@0 2
mas01cr@105 3 #if defined(O2_DEBUG)
mas01cr@104 4 void sigterm_action(int signal, siginfo_t *info, void *context) {
mas01cr@104 5 exit(128+signal);
mas01cr@104 6 }
mas01cr@104 7
mas01cr@104 8 void sighup_action(int signal, siginfo_t *info, void *context) {
mas01cr@104 9 // FIXME: reread any configuration files
mas01cr@104 10 }
mas01cr@105 11 #endif
mas01cr@0 12
mas01cr@32 13 void audioDB::error(const char* a, const char* b, const char *sysFunc) {
mas01cr@77 14 if(isServer) {
mas01cr@81 15 /* FIXME: I think this is leaky -- we never delete err. actually
mas01cr@81 16 deleting it is tricky, though; it gets placed into some
mas01cr@81 17 soap-internal struct with uncertain extent... -- CSR,
mas01cr@81 18 2007-10-01 */
mas01cr@77 19 char *err = new char[256]; /* FIXME: overflows */
mas01cr@77 20 snprintf(err, 255, "%s: %s\n%s", a, b, sysFunc ? strerror(errno) : "");
mas01cr@77 21 /* FIXME: actually we could usefully do with a properly structured
mas01cr@77 22 type, so that we can throw separate faultstring and details.
mas01cr@77 23 -- CSR, 2007-10-01 */
mas01cr@77 24 throw(err);
mas01cr@77 25 } else {
mas01cr@77 26 cerr << a << ": " << b << endl;
mas01cr@77 27 if (sysFunc) {
mas01cr@77 28 perror(sysFunc);
mas01cr@77 29 }
mas01cr@77 30 exit(1);
mas01cr@32 31 }
mas01cr@0 32 }
mas01cr@0 33
mas01cr@76 34 audioDB::audioDB(const unsigned argc, char* const argv[]): O2_AUDIODB_INITIALIZERS
mas01cr@76 35 {
mas01cr@0 36 if(processArgs(argc, argv)<0){
mas01cr@0 37 printf("No command found.\n");
mas01cr@0 38 cmdline_parser_print_version ();
mas01cr@0 39 if (strlen(gengetopt_args_info_purpose) > 0)
mas01cr@0 40 printf("%s\n", gengetopt_args_info_purpose);
mas01cr@0 41 printf("%s\n", gengetopt_args_info_usage);
mas01cr@0 42 printf("%s\n", gengetopt_args_info_help[1]);
mas01cr@0 43 printf("%s\n", gengetopt_args_info_help[2]);
mas01cr@0 44 printf("%s\n", gengetopt_args_info_help[0]);
mas01cr@151 45 error("No command found");
mas01cr@0 46 }
mas01cr@77 47
mas01cr@0 48 if(O2_ACTION(COM_SERVER))
mas01cr@0 49 startServer();
mas01cr@0 50
mas01cr@0 51 else if(O2_ACTION(COM_CREATE))
mas01cr@0 52 create(dbName);
mas01cr@0 53
mas01cr@0 54 else if(O2_ACTION(COM_INSERT))
mas01cr@0 55 insert(dbName, inFile);
mas01cr@0 56
mas01cr@0 57 else if(O2_ACTION(COM_BATCHINSERT))
mas01cr@0 58 batchinsert(dbName, inFile);
mas01cr@0 59
mas01cr@0 60 else if(O2_ACTION(COM_QUERY))
mas01cr@0 61 if(isClient)
mas01cr@0 62 ws_query(dbName, inFile, (char*)hostport);
mas01cr@0 63 else
mas01cr@76 64 query(dbName, inFile);
mas01cr@0 65
mas01cr@0 66 else if(O2_ACTION(COM_STATUS))
mas01cr@0 67 if(isClient)
mas01cr@0 68 ws_status(dbName,(char*)hostport);
mas01cr@0 69 else
mas01cr@0 70 status(dbName);
mas01cr@0 71
mas01cr@0 72 else if(O2_ACTION(COM_L2NORM))
mas01cr@0 73 l2norm(dbName);
mas01cr@0 74
mas01cr@0 75 else if(O2_ACTION(COM_DUMP))
mas01cr@0 76 dump(dbName);
mas01cr@0 77
mas01cr@0 78 else
mas01cr@0 79 error("Unrecognized command",command);
mas01cr@0 80 }
mas01cr@0 81
mas01cr@133 82 audioDB::audioDB(const unsigned argc, char* const argv[], adb__queryResponse *adbQueryResponse): O2_AUDIODB_INITIALIZERS
mas01cr@76 83 {
mas01cr@97 84 try {
mas01cr@151 85 isServer = 1; // FIXME: Hack
mas01cr@97 86 processArgs(argc, argv);
mas01cr@97 87 assert(O2_ACTION(COM_QUERY));
mas01cr@133 88 query(dbName, inFile, adbQueryResponse);
mas01cr@97 89 } catch(char *err) {
mas01cr@97 90 cleanup();
mas01cr@97 91 throw(err);
mas01cr@97 92 }
mas01cr@76 93 }
mas01cr@76 94
mas01cr@133 95 audioDB::audioDB(const unsigned argc, char* const argv[], adb__statusResponse *adbStatusResponse): O2_AUDIODB_INITIALIZERS
mas01cr@76 96 {
mas01cr@97 97 try {
mas01cr@151 98 isServer = 1; // FIXME: Hack
mas01cr@97 99 processArgs(argc, argv);
mas01cr@97 100 assert(O2_ACTION(COM_STATUS));
mas01cr@133 101 status(dbName, adbStatusResponse);
mas01cr@97 102 } catch(char *err) {
mas01cr@97 103 cleanup();
mas01cr@97 104 throw(err);
mas01cr@97 105 }
mas01cr@76 106 }
mas01cr@76 107
mas01cr@97 108 void audioDB::cleanup() {
mas01cr@122 109 cmdline_parser_free(&args_info);
mas01cr@0 110 if(indata)
mas01cr@0 111 munmap(indata,statbuf.st_size);
mas01cr@0 112 if(db)
mas01cr@176 113 munmap(db,getpagesize());
mas01cr@175 114 if(fileTable)
mas01cr@175 115 munmap(fileTable, dbH->trackTableOffset - dbH->fileTableOffset);
mas01cr@175 116 if(trackTable)
mas01cr@175 117 munmap(trackTable, dbH->dataOffset - dbH->trackTableOffset);
mas01cr@175 118 if(dataBuf)
mas01cr@175 119 munmap(dataBuf, dbH->timesTableOffset - dbH->dataOffset);
mas01cr@175 120 if(timesTable)
mas01cr@175 121 munmap(dataBuf, dbH->l2normTableOffset - dbH->timesTableOffset);
mas01cr@175 122 if(l2normTable)
mas01cr@175 123 munmap(l2normTable, dbH->dbSize - dbH->l2normTableOffset);
mas01cr@175 124
mas01cr@0 125 if(dbfid>0)
mas01cr@0 126 close(dbfid);
mas01cr@0 127 if(infid>0)
mas01cr@0 128 close(infid);
mas01cr@0 129 if(dbH)
mas01cr@0 130 delete dbH;
mas01cr@0 131 }
mas01cr@0 132
mas01cr@97 133 audioDB::~audioDB(){
mas01cr@97 134 cleanup();
mas01cr@97 135 }
mas01cr@97 136
mas01cr@0 137 int audioDB::processArgs(const unsigned argc, char* const argv[]){
mas01cr@0 138
mas01cr@0 139 if(argc<2){
mas01cr@0 140 cmdline_parser_print_version ();
mas01cr@0 141 if (strlen(gengetopt_args_info_purpose) > 0)
mas01cr@0 142 printf("%s\n", gengetopt_args_info_purpose);
mas01cr@0 143 printf("%s\n", gengetopt_args_info_usage);
mas01cr@0 144 printf("%s\n", gengetopt_args_info_help[1]);
mas01cr@0 145 printf("%s\n", gengetopt_args_info_help[2]);
mas01cr@0 146 printf("%s\n", gengetopt_args_info_help[0]);
mas01cr@0 147 exit(0);
mas01cr@0 148 }
mas01cr@0 149
mas01cr@0 150 if (cmdline_parser (argc, argv, &args_info) != 0)
mas01cr@151 151 error("Error parsing command line");
mas01cr@0 152
mas01cr@0 153 if(args_info.help_given){
mas01cr@0 154 cmdline_parser_print_help();
mas01cr@0 155 exit(0);
mas01cr@0 156 }
mas01cr@0 157
mas01cr@0 158 if(args_info.verbosity_given){
mas01cr@0 159 verbosity=args_info.verbosity_arg;
mas01cr@0 160 if(verbosity<0 || verbosity>10){
mas01cr@0 161 cerr << "Warning: verbosity out of range, setting to 1" << endl;
mas01cr@0 162 verbosity=1;
mas01cr@0 163 }
mas01cr@0 164 }
mas01cr@0 165
mas01cr@129 166 if(args_info.size_given) {
mas01cr@140 167 if (args_info.size_arg < 50 || args_info.size_arg > 4000) {
mas01cr@129 168 error("Size out of range", "");
mas01cr@129 169 }
mas01cr@129 170 size = args_info.size_arg * 1000000;
mas01cr@129 171 }
mas01cr@129 172
mas01mc@17 173 if(args_info.radius_given){
mas01mc@17 174 radius=args_info.radius_arg;
mas01mc@17 175 if(radius<=0 || radius>1000000000){
mas01cr@77 176 error("radius out of range");
mas01mc@17 177 }
mas01mc@17 178 else
mas01cr@60 179 if(verbosity>3) {
mas01mc@17 180 cerr << "Setting radius to " << radius << endl;
mas01cr@60 181 }
mas01mc@17 182 }
mas01mc@17 183
mas01cr@0 184 if(args_info.SERVER_given){
mas01cr@0 185 command=COM_SERVER;
mas01cr@0 186 port=args_info.SERVER_arg;
mas01cr@0 187 if(port<100 || port > 100000)
mas01cr@0 188 error("port out of range");
mas01cr@151 189 isServer = 1;
mas01cr@105 190 #if defined(O2_DEBUG)
mas01cr@104 191 struct sigaction sa;
mas01cr@104 192 sa.sa_sigaction = sigterm_action;
mas01cr@104 193 sa.sa_flags = SA_SIGINFO | SA_RESTART | SA_NODEFER;
mas01cr@104 194 sigaction(SIGTERM, &sa, NULL);
mas01cr@104 195 sa.sa_sigaction = sighup_action;
mas01cr@104 196 sa.sa_flags = SA_SIGINFO | SA_RESTART | SA_NODEFER;
mas01cr@104 197 sigaction(SIGHUP, &sa, NULL);
mas01cr@105 198 #endif
mas01cr@0 199 return 0;
mas01cr@0 200 }
mas01cr@0 201
mas01cr@0 202 // No return on client command, find database command
mas01cr@105 203 if(args_info.client_given){
mas01cr@105 204 command=COM_CLIENT;
mas01cr@105 205 hostport=args_info.client_arg;
mas01cr@105 206 isClient=1;
mas01cr@105 207 }
mas01cr@0 208
mas01cr@105 209 if(args_info.NEW_given){
mas01cr@105 210 command=COM_CREATE;
mas01cr@105 211 dbName=args_info.database_arg;
mas01cr@105 212 return 0;
mas01cr@105 213 }
mas01cr@0 214
mas01cr@105 215 if(args_info.STATUS_given){
mas01cr@105 216 command=COM_STATUS;
mas01cr@105 217 dbName=args_info.database_arg;
mas01cr@105 218 return 0;
mas01cr@105 219 }
mas01cr@0 220
mas01cr@105 221 if(args_info.DUMP_given){
mas01cr@105 222 command=COM_DUMP;
mas01cr@105 223 dbName=args_info.database_arg;
mas01cr@131 224 output = args_info.output_arg;
mas01cr@105 225 return 0;
mas01cr@105 226 }
mas01cr@0 227
mas01cr@105 228 if(args_info.L2NORM_given){
mas01cr@105 229 command=COM_L2NORM;
mas01cr@105 230 dbName=args_info.database_arg;
mas01cr@105 231 return 0;
mas01cr@105 232 }
mas01cr@0 233
mas01cr@105 234 if(args_info.INSERT_given){
mas01cr@105 235 command=COM_INSERT;
mas01cr@105 236 dbName=args_info.database_arg;
mas01cr@105 237 inFile=args_info.features_arg;
mas01cr@105 238 if(args_info.key_given)
mas01cr@105 239 key=args_info.key_arg;
mas01cr@105 240 if(args_info.times_given){
mas01cr@105 241 timesFileName=args_info.times_arg;
mas01cr@105 242 if(strlen(timesFileName)>0){
mas01cr@105 243 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@105 244 error("Could not open times file for reading", timesFileName);
mas01cr@105 245 usingTimes=1;
mas01cr@105 246 }
mas01cr@105 247 }
mas01cr@105 248 return 0;
mas01cr@105 249 }
mas01cr@105 250
mas01cr@105 251 if(args_info.BATCHINSERT_given){
mas01cr@105 252 command=COM_BATCHINSERT;
mas01cr@105 253 dbName=args_info.database_arg;
mas01cr@105 254 inFile=args_info.featureList_arg;
mas01cr@105 255 if(args_info.keyList_given)
mas01cr@105 256 key=args_info.keyList_arg; // INCONSISTENT NO CHECK
mas01cr@0 257
mas01cr@105 258 /* TO DO: REPLACE WITH
mas01cr@0 259 if(args_info.keyList_given){
mas01mc@18 260 trackFileName=args_info.keyList_arg;
mas01mc@18 261 if(strlen(trackFileName)>0 && !(trackFile = new ifstream(trackFileName,ios::in)))
mas01mc@18 262 error("Could not open keyList file for reading",trackFileName);
mas01cr@0 263 }
mas01cr@0 264 AND UPDATE BATCHINSERT()
mas01cr@105 265 */
mas01cr@105 266
mas01cr@105 267 if(args_info.timesList_given){
mas01cr@105 268 timesFileName=args_info.timesList_arg;
mas01cr@105 269 if(strlen(timesFileName)>0){
mas01cr@105 270 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@105 271 error("Could not open timesList file for reading", timesFileName);
mas01cr@105 272 usingTimes=1;
mas01cr@105 273 }
mas01cr@105 274 }
mas01cr@105 275 return 0;
mas01cr@105 276 }
mas01cr@105 277
mas01cr@105 278 // Query command and arguments
mas01cr@105 279 if(args_info.QUERY_given){
mas01cr@105 280 command=COM_QUERY;
mas01cr@105 281 dbName=args_info.database_arg;
mas01cr@105 282 inFile=args_info.features_arg;
mas01cr@105 283
mas01cr@105 284 if(args_info.keyList_given){
mas01cr@105 285 trackFileName=args_info.keyList_arg;
mas01cr@105 286 if(strlen(trackFileName)>0 && !(trackFile = new ifstream(trackFileName,ios::in)))
mas01cr@105 287 error("Could not open keyList file for reading",trackFileName);
mas01cr@105 288 }
mas01cr@105 289
mas01cr@105 290 if(args_info.times_given){
mas01cr@105 291 timesFileName=args_info.times_arg;
mas01cr@105 292 if(strlen(timesFileName)>0){
mas01cr@105 293 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@105 294 error("Could not open times file for reading", timesFileName);
mas01cr@105 295 usingTimes=1;
mas01cr@105 296 }
mas01cr@105 297 }
mas01cr@105 298
mas01cr@105 299 // query type
mas01cr@105 300 if(strncmp(args_info.QUERY_arg, "track", MAXSTR)==0)
mas01cr@105 301 queryType=O2_TRACK_QUERY;
mas01cr@105 302 else if(strncmp(args_info.QUERY_arg, "point", MAXSTR)==0)
mas01cr@105 303 queryType=O2_POINT_QUERY;
mas01cr@105 304 else if(strncmp(args_info.QUERY_arg, "sequence", MAXSTR)==0)
mas01cr@105 305 queryType=O2_SEQUENCE_QUERY;
mas01cr@105 306 else
mas01cr@105 307 error("unsupported query type",args_info.QUERY_arg);
mas01cr@105 308
mas01cr@105 309 if(!args_info.exhaustive_flag){
mas01cr@105 310 queryPoint = args_info.qpoint_arg;
mas01cr@105 311 usingQueryPoint=1;
mas01cr@105 312 if(queryPoint<0 || queryPoint >10000)
mas01cr@105 313 error("queryPoint out of range: 0 <= queryPoint <= 10000");
mas01cr@105 314 }
mas01cr@105 315
mas01cr@105 316 pointNN = args_info.pointnn_arg;
mas01cr@105 317 if(pointNN < 1 || pointNN > 1000) {
mas01cr@105 318 error("pointNN out of range: 1 <= pointNN <= 1000");
mas01cr@105 319 }
mas01cr@105 320 trackNN = args_info.resultlength_arg;
mas01cr@105 321 if(trackNN < 1 || trackNN > 1000) {
mas01cr@105 322 error("resultlength out of range: 1 <= resultlength <= 1000");
mas01cr@105 323 }
mas01cr@105 324 sequenceLength = args_info.sequencelength_arg;
mas01cr@105 325 if(sequenceLength < 1 || sequenceLength > 1000) {
mas01cr@105 326 error("seqlen out of range: 1 <= seqlen <= 1000");
mas01cr@105 327 }
mas01cr@105 328 sequenceHop = args_info.sequencehop_arg;
mas01cr@105 329 if(sequenceHop < 1 || sequenceHop > 1000) {
mas01cr@105 330 error("seqhop out of range: 1 <= seqhop <= 1000");
mas01cr@105 331 }
mas01cr@105 332 return 0;
mas01cr@105 333 }
mas01cr@105 334 return -1; // no command found
mas01cr@0 335 }
mas01cr@0 336
mas01cr@30 337 void audioDB::get_lock(int fd, bool exclusive) {
mas01cr@30 338 struct flock lock;
mas01cr@30 339 int status;
mas01cr@30 340
mas01cr@30 341 lock.l_type = exclusive ? F_WRLCK : F_RDLCK;
mas01cr@30 342 lock.l_whence = SEEK_SET;
mas01cr@30 343 lock.l_start = 0;
mas01cr@30 344 lock.l_len = 0; /* "the whole file" */
mas01cr@30 345
mas01cr@30 346 retry:
mas01cr@30 347 do {
mas01cr@30 348 status = fcntl(fd, F_SETLKW, &lock);
mas01cr@30 349 } while (status != 0 && errno == EINTR);
mas01cr@30 350
mas01cr@30 351 if (status) {
mas01cr@30 352 if (errno == EAGAIN) {
mas01cr@30 353 sleep(1);
mas01cr@30 354 goto retry;
mas01cr@30 355 } else {
mas01cr@32 356 error("fcntl lock error", "", "fcntl");
mas01cr@30 357 }
mas01cr@30 358 }
mas01cr@30 359 }
mas01cr@30 360
mas01cr@30 361 void audioDB::release_lock(int fd) {
mas01cr@30 362 struct flock lock;
mas01cr@30 363 int status;
mas01cr@30 364
mas01cr@30 365 lock.l_type = F_UNLCK;
mas01cr@30 366 lock.l_whence = SEEK_SET;
mas01cr@30 367 lock.l_start = 0;
mas01cr@30 368 lock.l_len = 0;
mas01cr@30 369
mas01cr@30 370 status = fcntl(fd, F_SETLKW, &lock);
mas01cr@30 371
mas01cr@30 372 if (status)
mas01cr@32 373 error("fcntl unlock error", "", "fcntl");
mas01cr@30 374 }
mas01cr@30 375
mas01cr@120 376 /* Make a new database.
mas01cr@120 377
mas01cr@120 378 The database consists of:
mas01cr@120 379
mas01cr@120 380 * a header (see dbTableHeader struct definition);
mas01cr@120 381 * keyTable: list of keys of tracks;
mas01cr@120 382 * trackTable: Maps implicit feature index to a feature vector
mas01cr@120 383 matrix (sizes of tracks)
mas01cr@120 384 * featureTable: Lots of doubles;
mas01cr@120 385 * timesTable: time points for each feature vector;
mas01cr@120 386 * l2normTable: squared l2norms for each feature vector.
mas01cr@120 387 */
mas01cr@0 388 void audioDB::create(const char* dbName){
mas01cr@31 389 if ((dbfid = open (dbName, O_RDWR|O_CREAT|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0)
mas01cr@32 390 error("Can't create database file", dbName, "open");
mas01cr@30 391 get_lock(dbfid, 1);
mas01cr@0 392
mas01cr@60 393 if(verbosity) {
mas01cr@0 394 cerr << "header size:" << O2_HEADERSIZE << endl;
mas01cr@60 395 }
mas01cr@0 396
mas01cr@0 397 dbH = new dbTableHeaderT();
mas01cr@0 398 assert(dbH);
mas01cr@0 399
mas01cr@140 400 unsigned int maxfiles = (unsigned int) rint((double) O2_MAXFILES * (double) size / (double) O2_DEFAULTDBSIZE);
mas01cr@140 401
mas01cr@0 402 // Initialize header
mas01cr@108 403 dbH->magic = O2_MAGIC;
mas01cr@108 404 dbH->version = O2_FORMAT_VERSION;
mas01cr@108 405 dbH->numFiles = 0;
mas01cr@108 406 dbH->dim = 0;
mas01cr@108 407 dbH->flags = 0;
mas01cr@108 408 dbH->length = 0;
mas01cr@177 409 dbH->fileTableOffset = ALIGN_PAGE_UP(O2_HEADERSIZE);
mas01cr@177 410 dbH->trackTableOffset = ALIGN_PAGE_UP(dbH->fileTableOffset + O2_FILETABLESIZE*maxfiles);
mas01cr@177 411 dbH->dataOffset = ALIGN_PAGE_UP(dbH->trackTableOffset + O2_TRACKTABLESIZE*maxfiles);
mas01cr@177 412 dbH->l2normTableOffset = ALIGN_PAGE_DOWN(size - maxfiles*O2_MEANNUMVECTORS*sizeof(double));
mas01cr@177 413 dbH->timesTableOffset = ALIGN_PAGE_DOWN(dbH->l2normTableOffset - maxfiles*O2_MEANNUMVECTORS*sizeof(double));
mas01cr@129 414 dbH->dbSize = size;
mas01cr@0 415
mas01cr@173 416 write(dbfid, dbH, O2_HEADERSIZE);
mas01cr@173 417
mas01cr@173 418 // go to the location corresponding to the last byte
mas01cr@173 419 if (lseek (dbfid, size - 1, SEEK_SET) == -1)
mas01cr@173 420 error("lseek error in db file", "", "lseek");
mas01cr@173 421
mas01cr@173 422 // write a dummy byte at the last location
mas01cr@173 423 if (write (dbfid, "", 1) != 1)
mas01cr@173 424 error("write error", "", "write");
mas01cr@173 425
mas01cr@60 426 if(verbosity) {
mas01cr@0 427 cerr << COM_CREATE << " " << dbName << endl;
mas01cr@60 428 }
mas01cr@0 429 }
mas01cr@0 430
mas01cr@0 431 void audioDB::drop(){
mas01cr@108 432 // FIXME: drop something? Should we even allow this?
mas01cr@0 433 }
mas01cr@0 434
mas01cr@177 435 void audioDB::initDBHeader(const char* dbName) {
mas01cr@106 436 if ((dbfid = open(dbName, forWrite ? O_RDWR : O_RDONLY)) < 0) {
mas01cr@32 437 error("Can't open database file", dbName, "open");
mas01cr@106 438 }
mas01cr@120 439
mas01cr@30 440 get_lock(dbfid, forWrite);
mas01cr@0 441 // Get the database header info
mas01cr@0 442 dbH = new dbTableHeaderT();
mas01cr@0 443 assert(dbH);
mas01cr@0 444
mas01cr@106 445 if(read(dbfid, (char *) dbH, O2_HEADERSIZE) != O2_HEADERSIZE) {
mas01cr@106 446 error("error reading db header", dbName, "read");
mas01cr@106 447 }
mas01cr@0 448
mas01cr@108 449 if(dbH->magic == O2_OLD_MAGIC) {
mas01cr@108 450 // FIXME: if anyone ever complains, write the program to convert
mas01cr@108 451 // from the old audioDB format to the new...
mas01cr@108 452 error("database file has old O2 header", dbName);
mas01cr@108 453 }
mas01cr@0 454
mas01cr@106 455 if(dbH->magic != O2_MAGIC) {
mas01cr@106 456 cerr << "expected: " << O2_MAGIC << ", got: " << dbH->magic << endl;
mas01cr@108 457 error("database file has incorrect header", dbName);
mas01cr@108 458 }
mas01cr@108 459
mas01cr@108 460 if(dbH->version != O2_FORMAT_VERSION) {
mas01cr@108 461 error("database file has incorect version", dbName);
mas01cr@0 462 }
mas01cr@0 463
mas01cr@128 464 // FIXME: when changing file format version, remove this workaround.
mas01cr@128 465 if(dbH->dbSize == 0) {
mas01cr@128 466 dbH->dbSize = O2_DEFAULTDBSIZE;
mas01cr@128 467 }
mas01cr@128 468
mas01cr@120 469 // Make some handy tables with correct types
mas01cr@174 470 #define CHECKED_MMAP(type, var, start, end) \
mas01cr@174 471 { void *tmp = mmap(0, ((end) - (start)), (PROT_READ | (forWrite ? PROT_WRITE : 0)), MAP_SHARED, dbfid, (start)); \
mas01cr@174 472 if(tmp == (void *) -1) { \
mas01cr@174 473 error("mmap error for db table", #var, "mmap"); \
mas01cr@174 474 } \
mas01cr@174 475 var = (type) tmp; \
mas01cr@174 476 }
mas01cr@174 477
mas01cr@176 478 CHECKED_MMAP(char *, db, 0, getpagesize());
mas01cr@176 479
mas01cr@177 480 if(forWrite) {
mas01cr@177 481 CHECKED_MMAP(char *, fileTable, dbH->fileTableOffset, dbH->trackTableOffset);
mas01cr@177 482
mas01cr@177 483 CHECKED_MMAP(unsigned *, trackTable, dbH->trackTableOffset, dbH->dataOffset);
mas01cr@177 484 CHECKED_MMAP(double *, dataBuf, dbH->dataOffset, dbH->timesTableOffset);
mas01cr@177 485 CHECKED_MMAP(double *, timesTable, dbH->timesTableOffset, dbH->l2normTableOffset);
mas01cr@177 486 CHECKED_MMAP(double *, l2normTable, dbH->l2normTableOffset, dbH->dbSize);
mas01cr@177 487 } else {
mas01cr@177 488 if(dbH->length > 0) {
mas01cr@177 489 CHECKED_MMAP(char *, fileTable, dbH->fileTableOffset, dbH->fileTableOffset + ALIGN_PAGE_UP(dbH->numFiles * O2_FILETABLESIZE));
mas01cr@177 490 CHECKED_MMAP(unsigned *, trackTable, dbH->trackTableOffset, dbH->trackTableOffset + ALIGN_PAGE_UP(dbH->numFiles * O2_TRACKTABLESIZE));
mas01cr@177 491 CHECKED_MMAP(double *, dataBuf, dbH->dataOffset, dbH->dataOffset + ALIGN_PAGE_UP(dbH->length));
mas01cr@177 492 CHECKED_MMAP(double *, timesTable, dbH->timesTableOffset, dbH->timesTableOffset + ALIGN_PAGE_UP(dbH->length / dbH->dim));
mas01cr@177 493 CHECKED_MMAP(double *, l2normTable, dbH->l2normTableOffset, dbH->l2normTableOffset + ALIGN_PAGE_UP(dbH->length / dbH->dim));
mas01cr@177 494 }
mas01cr@177 495 }
mas01cr@120 496 }
mas01cr@120 497
mas01cr@169 498 void audioDB::initInputFile (const char *inFile) {
mas01cr@169 499 if (inFile) {
mas01cr@169 500 if ((infid = open(inFile, O_RDONLY)) < 0) {
mas01cr@169 501 error("can't open input file for reading", inFile, "open");
mas01cr@169 502 }
mas01cr@169 503
mas01cr@169 504 if (fstat(infid, &statbuf) < 0) {
mas01cr@169 505 error("fstat error finding size of input", inFile, "fstat");
mas01cr@169 506 }
mas01cr@169 507
mas01cr@169 508 if(dbH->dim == 0 && dbH->length == 0) { // empty database
mas01cr@169 509 // initialize with input dimensionality
mas01cr@169 510 if(read(infid, &dbH->dim, sizeof(unsigned)) != sizeof(unsigned)) {
mas01cr@169 511 error("short read of input file", inFile);
mas01cr@169 512 }
mas01cr@169 513 if(dbH->dim == 0) {
mas01cr@169 514 error("dimensionality of zero in input file", inFile);
mas01cr@169 515 }
mas01cr@169 516 } else {
mas01cr@169 517 unsigned test;
mas01cr@169 518 if(read(infid, &test, sizeof(unsigned)) != sizeof(unsigned)) {
mas01cr@169 519 error("short read of input file", inFile);
mas01cr@169 520 }
mas01cr@169 521 if(dbH->dim == 0) {
mas01cr@169 522 error("dimensionality of zero in input file", inFile);
mas01cr@169 523 }
mas01cr@169 524 if(dbH->dim != test) {
mas01cr@169 525 cerr << "error: expected dimension: " << dbH->dim << ", got : " << test <<endl;
mas01cr@169 526 error("feature dimensions do not match database table dimensions", inFile);
mas01cr@169 527 }
mas01cr@169 528 }
mas01cr@169 529
mas01cr@169 530 if ((indata = (char *) mmap(0, statbuf.st_size, PROT_READ, MAP_SHARED, infid, 0)) == (caddr_t) -1) {
mas01cr@169 531 error("mmap error for input", inFile, "mmap");
mas01cr@169 532 }
mas01cr@169 533 }
mas01cr@169 534 }
mas01cr@169 535
mas01cr@177 536 void audioDB::initTables(const char* dbName, const char* inFile = 0) {
mas01cr@177 537 initDBHeader(dbName);
mas01cr@169 538 initInputFile(inFile);
mas01cr@0 539 }
mas01cr@0 540
mas01cr@177 541 void audioDB::insert(const char* dbName, const char* inFile) {
mas01cr@177 542 forWrite = true;
mas01cr@177 543 initTables(dbName, inFile);
mas01cr@0 544
mas01cr@0 545 if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 546 error("Must use timestamps with timestamped database","use --times");
mas01cr@0 547
mas01cr@0 548 // Check that there is room for at least 1 more file
mas01cr@0 549 if((char*)timesTable<((char*)dataBuf+dbH->length+statbuf.st_size-sizeof(int)))
mas01cr@169 550 error("Insert failed: no more room in database", inFile);
mas01cr@0 551
mas01cr@0 552 if(!key)
mas01cr@0 553 key=inFile;
mas01cr@0 554 // Linear scan of filenames check for pre-existing feature
mas01cr@0 555 unsigned alreadyInserted=0;
mas01cr@0 556 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 557 if(strncmp(fileTable + k*O2_FILETABLESIZE, key, strlen(key))==0){
mas01cr@0 558 alreadyInserted=1;
mas01cr@0 559 break;
mas01cr@0 560 }
mas01cr@0 561
mas01cr@0 562 if(alreadyInserted){
mas01cr@60 563 if(verbosity) {
mas01cr@0 564 cerr << "Warning: key already exists in database, ignoring: " <<inFile << endl;
mas01cr@60 565 }
mas01cr@0 566 return;
mas01cr@0 567 }
mas01cr@0 568
mas01mc@18 569 // Make a track index table of features to file indexes
mas01cr@0 570 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 571 if(!numVectors){
mas01cr@60 572 if(verbosity) {
mas01cr@0 573 cerr << "Warning: ignoring zero-length feature vector file:" << key << endl;
mas01cr@60 574 }
mas01cr@0 575 // CLEAN UP
mas01cr@0 576 munmap(indata,statbuf.st_size);
mas01cr@128 577 munmap(db,dbH->dbSize);
mas01cr@0 578 close(infid);
mas01cr@0 579 return;
mas01cr@0 580 }
mas01cr@0 581
mas01cr@0 582 strncpy(fileTable + dbH->numFiles*O2_FILETABLESIZE, key, strlen(key));
mas01cr@0 583
mas01cr@0 584 unsigned insertoffset = dbH->length;// Store current state
mas01cr@0 585
mas01cr@0 586 // Check times status and insert times from file
mas01cr@0 587 unsigned timesoffset=insertoffset/(dbH->dim*sizeof(double));
mas01cr@0 588 double* timesdata=timesTable+timesoffset;
mas01cr@0 589 assert(timesdata+numVectors<l2normTable);
mas01cr@0 590 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 591
mas01cr@0 592 // Increment file count
mas01cr@0 593 dbH->numFiles++;
mas01cr@0 594
mas01cr@0 595 // Update Header information
mas01cr@0 596 dbH->length+=(statbuf.st_size-sizeof(int));
mas01cr@0 597
mas01mc@18 598 // Update track to file index map
mas01mc@18 599 memcpy (trackTable+dbH->numFiles-1, &numVectors, sizeof(unsigned));
mas01cr@0 600
mas01cr@0 601 // Update the feature database
mas01cr@176 602 memcpy (((char *) dataBuf) + insertoffset, indata+sizeof(int), statbuf.st_size-sizeof(int));
mas01cr@0 603
mas01cr@0 604 // Norm the vectors on input if the database is already L2 normed
mas01cr@0 605 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@176 606 unitNormAndInsertL2((double*)(((char *) dataBuf) + insertoffset), dbH->dim, numVectors, 1); // append
mas01cr@0 607
mas01cr@0 608 // Report status
mas01cr@0 609 status(dbName);
mas01cr@60 610 if(verbosity) {
mas01cr@0 611 cerr << COM_INSERT << " " << dbName << " " << numVectors << " vectors "
mas01cr@0 612 << (statbuf.st_size-sizeof(int)) << " bytes." << endl;
mas01cr@60 613 }
mas01cr@0 614
mas01cr@176 615 // Copy the header back to the database
mas01cr@176 616 memcpy (db, dbH, sizeof(dbTableHeaderT));
mas01cr@176 617
mas01cr@0 618 // CLEAN UP
mas01cr@0 619 munmap(indata,statbuf.st_size);
mas01cr@0 620 close(infid);
mas01cr@0 621 }
mas01cr@0 622
mas01cr@0 623 void audioDB::insertTimeStamps(unsigned numVectors, ifstream* timesFile, double* timesdata){
mas01cr@0 624 unsigned numtimes=0;
mas01cr@0 625 if(usingTimes){
mas01cr@0 626 if(!(dbH->flags & O2_FLAG_TIMES) && !dbH->numFiles)
mas01cr@0 627 dbH->flags=dbH->flags|O2_FLAG_TIMES;
mas01cr@0 628 else if(!(dbH->flags&O2_FLAG_TIMES)){
mas01cr@0 629 cerr << "Warning: timestamp file used with non time-stamped database: ignoring timestamps" << endl;
mas01cr@0 630 usingTimes=0;
mas01cr@0 631 }
mas01cr@0 632
mas01cr@0 633 if(!timesFile->is_open()){
mas01cr@0 634 if(dbH->flags & O2_FLAG_TIMES){
mas01cr@0 635 munmap(indata,statbuf.st_size);
mas01cr@128 636 munmap(db,dbH->dbSize);
mas01cr@0 637 error("problem opening times file on timestamped database",timesFileName);
mas01cr@0 638 }
mas01cr@0 639 else{
mas01cr@0 640 cerr << "Warning: problem opening times file. But non-timestamped database, so ignoring times file." << endl;
mas01cr@0 641 usingTimes=0;
mas01cr@0 642 }
mas01cr@0 643 }
mas01cr@0 644
mas01cr@0 645 // Process time file
mas01cr@0 646 if(usingTimes){
mas01cr@0 647 do{
mas01cr@0 648 *timesFile>>*timesdata++;
mas01cr@0 649 if(timesFile->eof())
mas01cr@0 650 break;
mas01cr@0 651 numtimes++;
mas01cr@0 652 }while(!timesFile->eof() && numtimes<numVectors);
mas01cr@0 653 if(!timesFile->eof()){
mas01cr@0 654 double dummy;
mas01cr@0 655 do{
mas01cr@0 656 *timesFile>>dummy;
mas01cr@0 657 if(timesFile->eof())
mas01cr@0 658 break;
mas01cr@0 659 numtimes++;
mas01cr@0 660 }while(!timesFile->eof());
mas01cr@0 661 }
mas01cr@0 662 if(numtimes<numVectors || numtimes>numVectors+2){
mas01cr@0 663 munmap(indata,statbuf.st_size);
mas01cr@128 664 munmap(db,dbH->dbSize);
mas01cr@0 665 close(infid);
mas01cr@0 666 cerr << "expected " << numVectors << " found " << numtimes << endl;
mas01cr@0 667 error("Times file is incorrect length for features file",inFile);
mas01cr@0 668 }
mas01cr@60 669 if(verbosity>2) {
mas01cr@0 670 cerr << "numtimes: " << numtimes << endl;
mas01cr@60 671 }
mas01cr@0 672 }
mas01cr@0 673 }
mas01cr@0 674 }
mas01cr@0 675
mas01cr@120 676 void audioDB::batchinsert(const char* dbName, const char* inFile) {
mas01cr@0 677
mas01cr@177 678 forWrite = true;
mas01cr@177 679 initDBHeader(dbName);
mas01cr@0 680
mas01cr@0 681 if(!key)
mas01cr@0 682 key=inFile;
mas01cr@0 683 ifstream *filesIn = 0;
mas01cr@0 684 ifstream *keysIn = 0;
mas01cr@0 685 ifstream* thisTimesFile = 0;
mas01cr@0 686
mas01cr@0 687 if(!(filesIn = new ifstream(inFile)))
mas01cr@0 688 error("Could not open batch in file", inFile);
mas01cr@0 689 if(key && key!=inFile)
mas01cr@0 690 if(!(keysIn = new ifstream(key)))
mas01cr@0 691 error("Could not open batch key file",key);
mas01cr@0 692
mas01cr@0 693 if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 694 error("Must use timestamps with timestamped database","use --times");
mas01cr@0 695
mas01cr@0 696 unsigned totalVectors=0;
mas01cr@0 697 char *thisKey = new char[MAXSTR];
mas01cr@0 698 char *thisFile = new char[MAXSTR];
mas01cr@0 699 char *thisTimesFileName = new char[MAXSTR];
mas01cr@120 700
mas01cr@0 701 do{
mas01cr@0 702 filesIn->getline(thisFile,MAXSTR);
mas01cr@0 703 if(key && key!=inFile)
mas01cr@0 704 keysIn->getline(thisKey,MAXSTR);
mas01cr@0 705 else
mas01cr@0 706 thisKey = thisFile;
mas01cr@0 707 if(usingTimes)
mas01cr@0 708 timesFile->getline(thisTimesFileName,MAXSTR);
mas01cr@0 709
mas01cr@0 710 if(filesIn->eof())
mas01cr@0 711 break;
mas01cr@0 712
mas01cr@169 713 initInputFile(thisFile);
mas01cr@0 714
mas01cr@0 715 // Check that there is room for at least 1 more file
mas01cr@0 716 if((char*)timesTable<((char*)dataBuf+(dbH->length+statbuf.st_size-sizeof(int))))
mas01cr@169 717 error("batchinsert failed: no more room in database", thisFile);
mas01cr@0 718
mas01cr@0 719 // Linear scan of filenames check for pre-existing feature
mas01cr@0 720 unsigned alreadyInserted=0;
mas01cr@0 721
mas01cr@0 722 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 723 if(strncmp(fileTable + k*O2_FILETABLESIZE, thisKey, strlen(thisKey))==0){
mas01cr@0 724 alreadyInserted=1;
mas01cr@0 725 break;
mas01cr@0 726 }
mas01cr@0 727
mas01cr@0 728 if(alreadyInserted){
mas01cr@60 729 if(verbosity) {
mas01cr@0 730 cerr << "Warning: key already exists in database:" << thisKey << endl;
mas01cr@60 731 }
mas01cr@0 732 }
mas01cr@0 733 else{
mas01cr@0 734
mas01mc@18 735 // Make a track index table of features to file indexes
mas01cr@0 736 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 737 if(!numVectors){
mas01cr@60 738 if(verbosity) {
mas01cr@0 739 cerr << "Warning: ignoring zero-length feature vector file:" << thisKey << endl;
mas01cr@60 740 }
mas01cr@0 741 }
mas01cr@0 742 else{
mas01cr@0 743 if(usingTimes){
mas01cr@0 744 if(timesFile->eof())
mas01cr@0 745 error("not enough timestamp files in timesList");
mas01cr@0 746 thisTimesFile=new ifstream(thisTimesFileName,ios::in);
mas01cr@0 747 if(!thisTimesFile->is_open())
mas01cr@0 748 error("Cannot open timestamp file",thisTimesFileName);
mas01cr@0 749 unsigned insertoffset=dbH->length;
mas01cr@0 750 unsigned timesoffset=insertoffset/(dbH->dim*sizeof(double));
mas01cr@0 751 double* timesdata=timesTable+timesoffset;
mas01cr@0 752 assert(timesdata+numVectors<l2normTable);
mas01cr@0 753 insertTimeStamps(numVectors,thisTimesFile,timesdata);
mas01cr@0 754 if(thisTimesFile)
mas01cr@0 755 delete thisTimesFile;
mas01cr@0 756 }
mas01cr@0 757
mas01cr@0 758 strncpy(fileTable + dbH->numFiles*O2_FILETABLESIZE, thisKey, strlen(thisKey));
mas01cr@0 759
mas01cr@0 760 unsigned insertoffset = dbH->length;// Store current state
mas01cr@0 761
mas01cr@0 762 // Increment file count
mas01cr@0 763 dbH->numFiles++;
mas01cr@0 764
mas01cr@0 765 // Update Header information
mas01cr@0 766 dbH->length+=(statbuf.st_size-sizeof(int));
mas01cr@0 767
mas01mc@18 768 // Update track to file index map
mas01mc@18 769 memcpy (trackTable+dbH->numFiles-1, &numVectors, sizeof(unsigned));
mas01cr@0 770
mas01cr@0 771 // Update the feature database
mas01cr@176 772 memcpy (((char *) dataBuf) + insertoffset, indata+sizeof(int), statbuf.st_size-sizeof(int));
mas01cr@0 773
mas01cr@0 774 // Norm the vectors on input if the database is already L2 normed
mas01cr@0 775 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@176 776 unitNormAndInsertL2((double*)(((char *) dataBuf) + insertoffset), dbH->dim, numVectors, 1); // append
mas01cr@0 777
mas01cr@0 778 totalVectors+=numVectors;
mas01cr@176 779
mas01cr@176 780 // Copy the header back to the database
mas01cr@176 781 memcpy (db, dbH, sizeof(dbTableHeaderT));
mas01cr@0 782 }
mas01cr@0 783 }
mas01cr@0 784 // CLEAN UP
mas01cr@0 785 munmap(indata,statbuf.st_size);
mas01cr@0 786 close(infid);
mas01cr@0 787 }while(!filesIn->eof());
mas01mc@12 788
mas01cr@60 789 if(verbosity) {
mas01cr@0 790 cerr << COM_BATCHINSERT << " " << dbName << " " << totalVectors << " vectors "
mas01cr@0 791 << totalVectors*dbH->dim*sizeof(double) << " bytes." << endl;
mas01cr@60 792 }
mas01cr@0 793
mas01cr@0 794 // Report status
mas01cr@0 795 status(dbName);
mas01cr@0 796 }
mas01cr@0 797
mas01cr@76 798 // FIXME: this can't propagate the sequence length argument (used for
mas01cr@76 799 // dudCount). See adb__status() definition for the other half of
mas01cr@76 800 // this. -- CSR, 2007-10-01
mas01cr@0 801 void audioDB::ws_status(const char*dbName, char* hostport){
mas01cr@0 802 struct soap soap;
mas01cr@133 803 adb__statusResponse adbStatusResponse;
mas01cr@0 804
mas01cr@0 805 // Query an existing adb database
mas01cr@0 806 soap_init(&soap);
mas01cr@133 807 if(soap_call_adb__status(&soap,hostport,NULL,(char*)dbName,adbStatusResponse)==SOAP_OK) {
mas01cr@133 808 cout << "numFiles = " << adbStatusResponse.result.numFiles << endl;
mas01cr@133 809 cout << "dim = " << adbStatusResponse.result.dim << endl;
mas01cr@133 810 cout << "length = " << adbStatusResponse.result.length << endl;
mas01cr@133 811 cout << "dudCount = " << adbStatusResponse.result.dudCount << endl;
mas01cr@133 812 cout << "nullCount = " << adbStatusResponse.result.nullCount << endl;
mas01cr@133 813 cout << "flags = " << adbStatusResponse.result.flags << endl;
mas01cr@76 814 } else {
mas01cr@0 815 soap_print_fault(&soap,stderr);
mas01cr@76 816 }
mas01cr@0 817
mas01cr@0 818 soap_destroy(&soap);
mas01cr@0 819 soap_end(&soap);
mas01cr@0 820 soap_done(&soap);
mas01cr@0 821 }
mas01cr@0 822
mas01mc@18 823 void audioDB::ws_query(const char*dbName, const char *trackKey, const char* hostport){
mas01cr@0 824 struct soap soap;
mas01cr@133 825 adb__queryResponse adbQueryResponse;
mas01cr@0 826
mas01cr@0 827 soap_init(&soap);
mas01cr@0 828 if(soap_call_adb__query(&soap,hostport,NULL,
mas01mc@18 829 (char*)dbName,(char*)trackKey,(char*)trackFileName,(char*)timesFileName,
mas01cr@133 830 queryType, queryPoint, pointNN, trackNN, sequenceLength, adbQueryResponse)==SOAP_OK){
mas01cr@133 831 //std::cerr << "result list length:" << adbQueryResponse.result.__sizeRlist << std::endl;
mas01cr@133 832 for(int i=0; i<adbQueryResponse.result.__sizeRlist; i++)
mas01cr@133 833 std::cout << adbQueryResponse.result.Rlist[i] << " " << adbQueryResponse.result.Dist[i]
mas01cr@133 834 << " " << adbQueryResponse.result.Qpos[i] << " " << adbQueryResponse.result.Spos[i] << std::endl;
mas01cr@0 835 }
mas01cr@0 836 else
mas01cr@0 837 soap_print_fault(&soap,stderr);
mas01cr@0 838
mas01cr@0 839 soap_destroy(&soap);
mas01cr@0 840 soap_end(&soap);
mas01cr@0 841 soap_done(&soap);
mas01cr@0 842
mas01cr@0 843 }
mas01cr@0 844
mas01cr@0 845
mas01cr@133 846 void audioDB::status(const char* dbName, adb__statusResponse *adbStatusResponse){
mas01cr@0 847 if(!dbH)
mas01cr@177 848 initTables(dbName, 0);
mas01cr@0 849
mas01cr@0 850 unsigned dudCount=0;
mas01cr@0 851 unsigned nullCount=0;
mas01cr@0 852 for(unsigned k=0; k<dbH->numFiles; k++){
mas01mc@18 853 if(trackTable[k]<sequenceLength){
mas01cr@0 854 dudCount++;
mas01mc@18 855 if(!trackTable[k])
mas01cr@76 856 nullCount++;
mas01cr@0 857 }
mas01cr@0 858 }
mas01cr@76 859
mas01cr@133 860 if(adbStatusResponse == 0) {
mas01cr@76 861
mas01cr@76 862 // Update Header information
mas01cr@76 863 cout << "num files:" << dbH->numFiles << endl;
mas01cr@76 864 cout << "data dim:" << dbH->dim <<endl;
mas01cr@76 865 if(dbH->dim>0){
mas01cr@76 866 cout << "total vectors:" << dbH->length/(sizeof(double)*dbH->dim)<<endl;
mas01cr@108 867 cout << "vectors available:" << (dbH->timesTableOffset-(dbH->dataOffset+dbH->length))/(sizeof(double)*dbH->dim) << endl;
mas01cr@76 868 }
mas01cr@108 869 cout << "total bytes:" << dbH->length << " (" << (100.0*dbH->length)/(dbH->timesTableOffset-dbH->dataOffset) << "%)" << endl;
mas01cr@108 870 cout << "bytes available:" << dbH->timesTableOffset-(dbH->dataOffset+dbH->length) << " (" <<
mas01cr@108 871 (100.0*(dbH->timesTableOffset-(dbH->dataOffset+dbH->length)))/(dbH->timesTableOffset-dbH->dataOffset) << "%)" << endl;
mas01cr@76 872 cout << "flags:" << dbH->flags << endl;
mas01cr@76 873
mas01cr@76 874 cout << "null count: " << nullCount << " small sequence count " << dudCount-nullCount << endl;
mas01cr@76 875 } else {
mas01cr@133 876 adbStatusResponse->result.numFiles = dbH->numFiles;
mas01cr@133 877 adbStatusResponse->result.dim = dbH->dim;
mas01cr@133 878 adbStatusResponse->result.length = dbH->length;
mas01cr@133 879 adbStatusResponse->result.dudCount = dudCount;
mas01cr@133 880 adbStatusResponse->result.nullCount = nullCount;
mas01cr@133 881 adbStatusResponse->result.flags = dbH->flags;
mas01cr@76 882 }
mas01cr@0 883 }
mas01cr@0 884
mas01cr@0 885 void audioDB::dump(const char* dbName){
mas01cr@131 886 if(!dbH) {
mas01cr@177 887 initTables(dbName, 0);
mas01mc@17 888 }
mas01cr@0 889
mas01cr@131 890 if((mkdir(output, S_IRWXU|S_IRWXG|S_IRWXO)) < 0) {
mas01cr@131 891 error("error making output directory", output, "mkdir");
mas01cr@131 892 }
mas01cr@131 893
mas01cr@131 894 char *cwd = new char[PATH_MAX];
mas01cr@131 895
mas01cr@131 896 if ((getcwd(cwd, PATH_MAX)) == 0) {
mas01cr@131 897 error("error getting working directory", "", "getcwd");
mas01cr@131 898 }
mas01cr@131 899
mas01cr@131 900 if((chdir(output)) < 0) {
mas01cr@131 901 error("error changing working directory", output, "chdir");
mas01cr@131 902 }
mas01cr@131 903
mas01cr@131 904 int fLfd, tLfd = 0, kLfd;
mas01cr@131 905 FILE *fLFile, *tLFile = 0, *kLFile;
mas01cr@131 906
mas01cr@131 907 if ((fLfd = open("featureList.txt", O_CREAT|O_RDWR|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0) {
mas01cr@131 908 error("error creating featureList file", "featureList.txt", "open");
mas01cr@131 909 }
mas01cr@131 910 int times = dbH->flags & O2_FLAG_TIMES;
mas01cr@131 911 if (times) {
mas01cr@131 912 if ((tLfd = open("timesList.txt", O_CREAT|O_RDWR|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0) {
mas01cr@131 913 error("error creating timesList file", "timesList.txt", "open");
mas01cr@131 914 }
mas01cr@131 915 }
mas01cr@131 916 if ((kLfd = open("keyList.txt", O_CREAT|O_RDWR|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0) {
mas01cr@131 917 error("error creating keyList file", "keyList.txt", "open");
mas01cr@131 918 }
mas01cr@131 919
mas01cr@131 920 /* can these fail? I sincerely hope not. */
mas01cr@131 921 fLFile = fdopen(fLfd, "w");
mas01cr@131 922 if (times) {
mas01cr@131 923 tLFile = fdopen(tLfd, "w");
mas01cr@131 924 }
mas01cr@131 925 kLFile = fdopen(kLfd, "w");
mas01cr@131 926
mas01cr@131 927 char *fName = new char[256];
mas01cr@131 928 int ffd;
mas01cr@131 929 FILE *tFile;
mas01cr@131 930 unsigned pos = 0;
mas01cr@131 931 for(unsigned k = 0; k < dbH->numFiles; k++) {
mas01cr@131 932 fprintf(kLFile, "%s\n", fileTable + k*O2_FILETABLESIZE);
mas01cr@131 933 snprintf(fName, 256, "%05d.features", k);
mas01cr@131 934 if ((ffd = open(fName, O_CREAT|O_RDWR|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0) {
mas01cr@131 935 error("error creating feature file", fName, "open");
mas01cr@131 936 }
mas01cr@131 937 if ((write(ffd, &dbH->dim, sizeof(uint32_t))) < 0) {
mas01cr@131 938 error("error writing dimensions", fName, "write");
mas01cr@131 939 }
mas01cr@131 940
mas01cr@131 941 if ((write(ffd, dataBuf + pos * dbH->dim, trackTable[k] * dbH->dim * sizeof(double))) < 0) {
mas01cr@131 942 error("error writing data", fName, "write");
mas01cr@131 943 }
mas01cr@131 944 fprintf(fLFile, "%s\n", fName);
mas01cr@131 945 close(ffd);
mas01cr@131 946
mas01cr@131 947 if(times) {
mas01cr@131 948 snprintf(fName, 256, "%05d.times", k);
mas01cr@131 949 tFile = fopen(fName, "w");
mas01cr@131 950 for(unsigned i = 0; i < trackTable[k]; i++) {
mas01cr@131 951 // KLUDGE: specifying 16 digits of precision after the decimal
mas01cr@131 952 // point is (but check this!) sufficient to uniquely identify
mas01cr@131 953 // doubles; however, that will cause ugliness, as that's
mas01cr@131 954 // vastly too many for most values of interest. Moving to %a
mas01cr@131 955 // here and scanf() in the timesFile reading might fix this.
mas01cr@131 956 // -- CSR, 2007-10-19
mas01cr@131 957 fprintf(tFile, "%.16e\n", *(timesTable + pos + i));
mas01cr@131 958 }
mas01cr@131 959 fprintf(tLFile, "%s\n", fName);
mas01cr@131 960 }
mas01cr@131 961
mas01cr@131 962 pos += trackTable[k];
mas01cr@131 963 cout << fileTable+k*O2_FILETABLESIZE << " " << trackTable[k] << endl;
mas01cr@131 964 }
mas01cr@131 965
mas01cr@131 966 FILE *scriptFile;
mas01cr@131 967 scriptFile = fopen("restore.sh", "w");
mas01cr@131 968 fprintf(scriptFile, "\
mas01cr@131 969 #! /bin/sh\n\
mas01cr@131 970 #\n\
mas01cr@131 971 # usage: AUDIODB=/path/to/audioDB sh ./restore.sh <newdb>\n\
mas01cr@131 972 \n\
mas01cr@131 973 if [ -z \"${AUDIODB}\" ]; then echo set AUDIODB variable; exit 1; fi\n\
mas01cr@131 974 if [ -z \"$1\" ]; then echo usage: $0 newdb; exit 1; fi\n\n\
mas01cr@131 975 \"${AUDIODB}\" -d \"$1\" -N --size=%d\n", dbH->dbSize / 1000000);
mas01cr@131 976 if(dbH->flags & O2_FLAG_L2NORM) {
mas01cr@131 977 fprintf(scriptFile, "\"${AUDIODB}\" -d \"$1\" -L\n");
mas01cr@131 978 }
mas01cr@131 979 fprintf(scriptFile, "\"${AUDIODB}\" -d \"$1\" -B -F featureList.txt -K keyList.txt");
mas01cr@131 980 if(times) {
mas01cr@131 981 fprintf(scriptFile, " -T timesList.txt");
mas01cr@131 982 }
mas01cr@131 983 fprintf(scriptFile, "\n");
mas01cr@131 984 fclose(scriptFile);
mas01cr@131 985
mas01cr@131 986 if((chdir(cwd)) < 0) {
mas01cr@131 987 error("error changing working directory", cwd, "chdir");
mas01cr@131 988 }
mas01cr@131 989
mas01cr@131 990 fclose(fLFile);
mas01cr@131 991 if(times) {
mas01cr@131 992 fclose(tLFile);
mas01cr@131 993 }
mas01cr@131 994 fclose(kLFile);
mas01cr@131 995 delete[] fName;
mas01cr@131 996
mas01cr@0 997 status(dbName);
mas01cr@0 998 }
mas01cr@0 999
mas01cr@177 1000 void audioDB::l2norm(const char* dbName) {
mas01cr@177 1001 forWrite = true;
mas01cr@177 1002 initTables(dbName, 0);
mas01cr@0 1003 if(dbH->length>0){
mas01cr@0 1004 unsigned numVectors = dbH->length/(sizeof(double)*dbH->dim);
mas01cr@0 1005 unitNormAndInsertL2(dataBuf, dbH->dim, numVectors, 0); // No append
mas01cr@0 1006 }
mas01cr@0 1007 // Update database flags
mas01cr@0 1008 dbH->flags = dbH->flags|O2_FLAG_L2NORM;
mas01cr@0 1009 memcpy (db, dbH, O2_HEADERSIZE);
mas01cr@0 1010 }
mas01cr@0 1011
mas01cr@0 1012
mas01cr@0 1013
mas01cr@133 1014 void audioDB::query(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
mas01cr@0 1015 switch(queryType){
mas01cr@105 1016 case O2_POINT_QUERY:
mas01cr@133 1017 pointQuery(dbName, inFile, adbQueryResponse);
mas01cr@0 1018 break;
mas01cr@105 1019 case O2_SEQUENCE_QUERY:
mas01mc@17 1020 if(radius==0)
mas01cr@133 1021 trackSequenceQueryNN(dbName, inFile, adbQueryResponse);
mas01mc@17 1022 else
mas01cr@133 1023 trackSequenceQueryRad(dbName, inFile, adbQueryResponse);
mas01cr@0 1024 break;
mas01cr@105 1025 case O2_TRACK_QUERY:
mas01cr@133 1026 trackPointQuery(dbName, inFile, adbQueryResponse);
mas01cr@0 1027 break;
mas01cr@0 1028 default:
mas01cr@0 1029 error("unrecognized queryType in query()");
mas01cr@0 1030
mas01cr@0 1031 }
mas01cr@0 1032 }
mas01cr@0 1033
mas01cr@0 1034 //return ordinal position of key in keyTable
mas01cr@0 1035 unsigned audioDB::getKeyPos(char* key){
mas01cr@0 1036 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 1037 if(strncmp(fileTable + k*O2_FILETABLESIZE, key, strlen(key))==0)
mas01cr@0 1038 return k;
mas01cr@0 1039 error("Key not found",key);
mas01cr@0 1040 return O2_ERR_KEYNOTFOUND;
mas01cr@0 1041 }
mas01cr@0 1042
mas01cr@0 1043 // Basic point query engine
mas01cr@177 1044 void audioDB::pointQuery(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse) {
mas01cr@0 1045
mas01cr@177 1046 initTables(dbName, inFile);
mas01cr@0 1047
mas01cr@0 1048 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 1049 // we use stdout in this stub version
mas01cr@0 1050 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 1051
mas01cr@0 1052 double* query = (double*)(indata+sizeof(int));
mas01cr@0 1053 double* data = dataBuf;
mas01cr@0 1054 double* queryCopy = 0;
mas01cr@0 1055
mas01cr@0 1056 if( dbH->flags & O2_FLAG_L2NORM ){
mas01cr@0 1057 // Make a copy of the query
mas01cr@0 1058 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 1059 qNorm = new double[numVectors];
mas01cr@0 1060 assert(queryCopy&&qNorm);
mas01cr@0 1061 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 1062 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 1063 query = queryCopy;
mas01cr@0 1064 }
mas01cr@0 1065
mas01cr@0 1066 // Make temporary dynamic memory for results
mas01cr@0 1067 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01cr@0 1068 double distances[pointNN];
mas01cr@0 1069 unsigned qIndexes[pointNN];
mas01cr@0 1070 unsigned sIndexes[pointNN];
mas01cr@0 1071 for(unsigned k=0; k<pointNN; k++){
mas01cr@62 1072 distances[k]=-DBL_MAX;
mas01cr@0 1073 qIndexes[k]=~0;
mas01cr@0 1074 sIndexes[k]=~0;
mas01cr@0 1075 }
mas01cr@0 1076
mas01cr@0 1077 unsigned j=numVectors;
mas01cr@0 1078 unsigned k,l,n;
mas01cr@0 1079 double thisDist;
mas01cr@0 1080
mas01cr@0 1081 unsigned totalVecs=dbH->length/(dbH->dim*sizeof(double));
mas01cr@0 1082 double meanQdur = 0;
mas01cr@0 1083 double* timesdata = 0;
mas01cr@0 1084 double* dbdurs = 0;
mas01cr@0 1085
mas01cr@0 1086 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1087 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 1088 usingTimes=0;
mas01cr@0 1089 }
mas01cr@0 1090
mas01cr@0 1091 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 1092 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 1093
mas01cr@0 1094 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1095 timesdata = new double[numVectors];
mas01cr@0 1096 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1097 // Calculate durations of points
mas01cr@0 1098 for(k=0; k<numVectors-1; k++){
mas01cr@0 1099 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1100 meanQdur+=timesdata[k];
mas01cr@0 1101 }
mas01cr@0 1102 meanQdur/=k;
mas01cr@0 1103 // Individual exhaustive timepoint durations
mas01cr@0 1104 dbdurs = new double[totalVecs];
mas01cr@0 1105 for(k=0; k<totalVecs-1; k++)
mas01cr@0 1106 dbdurs[k]=timesTable[k+1]-timesTable[k];
mas01cr@0 1107 j--; // decrement vector counter by one
mas01cr@0 1108 }
mas01cr@0 1109
mas01cr@0 1110 if(usingQueryPoint)
mas01cr@0 1111 if(queryPoint>numVectors-1)
mas01cr@0 1112 error("queryPoint > numVectors in query");
mas01cr@0 1113 else{
mas01cr@60 1114 if(verbosity>1) {
mas01cr@0 1115 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@60 1116 }
mas01cr@0 1117 query=query+queryPoint*dbH->dim;
mas01cr@0 1118 numVectors=queryPoint+1;
mas01cr@0 1119 j=1;
mas01cr@0 1120 }
mas01cr@0 1121
mas01cr@0 1122 gettimeofday(&tv1, NULL);
mas01cr@0 1123 while(j--){ // query
mas01cr@0 1124 data=dataBuf;
mas01cr@0 1125 k=totalVecs; // number of database vectors
mas01cr@0 1126 while(k--){ // database
mas01cr@0 1127 thisDist=0;
mas01cr@0 1128 l=dbH->dim;
mas01cr@0 1129 double* q=query;
mas01cr@0 1130 while(l--)
mas01cr@0 1131 thisDist+=*q++**data++;
mas01cr@0 1132 if(!usingTimes ||
mas01cr@0 1133 (usingTimes
mas01cr@0 1134 && fabs(dbdurs[totalVecs-k-1]-timesdata[numVectors-j-1])<timesdata[numVectors-j-1]*timesTol)){
mas01cr@0 1135 n=pointNN;
mas01cr@0 1136 while(n--){
mas01cr@0 1137 if(thisDist>=distances[n]){
mas01cr@0 1138 if((n==0 || thisDist<=distances[n-1])){
mas01cr@0 1139 // Copy all values above up the queue
mas01cr@0 1140 for( l=pointNN-1 ; l >= n+1 ; l--){
mas01cr@0 1141 distances[l]=distances[l-1];
mas01cr@0 1142 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1143 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1144 }
mas01cr@0 1145 distances[n]=thisDist;
mas01cr@0 1146 qIndexes[n]=numVectors-j-1;
mas01cr@0 1147 sIndexes[n]=dbH->length/(sizeof(double)*dbH->dim)-k-1;
mas01cr@0 1148 break;
mas01cr@0 1149 }
mas01cr@0 1150 }
mas01cr@0 1151 else
mas01cr@0 1152 break;
mas01cr@0 1153 }
mas01cr@0 1154 }
mas01cr@0 1155 }
mas01cr@0 1156 // Move query pointer to next query point
mas01cr@0 1157 query+=dbH->dim;
mas01cr@0 1158 }
mas01cr@0 1159
mas01cr@0 1160 gettimeofday(&tv2, NULL);
mas01cr@60 1161 if(verbosity>1) {
mas01cr@0 1162 cerr << endl << " elapsed time:" << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01cr@60 1163 }
mas01cr@0 1164
mas01cr@133 1165 if(adbQueryResponse==0){
mas01cr@0 1166 // Output answer
mas01cr@0 1167 // Loop over nearest neighbours
mas01cr@0 1168 for(k=0; k < pointNN; k++){
mas01cr@0 1169 // Scan for key
mas01mc@18 1170 unsigned cumTrack=0;
mas01cr@0 1171 for(l=0 ; l<dbH->numFiles; l++){
mas01mc@18 1172 cumTrack+=trackTable[l];
mas01mc@18 1173 if(sIndexes[k]<cumTrack){
mas01cr@0 1174 cout << fileTable+l*O2_FILETABLESIZE << " " << distances[k] << " " << qIndexes[k] << " "
mas01mc@18 1175 << sIndexes[k]+trackTable[l]-cumTrack << endl;
mas01cr@0 1176 break;
mas01cr@0 1177 }
mas01cr@0 1178 }
mas01cr@0 1179 }
mas01cr@0 1180 }
mas01cr@0 1181 else{ // Process Web Services Query
mas01cr@89 1182 int listLen;
mas01cr@89 1183 for(k = 0; k < pointNN; k++) {
mas01cr@89 1184 if(distances[k] == -DBL_MAX)
mas01cr@89 1185 break;
mas01cr@89 1186 }
mas01cr@89 1187 listLen = k;
mas01cr@89 1188
mas01cr@133 1189 adbQueryResponse->result.__sizeRlist=listLen;
mas01cr@133 1190 adbQueryResponse->result.__sizeDist=listLen;
mas01cr@133 1191 adbQueryResponse->result.__sizeQpos=listLen;
mas01cr@133 1192 adbQueryResponse->result.__sizeSpos=listLen;
mas01cr@133 1193 adbQueryResponse->result.Rlist= new char*[listLen];
mas01cr@133 1194 adbQueryResponse->result.Dist = new double[listLen];
mas01cr@133 1195 adbQueryResponse->result.Qpos = new unsigned int[listLen];
mas01cr@133 1196 adbQueryResponse->result.Spos = new unsigned int[listLen];
mas01cr@133 1197 for(k=0; k<(unsigned)adbQueryResponse->result.__sizeRlist; k++){
mas01cr@133 1198 adbQueryResponse->result.Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@133 1199 adbQueryResponse->result.Dist[k]=distances[k];
mas01cr@133 1200 adbQueryResponse->result.Qpos[k]=qIndexes[k];
mas01mc@18 1201 unsigned cumTrack=0;
mas01cr@0 1202 for(l=0 ; l<dbH->numFiles; l++){
mas01mc@18 1203 cumTrack+=trackTable[l];
mas01mc@18 1204 if(sIndexes[k]<cumTrack){
mas01cr@133 1205 sprintf(adbQueryResponse->result.Rlist[k], "%s", fileTable+l*O2_FILETABLESIZE);
mas01cr@0 1206 break;
mas01cr@0 1207 }
mas01cr@0 1208 }
mas01cr@133 1209 adbQueryResponse->result.Spos[k]=sIndexes[k]+trackTable[l]-cumTrack;
mas01cr@0 1210 }
mas01cr@0 1211 }
mas01cr@0 1212
mas01cr@0 1213 // Clean up
mas01cr@0 1214 if(queryCopy)
mas01cr@0 1215 delete queryCopy;
mas01cr@0 1216 if(qNorm)
mas01cr@0 1217 delete qNorm;
mas01cr@0 1218 if(timesdata)
mas01cr@0 1219 delete timesdata;
mas01cr@0 1220 if(dbdurs)
mas01cr@0 1221 delete dbdurs;
mas01cr@0 1222 }
mas01cr@0 1223
mas01mc@18 1224 // trackPointQuery
mas01mc@18 1225 // return the trackNN closest tracks to the query track
mas01mc@18 1226 // uses average of pointNN points per track
mas01cr@177 1227 void audioDB::trackPointQuery(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse) {
mas01cr@177 1228 initTables(dbName, inFile);
mas01cr@0 1229
mas01cr@0 1230 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 1231 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 1232 double* query = (double*)(indata+sizeof(int));
mas01cr@0 1233 double* data = dataBuf;
mas01cr@0 1234 double* queryCopy = 0;
mas01cr@0 1235
mas01cr@0 1236 if( dbH->flags & O2_FLAG_L2NORM ){
mas01cr@0 1237 // Make a copy of the query
mas01cr@0 1238 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 1239 qNorm = new double[numVectors];
mas01cr@0 1240 assert(queryCopy&&qNorm);
mas01cr@0 1241 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 1242 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 1243 query = queryCopy;
mas01cr@0 1244 }
mas01cr@0 1245
mas01cr@0 1246 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01mc@18 1247 assert(trackNN>0 && trackNN<=O2_MAXNN);
mas01cr@0 1248
mas01cr@0 1249 // Make temporary dynamic memory for results
mas01mc@18 1250 double trackDistances[trackNN];
mas01mc@18 1251 unsigned trackIDs[trackNN];
mas01mc@18 1252 unsigned trackQIndexes[trackNN];
mas01mc@18 1253 unsigned trackSIndexes[trackNN];
mas01cr@0 1254
mas01cr@0 1255 double distances[pointNN];
mas01cr@0 1256 unsigned qIndexes[pointNN];
mas01cr@0 1257 unsigned sIndexes[pointNN];
mas01cr@0 1258
mas01cr@0 1259 unsigned j=numVectors; // number of query points
mas01mc@18 1260 unsigned k,l,n, track, trackOffset=0, processedTracks=0;
mas01cr@0 1261 double thisDist;
mas01cr@0 1262
mas01cr@0 1263 for(k=0; k<pointNN; k++){
mas01cr@66 1264 distances[k]=-DBL_MAX;
mas01cr@0 1265 qIndexes[k]=~0;
mas01cr@0 1266 sIndexes[k]=~0;
mas01cr@0 1267 }
mas01cr@0 1268
mas01mc@18 1269 for(k=0; k<trackNN; k++){
mas01cr@66 1270 trackDistances[k]=-DBL_MAX;
mas01mc@18 1271 trackQIndexes[k]=~0;
mas01mc@18 1272 trackSIndexes[k]=~0;
mas01mc@18 1273 trackIDs[k]=~0;
mas01cr@0 1274 }
mas01cr@0 1275
mas01cr@0 1276 double meanQdur = 0;
mas01cr@0 1277 double* timesdata = 0;
mas01cr@0 1278 double* meanDBdur = 0;
mas01cr@0 1279
mas01cr@0 1280 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1281 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 1282 usingTimes=0;
mas01cr@0 1283 }
mas01cr@0 1284
mas01cr@0 1285 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 1286 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 1287
mas01cr@0 1288 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1289 timesdata = new double[numVectors];
mas01cr@0 1290 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1291 // Calculate durations of points
mas01cr@0 1292 for(k=0; k<numVectors-1; k++){
mas01cr@0 1293 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1294 meanQdur+=timesdata[k];
mas01cr@0 1295 }
mas01cr@0 1296 meanQdur/=k;
mas01cr@0 1297 meanDBdur = new double[dbH->numFiles];
mas01cr@0 1298 for(k=0; k<dbH->numFiles; k++){
mas01cr@0 1299 meanDBdur[k]=0.0;
mas01mc@18 1300 for(j=0; j<trackTable[k]-1 ; j++)
mas01cr@0 1301 meanDBdur[k]+=timesTable[j+1]-timesTable[j];
mas01cr@0 1302 meanDBdur[k]/=j;
mas01cr@0 1303 }
mas01cr@0 1304 }
mas01cr@0 1305
mas01cr@0 1306 if(usingQueryPoint)
mas01cr@0 1307 if(queryPoint>numVectors-1)
mas01cr@0 1308 error("queryPoint > numVectors in query");
mas01cr@0 1309 else{
mas01cr@60 1310 if(verbosity>1) {
mas01cr@0 1311 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@60 1312 }
mas01cr@0 1313 query=query+queryPoint*dbH->dim;
mas01cr@0 1314 numVectors=queryPoint+1;
mas01cr@0 1315 }
mas01cr@0 1316
mas01mc@18 1317 // build track offset table
mas01mc@18 1318 unsigned *trackOffsetTable = new unsigned[dbH->numFiles];
mas01mc@18 1319 unsigned cumTrack=0;
mas01mc@18 1320 unsigned trackIndexOffset;
mas01cr@0 1321 for(k=0; k<dbH->numFiles;k++){
mas01mc@18 1322 trackOffsetTable[k]=cumTrack;
mas01mc@18 1323 cumTrack+=trackTable[k]*dbH->dim;
mas01cr@0 1324 }
mas01cr@0 1325
mas01cr@0 1326 char nextKey[MAXSTR];
mas01cr@0 1327
mas01cr@0 1328 gettimeofday(&tv1, NULL);
mas01cr@0 1329
mas01mc@18 1330 for(processedTracks=0, track=0 ; processedTracks < dbH->numFiles ; track++, processedTracks++){
mas01mc@18 1331 if(trackFile){
mas01mc@18 1332 if(!trackFile->eof()){
mas01mc@18 1333 trackFile->getline(nextKey,MAXSTR);
mas01mc@18 1334 track=getKeyPos(nextKey);
mas01cr@0 1335 }
mas01cr@0 1336 else
mas01cr@0 1337 break;
mas01cr@0 1338 }
mas01mc@18 1339 trackOffset=trackOffsetTable[track]; // numDoubles offset
mas01mc@18 1340 trackIndexOffset=trackOffset/dbH->dim; // numVectors offset
mas01cr@60 1341 if(verbosity>7) {
mas01mc@18 1342 cerr << track << "." << trackOffset/(dbH->dim) << "." << trackTable[track] << " | ";cerr.flush();
mas01cr@60 1343 }
mas01cr@0 1344
mas01cr@0 1345 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@0 1346 usingQueryPoint?query=queryCopy+queryPoint*dbH->dim:query=queryCopy;
mas01cr@0 1347 else
mas01cr@0 1348 usingQueryPoint?query=(double*)(indata+sizeof(int))+queryPoint*dbH->dim:query=(double*)(indata+sizeof(int));
mas01cr@0 1349 if(usingQueryPoint)
mas01cr@0 1350 j=1;
mas01cr@0 1351 else
mas01cr@0 1352 j=numVectors;
mas01cr@0 1353 while(j--){
mas01mc@18 1354 k=trackTable[track]; // number of vectors in track
mas01mc@18 1355 data=dataBuf+trackOffset; // data for track
mas01cr@0 1356 while(k--){
mas01cr@0 1357 thisDist=0;
mas01cr@0 1358 l=dbH->dim;
mas01cr@0 1359 double* q=query;
mas01cr@0 1360 while(l--)
mas01cr@0 1361 thisDist+=*q++**data++;
mas01cr@0 1362 if(!usingTimes ||
mas01cr@0 1363 (usingTimes
mas01mc@18 1364 && fabs(meanDBdur[track]-meanQdur)<meanQdur*timesTol)){
mas01cr@0 1365 n=pointNN;
mas01cr@0 1366 while(n--){
mas01cr@0 1367 if(thisDist>=distances[n]){
mas01cr@0 1368 if((n==0 || thisDist<=distances[n-1])){
mas01cr@0 1369 // Copy all values above up the queue
mas01cr@0 1370 for( l=pointNN-1 ; l > n ; l--){
mas01cr@0 1371 distances[l]=distances[l-1];
mas01cr@0 1372 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1373 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1374 }
mas01cr@0 1375 distances[n]=thisDist;
mas01cr@0 1376 qIndexes[n]=numVectors-j-1;
mas01mc@18 1377 sIndexes[n]=trackTable[track]-k-1;
mas01cr@0 1378 break;
mas01cr@0 1379 }
mas01cr@0 1380 }
mas01cr@0 1381 else
mas01cr@0 1382 break;
mas01cr@0 1383 }
mas01cr@0 1384 }
mas01mc@18 1385 } // track
mas01cr@0 1386 // Move query pointer to next query point
mas01cr@0 1387 query+=dbH->dim;
mas01cr@0 1388 } // query
mas01mc@18 1389 // Take the average of this track's distance
mas01mc@18 1390 // Test the track distances
mas01cr@0 1391 thisDist=0;
mas01cr@66 1392 for (n = 0; n < pointNN; n++) {
mas01cr@66 1393 if (distances[n] == -DBL_MAX) break;
mas01cr@66 1394 thisDist += distances[n];
mas01cr@66 1395 }
mas01cr@66 1396 thisDist /= n;
mas01cr@66 1397
mas01mc@18 1398 n=trackNN;
mas01cr@0 1399 while(n--){
mas01mc@18 1400 if(thisDist>=trackDistances[n]){
mas01mc@18 1401 if((n==0 || thisDist<=trackDistances[n-1])){
mas01cr@0 1402 // Copy all values above up the queue
mas01cr@74 1403 for( l=trackNN-1 ; l > n ; l--){
mas01mc@18 1404 trackDistances[l]=trackDistances[l-1];
mas01mc@18 1405 trackQIndexes[l]=trackQIndexes[l-1];
mas01mc@18 1406 trackSIndexes[l]=trackSIndexes[l-1];
mas01mc@18 1407 trackIDs[l]=trackIDs[l-1];
mas01cr@0 1408 }
mas01mc@18 1409 trackDistances[n]=thisDist;
mas01mc@18 1410 trackQIndexes[n]=qIndexes[0];
mas01mc@18 1411 trackSIndexes[n]=sIndexes[0];
mas01mc@18 1412 trackIDs[n]=track;
mas01cr@0 1413 break;
mas01cr@0 1414 }
mas01cr@0 1415 }
mas01cr@0 1416 else
mas01cr@0 1417 break;
mas01cr@0 1418 }
mas01cr@0 1419 for(unsigned k=0; k<pointNN; k++){
mas01cr@66 1420 distances[k]=-DBL_MAX;
mas01cr@0 1421 qIndexes[k]=~0;
mas01cr@0 1422 sIndexes[k]=~0;
mas01cr@0 1423 }
mas01mc@18 1424 } // tracks
mas01cr@0 1425 gettimeofday(&tv2, NULL);
mas01cr@0 1426
mas01cr@60 1427 if(verbosity>1) {
mas01mc@18 1428 cerr << endl << "processed tracks :" << processedTracks
mas01cr@0 1429 << " elapsed time:" << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01cr@60 1430 }
mas01cr@0 1431
mas01cr@133 1432 if(adbQueryResponse==0){
mas01cr@60 1433 if(verbosity>1) {
mas01cr@0 1434 cerr<<endl;
mas01cr@60 1435 }
mas01cr@0 1436 // Output answer
mas01cr@0 1437 // Loop over nearest neighbours
mas01mc@18 1438 for(k=0; k < min(trackNN,processedTracks); k++)
mas01mc@18 1439 cout << fileTable+trackIDs[k]*O2_FILETABLESIZE
mas01mc@18 1440 << " " << trackDistances[k] << " " << trackQIndexes[k] << " " << trackSIndexes[k] << endl;
mas01cr@0 1441 }
mas01cr@0 1442 else{ // Process Web Services Query
mas01mc@18 1443 int listLen = min(trackNN, processedTracks);
mas01cr@133 1444 adbQueryResponse->result.__sizeRlist=listLen;
mas01cr@133 1445 adbQueryResponse->result.__sizeDist=listLen;
mas01cr@133 1446 adbQueryResponse->result.__sizeQpos=listLen;
mas01cr@133 1447 adbQueryResponse->result.__sizeSpos=listLen;
mas01cr@133 1448 adbQueryResponse->result.Rlist= new char*[listLen];
mas01cr@133 1449 adbQueryResponse->result.Dist = new double[listLen];
mas01cr@133 1450 adbQueryResponse->result.Qpos = new unsigned int[listLen];
mas01cr@133 1451 adbQueryResponse->result.Spos = new unsigned int[listLen];
mas01cr@133 1452 for(k=0; k<(unsigned)adbQueryResponse->result.__sizeRlist; k++){
mas01cr@133 1453 adbQueryResponse->result.Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@133 1454 adbQueryResponse->result.Dist[k]=trackDistances[k];
mas01cr@133 1455 adbQueryResponse->result.Qpos[k]=trackQIndexes[k];
mas01cr@133 1456 adbQueryResponse->result.Spos[k]=trackSIndexes[k];
mas01cr@133 1457 sprintf(adbQueryResponse->result.Rlist[k], "%s", fileTable+trackIDs[k]*O2_FILETABLESIZE);
mas01cr@0 1458 }
mas01cr@0 1459 }
mas01cr@0 1460
mas01cr@0 1461
mas01cr@0 1462 // Clean up
mas01mc@18 1463 if(trackOffsetTable)
mas01mc@18 1464 delete trackOffsetTable;
mas01cr@0 1465 if(queryCopy)
mas01cr@0 1466 delete queryCopy;
mas01cr@0 1467 if(qNorm)
mas01cr@0 1468 delete qNorm;
mas01cr@0 1469 if(timesdata)
mas01cr@0 1470 delete timesdata;
mas01cr@0 1471 if(meanDBdur)
mas01cr@0 1472 delete meanDBdur;
mas01cr@0 1473
mas01cr@0 1474 }
mas01cr@0 1475
mas01cr@0 1476
mas01mc@20 1477 // k nearest-neighbor (k-NN) search between query and target tracks
mas01mc@20 1478 // efficient implementation based on matched filter
mas01mc@20 1479 // assumes normed shingles
mas01mc@20 1480 // outputs distances of retrieved shingles, max retreived = pointNN shingles per per track
mas01cr@133 1481 void audioDB::trackSequenceQueryNN(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
mas01cr@0 1482
mas01cr@177 1483 initTables(dbName, inFile);
mas01cr@0 1484
mas01cr@0 1485 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 1486 // we use stdout in this stub version
mas01cr@0 1487 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 1488 double* query = (double*)(indata+sizeof(int));
mas01cr@0 1489 double* queryCopy = 0;
mas01cr@0 1490
mas01cr@0 1491 double qMeanL2;
mas01cr@0 1492 double* sMeanL2;
mas01cr@0 1493
mas01cr@0 1494 unsigned USE_THRESH=0;
mas01cr@0 1495 double SILENCE_THRESH=0;
mas01cr@0 1496 double DIFF_THRESH=0;
mas01cr@0 1497
mas01cr@0 1498 if(!(dbH->flags & O2_FLAG_L2NORM) )
mas01cr@55 1499 error("Database must be L2 normed for sequence query","use -L2NORM");
mas01cr@55 1500
mas01cr@55 1501 if(numVectors<sequenceLength)
mas01cr@55 1502 error("Query shorter than requested sequence length", "maybe use -l");
mas01cr@0 1503
mas01cr@60 1504 if(verbosity>1) {
mas01cr@0 1505 cerr << "performing norms ... "; cerr.flush();
mas01cr@60 1506 }
mas01cr@0 1507 unsigned dbVectors = dbH->length/(sizeof(double)*dbH->dim);
mas01mc@20 1508
mas01cr@0 1509 // Make a copy of the query
mas01cr@0 1510 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 1511 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 1512 qNorm = new double[numVectors];
mas01cr@0 1513 sNorm = new double[dbVectors];
mas01cr@0 1514 sMeanL2=new double[dbH->numFiles];
mas01cr@0 1515 assert(qNorm&&sNorm&&queryCopy&&sMeanL2&&sequenceLength);
mas01cr@0 1516 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 1517 query = queryCopy;
mas01mc@20 1518
mas01cr@0 1519 // Make norm measurements relative to sequenceLength
mas01cr@0 1520 unsigned w = sequenceLength-1;
mas01cr@0 1521 unsigned i,j;
mas01cr@0 1522 double* ps;
mas01cr@0 1523 double tmp1,tmp2;
mas01mc@20 1524
mas01cr@0 1525 // Copy the L2 norm values to core to avoid disk random access later on
mas01cr@0 1526 memcpy(sNorm, l2normTable, dbVectors*sizeof(double));
mas01cr@125 1527 double* qnPtr = qNorm;
mas01cr@0 1528 double* snPtr = sNorm;
mas01cr@0 1529 for(i=0; i<dbH->numFiles; i++){
mas01mc@20 1530 if(trackTable[i]>=sequenceLength){
mas01cr@0 1531 tmp1=*snPtr;
mas01cr@0 1532 j=1;
mas01cr@0 1533 w=sequenceLength-1;
mas01cr@0 1534 while(w--)
mas01cr@0 1535 *snPtr+=snPtr[j++];
mas01cr@0 1536 ps = snPtr+1;
mas01mc@18 1537 w=trackTable[i]-sequenceLength; // +1 - 1
mas01cr@0 1538 while(w--){
mas01cr@0 1539 tmp2=*ps;
mas01mc@20 1540 *ps=*(ps-1)-tmp1+*(ps+sequenceLength-1);
mas01cr@0 1541 tmp1=tmp2;
mas01cr@0 1542 ps++;
mas01cr@0 1543 }
mas01mc@20 1544 ps = snPtr;
mas01mc@20 1545 w=trackTable[i]-sequenceLength+1;
mas01mc@20 1546 while(w--){
mas01mc@20 1547 *ps=sqrt(*ps);
mas01mc@20 1548 ps++;
mas01mc@20 1549 }
mas01cr@0 1550 }
mas01mc@18 1551 snPtr+=trackTable[i];
mas01cr@0 1552 }
mas01cr@0 1553
mas01cr@0 1554 double* pn = sMeanL2;
mas01cr@0 1555 w=dbH->numFiles;
mas01cr@0 1556 while(w--)
mas01cr@0 1557 *pn++=0.0;
mas01cr@0 1558 ps=sNorm;
mas01mc@18 1559 unsigned processedTracks=0;
mas01cr@0 1560 for(i=0; i<dbH->numFiles; i++){
mas01mc@18 1561 if(trackTable[i]>sequenceLength-1){
mas01cr@57 1562 w = trackTable[i]-sequenceLength+1;
mas01cr@0 1563 pn = sMeanL2+i;
mas01mc@20 1564 *pn=0;
mas01cr@0 1565 while(w--)
mas01mc@20 1566 if(*ps>0)
mas01mc@20 1567 *pn+=*ps++;
mas01cr@57 1568 *pn/=trackTable[i]-sequenceLength+1;
mas01cr@0 1569 SILENCE_THRESH+=*pn;
mas01mc@18 1570 processedTracks++;
mas01cr@0 1571 }
mas01mc@18 1572 ps = sNorm + trackTable[i];
mas01cr@0 1573 }
mas01cr@60 1574 if(verbosity>1) {
mas01mc@18 1575 cerr << "processedTracks: " << processedTracks << endl;
mas01cr@60 1576 }
mas01mc@20 1577
mas01mc@18 1578 SILENCE_THRESH/=processedTracks;
mas01cr@0 1579 USE_THRESH=1; // Turn thresholding on
mas01mc@20 1580 DIFF_THRESH=SILENCE_THRESH; // mean shingle power
mas01mc@20 1581 SILENCE_THRESH/=5; // 20% of the mean shingle power is SILENCE
mas01cr@60 1582 if(verbosity>4) {
mas01mc@20 1583 cerr << "silence thresh: " << SILENCE_THRESH;
mas01cr@60 1584 }
mas01cr@0 1585 w=sequenceLength-1;
mas01cr@0 1586 i=1;
mas01cr@125 1587 tmp1=*qnPtr;
mas01cr@0 1588 while(w--)
mas01cr@125 1589 *qnPtr+=qnPtr[i++];
mas01cr@125 1590 ps = qnPtr+1;
mas01mc@20 1591 w=numVectors-sequenceLength; // +1 -1
mas01cr@0 1592 while(w--){
mas01cr@0 1593 tmp2=*ps;
mas01mc@20 1594 *ps=*(ps-1)-tmp1+*(ps+sequenceLength-1);
mas01cr@0 1595 tmp1=tmp2;
mas01mc@20 1596 ps++;
mas01mc@20 1597 }
mas01cr@125 1598 ps = qnPtr;
mas01mc@20 1599 qMeanL2 = 0;
mas01mc@20 1600 w=numVectors-sequenceLength+1;
mas01mc@20 1601 while(w--){
mas01mc@20 1602 *ps=sqrt(*ps);
mas01mc@20 1603 qMeanL2+=*ps++;
mas01cr@0 1604 }
mas01cr@0 1605 qMeanL2 /= numVectors-sequenceLength+1;
mas01mc@20 1606
mas01cr@60 1607 if(verbosity>1) {
mas01cr@60 1608 cerr << "done." << endl;
mas01cr@60 1609 }
mas01cr@0 1610
mas01cr@60 1611 if(verbosity>1) {
mas01mc@18 1612 cerr << "matching tracks..." << endl;
mas01cr@60 1613 }
mas01cr@0 1614
mas01cr@0 1615 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01mc@18 1616 assert(trackNN>0 && trackNN<=O2_MAXNN);
mas01cr@0 1617
mas01cr@0 1618 // Make temporary dynamic memory for results
mas01mc@18 1619 double trackDistances[trackNN];
mas01mc@18 1620 unsigned trackIDs[trackNN];
mas01mc@18 1621 unsigned trackQIndexes[trackNN];
mas01mc@18 1622 unsigned trackSIndexes[trackNN];
mas01cr@0 1623
mas01cr@0 1624 double distances[pointNN];
mas01cr@0 1625 unsigned qIndexes[pointNN];
mas01cr@0 1626 unsigned sIndexes[pointNN];
mas01cr@0 1627
mas01cr@0 1628
mas01mc@18 1629 unsigned k,l,m,n,track,trackOffset=0, HOP_SIZE=sequenceHop, wL=sequenceLength;
mas01cr@0 1630 double thisDist;
mas01cr@0 1631
mas01cr@0 1632 for(k=0; k<pointNN; k++){
mas01mc@20 1633 distances[k]=1.0e6;
mas01cr@0 1634 qIndexes[k]=~0;
mas01cr@0 1635 sIndexes[k]=~0;
mas01cr@0 1636 }
mas01cr@0 1637
mas01mc@18 1638 for(k=0; k<trackNN; k++){
mas01mc@20 1639 trackDistances[k]=1.0e6;
mas01mc@18 1640 trackQIndexes[k]=~0;
mas01mc@18 1641 trackSIndexes[k]=~0;
mas01mc@18 1642 trackIDs[k]=~0;
mas01cr@0 1643 }
mas01cr@0 1644
mas01cr@0 1645 // Timestamp and durations processing
mas01cr@0 1646 double meanQdur = 0;
mas01cr@0 1647 double* timesdata = 0;
mas01cr@0 1648 double* meanDBdur = 0;
mas01cr@0 1649
mas01cr@0 1650 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1651 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 1652 usingTimes=0;
mas01cr@0 1653 }
mas01cr@0 1654
mas01cr@0 1655 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 1656 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 1657
mas01cr@0 1658 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1659 timesdata = new double[numVectors];
mas01cr@0 1660 assert(timesdata);
mas01cr@0 1661 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1662 // Calculate durations of points
mas01cr@0 1663 for(k=0; k<numVectors-1; k++){
mas01cr@0 1664 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1665 meanQdur+=timesdata[k];
mas01cr@0 1666 }
mas01cr@0 1667 meanQdur/=k;
mas01cr@60 1668 if(verbosity>1) {
mas01cr@0 1669 cerr << "mean query file duration: " << meanQdur << endl;
mas01cr@60 1670 }
mas01cr@0 1671 meanDBdur = new double[dbH->numFiles];
mas01cr@0 1672 assert(meanDBdur);
mas01cr@0 1673 for(k=0; k<dbH->numFiles; k++){
mas01cr@0 1674 meanDBdur[k]=0.0;
mas01mc@18 1675 for(j=0; j<trackTable[k]-1 ; j++)
mas01cr@0 1676 meanDBdur[k]+=timesTable[j+1]-timesTable[j];
mas01cr@0 1677 meanDBdur[k]/=j;
mas01cr@0 1678 }
mas01cr@0 1679 }
mas01cr@0 1680
mas01cr@0 1681 if(usingQueryPoint)
mas01cr@0 1682 if(queryPoint>numVectors || queryPoint>numVectors-wL+1)
mas01cr@0 1683 error("queryPoint > numVectors-wL+1 in query");
mas01cr@0 1684 else{
mas01cr@60 1685 if(verbosity>1) {
mas01cr@0 1686 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@60 1687 }
mas01cr@0 1688 query=query+queryPoint*dbH->dim;
mas01cr@125 1689 qnPtr=qnPtr+queryPoint;
mas01cr@0 1690 numVectors=wL;
mas01cr@0 1691 }
mas01cr@0 1692
mas01mc@20 1693 double ** D = 0; // Differences query and target
mas01cr@0 1694 double ** DD = 0; // Matched filter distance
mas01cr@0 1695
mas01cr@0 1696 D = new double*[numVectors];
mas01cr@0 1697 assert(D);
mas01cr@0 1698 DD = new double*[numVectors];
mas01cr@0 1699 assert(DD);
mas01cr@0 1700
mas01cr@0 1701 gettimeofday(&tv1, NULL);
mas01mc@18 1702 processedTracks=0;
mas01mc@18 1703 unsigned successfulTracks=0;
mas01cr@0 1704
mas01cr@0 1705 double* qp;
mas01cr@0 1706 double* sp;
mas01cr@0 1707 double* dp;
mas01cr@0 1708
mas01mc@18 1709 // build track offset table
mas01mc@18 1710 unsigned *trackOffsetTable = new unsigned[dbH->numFiles];
mas01mc@18 1711 unsigned cumTrack=0;
mas01mc@18 1712 unsigned trackIndexOffset;
mas01cr@0 1713 for(k=0; k<dbH->numFiles;k++){
mas01mc@18 1714 trackOffsetTable[k]=cumTrack;
mas01mc@18 1715 cumTrack+=trackTable[k]*dbH->dim;
mas01cr@0 1716 }
mas01cr@0 1717
mas01cr@0 1718 char nextKey [MAXSTR];
mas01mc@20 1719
mas01mc@20 1720 // chi^2 statistics
mas01mc@20 1721 double sampleCount = 0;
mas01mc@20 1722 double sampleSum = 0;
mas01mc@20 1723 double logSampleSum = 0;
mas01mc@20 1724 double minSample = 1e9;
mas01mc@20 1725 double maxSample = 0;
mas01mc@20 1726
mas01mc@20 1727 // Track loop
mas01mc@18 1728 for(processedTracks=0, track=0 ; processedTracks < dbH->numFiles ; track++, processedTracks++){
mas01cr@0 1729
mas01mc@18 1730 // get trackID from file if using a control file
mas01mc@18 1731 if(trackFile){
mas01mc@18 1732 if(!trackFile->eof()){
mas01mc@18 1733 trackFile->getline(nextKey,MAXSTR);
mas01mc@18 1734 track=getKeyPos(nextKey);
mas01cr@0 1735 }
mas01cr@0 1736 else
mas01cr@0 1737 break;
mas01cr@0 1738 }
mas01mc@12 1739
mas01mc@18 1740 trackOffset=trackOffsetTable[track]; // numDoubles offset
mas01mc@18 1741 trackIndexOffset=trackOffset/dbH->dim; // numVectors offset
mas01cr@0 1742
mas01cr@57 1743 if(sequenceLength<=trackTable[track]){ // test for short sequences
mas01cr@0 1744
mas01cr@60 1745 if(verbosity>7) {
mas01mc@18 1746 cerr << track << "." << trackIndexOffset << "." << trackTable[track] << " | ";cerr.flush();
mas01cr@60 1747 }
mas01cr@0 1748
mas01mc@20 1749 // Sum products matrix
mas01cr@0 1750 for(j=0; j<numVectors;j++){
mas01mc@18 1751 D[j]=new double[trackTable[track]];
mas01cr@0 1752 assert(D[j]);
mas01cr@0 1753
mas01cr@0 1754 }
mas01cr@0 1755
mas01cr@0 1756 // Matched filter matrix
mas01cr@0 1757 for(j=0; j<numVectors;j++){
mas01mc@18 1758 DD[j]=new double[trackTable[track]];
mas01cr@0 1759 assert(DD[j]);
mas01cr@0 1760 }
mas01cr@0 1761
mas01mc@20 1762 // Dot product
mas01cr@0 1763 for(j=0; j<numVectors; j++)
mas01mc@18 1764 for(k=0; k<trackTable[track]; k++){
mas01cr@0 1765 qp=query+j*dbH->dim;
mas01mc@18 1766 sp=dataBuf+trackOffset+k*dbH->dim;
mas01cr@0 1767 DD[j][k]=0.0; // Initialize matched filter array
mas01cr@0 1768 dp=&D[j][k]; // point to correlation cell j,k
mas01cr@0 1769 *dp=0.0; // initialize correlation cell
mas01cr@0 1770 l=dbH->dim; // size of vectors
mas01cr@0 1771 while(l--)
mas01cr@0 1772 *dp+=*qp++**sp++;
mas01cr@0 1773 }
mas01cr@0 1774
mas01cr@0 1775 // Matched Filter
mas01cr@0 1776 // HOP SIZE == 1
mas01cr@0 1777 double* spd;
mas01cr@0 1778 if(HOP_SIZE==1){ // HOP_SIZE = shingleHop
mas01cr@0 1779 for(w=0; w<wL; w++)
mas01cr@0 1780 for(j=0; j<numVectors-w; j++){
mas01cr@0 1781 sp=DD[j];
mas01cr@0 1782 spd=D[j+w]+w;
mas01mc@18 1783 k=trackTable[track]-w;
mas01cr@0 1784 while(k--)
mas01cr@0 1785 *sp+++=*spd++;
mas01cr@0 1786 }
mas01cr@0 1787 }
mas01mc@20 1788
mas01cr@0 1789 else{ // HOP_SIZE != 1
mas01cr@0 1790 for(w=0; w<wL; w++)
mas01cr@0 1791 for(j=0; j<numVectors-w; j+=HOP_SIZE){
mas01cr@0 1792 sp=DD[j];
mas01cr@0 1793 spd=D[j+w]+w;
mas01mc@18 1794 for(k=0; k<trackTable[track]-w; k+=HOP_SIZE){
mas01cr@0 1795 *sp+=*spd;
mas01cr@0 1796 sp+=HOP_SIZE;
mas01cr@0 1797 spd+=HOP_SIZE;
mas01cr@0 1798 }
mas01cr@0 1799 }
mas01cr@0 1800 }
mas01cr@0 1801
mas01cr@60 1802 if(verbosity>3 && usingTimes) {
mas01mc@18 1803 cerr << "meanQdur=" << meanQdur << " meanDBdur=" << meanDBdur[track] << endl;
mas01cr@0 1804 cerr.flush();
mas01cr@0 1805 }
mas01cr@0 1806
mas01cr@0 1807 if(!usingTimes ||
mas01cr@0 1808 (usingTimes
mas01mc@18 1809 && fabs(meanDBdur[track]-meanQdur)<meanQdur*timesTol)){
mas01cr@0 1810
mas01cr@60 1811 if(verbosity>3 && usingTimes) {
mas01cr@0 1812 cerr << "within duration tolerance." << endl;
mas01cr@0 1813 cerr.flush();
mas01cr@0 1814 }
mas01cr@0 1815
mas01cr@0 1816 // Search for minimum distance by shingles (concatenated vectors)
mas01cr@53 1817 for(j=0;j<=numVectors-wL;j+=HOP_SIZE)
mas01cr@53 1818 for(k=0;k<=trackTable[track]-wL;k+=HOP_SIZE){
mas01cr@125 1819 thisDist=2-(2/(qnPtr[j]*sNorm[trackIndexOffset+k]))*DD[j][k];
mas01cr@60 1820 if(verbosity>10) {
mas01cr@125 1821 cerr << thisDist << " " << qnPtr[j] << " " << sNorm[trackIndexOffset+k] << endl;
mas01cr@60 1822 }
mas01mc@20 1823 // Gather chi^2 statistics
mas01mc@20 1824 if(thisDist<minSample)
mas01mc@20 1825 minSample=thisDist;
mas01mc@20 1826 else if(thisDist>maxSample)
mas01mc@20 1827 maxSample=thisDist;
mas01mc@20 1828 if(thisDist>1e-9){
mas01mc@20 1829 sampleCount++;
mas01mc@20 1830 sampleSum+=thisDist;
mas01mc@20 1831 logSampleSum+=log(thisDist);
mas01mc@20 1832 }
mas01mc@20 1833
mas01cr@125 1834 // diffL2 = fabs(qnPtr[j] - sNorm[trackIndexOffset+k]);
mas01cr@0 1835 // Power test
mas01cr@0 1836 if(!USE_THRESH ||
mas01cr@0 1837 // Threshold on mean L2 of Q and S sequences
mas01cr@125 1838 (USE_THRESH && qnPtr[j]>SILENCE_THRESH && sNorm[trackIndexOffset+k]>SILENCE_THRESH &&
mas01cr@0 1839 // Are both query and target windows above mean energy?
mas01cr@125 1840 (qnPtr[j]>qMeanL2*.25 && sNorm[trackIndexOffset+k]>sMeanL2[track]*.25))) // && diffL2 < DIFF_THRESH )))
mas01mc@20 1841 thisDist=thisDist; // Computed above
mas01cr@0 1842 else
mas01mc@20 1843 thisDist=1000000.0;
mas01mc@20 1844
mas01mc@20 1845 // k-NN match algorithm
mas01cr@58 1846 m=pointNN;
mas01mc@20 1847 while(m--){
mas01mc@20 1848 if(thisDist<=distances[m])
mas01mc@20 1849 if(m==0 || thisDist>=distances[m-1]){
mas01cr@0 1850 // Shuffle distances up the list
mas01cr@0 1851 for(l=pointNN-1; l>m; l--){
mas01cr@0 1852 distances[l]=distances[l-1];
mas01cr@0 1853 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1854 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1855 }
mas01cr@0 1856 distances[m]=thisDist;
mas01cr@0 1857 if(usingQueryPoint)
mas01cr@0 1858 qIndexes[m]=queryPoint;
mas01cr@0 1859 else
mas01cr@0 1860 qIndexes[m]=j;
mas01cr@0 1861 sIndexes[m]=k;
mas01cr@0 1862 break;
mas01mc@20 1863 }
mas01cr@0 1864 }
mas01cr@0 1865 }
mas01cr@0 1866 // Calculate the mean of the N-Best matches
mas01cr@0 1867 thisDist=0.0;
mas01cr@53 1868 for(m=0; m<pointNN; m++) {
mas01cr@53 1869 if (distances[m] == 1000000.0) break;
mas01mc@20 1870 thisDist+=distances[m];
mas01cr@53 1871 }
mas01cr@53 1872 thisDist/=m;
mas01cr@0 1873
mas01mc@12 1874 // Let's see the distances then...
mas01cr@60 1875 if(verbosity>3) {
mas01mc@18 1876 cerr << fileTable+track*O2_FILETABLESIZE << " " << thisDist << endl;
mas01cr@60 1877 }
mas01mc@12 1878
mas01mc@20 1879
mas01mc@18 1880 // All the track stuff goes here
mas01cr@58 1881 n=trackNN;
mas01cr@0 1882 while(n--){
mas01mc@20 1883 if(thisDist<=trackDistances[n]){
mas01mc@20 1884 if((n==0 || thisDist>=trackDistances[n-1])){
mas01cr@0 1885 // Copy all values above up the queue
mas01mc@18 1886 for( l=trackNN-1 ; l > n ; l--){
mas01mc@18 1887 trackDistances[l]=trackDistances[l-1];
mas01mc@18 1888 trackQIndexes[l]=trackQIndexes[l-1];
mas01mc@18 1889 trackSIndexes[l]=trackSIndexes[l-1];
mas01mc@18 1890 trackIDs[l]=trackIDs[l-1];
mas01cr@0 1891 }
mas01mc@18 1892 trackDistances[n]=thisDist;
mas01mc@18 1893 trackQIndexes[n]=qIndexes[0];
mas01mc@18 1894 trackSIndexes[n]=sIndexes[0];
mas01mc@18 1895 successfulTracks++;
mas01mc@18 1896 trackIDs[n]=track;
mas01cr@0 1897 break;
mas01cr@0 1898 }
mas01cr@0 1899 }
mas01cr@0 1900 else
mas01cr@0 1901 break;
mas01cr@0 1902 }
mas01cr@0 1903 } // Duration match
mas01mc@20 1904
mas01mc@18 1905 // Clean up current track
mas01cr@0 1906 if(D!=NULL){
mas01cr@0 1907 for(j=0; j<numVectors; j++)
mas01cr@0 1908 delete[] D[j];
mas01cr@0 1909 }
mas01cr@0 1910
mas01cr@0 1911 if(DD!=NULL){
mas01cr@0 1912 for(j=0; j<numVectors; j++)
mas01cr@0 1913 delete[] DD[j];
mas01cr@0 1914 }
mas01cr@0 1915 }
mas01mc@20 1916 // per-track reset array values
mas01mc@20 1917 for(unsigned k=0; k<pointNN; k++){
mas01mc@20 1918 distances[k]=1.0e6;
mas01mc@20 1919 qIndexes[k]=~0;
mas01mc@20 1920 sIndexes[k]=~0;
mas01mc@20 1921 }
mas01cr@0 1922 }
mas01cr@0 1923
mas01cr@0 1924 gettimeofday(&tv2,NULL);
mas01cr@60 1925 if(verbosity>1) {
mas01mc@18 1926 cerr << endl << "processed tracks :" << processedTracks << " matched tracks: " << successfulTracks << " elapsed time:"
mas01cr@0 1927 << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01mc@20 1928 cerr << "sampleCount: " << sampleCount << " sampleSum: " << sampleSum << " logSampleSum: " << logSampleSum
mas01mc@20 1929 << " minSample: " << minSample << " maxSample: " << maxSample << endl;
mas01mc@20 1930 }
mas01cr@133 1931 if(adbQueryResponse==0){
mas01cr@60 1932 if(verbosity>1) {
mas01cr@0 1933 cerr<<endl;
mas01cr@60 1934 }
mas01cr@0 1935 // Output answer
mas01cr@0 1936 // Loop over nearest neighbours
mas01mc@18 1937 for(k=0; k < min(trackNN,successfulTracks); k++)
mas01mc@20 1938 cout << fileTable+trackIDs[k]*O2_FILETABLESIZE << " " << trackDistances[k] << " "
mas01mc@20 1939 << trackQIndexes[k] << " " << trackSIndexes[k] << endl;
mas01cr@0 1940 }
mas01cr@0 1941 else{ // Process Web Services Query
mas01mc@18 1942 int listLen = min(trackNN, processedTracks);
mas01cr@133 1943 adbQueryResponse->result.__sizeRlist=listLen;
mas01cr@133 1944 adbQueryResponse->result.__sizeDist=listLen;
mas01cr@133 1945 adbQueryResponse->result.__sizeQpos=listLen;
mas01cr@133 1946 adbQueryResponse->result.__sizeSpos=listLen;
mas01cr@133 1947 adbQueryResponse->result.Rlist= new char*[listLen];
mas01cr@133 1948 adbQueryResponse->result.Dist = new double[listLen];
mas01cr@133 1949 adbQueryResponse->result.Qpos = new unsigned int[listLen];
mas01cr@133 1950 adbQueryResponse->result.Spos = new unsigned int[listLen];
mas01cr@133 1951 for(k=0; k<(unsigned)adbQueryResponse->result.__sizeRlist; k++){
mas01cr@133 1952 adbQueryResponse->result.Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@133 1953 adbQueryResponse->result.Dist[k]=trackDistances[k];
mas01cr@133 1954 adbQueryResponse->result.Qpos[k]=trackQIndexes[k];
mas01cr@133 1955 adbQueryResponse->result.Spos[k]=trackSIndexes[k];
mas01cr@133 1956 sprintf(adbQueryResponse->result.Rlist[k], "%s", fileTable+trackIDs[k]*O2_FILETABLESIZE);
mas01cr@0 1957 }
mas01cr@0 1958 }
mas01cr@0 1959
mas01cr@0 1960
mas01cr@0 1961 // Clean up
mas01mc@18 1962 if(trackOffsetTable)
mas01mc@20 1963 delete[] trackOffsetTable;
mas01cr@0 1964 if(queryCopy)
mas01mc@20 1965 delete[] queryCopy;
mas01cr@121 1966 if(qNorm)
mas01cr@121 1967 delete[] qNorm;
mas01cr@121 1968 if(sNorm)
mas01cr@121 1969 delete[] sNorm;
mas01cr@121 1970 if(sMeanL2)
mas01cr@121 1971 delete[] sMeanL2;
mas01cr@0 1972 if(D)
mas01cr@0 1973 delete[] D;
mas01cr@0 1974 if(DD)
mas01cr@0 1975 delete[] DD;
mas01cr@0 1976 if(timesdata)
mas01mc@20 1977 delete[] timesdata;
mas01cr@0 1978 if(meanDBdur)
mas01mc@20 1979 delete[] meanDBdur;
mas01cr@0 1980
mas01cr@0 1981
mas01cr@0 1982 }
mas01cr@0 1983
mas01mc@20 1984 // Radius search between query and target tracks
mas01mc@20 1985 // efficient implementation based on matched filter
mas01mc@20 1986 // assumes normed shingles
mas01mc@20 1987 // outputs count of retrieved shingles, max retreived = one shingle per query shingle per track
mas01cr@133 1988 void audioDB::trackSequenceQueryRad(const char* dbName, const char* inFile, adb__queryResponse *adbQueryResponse){
mas01mc@17 1989
mas01cr@177 1990 initTables(dbName, inFile);
mas01mc@17 1991
mas01mc@17 1992 // For each input vector, find the closest pointNN matching output vectors and report
mas01mc@17 1993 // we use stdout in this stub version
mas01mc@17 1994 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01mc@17 1995 double* query = (double*)(indata+sizeof(int));
mas01mc@17 1996 double* queryCopy = 0;
mas01mc@17 1997
mas01mc@17 1998 double qMeanL2;
mas01mc@17 1999 double* sMeanL2;
mas01mc@17 2000
mas01mc@17 2001 unsigned USE_THRESH=0;
mas01mc@17 2002 double SILENCE_THRESH=0;
mas01mc@17 2003 double DIFF_THRESH=0;
mas01mc@17 2004
mas01mc@17 2005 if(!(dbH->flags & O2_FLAG_L2NORM) )
mas01mc@17 2006 error("Database must be L2 normed for sequence query","use -l2norm");
mas01mc@17 2007
mas01cr@60 2008 if(verbosity>1) {
mas01mc@17 2009 cerr << "performing norms ... "; cerr.flush();
mas01cr@60 2010 }
mas01mc@17 2011 unsigned dbVectors = dbH->length/(sizeof(double)*dbH->dim);
mas01mc@18 2012
mas01mc@17 2013 // Make a copy of the query
mas01mc@17 2014 queryCopy = new double[numVectors*dbH->dim];
mas01mc@17 2015 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01mc@17 2016 qNorm = new double[numVectors];
mas01mc@17 2017 sNorm = new double[dbVectors];
mas01mc@17 2018 sMeanL2=new double[dbH->numFiles];
mas01mc@17 2019 assert(qNorm&&sNorm&&queryCopy&&sMeanL2&&sequenceLength);
mas01mc@17 2020 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01mc@17 2021 query = queryCopy;
mas01mc@18 2022
mas01mc@17 2023 // Make norm measurements relative to sequenceLength
mas01mc@17 2024 unsigned w = sequenceLength-1;
mas01mc@17 2025 unsigned i,j;
mas01mc@17 2026 double* ps;
mas01mc@17 2027 double tmp1,tmp2;
mas01mc@18 2028
mas01mc@17 2029 // Copy the L2 norm values to core to avoid disk random access later on
mas01mc@17 2030 memcpy(sNorm, l2normTable, dbVectors*sizeof(double));
mas01mc@17 2031 double* snPtr = sNorm;
mas01cr@125 2032 double* qnPtr = qNorm;
mas01mc@17 2033 for(i=0; i<dbH->numFiles; i++){
mas01mc@18 2034 if(trackTable[i]>=sequenceLength){
mas01mc@17 2035 tmp1=*snPtr;
mas01mc@17 2036 j=1;
mas01mc@17 2037 w=sequenceLength-1;
mas01mc@17 2038 while(w--)
mas01mc@17 2039 *snPtr+=snPtr[j++];
mas01mc@17 2040 ps = snPtr+1;
mas01mc@18 2041 w=trackTable[i]-sequenceLength; // +1 - 1
mas01mc@17 2042 while(w--){
mas01mc@17 2043 tmp2=*ps;
mas01mc@17 2044 *ps=*(ps-1)-tmp1+*(ps+sequenceLength-1);
mas01mc@17 2045 tmp1=tmp2;
mas01mc@17 2046 ps++;
mas01mc@17 2047 }
mas01mc@17 2048 ps = snPtr;
mas01mc@18 2049 w=trackTable[i]-sequenceLength+1;
mas01mc@17 2050 while(w--){
mas01mc@17 2051 *ps=sqrt(*ps);
mas01mc@17 2052 ps++;
mas01mc@17 2053 }
mas01mc@17 2054 }
mas01mc@18 2055 snPtr+=trackTable[i];
mas01mc@17 2056 }
mas01mc@17 2057
mas01mc@17 2058 double* pn = sMeanL2;
mas01mc@17 2059 w=dbH->numFiles;
mas01mc@17 2060 while(w--)
mas01mc@17 2061 *pn++=0.0;
mas01mc@17 2062 ps=sNorm;
mas01mc@18 2063 unsigned processedTracks=0;
mas01mc@17 2064 for(i=0; i<dbH->numFiles; i++){
mas01mc@18 2065 if(trackTable[i]>sequenceLength-1){
mas01cr@70 2066 w = trackTable[i]-sequenceLength+1;
mas01mc@17 2067 pn = sMeanL2+i;
mas01mc@17 2068 *pn=0;
mas01mc@17 2069 while(w--)
mas01mc@17 2070 if(*ps>0)
mas01mc@17 2071 *pn+=*ps++;
mas01cr@70 2072 *pn/=trackTable[i]-sequenceLength+1;
mas01mc@17 2073 SILENCE_THRESH+=*pn;
mas01mc@18 2074 processedTracks++;
mas01mc@17 2075 }
mas01mc@18 2076 ps = sNorm + trackTable[i];
mas01mc@17 2077 }
mas01cr@60 2078 if(verbosity>1) {
mas01mc@18 2079 cerr << "processedTracks: " << processedTracks << endl;
mas01cr@60 2080 }
mas01mc@17 2081
mas01mc@18 2082 SILENCE_THRESH/=processedTracks;
mas01mc@17 2083 USE_THRESH=1; // Turn thresholding on
mas01mc@18 2084 DIFF_THRESH=SILENCE_THRESH; // mean shingle power
mas01mc@17 2085 SILENCE_THRESH/=5; // 20% of the mean shingle power is SILENCE
mas01cr@60 2086 if(verbosity>4) {
mas01mc@17 2087 cerr << "silence thresh: " << SILENCE_THRESH;
mas01cr@60 2088 }
mas01mc@17 2089 w=sequenceLength-1;
mas01mc@17 2090 i=1;
mas01cr@125 2091 tmp1=*qnPtr;
mas01mc@17 2092 while(w--)
mas01cr@125 2093 *qnPtr+=qnPtr[i++];
mas01cr@125 2094 ps = qnPtr+1;
mas01mc@17 2095 w=numVectors-sequenceLength; // +1 -1
mas01mc@17 2096 while(w--){
mas01mc@17 2097 tmp2=*ps;
mas01mc@17 2098 *ps=*(ps-1)-tmp1+*(ps+sequenceLength-1);
mas01mc@17 2099 tmp1=tmp2;
mas01mc@17 2100 ps++;
mas01mc@17 2101 }
mas01cr@125 2102 ps = qnPtr;
mas01mc@17 2103 qMeanL2 = 0;
mas01mc@17 2104 w=numVectors-sequenceLength+1;
mas01mc@17 2105 while(w--){
mas01mc@17 2106 *ps=sqrt(*ps);
mas01mc@17 2107 qMeanL2+=*ps++;
mas01mc@17 2108 }
mas01mc@17 2109 qMeanL2 /= numVectors-sequenceLength+1;
mas01mc@17 2110
mas01cr@60 2111 if(verbosity>1) {
mas01mc@17 2112 cerr << "done." << endl;
mas01cr@60 2113 }
mas01mc@17 2114
mas01cr@60 2115 if(verbosity>1) {
mas01mc@18 2116 cerr << "matching tracks..." << endl;
mas01cr@60 2117 }
mas01mc@17 2118
mas01mc@17 2119 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01mc@18 2120 assert(trackNN>0 && trackNN<=O2_MAXNN);
mas01mc@17 2121
mas01mc@17 2122 // Make temporary dynamic memory for results
mas01mc@18 2123 double trackDistances[trackNN];
mas01mc@18 2124 unsigned trackIDs[trackNN];
mas01mc@18 2125 unsigned trackQIndexes[trackNN];
mas01mc@18 2126 unsigned trackSIndexes[trackNN];
mas01mc@17 2127
mas01mc@17 2128 double distances[pointNN];
mas01mc@17 2129 unsigned qIndexes[pointNN];
mas01mc@17 2130 unsigned sIndexes[pointNN];
mas01mc@17 2131
mas01mc@17 2132
mas01cr@59 2133 unsigned k,l,n,track,trackOffset=0, HOP_SIZE=sequenceHop, wL=sequenceLength;
mas01mc@17 2134 double thisDist;
mas01mc@17 2135
mas01mc@17 2136 for(k=0; k<pointNN; k++){
mas01mc@17 2137 distances[k]=0.0;
mas01mc@17 2138 qIndexes[k]=~0;
mas01mc@17 2139 sIndexes[k]=~0;
mas01mc@17 2140 }
mas01mc@17 2141
mas01mc@18 2142 for(k=0; k<trackNN; k++){
mas01mc@18 2143 trackDistances[k]=0.0;
mas01mc@18 2144 trackQIndexes[k]=~0;
mas01mc@18 2145 trackSIndexes[k]=~0;
mas01mc@18 2146 trackIDs[k]=~0;
mas01mc@17 2147 }
mas01mc@17 2148
mas01mc@17 2149 // Timestamp and durations processing
mas01mc@17 2150 double meanQdur = 0;
mas01mc@17 2151 double* timesdata = 0;
mas01mc@17 2152 double* meanDBdur = 0;
mas01mc@17 2153
mas01mc@17 2154 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01mc@17 2155 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01mc@17 2156 usingTimes=0;
mas01mc@17 2157 }
mas01mc@17 2158
mas01mc@17 2159 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01mc@17 2160 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01mc@17 2161
mas01mc@17 2162 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01mc@17 2163 timesdata = new double[numVectors];
mas01mc@17 2164 assert(timesdata);
mas01mc@17 2165 insertTimeStamps(numVectors, timesFile, timesdata);
mas01mc@17 2166 // Calculate durations of points
mas01mc@17 2167 for(k=0; k<numVectors-1; k++){
mas01mc@17 2168 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01mc@17 2169 meanQdur+=timesdata[k];
mas01mc@17 2170 }
mas01mc@17 2171 meanQdur/=k;
mas01cr@60 2172 if(verbosity>1) {
mas01mc@17 2173 cerr << "mean query file duration: " << meanQdur << endl;
mas01cr@60 2174 }
mas01mc@17 2175 meanDBdur = new double[dbH->numFiles];
mas01mc@17 2176 assert(meanDBdur);
mas01mc@17 2177 for(k=0; k<dbH->numFiles; k++){
mas01mc@17 2178 meanDBdur[k]=0.0;
mas01mc@18 2179 for(j=0; j<trackTable[k]-1 ; j++)
mas01mc@17 2180 meanDBdur[k]+=timesTable[j+1]-timesTable[j];
mas01mc@17 2181 meanDBdur[k]/=j;
mas01mc@17 2182 }
mas01mc@17 2183 }
mas01mc@17 2184
mas01mc@17 2185 if(usingQueryPoint)
mas01mc@17 2186 if(queryPoint>numVectors || queryPoint>numVectors-wL+1)
mas01mc@17 2187 error("queryPoint > numVectors-wL+1 in query");
mas01mc@17 2188 else{
mas01cr@60 2189 if(verbosity>1) {
mas01mc@17 2190 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@60 2191 }
mas01mc@17 2192 query=query+queryPoint*dbH->dim;
mas01cr@125 2193 qnPtr=qnPtr+queryPoint;
mas01mc@17 2194 numVectors=wL;
mas01mc@17 2195 }
mas01mc@17 2196
mas01mc@17 2197 double ** D = 0; // Differences query and target
mas01mc@17 2198 double ** DD = 0; // Matched filter distance
mas01mc@17 2199
mas01mc@17 2200 D = new double*[numVectors];
mas01mc@17 2201 assert(D);
mas01mc@17 2202 DD = new double*[numVectors];
mas01mc@17 2203 assert(DD);
mas01mc@17 2204
mas01mc@17 2205 gettimeofday(&tv1, NULL);
mas01mc@18 2206 processedTracks=0;
mas01mc@18 2207 unsigned successfulTracks=0;
mas01mc@17 2208
mas01mc@17 2209 double* qp;
mas01mc@17 2210 double* sp;
mas01mc@17 2211 double* dp;
mas01mc@17 2212
mas01mc@18 2213 // build track offset table
mas01mc@18 2214 unsigned *trackOffsetTable = new unsigned[dbH->numFiles];
mas01mc@18 2215 unsigned cumTrack=0;
mas01mc@18 2216 unsigned trackIndexOffset;
mas01mc@17 2217 for(k=0; k<dbH->numFiles;k++){
mas01mc@18 2218 trackOffsetTable[k]=cumTrack;
mas01mc@18 2219 cumTrack+=trackTable[k]*dbH->dim;
mas01mc@17 2220 }
mas01mc@17 2221
mas01mc@17 2222 char nextKey [MAXSTR];
mas01mc@17 2223
mas01mc@17 2224 // chi^2 statistics
mas01mc@17 2225 double sampleCount = 0;
mas01mc@17 2226 double sampleSum = 0;
mas01mc@17 2227 double logSampleSum = 0;
mas01mc@17 2228 double minSample = 1e9;
mas01mc@17 2229 double maxSample = 0;
mas01mc@17 2230
mas01mc@17 2231 // Track loop
mas01mc@18 2232 for(processedTracks=0, track=0 ; processedTracks < dbH->numFiles ; track++, processedTracks++){
mas01mc@17 2233
mas01mc@18 2234 // get trackID from file if using a control file
mas01mc@18 2235 if(trackFile){
mas01mc@18 2236 if(!trackFile->eof()){
mas01mc@18 2237 trackFile->getline(nextKey,MAXSTR);
mas01mc@18 2238 track=getKeyPos(nextKey);
mas01mc@17 2239 }
mas01mc@17 2240 else
mas01mc@17 2241 break;
mas01mc@17 2242 }
mas01mc@17 2243
mas01mc@18 2244 trackOffset=trackOffsetTable[track]; // numDoubles offset
mas01mc@18 2245 trackIndexOffset=trackOffset/dbH->dim; // numVectors offset
mas01mc@17 2246
mas01cr@70 2247 if(sequenceLength<=trackTable[track]){ // test for short sequences
mas01mc@17 2248
mas01cr@60 2249 if(verbosity>7) {
mas01mc@18 2250 cerr << track << "." << trackIndexOffset << "." << trackTable[track] << " | ";cerr.flush();
mas01cr@60 2251 }
mas01cr@60 2252
mas01mc@17 2253 // Sum products matrix
mas01mc@17 2254 for(j=0; j<numVectors;j++){
mas01mc@18 2255 D[j]=new double[trackTable[track]];
mas01mc@17 2256 assert(D[j]);
mas01mc@17 2257
mas01mc@17 2258 }
mas01mc@17 2259
mas01mc@17 2260 // Matched filter matrix
mas01mc@17 2261 for(j=0; j<numVectors;j++){
mas01mc@18 2262 DD[j]=new double[trackTable[track]];
mas01mc@17 2263 assert(DD[j]);
mas01mc@17 2264 }
mas01mc@17 2265
mas01mc@17 2266 // Dot product
mas01mc@17 2267 for(j=0; j<numVectors; j++)
mas01mc@18 2268 for(k=0; k<trackTable[track]; k++){
mas01mc@17 2269 qp=query+j*dbH->dim;
mas01mc@18 2270 sp=dataBuf+trackOffset+k*dbH->dim;
mas01mc@17 2271 DD[j][k]=0.0; // Initialize matched filter array
mas01mc@17 2272 dp=&D[j][k]; // point to correlation cell j,k
mas01mc@17 2273 *dp=0.0; // initialize correlation cell
mas01mc@17 2274 l=dbH->dim; // size of vectors
mas01mc@17 2275 while(l--)
mas01mc@17 2276 *dp+=*qp++**sp++;
mas01mc@17 2277 }
mas01mc@17 2278
mas01mc@17 2279 // Matched Filter
mas01mc@17 2280 // HOP SIZE == 1
mas01mc@17 2281 double* spd;
mas01mc@17 2282 if(HOP_SIZE==1){ // HOP_SIZE = shingleHop
mas01mc@17 2283 for(w=0; w<wL; w++)
mas01mc@17 2284 for(j=0; j<numVectors-w; j++){
mas01mc@17 2285 sp=DD[j];
mas01mc@17 2286 spd=D[j+w]+w;
mas01mc@18 2287 k=trackTable[track]-w;
mas01mc@17 2288 while(k--)
mas01mc@17 2289 *sp+++=*spd++;
mas01mc@17 2290 }
mas01mc@17 2291 }
mas01mc@17 2292
mas01mc@17 2293 else{ // HOP_SIZE != 1
mas01mc@17 2294 for(w=0; w<wL; w++)
mas01mc@17 2295 for(j=0; j<numVectors-w; j+=HOP_SIZE){
mas01mc@17 2296 sp=DD[j];
mas01mc@17 2297 spd=D[j+w]+w;
mas01mc@18 2298 for(k=0; k<trackTable[track]-w; k+=HOP_SIZE){
mas01mc@17 2299 *sp+=*spd;
mas01mc@17 2300 sp+=HOP_SIZE;
mas01mc@17 2301 spd+=HOP_SIZE;
mas01mc@17 2302 }
mas01mc@17 2303 }
mas01mc@17 2304 }
mas01mc@17 2305
mas01cr@60 2306 if(verbosity>3 && usingTimes) {
mas01mc@18 2307 cerr << "meanQdur=" << meanQdur << " meanDBdur=" << meanDBdur[track] << endl;
mas01mc@17 2308 cerr.flush();
mas01mc@17 2309 }
mas01mc@17 2310
mas01mc@17 2311 if(!usingTimes ||
mas01mc@17 2312 (usingTimes
mas01mc@18 2313 && fabs(meanDBdur[track]-meanQdur)<meanQdur*timesTol)){
mas01mc@17 2314
mas01cr@60 2315 if(verbosity>3 && usingTimes) {
mas01mc@17 2316 cerr << "within duration tolerance." << endl;
mas01mc@17 2317 cerr.flush();
mas01mc@17 2318 }
mas01mc@17 2319
mas01mc@17 2320 // Search for minimum distance by shingles (concatenated vectors)
mas01cr@70 2321 for(j=0;j<=numVectors-wL;j+=HOP_SIZE)
mas01cr@70 2322 for(k=0;k<=trackTable[track]-wL;k+=HOP_SIZE){
mas01cr@125 2323 thisDist=2-(2/(qnPtr[j]*sNorm[trackIndexOffset+k]))*DD[j][k];
mas01cr@60 2324 if(verbosity>10) {
mas01cr@125 2325 cerr << thisDist << " " << qnPtr[j] << " " << sNorm[trackIndexOffset+k] << endl;
mas01cr@60 2326 }
mas01mc@17 2327 // Gather chi^2 statistics
mas01mc@17 2328 if(thisDist<minSample)
mas01mc@17 2329 minSample=thisDist;
mas01mc@17 2330 else if(thisDist>maxSample)
mas01mc@17 2331 maxSample=thisDist;
mas01mc@17 2332 if(thisDist>1e-9){
mas01mc@17 2333 sampleCount++;
mas01mc@17 2334 sampleSum+=thisDist;
mas01mc@17 2335 logSampleSum+=log(thisDist);
mas01mc@17 2336 }
mas01mc@17 2337
mas01cr@125 2338 // diffL2 = fabs(qnPtr[j] - sNorm[trackIndexOffset+k]);
mas01mc@17 2339 // Power test
mas01mc@17 2340 if(!USE_THRESH ||
mas01mc@17 2341 // Threshold on mean L2 of Q and S sequences
mas01cr@125 2342 (USE_THRESH && qnPtr[j]>SILENCE_THRESH && sNorm[trackIndexOffset+k]>SILENCE_THRESH &&
mas01mc@17 2343 // Are both query and target windows above mean energy?
mas01cr@125 2344 (qnPtr[j]>qMeanL2*.25 && sNorm[trackIndexOffset+k]>sMeanL2[track]*.25))) // && diffL2 < DIFF_THRESH )))
mas01mc@17 2345 thisDist=thisDist; // Computed above
mas01mc@17 2346 else
mas01mc@17 2347 thisDist=1000000.0;
mas01mc@17 2348 if(thisDist>=0 && thisDist<=radius){
mas01mc@17 2349 distances[0]++; // increment count
mas01mc@18 2350 break; // only need one track point per query point
mas01mc@17 2351 }
mas01mc@17 2352 }
mas01mc@17 2353 // How many points were below threshold ?
mas01mc@17 2354 thisDist=distances[0];
mas01mc@17 2355
mas01mc@17 2356 // Let's see the distances then...
mas01cr@60 2357 if(verbosity>3) {
mas01mc@18 2358 cerr << fileTable+track*O2_FILETABLESIZE << " " << thisDist << endl;
mas01cr@60 2359 }
mas01mc@17 2360
mas01mc@18 2361 // All the track stuff goes here
mas01mc@18 2362 n=trackNN;
mas01mc@17 2363 while(n--){
mas01mc@18 2364 if(thisDist>trackDistances[n]){
mas01mc@18 2365 if((n==0 || thisDist<=trackDistances[n-1])){
mas01mc@17 2366 // Copy all values above up the queue
mas01mc@18 2367 for( l=trackNN-1 ; l > n ; l--){
mas01mc@18 2368 trackDistances[l]=trackDistances[l-1];
mas01mc@18 2369 trackQIndexes[l]=trackQIndexes[l-1];
mas01mc@18 2370 trackSIndexes[l]=trackSIndexes[l-1];
mas01mc@18 2371 trackIDs[l]=trackIDs[l-1];
mas01mc@17 2372 }
mas01mc@18 2373 trackDistances[n]=thisDist;
mas01mc@18 2374 trackQIndexes[n]=qIndexes[0];
mas01mc@18 2375 trackSIndexes[n]=sIndexes[0];
mas01mc@18 2376 successfulTracks++;
mas01mc@18 2377 trackIDs[n]=track;
mas01mc@17 2378 break;
mas01mc@17 2379 }
mas01mc@17 2380 }
mas01mc@17 2381 else
mas01mc@17 2382 break;
mas01mc@17 2383 }
mas01mc@17 2384 } // Duration match
mas01mc@17 2385
mas01mc@18 2386 // Clean up current track
mas01mc@17 2387 if(D!=NULL){
mas01mc@17 2388 for(j=0; j<numVectors; j++)
mas01mc@17 2389 delete[] D[j];
mas01mc@17 2390 }
mas01mc@17 2391
mas01mc@17 2392 if(DD!=NULL){
mas01mc@17 2393 for(j=0; j<numVectors; j++)
mas01mc@17 2394 delete[] DD[j];
mas01mc@17 2395 }
mas01mc@17 2396 }
mas01mc@18 2397 // per-track reset array values
mas01mc@17 2398 for(unsigned k=0; k<pointNN; k++){
mas01mc@17 2399 distances[k]=0.0;
mas01mc@17 2400 qIndexes[k]=~0;
mas01mc@17 2401 sIndexes[k]=~0;
mas01mc@17 2402 }
mas01mc@17 2403 }
mas01mc@17 2404
mas01mc@17 2405 gettimeofday(&tv2,NULL);
mas01cr@60 2406 if(verbosity>1) {
mas01mc@18 2407 cerr << endl << "processed tracks :" << processedTracks << " matched tracks: " << successfulTracks << " elapsed time:"
mas01mc@17 2408 << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01mc@17 2409 cerr << "sampleCount: " << sampleCount << " sampleSum: " << sampleSum << " logSampleSum: " << logSampleSum
mas01mc@17 2410 << " minSample: " << minSample << " maxSample: " << maxSample << endl;
mas01mc@17 2411 }
mas01mc@17 2412
mas01cr@133 2413 if(adbQueryResponse==0){
mas01cr@60 2414 if(verbosity>1) {
mas01mc@17 2415 cerr<<endl;
mas01cr@60 2416 }
mas01mc@17 2417 // Output answer
mas01mc@17 2418 // Loop over nearest neighbours
mas01mc@18 2419 for(k=0; k < min(trackNN,successfulTracks); k++)
mas01mc@18 2420 cout << fileTable+trackIDs[k]*O2_FILETABLESIZE << " " << trackDistances[k] << endl;
mas01mc@17 2421 }
mas01mc@17 2422 else{ // Process Web Services Query
mas01mc@18 2423 int listLen = min(trackNN, processedTracks);
mas01cr@133 2424 adbQueryResponse->result.__sizeRlist=listLen;
mas01cr@133 2425 adbQueryResponse->result.__sizeDist=listLen;
mas01cr@133 2426 adbQueryResponse->result.__sizeQpos=listLen;
mas01cr@133 2427 adbQueryResponse->result.__sizeSpos=listLen;
mas01cr@133 2428 adbQueryResponse->result.Rlist= new char*[listLen];
mas01cr@133 2429 adbQueryResponse->result.Dist = new double[listLen];
mas01cr@133 2430 adbQueryResponse->result.Qpos = new unsigned int[listLen];
mas01cr@133 2431 adbQueryResponse->result.Spos = new unsigned int[listLen];
mas01cr@133 2432 for(k=0; k<(unsigned)adbQueryResponse->result.__sizeRlist; k++){
mas01cr@133 2433 adbQueryResponse->result.Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@133 2434 adbQueryResponse->result.Dist[k]=trackDistances[k];
mas01cr@133 2435 adbQueryResponse->result.Qpos[k]=trackQIndexes[k];
mas01cr@133 2436 adbQueryResponse->result.Spos[k]=trackSIndexes[k];
mas01cr@133 2437 sprintf(adbQueryResponse->result.Rlist[k], "%s", fileTable+trackIDs[k]*O2_FILETABLESIZE);
mas01mc@17 2438 }
mas01mc@17 2439 }
mas01mc@17 2440
mas01mc@17 2441 // Clean up
mas01mc@18 2442 if(trackOffsetTable)
mas01mc@18 2443 delete[] trackOffsetTable;
mas01mc@17 2444 if(queryCopy)
mas01mc@17 2445 delete[] queryCopy;
mas01cr@121 2446 if(qNorm)
mas01cr@121 2447 delete[] qNorm;
mas01cr@121 2448 if(sNorm)
mas01cr@121 2449 delete[] sNorm;
mas01cr@121 2450 if(sMeanL2)
mas01cr@121 2451 delete[] sMeanL2;
mas01mc@17 2452 if(D)
mas01mc@17 2453 delete[] D;
mas01mc@17 2454 if(DD)
mas01mc@17 2455 delete[] DD;
mas01mc@17 2456 if(timesdata)
mas01mc@17 2457 delete[] timesdata;
mas01mc@17 2458 if(meanDBdur)
mas01mc@17 2459 delete[] meanDBdur;
mas01mc@17 2460
mas01mc@17 2461
mas01mc@17 2462 }
mas01mc@17 2463
mas01cr@0 2464 // Unit norm block of features
mas01cr@0 2465 void audioDB::unitNorm(double* X, unsigned dim, unsigned n, double* qNorm){
mas01cr@0 2466 unsigned d;
mas01cr@59 2467 double L2, *p;
mas01cr@60 2468 if(verbosity>2) {
mas01cr@0 2469 cerr << "norming " << n << " vectors...";cerr.flush();
mas01cr@60 2470 }
mas01cr@0 2471 while(n--){
mas01cr@0 2472 p=X;
mas01cr@0 2473 L2=0.0;
mas01cr@0 2474 d=dim;
mas01cr@0 2475 while(d--){
mas01cr@0 2476 L2+=*p**p;
mas01cr@0 2477 p++;
mas01cr@0 2478 }
mas01mc@17 2479 /* L2=sqrt(L2);*/
mas01cr@0 2480 if(qNorm)
mas01cr@0 2481 *qNorm++=L2;
mas01mc@17 2482 /*
mas01cr@0 2483 oneOverL2 = 1.0/L2;
mas01cr@0 2484 d=dim;
mas01cr@0 2485 while(d--){
mas01cr@0 2486 *X*=oneOverL2;
mas01cr@0 2487 X++;
mas01mc@17 2488 */
mas01mc@17 2489 X+=dim;
mas01cr@0 2490 }
mas01cr@60 2491 if(verbosity>2) {
mas01cr@0 2492 cerr << "done..." << endl;
mas01cr@60 2493 }
mas01cr@0 2494 }
mas01cr@0 2495
mas01cr@0 2496 // Unit norm block of features
mas01cr@0 2497 void audioDB::unitNormAndInsertL2(double* X, unsigned dim, unsigned n, unsigned append=0){
mas01cr@0 2498 unsigned d;
mas01cr@59 2499 double *p;
mas01cr@0 2500 unsigned nn = n;
mas01cr@0 2501
mas01cr@0 2502 assert(l2normTable);
mas01cr@0 2503
mas01cr@0 2504 if( !append && (dbH->flags & O2_FLAG_L2NORM) )
mas01cr@0 2505 error("Database is already L2 normed", "automatic norm on insert is enabled");
mas01cr@0 2506
mas01cr@60 2507 if(verbosity>2) {
mas01cr@0 2508 cerr << "norming " << n << " vectors...";cerr.flush();
mas01cr@60 2509 }
mas01cr@0 2510
mas01cr@0 2511 double* l2buf = new double[n];
mas01cr@0 2512 double* l2ptr = l2buf;
mas01cr@0 2513 assert(l2buf);
mas01cr@0 2514 assert(X);
mas01cr@0 2515
mas01cr@0 2516 while(nn--){
mas01cr@0 2517 p=X;
mas01cr@0 2518 *l2ptr=0.0;
mas01cr@0 2519 d=dim;
mas01cr@0 2520 while(d--){
mas01cr@0 2521 *l2ptr+=*p**p;
mas01cr@0 2522 p++;
mas01cr@0 2523 }
mas01mc@17 2524 l2ptr++;
mas01mc@17 2525 /*
mas01mc@17 2526 oneOverL2 = 1.0/(*l2ptr++);
mas01mc@17 2527 d=dim;
mas01mc@17 2528 while(d--){
mas01cr@0 2529 *X*=oneOverL2;
mas01cr@0 2530 X++;
mas01mc@17 2531 }
mas01mc@17 2532 */
mas01mc@17 2533 X+=dim;
mas01cr@0 2534 }
mas01cr@0 2535 unsigned offset;
mas01cr@84 2536 if(append) {
mas01cr@84 2537 // FIXME: a hack, a very palpable hack: the vectors have already
mas01cr@84 2538 // been inserted, and dbH->length has already been updated. We
mas01cr@84 2539 // need to subtract off again the number of vectors that we've
mas01cr@84 2540 // inserted this time...
mas01cr@84 2541 offset=(dbH->length/(dbH->dim*sizeof(double)))-n; // number of vectors
mas01cr@84 2542 } else {
mas01cr@0 2543 offset=0;
mas01cr@84 2544 }
mas01cr@0 2545 memcpy(l2normTable+offset, l2buf, n*sizeof(double));
mas01cr@0 2546 if(l2buf)
mas01mc@17 2547 delete[] l2buf;
mas01cr@60 2548 if(verbosity>2) {
mas01cr@0 2549 cerr << "done..." << endl;
mas01cr@60 2550 }
mas01cr@0 2551 }
mas01cr@0 2552
mas01cr@0 2553
mas01cr@0 2554 // Start an audioDB server on the host
mas01cr@0 2555 void audioDB::startServer(){
mas01cr@0 2556 struct soap soap;
mas01cr@0 2557 int m, s; // master and slave sockets
mas01cr@0 2558 soap_init(&soap);
mas01cr@92 2559 // FIXME: largely this use of SO_REUSEADDR is to make writing (and
mas01cr@92 2560 // running) test cases more convenient, so that multiple test runs
mas01cr@92 2561 // in close succession don't fail because of a bin() error.
mas01cr@92 2562 // Investigate whether there are any potential drawbacks in this,
mas01cr@92 2563 // and also whether there's a better way to write the tests. --
mas01cr@92 2564 // CSR, 2007-10-03
mas01cr@92 2565 soap.bind_flags |= SO_REUSEADDR;
mas01cr@0 2566 m = soap_bind(&soap, NULL, port, 100);
mas01cr@0 2567 if (m < 0)
mas01cr@0 2568 soap_print_fault(&soap, stderr);
mas01cr@0 2569 else
mas01cr@0 2570 {
mas01cr@0 2571 fprintf(stderr, "Socket connection successful: master socket = %d\n", m);
mas01cr@0 2572 for (int i = 1; ; i++)
mas01cr@0 2573 {
mas01cr@0 2574 s = soap_accept(&soap);
mas01cr@0 2575 if (s < 0)
mas01cr@0 2576 {
mas01cr@0 2577 soap_print_fault(&soap, stderr);
mas01cr@0 2578 break;
mas01cr@0 2579 }
mas01cr@75 2580 fprintf(stderr, "%d: accepted connection from IP=%lu.%lu.%lu.%lu socket=%d\n", i,
mas01cr@0 2581 (soap.ip >> 24)&0xFF, (soap.ip >> 16)&0xFF, (soap.ip >> 8)&0xFF, soap.ip&0xFF, s);
mas01cr@0 2582 if (soap_serve(&soap) != SOAP_OK) // process RPC request
mas01cr@0 2583 soap_print_fault(&soap, stderr); // print error
mas01cr@0 2584 fprintf(stderr, "request served\n");
mas01cr@0 2585 soap_destroy(&soap); // clean up class instances
mas01cr@0 2586 soap_end(&soap); // clean up everything and close socket
mas01cr@0 2587 }
mas01cr@0 2588 }
mas01cr@0 2589 soap_done(&soap); // close master socket and detach environment
mas01cr@0 2590 }
mas01cr@0 2591
mas01cr@0 2592
mas01cr@0 2593 // web services
mas01cr@0 2594
mas01cr@0 2595 // SERVER SIDE
mas01cr@133 2596 int adb__status(struct soap* soap, xsd__string dbName, adb__statusResponse &adbStatusResponse){
mas01cr@21 2597 char* const argv[]={"audioDB",COM_STATUS,"-d",dbName};
mas01cr@21 2598 const unsigned argc = 4;
mas01cr@77 2599 try {
mas01cr@133 2600 audioDB(argc, argv, &adbStatusResponse);
mas01cr@77 2601 return SOAP_OK;
mas01cr@77 2602 } catch(char *err) {
mas01cr@77 2603 soap_receiver_fault(soap, err, "");
mas01cr@77 2604 return SOAP_FAULT;
mas01cr@77 2605 }
mas01cr@0 2606 }
mas01cr@0 2607
mas01cr@0 2608 // Literal translation of command line to web service
mas01cr@0 2609
mas01cr@133 2610 int adb__query(struct soap* soap, xsd__string dbName, xsd__string qKey, xsd__string keyList, xsd__string timesFileName, xsd__int qType, xsd__int qPos, xsd__int pointNN, xsd__int trackNN, xsd__int seqLen, adb__queryResponse &adbQueryResponse){
mas01cr@0 2611 char queryType[256];
mas01cr@0 2612 for(int k=0; k<256; k++)
mas01cr@0 2613 queryType[k]='\0';
mas01cr@105 2614 if(qType == O2_POINT_QUERY)
mas01cr@0 2615 strncpy(queryType, "point", strlen("point"));
mas01cr@105 2616 else if (qType == O2_SEQUENCE_QUERY)
mas01cr@0 2617 strncpy(queryType, "sequence", strlen("sequence"));
mas01cr@105 2618 else if(qType == O2_TRACK_QUERY)
mas01mc@18 2619 strncpy(queryType,"track", strlen("track"));
mas01cr@0 2620 else
mas01cr@0 2621 strncpy(queryType, "", strlen(""));
mas01cr@0 2622
mas01cr@0 2623 if(pointNN==0)
mas01cr@0 2624 pointNN=10;
mas01mc@18 2625 if(trackNN==0)
mas01mc@18 2626 trackNN=10;
mas01cr@0 2627 if(seqLen==0)
mas01cr@0 2628 seqLen=16;
mas01cr@0 2629
mas01cr@0 2630 char qPosStr[256];
mas01cr@0 2631 sprintf(qPosStr, "%d", qPos);
mas01cr@0 2632 char pointNNStr[256];
mas01cr@0 2633 sprintf(pointNNStr,"%d",pointNN);
mas01mc@18 2634 char trackNNStr[256];
mas01mc@18 2635 sprintf(trackNNStr,"%d",trackNN);
mas01cr@0 2636 char seqLenStr[256];
mas01cr@0 2637 sprintf(seqLenStr,"%d",seqLen);
mas01cr@0 2638
mas01cr@0 2639 const char* argv[] ={
mas01cr@0 2640 "./audioDB",
mas01cr@0 2641 COM_QUERY,
mas01cr@0 2642 queryType, // Need to pass a parameter
mas01cr@0 2643 COM_DATABASE,
mas01cr@166 2644 ENSURE_STRING(dbName),
mas01cr@0 2645 COM_FEATURES,
mas01cr@166 2646 ENSURE_STRING(qKey),
mas01cr@0 2647 COM_KEYLIST,
mas01cr@166 2648 ENSURE_STRING(keyList),
mas01cr@0 2649 COM_TIMES,
mas01cr@166 2650 ENSURE_STRING(timesFileName),
mas01cr@0 2651 COM_QPOINT,
mas01cr@0 2652 qPosStr,
mas01cr@0 2653 COM_POINTNN,
mas01cr@0 2654 pointNNStr,
mas01mc@18 2655 COM_TRACKNN,
mas01mc@18 2656 trackNNStr, // Need to pass a parameter
mas01cr@0 2657 COM_SEQLEN,
mas01cr@0 2658 seqLenStr
mas01cr@0 2659 };
mas01cr@0 2660
mas01cr@0 2661 const unsigned argc = 19;
mas01cr@79 2662 try {
mas01cr@133 2663 audioDB(argc, (char* const*)argv, &adbQueryResponse);
mas01cr@79 2664 return SOAP_OK;
mas01cr@79 2665 } catch (char *err) {
mas01cr@79 2666 soap_receiver_fault(soap, err, "");
mas01cr@79 2667 return SOAP_FAULT;
mas01cr@79 2668 }
mas01cr@0 2669 }
mas01cr@0 2670
mas01cr@0 2671 int main(const unsigned argc, char* const argv[]){
mas01cr@0 2672 audioDB(argc, argv);
mas01cr@0 2673 }