annotate audioDB.cpp @ 0:9eab3e0f3068

Initial import
author mas01cr
date Fri, 20 Jul 2007 15:51:39 +0000
parents
children 12be3560ff0f 69eb22e09772
rev   line source
mas01cr@0 1 /* audioDB.cpp
mas01cr@0 2
mas01cr@0 3 audioDB version 1.0
mas01cr@0 4
mas01cr@0 5 A feature vector database management system for content-based retrieval.
mas01cr@0 6
mas01cr@0 7 Usage: audioDB [OPTIONS]...
mas01cr@0 8
mas01cr@0 9 --full-help Print help, including hidden options, and exit
mas01cr@0 10 -V, --version Print version and exit
mas01cr@0 11 -H, --help print help on audioDB usage and exit.
mas01cr@0 12 -v, --verbosity=detail level of detail of operational information.
mas01cr@0 13 (default=`1')
mas01cr@0 14
mas01cr@0 15 Database Setup:
mas01cr@0 16 All database operations require a database argument.
mas01cr@0 17
mas01cr@0 18 Database commands are UPPER CASE. Command options are lower case.
mas01cr@0 19
mas01cr@0 20 -d, --database=filename database file required by Database commands.
mas01cr@0 21 -N, --NEW make a new (initially empty) database.
mas01cr@0 22 -S, --STATUS output database information to stdout.
mas01cr@0 23 -D, --DUMP output all entries: index key size.
mas01cr@0 24 -L, --L2NORM unit norm vectors and norm all future inserts.
mas01cr@0 25
mas01cr@0 26 Database Insertion:
mas01cr@0 27 The following commands insert feature files, with optional keys and
mas01cr@0 28 timestamps.
mas01cr@0 29
mas01cr@0 30 -I, --INSERT add feature vectors to an existing database.
mas01cr@0 31 -U, --UPDATE replace inserted vectors associated with key
mas01cr@0 32 with new input vectors.
mas01cr@0 33 -f, --features=filename binary series of vectors file {int sz:ieee
mas01cr@0 34 double[][sz]:eof}.
mas01cr@0 35 -t, --times=filename list of time points (ascii) for feature vectors.
mas01cr@0 36 -k, --key=identifier unique identifier associated with features.
mas01cr@0 37
mas01cr@0 38 -B, --BATCHINSERT add feature vectors named in a --featureList
mas01cr@0 39 file (with optional keys in a --keyList file)
mas01cr@0 40 to the named database.
mas01cr@0 41 -F, --featureList=filename text file containing list of binary feature
mas01cr@0 42 vector files to process
mas01cr@0 43 -T, --timesList=filename text file containing list of ascii --times for
mas01cr@0 44 each --features file in --featureList.
mas01cr@0 45 -K, --keyList=filename text file containing list of unique identifiers
mas01cr@0 46 associated with --features.
mas01cr@0 47
mas01cr@0 48 Database Search:
mas01cr@0 49 Thse commands control the retrieval behaviour.
mas01cr@0 50
mas01cr@0 51 -Q, --QUERY=searchtype content-based search on --database using
mas01cr@0 52 --features as a query. Optionally restrict the
mas01cr@0 53 search to those segments identified in a
mas01cr@0 54 --keyList. (possible values="point",
mas01cr@0 55 "segment", "sequence")
mas01cr@0 56 -p, --qpoint=position ordinal position of query start point in
mas01cr@0 57 --features file. (default=`0')
mas01cr@0 58 -e, --exhaustive exhaustive search: iterate through all query
mas01cr@0 59 vectors in search. Overrides --qpoint.
mas01cr@0 60 (default=off)
mas01cr@0 61 -n, --pointnn=numpoints number of point nearest neighbours to use in
mas01cr@0 62 retrieval. (default=`10')
mas01cr@0 63 -R, --radius=DOUBLE radius search, returns all
mas01cr@0 64 points/segments/sequences inside given radius.
mas01cr@0 65 (default=`1.0')
mas01cr@0 66 -x, --expandfactor=DOUBLE time compress/expand factor of result length to
mas01cr@0 67 query length [1.0 .. 100.0]. (default=`1.1')
mas01cr@0 68 -o, --rotate rotate query vectors for rotationally invariant
mas01cr@0 69 search. (default=off)
mas01cr@0 70 -r, --resultlength=length maximum length of the result list.
mas01cr@0 71 (default=`10')
mas01cr@0 72 -l, --sequencelength=length length of sequences for sequence search.
mas01cr@0 73 (default=`16')
mas01cr@0 74 -h, --sequencehop=hop hop size of sequence window for sequence search.
mas01cr@0 75 (default=`1')
mas01cr@0 76
mas01cr@0 77 Web Services:
mas01cr@0 78 These commands enable the database process to establish a connection via the
mas01cr@0 79 internet and operate as separate client and server processes.
mas01cr@0 80
mas01cr@0 81 -s, --SERVER=port run as standalone web service on named port.
mas01cr@0 82 (default=`80011')
mas01cr@0 83 -c, --client=hostname:port run as a client using named host service.
mas01cr@0 84
mas01cr@0 85 Copyright (C) 2007 Michael Casey, Goldsmiths, University of London
mas01cr@0 86
mas01cr@0 87 outputs:
mas01cr@0 88
mas01cr@0 89 key1 distance1 qpos1 spos1
mas01cr@0 90 key2 distance2 qpos2 spos2
mas01cr@0 91 ...
mas01cr@0 92 keyN distanceN qposN sposN
mas01cr@0 93
mas01cr@0 94 */
mas01cr@0 95
mas01cr@0 96 #include "audioDB.h"
mas01cr@0 97
mas01cr@0 98 #define O2_DEBUG
mas01cr@0 99
mas01cr@0 100 void audioDB::error(const char* a, const char* b){
mas01cr@0 101 cerr << a << ":" << b << endl;
mas01cr@0 102 exit(1);
mas01cr@0 103 }
mas01cr@0 104
mas01cr@0 105 audioDB::audioDB(const unsigned argc, char* const argv[], adb__queryResult *adbQueryResult):
mas01cr@0 106 dim(0),
mas01cr@0 107 dbName(0),
mas01cr@0 108 inFile(0),
mas01cr@0 109 key(0),
mas01cr@0 110 segFile(0),
mas01cr@0 111 segFileName(0),
mas01cr@0 112 timesFile(0),
mas01cr@0 113 timesFileName(0),
mas01cr@0 114 usingTimes(0),
mas01cr@0 115 command(0),
mas01cr@0 116 dbfid(0),
mas01cr@0 117 db(0),
mas01cr@0 118 dbH(0),
mas01cr@0 119 infid(0),
mas01cr@0 120 indata(0),
mas01cr@0 121 queryType(O2_FLAG_POINT_QUERY),
mas01cr@0 122 verbosity(1),
mas01cr@0 123 pointNN(O2_DEFAULT_POINTNN),
mas01cr@0 124 segNN(O2_DEFAULT_SEGNN),
mas01cr@0 125 segTable(0),
mas01cr@0 126 fileTable(0),
mas01cr@0 127 dataBuf(0),
mas01cr@0 128 l2normTable(0),
mas01cr@0 129 timesTable(0),
mas01cr@0 130 qNorm(0),
mas01cr@0 131 sequenceLength(16),
mas01cr@0 132 sequenceHop(1),
mas01cr@0 133 queryPoint(0),
mas01cr@0 134 usingQueryPoint(0),
mas01cr@0 135 isClient(0),
mas01cr@0 136 isServer(0),
mas01cr@0 137 port(0),
mas01cr@0 138 timesTol(0.1){
mas01cr@0 139
mas01cr@0 140 if(processArgs(argc, argv)<0){
mas01cr@0 141 printf("No command found.\n");
mas01cr@0 142 cmdline_parser_print_version ();
mas01cr@0 143 if (strlen(gengetopt_args_info_purpose) > 0)
mas01cr@0 144 printf("%s\n", gengetopt_args_info_purpose);
mas01cr@0 145 printf("%s\n", gengetopt_args_info_usage);
mas01cr@0 146 printf("%s\n", gengetopt_args_info_help[1]);
mas01cr@0 147 printf("%s\n", gengetopt_args_info_help[2]);
mas01cr@0 148 printf("%s\n", gengetopt_args_info_help[0]);
mas01cr@0 149 exit(1);
mas01cr@0 150 }
mas01cr@0 151
mas01cr@0 152 if(O2_ACTION(COM_SERVER))
mas01cr@0 153 startServer();
mas01cr@0 154
mas01cr@0 155 else if(O2_ACTION(COM_CREATE))
mas01cr@0 156 create(dbName);
mas01cr@0 157
mas01cr@0 158 else if(O2_ACTION(COM_INSERT))
mas01cr@0 159 insert(dbName, inFile);
mas01cr@0 160
mas01cr@0 161 else if(O2_ACTION(COM_BATCHINSERT))
mas01cr@0 162 batchinsert(dbName, inFile);
mas01cr@0 163
mas01cr@0 164 else if(O2_ACTION(COM_QUERY))
mas01cr@0 165 if(isClient)
mas01cr@0 166 ws_query(dbName, inFile, (char*)hostport);
mas01cr@0 167 else
mas01cr@0 168 query(dbName, inFile, adbQueryResult);
mas01cr@0 169
mas01cr@0 170 else if(O2_ACTION(COM_STATUS))
mas01cr@0 171 if(isClient)
mas01cr@0 172 ws_status(dbName,(char*)hostport);
mas01cr@0 173 else
mas01cr@0 174 status(dbName);
mas01cr@0 175
mas01cr@0 176 else if(O2_ACTION(COM_L2NORM))
mas01cr@0 177 l2norm(dbName);
mas01cr@0 178
mas01cr@0 179 else if(O2_ACTION(COM_DUMP))
mas01cr@0 180 dump(dbName);
mas01cr@0 181
mas01cr@0 182 else
mas01cr@0 183 error("Unrecognized command",command);
mas01cr@0 184 }
mas01cr@0 185
mas01cr@0 186 audioDB::~audioDB(){
mas01cr@0 187 // Clean up
mas01cr@0 188 if(indata)
mas01cr@0 189 munmap(indata,statbuf.st_size);
mas01cr@0 190 if(db)
mas01cr@0 191 munmap(db,O2_DEFAULTDBSIZE);
mas01cr@0 192 if(dbfid>0)
mas01cr@0 193 close(dbfid);
mas01cr@0 194 if(infid>0)
mas01cr@0 195 close(infid);
mas01cr@0 196 if(dbH)
mas01cr@0 197 delete dbH;
mas01cr@0 198 }
mas01cr@0 199
mas01cr@0 200 int audioDB::processArgs(const unsigned argc, char* const argv[]){
mas01cr@0 201
mas01cr@0 202 if(argc<2){
mas01cr@0 203 cmdline_parser_print_version ();
mas01cr@0 204 if (strlen(gengetopt_args_info_purpose) > 0)
mas01cr@0 205 printf("%s\n", gengetopt_args_info_purpose);
mas01cr@0 206 printf("%s\n", gengetopt_args_info_usage);
mas01cr@0 207 printf("%s\n", gengetopt_args_info_help[1]);
mas01cr@0 208 printf("%s\n", gengetopt_args_info_help[2]);
mas01cr@0 209 printf("%s\n", gengetopt_args_info_help[0]);
mas01cr@0 210 exit(0);
mas01cr@0 211 }
mas01cr@0 212
mas01cr@0 213 if (cmdline_parser (argc, argv, &args_info) != 0)
mas01cr@0 214 exit(1) ;
mas01cr@0 215
mas01cr@0 216 if(args_info.help_given){
mas01cr@0 217 cmdline_parser_print_help();
mas01cr@0 218 exit(0);
mas01cr@0 219 }
mas01cr@0 220
mas01cr@0 221 if(args_info.verbosity_given){
mas01cr@0 222 verbosity=args_info.verbosity_arg;
mas01cr@0 223 if(verbosity<0 || verbosity>10){
mas01cr@0 224 cerr << "Warning: verbosity out of range, setting to 1" << endl;
mas01cr@0 225 verbosity=1;
mas01cr@0 226 }
mas01cr@0 227 }
mas01cr@0 228
mas01cr@0 229 if(args_info.SERVER_given){
mas01cr@0 230 command=COM_SERVER;
mas01cr@0 231 port=args_info.SERVER_arg;
mas01cr@0 232 if(port<100 || port > 100000)
mas01cr@0 233 error("port out of range");
mas01cr@0 234 isServer=1;
mas01cr@0 235 return 0;
mas01cr@0 236 }
mas01cr@0 237
mas01cr@0 238 // No return on client command, find database command
mas01cr@0 239 if(args_info.client_given){
mas01cr@0 240 command=COM_CLIENT;
mas01cr@0 241 hostport=args_info.client_arg;
mas01cr@0 242 isClient=1;
mas01cr@0 243 }
mas01cr@0 244
mas01cr@0 245 if(args_info.NEW_given){
mas01cr@0 246 command=COM_CREATE;
mas01cr@0 247 dbName=args_info.database_arg;
mas01cr@0 248 return 0;
mas01cr@0 249 }
mas01cr@0 250
mas01cr@0 251 if(args_info.STATUS_given){
mas01cr@0 252 command=COM_STATUS;
mas01cr@0 253 dbName=args_info.database_arg;
mas01cr@0 254 return 0;
mas01cr@0 255 }
mas01cr@0 256
mas01cr@0 257 if(args_info.DUMP_given){
mas01cr@0 258 command=COM_DUMP;
mas01cr@0 259 dbName=args_info.database_arg;
mas01cr@0 260 return 0;
mas01cr@0 261 }
mas01cr@0 262
mas01cr@0 263 if(args_info.L2NORM_given){
mas01cr@0 264 command=COM_L2NORM;
mas01cr@0 265 dbName=args_info.database_arg;
mas01cr@0 266 return 0;
mas01cr@0 267 }
mas01cr@0 268
mas01cr@0 269 if(args_info.INSERT_given){
mas01cr@0 270 command=COM_INSERT;
mas01cr@0 271 dbName=args_info.database_arg;
mas01cr@0 272 inFile=args_info.features_arg;
mas01cr@0 273 if(args_info.key_given)
mas01cr@0 274 key=args_info.key_arg;
mas01cr@0 275 if(args_info.times_given){
mas01cr@0 276 timesFileName=args_info.times_arg;
mas01cr@0 277 if(strlen(timesFileName)>0){
mas01cr@0 278 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@0 279 error("Could not open times file for reading", timesFileName);
mas01cr@0 280 usingTimes=1;
mas01cr@0 281 }
mas01cr@0 282 }
mas01cr@0 283 return 0;
mas01cr@0 284 }
mas01cr@0 285
mas01cr@0 286 if(args_info.BATCHINSERT_given){
mas01cr@0 287 command=COM_BATCHINSERT;
mas01cr@0 288 dbName=args_info.database_arg;
mas01cr@0 289 inFile=args_info.featureList_arg;
mas01cr@0 290 if(args_info.keyList_given)
mas01cr@0 291 key=args_info.keyList_arg; // INCONSISTENT NO CHECK
mas01cr@0 292
mas01cr@0 293 /* TO DO: REPLACE WITH
mas01cr@0 294 if(args_info.keyList_given){
mas01cr@0 295 segFileName=args_info.keyList_arg;
mas01cr@0 296 if(strlen(segFileName)>0 && !(segFile = new ifstream(segFileName,ios::in)))
mas01cr@0 297 error("Could not open keyList file for reading",segFileName);
mas01cr@0 298 }
mas01cr@0 299 AND UPDATE BATCHINSERT()
mas01cr@0 300 */
mas01cr@0 301
mas01cr@0 302 if(args_info.timesList_given){
mas01cr@0 303 timesFileName=args_info.timesList_arg;
mas01cr@0 304 if(strlen(timesFileName)>0){
mas01cr@0 305 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@0 306 error("Could not open timesList file for reading", timesFileName);
mas01cr@0 307 usingTimes=1;
mas01cr@0 308 }
mas01cr@0 309 }
mas01cr@0 310 return 0;
mas01cr@0 311 }
mas01cr@0 312
mas01cr@0 313 // Query command and arguments
mas01cr@0 314 if(args_info.QUERY_given){
mas01cr@0 315 command=COM_QUERY;
mas01cr@0 316 dbName=args_info.database_arg;
mas01cr@0 317 inFile=args_info.features_arg;
mas01cr@0 318
mas01cr@0 319 if(args_info.keyList_given){
mas01cr@0 320 segFileName=args_info.keyList_arg;
mas01cr@0 321 if(strlen(segFileName)>0 && !(segFile = new ifstream(segFileName,ios::in)))
mas01cr@0 322 error("Could not open keyList file for reading",segFileName);
mas01cr@0 323 }
mas01cr@0 324
mas01cr@0 325 if(args_info.times_given){
mas01cr@0 326 timesFileName=args_info.times_arg;
mas01cr@0 327 if(strlen(timesFileName)>0){
mas01cr@0 328 if(!(timesFile = new ifstream(timesFileName,ios::in)))
mas01cr@0 329 error("Could not open times file for reading", timesFileName);
mas01cr@0 330 usingTimes=1;
mas01cr@0 331 }
mas01cr@0 332 }
mas01cr@0 333
mas01cr@0 334 // query type
mas01cr@0 335 if(strncmp(args_info.QUERY_arg, "segment", MAXSTR)==0)
mas01cr@0 336 queryType=O2_FLAG_SEG_QUERY;
mas01cr@0 337 else if(strncmp(args_info.QUERY_arg, "point", MAXSTR)==0)
mas01cr@0 338 queryType=O2_FLAG_POINT_QUERY;
mas01cr@0 339 else if(strncmp(args_info.QUERY_arg, "sequence", MAXSTR)==0)
mas01cr@0 340 queryType=O2_FLAG_SEQUENCE_QUERY;
mas01cr@0 341 else
mas01cr@0 342 error("unsupported query type",args_info.QUERY_arg);
mas01cr@0 343
mas01cr@0 344 if(!args_info.exhaustive_flag){
mas01cr@0 345 queryPoint = args_info.qpoint_arg;
mas01cr@0 346 usingQueryPoint=1;
mas01cr@0 347 if(queryPoint<0 || queryPoint >10000)
mas01cr@0 348 error("queryPoint out of range: 0 <= queryPoint <= 10000");
mas01cr@0 349 }
mas01cr@0 350
mas01cr@0 351
mas01cr@0 352 pointNN=args_info.pointnn_arg;
mas01cr@0 353 if(pointNN<1 || pointNN >1000)
mas01cr@0 354 error("pointNN out of range: 1 <= pointNN <= 1000");
mas01cr@0 355
mas01cr@0 356
mas01cr@0 357
mas01cr@0 358 segNN=args_info.resultlength_arg;
mas01cr@0 359 if(segNN<1 || segNN >1000)
mas01cr@0 360 error("resultlength out of range: 1 <= resultlength <= 1000");
mas01cr@0 361
mas01cr@0 362
mas01cr@0 363 sequenceLength=args_info.sequencelength_arg;
mas01cr@0 364 if(sequenceLength<1 || sequenceLength >1000)
mas01cr@0 365 error("seqlen out of range: 1 <= seqlen <= 1000");
mas01cr@0 366
mas01cr@0 367 sequenceHop=args_info.sequencehop_arg;
mas01cr@0 368 if(sequenceHop<1 || sequenceHop >1000)
mas01cr@0 369 error("seqhop out of range: 1 <= seqhop <= 1000");
mas01cr@0 370
mas01cr@0 371 return 0;
mas01cr@0 372 }
mas01cr@0 373 return -1; // no command found
mas01cr@0 374 }
mas01cr@0 375
mas01cr@0 376 /* Make a new database
mas01cr@0 377
mas01cr@0 378 The database consists of:
mas01cr@0 379
mas01cr@0 380 header
mas01cr@0 381 ---------------------------------------------------------------------------------
mas01cr@0 382 | magic 4 bytes| numFiles 4 bytes | dim 4 bytes | length 4 bytes |flags 4 bytes |
mas01cr@0 383 ---------------------------------------------------------------------------------
mas01cr@0 384
mas01cr@0 385
mas01cr@0 386 keyTable : list of keys of segments
mas01cr@0 387 --------------------------------------------------------------------------
mas01cr@0 388 | key 256 bytes |
mas01cr@0 389 --------------------------------------------------------------------------
mas01cr@0 390 O2_MAXFILES*02_FILENAMELENGTH
mas01cr@0 391
mas01cr@0 392 segTable : Maps implicit feature index to a feature vector matrix
mas01cr@0 393 --------------------------------------------------------------------------
mas01cr@0 394 | numVectors (4 bytes) |
mas01cr@0 395 --------------------------------------------------------------------------
mas01cr@0 396 O2_MAXFILES * 02_MEANNUMFEATURES * sizeof(INT)
mas01cr@0 397
mas01cr@0 398 featureTable
mas01cr@0 399 --------------------------------------------------------------------------
mas01cr@0 400 | v1 v2 v3 ... vd (double) |
mas01cr@0 401 --------------------------------------------------------------------------
mas01cr@0 402 O2_MAXFILES * 02_MEANNUMFEATURES * DIM * sizeof(DOUBLE)
mas01cr@0 403
mas01cr@0 404 timesTable
mas01cr@0 405 --------------------------------------------------------------------------
mas01cr@0 406 | timestamp (double) |
mas01cr@0 407 --------------------------------------------------------------------------
mas01cr@0 408 O2_MAXFILES * 02_MEANNUMFEATURES * sizeof(DOUBLE)
mas01cr@0 409
mas01cr@0 410 l2normTable
mas01cr@0 411 --------------------------------------------------------------------------
mas01cr@0 412 | nm (double) |
mas01cr@0 413 --------------------------------------------------------------------------
mas01cr@0 414 O2_MAXFILES * 02_MEANNUMFEATURES * sizeof(DOUBLE)
mas01cr@0 415
mas01cr@0 416 */
mas01cr@0 417
mas01cr@0 418 void audioDB::create(const char* dbName){
mas01cr@0 419 if ((dbfid = open (dbName, O_RDWR | O_CREAT | O_TRUNC)) < 0)
mas01cr@0 420 error("Can't open database file:", dbName);
mas01cr@0 421
mas01cr@0 422 // go to the location corresponding to the last byte
mas01cr@0 423 if (lseek (dbfid, O2_DEFAULTDBSIZE - 1, SEEK_SET) == -1)
mas01cr@0 424 error("lseek error in db file");
mas01cr@0 425
mas01cr@0 426 // write a dummy byte at the last location
mas01cr@0 427 if (write (dbfid, "", 1) != 1)
mas01cr@0 428 error("write error");
mas01cr@0 429
mas01cr@0 430 // mmap the output file
mas01cr@0 431 if(verbosity)
mas01cr@0 432 cerr << "header size:" << O2_HEADERSIZE << endl;
mas01cr@0 433 if ((db = (char*) mmap(0, O2_DEFAULTDBSIZE, PROT_READ | PROT_WRITE,
mas01cr@0 434 MAP_SHARED, dbfid, 0)) == (caddr_t) -1)
mas01cr@0 435 error("mmap error for creating database");
mas01cr@0 436
mas01cr@0 437 dbH = new dbTableHeaderT();
mas01cr@0 438 assert(dbH);
mas01cr@0 439
mas01cr@0 440 // Initialize header
mas01cr@0 441 dbH->magic=O2_MAGIC;
mas01cr@0 442 dbH->numFiles=0;
mas01cr@0 443 dbH->length=0;
mas01cr@0 444 dbH->dim=0;
mas01cr@0 445 dbH->flags=0; //O2_FLAG_L2NORM;
mas01cr@0 446
mas01cr@0 447 memcpy (db, dbH, O2_HEADERSIZE);
mas01cr@0 448 if(verbosity)
mas01cr@0 449 cerr << COM_CREATE << " " << dbName << endl;
mas01cr@0 450
mas01cr@0 451 }
mas01cr@0 452
mas01cr@0 453
mas01cr@0 454 void audioDB::drop(){
mas01cr@0 455
mas01cr@0 456
mas01cr@0 457 }
mas01cr@0 458
mas01cr@0 459 // initTables - memory map files passed as arguments
mas01cr@0 460 // Precondition: database has already been created
mas01cr@0 461 void audioDB::initTables(const char* dbName, const char* inFile=0){
mas01cr@0 462 if ((dbfid = open (dbName, O_RDWR)) < 0)
mas01cr@0 463 error("Can't open database file:", dbName);
mas01cr@0 464
mas01cr@0 465 // open the input file
mas01cr@0 466 if (inFile && (infid = open (inFile, O_RDONLY)) < 0)
mas01cr@0 467 error("can't open feature file for reading", inFile);
mas01cr@0 468
mas01cr@0 469 // find size of input file
mas01cr@0 470 if (inFile && fstat (infid,&statbuf) < 0)
mas01cr@0 471 error("fstat error finding size of input");
mas01cr@0 472
mas01cr@0 473 // Get the database header info
mas01cr@0 474 dbH = new dbTableHeaderT();
mas01cr@0 475 assert(dbH);
mas01cr@0 476
mas01cr@0 477 if(read(dbfid,(char*)dbH,sizeof(dbTableHeaderT))!=sizeof(dbTableHeaderT))
mas01cr@0 478 error("error reading db header");
mas01cr@0 479
mas01cr@0 480 fileTableOffset = O2_HEADERSIZE;
mas01cr@0 481 segTableOffset = fileTableOffset + O2_FILETABLESIZE*O2_MAXFILES;
mas01cr@0 482 dataoffset = segTableOffset + O2_SEGTABLESIZE*O2_MAXFILES;
mas01cr@0 483 l2normTableOffset = O2_DEFAULTDBSIZE - O2_MAXFILES*O2_MEANNUMVECTORS*sizeof(double);
mas01cr@0 484 timesTableOffset = l2normTableOffset - O2_MAXFILES*O2_MEANNUMVECTORS*sizeof(double);
mas01cr@0 485
mas01cr@0 486 if(dbH->magic!=O2_MAGIC){
mas01cr@0 487 cerr << "expected: " << O2_MAGIC << ", got:" << dbH->magic << endl;
mas01cr@0 488 error("database file has incorrect header",dbName);
mas01cr@0 489 }
mas01cr@0 490
mas01cr@0 491 if(inFile)
mas01cr@0 492 if(dbH->dim==0 && dbH->length==0) // empty database
mas01cr@0 493 read(infid,&dbH->dim,sizeof(unsigned)); // initialize with input dimensionality
mas01cr@0 494 else {
mas01cr@0 495 unsigned test;
mas01cr@0 496 read(infid,&test,sizeof(unsigned));
mas01cr@0 497 if(dbH->dim!=test){
mas01cr@0 498 cerr << "error: expected dimension: " << dbH->dim << ", got :" << test <<endl;
mas01cr@0 499 error("feature dimensions do not match database table dimensions");
mas01cr@0 500 }
mas01cr@0 501 }
mas01cr@0 502
mas01cr@0 503 // mmap the input file
mas01cr@0 504 if (inFile && (indata = (char*)mmap (0, statbuf.st_size, PROT_READ, MAP_SHARED, infid, 0))
mas01cr@0 505 == (caddr_t) -1)
mas01cr@0 506 error("mmap error for input");
mas01cr@0 507
mas01cr@0 508 // mmap the database file
mas01cr@0 509 if ((db = (char*) mmap(0, O2_DEFAULTDBSIZE, PROT_READ | PROT_WRITE,
mas01cr@0 510 MAP_SHARED, dbfid, 0)) == (caddr_t) -1)
mas01cr@0 511 error("mmap error for creating database");
mas01cr@0 512
mas01cr@0 513 // Make some handy tables with correct types
mas01cr@0 514 fileTable= (char*)(db+fileTableOffset);
mas01cr@0 515 segTable = (unsigned*)(db+segTableOffset);
mas01cr@0 516 dataBuf = (double*)(db+dataoffset);
mas01cr@0 517 l2normTable = (double*)(db+l2normTableOffset);
mas01cr@0 518 timesTable = (double*)(db+timesTableOffset);
mas01cr@0 519
mas01cr@0 520 }
mas01cr@0 521
mas01cr@0 522 void audioDB::insert(const char* dbName, const char* inFile){
mas01cr@0 523
mas01cr@0 524 initTables(dbName, inFile);
mas01cr@0 525
mas01cr@0 526 if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 527 error("Must use timestamps with timestamped database","use --times");
mas01cr@0 528
mas01cr@0 529 // Check that there is room for at least 1 more file
mas01cr@0 530 if((char*)timesTable<((char*)dataBuf+dbH->length+statbuf.st_size-sizeof(int)))
mas01cr@0 531 error("No more room in database","insert failed: reason database is full.");
mas01cr@0 532
mas01cr@0 533 if(!key)
mas01cr@0 534 key=inFile;
mas01cr@0 535 // Linear scan of filenames check for pre-existing feature
mas01cr@0 536 unsigned alreadyInserted=0;
mas01cr@0 537 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 538 if(strncmp(fileTable + k*O2_FILETABLESIZE, key, strlen(key))==0){
mas01cr@0 539 alreadyInserted=1;
mas01cr@0 540 break;
mas01cr@0 541 }
mas01cr@0 542
mas01cr@0 543 if(alreadyInserted){
mas01cr@0 544 if(verbosity)
mas01cr@0 545 cerr << "Warning: key already exists in database, ignoring: " <<inFile << endl;
mas01cr@0 546 return;
mas01cr@0 547 }
mas01cr@0 548
mas01cr@0 549 // Make a segment index table of features to file indexes
mas01cr@0 550 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 551 if(!numVectors){
mas01cr@0 552 if(verbosity)
mas01cr@0 553 cerr << "Warning: ignoring zero-length feature vector file:" << key << endl;
mas01cr@0 554 // CLEAN UP
mas01cr@0 555 munmap(indata,statbuf.st_size);
mas01cr@0 556 munmap(db,O2_DEFAULTDBSIZE);
mas01cr@0 557 close(infid);
mas01cr@0 558 return;
mas01cr@0 559 }
mas01cr@0 560
mas01cr@0 561 strncpy(fileTable + dbH->numFiles*O2_FILETABLESIZE, key, strlen(key));
mas01cr@0 562
mas01cr@0 563 unsigned insertoffset = dbH->length;// Store current state
mas01cr@0 564
mas01cr@0 565 // Check times status and insert times from file
mas01cr@0 566 unsigned timesoffset=insertoffset/(dbH->dim*sizeof(double));
mas01cr@0 567 double* timesdata=timesTable+timesoffset;
mas01cr@0 568 assert(timesdata+numVectors<l2normTable);
mas01cr@0 569 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 570
mas01cr@0 571 // Increment file count
mas01cr@0 572 dbH->numFiles++;
mas01cr@0 573
mas01cr@0 574 // Update Header information
mas01cr@0 575 dbH->length+=(statbuf.st_size-sizeof(int));
mas01cr@0 576
mas01cr@0 577 // Copy the header back to the database
mas01cr@0 578 memcpy (db, dbH, sizeof(dbTableHeaderT));
mas01cr@0 579
mas01cr@0 580 // Update segment to file index map
mas01cr@0 581 //memcpy (db+segTableOffset+(dbH->numFiles-1)*sizeof(unsigned), &numVectors, sizeof(unsigned));
mas01cr@0 582 memcpy (segTable+dbH->numFiles-1, &numVectors, sizeof(unsigned));
mas01cr@0 583
mas01cr@0 584 // Update the feature database
mas01cr@0 585 memcpy (db+dataoffset+insertoffset, indata+sizeof(int), statbuf.st_size-sizeof(int));
mas01cr@0 586
mas01cr@0 587 // Norm the vectors on input if the database is already L2 normed
mas01cr@0 588 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@0 589 unitNormAndInsertL2((double*)(db+dataoffset+insertoffset), dbH->dim, numVectors, 1); // append
mas01cr@0 590
mas01cr@0 591 // Report status
mas01cr@0 592 status(dbName);
mas01cr@0 593 if(verbosity)
mas01cr@0 594 cerr << COM_INSERT << " " << dbName << " " << numVectors << " vectors "
mas01cr@0 595 << (statbuf.st_size-sizeof(int)) << " bytes." << endl;
mas01cr@0 596
mas01cr@0 597 // CLEAN UP
mas01cr@0 598 munmap(indata,statbuf.st_size);
mas01cr@0 599 close(infid);
mas01cr@0 600 }
mas01cr@0 601
mas01cr@0 602 void audioDB::insertTimeStamps(unsigned numVectors, ifstream* timesFile, double* timesdata){
mas01cr@0 603 unsigned numtimes=0;
mas01cr@0 604 if(usingTimes){
mas01cr@0 605 if(!(dbH->flags & O2_FLAG_TIMES) && !dbH->numFiles)
mas01cr@0 606 dbH->flags=dbH->flags|O2_FLAG_TIMES;
mas01cr@0 607 else if(!(dbH->flags&O2_FLAG_TIMES)){
mas01cr@0 608 cerr << "Warning: timestamp file used with non time-stamped database: ignoring timestamps" << endl;
mas01cr@0 609 usingTimes=0;
mas01cr@0 610 }
mas01cr@0 611
mas01cr@0 612 if(!timesFile->is_open()){
mas01cr@0 613 if(dbH->flags & O2_FLAG_TIMES){
mas01cr@0 614 munmap(indata,statbuf.st_size);
mas01cr@0 615 munmap(db,O2_DEFAULTDBSIZE);
mas01cr@0 616 error("problem opening times file on timestamped database",timesFileName);
mas01cr@0 617 }
mas01cr@0 618 else{
mas01cr@0 619 cerr << "Warning: problem opening times file. But non-timestamped database, so ignoring times file." << endl;
mas01cr@0 620 usingTimes=0;
mas01cr@0 621 }
mas01cr@0 622 }
mas01cr@0 623
mas01cr@0 624 // Process time file
mas01cr@0 625 if(usingTimes){
mas01cr@0 626 do{
mas01cr@0 627 *timesFile>>*timesdata++;
mas01cr@0 628 if(timesFile->eof())
mas01cr@0 629 break;
mas01cr@0 630 numtimes++;
mas01cr@0 631 }while(!timesFile->eof() && numtimes<numVectors);
mas01cr@0 632 if(!timesFile->eof()){
mas01cr@0 633 double dummy;
mas01cr@0 634 do{
mas01cr@0 635 *timesFile>>dummy;
mas01cr@0 636 if(timesFile->eof())
mas01cr@0 637 break;
mas01cr@0 638 numtimes++;
mas01cr@0 639 }while(!timesFile->eof());
mas01cr@0 640 }
mas01cr@0 641 if(numtimes<numVectors || numtimes>numVectors+2){
mas01cr@0 642 munmap(indata,statbuf.st_size);
mas01cr@0 643 munmap(db,O2_DEFAULTDBSIZE);
mas01cr@0 644 close(infid);
mas01cr@0 645 cerr << "expected " << numVectors << " found " << numtimes << endl;
mas01cr@0 646 error("Times file is incorrect length for features file",inFile);
mas01cr@0 647 }
mas01cr@0 648 if(verbosity>2)
mas01cr@0 649 cerr << "numtimes: " << numtimes << endl;
mas01cr@0 650 }
mas01cr@0 651 }
mas01cr@0 652 }
mas01cr@0 653
mas01cr@0 654 void audioDB::batchinsert(const char* dbName, const char* inFile){
mas01cr@0 655
mas01cr@0 656 if ((dbfid = open (dbName, O_RDWR)) < 0)
mas01cr@0 657 error("Can't open database file:", dbName);
mas01cr@0 658
mas01cr@0 659 if(!key)
mas01cr@0 660 key=inFile;
mas01cr@0 661 ifstream *filesIn = 0;
mas01cr@0 662 ifstream *keysIn = 0;
mas01cr@0 663 ifstream* thisTimesFile = 0;
mas01cr@0 664
mas01cr@0 665 if(!(filesIn = new ifstream(inFile)))
mas01cr@0 666 error("Could not open batch in file", inFile);
mas01cr@0 667 if(key && key!=inFile)
mas01cr@0 668 if(!(keysIn = new ifstream(key)))
mas01cr@0 669 error("Could not open batch key file",key);
mas01cr@0 670
mas01cr@0 671 // Get the database header info
mas01cr@0 672 dbH = new dbTableHeaderT();
mas01cr@0 673 assert(dbH);
mas01cr@0 674
mas01cr@0 675 if(read(dbfid,(char*)dbH,sizeof(dbTableHeaderT))!=sizeof(dbTableHeaderT))
mas01cr@0 676 error("error reading db header");
mas01cr@0 677
mas01cr@0 678 if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 679 error("Must use timestamps with timestamped database","use --times");
mas01cr@0 680
mas01cr@0 681 fileTableOffset = O2_HEADERSIZE;
mas01cr@0 682 segTableOffset = fileTableOffset + O2_FILETABLESIZE*O2_MAXFILES;
mas01cr@0 683 dataoffset = segTableOffset + O2_SEGTABLESIZE*O2_MAXFILES;
mas01cr@0 684 l2normTableOffset = O2_DEFAULTDBSIZE - O2_MAXFILES*O2_MEANNUMVECTORS*sizeof(double);
mas01cr@0 685 timesTableOffset = l2normTableOffset - O2_MAXFILES*O2_MEANNUMVECTORS*sizeof(double);
mas01cr@0 686
mas01cr@0 687 if(dbH->magic!=O2_MAGIC){
mas01cr@0 688 cerr << "expected:" << O2_MAGIC << ", got:" << dbH->magic << endl;
mas01cr@0 689 error("database file has incorrect header",dbName);
mas01cr@0 690 }
mas01cr@0 691
mas01cr@0 692
mas01cr@0 693 // mmap the database file
mas01cr@0 694 if ((db = (char*) mmap(0, O2_DEFAULTDBSIZE, PROT_READ | PROT_WRITE,
mas01cr@0 695 MAP_SHARED, dbfid, 0)) == (caddr_t) -1)
mas01cr@0 696 error("mmap error for creating database");
mas01cr@0 697
mas01cr@0 698 // Make some handy tables with correct types
mas01cr@0 699 fileTable= (char*)(db+fileTableOffset);
mas01cr@0 700 segTable = (unsigned*)(db+segTableOffset);
mas01cr@0 701 dataBuf = (double*)(db+dataoffset);
mas01cr@0 702 l2normTable = (double*)(db+l2normTableOffset);
mas01cr@0 703 timesTable = (double*)(db+timesTableOffset);
mas01cr@0 704
mas01cr@0 705 unsigned totalVectors=0;
mas01cr@0 706 char *thisKey = new char[MAXSTR];
mas01cr@0 707 char *thisFile = new char[MAXSTR];
mas01cr@0 708 char *thisTimesFileName = new char[MAXSTR];
mas01cr@0 709
mas01cr@0 710 do{
mas01cr@0 711 filesIn->getline(thisFile,MAXSTR);
mas01cr@0 712 if(key && key!=inFile)
mas01cr@0 713 keysIn->getline(thisKey,MAXSTR);
mas01cr@0 714 else
mas01cr@0 715 thisKey = thisFile;
mas01cr@0 716 if(usingTimes)
mas01cr@0 717 timesFile->getline(thisTimesFileName,MAXSTR);
mas01cr@0 718
mas01cr@0 719 if(filesIn->eof())
mas01cr@0 720 break;
mas01cr@0 721
mas01cr@0 722 // open the input file
mas01cr@0 723 if (thisFile && (infid = open (thisFile, O_RDONLY)) < 0)
mas01cr@0 724 error("can't open feature file for reading", thisFile);
mas01cr@0 725
mas01cr@0 726 // find size of input file
mas01cr@0 727 if (thisFile && fstat (infid,&statbuf) < 0)
mas01cr@0 728 error("fstat error finding size of input");
mas01cr@0 729
mas01cr@0 730 // Check that there is room for at least 1 more file
mas01cr@0 731 if((char*)timesTable<((char*)dataBuf+(dbH->length+statbuf.st_size-sizeof(int))))
mas01cr@0 732 error("No more room in database","insert failed: reason database is full.");
mas01cr@0 733
mas01cr@0 734 if(thisFile)
mas01cr@0 735 if(dbH->dim==0 && dbH->length==0) // empty database
mas01cr@0 736 read(infid,&dbH->dim,sizeof(unsigned)); // initialize with input dimensionality
mas01cr@0 737 else {
mas01cr@0 738 unsigned test;
mas01cr@0 739 read(infid,&test,sizeof(unsigned));
mas01cr@0 740 if(dbH->dim!=test){
mas01cr@0 741 cerr << "error: expected dimension: " << dbH->dim << ", got :" << test <<endl;
mas01cr@0 742 error("feature dimensions do not match database table dimensions");
mas01cr@0 743 }
mas01cr@0 744 }
mas01cr@0 745
mas01cr@0 746 // mmap the input file
mas01cr@0 747 if (thisFile && (indata = (char*)mmap (0, statbuf.st_size, PROT_READ, MAP_SHARED, infid, 0))
mas01cr@0 748 == (caddr_t) -1)
mas01cr@0 749 error("mmap error for input");
mas01cr@0 750
mas01cr@0 751
mas01cr@0 752 // Linear scan of filenames check for pre-existing feature
mas01cr@0 753 unsigned alreadyInserted=0;
mas01cr@0 754
mas01cr@0 755 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 756 if(strncmp(fileTable + k*O2_FILETABLESIZE, thisKey, strlen(thisKey))==0){
mas01cr@0 757 alreadyInserted=1;
mas01cr@0 758 break;
mas01cr@0 759 }
mas01cr@0 760
mas01cr@0 761 if(alreadyInserted){
mas01cr@0 762 if(verbosity)
mas01cr@0 763 cerr << "Warning: key already exists in database:" << thisKey << endl;
mas01cr@0 764 }
mas01cr@0 765 else{
mas01cr@0 766
mas01cr@0 767 // Make a segment index table of features to file indexes
mas01cr@0 768 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 769 if(!numVectors){
mas01cr@0 770 if(verbosity)
mas01cr@0 771 cerr << "Warning: ignoring zero-length feature vector file:" << thisKey << endl;
mas01cr@0 772 }
mas01cr@0 773 else{
mas01cr@0 774 if(usingTimes){
mas01cr@0 775 if(timesFile->eof())
mas01cr@0 776 error("not enough timestamp files in timesList");
mas01cr@0 777 thisTimesFile=new ifstream(thisTimesFileName,ios::in);
mas01cr@0 778 if(!thisTimesFile->is_open())
mas01cr@0 779 error("Cannot open timestamp file",thisTimesFileName);
mas01cr@0 780 unsigned insertoffset=dbH->length;
mas01cr@0 781 unsigned timesoffset=insertoffset/(dbH->dim*sizeof(double));
mas01cr@0 782 double* timesdata=timesTable+timesoffset;
mas01cr@0 783 assert(timesdata+numVectors<l2normTable);
mas01cr@0 784 insertTimeStamps(numVectors,thisTimesFile,timesdata);
mas01cr@0 785 if(thisTimesFile)
mas01cr@0 786 delete thisTimesFile;
mas01cr@0 787 }
mas01cr@0 788
mas01cr@0 789 strncpy(fileTable + dbH->numFiles*O2_FILETABLESIZE, thisKey, strlen(thisKey));
mas01cr@0 790
mas01cr@0 791 unsigned insertoffset = dbH->length;// Store current state
mas01cr@0 792
mas01cr@0 793 // Increment file count
mas01cr@0 794 dbH->numFiles++;
mas01cr@0 795
mas01cr@0 796 // Update Header information
mas01cr@0 797 dbH->length+=(statbuf.st_size-sizeof(int));
mas01cr@0 798 // Copy the header back to the database
mas01cr@0 799 memcpy (db, dbH, sizeof(dbTableHeaderT));
mas01cr@0 800
mas01cr@0 801 // Update segment to file index map
mas01cr@0 802 //memcpy (db+segTableOffset+(dbH->numFiles-1)*sizeof(unsigned), &numVectors, sizeof(unsigned));
mas01cr@0 803 memcpy (segTable+dbH->numFiles-1, &numVectors, sizeof(unsigned));
mas01cr@0 804
mas01cr@0 805 // Update the feature database
mas01cr@0 806 memcpy (db+dataoffset+insertoffset, indata+sizeof(int), statbuf.st_size-sizeof(int));
mas01cr@0 807
mas01cr@0 808 // Norm the vectors on input if the database is already L2 normed
mas01cr@0 809 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@0 810 unitNormAndInsertL2((double*)(db+dataoffset+insertoffset), dbH->dim, numVectors, 1); // append
mas01cr@0 811
mas01cr@0 812 totalVectors+=numVectors;
mas01cr@0 813 }
mas01cr@0 814 }
mas01cr@0 815 // CLEAN UP
mas01cr@0 816 munmap(indata,statbuf.st_size);
mas01cr@0 817 close(infid);
mas01cr@0 818 }while(!filesIn->eof());
mas01cr@0 819
mas01cr@0 820 if(verbosity)
mas01cr@0 821 cerr << COM_BATCHINSERT << " " << dbName << " " << totalVectors << " vectors "
mas01cr@0 822 << totalVectors*dbH->dim*sizeof(double) << " bytes." << endl;
mas01cr@0 823
mas01cr@0 824 // Report status
mas01cr@0 825 status(dbName);
mas01cr@0 826
mas01cr@0 827 }
mas01cr@0 828
mas01cr@0 829 void audioDB::ws_status(const char*dbName, char* hostport){
mas01cr@0 830 struct soap soap;
mas01cr@0 831 int adbStatusResult;
mas01cr@0 832
mas01cr@0 833 // Query an existing adb database
mas01cr@0 834 soap_init(&soap);
mas01cr@0 835 if(soap_call_adb__status(&soap,hostport,NULL,(char*)dbName,adbStatusResult)==SOAP_OK)
mas01cr@0 836 std::cout << "result = " << adbStatusResult << std::endl;
mas01cr@0 837 else
mas01cr@0 838 soap_print_fault(&soap,stderr);
mas01cr@0 839
mas01cr@0 840 soap_destroy(&soap);
mas01cr@0 841 soap_end(&soap);
mas01cr@0 842 soap_done(&soap);
mas01cr@0 843 }
mas01cr@0 844
mas01cr@0 845 void audioDB::ws_query(const char*dbName, const char *segKey, const char* hostport){
mas01cr@0 846 struct soap soap;
mas01cr@0 847 adb__queryResult adbQueryResult;
mas01cr@0 848
mas01cr@0 849 soap_init(&soap);
mas01cr@0 850 if(soap_call_adb__query(&soap,hostport,NULL,
mas01cr@0 851 (char*)dbName,(char*)segKey,(char*)segFileName,(char*)timesFileName,
mas01cr@0 852 queryType, queryPoint, pointNN, segNN, sequenceLength, adbQueryResult)==SOAP_OK){
mas01cr@0 853 //std::cerr << "result list length:" << adbQueryResult.__sizeRlist << std::endl;
mas01cr@0 854 for(int i=0; i<adbQueryResult.__sizeRlist; i++)
mas01cr@0 855 std::cout << adbQueryResult.Rlist[i] << " " << adbQueryResult.Dist[i]
mas01cr@0 856 << " " << adbQueryResult.Qpos[i] << " " << adbQueryResult.Spos[i] << std::endl;
mas01cr@0 857 }
mas01cr@0 858 else
mas01cr@0 859 soap_print_fault(&soap,stderr);
mas01cr@0 860
mas01cr@0 861 soap_destroy(&soap);
mas01cr@0 862 soap_end(&soap);
mas01cr@0 863 soap_done(&soap);
mas01cr@0 864
mas01cr@0 865 }
mas01cr@0 866
mas01cr@0 867
mas01cr@0 868 void audioDB::status(const char* dbName){
mas01cr@0 869 if(!dbH)
mas01cr@0 870 initTables(dbName, 0);
mas01cr@0 871
mas01cr@0 872 // Update Header information
mas01cr@0 873 cout << "num files:" << dbH->numFiles << endl;
mas01cr@0 874 cout << "data dim:" << dbH->dim <<endl;
mas01cr@0 875 if(dbH->dim>0){
mas01cr@0 876 cout << "total vectors:" << dbH->length/(sizeof(double)*dbH->dim)<<endl;
mas01cr@0 877 cout << "vectors available:" << (timesTableOffset-(dataoffset+dbH->length))/(sizeof(double)*dbH->dim) << endl;
mas01cr@0 878 }
mas01cr@0 879 cout << "total bytes:" << dbH->length << " (" << (100.0*dbH->length)/(timesTableOffset-dataoffset) << "%)" << endl;
mas01cr@0 880 cout << "bytes available:" << timesTableOffset-(dataoffset+dbH->length) << " (" <<
mas01cr@0 881 (100.0*(timesTableOffset-(dataoffset+dbH->length)))/(timesTableOffset-dataoffset) << "%)" << endl;
mas01cr@0 882 cout << "flags:" << dbH->flags << endl;
mas01cr@0 883
mas01cr@0 884 unsigned dudCount=0;
mas01cr@0 885 unsigned nullCount=0;
mas01cr@0 886 for(unsigned k=0; k<dbH->numFiles; k++){
mas01cr@0 887 if(segTable[k]<sequenceLength){
mas01cr@0 888 dudCount++;
mas01cr@0 889 if(!segTable[k])
mas01cr@0 890 nullCount++;
mas01cr@0 891 }
mas01cr@0 892 }
mas01cr@0 893 cout << "null count: " << nullCount << " small sequence count " << dudCount-nullCount << endl;
mas01cr@0 894 }
mas01cr@0 895
mas01cr@0 896
mas01cr@0 897 void audioDB::dump(const char* dbName){
mas01cr@0 898 if(!dbH)
mas01cr@0 899 initTables(dbName,0);
mas01cr@0 900
mas01cr@0 901 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 902 cout << fileTable+k*O2_FILETABLESIZE << " " << segTable[k] << endl;
mas01cr@0 903
mas01cr@0 904 status(dbName);
mas01cr@0 905 }
mas01cr@0 906
mas01cr@0 907 void audioDB::l2norm(const char* dbName){
mas01cr@0 908 initTables(dbName,0);
mas01cr@0 909 if(dbH->length>0){
mas01cr@0 910 unsigned numVectors = dbH->length/(sizeof(double)*dbH->dim);
mas01cr@0 911 unitNormAndInsertL2(dataBuf, dbH->dim, numVectors, 0); // No append
mas01cr@0 912 }
mas01cr@0 913 // Update database flags
mas01cr@0 914 dbH->flags = dbH->flags|O2_FLAG_L2NORM;
mas01cr@0 915 memcpy (db, dbH, O2_HEADERSIZE);
mas01cr@0 916 }
mas01cr@0 917
mas01cr@0 918
mas01cr@0 919
mas01cr@0 920 void audioDB::query(const char* dbName, const char* inFile, adb__queryResult *adbQueryResult){
mas01cr@0 921 switch(queryType){
mas01cr@0 922 case O2_FLAG_POINT_QUERY:
mas01cr@0 923 pointQuery(dbName, inFile, adbQueryResult);
mas01cr@0 924 break;
mas01cr@0 925 case O2_FLAG_SEQUENCE_QUERY:
mas01cr@0 926 segSequenceQuery(dbName, inFile, adbQueryResult);
mas01cr@0 927 break;
mas01cr@0 928 case O2_FLAG_SEG_QUERY:
mas01cr@0 929 segPointQuery(dbName, inFile, adbQueryResult);
mas01cr@0 930 break;
mas01cr@0 931 default:
mas01cr@0 932 error("unrecognized queryType in query()");
mas01cr@0 933
mas01cr@0 934 }
mas01cr@0 935 }
mas01cr@0 936
mas01cr@0 937 //return ordinal position of key in keyTable
mas01cr@0 938 unsigned audioDB::getKeyPos(char* key){
mas01cr@0 939 for(unsigned k=0; k<dbH->numFiles; k++)
mas01cr@0 940 if(strncmp(fileTable + k*O2_FILETABLESIZE, key, strlen(key))==0)
mas01cr@0 941 return k;
mas01cr@0 942 error("Key not found",key);
mas01cr@0 943 return O2_ERR_KEYNOTFOUND;
mas01cr@0 944 }
mas01cr@0 945
mas01cr@0 946 // Basic point query engine
mas01cr@0 947 void audioDB::pointQuery(const char* dbName, const char* inFile, adb__queryResult *adbQueryResult){
mas01cr@0 948
mas01cr@0 949 initTables(dbName, inFile);
mas01cr@0 950
mas01cr@0 951 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 952 // we use stdout in this stub version
mas01cr@0 953 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 954
mas01cr@0 955 double* query = (double*)(indata+sizeof(int));
mas01cr@0 956 double* data = dataBuf;
mas01cr@0 957 double* queryCopy = 0;
mas01cr@0 958
mas01cr@0 959 if( dbH->flags & O2_FLAG_L2NORM ){
mas01cr@0 960 // Make a copy of the query
mas01cr@0 961 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 962 qNorm = new double[numVectors];
mas01cr@0 963 assert(queryCopy&&qNorm);
mas01cr@0 964 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 965 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 966 query = queryCopy;
mas01cr@0 967 }
mas01cr@0 968
mas01cr@0 969 // Make temporary dynamic memory for results
mas01cr@0 970 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01cr@0 971 double distances[pointNN];
mas01cr@0 972 unsigned qIndexes[pointNN];
mas01cr@0 973 unsigned sIndexes[pointNN];
mas01cr@0 974 for(unsigned k=0; k<pointNN; k++){
mas01cr@0 975 distances[k]=0.0;
mas01cr@0 976 qIndexes[k]=~0;
mas01cr@0 977 sIndexes[k]=~0;
mas01cr@0 978 }
mas01cr@0 979
mas01cr@0 980 unsigned j=numVectors;
mas01cr@0 981 unsigned k,l,n;
mas01cr@0 982 double thisDist;
mas01cr@0 983
mas01cr@0 984 unsigned totalVecs=dbH->length/(dbH->dim*sizeof(double));
mas01cr@0 985 double meanQdur = 0;
mas01cr@0 986 double* timesdata = 0;
mas01cr@0 987 double* dbdurs = 0;
mas01cr@0 988
mas01cr@0 989 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 990 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 991 usingTimes=0;
mas01cr@0 992 }
mas01cr@0 993
mas01cr@0 994 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 995 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 996
mas01cr@0 997 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 998 timesdata = new double[numVectors];
mas01cr@0 999 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1000 // Calculate durations of points
mas01cr@0 1001 for(k=0; k<numVectors-1; k++){
mas01cr@0 1002 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1003 meanQdur+=timesdata[k];
mas01cr@0 1004 }
mas01cr@0 1005 meanQdur/=k;
mas01cr@0 1006 // Individual exhaustive timepoint durations
mas01cr@0 1007 dbdurs = new double[totalVecs];
mas01cr@0 1008 for(k=0; k<totalVecs-1; k++)
mas01cr@0 1009 dbdurs[k]=timesTable[k+1]-timesTable[k];
mas01cr@0 1010 j--; // decrement vector counter by one
mas01cr@0 1011 }
mas01cr@0 1012
mas01cr@0 1013 if(usingQueryPoint)
mas01cr@0 1014 if(queryPoint>numVectors-1)
mas01cr@0 1015 error("queryPoint > numVectors in query");
mas01cr@0 1016 else{
mas01cr@0 1017 if(verbosity>1)
mas01cr@0 1018 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@0 1019 query=query+queryPoint*dbH->dim;
mas01cr@0 1020 numVectors=queryPoint+1;
mas01cr@0 1021 j=1;
mas01cr@0 1022 }
mas01cr@0 1023
mas01cr@0 1024 gettimeofday(&tv1, NULL);
mas01cr@0 1025 while(j--){ // query
mas01cr@0 1026 data=dataBuf;
mas01cr@0 1027 k=totalVecs; // number of database vectors
mas01cr@0 1028 while(k--){ // database
mas01cr@0 1029 thisDist=0;
mas01cr@0 1030 l=dbH->dim;
mas01cr@0 1031 double* q=query;
mas01cr@0 1032 while(l--)
mas01cr@0 1033 thisDist+=*q++**data++;
mas01cr@0 1034 if(!usingTimes ||
mas01cr@0 1035 (usingTimes
mas01cr@0 1036 && fabs(dbdurs[totalVecs-k-1]-timesdata[numVectors-j-1])<timesdata[numVectors-j-1]*timesTol)){
mas01cr@0 1037 n=pointNN;
mas01cr@0 1038 while(n--){
mas01cr@0 1039 if(thisDist>=distances[n]){
mas01cr@0 1040 if((n==0 || thisDist<=distances[n-1])){
mas01cr@0 1041 // Copy all values above up the queue
mas01cr@0 1042 for( l=pointNN-1 ; l >= n+1 ; l--){
mas01cr@0 1043 distances[l]=distances[l-1];
mas01cr@0 1044 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1045 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1046 }
mas01cr@0 1047 distances[n]=thisDist;
mas01cr@0 1048 qIndexes[n]=numVectors-j-1;
mas01cr@0 1049 sIndexes[n]=dbH->length/(sizeof(double)*dbH->dim)-k-1;
mas01cr@0 1050 break;
mas01cr@0 1051 }
mas01cr@0 1052 }
mas01cr@0 1053 else
mas01cr@0 1054 break;
mas01cr@0 1055 }
mas01cr@0 1056 }
mas01cr@0 1057 }
mas01cr@0 1058 // Move query pointer to next query point
mas01cr@0 1059 query+=dbH->dim;
mas01cr@0 1060 }
mas01cr@0 1061
mas01cr@0 1062 gettimeofday(&tv2, NULL);
mas01cr@0 1063 if(verbosity>1)
mas01cr@0 1064 cerr << endl << " elapsed time:" << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01cr@0 1065
mas01cr@0 1066 if(adbQueryResult==0){
mas01cr@0 1067 // Output answer
mas01cr@0 1068 // Loop over nearest neighbours
mas01cr@0 1069 for(k=0; k < pointNN; k++){
mas01cr@0 1070 // Scan for key
mas01cr@0 1071 unsigned cumSeg=0;
mas01cr@0 1072 for(l=0 ; l<dbH->numFiles; l++){
mas01cr@0 1073 cumSeg+=segTable[l];
mas01cr@0 1074 if(sIndexes[k]<cumSeg){
mas01cr@0 1075 cout << fileTable+l*O2_FILETABLESIZE << " " << distances[k] << " " << qIndexes[k] << " "
mas01cr@0 1076 << sIndexes[k]+segTable[l]-cumSeg << endl;
mas01cr@0 1077 break;
mas01cr@0 1078 }
mas01cr@0 1079 }
mas01cr@0 1080 }
mas01cr@0 1081 }
mas01cr@0 1082 else{ // Process Web Services Query
mas01cr@0 1083 int listLen = pointNN;
mas01cr@0 1084 adbQueryResult->__sizeRlist=listLen;
mas01cr@0 1085 adbQueryResult->__sizeDist=listLen;
mas01cr@0 1086 adbQueryResult->__sizeQpos=listLen;
mas01cr@0 1087 adbQueryResult->__sizeSpos=listLen;
mas01cr@0 1088 adbQueryResult->Rlist= new char*[listLen];
mas01cr@0 1089 adbQueryResult->Dist = new double[listLen];
mas01cr@0 1090 adbQueryResult->Qpos = new int[listLen];
mas01cr@0 1091 adbQueryResult->Spos = new int[listLen];
mas01cr@0 1092 for(k=0; k<adbQueryResult->__sizeRlist; k++){
mas01cr@0 1093 adbQueryResult->Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@0 1094 adbQueryResult->Dist[k]=distances[k];
mas01cr@0 1095 adbQueryResult->Qpos[k]=qIndexes[k];
mas01cr@0 1096 unsigned cumSeg=0;
mas01cr@0 1097 for(l=0 ; l<dbH->numFiles; l++){
mas01cr@0 1098 cumSeg+=segTable[l];
mas01cr@0 1099 if(sIndexes[k]<cumSeg){
mas01cr@0 1100 sprintf(adbQueryResult->Rlist[k], "%s", fileTable+l*O2_FILETABLESIZE);
mas01cr@0 1101 break;
mas01cr@0 1102 }
mas01cr@0 1103 }
mas01cr@0 1104 adbQueryResult->Spos[k]=sIndexes[k]+segTable[l]-cumSeg;
mas01cr@0 1105 }
mas01cr@0 1106 }
mas01cr@0 1107
mas01cr@0 1108 // Clean up
mas01cr@0 1109 if(queryCopy)
mas01cr@0 1110 delete queryCopy;
mas01cr@0 1111 if(qNorm)
mas01cr@0 1112 delete qNorm;
mas01cr@0 1113 if(timesdata)
mas01cr@0 1114 delete timesdata;
mas01cr@0 1115 if(dbdurs)
mas01cr@0 1116 delete dbdurs;
mas01cr@0 1117 }
mas01cr@0 1118
mas01cr@0 1119 void audioDB::sequenceQuery(const char* dbName, const char* inFile, adb__queryResult *adbQueryResult){
mas01cr@0 1120
mas01cr@0 1121 }
mas01cr@0 1122
mas01cr@0 1123 // segPointQuery
mas01cr@0 1124 // return the segNN closest segs to the query seg
mas01cr@0 1125 // uses average of pointNN points per seg
mas01cr@0 1126 void audioDB::segPointQuery(const char* dbName, const char* inFile, adb__queryResult *adbQueryResult){
mas01cr@0 1127 initTables(dbName, inFile);
mas01cr@0 1128
mas01cr@0 1129 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 1130 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 1131 unsigned numSegs = dbH->numFiles;
mas01cr@0 1132
mas01cr@0 1133 double* query = (double*)(indata+sizeof(int));
mas01cr@0 1134 double* data = dataBuf;
mas01cr@0 1135 double* queryCopy = 0;
mas01cr@0 1136
mas01cr@0 1137 if( dbH->flags & O2_FLAG_L2NORM ){
mas01cr@0 1138 // Make a copy of the query
mas01cr@0 1139 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 1140 qNorm = new double[numVectors];
mas01cr@0 1141 assert(queryCopy&&qNorm);
mas01cr@0 1142 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 1143 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 1144 query = queryCopy;
mas01cr@0 1145 }
mas01cr@0 1146
mas01cr@0 1147 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01cr@0 1148 assert(segNN>0 && segNN<=O2_MAXNN);
mas01cr@0 1149
mas01cr@0 1150 // Make temporary dynamic memory for results
mas01cr@0 1151 double segDistances[segNN];
mas01cr@0 1152 unsigned segIDs[segNN];
mas01cr@0 1153 unsigned segQIndexes[segNN];
mas01cr@0 1154 unsigned segSIndexes[segNN];
mas01cr@0 1155
mas01cr@0 1156 double distances[pointNN];
mas01cr@0 1157 unsigned qIndexes[pointNN];
mas01cr@0 1158 unsigned sIndexes[pointNN];
mas01cr@0 1159
mas01cr@0 1160 unsigned j=numVectors; // number of query points
mas01cr@0 1161 unsigned k,l,n, seg, segOffset=0, processedSegs=0;
mas01cr@0 1162 double thisDist;
mas01cr@0 1163
mas01cr@0 1164 for(k=0; k<pointNN; k++){
mas01cr@0 1165 distances[k]=0.0;
mas01cr@0 1166 qIndexes[k]=~0;
mas01cr@0 1167 sIndexes[k]=~0;
mas01cr@0 1168 }
mas01cr@0 1169
mas01cr@0 1170 for(k=0; k<segNN; k++){
mas01cr@0 1171 segDistances[k]=0.0;
mas01cr@0 1172 segQIndexes[k]=~0;
mas01cr@0 1173 segSIndexes[k]=~0;
mas01cr@0 1174 segIDs[k]=~0;
mas01cr@0 1175 }
mas01cr@0 1176
mas01cr@0 1177 double meanQdur = 0;
mas01cr@0 1178 double* timesdata = 0;
mas01cr@0 1179 double* meanDBdur = 0;
mas01cr@0 1180
mas01cr@0 1181 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1182 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 1183 usingTimes=0;
mas01cr@0 1184 }
mas01cr@0 1185
mas01cr@0 1186 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 1187 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 1188
mas01cr@0 1189 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1190 timesdata = new double[numVectors];
mas01cr@0 1191 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1192 // Calculate durations of points
mas01cr@0 1193 for(k=0; k<numVectors-1; k++){
mas01cr@0 1194 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1195 meanQdur+=timesdata[k];
mas01cr@0 1196 }
mas01cr@0 1197 meanQdur/=k;
mas01cr@0 1198 meanDBdur = new double[dbH->numFiles];
mas01cr@0 1199 for(k=0; k<dbH->numFiles; k++){
mas01cr@0 1200 meanDBdur[k]=0.0;
mas01cr@0 1201 for(j=0; j<segTable[k]-1 ; j++)
mas01cr@0 1202 meanDBdur[k]+=timesTable[j+1]-timesTable[j];
mas01cr@0 1203 meanDBdur[k]/=j;
mas01cr@0 1204 }
mas01cr@0 1205 }
mas01cr@0 1206
mas01cr@0 1207 if(usingQueryPoint)
mas01cr@0 1208 if(queryPoint>numVectors-1)
mas01cr@0 1209 error("queryPoint > numVectors in query");
mas01cr@0 1210 else{
mas01cr@0 1211 if(verbosity>1)
mas01cr@0 1212 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@0 1213 query=query+queryPoint*dbH->dim;
mas01cr@0 1214 numVectors=queryPoint+1;
mas01cr@0 1215 }
mas01cr@0 1216
mas01cr@0 1217 // build segment offset table
mas01cr@0 1218 unsigned *segOffsetTable = new unsigned[dbH->numFiles];
mas01cr@0 1219 unsigned cumSeg=0;
mas01cr@0 1220 unsigned segIndexOffset;
mas01cr@0 1221 for(k=0; k<dbH->numFiles;k++){
mas01cr@0 1222 segOffsetTable[k]=cumSeg;
mas01cr@0 1223 cumSeg+=segTable[k]*dbH->dim;
mas01cr@0 1224 }
mas01cr@0 1225
mas01cr@0 1226 char nextKey[MAXSTR];
mas01cr@0 1227
mas01cr@0 1228 gettimeofday(&tv1, NULL);
mas01cr@0 1229
mas01cr@0 1230 for(seg=0 ; seg < dbH->numFiles ; seg++, processedSegs++){
mas01cr@0 1231 if(segFile){
mas01cr@0 1232 if(!segFile->eof()){
mas01cr@0 1233 //*segFile>>seg;
mas01cr@0 1234 segFile->getline(nextKey,MAXSTR);
mas01cr@0 1235 seg=getKeyPos(nextKey);
mas01cr@0 1236 }
mas01cr@0 1237 else
mas01cr@0 1238 break;
mas01cr@0 1239 }
mas01cr@0 1240 segOffset=segOffsetTable[seg]; // numDoubles offset
mas01cr@0 1241 segIndexOffset=segOffset/dbH->dim; // numVectors offset
mas01cr@0 1242 if(verbosity>7)
mas01cr@0 1243 cerr << seg << "." << segOffset/(dbH->dim) << "." << segTable[seg] << " | ";cerr.flush();
mas01cr@0 1244
mas01cr@0 1245 if(dbH->flags & O2_FLAG_L2NORM)
mas01cr@0 1246 usingQueryPoint?query=queryCopy+queryPoint*dbH->dim:query=queryCopy;
mas01cr@0 1247 else
mas01cr@0 1248 usingQueryPoint?query=(double*)(indata+sizeof(int))+queryPoint*dbH->dim:query=(double*)(indata+sizeof(int));
mas01cr@0 1249 if(usingQueryPoint)
mas01cr@0 1250 j=1;
mas01cr@0 1251 else
mas01cr@0 1252 j=numVectors;
mas01cr@0 1253 while(j--){
mas01cr@0 1254 k=segTable[seg]; // number of vectors in seg
mas01cr@0 1255 data=dataBuf+segOffset; // data for seg
mas01cr@0 1256 while(k--){
mas01cr@0 1257 thisDist=0;
mas01cr@0 1258 l=dbH->dim;
mas01cr@0 1259 double* q=query;
mas01cr@0 1260 while(l--)
mas01cr@0 1261 thisDist+=*q++**data++;
mas01cr@0 1262 if(!usingTimes ||
mas01cr@0 1263 (usingTimes
mas01cr@0 1264 && fabs(meanDBdur[seg]-meanQdur)<meanQdur*timesTol)){
mas01cr@0 1265 n=pointNN;
mas01cr@0 1266 while(n--){
mas01cr@0 1267 if(thisDist>=distances[n]){
mas01cr@0 1268 if((n==0 || thisDist<=distances[n-1])){
mas01cr@0 1269 // Copy all values above up the queue
mas01cr@0 1270 for( l=pointNN-1 ; l > n ; l--){
mas01cr@0 1271 distances[l]=distances[l-1];
mas01cr@0 1272 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1273 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1274 }
mas01cr@0 1275 distances[n]=thisDist;
mas01cr@0 1276 qIndexes[n]=numVectors-j-1;
mas01cr@0 1277 sIndexes[n]=segTable[seg]-k-1;
mas01cr@0 1278 break;
mas01cr@0 1279 }
mas01cr@0 1280 }
mas01cr@0 1281 else
mas01cr@0 1282 break;
mas01cr@0 1283 }
mas01cr@0 1284 }
mas01cr@0 1285 } // seg
mas01cr@0 1286 // Move query pointer to next query point
mas01cr@0 1287 query+=dbH->dim;
mas01cr@0 1288 } // query
mas01cr@0 1289 // Take the average of this seg's distance
mas01cr@0 1290 // Test the seg distances
mas01cr@0 1291 thisDist=0;
mas01cr@0 1292 n=pointNN;
mas01cr@0 1293 while(n--)
mas01cr@0 1294 thisDist+=distances[pointNN-n-1];
mas01cr@0 1295 thisDist/=pointNN;
mas01cr@0 1296 n=segNN;
mas01cr@0 1297 while(n--){
mas01cr@0 1298 if(thisDist>=segDistances[n]){
mas01cr@0 1299 if((n==0 || thisDist<=segDistances[n-1])){
mas01cr@0 1300 // Copy all values above up the queue
mas01cr@0 1301 for( l=pointNN-1 ; l > n ; l--){
mas01cr@0 1302 segDistances[l]=segDistances[l-1];
mas01cr@0 1303 segQIndexes[l]=segQIndexes[l-1];
mas01cr@0 1304 segSIndexes[l]=segSIndexes[l-1];
mas01cr@0 1305 segIDs[l]=segIDs[l-1];
mas01cr@0 1306 }
mas01cr@0 1307 segDistances[n]=thisDist;
mas01cr@0 1308 segQIndexes[n]=qIndexes[0];
mas01cr@0 1309 segSIndexes[n]=sIndexes[0];
mas01cr@0 1310 segIDs[n]=seg;
mas01cr@0 1311 break;
mas01cr@0 1312 }
mas01cr@0 1313 }
mas01cr@0 1314 else
mas01cr@0 1315 break;
mas01cr@0 1316 }
mas01cr@0 1317 for(unsigned k=0; k<pointNN; k++){
mas01cr@0 1318 distances[k]=0.0;
mas01cr@0 1319 qIndexes[k]=~0;
mas01cr@0 1320 sIndexes[k]=~0;
mas01cr@0 1321 }
mas01cr@0 1322 } // segs
mas01cr@0 1323 gettimeofday(&tv2, NULL);
mas01cr@0 1324
mas01cr@0 1325 if(verbosity>1)
mas01cr@0 1326 cerr << endl << "processed segs :" << processedSegs
mas01cr@0 1327 << " elapsed time:" << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01cr@0 1328
mas01cr@0 1329 if(adbQueryResult==0){
mas01cr@0 1330 if(verbosity>1)
mas01cr@0 1331 cerr<<endl;
mas01cr@0 1332 // Output answer
mas01cr@0 1333 // Loop over nearest neighbours
mas01cr@0 1334 for(k=0; k < min(segNN,processedSegs); k++)
mas01cr@0 1335 cout << fileTable+segIDs[k]*O2_FILETABLESIZE
mas01cr@0 1336 << " " << segDistances[k] << " " << segQIndexes[k] << " " << segSIndexes[k] << endl;
mas01cr@0 1337 }
mas01cr@0 1338 else{ // Process Web Services Query
mas01cr@0 1339 int listLen = min(segNN, processedSegs);
mas01cr@0 1340 adbQueryResult->__sizeRlist=listLen;
mas01cr@0 1341 adbQueryResult->__sizeDist=listLen;
mas01cr@0 1342 adbQueryResult->__sizeQpos=listLen;
mas01cr@0 1343 adbQueryResult->__sizeSpos=listLen;
mas01cr@0 1344 adbQueryResult->Rlist= new char*[listLen];
mas01cr@0 1345 adbQueryResult->Dist = new double[listLen];
mas01cr@0 1346 adbQueryResult->Qpos = new int[listLen];
mas01cr@0 1347 adbQueryResult->Spos = new int[listLen];
mas01cr@0 1348 for(k=0; k<adbQueryResult->__sizeRlist; k++){
mas01cr@0 1349 adbQueryResult->Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@0 1350 adbQueryResult->Dist[k]=segDistances[k];
mas01cr@0 1351 adbQueryResult->Qpos[k]=segQIndexes[k];
mas01cr@0 1352 adbQueryResult->Spos[k]=segSIndexes[k];
mas01cr@0 1353 sprintf(adbQueryResult->Rlist[k], "%s", fileTable+segIDs[k]*O2_FILETABLESIZE);
mas01cr@0 1354 }
mas01cr@0 1355 }
mas01cr@0 1356
mas01cr@0 1357
mas01cr@0 1358 // Clean up
mas01cr@0 1359 if(segOffsetTable)
mas01cr@0 1360 delete segOffsetTable;
mas01cr@0 1361 if(queryCopy)
mas01cr@0 1362 delete queryCopy;
mas01cr@0 1363 if(qNorm)
mas01cr@0 1364 delete qNorm;
mas01cr@0 1365 if(timesdata)
mas01cr@0 1366 delete timesdata;
mas01cr@0 1367 if(meanDBdur)
mas01cr@0 1368 delete meanDBdur;
mas01cr@0 1369
mas01cr@0 1370 }
mas01cr@0 1371
mas01cr@0 1372 void audioDB::deleteDB(const char* dbName, const char* inFile){
mas01cr@0 1373
mas01cr@0 1374 }
mas01cr@0 1375
mas01cr@0 1376 // NBest matched filter distance between query and target segs
mas01cr@0 1377 // efficient implementation
mas01cr@0 1378 // outputs average of N minimum matched filter distances
mas01cr@0 1379 void audioDB::segSequenceQuery(const char* dbName, const char* inFile, adb__queryResult *adbQueryResult){
mas01cr@0 1380
mas01cr@0 1381 initTables(dbName, inFile);
mas01cr@0 1382
mas01cr@0 1383 // For each input vector, find the closest pointNN matching output vectors and report
mas01cr@0 1384 // we use stdout in this stub version
mas01cr@0 1385 unsigned numVectors = (statbuf.st_size-sizeof(int))/(sizeof(double)*dbH->dim);
mas01cr@0 1386 unsigned numSegs = dbH->numFiles;
mas01cr@0 1387
mas01cr@0 1388 double* query = (double*)(indata+sizeof(int));
mas01cr@0 1389 double* data = dataBuf;
mas01cr@0 1390 double* queryCopy = 0;
mas01cr@0 1391
mas01cr@0 1392 double qMeanL2;
mas01cr@0 1393 double* sMeanL2;
mas01cr@0 1394
mas01cr@0 1395 unsigned USE_THRESH=0;
mas01cr@0 1396 double SILENCE_THRESH=0;
mas01cr@0 1397 double DIFF_THRESH=0;
mas01cr@0 1398
mas01cr@0 1399 if(!(dbH->flags & O2_FLAG_L2NORM) )
mas01cr@0 1400 error("Database must be L2 normed for sequence query","use -l2norm");
mas01cr@0 1401
mas01cr@0 1402 if(verbosity>1)
mas01cr@0 1403 cerr << "performing norms ... "; cerr.flush();
mas01cr@0 1404 unsigned dbVectors = dbH->length/(sizeof(double)*dbH->dim);
mas01cr@0 1405 // Make a copy of the query
mas01cr@0 1406 queryCopy = new double[numVectors*dbH->dim];
mas01cr@0 1407 memcpy(queryCopy, query, numVectors*dbH->dim*sizeof(double));
mas01cr@0 1408 qNorm = new double[numVectors];
mas01cr@0 1409 sNorm = new double[dbVectors];
mas01cr@0 1410 sMeanL2=new double[dbH->numFiles];
mas01cr@0 1411 assert(qNorm&&sNorm&&queryCopy&&sMeanL2&&sequenceLength);
mas01cr@0 1412 unitNorm(queryCopy, dbH->dim, numVectors, qNorm);
mas01cr@0 1413 query = queryCopy;
mas01cr@0 1414 // Make norm measurements relative to sequenceLength
mas01cr@0 1415 unsigned w = sequenceLength-1;
mas01cr@0 1416 unsigned i,j;
mas01cr@0 1417 double* ps;
mas01cr@0 1418 double tmp1,tmp2;
mas01cr@0 1419 // Copy the L2 norm values to core to avoid disk random access later on
mas01cr@0 1420 memcpy(sNorm, l2normTable, dbVectors*sizeof(double));
mas01cr@0 1421 double* snPtr = sNorm;
mas01cr@0 1422 for(i=0; i<dbH->numFiles; i++){
mas01cr@0 1423 if(segTable[i]>sequenceLength){
mas01cr@0 1424 tmp1=*snPtr;
mas01cr@0 1425 j=1;
mas01cr@0 1426 w=sequenceLength-1;
mas01cr@0 1427 while(w--)
mas01cr@0 1428 *snPtr+=snPtr[j++];
mas01cr@0 1429 ps = snPtr+1;
mas01cr@0 1430 w=segTable[i]-sequenceLength; // +1 - 1
mas01cr@0 1431 while(w--){
mas01cr@0 1432 tmp2=*ps;
mas01cr@0 1433 *ps=*(ps-1)-tmp1+*(ps+sequenceLength);
mas01cr@0 1434 tmp1=tmp2;
mas01cr@0 1435 ps++;
mas01cr@0 1436 }
mas01cr@0 1437 }
mas01cr@0 1438 snPtr+=segTable[i];
mas01cr@0 1439 }
mas01cr@0 1440
mas01cr@0 1441 double* pn = sMeanL2;
mas01cr@0 1442 w=dbH->numFiles;
mas01cr@0 1443 while(w--)
mas01cr@0 1444 *pn++=0.0;
mas01cr@0 1445 ps=sNorm;
mas01cr@0 1446 unsigned processedSegs=0;
mas01cr@0 1447 for(i=0; i<dbH->numFiles; i++){
mas01cr@0 1448 if(segTable[i]>sequenceLength-1){
mas01cr@0 1449 w = segTable[i]-sequenceLength+1;
mas01cr@0 1450 pn = sMeanL2+i;
mas01cr@0 1451 while(w--)
mas01cr@0 1452 *pn+=*ps++;
mas01cr@0 1453 *pn/=segTable[i]-sequenceLength+1;
mas01cr@0 1454 SILENCE_THRESH+=*pn;
mas01cr@0 1455 processedSegs++;
mas01cr@0 1456 }
mas01cr@0 1457 }
mas01cr@0 1458 if(verbosity>1)
mas01cr@0 1459 cerr << "processedSegs: " << processedSegs << endl;
mas01cr@0 1460 SILENCE_THRESH/=processedSegs;
mas01cr@0 1461 USE_THRESH=1; // Turn thresholding on
mas01cr@0 1462 DIFF_THRESH=SILENCE_THRESH/=2; // 50% of the mean shingle power
mas01cr@0 1463 SILENCE_THRESH/=10; // 10% of the mean shingle power is SILENCE
mas01cr@0 1464
mas01cr@0 1465 w=sequenceLength-1;
mas01cr@0 1466 i=1;
mas01cr@0 1467 tmp1=*qNorm;
mas01cr@0 1468 while(w--)
mas01cr@0 1469 *qNorm+=qNorm[i++];
mas01cr@0 1470 ps = qNorm+1;
mas01cr@0 1471 qMeanL2 = *qNorm;
mas01cr@0 1472 w=numVectors-sequenceLength;
mas01cr@0 1473 while(w--){
mas01cr@0 1474 tmp2=*ps;
mas01cr@0 1475 *ps=*(ps-1)-tmp1+*(ps+sequenceLength);
mas01cr@0 1476 tmp1=tmp2;
mas01cr@0 1477 qMeanL2+=*ps;
mas01cr@0 1478 *ps++;
mas01cr@0 1479 }
mas01cr@0 1480 qMeanL2 /= numVectors-sequenceLength+1;
mas01cr@0 1481 if(verbosity>1)
mas01cr@0 1482 cerr << "done." << endl;
mas01cr@0 1483
mas01cr@0 1484
mas01cr@0 1485 if(verbosity>1)
mas01cr@0 1486 cerr << "matching segs..." << endl;
mas01cr@0 1487
mas01cr@0 1488 assert(pointNN>0 && pointNN<=O2_MAXNN);
mas01cr@0 1489 assert(segNN>0 && segNN<=O2_MAXNN);
mas01cr@0 1490
mas01cr@0 1491 // Make temporary dynamic memory for results
mas01cr@0 1492 double segDistances[segNN];
mas01cr@0 1493 unsigned segIDs[segNN];
mas01cr@0 1494 unsigned segQIndexes[segNN];
mas01cr@0 1495 unsigned segSIndexes[segNN];
mas01cr@0 1496
mas01cr@0 1497 double distances[pointNN];
mas01cr@0 1498 unsigned qIndexes[pointNN];
mas01cr@0 1499 unsigned sIndexes[pointNN];
mas01cr@0 1500
mas01cr@0 1501
mas01cr@0 1502 unsigned k,l,m,n,seg,segOffset=0, HOP_SIZE=sequenceHop, wL=sequenceLength;
mas01cr@0 1503 double thisDist;
mas01cr@0 1504 double oneOverWL=1.0/wL;
mas01cr@0 1505
mas01cr@0 1506 for(k=0; k<pointNN; k++){
mas01cr@0 1507 distances[k]=0.0;
mas01cr@0 1508 qIndexes[k]=~0;
mas01cr@0 1509 sIndexes[k]=~0;
mas01cr@0 1510 }
mas01cr@0 1511
mas01cr@0 1512 for(k=0; k<segNN; k++){
mas01cr@0 1513 segDistances[k]=0.0;
mas01cr@0 1514 segQIndexes[k]=~0;
mas01cr@0 1515 segSIndexes[k]=~0;
mas01cr@0 1516 segIDs[k]=~0;
mas01cr@0 1517 }
mas01cr@0 1518
mas01cr@0 1519 // Timestamp and durations processing
mas01cr@0 1520 double meanQdur = 0;
mas01cr@0 1521 double* timesdata = 0;
mas01cr@0 1522 double* meanDBdur = 0;
mas01cr@0 1523
mas01cr@0 1524 if(usingTimes && !(dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1525 cerr << "warning: ignoring query timestamps for non-timestamped database" << endl;
mas01cr@0 1526 usingTimes=0;
mas01cr@0 1527 }
mas01cr@0 1528
mas01cr@0 1529 else if(!usingTimes && (dbH->flags & O2_FLAG_TIMES))
mas01cr@0 1530 cerr << "warning: no timestamps given for query. Ignoring database timestamps." << endl;
mas01cr@0 1531
mas01cr@0 1532 else if(usingTimes && (dbH->flags & O2_FLAG_TIMES)){
mas01cr@0 1533 timesdata = new double[numVectors];
mas01cr@0 1534 assert(timesdata);
mas01cr@0 1535 insertTimeStamps(numVectors, timesFile, timesdata);
mas01cr@0 1536 // Calculate durations of points
mas01cr@0 1537 for(k=0; k<numVectors-1; k++){
mas01cr@0 1538 timesdata[k]=timesdata[k+1]-timesdata[k];
mas01cr@0 1539 meanQdur+=timesdata[k];
mas01cr@0 1540 }
mas01cr@0 1541 meanQdur/=k;
mas01cr@0 1542 if(verbosity>1)
mas01cr@0 1543 cerr << "mean query file duration: " << meanQdur << endl;
mas01cr@0 1544 meanDBdur = new double[dbH->numFiles];
mas01cr@0 1545 assert(meanDBdur);
mas01cr@0 1546 for(k=0; k<dbH->numFiles; k++){
mas01cr@0 1547 meanDBdur[k]=0.0;
mas01cr@0 1548 for(j=0; j<segTable[k]-1 ; j++)
mas01cr@0 1549 meanDBdur[k]+=timesTable[j+1]-timesTable[j];
mas01cr@0 1550 meanDBdur[k]/=j;
mas01cr@0 1551 }
mas01cr@0 1552 }
mas01cr@0 1553
mas01cr@0 1554 if(usingQueryPoint)
mas01cr@0 1555 if(queryPoint>numVectors || queryPoint>numVectors-wL+1)
mas01cr@0 1556 error("queryPoint > numVectors-wL+1 in query");
mas01cr@0 1557 else{
mas01cr@0 1558 if(verbosity>1)
mas01cr@0 1559 cerr << "query point: " << queryPoint << endl; cerr.flush();
mas01cr@0 1560 query=query+queryPoint*dbH->dim;
mas01cr@0 1561 qNorm=qNorm+queryPoint;
mas01cr@0 1562 numVectors=wL;
mas01cr@0 1563 }
mas01cr@0 1564
mas01cr@0 1565 double ** D = 0; // Cross-correlation between query and target
mas01cr@0 1566 double ** DD = 0; // Matched filter distance
mas01cr@0 1567
mas01cr@0 1568 D = new double*[numVectors];
mas01cr@0 1569 assert(D);
mas01cr@0 1570 DD = new double*[numVectors];
mas01cr@0 1571 assert(DD);
mas01cr@0 1572
mas01cr@0 1573 gettimeofday(&tv1, NULL);
mas01cr@0 1574 processedSegs=0;
mas01cr@0 1575 unsigned successfulSegs=0;
mas01cr@0 1576
mas01cr@0 1577 double* qp;
mas01cr@0 1578 double* sp;
mas01cr@0 1579 double* dp;
mas01cr@0 1580 double diffL2;
mas01cr@0 1581
mas01cr@0 1582 // build segment offset table
mas01cr@0 1583 unsigned *segOffsetTable = new unsigned[dbH->numFiles];
mas01cr@0 1584 unsigned cumSeg=0;
mas01cr@0 1585 unsigned segIndexOffset;
mas01cr@0 1586 for(k=0; k<dbH->numFiles;k++){
mas01cr@0 1587 segOffsetTable[k]=cumSeg;
mas01cr@0 1588 cumSeg+=segTable[k]*dbH->dim;
mas01cr@0 1589 }
mas01cr@0 1590
mas01cr@0 1591 char nextKey [MAXSTR];
mas01cr@0 1592 for(seg=0 ; seg < dbH->numFiles ; seg++, processedSegs++){
mas01cr@0 1593
mas01cr@0 1594 // get segID from file if using a control file
mas01cr@0 1595 if(segFile){
mas01cr@0 1596 if(!segFile->eof()){
mas01cr@0 1597 segFile->getline(nextKey,MAXSTR);
mas01cr@0 1598 seg=getKeyPos(nextKey);
mas01cr@0 1599 }
mas01cr@0 1600 else
mas01cr@0 1601 break;
mas01cr@0 1602 }
mas01cr@0 1603 segOffset=segOffsetTable[seg]; // numDoubles offset
mas01cr@0 1604 segIndexOffset=segOffset/dbH->dim; // numVectors offset
mas01cr@0 1605
mas01cr@0 1606 if(sequenceLength<segTable[seg]){ // test for short sequences
mas01cr@0 1607
mas01cr@0 1608 if(verbosity>7)
mas01cr@0 1609 cerr << seg << "." << segIndexOffset << "." << segTable[seg] << " | ";cerr.flush();
mas01cr@0 1610
mas01cr@0 1611 // Cross-correlation matrix
mas01cr@0 1612 for(j=0; j<numVectors;j++){
mas01cr@0 1613 D[j]=new double[segTable[seg]];
mas01cr@0 1614 assert(D[j]);
mas01cr@0 1615
mas01cr@0 1616 }
mas01cr@0 1617
mas01cr@0 1618 // Matched filter matrix
mas01cr@0 1619 for(j=0; j<numVectors;j++){
mas01cr@0 1620 DD[j]=new double[segTable[seg]];
mas01cr@0 1621 assert(DD[j]);
mas01cr@0 1622 }
mas01cr@0 1623
mas01cr@0 1624 // Cross Correlation
mas01cr@0 1625 for(j=0; j<numVectors; j++)
mas01cr@0 1626 for(k=0; k<segTable[seg]; k++){
mas01cr@0 1627 qp=query+j*dbH->dim;
mas01cr@0 1628 sp=dataBuf+segOffset+k*dbH->dim;
mas01cr@0 1629 DD[j][k]=0.0; // Initialize matched filter array
mas01cr@0 1630 dp=&D[j][k]; // point to correlation cell j,k
mas01cr@0 1631 *dp=0.0; // initialize correlation cell
mas01cr@0 1632 l=dbH->dim; // size of vectors
mas01cr@0 1633 while(l--)
mas01cr@0 1634 *dp+=*qp++**sp++;
mas01cr@0 1635 }
mas01cr@0 1636
mas01cr@0 1637 // Matched Filter
mas01cr@0 1638 // HOP SIZE == 1
mas01cr@0 1639 double* spd;
mas01cr@0 1640 if(HOP_SIZE==1){ // HOP_SIZE = shingleHop
mas01cr@0 1641 for(w=0; w<wL; w++)
mas01cr@0 1642 for(j=0; j<numVectors-w; j++){
mas01cr@0 1643 sp=DD[j];
mas01cr@0 1644 spd=D[j+w]+w;
mas01cr@0 1645 k=segTable[seg]-w;
mas01cr@0 1646 while(k--)
mas01cr@0 1647 *sp+++=*spd++;
mas01cr@0 1648 }
mas01cr@0 1649 }
mas01cr@0 1650 else{ // HOP_SIZE != 1
mas01cr@0 1651 for(w=0; w<wL; w++)
mas01cr@0 1652 for(j=0; j<numVectors-w; j+=HOP_SIZE){
mas01cr@0 1653 sp=DD[j];
mas01cr@0 1654 spd=D[j+w]+w;
mas01cr@0 1655 for(k=0; k<segTable[seg]-w; k+=HOP_SIZE){
mas01cr@0 1656 *sp+=*spd;
mas01cr@0 1657 sp+=HOP_SIZE;
mas01cr@0 1658 spd+=HOP_SIZE;
mas01cr@0 1659 }
mas01cr@0 1660 }
mas01cr@0 1661 }
mas01cr@0 1662
mas01cr@0 1663 if(verbosity>3){
mas01cr@0 1664 cerr << "meanQdur=" << meanQdur << " meanDBdur=" << meanDBdur[seg] << endl;
mas01cr@0 1665 cerr.flush();
mas01cr@0 1666 }
mas01cr@0 1667
mas01cr@0 1668 if(!usingTimes ||
mas01cr@0 1669 (usingTimes
mas01cr@0 1670 && fabs(meanDBdur[seg]-meanQdur)<meanQdur*timesTol)){
mas01cr@0 1671
mas01cr@0 1672 if(verbosity>3){
mas01cr@0 1673 cerr << "within duration tolerance." << endl;
mas01cr@0 1674 cerr.flush();
mas01cr@0 1675 }
mas01cr@0 1676
mas01cr@0 1677 // Search for minimum distance by shingles (concatenated vectors)
mas01cr@0 1678 for(j=0;j<numVectors-wL+1;j+=HOP_SIZE)
mas01cr@0 1679 for(k=0;k<segTable[seg]-wL+1;k+=HOP_SIZE){
mas01cr@0 1680
mas01cr@0 1681 diffL2 = fabs(qNorm[j] - sNorm[k]);
mas01cr@0 1682 // Power test
mas01cr@0 1683 if(!USE_THRESH ||
mas01cr@0 1684 // Threshold on mean L2 of Q and S sequences
mas01cr@0 1685 (USE_THRESH && qNorm[j]>SILENCE_THRESH && sNorm[k]>SILENCE_THRESH &&
mas01cr@0 1686 // Are both query and target windows above mean energy?
mas01cr@0 1687 (qNorm[j]>qMeanL2*.25 && sNorm[k]>sMeanL2[seg]*.25 && diffL2 < DIFF_THRESH )))
mas01cr@0 1688 thisDist=DD[j][k]*oneOverWL;
mas01cr@0 1689 else
mas01cr@0 1690 thisDist=0.0;
mas01cr@0 1691
mas01cr@0 1692 // NBest match algorithm
mas01cr@0 1693 for(m=0; m<pointNN; m++){
mas01cr@0 1694 if(thisDist>=distances[m]){
mas01cr@0 1695 // Shuffle distances up the list
mas01cr@0 1696 for(l=pointNN-1; l>m; l--){
mas01cr@0 1697 distances[l]=distances[l-1];
mas01cr@0 1698 qIndexes[l]=qIndexes[l-1];
mas01cr@0 1699 sIndexes[l]=sIndexes[l-1];
mas01cr@0 1700 }
mas01cr@0 1701 distances[m]=thisDist;
mas01cr@0 1702 if(usingQueryPoint)
mas01cr@0 1703 qIndexes[m]=queryPoint;
mas01cr@0 1704 else
mas01cr@0 1705 qIndexes[m]=j;
mas01cr@0 1706 sIndexes[m]=k;
mas01cr@0 1707 break;
mas01cr@0 1708 }
mas01cr@0 1709 }
mas01cr@0 1710 }
mas01cr@0 1711 // Calculate the mean of the N-Best matches
mas01cr@0 1712 thisDist=0.0;
mas01cr@0 1713 for(m=0; m<pointNN; m++)
mas01cr@0 1714 thisDist+=distances[m];
mas01cr@0 1715 thisDist/=pointNN;
mas01cr@0 1716
mas01cr@0 1717 // All the seg stuff goes here
mas01cr@0 1718 n=segNN;
mas01cr@0 1719 while(n--){
mas01cr@0 1720 if(thisDist>=segDistances[n]){
mas01cr@0 1721 if((n==0 || thisDist<=segDistances[n-1])){
mas01cr@0 1722 // Copy all values above up the queue
mas01cr@0 1723 for( l=segNN-1 ; l > n ; l--){
mas01cr@0 1724 segDistances[l]=segDistances[l-1];
mas01cr@0 1725 segQIndexes[l]=segQIndexes[l-1];
mas01cr@0 1726 segSIndexes[l]=segSIndexes[l-1];
mas01cr@0 1727 segIDs[l]=segIDs[l-1];
mas01cr@0 1728 }
mas01cr@0 1729 segDistances[n]=thisDist;
mas01cr@0 1730 segQIndexes[n]=qIndexes[0];
mas01cr@0 1731 segSIndexes[n]=sIndexes[0];
mas01cr@0 1732 successfulSegs++;
mas01cr@0 1733 segIDs[n]=seg;
mas01cr@0 1734 break;
mas01cr@0 1735 }
mas01cr@0 1736 }
mas01cr@0 1737 else
mas01cr@0 1738 break;
mas01cr@0 1739 }
mas01cr@0 1740 } // Duration match
mas01cr@0 1741
mas01cr@0 1742 // per-seg reset array values
mas01cr@0 1743 for(unsigned k=0; k<pointNN; k++){
mas01cr@0 1744 distances[k]=0.0;
mas01cr@0 1745 qIndexes[k]=~0;
mas01cr@0 1746 sIndexes[k]=~0;
mas01cr@0 1747 }
mas01cr@0 1748
mas01cr@0 1749 // Clean up current seg
mas01cr@0 1750 if(D!=NULL){
mas01cr@0 1751 for(j=0; j<numVectors; j++)
mas01cr@0 1752 delete[] D[j];
mas01cr@0 1753 }
mas01cr@0 1754
mas01cr@0 1755 if(DD!=NULL){
mas01cr@0 1756 for(j=0; j<numVectors; j++)
mas01cr@0 1757 delete[] DD[j];
mas01cr@0 1758 }
mas01cr@0 1759 }
mas01cr@0 1760 }
mas01cr@0 1761
mas01cr@0 1762 gettimeofday(&tv2,NULL);
mas01cr@0 1763 if(verbosity>1)
mas01cr@0 1764 cerr << endl << "processed segs :" << processedSegs << " matched segments: " << successfulSegs << " elapsed time:"
mas01cr@0 1765 << ( tv2.tv_sec*1000 + tv2.tv_usec/1000 ) - ( tv1.tv_sec*1000+tv1.tv_usec/1000 ) << " msec" << endl;
mas01cr@0 1766
mas01cr@0 1767 if(adbQueryResult==0){
mas01cr@0 1768 if(verbosity>1)
mas01cr@0 1769 cerr<<endl;
mas01cr@0 1770 // Output answer
mas01cr@0 1771 // Loop over nearest neighbours
mas01cr@0 1772 for(k=0; k < min(segNN,successfulSegs); k++)
mas01cr@0 1773 cout << fileTable+segIDs[k]*O2_FILETABLESIZE << " " << segDistances[k] << " " << segQIndexes[k] << " " << segSIndexes[k] << endl;
mas01cr@0 1774 }
mas01cr@0 1775 else{ // Process Web Services Query
mas01cr@0 1776 int listLen = min(segNN, processedSegs);
mas01cr@0 1777 adbQueryResult->__sizeRlist=listLen;
mas01cr@0 1778 adbQueryResult->__sizeDist=listLen;
mas01cr@0 1779 adbQueryResult->__sizeQpos=listLen;
mas01cr@0 1780 adbQueryResult->__sizeSpos=listLen;
mas01cr@0 1781 adbQueryResult->Rlist= new char*[listLen];
mas01cr@0 1782 adbQueryResult->Dist = new double[listLen];
mas01cr@0 1783 adbQueryResult->Qpos = new int[listLen];
mas01cr@0 1784 adbQueryResult->Spos = new int[listLen];
mas01cr@0 1785 for(k=0; k<adbQueryResult->__sizeRlist; k++){
mas01cr@0 1786 adbQueryResult->Rlist[k]=new char[O2_MAXFILESTR];
mas01cr@0 1787 adbQueryResult->Dist[k]=segDistances[k];
mas01cr@0 1788 adbQueryResult->Qpos[k]=segQIndexes[k];
mas01cr@0 1789 adbQueryResult->Spos[k]=segSIndexes[k];
mas01cr@0 1790 sprintf(adbQueryResult->Rlist[k], "%s", fileTable+segIDs[k]*O2_FILETABLESIZE);
mas01cr@0 1791 }
mas01cr@0 1792 }
mas01cr@0 1793
mas01cr@0 1794
mas01cr@0 1795 // Clean up
mas01cr@0 1796 if(segOffsetTable)
mas01cr@0 1797 delete segOffsetTable;
mas01cr@0 1798 if(queryCopy)
mas01cr@0 1799 delete queryCopy;
mas01cr@0 1800 //if(qNorm)
mas01cr@0 1801 //delete qNorm;
mas01cr@0 1802 if(D)
mas01cr@0 1803 delete[] D;
mas01cr@0 1804 if(DD)
mas01cr@0 1805 delete[] DD;
mas01cr@0 1806 if(timesdata)
mas01cr@0 1807 delete timesdata;
mas01cr@0 1808 if(meanDBdur)
mas01cr@0 1809 delete meanDBdur;
mas01cr@0 1810
mas01cr@0 1811
mas01cr@0 1812 }
mas01cr@0 1813
mas01cr@0 1814 void audioDB::normalize(double* X, int dim, int n){
mas01cr@0 1815 unsigned c = n*dim;
mas01cr@0 1816 double minval,maxval,v,*p;
mas01cr@0 1817
mas01cr@0 1818 p=X;
mas01cr@0 1819 while(c--){
mas01cr@0 1820 v=*p++;
mas01cr@0 1821 if(v<minval)
mas01cr@0 1822 minval=v;
mas01cr@0 1823 else if(v>maxval)
mas01cr@0 1824 maxval=v;
mas01cr@0 1825 }
mas01cr@0 1826
mas01cr@0 1827 normalize(X, dim, n, minval, maxval);
mas01cr@0 1828
mas01cr@0 1829 }
mas01cr@0 1830
mas01cr@0 1831 void audioDB::normalize(double* X, int dim, int n, double minval, double maxval){
mas01cr@0 1832 unsigned c = n*dim;
mas01cr@0 1833 double *p;
mas01cr@0 1834
mas01cr@0 1835
mas01cr@0 1836 if(maxval==minval)
mas01cr@0 1837 return;
mas01cr@0 1838
mas01cr@0 1839 maxval=1.0/(maxval-minval);
mas01cr@0 1840 c=n*dim;
mas01cr@0 1841 p=X;
mas01cr@0 1842
mas01cr@0 1843 while(c--){
mas01cr@0 1844 *p=(*p-minval)*maxval;
mas01cr@0 1845 p++;
mas01cr@0 1846 }
mas01cr@0 1847 }
mas01cr@0 1848
mas01cr@0 1849 // Unit norm block of features
mas01cr@0 1850 void audioDB::unitNorm(double* X, unsigned dim, unsigned n, double* qNorm){
mas01cr@0 1851 unsigned d;
mas01cr@0 1852 double L2, oneOverL2, *p;
mas01cr@0 1853 if(verbosity>2)
mas01cr@0 1854 cerr << "norming " << n << " vectors...";cerr.flush();
mas01cr@0 1855 while(n--){
mas01cr@0 1856 p=X;
mas01cr@0 1857 L2=0.0;
mas01cr@0 1858 d=dim;
mas01cr@0 1859 while(d--){
mas01cr@0 1860 L2+=*p**p;
mas01cr@0 1861 p++;
mas01cr@0 1862 }
mas01cr@0 1863 L2=sqrt(L2);
mas01cr@0 1864 if(qNorm)
mas01cr@0 1865 *qNorm++=L2;
mas01cr@0 1866 oneOverL2 = 1.0/L2;
mas01cr@0 1867 d=dim;
mas01cr@0 1868 while(d--){
mas01cr@0 1869 *X*=oneOverL2;
mas01cr@0 1870 X++;
mas01cr@0 1871 }
mas01cr@0 1872 }
mas01cr@0 1873 if(verbosity>2)
mas01cr@0 1874 cerr << "done..." << endl;
mas01cr@0 1875 }
mas01cr@0 1876
mas01cr@0 1877 // Unit norm block of features
mas01cr@0 1878 void audioDB::unitNormAndInsertL2(double* X, unsigned dim, unsigned n, unsigned append=0){
mas01cr@0 1879 unsigned d;
mas01cr@0 1880 double L2, oneOverL2, *p;
mas01cr@0 1881 unsigned nn = n;
mas01cr@0 1882
mas01cr@0 1883 assert(l2normTable);
mas01cr@0 1884
mas01cr@0 1885 if( !append && (dbH->flags & O2_FLAG_L2NORM) )
mas01cr@0 1886 error("Database is already L2 normed", "automatic norm on insert is enabled");
mas01cr@0 1887
mas01cr@0 1888 if(verbosity>2)
mas01cr@0 1889 cerr << "norming " << n << " vectors...";cerr.flush();
mas01cr@0 1890
mas01cr@0 1891 double* l2buf = new double[n];
mas01cr@0 1892 double* l2ptr = l2buf;
mas01cr@0 1893 assert(l2buf);
mas01cr@0 1894 assert(X);
mas01cr@0 1895
mas01cr@0 1896 while(nn--){
mas01cr@0 1897 p=X;
mas01cr@0 1898 *l2ptr=0.0;
mas01cr@0 1899 d=dim;
mas01cr@0 1900 while(d--){
mas01cr@0 1901 *l2ptr+=*p**p;
mas01cr@0 1902 p++;
mas01cr@0 1903 }
mas01cr@0 1904 *l2ptr=sqrt(*l2ptr);
mas01cr@0 1905 oneOverL2 = 1.0/(*l2ptr++);
mas01cr@0 1906 d=dim;
mas01cr@0 1907 while(d--){
mas01cr@0 1908 *X*=oneOverL2;
mas01cr@0 1909 X++;
mas01cr@0 1910 }
mas01cr@0 1911 }
mas01cr@0 1912 unsigned offset;
mas01cr@0 1913 if(append)
mas01cr@0 1914 offset=dbH->length/(dbH->dim*sizeof(double)); // number of vectors
mas01cr@0 1915 else
mas01cr@0 1916 offset=0;
mas01cr@0 1917 memcpy(l2normTable+offset, l2buf, n*sizeof(double));
mas01cr@0 1918 if(l2buf)
mas01cr@0 1919 delete l2buf;
mas01cr@0 1920 if(verbosity>2)
mas01cr@0 1921 cerr << "done..." << endl;
mas01cr@0 1922 }
mas01cr@0 1923
mas01cr@0 1924
mas01cr@0 1925 // Start an audioDB server on the host
mas01cr@0 1926 void audioDB::startServer(){
mas01cr@0 1927 struct soap soap;
mas01cr@0 1928 int m, s; // master and slave sockets
mas01cr@0 1929 soap_init(&soap);
mas01cr@0 1930 m = soap_bind(&soap, NULL, port, 100);
mas01cr@0 1931 if (m < 0)
mas01cr@0 1932 soap_print_fault(&soap, stderr);
mas01cr@0 1933 else
mas01cr@0 1934 {
mas01cr@0 1935 fprintf(stderr, "Socket connection successful: master socket = %d\n", m);
mas01cr@0 1936 for (int i = 1; ; i++)
mas01cr@0 1937 {
mas01cr@0 1938 s = soap_accept(&soap);
mas01cr@0 1939 if (s < 0)
mas01cr@0 1940 {
mas01cr@0 1941 soap_print_fault(&soap, stderr);
mas01cr@0 1942 break;
mas01cr@0 1943 }
mas01cr@0 1944 fprintf(stderr, "%d: accepted connection from IP=%d.%d.%d.%d socket=%d\n", i,
mas01cr@0 1945 (soap.ip >> 24)&0xFF, (soap.ip >> 16)&0xFF, (soap.ip >> 8)&0xFF, soap.ip&0xFF, s);
mas01cr@0 1946 if (soap_serve(&soap) != SOAP_OK) // process RPC request
mas01cr@0 1947 soap_print_fault(&soap, stderr); // print error
mas01cr@0 1948 fprintf(stderr, "request served\n");
mas01cr@0 1949 soap_destroy(&soap); // clean up class instances
mas01cr@0 1950 soap_end(&soap); // clean up everything and close socket
mas01cr@0 1951 }
mas01cr@0 1952 }
mas01cr@0 1953 soap_done(&soap); // close master socket and detach environment
mas01cr@0 1954 }
mas01cr@0 1955
mas01cr@0 1956
mas01cr@0 1957 // web services
mas01cr@0 1958
mas01cr@0 1959 // SERVER SIDE
mas01cr@0 1960 int adb__status(struct soap* soap, xsd__string dbName, xsd__int &adbCreateResult){
mas01cr@0 1961 char* const argv[]={"audioDB",COM_STATUS,dbName};
mas01cr@0 1962 const unsigned argc = 3;
mas01cr@0 1963 audioDB(argc,argv);
mas01cr@0 1964 adbCreateResult=100;
mas01cr@0 1965 return SOAP_OK;
mas01cr@0 1966 }
mas01cr@0 1967
mas01cr@0 1968 // Literal translation of command line to web service
mas01cr@0 1969
mas01cr@0 1970 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 segNN, xsd__int seqLen, adb__queryResult &adbQueryResult){
mas01cr@0 1971 char queryType[256];
mas01cr@0 1972 for(int k=0; k<256; k++)
mas01cr@0 1973 queryType[k]='\0';
mas01cr@0 1974 if(qType == O2_FLAG_POINT_QUERY)
mas01cr@0 1975 strncpy(queryType, "point", strlen("point"));
mas01cr@0 1976 else if (qType == O2_FLAG_SEQUENCE_QUERY)
mas01cr@0 1977 strncpy(queryType, "sequence", strlen("sequence"));
mas01cr@0 1978 else if(qType == O2_FLAG_SEG_QUERY)
mas01cr@0 1979 strncpy(queryType,"segment", strlen("segment"));
mas01cr@0 1980 else
mas01cr@0 1981 strncpy(queryType, "", strlen(""));
mas01cr@0 1982
mas01cr@0 1983 if(pointNN==0)
mas01cr@0 1984 pointNN=10;
mas01cr@0 1985 if(segNN==0)
mas01cr@0 1986 segNN=10;
mas01cr@0 1987 if(seqLen==0)
mas01cr@0 1988 seqLen=16;
mas01cr@0 1989
mas01cr@0 1990 char qPosStr[256];
mas01cr@0 1991 sprintf(qPosStr, "%d", qPos);
mas01cr@0 1992 char pointNNStr[256];
mas01cr@0 1993 sprintf(pointNNStr,"%d",pointNN);
mas01cr@0 1994 char segNNStr[256];
mas01cr@0 1995 sprintf(segNNStr,"%d",segNN);
mas01cr@0 1996 char seqLenStr[256];
mas01cr@0 1997 sprintf(seqLenStr,"%d",seqLen);
mas01cr@0 1998
mas01cr@0 1999 const char* argv[] ={
mas01cr@0 2000 "./audioDB",
mas01cr@0 2001 COM_QUERY,
mas01cr@0 2002 queryType, // Need to pass a parameter
mas01cr@0 2003 COM_DATABASE,
mas01cr@0 2004 dbName,
mas01cr@0 2005 COM_FEATURES,
mas01cr@0 2006 qKey,
mas01cr@0 2007 COM_KEYLIST,
mas01cr@0 2008 keyList==0?"":keyList,
mas01cr@0 2009 COM_TIMES,
mas01cr@0 2010 timesFileName==0?"":timesFileName,
mas01cr@0 2011 COM_QPOINT,
mas01cr@0 2012 qPosStr,
mas01cr@0 2013 COM_POINTNN,
mas01cr@0 2014 pointNNStr,
mas01cr@0 2015 COM_SEGNN,
mas01cr@0 2016 segNNStr, // Need to pass a parameter
mas01cr@0 2017 COM_SEQLEN,
mas01cr@0 2018 seqLenStr
mas01cr@0 2019 };
mas01cr@0 2020
mas01cr@0 2021 const unsigned argc = 19;
mas01cr@0 2022 audioDB(argc, (char* const*)argv, &adbQueryResult);
mas01cr@0 2023 return SOAP_OK;
mas01cr@0 2024 }
mas01cr@0 2025
mas01cr@0 2026 int main(const unsigned argc, char* const argv[]){
mas01cr@0 2027 audioDB(argc, argv);
mas01cr@0 2028 }
mas01cr@0 2029
mas01cr@0 2030