annotate audioDB.cpp @ 178:7bbe5d48a7ef no-big-mmap

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