mas01mc@292
|
1 // LSH indexing
|
mas01mc@292
|
2 //
|
mas01mc@292
|
3 // Construct a persistent LSH table structure
|
mas01mc@292
|
4 // Store at the same location as dbName
|
mas01mc@292
|
5 // Naming convention:
|
mas01mc@292
|
6 // dbName.lsh.${radius}.${sequenceLength}
|
mas01mc@292
|
7 //
|
mas01mc@292
|
8 //
|
mas01mc@292
|
9 // Author: Michael Casey
|
mas01mc@292
|
10 // Date: 23 June 2008
|
mas01mc@319
|
11 //
|
mas01mc@319
|
12 // 19th August 2008 - added O2_FLAG_LARGE_ADB support
|
mas01mc@292
|
13
|
mas01mc@292
|
14 #include "audioDB.h"
|
mas01mc@292
|
15 #include "ReporterBase.h"
|
mas01mc@292
|
16
|
mas01mc@292
|
17
|
mas01mc@292
|
18 /************************* LSH point index to audioDB conversion *****************/
|
mas01mc@319
|
19 Uns32T audioDB::index_to_trackID(Uns32T lshID, Uns32T nPntBits){
|
mas01mc@319
|
20 assert(nPntBits);
|
mas01mc@319
|
21 return lshID>>nPntBits;
|
mas01mc@292
|
22 }
|
mas01mc@292
|
23
|
mas01mc@319
|
24 Uns32T audioDB::index_to_trackPos(Uns32T lshID, Uns32T nPntBits){
|
mas01mc@319
|
25 assert(nPntBits);
|
mas01mc@319
|
26 return lshID&((1<<nPntBits)-1);
|
mas01mc@292
|
27 }
|
mas01mc@292
|
28
|
mas01mc@319
|
29 Uns32T audioDB::index_from_trackInfo(Uns32T trackID, Uns32T spos, Uns32T nPntBits){
|
mas01mc@319
|
30 assert(nPntBits);
|
mas01mc@319
|
31 return (trackID << nPntBits) | spos;
|
mas01mc@292
|
32 }
|
mas01mc@292
|
33
|
mas01mc@292
|
34 /************************* LSH indexing and query initialization *****************/
|
mas01mc@292
|
35
|
mas01mc@292
|
36 char* audioDB::index_get_name(const char*dbName, double radius, Uns32T sequenceLength){
|
mas01mc@292
|
37 char* indexName = new char[MAXSTR];
|
mas01mc@292
|
38 // Attempt to make new file
|
mas01mc@292
|
39 if(strlen(dbName) > (MAXSTR - 32))
|
mas01mc@292
|
40 error("dbName is too long for LSH index filename appendages");
|
mas01mc@292
|
41 strncpy(indexName, dbName, MAXSTR);
|
mas01mc@292
|
42 sprintf(indexName+strlen(dbName), ".lsh.%019.9f.%d", radius, sequenceLength);
|
mas01mc@292
|
43 return indexName;
|
mas01mc@292
|
44 }
|
mas01mc@292
|
45
|
mas01mc@292
|
46 // return true if index exists else return false
|
mas01mc@292
|
47 int audioDB::index_exists(const char* dbName, double radius, Uns32T sequenceLength){
|
mas01mc@292
|
48 // Test to see if file exists
|
mas01mc@292
|
49 char* indexName = index_get_name(dbName, radius, sequenceLength);
|
mas01mc@292
|
50 lshfid = open (indexName, O_RDONLY);
|
mas01mc@292
|
51 delete[] indexName;
|
mas01mc@292
|
52 close(lshfid);
|
mas01mc@292
|
53
|
mas01mc@292
|
54 if(lshfid<0)
|
mas01mc@292
|
55 return false;
|
mas01mc@292
|
56 else
|
mas01mc@292
|
57 return true;
|
mas01mc@292
|
58 }
|
mas01mc@292
|
59
|
mas01mc@321
|
60 // If we are a server and have a memory-resident index, check the indexName against the resident index (using get_indexName())
|
mas01mc@321
|
61 // If they match, i.e. path+dbName_resident == path+dbName_requested, use
|
mas01mc@321
|
62 // the memory-resident index.
|
mas01mc@321
|
63 // Else allocate a new LSH instance and load the index from disk
|
mas01mc@308
|
64 LSH* audioDB::index_allocate(char* indexName, bool load_hashTables){
|
mas01mc@308
|
65 LSH* gIndx=SERVER_LSH_INDEX_SINGLETON;
|
mas01mc@308
|
66 if(isServer && gIndx && (strncmp(gIndx->get_indexName(), indexName, MAXSTR)==0) )
|
mas01mc@308
|
67 audioDB::lsh = gIndx; // Use the global SERVER resident index
|
mas01mc@308
|
68 else{
|
mas01mc@308
|
69 if(audioDB::lsh)
|
mas01mc@308
|
70 delete audioDB::lsh;
|
mas01mc@308
|
71 audioDB::lsh = new LSH(indexName, load_hashTables);
|
mas01mc@308
|
72 }
|
mas01mc@308
|
73 assert(audioDB::lsh);
|
mas01mc@308
|
74 return audioDB::lsh;
|
mas01mc@308
|
75 }
|
mas01mc@308
|
76
|
mas01mc@292
|
77 vector<vector<float> >* audioDB::index_initialize_shingles(Uns32T sz){
|
mas01mc@292
|
78 if(vv)
|
mas01mc@292
|
79 delete vv;
|
mas01mc@292
|
80 vv = new vector<vector<float> >(sz);
|
mas01mc@292
|
81 for(Uns32T i=0 ; i < sz ; i++)
|
mas01mc@292
|
82 (*vv)[i]=vector<float>(dbH->dim*sequenceLength); // allocate shingle storage
|
mas01mc@292
|
83 return vv;
|
mas01mc@292
|
84 }
|
mas01mc@292
|
85
|
mas01mc@292
|
86 /******************** LSH indexing audioDB database access forall s \in {S} ***********************/
|
mas01mc@292
|
87
|
mas01mc@292
|
88 // Prepare the AudioDB database for read access and allocate auxillary memory
|
mas01mc@292
|
89 void audioDB::index_initialize(double **snp, double **vsnp, double **spp, double **vspp, Uns32T *dvp) {
|
mas01mc@319
|
90 if (!(dbH->flags & O2_FLAG_POWER)) {
|
mas01mc@319
|
91 error("INDEXed database must be power-enabled", dbName);
|
mas01mc@319
|
92 }
|
mas01mc@319
|
93
|
mas01mc@319
|
94 double *snpp = *snp, *sppp = 0;
|
mas01mc@319
|
95
|
mas01mc@292
|
96 *dvp = dbH->length / (dbH->dim * sizeof(double)); // number of database vectors
|
mas01mc@292
|
97 *snp = new double[*dvp]; // songs norm pointer: L2 norm table for each vector
|
mas01mc@292
|
98 *spp = new double[*dvp]; // song powertable pointer
|
mas01mc@292
|
99 sppp = *spp;
|
mas01mc@319
|
100 memcpy(*snp, l2normTable, *dvp * sizeof(double));
|
mas01mc@292
|
101 memcpy(*spp, powerTable, *dvp * sizeof(double));
|
mas01mc@319
|
102
|
mas01mc@319
|
103
|
mas01mc@292
|
104 for(Uns32T i = 0; i < dbH->numFiles; i++){
|
mas01mc@292
|
105 if(trackTable[i] >= sequenceLength) {
|
mas01mc@292
|
106 sequence_sum(snpp, trackTable[i], sequenceLength);
|
mas01mc@292
|
107 sequence_sqrt(snpp, trackTable[i], sequenceLength);
|
mas01mc@292
|
108
|
mas01mc@292
|
109 sequence_sum(sppp, trackTable[i], sequenceLength);
|
mas01mc@292
|
110 sequence_average(sppp, trackTable[i], sequenceLength);
|
mas01mc@292
|
111 }
|
mas01mc@292
|
112 snpp += trackTable[i];
|
mas01mc@292
|
113 sppp += trackTable[i];
|
mas01mc@292
|
114 }
|
mas01mc@319
|
115
|
mas01mc@292
|
116 *vsnp = *snp;
|
mas01mc@292
|
117 *vspp = *spp;
|
mas01mc@319
|
118
|
mas01mc@292
|
119 // Move the feature vector read pointer to start of fetures in database
|
mas01mc@292
|
120 lseek(dbfid, dbH->dataOffset, SEEK_SET);
|
mas01mc@292
|
121 }
|
mas01mc@292
|
122
|
mas01mc@292
|
123
|
mas01mc@292
|
124 /************************ LSH indexing ***********************************/
|
mas01mc@292
|
125 void audioDB::index_index_db(const char* dbName){
|
mas01mc@292
|
126 char* newIndexName;
|
mas01mc@292
|
127 double *fvp = 0, *sNorm = 0, *snPtr = 0, *sPower = 0, *spPtr = 0;
|
mas01mc@292
|
128 Uns32T dbVectors = 0;
|
mas01mc@292
|
129
|
mas01mc@320
|
130
|
mas01mc@292
|
131 printf("INDEX: initializing header\n");
|
mas01mc@292
|
132 // Check if audioDB exists, initialize header and open database for read
|
mas01mc@292
|
133 forWrite = false;
|
mas01mc@292
|
134 initDBHeader(dbName);
|
mas01mc@292
|
135
|
mas01mc@320
|
136 if(dbH->flags & O2_FLAG_POWER)
|
mas01mc@320
|
137 usingPower = true;
|
mas01mc@320
|
138
|
mas01mc@320
|
139 if(dbH->flags & O2_FLAG_TIMES)
|
mas01mc@320
|
140 usingTimes = true;
|
mas01mc@320
|
141
|
mas01mc@292
|
142 newIndexName = index_get_name(dbName, radius, sequenceLength);
|
mas01mc@292
|
143
|
mas01mc@292
|
144 // Set unit norming flag override
|
mas01mc@292
|
145 audioDB::normalizedDistance = !audioDB::no_unit_norming;
|
mas01mc@292
|
146
|
mas01mc@320
|
147 printf("INDEX: dim %d\n", (int)dbH->dim);
|
mas01mc@292
|
148 printf("INDEX: R %f\n", radius);
|
mas01mc@292
|
149 printf("INDEX: seqlen %d\n", sequenceLength);
|
mas01mc@292
|
150 printf("INDEX: lsh_w %f\n", lsh_param_w);
|
mas01mc@292
|
151 printf("INDEX: lsh_k %d\n", lsh_param_k);
|
mas01mc@292
|
152 printf("INDEX: lsh_m %d\n", lsh_param_m);
|
mas01mc@292
|
153 printf("INDEX: lsh_N %d\n", lsh_param_N);
|
mas01mc@296
|
154 printf("INDEX: lsh_C %d\n", lsh_param_ncols);
|
mas01mc@292
|
155 printf("INDEX: lsh_b %d\n", lsh_param_b);
|
mas01mc@292
|
156 printf("INDEX: normalized? %s\n", normalizedDistance?"true":"false");
|
mas01mc@292
|
157 fflush(stdout);
|
mas01mc@292
|
158
|
mas01mc@292
|
159
|
mas01mc@292
|
160 if((lshfid = open(newIndexName,O_RDONLY))<0){
|
mas01mc@292
|
161 printf("INDEX: constructing new LSH index\n");
|
mas01mc@292
|
162 printf("INDEX: making index file %s\n", newIndexName);
|
mas01mc@292
|
163 fflush(stdout);
|
mas01mc@292
|
164 // Construct new LSH index
|
mas01mc@292
|
165 lsh = new LSH((float)lsh_param_w, lsh_param_k,
|
mas01mc@292
|
166 lsh_param_m,
|
mas01mc@292
|
167 (Uns32T)(sequenceLength*dbH->dim),
|
mas01mc@292
|
168 lsh_param_N,
|
mas01mc@292
|
169 lsh_param_ncols,
|
mas01mc@292
|
170 (float)radius);
|
mas01mc@292
|
171 assert(lsh);
|
mas01mc@292
|
172
|
mas01mc@292
|
173 Uns32T endTrack = lsh_param_b;
|
mas01mc@292
|
174 if( endTrack > dbH->numFiles)
|
mas01mc@292
|
175 endTrack = dbH->numFiles;
|
mas01mc@292
|
176 // Insert up to lsh_param_b tracks
|
mas01mc@320
|
177 if( ! (dbH->flags & O2_FLAG_LARGE_ADB) ){
|
mas01mc@320
|
178 index_initialize(&sNorm, &snPtr, &sPower, &spPtr, &dbVectors);
|
mas01mc@319
|
179 }
|
mas01mc@320
|
180 index_insert_tracks(0, endTrack, &fvp, &sNorm, &snPtr, &sPower, &spPtr);
|
mas01mc@292
|
181 lsh->serialize(newIndexName, lsh_in_core?O2_SERIAL_FILEFORMAT2:O2_SERIAL_FILEFORMAT1);
|
mas01mc@292
|
182
|
mas01mc@292
|
183 // Clean up
|
mas01mc@292
|
184 delete lsh;
|
mas01mc@308
|
185 lsh = 0;
|
mas01mc@292
|
186 close(lshfid);
|
mas01mc@292
|
187 }
|
mas01mc@292
|
188
|
mas01mc@292
|
189 // Attempt to open LSH file
|
mas01mc@292
|
190 if((lshfid = open(newIndexName,O_RDONLY))>0){
|
mas01mc@292
|
191 printf("INDEX: merging with existing LSH index\n");
|
mas01mc@292
|
192 fflush(stdout);
|
mas01mc@292
|
193
|
mas01mc@292
|
194 // Get the lsh header info and find how many tracks are inserted already
|
mas01mc@292
|
195 lsh = new LSH(newIndexName, false); // lshInCore=false to avoid loading hashTables here
|
mas01mc@292
|
196 assert(lsh);
|
mas01mc@319
|
197 Uns32T maxs = index_to_trackID(lsh->get_maxp(), lsh_n_point_bits)+1;
|
mas01mc@292
|
198 delete lsh;
|
mas01mc@308
|
199 lsh = 0;
|
mas01mc@292
|
200
|
mas01mc@292
|
201 // This allows for updating index after more tracks are inserted into audioDB
|
mas01mc@292
|
202 for(Uns32T startTrack = maxs; startTrack < dbH->numFiles; startTrack+=lsh_param_b){
|
mas01mc@292
|
203
|
mas01mc@292
|
204 Uns32T endTrack = startTrack + lsh_param_b;
|
mas01mc@292
|
205 if( endTrack > dbH->numFiles)
|
mas01mc@292
|
206 endTrack = dbH->numFiles;
|
mas01mc@292
|
207 printf("Indexing track range: %d - %d\n", startTrack, endTrack);
|
mas01mc@292
|
208 fflush(stdout);
|
mas01mc@292
|
209 lsh = new LSH(newIndexName, lsh_in_core); // Initialize core memory for LSH tables
|
mas01mc@292
|
210 assert(lsh);
|
mas01mc@292
|
211
|
mas01mc@292
|
212 // Insert up to lsh_param_b database tracks
|
mas01mc@292
|
213 index_insert_tracks(startTrack, endTrack, &fvp, &sNorm, &snPtr, &sPower, &spPtr);
|
mas01mc@292
|
214
|
mas01mc@292
|
215 // Serialize to file
|
mas01mc@292
|
216 lsh->serialize(newIndexName, lsh_in_core?O2_SERIAL_FILEFORMAT2:O2_SERIAL_FILEFORMAT1); // Serialize core LSH heap to disk
|
mas01mc@292
|
217 delete lsh;
|
mas01mc@308
|
218 lsh = 0;
|
mas01mc@292
|
219 }
|
mas01mc@292
|
220
|
mas01mc@292
|
221 close(lshfid);
|
mas01mc@292
|
222 printf("INDEX: done constructing LSH index.\n");
|
mas01mc@292
|
223 fflush(stdout);
|
mas01mc@292
|
224
|
mas01mc@292
|
225 }
|
mas01mc@292
|
226 else{
|
mas01mc@292
|
227 error("Something's wrong with LSH index file");
|
mas01mc@292
|
228 exit(1);
|
mas01mc@292
|
229 }
|
mas01mc@292
|
230
|
mas01mc@292
|
231 delete[] newIndexName;
|
mas01mc@320
|
232 delete[] sNorm;
|
mas01mc@320
|
233 delete[] sPower;
|
mas01mc@292
|
234 }
|
mas01mc@292
|
235
|
mas01mc@319
|
236
|
mas01mc@319
|
237 // initialize auxillary track data from filesystem
|
mas01mc@319
|
238 // pre-conditions:
|
mas01mc@319
|
239 // dbH->flags & O2_FLAG_LARGE_ADB
|
mas01mc@319
|
240 // feature data allocated and copied (fvp)
|
mas01mc@319
|
241 //
|
mas01mc@319
|
242 // post-conditions:
|
mas01mc@319
|
243 // allocated power data
|
mas01mc@319
|
244 // allocated l2norm data
|
mas01mc@319
|
245 //
|
mas01mc@319
|
246 void audioDB::init_track_aux_data(Uns32T trackID, double* fvp, double** sNormpp,double** snPtrp, double** sPowerp, double** spPtrp){
|
mas01mc@319
|
247 if( !(dbH->flags & O2_FLAG_LARGE_ADB) )
|
mas01mc@319
|
248 error("error: init_track_large_adb required O2_FLAG_LARGE_ADB");
|
mas01mc@319
|
249
|
mas01mc@319
|
250 // Allocate and read the power sequence
|
mas01mc@319
|
251 if(trackTable[trackID]>=sequenceLength){
|
mas01mc@321
|
252
|
mas01mc@321
|
253 char* prefixedString = new char[O2_MAXFILESTR];
|
mas01mc@321
|
254 char* tmpStr = prefixedString;
|
mas01mc@319
|
255 // Open and check dimensions of power file
|
mas01mc@321
|
256 strncpy(prefixedString, powerFileNameTable+trackID*O2_FILETABLE_ENTRY_SIZE, O2_MAXFILESTR);
|
mas01mc@321
|
257 prefix_name((char ** const)&prefixedString, adb_feature_root);
|
mas01mc@321
|
258 if(prefixedString!=tmpStr)
|
mas01mc@321
|
259 delete[] tmpStr;
|
mas01mc@321
|
260 powerfd = open(prefixedString, O_RDONLY);
|
mas01mc@319
|
261 if (powerfd < 0) {
|
mas01mc@321
|
262 error("failed to open power file", prefixedString);
|
mas01mc@319
|
263 }
|
mas01mc@319
|
264 if (fstat(powerfd, &statbuf) < 0) {
|
mas01mc@321
|
265 error("fstat error finding size of power file", prefixedString, "fstat");
|
mas01mc@319
|
266 }
|
mas01mc@319
|
267
|
mas01mc@319
|
268 if( (statbuf.st_size - sizeof(int)) / (sizeof(double)) != trackTable[trackID] )
|
mas01mc@321
|
269 error("Dimension mismatch: numPowers != numVectors", prefixedString);
|
mas01mc@320
|
270
|
mas01mc@319
|
271 *sPowerp = new double[trackTable[trackID]]; // Allocate memory for power values
|
mas01mc@319
|
272 assert(*sPowerp);
|
mas01mc@319
|
273 *spPtrp = *sPowerp;
|
mas01mc@319
|
274 insertPowerData(trackTable[trackID], powerfd, *sPowerp);
|
mas01mc@319
|
275 if (0 < powerfd) {
|
mas01mc@319
|
276 close(powerfd);
|
mas01mc@319
|
277 }
|
mas01mc@319
|
278
|
mas01mc@319
|
279 sequence_sum(*sPowerp, trackTable[trackID], sequenceLength);
|
mas01mc@319
|
280 sequence_average(*sPowerp, trackTable[trackID], sequenceLength);
|
mas01mc@319
|
281 powerTable = 0;
|
mas01mc@319
|
282
|
mas01mc@319
|
283 // Allocate and calculate the l2norm sequence
|
mas01mc@319
|
284 *sNormpp = new double[trackTable[trackID]];
|
mas01mc@319
|
285 assert(*sNormpp);
|
mas01mc@319
|
286 *snPtrp = *sNormpp;
|
mas01mc@319
|
287 unitNorm(fvp, dbH->dim, trackTable[trackID], *sNormpp);
|
mas01mc@319
|
288 sequence_sum(*sNormpp, trackTable[trackID], sequenceLength);
|
mas01mc@319
|
289 sequence_sqrt(*sNormpp, trackTable[trackID], sequenceLength);
|
mas01mc@319
|
290 }
|
mas01mc@319
|
291 }
|
mas01mc@319
|
292
|
mas01mc@292
|
293 void audioDB::index_insert_tracks(Uns32T start_track, Uns32T end_track,
|
mas01mc@292
|
294 double** fvpp, double** sNormpp,double** snPtrp,
|
mas01mc@292
|
295 double** sPowerp, double** spPtrp){
|
mas01mc@292
|
296 size_t nfv = 0;
|
mas01mc@292
|
297 double* fvp = 0; // Keep pointer for memory allocation and free() for track data
|
mas01mc@292
|
298 Uns32T trackID = 0;
|
mas01mc@292
|
299
|
mas01mc@292
|
300 VERB_LOG(1, "indexing tracks...");
|
mas01mc@292
|
301
|
mas01mc@319
|
302 int trackfd = dbfid;
|
mas01mc@292
|
303 for(trackID = start_track ; trackID < end_track ; trackID++ ){
|
mas01mc@319
|
304 if( dbH->flags & O2_FLAG_LARGE_ADB ){
|
mas01mc@321
|
305 char* prefixedString = new char[O2_MAXFILESTR];
|
mas01mc@321
|
306 char* tmpStr = prefixedString;
|
mas01mc@319
|
307 // Open and check dimensions of feature file
|
mas01mc@321
|
308 strncpy(prefixedString, featureFileNameTable+trackID*O2_FILETABLE_ENTRY_SIZE, O2_MAXFILESTR);
|
mas01mc@321
|
309 prefix_name((char ** const) &prefixedString, adb_feature_root);
|
mas01mc@321
|
310 if(prefixedString!=tmpStr)
|
mas01mc@321
|
311 delete[] tmpStr;
|
mas01mc@321
|
312 initInputFile(prefixedString, false); // nommap, file pointer at correct position
|
mas01mc@319
|
313 trackfd = infid;
|
mas01mc@319
|
314 }
|
mas01mc@319
|
315 read_data(trackfd, trackID, &fvp, &nfv); // over-writes fvp and nfv
|
mas01mc@292
|
316 *fvpp = fvp; // Protect memory allocation and free() for track data
|
mas01mc@319
|
317
|
mas01mc@319
|
318 if( dbH->flags & O2_FLAG_LARGE_ADB )
|
mas01mc@319
|
319 // Load power and calculate power and l2norm sequence sums
|
mas01mc@319
|
320 init_track_aux_data(trackID, fvp, sNormpp, snPtrp, sPowerp, spPtrp);
|
mas01mc@319
|
321
|
mas01mc@292
|
322 if(!index_insert_track(trackID, fvpp, snPtrp, spPtrp))
|
mas01mc@292
|
323 break;
|
mas01mc@319
|
324 if ( dbH->flags & O2_FLAG_LARGE_ADB ){
|
mas01mc@319
|
325 close(infid);
|
mas01mc@320
|
326 delete[] *sNormpp;
|
mas01mc@320
|
327 delete[] *sPowerp;
|
mas01mc@319
|
328 *sNormpp = *sPowerp = *snPtrp = *snPtrp = 0;
|
mas01mc@319
|
329 }
|
mas01mc@319
|
330 } // end for(trackID = start_track ; ... )
|
mas01mc@292
|
331 std::cout << "finished inserting." << endl;
|
mas01mc@292
|
332 }
|
mas01mc@292
|
333
|
mas01mc@292
|
334 int audioDB::index_insert_track(Uns32T trackID, double** fvpp, double** snpp, double** sppp){
|
mas01mc@292
|
335 // Loop over the current input track's vectors
|
mas01cr@305
|
336 Uns32T numVecs = 0;
|
mas01cr@305
|
337 if (trackTable[trackID] > O2_MAXTRACKLEN) {
|
mas01cr@305
|
338 if (O2_MAXTRACKLEN < sequenceLength - 1) {
|
mas01cr@305
|
339 numVecs = 0;
|
mas01cr@305
|
340 } else {
|
mas01cr@305
|
341 numVecs = O2_MAXTRACKLEN - sequenceLength + 1;
|
mas01cr@305
|
342 }
|
mas01cr@305
|
343 } else {
|
mas01cr@305
|
344 if (trackTable[trackID] < sequenceLength - 1) {
|
mas01cr@305
|
345 numVecs = 0;
|
mas01cr@305
|
346 } else {
|
mas01cr@305
|
347 numVecs = trackTable[trackID] - sequenceLength + 1;
|
mas01cr@305
|
348 }
|
mas01cr@305
|
349 }
|
mas01mc@292
|
350
|
mas01mc@319
|
351 Uns32T numVecsAboveThreshold = 0, collisionCount = 0;
|
mas01mc@319
|
352 if(numVecs){
|
mas01mc@319
|
353 vv = index_initialize_shingles(numVecs);
|
mas01mc@319
|
354
|
mas01mc@319
|
355 for( Uns32T pointID = 0 ; pointID < numVecs; pointID++ )
|
mas01mc@319
|
356 index_make_shingle(vv, pointID, *fvpp, dbH->dim, sequenceLength);
|
mas01mc@319
|
357
|
mas01mc@319
|
358 numVecsAboveThreshold = index_norm_shingles(vv, *snpp, *sppp);
|
mas01mc@319
|
359 collisionCount = index_insert_shingles(vv, trackID, *sppp);
|
mas01mc@319
|
360 }
|
mas01mc@292
|
361 float meanCollisionCount = numVecsAboveThreshold?(float)collisionCount/numVecsAboveThreshold:0;
|
mas01mc@292
|
362
|
mas01mc@292
|
363 /* index_norm_shingles() only goes as far as the end of the
|
mas01mc@292
|
364 sequence, which is right, but the space allocated is for the
|
mas01mc@292
|
365 whole track. */
|
mas01mc@292
|
366
|
mas01mc@292
|
367 /* But numVecs will be <trackTable[track] if trackTable[track]>O2_MAXTRACKLEN
|
mas01mc@292
|
368 * So let's be certain the pointers are in the correct place
|
mas01mc@292
|
369 */
|
mas01mc@292
|
370
|
mas01mc@319
|
371 if( !(dbH->flags & O2_FLAG_LARGE_ADB) ){
|
mas01mc@319
|
372 *snpp += trackTable[trackID];
|
mas01mc@319
|
373 *sppp += trackTable[trackID];
|
mas01mc@319
|
374 *fvpp += trackTable[trackID] * dbH->dim;
|
mas01mc@319
|
375 }
|
mas01mc@292
|
376
|
mas01mc@292
|
377 std::cout << " n=" << trackTable[trackID] << " n'=" << numVecsAboveThreshold << " E[#c]=" << lsh->get_mean_collision_rate() << " E[#p]=" << meanCollisionCount << endl;
|
mas01mc@292
|
378 std::cout.flush();
|
mas01mc@292
|
379 return true;
|
mas01mc@292
|
380 }
|
mas01mc@292
|
381
|
mas01mc@292
|
382 Uns32T audioDB::index_insert_shingles(vector<vector<float> >* vv, Uns32T trackID, double* spp){
|
mas01mc@292
|
383 Uns32T collisionCount = 0;
|
mas01mc@292
|
384 cout << "[" << trackID << "]" << fileTable+trackID*O2_FILETABLE_ENTRY_SIZE;
|
mas01mc@323
|
385 for( Uns32T pointID=0 ; pointID < (*vv).size(); pointID+=sequenceHop){
|
mas01mc@323
|
386 if(!use_absolute_threshold || (use_absolute_threshold && (*spp >= absolute_threshold)))
|
mas01mc@319
|
387 collisionCount += lsh->insert_point((*vv)[pointID], index_from_trackInfo(trackID, pointID, lsh_n_point_bits));
|
mas01mc@323
|
388 spp+=sequenceHop;
|
mas01mc@311
|
389 }
|
mas01mc@292
|
390 return collisionCount;
|
mas01mc@292
|
391 }
|
mas01mc@292
|
392
|
mas01mc@292
|
393 /********************* LSH shingle construction ***************************/
|
mas01mc@292
|
394
|
mas01mc@292
|
395 // Construct shingles out of a feature matrix
|
mas01mc@292
|
396 // inputs:
|
mas01mc@292
|
397 // idx is vector index in feature matrix
|
mas01mc@292
|
398 // fvp is base feature matrix pointer double* [numVecs x dbH->dim]
|
mas01mc@292
|
399 //
|
mas01mc@292
|
400 // pre-conditions:
|
mas01mc@292
|
401 // dbH->dim
|
mas01mc@292
|
402 // sequenceLength
|
mas01mc@292
|
403 // idx < numVectors - sequenceLength + 1
|
mas01mc@292
|
404 //
|
mas01mc@292
|
405 // post-conditions:
|
mas01mc@292
|
406 // (*vv)[idx] contains a shingle with dbH->dim*sequenceLength float values
|
mas01mc@292
|
407
|
mas01mc@292
|
408 void audioDB::index_make_shingle(vector<vector<float> >* vv, Uns32T idx, double* fvp, Uns32T dim, Uns32T seqLen){
|
mas01mc@292
|
409 assert(idx<(*vv).size());
|
mas01mc@292
|
410 vector<float>::iterator ve = (*vv)[idx].end();
|
mas01mc@292
|
411 vi=(*vv)[idx].begin(); // shingle iterator
|
mas01mc@292
|
412 // First feature vector in shingle
|
mas01mc@292
|
413 if(idx==0){
|
mas01mc@292
|
414 while(vi!=ve)
|
mas01mc@292
|
415 *vi++ = (float)(*fvp++);
|
mas01mc@292
|
416 }
|
mas01mc@292
|
417 // Not first feature vector in shingle
|
mas01mc@292
|
418 else{
|
mas01mc@292
|
419 vector<float>::iterator ui=(*vv)[idx-1].begin() + dim; // previous shingle iterator
|
mas01mc@292
|
420 // Previous seqLen-1 dim-vectors
|
mas01mc@292
|
421 while(vi!=ve-dim)
|
mas01mc@292
|
422 *vi++=*ui++;
|
mas01mc@292
|
423 // Move data pointer to next feature vector
|
mas01mc@292
|
424 fvp += ( seqLen + idx - 1 ) * dim ;
|
mas01mc@292
|
425 // New d-vector
|
mas01mc@292
|
426 while(vi!=ve)
|
mas01mc@292
|
427 *vi++ = (float)(*fvp++);
|
mas01mc@292
|
428 }
|
mas01mc@292
|
429 }
|
mas01mc@292
|
430
|
mas01mc@292
|
431 // norm shingles
|
mas01mc@292
|
432 // in-place norming, no deletions
|
mas01mc@292
|
433 // If using power, return number of shingles above power threshold
|
mas01mc@292
|
434 int audioDB::index_norm_shingles(vector<vector<float> >* vv, double* snp, double* spp){
|
mas01mc@292
|
435 int z = 0; // number of above-threshold shingles
|
mas01mc@292
|
436 float l2norm;
|
mas01mc@292
|
437 double power;
|
mas01mc@292
|
438 float oneOverRadius = 1./(float)sqrt(radius); // Passed radius is really radius^2
|
mas01mc@292
|
439 float oneOverSqrtl2NormDivRad = oneOverRadius;
|
mas01mc@292
|
440 if(!spp)
|
mas01mc@292
|
441 error("LSH indexing and query requires a power feature using -w or -W");
|
mas01mc@292
|
442 Uns32T shingleSize = sequenceLength*dbH->dim;
|
mas01mc@292
|
443 for(Uns32T a=0; a<(*vv).size(); a++){
|
mas01mc@292
|
444 l2norm = (float)(*snp++);
|
mas01mc@292
|
445 if(audioDB::normalizedDistance)
|
mas01mc@292
|
446 oneOverSqrtl2NormDivRad = (1./l2norm)*oneOverRadius;
|
mas01mc@292
|
447
|
mas01mc@292
|
448 for(Uns32T b=0; b < shingleSize ; b++)
|
mas01mc@292
|
449 (*vv)[a][b]*=oneOverSqrtl2NormDivRad;
|
mas01mc@292
|
450
|
mas01mc@292
|
451 power = *spp++;
|
mas01mc@292
|
452 if(use_absolute_threshold){
|
mas01mc@292
|
453 if ( power >= absolute_threshold )
|
mas01mc@292
|
454 z++;
|
mas01mc@292
|
455 }
|
mas01mc@292
|
456 else
|
mas01mc@292
|
457 z++;
|
mas01mc@292
|
458 }
|
mas01mc@292
|
459 return z;
|
mas01mc@292
|
460 }
|
mas01mc@292
|
461
|
mas01mc@292
|
462
|
mas01mc@292
|
463 /*********************** LSH retrieval ****************************/
|
mas01mc@292
|
464
|
mas01mc@292
|
465
|
mas01mc@292
|
466 // return true if indexed query performed else return false
|
mas01mc@292
|
467 int audioDB::index_init_query(const char* dbName){
|
mas01mc@292
|
468
|
mas01mc@292
|
469 if(!(index_exists(dbName, radius, sequenceLength)))
|
mas01mc@292
|
470 return false;
|
mas01mc@292
|
471
|
mas01mc@292
|
472 char* indexName = index_get_name(dbName, radius, sequenceLength);
|
mas01mc@292
|
473
|
mas01mc@292
|
474 // Test to see if file exists
|
mas01mc@292
|
475 if((lshfid = open (indexName, O_RDONLY)) < 0){
|
mas01mc@292
|
476 delete[] indexName;
|
mas01mc@292
|
477 return false;
|
mas01mc@292
|
478 }
|
mas01mc@292
|
479
|
mas01mc@308
|
480 lsh = index_allocate(indexName, false); // Get the header only here
|
mas01mc@292
|
481 sequenceLength = lsh->get_lshHeader()->dataDim / dbH->dim; // shingleDim / vectorDim
|
mas01mc@292
|
482
|
mas01mc@311
|
483 if(lsh!=SERVER_LSH_INDEX_SINGLETON){
|
mas01mc@308
|
484 if( fabs(radius - lsh->get_radius())>fabs(O2_DISTANCE_TOLERANCE))
|
mas01mc@308
|
485 printf("*** Warning: adb_radius (%f) != lsh_radius (%f) ***\n", radius, lsh->get_radius());
|
mas01mc@320
|
486 printf("INDEX: dim %d\n", (int)dbH->dim);
|
mas01mc@308
|
487 printf("INDEX: R %f\n", lsh->get_radius());
|
mas01mc@308
|
488 printf("INDEX: seqlen %d\n", sequenceLength);
|
mas01mc@308
|
489 printf("INDEX: w %f\n", lsh->get_lshHeader()->get_binWidth());
|
mas01mc@308
|
490 printf("INDEX: k %d\n", lsh->get_lshHeader()->get_numFuns());
|
mas01mc@308
|
491 printf("INDEX: L (m*(m-1))/2 %d\n", lsh->get_lshHeader()->get_numTables());
|
mas01mc@308
|
492 printf("INDEX: N %d\n", lsh->get_lshHeader()->get_numRows());
|
mas01mc@319
|
493 printf("INDEX: s %d\n", index_to_trackID(lsh->get_maxp(), lsh_n_point_bits));
|
mas01mc@308
|
494 printf("INDEX: Opened LSH index file %s\n", indexName);
|
mas01mc@308
|
495 fflush(stdout);
|
mas01mc@308
|
496 }
|
mas01mc@292
|
497
|
mas01mc@292
|
498 // Check to see if we are loading hash tables into core, and do so if true
|
mas01mc@292
|
499 if((lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2) || lsh_in_core){
|
mas01mc@308
|
500 if(SERVER_LSH_INDEX_SINGLETON)
|
mas01mc@308
|
501 fprintf(stderr,"INDEX: using persistent hash tables: %s\n", lsh->get_indexName());
|
mas01mc@308
|
502 else
|
mas01mc@308
|
503 printf("INDEX: loading hash tables into core %s\n", (lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2)?"FORMAT2":"FORMAT1");
|
mas01mc@308
|
504 lsh = index_allocate(indexName, true);
|
mas01mc@292
|
505 }
|
mas01mc@292
|
506
|
mas01mc@292
|
507 delete[] indexName;
|
mas01mc@292
|
508 return true;
|
mas01mc@292
|
509 }
|
mas01mc@292
|
510
|
mas01mc@292
|
511 // *Static* approximate NN point reporter callback method for lshlib
|
mas01mc@292
|
512 void audioDB::index_add_point_approximate(void* instancePtr, Uns32T pointID, Uns32T qpos, float dist){
|
mas01mc@292
|
513 assert(instancePtr); // We need an instance for this callback
|
mas01mc@292
|
514 audioDB* myself = (audioDB*) instancePtr; // Use explicit cast to recover "this" instance
|
mas01mc@319
|
515 Uns32T trackID = index_to_trackID(pointID, myself->lsh_n_point_bits);
|
mas01mc@319
|
516 Uns32T spos = index_to_trackPos(pointID, myself->lsh_n_point_bits);
|
mas01mc@292
|
517 // Skip identity in query_from_key
|
mas01mc@292
|
518 if( !myself->query_from_key || (myself->query_from_key && ( trackID != myself->query_from_key_index )) )
|
mas01mc@292
|
519 myself->reporter->add_point(trackID, qpos, spos, dist);
|
mas01mc@292
|
520 }
|
mas01mc@292
|
521
|
mas01mc@292
|
522 // *Static* exact NN point reporter callback method for lshlib
|
mas01mc@292
|
523 // Maintain a queue of points to pass to query_points() for exact evaluation
|
mas01mc@292
|
524 void audioDB::index_add_point_exact(void* instancePtr, Uns32T pointID, Uns32T qpos, float dist){
|
mas01mc@292
|
525 assert(instancePtr); // We need an instance for this callback
|
mas01mc@292
|
526 audioDB* myself = (audioDB*) instancePtr; // Use explicit cast to recover "this" instance
|
mas01mc@319
|
527 Uns32T trackID = index_to_trackID(pointID, myself->lsh_n_point_bits);
|
mas01mc@319
|
528 Uns32T spos = index_to_trackPos(pointID, myself->lsh_n_point_bits);
|
mas01mc@292
|
529 // Skip identity in query_from_key
|
mas01mc@292
|
530 if( !myself->query_from_key || (myself->query_from_key && ( trackID != myself->query_from_key_index )) )
|
mas01mc@292
|
531 myself->index_insert_exact_evaluation_queue(trackID, qpos, spos);
|
mas01mc@292
|
532 }
|
mas01mc@292
|
533
|
mas01mc@292
|
534 void audioDB::initialize_exact_evalutation_queue(){
|
mas01mc@292
|
535 if(exact_evaluation_queue)
|
mas01mc@292
|
536 delete exact_evaluation_queue;
|
mas01mc@292
|
537 exact_evaluation_queue = new priority_queue<PointPair, std::vector<PointPair>, std::less<PointPair> >;
|
mas01mc@292
|
538 }
|
mas01mc@292
|
539
|
mas01mc@292
|
540 void audioDB::index_insert_exact_evaluation_queue(Uns32T trackID, Uns32T qpos, Uns32T spos){
|
mas01mc@292
|
541 PointPair p(trackID, qpos, spos);
|
mas01mc@292
|
542 exact_evaluation_queue->push(p);
|
mas01mc@292
|
543 }
|
mas01mc@292
|
544
|
mas01mc@292
|
545 // return 0: if index does not exist
|
mas01mc@292
|
546 // return nqv: if index exists
|
mas01mc@292
|
547 int audioDB::index_query_loop(const char* dbName, Uns32T queryIndex) {
|
mas01mc@292
|
548
|
mas01mc@320
|
549 unsigned int numVectors = 0;
|
mas01mc@320
|
550 double *query = 0, *query_data = 0;
|
mas01mc@320
|
551 double *qNorm = 0, *qnPtr = 0, *qPower = 0, *qpPtr = 0;
|
mas01mc@320
|
552 double meanQdur = 0;
|
mas01mc@292
|
553 void (*add_point_func)(void*,Uns32T,Uns32T,float);
|
mas01mc@292
|
554
|
mas01mc@292
|
555 // Set the point-reporter callback based on the value of lsh_exact
|
mas01mc@292
|
556 if(lsh_exact){
|
mas01mc@292
|
557 initialize_exact_evalutation_queue();
|
mas01mc@292
|
558 add_point_func = &index_add_point_exact;
|
mas01mc@292
|
559 }
|
mas01mc@292
|
560 else
|
mas01mc@292
|
561 add_point_func = &index_add_point_approximate;
|
mas01mc@292
|
562
|
mas01mc@292
|
563 if(!index_init_query(dbName)) // sets-up LSH index structures for querying
|
mas01mc@292
|
564 return 0;
|
mas01mc@292
|
565
|
mas01mc@292
|
566 char* database = index_get_name(dbName, radius, sequenceLength);
|
mas01mc@292
|
567
|
mas01mc@292
|
568 if(query_from_key)
|
mas01mc@292
|
569 set_up_query_from_key(&query_data, &query, &qNorm, &qnPtr, &qPower, &qpPtr, &meanQdur, &numVectors, queryIndex);
|
mas01mc@292
|
570 else
|
mas01mc@292
|
571 set_up_query(&query_data, &query, &qNorm, &qnPtr, &qPower, &qpPtr, &meanQdur, &numVectors); // get query vectors
|
mas01mc@292
|
572
|
mas01mc@292
|
573 VERB_LOG(1, "retrieving tracks...");
|
mas01mc@292
|
574
|
mas01mc@292
|
575 assert(pointNN>0 && pointNN<=O2_MAXNN);
|
mas01mc@292
|
576 assert(trackNN>0 && trackNN<=O2_MAXNN);
|
mas01mc@292
|
577
|
mas01mc@292
|
578 gettimeofday(&tv1, NULL);
|
mas01mc@292
|
579 // query vector index
|
mas01mc@292
|
580 Uns32T Nq = (numVectors>O2_MAXTRACKLEN?O2_MAXTRACKLEN:numVectors) - sequenceLength + 1;
|
mas01mc@292
|
581 vv = index_initialize_shingles(Nq); // allocate memory to copy query vectors to shingles
|
mas01mc@292
|
582 cout << "Nq=" << Nq; cout.flush();
|
mas01mc@292
|
583 // Construct shingles from query features
|
mas01mc@292
|
584 for( Uns32T pointID = 0 ; pointID < Nq ; pointID++ )
|
mas01mc@292
|
585 index_make_shingle(vv, pointID, query, dbH->dim, sequenceLength);
|
mas01mc@292
|
586
|
mas01mc@292
|
587 // Normalize query vectors
|
mas01mc@292
|
588 Uns32T numVecsAboveThreshold = index_norm_shingles( vv, qnPtr, qpPtr );
|
mas01mc@292
|
589 cout << " Nq'=" << numVecsAboveThreshold << endl; cout.flush();
|
mas01mc@292
|
590
|
mas01mc@292
|
591 // Nq contains number of inspected points in query file,
|
mas01mc@292
|
592 // numVecsAboveThreshold is number of points with power >= absolute_threshold
|
mas01mc@292
|
593 double* qpp = qpPtr; // Keep original qpPtr for possible exact evaluation
|
mas01mc@292
|
594 if(usingQueryPoint && numVecsAboveThreshold){
|
mas01mc@292
|
595 if((lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2) || lsh_in_core)
|
mas01mc@292
|
596 lsh->retrieve_point((*vv)[0], queryPoint, add_point_func, (void*)this);
|
mas01mc@292
|
597 else
|
mas01mc@292
|
598 lsh->serial_retrieve_point(database, (*vv)[0], queryPoint, add_point_func, (void*)this);
|
mas01mc@292
|
599 }
|
mas01mc@292
|
600 else if(numVecsAboveThreshold)
|
mas01mc@292
|
601 for( Uns32T pointID = 0 ; pointID < Nq; pointID++ )
|
mas01mc@292
|
602 if(!use_absolute_threshold || (use_absolute_threshold && (*qpp++ >= absolute_threshold)))
|
mas01mc@292
|
603 if((lsh->get_lshHeader()->flags&O2_SERIAL_FILEFORMAT2) || lsh_in_core)
|
mas01mc@292
|
604 lsh->retrieve_point((*vv)[pointID], pointID, add_point_func, (void*)this);
|
mas01mc@292
|
605 else
|
mas01mc@292
|
606 lsh->serial_retrieve_point(database, (*vv)[pointID], pointID, add_point_func, (void*)this);
|
mas01mc@292
|
607
|
mas01mc@292
|
608 if(lsh_exact)
|
mas01mc@292
|
609 // Perform exact distance computation on point pairs in exact_evaluation_queue
|
mas01mc@292
|
610 query_loop_points(query, qnPtr, qpPtr, meanQdur, numVectors);
|
mas01mc@292
|
611
|
mas01mc@292
|
612 gettimeofday(&tv2,NULL);
|
mas01mc@292
|
613 VERB_LOG(1,"elapsed time: %ld msec\n",
|
mas01mc@292
|
614 (tv2.tv_sec*1000 + tv2.tv_usec/1000) -
|
mas01mc@292
|
615 (tv1.tv_sec*1000 + tv1.tv_usec/1000))
|
mas01mc@292
|
616
|
mas01mc@292
|
617 // Close the index file
|
mas01mc@292
|
618 close(lshfid);
|
mas01mc@292
|
619
|
mas01mc@292
|
620 // Clean up
|
mas01mc@292
|
621 if(query_data)
|
mas01mc@292
|
622 delete[] query_data;
|
mas01mc@292
|
623 if(qNorm)
|
mas01mc@292
|
624 delete[] qNorm;
|
mas01mc@292
|
625 if(qPower)
|
mas01mc@292
|
626 delete[] qPower;
|
mas01mc@292
|
627 if(database)
|
mas01mc@292
|
628 delete[] database;
|
mas01mc@292
|
629
|
mas01mc@292
|
630 return Nq;
|
mas01mc@292
|
631 }
|
mas01mc@292
|
632
|