annotate lshlib.cpp @ 534:57e459f62788

Removed LSH_N_POINT_BITS coding for LSH index. Now uses binay search via STL lower_bound to locate tracks and positions from global pointID searching over cumulative track lengths. MERGE of branches/multiprobeLSH -r 819:821 onto trunk. This is a non backward-compatible change; WARNING generated on attempt to use INDEXING with older audioDB databases. Only INDEXES are broken, not ADB instances.
author mas01mc
date Wed, 04 Feb 2009 10:45:57 +0000
parents 01cb73dfcddd
children fdcd436d7cbd 9119f2fa3efe
rev   line source
mas01mc@292 1 #include "lshlib.h"
mas01mc@292 2
mas01mc@292 3
mas01mc@292 4 void err(char*s){cout << s << endl;exit(2);}
mas01mc@292 5
mas01mc@292 6 Uns32T get_page_logn(){
mas01mc@292 7 int pagesz = (int)sysconf(_SC_PAGESIZE);
mas01mc@292 8 return (Uns32T)log2((double)pagesz);
mas01mc@292 9 }
mas01mc@292 10
mas01cr@370 11 unsigned align_up(unsigned x, unsigned w) { return (((x) + ((1<<w)-1)) & ~((1<<w)-1)); }
mas01mc@292 12
mas01mc@292 13 void H::error(const char* a, const char* b, const char *sysFunc) {
mas01mc@292 14 cerr << a << ": " << b << endl;
mas01mc@292 15 if (sysFunc) {
mas01mc@292 16 perror(sysFunc);
mas01mc@292 17 }
mas01mc@292 18 exit(1);
mas01mc@292 19 }
mas01mc@292 20
mas01mc@293 21 H::H(){
mas01mc@293 22 // Delay initialization of lsh functions until we know the parameters
mas01mc@293 23 }
mas01mc@293 24
mas01mc@293 25 H::H(Uns32T kk, Uns32T mm, Uns32T dd, Uns32T NN, Uns32T CC, float ww, float rr):
mas01mc@292 26 #ifdef USE_U_FUNCTIONS
mas01mc@292 27 use_u_functions(true),
mas01mc@292 28 #else
mas01mc@292 29 use_u_functions(false),
mas01mc@292 30 #endif
mas01mc@293 31 maxp(0),
mas01mc@292 32 bucketCount(0),
mas01mc@292 33 pointCount(0),
mas01mc@292 34 N(NN),
mas01mc@292 35 C(CC),
mas01mc@292 36 k(kk),
mas01mc@292 37 m(mm),
mas01mc@293 38 L((mm*(mm-1))/2),
mas01mc@293 39 d(dd),
mas01mc@293 40 w(ww),
mas01mc@293 41 radius(rr)
mas01mc@292 42 {
mas01mc@292 43
mas01mc@292 44 if(m<2){
mas01mc@292 45 m=2;
mas01mc@292 46 L=1; // check value of L
mas01mc@292 47 cout << "warning: setting m=2, L=1" << endl;
mas01mc@292 48 }
mas01mc@292 49 if(use_u_functions && k%2){
mas01mc@292 50 k++; // make sure k is even
mas01mc@292 51 cout << "warning: setting k even" << endl;
mas01mc@292 52 }
mas01mc@293 53
mas01mc@293 54 // We have the necessary parameters, so construct hashfunction datastructures
mas01mc@293 55 initialize_lsh_functions();
mas01mc@292 56 }
mas01mc@292 57
mas01mc@293 58 void H::initialize_lsh_functions(){
mas01mc@292 59 H::P = UH_PRIME_DEFAULT;
mas01mc@292 60
mas01mc@292 61 /* FIXME: don't use time(); instead use /dev/random or similar */
mas01mc@292 62 /* FIXME: write out the seed somewhere, so that we can get
mas01mc@292 63 repeatability */
mas01mc@292 64 #ifdef MT19937
mas01mc@292 65 init_genrand(time(NULL));
mas01mc@292 66 #else
mas01mc@292 67 srand(time(NULL)); // seed random number generator
mas01mc@292 68 #endif
mas01mc@293 69 Uns32T i,j, kk;
mas01mc@293 70 #ifdef USE_U_FUNCTIONS
mas01mc@293 71 H::A = new float**[ H::m ]; // m x k x d random projectors
mas01mc@293 72 H::b = new float*[ H::m ]; // m x k random biases
mas01mc@293 73 #else
mas01mc@293 74 H::A = new float**[ H::L ]; // m x k x d random projectors
mas01mc@293 75 H::b = new float*[ H::L ]; // m x k random biases
mas01mc@293 76 #endif
mas01mc@293 77 H::g = new Uns32T*[ H::L ]; // L x k random projections
mas01mc@293 78 assert( H::g && H::A && H::b ); // failure
mas01mc@293 79 #ifdef USE_U_FUNCTIONS
mas01mc@293 80 // Use m \times u_i functions \in R^{(k/2) \times (d)}
mas01mc@293 81 // Combine to make L=m(m-1)/2 hash functions \in R^{k \times d}
mas01mc@293 82 for( j = 0; j < H::m ; j++ ){ // m functions u_i(v)
mas01mc@293 83 H::A[j] = new float*[ H::k/2 ]; // k/2 x d 2-stable distribution coefficients
mas01mc@293 84 H::b[j] = new float[ H::k/2 ]; // bias
mas01mc@293 85 assert( H::A[j] && H::b[j] ); // failure
mas01mc@293 86 for( kk = 0; kk < H::k/2 ; kk++ ){
mas01mc@293 87 H::A[j][kk] = new float[ H::d ];
mas01mc@293 88 assert( H::A[j][kk] ); // failure
mas01mc@293 89 for(Uns32T i = 0 ; i < H::d ; i++ )
mas01mc@293 90 H::A[j][kk][i] = H::randn(); // Normal
mas01mc@293 91 H::b[j][kk] = H::ranf()*H::w; // Uniform
mas01mc@293 92 }
mas01mc@293 93 }
mas01mc@293 94 #else
mas01mc@293 95 // Use m \times u_i functions \in R^{k \times (d)}
mas01mc@293 96 // Combine to make L=m(m-1)/2 hash functions \in R^{k \times d}
mas01mc@293 97 for( j = 0; j < H::L ; j++ ){ // m functions u_i(v)
mas01mc@293 98 H::A[j] = new float*[ H::k ]; // k x d 2-stable distribution coefficients
mas01mc@293 99 H::b[j] = new float[ H::k ]; // bias
mas01mc@293 100 assert( H::A[j] && H::b[j] ); // failure
mas01mc@293 101 for( kk = 0; kk < H::k ; kk++ ){
mas01mc@293 102 H::A[j][kk] = new float[ H::d ];
mas01mc@293 103 assert( H::A[j][kk] ); // failure
mas01mc@293 104 for(Uns32T i = 0 ; i < H::d ; i++ )
mas01mc@293 105 H::A[j][kk][i] = H::randn(); // Normal
mas01mc@293 106 H::b[j][kk] = H::ranf()*H::w; // Uniform
mas01mc@293 107 }
mas01mc@293 108 }
mas01mc@293 109 #endif
mas01mc@293 110
mas01mc@293 111 // Storage for LSH hash function output (Uns32T)
mas01mc@293 112 for( j = 0 ; j < H::L ; j++ ){ // L functions g_j(u_a, u_b) a,b \in nchoosek(m,2)
mas01mc@293 113 H::g[j] = new Uns32T[ H::k ]; // k x 32-bit hash values, gj(v)=[x0 x1 ... xk-1] xk \in Z
mas01mc@293 114 assert( H::g[j] );
mas01mc@292 115 }
mas01mc@292 116
mas01mc@293 117 // LSH Hash tables
mas01mc@293 118 H::h = new bucket**[ H::L ];
mas01mc@293 119 assert( H::h );
mas01mc@292 120 for( j = 0 ; j < H::L ; j++ ){
mas01mc@292 121 H::h[j] = new bucket*[ H::N ];
mas01mc@292 122 assert( H::h[j] );
mas01mc@292 123 for( i = 0 ; i < H::N ; i++)
mas01mc@292 124 H::h[j][i] = 0;
mas01mc@292 125 }
mas01mc@293 126
mas01mc@293 127 // Standard hash functions
mas01mc@293 128 H::r1 = new Uns32T*[ H::L ];
mas01mc@293 129 H::r2 = new Uns32T*[ H::L ];
mas01mc@293 130 assert( H::r1 && H::r2 ); // failure
mas01mc@293 131 for( j = 0 ; j < H::L ; j++ ){
mas01mc@293 132 H::r1[ j ] = new Uns32T[ H::k ];
mas01mc@293 133 H::r2[ j ] = new Uns32T[ H::k ];
mas01mc@293 134 assert( H::r1[j] && H::r2[j] ); // failure
mas01mc@293 135 for( i = 0; i<H::k; i++){
mas01mc@293 136 H::r1[j][i] = randr();
mas01mc@293 137 H::r2[j][i] = randr();
mas01mc@293 138 }
mas01mc@293 139 }
mas01mc@293 140
mas01mc@534 141 // Storage for whole or partial function evaluation depending on USE_U_FUNCTIONS
mas01mc@293 142 H::initialize_partial_functions();
mas01mc@293 143 }
mas01mc@293 144
mas01mc@293 145 void H::initialize_partial_functions(){
mas01mc@293 146
mas01mc@293 147 #ifdef USE_U_FUNCTIONS
mas01mc@293 148 H::uu = vector<vector<Uns32T> >(H::m);
mas01mc@293 149 for( Uns32T aa=0 ; aa < H::m ; aa++ )
mas01mc@293 150 H::uu[aa] = vector<Uns32T>( H::k/2 );
mas01mc@293 151 #endif
mas01mc@293 152 }
mas01mc@293 153
mas01mc@293 154
mas01mc@293 155 // Generate z ~ N(0,1)
mas01mc@293 156 float H::randn(){
mas01mc@293 157 // Box-Muller
mas01mc@293 158 float x1, x2;
mas01mc@293 159 do{
mas01mc@293 160 x1 = ranf();
mas01mc@293 161 } while (x1 == 0); // cannot take log of 0
mas01mc@293 162 x2 = ranf();
mas01mc@293 163 float z;
mas01mc@293 164 z = sqrtf(-2.0 * logf(x1)) * cosf(2.0 * M_PI * x2);
mas01mc@293 165 return z;
mas01mc@293 166 }
mas01mc@293 167
mas01mc@293 168 float H::ranf(){
mas01mc@293 169 #ifdef MT19937
mas01mc@293 170 return (float) genrand_real2();
mas01mc@293 171 #else
mas01mc@293 172 return (float)( (double)rand() / ((double)(RAND_MAX)+(double)(1)) );
mas01mc@293 173 #endif
mas01mc@293 174 }
mas01mc@293 175
mas01mc@293 176 // range is 1..2^29
mas01mc@293 177 /* FIXME: that looks like an ... odd range. Still. */
mas01mc@293 178 Uns32T H::randr(){
mas01mc@293 179 #ifdef MT19937
mas01mc@293 180 return (Uns32T)((genrand_int32() >> 3) + 1);
mas01mc@293 181 #else
mas01mc@293 182 return (Uns32T) ((rand() >> 2) + 1);
mas01mc@293 183 #endif
mas01mc@292 184 }
mas01mc@292 185
mas01mc@292 186 // Destruct hash tables
mas01mc@292 187 H::~H(){
mas01mc@293 188 Uns32T i,j,kk;
mas01mc@340 189 bucket** pp;
mas01mc@293 190 #ifdef USE_U_FUNCTIONS
mas01mc@293 191 for( j = 0 ; j < H::m ; j++ ){
mas01mc@293 192 for( kk = 0 ; kk < H::k/2 ; kk++ )
mas01mc@293 193 delete[] A[j][kk];
mas01mc@293 194 delete[] A[j];
mas01mc@293 195 }
mas01mc@293 196 delete[] A;
mas01mc@293 197 for( j = 0 ; j < H::m ; j++ )
mas01mc@293 198 delete[] b[j];
mas01mc@293 199 delete[] b;
mas01mc@293 200 #else
mas01mc@293 201 for( j = 0 ; j < H::L ; j++ ){
mas01mc@293 202 for( kk = 0 ; kk < H::k ; kk++ )
mas01mc@293 203 delete[] A[j][kk];
mas01mc@293 204 delete[] A[j];
mas01mc@293 205 }
mas01mc@293 206 delete[] A;
mas01mc@293 207 for( j = 0 ; j < H::L ; j++ )
mas01mc@293 208 delete[] b[j];
mas01mc@293 209 delete[] b;
mas01mc@293 210 #endif
mas01mc@293 211
mas01mc@293 212 for( j = 0 ; j < H::L ; j++ )
mas01mc@293 213 delete[] g[j];
mas01mc@293 214 delete[] g;
mas01mc@292 215 for( j=0 ; j < H::L ; j++ ){
mas01mc@292 216 delete[] H::r1[ j ];
mas01mc@292 217 delete[] H::r2[ j ];
mas01mc@340 218 for(i = 0; i< H::N ; i++){
mas01mc@340 219 pp = 0;
mas01mc@340 220 #ifdef LSH_CORE_ARRAY
mas01mc@340 221 if(H::h[ j ][ i ])
mas01mc@340 222 if(H::h[ j ][ i ]->t2 & LSH_CORE_ARRAY_BIT){
mas01mc@340 223 pp = get_pointer_to_bucket_linked_list(H::h[ j ][ i ]);
mas01mc@340 224 if(*pp){
mas01mc@344 225 (*pp)->snext.ptr=0; // Head of list uses snext as a non-pointer value
mas01mc@340 226 delete *pp; // Now the destructor can do its work properly
mas01mc@340 227 }
mas01mc@340 228 free(H::h[ j ][ i ]->next);
mas01mc@340 229 H::h[ j ][ i ]->next = 0; // Zero next pointer
mas01mc@344 230 H::h[ j ][ i ]->snext.ptr = 0; // Zero head-of-list pointer as above
mas01mc@340 231 }
mas01mc@340 232 #endif
mas01mc@292 233 delete H::h[ j ][ i ];
mas01mc@340 234 }
mas01mc@292 235 delete[] H::h[ j ];
mas01mc@292 236 }
mas01mc@474 237 delete[] H::r1;
mas01mc@474 238 delete[] H::r2;
mas01mc@474 239 delete[] H::h;
mas01mc@292 240 }
mas01mc@292 241
mas01mc@292 242
mas01mc@293 243 // Compute all hash functions for vector v
mas01mc@293 244 // #ifdef USE_U_FUNCTIONS use Combination of m \times h_i \in R^{(k/2) \times d}
mas01mc@293 245 // to make L \times g_j functions \in Z^k
mas01mc@293 246 void H::compute_hash_functions(vector<float>& v){ // v \in R^d
mas01mc@293 247 float iw = 1. / H::w; // hash bucket width
mas01mc@293 248 Uns32T aa, kk;
mas01mc@293 249 if( v.size() != H::d )
mas01mc@293 250 error("v.size != H::d","","compute_hash_functions"); // check input vector dimensionality
mas01mc@293 251 double tmp = 0;
mas01mc@293 252 float *pA, *pb;
mas01mc@293 253 Uns32T *pg;
mas01mc@293 254 int dd;
mas01mc@293 255 vector<float>::iterator vi;
mas01mc@293 256 vector<Uns32T>::iterator ui;
mas01mc@293 257
mas01mc@293 258 #ifdef USE_U_FUNCTIONS
mas01mc@293 259 Uns32T bb;
mas01mc@293 260 // Store m dot products to expand
mas01mc@293 261 for( aa=0; aa < H::m ; aa++ ){
mas01mc@293 262 ui = H::uu[aa].begin();
mas01mc@293 263 for( kk = 0 ; kk < H::k/2 ; kk++ ){
mas01mc@293 264 pb = *( H::b + aa ) + kk;
mas01mc@293 265 pA = * ( * ( H::A + aa ) + kk );
mas01mc@293 266 dd = H::d;
mas01mc@293 267 tmp = 0.;
mas01mc@293 268 vi = v.begin();
mas01mc@293 269 while( dd-- )
mas01mc@293 270 tmp += *pA++ * *vi++; // project
mas01mc@293 271 tmp += *pb; // translate
mas01mc@293 272 tmp *= iw; // scale
mas01mc@293 273 *ui++ = (Uns32T) floor(tmp); // floor
mas01mc@293 274 }
mas01mc@293 275 }
mas01mc@293 276 // Binomial combinations of functions u_{a,b} \in Z^{(k/2) \times d}
mas01mc@293 277 Uns32T j;
mas01mc@293 278 for( aa=0, j=0 ; aa < H::m-1 ; aa++ )
mas01mc@293 279 for( bb = aa + 1 ; bb < H::m ; bb++, j++ ){
mas01mc@293 280 pg= *( H::g + j ); // L \times functions g_j(v) \in Z^k
mas01mc@293 281 // u_1 \in Z^{(k/2) \times d}
mas01mc@293 282 ui = H::uu[aa].begin();
mas01mc@293 283 kk=H::k/2;
mas01mc@293 284 while( kk-- )
mas01mc@293 285 *pg++ = *ui++; // hash function g_j(v)=[x1 x2 ... x(k/2)]; xk \in Z
mas01mc@293 286 // u_2 \in Z^{(k/2) \times d}
mas01mc@293 287 ui = H::uu[bb].begin();
mas01mc@293 288 kk=H::k/2;
mas01mc@293 289 while( kk--)
mas01mc@293 290 *pg++ = *ui++; // hash function g_j(v)=[x(k/2+1) x(k/2+2) ... xk]; xk \in Z
mas01mc@293 291 }
mas01mc@293 292 #else
mas01mc@293 293 for( aa=0; aa < H::L ; aa++ ){
mas01mc@514 294 pg= *( H::g + aa ); // L \times functions g_j(v) \in Z^k
mas01mc@514 295 for( kk = 0 ; kk != H::k ; kk++ ){
mas01mc@293 296 pb = *( H::b + aa ) + kk;
mas01mc@293 297 pA = * ( * ( H::A + aa ) + kk );
mas01mc@293 298 dd = H::d;
mas01mc@293 299 tmp = 0.;
mas01mc@293 300 vi = v.begin();
mas01mc@293 301 while( dd-- )
mas01mc@293 302 tmp += *pA++ * *vi++; // project
mas01mc@293 303 tmp += *pb; // translate
mas01mc@293 304 tmp *= iw; // scale
mas01mc@514 305 *pg++ = (Uns32T) (floor(tmp)); // hash function g_j(v)=[x1 x2 ... xk]; xk \in Z
mas01mc@293 306 }
mas01mc@293 307 }
mas01mc@293 308 #endif
mas01mc@293 309 }
mas01mc@293 310
mas01mc@292 311 // make hash value \in Z
mas01mc@293 312 void H::generate_hash_keys(Uns32T*g, Uns32T* r1, Uns32T* r2){
mas01mc@293 313 H::t1 = computeProductModDefaultPrime( g, r1, H::k ) % H::N;
mas01mc@293 314 H::t2 = computeProductModDefaultPrime( g, r2, H::k );
mas01mc@292 315 }
mas01mc@292 316
mas01mc@292 317 #define CR_ASSERT(b){if(!(b)){fprintf(stderr, "ASSERT failed on line %d, file %s.\n", __LINE__, __FILE__); exit(1);}}
mas01mc@292 318
mas01mc@292 319 // Computes (a.b) mod UH_PRIME_DEFAULT
mas01mc@293 320 inline Uns32T H::computeProductModDefaultPrime(Uns32T *a, Uns32T *b, IntT size){
mas01mc@292 321 LongUns64T h = 0;
mas01mc@292 322
mas01mc@292 323 for(IntT i = 0; i < size; i++){
mas01mc@292 324 h = h + (LongUns64T)a[i] * (LongUns64T)b[i];
mas01mc@292 325 h = (h & TWO_TO_32_MINUS_1) + 5 * (h >> 32);
mas01mc@292 326 if (h >= UH_PRIME_DEFAULT) {
mas01mc@292 327 h = h - UH_PRIME_DEFAULT;
mas01mc@292 328 }
mas01mc@292 329 CR_ASSERT(h < UH_PRIME_DEFAULT);
mas01mc@292 330 }
mas01mc@292 331 return h;
mas01mc@292 332 }
mas01mc@292 333
mas01mc@292 334 Uns32T H::bucket_insert_point(bucket **pp){
mas01mc@296 335 collisionCount = 0;
mas01mc@340 336 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@292 337 if(!*pp){
mas01mc@292 338 *pp = new bucket();
mas01mc@292 339 (*pp)->t2 = 0; // Use t2 as a collision counter for the row
mas01mc@292 340 (*pp)->next = new bucket();
mas01mc@292 341 }
mas01mc@340 342 // The list head holds point collision count
mas01mc@340 343 if( (*pp)->t2 & LSH_CORE_ARRAY_BIT )
mas01mc@340 344 bucket_insert_point(get_pointer_to_bucket_linked_list(*pp)); // recurse
mas01mc@340 345 else{
mas01mc@340 346 collisionCount = (*pp)->t2;
mas01mc@340 347 if(collisionCount < H::C){ // Block if row is full
mas01mc@340 348 (*pp)->t2++; // Increment collision counter (numPoints in row)
mas01mc@340 349 pointCount++;
mas01mc@340 350 collisionCount++;
mas01mc@340 351 // Locate the bucket linked list
mas01mc@340 352 __bucket_insert_point( (*pp)->next );
mas01mc@340 353 }
mas01mc@292 354 }
mas01mc@340 355 #else // NOT USING LSH_LIST_HEAD_COUNTERS
mas01mc@340 356 if(!*pp)
mas01mc@340 357 *pp = new bucket();
mas01mc@296 358 pointCount++;
mas01mc@296 359 __bucket_insert_point(*pp); // No collision count storage
mas01mc@292 360 #endif
mas01mc@292 361 return collisionCount;
mas01mc@292 362 }
mas01mc@292 363
mas01mc@340 364 // insert points into hashtable row collision chain
mas01mc@292 365 void H::__bucket_insert_point(bucket* p){
mas01mc@292 366 if(p->t2 == IFLAG){ // initialization flag, is it in the domain of t2?
mas01mc@292 367 p->t2 = H::t2;
mas01mc@292 368 bucketCount++; // Record start of new point-locale collision chain
mas01mc@344 369 p->snext.ptr = new sbucket();
mas01mc@344 370 __sbucket_insert_point(p->snext.ptr);
mas01mc@292 371 return;
mas01mc@292 372 }
mas01mc@292 373
mas01mc@292 374 if(p->t2 == H::t2){
mas01mc@344 375 __sbucket_insert_point(p->snext.ptr);
mas01mc@292 376 return;
mas01mc@292 377 }
mas01mc@292 378
mas01mc@292 379 if(p->next){
mas01mc@340 380 // Construct list in t2 order
mas01mc@340 381 if(H::t2 < p->next->t2){
mas01mc@340 382 bucket* tmp = new bucket();
mas01mc@340 383 tmp->next = p->next;
mas01mc@340 384 p->next = tmp;
mas01mc@340 385 __bucket_insert_point(tmp);
mas01mc@340 386 }
mas01mc@340 387 else
mas01mc@340 388 __bucket_insert_point(p->next);
mas01mc@292 389 }
mas01mc@340 390 else {
mas01mc@292 391 p->next = new bucket();
mas01mc@292 392 __bucket_insert_point(p->next);
mas01mc@292 393 }
mas01mc@292 394 }
mas01mc@292 395
mas01mc@340 396 // insert points into point-locale collision chain
mas01mc@292 397 void H::__sbucket_insert_point(sbucket* p){
mas01mc@292 398 if(p->pointID==IFLAG){
mas01mc@292 399 p->pointID = H::p;
mas01mc@292 400 return;
mas01mc@292 401 }
mas01mc@340 402
mas01mc@292 403 // Search for pointID
mas01mc@340 404 if(p->snext){
mas01mc@292 405 __sbucket_insert_point(p->snext);
mas01mc@292 406 }
mas01mc@292 407 else{
mas01mc@340 408 // Make new point collision bucket at end of list
mas01mc@340 409 p->snext = new sbucket();
mas01mc@340 410 __sbucket_insert_point(p->snext);
mas01mc@292 411 }
mas01mc@292 412 }
mas01mc@292 413
mas01mc@293 414 inline bucket** H::get_bucket(int j){
mas01mc@292 415 return *(h+j);
mas01mc@292 416 }
mas01mc@292 417
mas01mc@340 418 // Find the linked-list pointer at the end of the CORE_ARRAY
mas01mc@340 419 bucket** H::get_pointer_to_bucket_linked_list(bucket* rowPtr){
mas01mc@344 420 Uns32T numBuckets = rowPtr->snext.numBuckets; // Cast pointer to unsigned int
mas01mc@343 421 Uns32T numPoints = rowPtr->t2 & 0x7FFFFFFF; // Value is stored in low 31 bits of t2 field
mas01mc@343 422 bucket** listPtr = reinterpret_cast<bucket**> (reinterpret_cast<unsigned int*>(rowPtr->next)+numPoints+numBuckets+1);
mas01mc@343 423 return listPtr;
mas01mc@340 424 }
mas01mc@340 425
mas01mc@293 426 // Interface to Locality Sensitive Hashing G
mas01mc@293 427 G::G(float ww, Uns32T kk,Uns32T mm, Uns32T dd, Uns32T NN, Uns32T CC, float rr):
mas01mc@293 428 H(kk,mm,dd,NN,CC,ww,rr), // constructor to initialize data structures
mas01mc@308 429 indexName(0),
mas01mc@293 430 lshHeader(0),
mas01mc@292 431 calling_instance(0),
mas01mc@293 432 add_point_callback(0)
mas01mc@292 433 {
mas01mc@293 434
mas01mc@292 435 }
mas01mc@292 436
mas01mc@292 437 // Serialize from file LSH constructor
mas01mc@292 438 // Read parameters from database file
mas01mc@292 439 // Load the hash functions, close the database
mas01mc@292 440 // Optionally load the LSH tables into head-allocated lists in core
mas01mc@292 441 G::G(char* filename, bool lshInCoreFlag):
mas01mc@293 442 H(), // default base-class constructor call delays data-structure initialization
mas01mc@309 443 indexName(0),
mas01mc@293 444 lshHeader(0),
mas01mc@292 445 calling_instance(0),
mas01mc@292 446 add_point_callback(0)
mas01mc@292 447 {
mas01mc@474 448 FILE* dbFile = 0;
mas01mc@292 449 int dbfid = unserialize_lsh_header(filename);
mas01mc@309 450 indexName = new char[O2_INDEX_MAXSTR];
mas01mc@309 451 strncpy(indexName, filename, O2_INDEX_MAXSTR); // COPY THE CONTENTS TO THE NEW POINTER
mas01mc@293 452 H::initialize_lsh_functions(); // Base-class data-structure initialization
mas01mc@293 453 unserialize_lsh_functions(dbfid); // populate with on-disk hashfunction values
mas01mc@292 454
mas01mc@292 455 // Format1 only needs unserializing if specifically requested
mas01mc@292 456 if(!(lshHeader->flags&O2_SERIAL_FILEFORMAT2) && lshInCoreFlag){
mas01mc@292 457 unserialize_lsh_hashtables_format1(dbfid);
mas01mc@292 458 }
mas01mc@292 459
mas01mc@292 460 // Format2 always needs unserializing
mas01mc@292 461 if(lshHeader->flags&O2_SERIAL_FILEFORMAT2 && lshInCoreFlag){
mas01mc@474 462 dbFile = fdopen(dbfid, "rb");
mas01mc@336 463 if(!dbFile)
mas01mc@336 464 error("Cannot open LSH file for reading", filename);
mas01mc@336 465 unserialize_lsh_hashtables_format2(dbFile);
mas01mc@292 466 }
mas01mc@336 467 serial_close(dbfid);
mas01mc@474 468 if(dbFile){
mas01mc@474 469 fclose(dbFile);
mas01mc@474 470 dbFile = 0;
mas01mc@474 471 }
mas01mc@336 472 }
mas01mc@292 473
mas01mc@292 474 G::~G(){
mas01mc@292 475 delete lshHeader;
mas01mc@474 476 delete[] indexName;
mas01mc@292 477 }
mas01mc@292 478
mas01mc@292 479 // single point insertion; inserted values are hash value and pointID
mas01mc@292 480 Uns32T G::insert_point(vector<float>& v, Uns32T pp){
mas01mc@292 481 Uns32T collisionCount = 0;
mas01mc@292 482 H::p = pp;
mas01mc@299 483 if(H::maxp && pp<=H::maxp)
mas01mc@296 484 error("points must be indexed in strict ascending order", "LSH::insert_point(vector<float>&, Uns32T pointID)");
mas01mc@296 485 H::maxp=pp; // Store highest pointID in database
mas01mc@293 486 H::compute_hash_functions( v );
mas01mc@292 487 for(Uns32T j = 0 ; j < H::L ; j++ ){ // insertion
mas01mc@293 488 H::generate_hash_keys( *( H::g + j ), *( H::r1 + j ), *( H::r2 + j ) );
mas01mc@292 489 collisionCount += bucket_insert_point( *(h + j) + t1 );
mas01mc@292 490 }
mas01mc@292 491 return collisionCount;
mas01mc@292 492 }
mas01mc@292 493
mas01mc@292 494
mas01mc@292 495 // batch insert for a point set
mas01mc@292 496 // inserted values are vector hash value and pointID starting at basePointID
mas01mc@292 497 void G::insert_point_set(vector<vector<float> >& vv, Uns32T basePointID){
mas01mc@292 498 for(Uns32T point=0; point<vv.size(); point++)
mas01mc@292 499 insert_point(vv[point], basePointID+point);
mas01mc@292 500 }
mas01mc@292 501
mas01mc@292 502 // point retrieval routine
mas01mc@292 503 void G::retrieve_point(vector<float>& v, Uns32T qpos, ReporterCallbackPtr add_point, void* caller){
mas01mc@292 504 calling_instance = caller;
mas01mc@292 505 add_point_callback = add_point;
mas01mc@293 506 H::compute_hash_functions( v );
mas01mc@292 507 for(Uns32T j = 0 ; j < H::L ; j++ ){
mas01mc@293 508 H::generate_hash_keys( *( H::g + j ), *( H::r1 + j ), *( H::r2 + j ) );
mas01cr@370 509 if( bucket* bPtr = *(get_bucket(j) + get_t1()) ) {
mas01mc@340 510 #ifdef LSH_LIST_HEAD_COUNTERS
mas01cr@370 511 if(bPtr->t2&LSH_CORE_ARRAY_BIT) {
mas01mc@340 512 retrieve_from_core_hashtable_array((Uns32T*)(bPtr->next), qpos);
mas01cr@370 513 } else {
mas01mc@340 514 bucket_chain_point( bPtr->next, qpos);
mas01cr@370 515 }
mas01mc@292 516 #else
mas01cr@370 517 bucket_chain_point( bPtr , qpos);
mas01mc@292 518 #endif
mas01cr@370 519 }
mas01mc@292 520 }
mas01mc@292 521 }
mas01mc@292 522
mas01mc@292 523 void G::retrieve_point_set(vector<vector<float> >& vv, ReporterCallbackPtr add_point, void* caller){
mas01mc@292 524 for(Uns32T qpos = 0 ; qpos < vv.size() ; qpos++ )
mas01mc@292 525 retrieve_point(vv[qpos], qpos, add_point, caller);
mas01mc@292 526 }
mas01mc@292 527
mas01mc@292 528 // export lsh tables to table structure on disk
mas01mc@292 529 //
mas01mc@292 530 // LSH TABLE STRUCTURE
mas01mc@292 531 // ---header 64 bytes ---
mas01mc@292 532 // [magic #tables #rows #cols elementSize databaseSize version flags dim #funs 0 0 0 0 0 0]
mas01mc@292 533 //
mas01mc@292 534 // ---random projections L x k x d float ---
mas01mc@292 535 // A[0][0][0] A[0][0][1] ... A[0][0][d-1]
mas01mc@292 536 // A[0][1][0] A[0][1][1] ... A[1][1][d-1]
mas01mc@292 537 // ...
mas01mc@292 538 // A[0][K-1][0] A[0][1][1] ... A[0][k-1][d-1]
mas01mc@292 539 // ...
mas01mc@292 540 // ...
mas01mc@292 541 // A[L-1][0][0] A[M-1][0][1] ... A[L-1][0][d-1]
mas01mc@292 542 // A[L-1][1][0] A[M-1][1][1] ... A[L-1][1][d-1]
mas01mc@292 543 // ...
mas01mc@292 544 // A[L-1][k-1][0] A[M-1][1][1] ... A[L-1][k-1][d-1]
mas01mc@292 545 //
mas01mc@292 546 // ---bias L x k float ---
mas01mc@292 547 // b[0][0] b[0][1] ... b[0][k-1]
mas01mc@292 548 // b[1][0] b[1][1] ... b[1][k-1]
mas01mc@292 549 // ...
mas01mc@292 550 // b[L-1][0] b[L-1][1] ... b[L-1][k-1]
mas01mc@292 551 //
mas01mc@292 552 // ---random r1 L x k float ---
mas01mc@292 553 // r1[0][0] r1[0][1] ... r1[0][k-1]
mas01mc@292 554 // r1[1][0] r1[1][1] ... r1[1][k-1]
mas01mc@292 555 // ...
mas01mc@292 556 // r1[L-1][0] r1[L-1][1] ... r1[L-1][k-1]
mas01mc@292 557 //
mas01mc@292 558 // ---random r2 L x k float ---
mas01mc@292 559 // r2[0][0] r2[0][1] ... r2[0][k-1]
mas01mc@292 560 // r2[1][0] r2[1][1] ... r2[1][k-1]
mas01mc@292 561 // ...
mas01mc@292 562 // r2[L-1][0] r2[L-1][1] ... r2[L-1][k-1]
mas01mc@292 563 //
mas01mc@293 564 // ******* HASHTABLES FORMAT1 (optimized for LSH_ON_DISK retrieval) *******
mas01mc@292 565 // ---hash table 0: N x C x 8 ---
mas01mc@292 566 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 567 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 568 // ...
mas01mc@292 569 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 570 //
mas01mc@292 571 // ---hash table 1: N x C x 8 ---
mas01mc@292 572 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 573 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 574 // ...
mas01mc@292 575 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 576 //
mas01mc@292 577 // ...
mas01mc@292 578 //
mas01mc@292 579 // ---hash table L-1: N x C x 8 ---
mas01mc@292 580 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 581 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 582 // ...
mas01mc@292 583 // [t2 pointID][t2 pointID]...[t2 pointID]
mas01mc@292 584 //
mas01mc@293 585 // ******* HASHTABLES FORMAT2 (optimized for LSH_IN_CORE retrieval) *******
mas01mc@293 586 //
mas01mc@293 587 // State machine controlled by regular expression.
mas01mc@293 588 // legend:
mas01mc@293 589 //
mas01mc@306 590 // O2_SERIAL_TOKEN_T1 = 0xFFFFFFFCU
mas01mc@306 591 // O2_SERIAL_TOKEN_T2 = 0xFFFFFFFDU
mas01mc@306 592 // O2_SERIAL_TOKEN_ENDTABLE = 0xFFFFFFFEU
mas01mc@293 593 //
mas01mc@306 594 // T1 - T1 hash token
mas01mc@306 595 // t1 - t1 hash key (t1 range 0..2^29-1)
mas01mc@306 596 // T2 - T2 token
mas01mc@293 597 // t2 - t2 hash key (range 1..2^32-6)
mas01mc@293 598 // p - point identifier (range 0..2^32-1)
mas01mc@306 599 // E - end hash table token
mas01mc@293 600 // {...} required arguments
mas01mc@293 601 // [...] optional arguments
mas01mc@293 602 // * - match zero or more occurences
mas01mc@293 603 // + - match one or more occurences
mas01mc@293 604 // {...}^L - repeat argument L times
mas01mc@293 605 //
mas01mc@293 606 // FORMAT2 Regular expression:
mas01mc@306 607 // { [T1 t1 [T2 t2 p+]+ ]* E }^L
mas01mc@293 608 //
mas01mc@292 609
mas01mc@292 610 // Serial header constructors
mas01mc@292 611 SerialHeader::SerialHeader(){;}
mas01mc@296 612 SerialHeader::SerialHeader(float W, Uns32T L, Uns32T N, Uns32T C, Uns32T k, Uns32T d, float r, Uns32T p, Uns32T FMT, Uns32T pc):
mas01mc@292 613 lshMagic(O2_SERIAL_MAGIC),
mas01mc@292 614 binWidth(W),
mas01mc@292 615 numTables(L),
mas01mc@292 616 numRows(N),
mas01mc@292 617 numCols(C),
mas01mc@292 618 elementSize(O2_SERIAL_ELEMENT_SIZE),
mas01mc@296 619 version(O2_SERIAL_VERSION),
mas01mc@296 620 size(0), // we are deprecating this value
mas01mc@292 621 flags(FMT),
mas01mc@292 622 dataDim(d),
mas01mc@292 623 numFuns(k),
mas01mc@292 624 radius(r),
mas01mc@296 625 maxp(p),
mas01mc@296 626 size_long((unsigned long long)L * align_up((unsigned long long)N * C * O2_SERIAL_ELEMENT_SIZE, get_page_logn()) // hash tables
mas01mc@296 627 + align_up(O2_SERIAL_HEADER_SIZE + // header + hash functions
mas01mc@296 628 (unsigned long long)L*k*( sizeof(float)*d+2*sizeof(Uns32T)+sizeof(float)),get_page_logn())),
mas01mc@296 629 pointCount(pc){
mas01mc@296 630
mas01mc@296 631 if(FMT==O2_SERIAL_FILEFORMAT2)
mas01mc@296 632 size_long = (unsigned long long)align_up(O2_SERIAL_HEADER_SIZE
mas01mc@296 633 + (unsigned long long)L*k*(sizeof(float)*d+2+sizeof(Uns32T)
mas01mc@296 634 +sizeof(float)) + (unsigned long long)pc*16UL,get_page_logn());
mas01mc@296 635 } // header
mas01mc@292 636
mas01mc@292 637 float* G::get_serial_hashfunction_base(char* db){
mas01mc@292 638 if(db&&lshHeader)
mas01mc@292 639 return (float*)(db+O2_SERIAL_HEADER_SIZE);
mas01mc@292 640 else return NULL;
mas01mc@292 641 }
mas01mc@292 642
mas01mc@292 643 SerialElementT* G::get_serial_hashtable_base(char* db){
mas01mc@292 644 if(db&&lshHeader)
mas01mc@292 645 return (SerialElementT*)(db+get_serial_hashtable_offset());
mas01mc@292 646 else
mas01mc@292 647 return NULL;
mas01mc@292 648 }
mas01mc@292 649
mas01mc@292 650 Uns32T G::get_serial_hashtable_offset(){
mas01mc@292 651 if(lshHeader)
mas01mc@292 652 return align_up(O2_SERIAL_HEADER_SIZE +
mas01mc@292 653 L*lshHeader->numFuns*( sizeof(float)*lshHeader->dataDim+2*sizeof(Uns32T)+sizeof(float)),get_page_logn());
mas01mc@292 654 else
mas01mc@292 655 return 0;
mas01mc@292 656 }
mas01mc@292 657
mas01mc@292 658 void G::serialize(char* filename, Uns32T serialFormat){
mas01mc@292 659 int dbfid;
mas01mc@292 660 char* db;
mas01mc@292 661 int dbIsNew=0;
mas01mc@474 662 FILE* dbFile = 0;
mas01mc@292 663 // Check requested serialFormat
mas01mc@292 664 if(!(serialFormat==O2_SERIAL_FILEFORMAT1 || serialFormat==O2_SERIAL_FILEFORMAT2))
mas01mc@292 665 error("Unrecognized serial file format request: ", "serialize()");
mas01mc@296 666
mas01mc@292 667 // Test to see if file exists
mas01cr@370 668 if((dbfid = open (filename, O_RDONLY)) < 0) {
mas01mc@292 669 // If it doesn't, then create the file (CREATE)
mas01cr@370 670 if(errno == ENOENT) {
mas01mc@292 671 // Create the file
mas01mc@292 672 std::cout << "Creating new serialized LSH database:" << filename << "...";
mas01mc@292 673 std::cout.flush();
mas01mc@292 674 serial_create(filename, serialFormat);
mas01mc@292 675 dbIsNew=1;
mas01cr@370 676 } else {
mas01mc@292 677 // The file can't be opened
mas01mc@292 678 error("Can't open the file", filename, "open");
mas01cr@370 679 }
mas01cr@370 680 }
mas01mc@292 681
mas01mc@292 682 // Load the on-disk header into core
mas01mc@292 683 dbfid = serial_open(filename, 1); // open for write
mas01mc@292 684 db = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 1);// get database pointer
mas01mc@292 685 serial_get_header(db); // read header
mas01mc@292 686 serial_munmap(db, O2_SERIAL_HEADER_SIZE); // drop mmap
mas01mc@292 687
mas01mc@292 688 // Check compatibility of core and disk data structures
mas01mc@292 689 if( !serial_can_merge(serialFormat) )
mas01mc@292 690 error("Incompatible core and serial LSH, data structure dimensions mismatch.");
mas01mc@292 691
mas01mc@292 692 // For new LSH databases write the hashfunctions
mas01mc@292 693 if(dbIsNew)
mas01mc@292 694 serialize_lsh_hashfunctions(dbfid);
mas01mc@292 695 // Write the hashtables in the requested format
mas01mc@292 696 if(serialFormat == O2_SERIAL_FILEFORMAT1)
mas01mc@292 697 serialize_lsh_hashtables_format1(dbfid, !dbIsNew);
mas01mc@336 698 else{
mas01mc@474 699 dbFile = fdopen(dbfid, "r+b");
mas01mc@336 700 if(!dbFile)
mas01mc@336 701 error("Cannot open LSH file for writing",filename);
mas01mc@336 702 serialize_lsh_hashtables_format2(dbFile, !dbIsNew);
mas01mc@336 703 fflush(dbFile);
mas01mc@336 704 }
mas01mc@292 705
mas01mc@336 706 if(!dbIsNew) {
mas01mc@292 707 db = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 1);// get database pointer
mas01mc@292 708 //serial_get_header(db); // read header
mas01mc@293 709 cout << "maxp = " << H::maxp << endl;
mas01mc@293 710 lshHeader->maxp=H::maxp;
mas01mc@292 711 // Default to FILEFORMAT1
mas01mc@292 712 if(!(lshHeader->flags&O2_SERIAL_FILEFORMAT2))
mas01mc@294 713 lshHeader->flags|=O2_SERIAL_FILEFORMAT1;
mas01mc@292 714 memcpy((char*)db, (char*)lshHeader, sizeof(SerialHeaderT));
mas01mc@292 715 serial_munmap(db, O2_SERIAL_HEADER_SIZE); // drop mmap
mas01mc@336 716 }
mas01mc@336 717 serial_close(dbfid);
mas01mc@474 718 if(dbFile){
mas01mc@474 719 fclose(dbFile);
mas01mc@474 720 dbFile = 0;
mas01mc@474 721 }
mas01mc@292 722 }
mas01mc@292 723
mas01mc@292 724 // Test to see if core structure and requested format is
mas01mc@292 725 // compatible with currently opened database
mas01mc@292 726 int G::serial_can_merge(Uns32T format){
mas01mc@292 727 SerialHeaderT* that = lshHeader;
mas01mc@292 728 if( (format==O2_SERIAL_FILEFORMAT2 && !that->flags&O2_SERIAL_FILEFORMAT2)
mas01mc@292 729 || (format!=O2_SERIAL_FILEFORMAT2 && that->flags&O2_SERIAL_FILEFORMAT2)
mas01mc@292 730 || !( this->w == that->binWidth &&
mas01mc@296 731 this->L == that->numTables &&
mas01mc@296 732 this->N == that->numRows &&
mas01mc@296 733 this->k == that->numFuns &&
mas01mc@296 734 this->d == that->dataDim &&
mas01mc@296 735 sizeof(SerialElementT) == that->elementSize &&
mas01mc@296 736 this->radius == that->radius)){
mas01mc@292 737 serial_print_header(format);
mas01mc@292 738 return 0;
mas01mc@292 739 }
mas01mc@292 740 else
mas01mc@292 741 return 1;
mas01mc@292 742 }
mas01mc@292 743
mas01mc@292 744 // Used as an error message for serial_can_merge()
mas01mc@292 745 void G::serial_print_header(Uns32T format){
mas01mc@292 746 std::cout << "Fc:" << format << " Fs:" << lshHeader->flags << endl;
mas01mc@292 747 std::cout << "Wc:" << w << " Ls:" << lshHeader->binWidth << endl;
mas01mc@292 748 std::cout << "Lc:" << L << " Ls:" << lshHeader->numTables << endl;
mas01mc@292 749 std::cout << "Nc:" << N << " Ns:" << lshHeader->numRows << endl;
mas01mc@292 750 std::cout << "kc:" << k << " ks:" << lshHeader->numFuns << endl;
mas01mc@292 751 std::cout << "dc:" << d << " ds:" << lshHeader->dataDim << endl;
mas01mc@292 752 std::cout << "sc:" << sizeof(SerialElementT) << " ss:" << lshHeader->elementSize << endl;
mas01mc@292 753 std::cout << "rc:" << this->radius << " rs:" << lshHeader->radius << endl;
mas01mc@292 754 }
mas01mc@292 755
mas01mc@292 756 int G::serialize_lsh_hashfunctions(int fid){
mas01mc@292 757 float* pf;
mas01mc@292 758 Uns32T *pu;
mas01mc@292 759 Uns32T x,y,z;
mas01mc@292 760
mas01mc@293 761 char* db = serial_mmap(fid, get_serial_hashtable_offset(), 1);// get database pointer
mas01mc@292 762 pf = get_serial_hashfunction_base(db);
mas01mc@292 763
mas01mc@292 764 // HASH FUNCTIONS
mas01mc@292 765 // Write the random projectors A[][][]
mas01mc@292 766 #ifdef USE_U_FUNCTIONS
mas01mc@292 767 for( x = 0 ; x < H::m ; x++ )
mas01mc@292 768 for( y = 0 ; y < H::k/2 ; y++ )
mas01mc@292 769 #else
mas01mc@292 770 for( x = 0 ; x < H::L ; x++ )
mas01mc@292 771 for( y = 0 ; y < H::k ; y++ )
mas01mc@292 772 #endif
mas01mc@292 773 for( z = 0 ; z < d ; z++ )
mas01mc@293 774 *pf++ = H::A[x][y][z];
mas01mc@292 775
mas01mc@292 776 // Write the random biases b[][]
mas01mc@292 777 #ifdef USE_U_FUNCTIONS
mas01mc@292 778 for( x = 0 ; x < H::m ; x++ )
mas01mc@292 779 for( y = 0 ; y < H::k/2 ; y++ )
mas01mc@292 780 #else
mas01mc@292 781 for( x = 0 ; x < H::L ; x++ )
mas01mc@292 782 for( y = 0 ; y < H::k ; y++ )
mas01mc@292 783 #endif
mas01mc@293 784 *pf++ = H::b[x][y];
mas01mc@292 785
mas01mc@292 786 pu = (Uns32T*)pf;
mas01mc@292 787
mas01mc@292 788 // Write the Z projectors r1[][]
mas01mc@292 789 for( x = 0 ; x < H::L ; x++)
mas01mc@292 790 for( y = 0 ; y < H::k ; y++)
mas01mc@293 791 *pu++ = H::r1[x][y];
mas01mc@292 792
mas01mc@292 793 // Write the Z projectors r2[][]
mas01mc@292 794 for( x = 0 ; x < H::L ; x++)
mas01mc@292 795 for( y = 0; y < H::k ; y++)
mas01mc@293 796 *pu++ = H::r2[x][y];
mas01mc@292 797
mas01mc@292 798 serial_munmap(db, get_serial_hashtable_offset());
mas01mc@292 799 return 1;
mas01mc@292 800 }
mas01mc@292 801
mas01mc@292 802 int G::serialize_lsh_hashtables_format1(int fid, int merge){
mas01mc@292 803 SerialElementT *pe, *pt;
mas01mc@292 804 Uns32T x,y;
mas01mc@292 805
mas01mc@292 806 if( merge && !serial_can_merge(O2_SERIAL_FILEFORMAT1) )
mas01mc@292 807 error("Cannot merge core and serial LSH, data structure dimensions mismatch.");
mas01mc@292 808
mas01mc@292 809 Uns32T hashTableSize=sizeof(SerialElementT)*lshHeader->numRows*lshHeader->numCols;
mas01mc@292 810 Uns32T colCount, meanColCount, colCountN, maxColCount, minColCount;
mas01mc@292 811 // Write the hash tables
mas01mc@292 812 for( x = 0 ; x < H::L ; x++ ){
mas01mc@292 813 std::cout << (merge ? "merging":"writing") << " hash table " << x << " FORMAT1...";
mas01mc@292 814 std::cout.flush();
mas01mc@292 815 // memory map a single hash table for sequential access
mas01mc@292 816 // Align each hash table to page boundary
mas01mc@292 817 char* dbtable = serial_mmap(fid, hashTableSize, 1,
mas01mc@292 818 align_up(get_serial_hashtable_offset()+x*hashTableSize, get_page_logn()));
mas01mc@324 819 #ifdef __CYGWIN__
mas01mc@324 820 // No madvise in CYGWIN
mas01mc@324 821 #else
mas01mc@292 822 if(madvise(dbtable, hashTableSize, MADV_SEQUENTIAL)<0)
mas01mc@292 823 error("could not advise hashtable memory","","madvise");
mas01mc@324 824 #endif
mas01mc@292 825 maxColCount=0;
mas01mc@292 826 minColCount=O2_SERIAL_MAX_COLS;
mas01mc@292 827 meanColCount=0;
mas01mc@292 828 colCountN=0;
mas01mc@292 829 pt=(SerialElementT*)dbtable;
mas01mc@292 830 for( y = 0 ; y < H::N ; y++ ){
mas01mc@292 831 // Move disk pointer to beginning of row
mas01mc@292 832 pe=pt+y*lshHeader->numCols;
mas01mc@292 833
mas01mc@292 834 colCount=0;
mas01cr@370 835 if(bucket* bPtr = h[x][y]) {
mas01cr@370 836 if(merge) {
mas01mc@340 837 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@292 838 serial_merge_hashtable_row_format1(pe, bPtr->next, colCount); // skip collision counter bucket
mas01cr@370 839 } else {
mas01mc@292 840 serial_write_hashtable_row_format1(pe, bPtr->next, colCount); // skip collision counter bucket
mas01mc@292 841 #else
mas01cr@370 842 serial_merge_hashtable_row_format1(pe, bPtr, colCount);
mas01cr@370 843 } else {
mas01cr@370 844 serial_write_hashtable_row_format1(pe, bPtr, colCount);
mas01mc@292 845 #endif
mas01cr@370 846 }
mas01cr@370 847 }
mas01mc@292 848 if(colCount){
mas01mc@292 849 if(colCount<minColCount)
mas01mc@292 850 minColCount=colCount;
mas01mc@292 851 if(colCount>maxColCount)
mas01mc@292 852 maxColCount=colCount;
mas01mc@292 853 meanColCount+=colCount;
mas01mc@292 854 colCountN++;
mas01mc@292 855 }
mas01mc@292 856 }
mas01mc@292 857 if(colCountN)
mas01mc@292 858 std::cout << "#rows with collisions =" << colCountN << ", mean = " << meanColCount/(float)colCountN
mas01mc@292 859 << ", min = " << minColCount << ", max = " << maxColCount
mas01mc@292 860 << endl;
mas01mc@292 861 serial_munmap(dbtable, hashTableSize);
mas01mc@292 862 }
mas01mc@292 863
mas01mc@292 864 // We're done writing
mas01mc@292 865 return 1;
mas01mc@292 866 }
mas01mc@292 867
mas01mc@292 868 void G::serial_merge_hashtable_row_format1(SerialElementT* pr, bucket* b, Uns32T& colCount){
mas01mc@292 869 while(b && b->t2!=IFLAG){
mas01mc@292 870 SerialElementT*pe=pr; // reset disk pointer to beginning of row
mas01mc@344 871 serial_merge_element_format1(pe, b->snext.ptr, b->t2, colCount);
mas01mc@292 872 b=b->next;
mas01mc@292 873 }
mas01mc@292 874 }
mas01mc@292 875
mas01mc@292 876 void G::serial_merge_element_format1(SerialElementT* pe, sbucket* sb, Uns32T t2, Uns32T& colCount){
mas01mc@292 877 while(sb){
mas01mc@292 878 if(colCount==lshHeader->numCols){
mas01mc@292 879 std::cout << "!point-chain full " << endl;
mas01mc@292 880 return;
mas01mc@292 881 }
mas01mc@292 882 Uns32T c=0;
mas01mc@292 883 // Merge collision chains
mas01mc@292 884 while(c<lshHeader->numCols){
mas01mc@292 885 if( (pe+c)->hashValue==IFLAG){
mas01mc@292 886 (pe+c)->hashValue=t2;
mas01mc@292 887 (pe+c)->pointID=sb->pointID;
mas01mc@292 888 colCount=c+1;
mas01mc@292 889 if(c+1<lshHeader->numCols)
mas01mc@292 890 (pe+c+1)->hashValue=IFLAG;
mas01mc@292 891 break;
mas01mc@292 892 }
mas01mc@292 893 c++;
mas01mc@292 894 }
mas01mc@292 895 sb=sb->snext;
mas01mc@292 896 }
mas01mc@292 897 return;
mas01mc@292 898 }
mas01mc@292 899
mas01mc@292 900 void G::serial_write_hashtable_row_format1(SerialElementT*& pe, bucket* b, Uns32T& colCount){
mas01mc@292 901 pe->hashValue=IFLAG;
mas01mc@292 902 while(b && b->t2!=IFLAG){
mas01mc@344 903 serial_write_element_format1(pe, b->snext.ptr, b->t2, colCount);
mas01mc@292 904 b=b->next;
mas01mc@292 905 }
mas01mc@292 906 }
mas01mc@292 907
mas01mc@292 908 void G::serial_write_element_format1(SerialElementT*& pe, sbucket* sb, Uns32T t2, Uns32T& colCount){
mas01mc@292 909 while(sb){
mas01mc@292 910 if(colCount==lshHeader->numCols){
mas01mc@292 911 std::cout << "!point-chain full " << endl;
mas01mc@292 912 return;
mas01mc@292 913 }
mas01mc@292 914 pe->hashValue=t2;
mas01mc@292 915 pe->pointID=sb->pointID;
mas01mc@292 916 pe++;
mas01mc@292 917 colCount++;
mas01mc@292 918 sb=sb->snext;
mas01mc@292 919 }
mas01mc@292 920 pe->hashValue=IFLAG;
mas01mc@292 921 return;
mas01mc@292 922 }
mas01mc@292 923
mas01mc@336 924 int G::serialize_lsh_hashtables_format2(FILE* dbFile, int merge){
mas01mc@292 925 Uns32T x,y;
mas01mc@292 926
mas01mc@292 927 if( merge && !serial_can_merge(O2_SERIAL_FILEFORMAT2) )
mas01mc@292 928 error("Cannot merge core and serial LSH, data structure dimensions mismatch.");
mas01mc@292 929
mas01mc@340 930 // We must pereform FORMAT2 merges in core FORMAT1 (dynamic list structure)
mas01mc@292 931 if(merge)
mas01mc@340 932 unserialize_lsh_hashtables_format2(dbFile, merge);
mas01mc@292 933
mas01mc@292 934 Uns32T colCount, meanColCount, colCountN, maxColCount, minColCount, t1;
mas01mc@336 935 if(fseek(dbFile, get_serial_hashtable_offset(), SEEK_SET)){
mas01mc@336 936 fclose(dbFile);
mas01mc@336 937 error("fSeek error in serialize_lsh_hashtables_format2");
mas01mc@336 938 }
mas01mc@292 939
mas01mc@292 940 // Write the hash tables
mas01mc@292 941 for( x = 0 ; x < H::L ; x++ ){
mas01mc@292 942 std::cout << (merge ? "merging":"writing") << " hash table " << x << " FORMAT2...";
mas01mc@292 943 std::cout.flush();
mas01mc@292 944 maxColCount=0;
mas01mc@292 945 minColCount=O2_SERIAL_MAX_COLS;
mas01mc@292 946 meanColCount=0;
mas01mc@292 947 colCountN=0;
mas01mc@292 948 for( y = 0 ; y < H::N ; y++ ){
mas01mc@292 949 colCount=0;
mas01mc@292 950 if(bucket* bPtr = h[x][y]){
mas01mc@306 951 // Check for empty row (even though row was allocated)
mas01mc@340 952 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@306 953 if(bPtr->next->t2==IFLAG){
mas01mc@336 954 fclose(dbFile);
mas01mc@306 955 error("b->next->t2==IFLAG","serialize_lsh_hashtables_format2()");
mas01mc@306 956 }
mas01mc@306 957 #else
mas01mc@306 958 if(bPtr->t2==IFLAG){
mas01mc@336 959 fclose(dbFile);
mas01mc@306 960 error("b->t2==IFLAG","serialize_lsh_hashtables_format2()");
mas01mc@306 961 }
mas01mc@306 962 #endif
mas01mc@306 963 t1 = O2_SERIAL_TOKEN_T1;
mas01mc@340 964 WRITE_UNS32(&t1, "[T1]");
mas01mc@306 965 t1 = y;
mas01mc@340 966 WRITE_UNS32(&t1, "[t1]");
mas01mc@340 967 #ifdef LSH_CORE_ARRAY
mas01mc@340 968 t1 = count_buckets_and_points_hashtable_row(bPtr);
mas01mc@340 969 WRITE_UNS32(&t1,"[count]"); // write numElements
mas01mc@340 970 #endif
mas01mc@340 971 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@336 972 serial_write_hashtable_row_format2(dbFile, bPtr->next, colCount); // skip collision counter bucket
mas01mc@292 973 #else
mas01mc@340 974 serial_write_hashtable_row_format2(dbFile, bPtr, colCount);
mas01mc@292 975 #endif
mas01mc@292 976 }
mas01mc@292 977 if(colCount){
mas01mc@292 978 if(colCount<minColCount)
mas01mc@292 979 minColCount=colCount;
mas01mc@292 980 if(colCount>maxColCount)
mas01mc@292 981 maxColCount=colCount;
mas01mc@292 982 meanColCount+=colCount;
mas01mc@292 983 colCountN++;
mas01mc@292 984 }
mas01mc@292 985 }
mas01mc@292 986 // Write END of table marker
mas01mc@306 987 t1 = O2_SERIAL_TOKEN_ENDTABLE;
mas01mc@340 988 WRITE_UNS32(&t1,"[end]");
mas01mc@292 989 if(colCountN)
mas01mc@292 990 std::cout << "#rows with collisions =" << colCountN << ", mean = " << meanColCount/(float)colCountN
mas01mc@292 991 << ", min = " << minColCount << ", max = " << maxColCount
mas01mc@292 992 << endl;
mas01mc@340 993 }
mas01mc@292 994 // We're done writing
mas01mc@292 995 return 1;
mas01mc@292 996 }
mas01mc@292 997
mas01mc@340 998 Uns32T G::count_buckets_and_points_hashtable_row(bucket* bPtr){
mas01mc@340 999 Uns32T total_count = 0;
mas01mc@340 1000 bucket* p = 0;
mas01mc@340 1001
mas01mc@340 1002 // count points
mas01mc@340 1003 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@340 1004 total_count = bPtr->t2; // points already counted
mas01mc@340 1005 p = bPtr->next;
mas01mc@340 1006 #else
mas01mc@340 1007 total_count = count_points_hashtable_row(bPtr);
mas01mc@340 1008 p = bPtr;
mas01mc@340 1009 #endif
mas01mc@340 1010
mas01mc@340 1011 // count buckets
mas01mc@340 1012 do{
mas01mc@340 1013 total_count++;
mas01mc@340 1014 }while((p=p->next));
mas01mc@340 1015
mas01mc@340 1016 return total_count;
mas01mc@340 1017 }
mas01mc@340 1018
mas01mc@340 1019 Uns32T G::count_points_hashtable_row(bucket* bPtr){
mas01mc@340 1020 Uns32T point_count = 0;
mas01mc@340 1021 bucket* p = bPtr;
mas01mc@340 1022 sbucket* s = 0;
mas01mc@340 1023 while(p){
mas01mc@344 1024 s = p->snext.ptr;
mas01mc@340 1025 while(s){
mas01mc@340 1026 point_count++;
mas01mc@340 1027 s=s->snext;
mas01mc@340 1028 }
mas01mc@340 1029 p=p->next;
mas01mc@340 1030 }
mas01mc@340 1031 return point_count;
mas01mc@340 1032 }
mas01mc@340 1033
mas01mc@336 1034 void G::serial_write_hashtable_row_format2(FILE* dbFile, bucket* b, Uns32T& colCount){
mas01mc@292 1035 while(b && b->t2!=IFLAG){
mas01mc@344 1036 if(!b->snext.ptr){
mas01mc@336 1037 fclose(dbFile);
mas01mc@306 1038 error("Empty collision chain in serial_write_hashtable_row_format2()");
mas01mc@306 1039 }
mas01mc@306 1040 t2 = O2_SERIAL_TOKEN_T2;
mas01mc@336 1041 if( fwrite(&t2, sizeof(Uns32T), 1, dbFile) != 1 ){
mas01mc@336 1042 fclose(dbFile);
mas01mc@292 1043 error("write error in serial_write_hashtable_row_format2()");
mas01mc@292 1044 }
mas01mc@292 1045 t2 = b->t2;
mas01mc@336 1046 if( fwrite(&t2, sizeof(Uns32T), 1, dbFile) != 1 ){
mas01mc@336 1047 fclose(dbFile);
mas01mc@292 1048 error("write error in serial_write_hashtable_row_format2()");
mas01mc@292 1049 }
mas01mc@344 1050 serial_write_element_format2(dbFile, b->snext.ptr, colCount);
mas01mc@292 1051 b=b->next;
mas01mc@292 1052 }
mas01mc@292 1053 }
mas01mc@292 1054
mas01mc@336 1055 void G::serial_write_element_format2(FILE* dbFile, sbucket* sb, Uns32T& colCount){
mas01mc@292 1056 while(sb){
mas01mc@336 1057 if(fwrite(&sb->pointID, sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1058 fclose(dbFile);
mas01mc@292 1059 error("Write error in serial_write_element_format2()");
mas01mc@292 1060 }
mas01mc@292 1061 colCount++;
mas01mc@292 1062 sb=sb->snext;
mas01mc@292 1063 }
mas01mc@292 1064 }
mas01mc@292 1065
mas01mc@292 1066
mas01mc@292 1067 int G::serial_create(char* filename, Uns32T FMT){
mas01mc@292 1068 return serial_create(filename, w, L, N, C, k, d, FMT);
mas01mc@292 1069 }
mas01mc@292 1070
mas01mc@292 1071
mas01mc@292 1072 int G::serial_create(char* filename, float binWidth, Uns32T numTables, Uns32T numRows, Uns32T numCols,
mas01mc@292 1073 Uns32T numFuns, Uns32T dim, Uns32T FMT){
mas01mc@292 1074
mas01mc@292 1075 if(numTables > O2_SERIAL_MAX_TABLES || numRows > O2_SERIAL_MAX_ROWS
mas01mc@292 1076 || numCols > O2_SERIAL_MAX_COLS || numFuns > O2_SERIAL_MAX_FUNS
mas01mc@292 1077 || dim>O2_SERIAL_MAX_DIM){
mas01mc@292 1078 error("LSH parameters out of bounds for serialization");
mas01mc@292 1079 }
mas01mc@292 1080
mas01mc@292 1081 int dbfid;
mas01mc@292 1082 if ((dbfid = open (filename, O_RDWR|O_CREAT|O_EXCL, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) < 0)
mas01mc@292 1083 error("Can't create serial file", filename, "open");
mas01mc@292 1084 get_lock(dbfid, 1);
mas01mc@292 1085
mas01mc@292 1086 // Make header first to get size of serialized database
mas01mc@296 1087 lshHeader = new SerialHeaderT(binWidth, numTables, numRows, numCols, numFuns, dim, radius, maxp, FMT, pointCount);
mas01mc@296 1088
mas01mc@296 1089 cout << "file size: <=" << lshHeader->get_size()/1024UL << "KB" << endl;
mas01mc@296 1090 if(lshHeader->get_size()>O2_SERIAL_MAXFILESIZE)
mas01mc@296 1091 error("Maximum size of LSH file exceded: > 4000MB");
mas01mc@296 1092
mas01mc@292 1093 // go to the location corresponding to the last byte
mas01mc@292 1094 if (lseek (dbfid, lshHeader->get_size() - 1, SEEK_SET) == -1)
mas01mc@292 1095 error("lseek error in db file", "", "lseek");
mas01mc@292 1096
mas01mc@292 1097 // write a dummy byte at the last location
mas01mc@292 1098 if (write (dbfid, "", 1) != 1)
mas01mc@292 1099 error("write error", "", "write");
mas01mc@292 1100
mas01mc@293 1101 char* db = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 1);
mas01mc@296 1102
mas01mc@292 1103 memcpy (db, lshHeader, O2_SERIAL_HEADER_SIZE);
mas01mc@296 1104
mas01mc@292 1105 serial_munmap(db, O2_SERIAL_HEADER_SIZE);
mas01mc@296 1106
mas01mc@292 1107 close(dbfid);
mas01mc@292 1108
mas01mc@292 1109 std::cout << "done initializing tables." << endl;
mas01mc@292 1110
mas01mc@292 1111 return 1;
mas01mc@292 1112 }
mas01mc@292 1113
mas01mc@292 1114 char* G::serial_mmap(int dbfid, Uns32T memSize, Uns32T forWrite, off_t offset){
mas01mc@293 1115 char* db;
mas01mc@292 1116 if(forWrite){
mas01mc@292 1117 if ((db = (char*) mmap(0, memSize, PROT_READ | PROT_WRITE,
mas01mc@292 1118 MAP_SHARED, dbfid, offset)) == (caddr_t) -1)
mas01mc@292 1119 error("mmap error in request for writable serialized database", "", "mmap");
mas01mc@292 1120 }
mas01mc@292 1121 else if ((db = (char*) mmap(0, memSize, PROT_READ, MAP_SHARED, dbfid, offset)) == (caddr_t) -1)
mas01mc@292 1122 error("mmap error in read-only serialized database", "", "mmap");
mas01mc@292 1123
mas01mc@292 1124 return db;
mas01mc@292 1125 }
mas01mc@292 1126
mas01mc@292 1127 SerialHeaderT* G::serial_get_header(char* db){
mas01mc@292 1128 lshHeader = new SerialHeaderT();
mas01mc@292 1129 memcpy((char*)lshHeader, db, sizeof(SerialHeaderT));
mas01mc@292 1130
mas01mc@292 1131 if(lshHeader->lshMagic!=O2_SERIAL_MAGIC)
mas01mc@292 1132 error("Not an LSH database file");
mas01mc@292 1133
mas01mc@292 1134 return lshHeader;
mas01mc@292 1135 }
mas01mc@292 1136
mas01mc@292 1137 void G::serial_munmap(char* db, Uns32T N){
mas01mc@292 1138 munmap(db, N);
mas01mc@292 1139 }
mas01mc@292 1140
mas01mc@292 1141 int G::serial_open(char* filename, int writeFlag){
mas01mc@292 1142 int dbfid;
mas01mc@292 1143 if(writeFlag){
mas01mc@292 1144 if ((dbfid = open (filename, O_RDWR)) < 0)
mas01mc@292 1145 error("Can't open serial file for read/write", filename, "open");
mas01mc@292 1146 get_lock(dbfid, writeFlag);
mas01mc@292 1147 }
mas01mc@292 1148 else{
mas01mc@292 1149 if ((dbfid = open (filename, O_RDONLY)) < 0)
mas01mc@292 1150 error("Can't open serial file for read", filename, "open");
mas01mc@292 1151 get_lock(dbfid, 0);
mas01mc@292 1152 }
mas01mc@292 1153
mas01mc@292 1154 return dbfid;
mas01mc@292 1155 }
mas01mc@292 1156
mas01mc@292 1157 void G::serial_close(int dbfid){
mas01mc@292 1158
mas01mc@292 1159 release_lock(dbfid);
mas01mc@292 1160 close(dbfid);
mas01mc@292 1161 }
mas01mc@292 1162
mas01mc@292 1163 int G::unserialize_lsh_header(char* filename){
mas01mc@292 1164
mas01mc@292 1165 int dbfid;
mas01mc@292 1166 char* db;
mas01mc@292 1167 // Test to see if file exists
mas01mc@292 1168 if((dbfid = open (filename, O_RDONLY)) < 0)
mas01mc@292 1169 error("Can't open the file", filename, "open");
mas01mc@292 1170 close(dbfid);
mas01mc@292 1171 dbfid = serial_open(filename, 0); // open for read
mas01mc@292 1172 db = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 0);// get database pointer
mas01mc@292 1173 serial_get_header(db); // read header
mas01mc@292 1174 serial_munmap(db, O2_SERIAL_HEADER_SIZE); // drop mmap
mas01mc@292 1175
mas01mc@292 1176 // Unserialize header parameters
mas01mc@292 1177 H::L = lshHeader->numTables;
mas01mc@292 1178 H::m = (Uns32T)( (1.0 + sqrt(1 + 8.0*(int)H::L)) / 2.0);
mas01mc@292 1179 H::N = lshHeader->numRows;
mas01mc@292 1180 H::C = lshHeader->numCols;
mas01mc@292 1181 H::k = lshHeader->numFuns;
mas01mc@292 1182 H::d = lshHeader->dataDim;
mas01mc@293 1183 H::w = lshHeader->binWidth;
mas01mc@293 1184 H::radius = lshHeader->radius;
mas01mc@293 1185 H::maxp = lshHeader->maxp;
mas01mc@296 1186 H::pointCount = lshHeader->pointCount;
mas01mc@292 1187
mas01mc@292 1188 return dbfid;
mas01mc@292 1189 }
mas01mc@292 1190
mas01mc@292 1191 // unserialize the LSH parameters
mas01mc@292 1192 // we leave the LSH tree on disk as a flat file
mas01mc@292 1193 // it is this flat file that we search by memory mapping
mas01mc@292 1194 void G::unserialize_lsh_functions(int dbfid){
mas01mc@292 1195 Uns32T j, kk;
mas01mc@292 1196 float* pf;
mas01mc@292 1197 Uns32T* pu;
mas01mc@292 1198
mas01mc@292 1199 // Load the hash functions into core
mas01mc@292 1200 char* db = serial_mmap(dbfid, get_serial_hashtable_offset(), 0);// get database pointer again
mas01mc@292 1201
mas01mc@292 1202 pf = get_serial_hashfunction_base(db);
mas01mc@292 1203
mas01mc@292 1204 #ifdef USE_U_FUNCTIONS
mas01mc@292 1205 for( j = 0 ; j < H::m ; j++ ){ // L functions gj(v)
mas01mc@292 1206 for( kk = 0 ; kk < H::k/2 ; kk++ ){ // Normally distributed hash functions
mas01mc@292 1207 #else
mas01mc@292 1208 for( j = 0 ; j < H::L ; j++ ){ // L functions gj(v)
mas01mc@292 1209 for( kk = 0 ; kk < H::k ; kk++ ){ // Normally distributed hash functions
mas01mc@292 1210 #endif
mas01mc@292 1211 for(Uns32T i = 0 ; i < H::d ; i++ )
mas01mc@293 1212 H::A[j][kk][i] = *pf++; // Normally distributed random vectors
mas01mc@292 1213 }
mas01mc@292 1214 }
mas01mc@292 1215 #ifdef USE_U_FUNCTIONS
mas01mc@292 1216 for( j = 0 ; j < H::m ; j++ ) // biases b
mas01mc@292 1217 for( kk = 0 ; kk < H::k/2 ; kk++ )
mas01mc@292 1218 #else
mas01mc@292 1219 for( j = 0 ; j < H::L ; j++ ) // biases b
mas01mc@292 1220 for( kk = 0 ; kk < H::k ; kk++ )
mas01mc@292 1221 #endif
mas01mc@293 1222 H::b[j][kk] = *pf++;
mas01mc@292 1223
mas01mc@292 1224 pu = (Uns32T*)pf;
mas01mc@292 1225 for( j = 0 ; j < H::L ; j++ ) // Z projectors r1
mas01mc@292 1226 for( kk = 0 ; kk < H::k ; kk++ )
mas01mc@292 1227 H::r1[j][kk] = *pu++;
mas01mc@292 1228
mas01mc@292 1229 for( j = 0 ; j < H::L ; j++ ) // Z projectors r2
mas01mc@292 1230 for( kk = 0 ; kk < H::k ; kk++ )
mas01mc@292 1231 H::r2[j][kk] = *pu++;
mas01mc@292 1232
mas01mc@293 1233 serial_munmap(db, get_serial_hashtable_offset());
mas01mc@292 1234 }
mas01mc@292 1235
mas01mc@292 1236 void G::unserialize_lsh_hashtables_format1(int fid){
mas01mc@292 1237 SerialElementT *pe, *pt;
mas01mc@292 1238 Uns32T x,y;
mas01mc@292 1239 Uns32T hashTableSize=sizeof(SerialElementT)*lshHeader->numRows*lshHeader->numCols;
mas01mc@292 1240 // Read the hash tables into core
mas01mc@292 1241 for( x = 0 ; x < H::L ; x++ ){
mas01mc@292 1242 // memory map a single hash table
mas01mc@292 1243 // Align each hash table to page boundary
mas01mc@292 1244 char* dbtable = serial_mmap(fid, hashTableSize, 0,
mas01mc@292 1245 align_up(get_serial_hashtable_offset()+x*hashTableSize, get_page_logn()));
mas01mc@324 1246 #ifdef __CYGWIN__
mas01mc@324 1247 // No madvise in CYGWIN
mas01mc@324 1248 #else
mas01mc@292 1249 if(madvise(dbtable, hashTableSize, MADV_SEQUENTIAL)<0)
mas01mc@292 1250 error("could not advise hashtable memory","","madvise");
mas01mc@324 1251 #endif
mas01mc@292 1252 pt=(SerialElementT*)dbtable;
mas01mc@292 1253 for( y = 0 ; y < H::N ; y++ ){
mas01mc@292 1254 // Move disk pointer to beginning of row
mas01mc@292 1255 pe=pt+y*lshHeader->numCols;
mas01mc@292 1256 unserialize_hashtable_row_format1(pe, h[x]+y);
mas01mc@340 1257 #ifdef LSH_DUMP_CORE_TABLES
mas01mc@292 1258 printf("S[%d,%d]", x, y);
mas01mc@292 1259 serial_bucket_dump(pe);
mas01mc@292 1260 printf("C[%d,%d]", x, y);
mas01mc@292 1261 dump_hashtable_row(h[x][y]);
mas01mc@292 1262 #endif
mas01mc@292 1263 }
mas01mc@292 1264 serial_munmap(dbtable, hashTableSize);
mas01mc@292 1265 }
mas01mc@292 1266 }
mas01mc@292 1267
mas01mc@292 1268 void G::unserialize_hashtable_row_format1(SerialElementT* pe, bucket** b){
mas01mc@292 1269 Uns32T colCount = 0;
mas01mc@292 1270 while(colCount!=lshHeader->numCols && pe->hashValue !=IFLAG){
mas01mc@292 1271 H::p = pe->pointID; // current point ID
mas01mc@292 1272 t2 = pe->hashValue;
mas01mc@292 1273 bucket_insert_point(b);
mas01mc@292 1274 pe++;
mas01mc@292 1275 colCount++;
mas01mc@292 1276 }
mas01mc@292 1277 }
mas01mc@292 1278
mas01mc@340 1279 void G::unserialize_lsh_hashtables_format2(FILE* dbFile, bool forMerge){
mas01mc@292 1280 Uns32T x=0,y=0;
mas01mc@292 1281
mas01mc@292 1282 // Seek to hashtable base offset
mas01mc@336 1283 if(fseek(dbFile, get_serial_hashtable_offset(), SEEK_SET)){
mas01mc@336 1284 fclose(dbFile);
mas01mc@336 1285 error("fSeek error in unserialize_lsh_hashtables_format2");
mas01mc@292 1286 }
mas01mc@292 1287
mas01mc@292 1288 // Read the hash tables into core (structure is given in header)
mas01mc@292 1289 while( x < H::L){
mas01mc@336 1290 if(fread(&(H::t1), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1291 fclose(dbFile);
mas01mc@292 1292 error("Read error","unserialize_lsh_hashtables_format2()");
mas01mc@292 1293 }
mas01mc@306 1294 if(H::t1==O2_SERIAL_TOKEN_ENDTABLE)
mas01mc@292 1295 x++; // End of table
mas01mc@292 1296 else
mas01mc@292 1297 while(y < H::N){
mas01mc@306 1298 // Read a row and move file pointer to beginning of next row or _bittable
mas01mc@306 1299 if(!(H::t1==O2_SERIAL_TOKEN_T1)){
mas01mc@336 1300 fclose(dbFile);
mas01mc@306 1301 error("State matchine error T1","unserialize_lsh_hashtables_format2()");
mas01mc@292 1302 }
mas01mc@336 1303 if(fread(&(H::t1), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1304 fclose(dbFile);
mas01mc@306 1305 error("Read error: t1","unserialize_lsh_hashtables_format2()");
mas01mc@306 1306 }
mas01mc@306 1307 y = H::t1;
mas01mc@292 1308 if(y>=H::N){
mas01mc@336 1309 fclose(dbFile);
mas01mc@292 1310 error("Unserialized hashtable row pointer out of range","unserialize_lsh_hashtables_format2()");
mas01mc@292 1311 }
mas01mc@340 1312 Uns32T token = 0;
mas01mc@340 1313 #ifdef LSH_CORE_ARRAY
mas01mc@340 1314 Uns32T numElements;
mas01mc@340 1315 if(fread(&numElements, sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@340 1316 fclose(dbFile);
mas01mc@340 1317 error("Read error: numElements","unserialize_lsh_hashtables_format2()");
mas01mc@340 1318 }
mas01mc@292 1319
mas01mc@340 1320 // BACKWARD COMPATIBILITY: check to see if T2 or END token was read
mas01mc@340 1321 if(numElements==O2_SERIAL_TOKEN_T2 || numElements==O2_SERIAL_TOKEN_ENDTABLE ){
mas01mc@340 1322 forMerge=true; // Force use of dynamic linked list core format
mas01mc@340 1323 token = numElements;
mas01mc@340 1324 }
mas01mc@340 1325
mas01mc@340 1326 if(forMerge)
mas01mc@340 1327 // Use linked list CORE format
mas01mc@340 1328 token = unserialize_hashtable_row_format2(dbFile, h[x]+y, token);
mas01mc@340 1329 else
mas01mc@340 1330 // Use ARRAY CORE format with numElements counter
mas01mc@340 1331 token = unserialize_hashtable_row_to_array(dbFile, h[x]+y, numElements);
mas01mc@340 1332 #else
mas01mc@340 1333 token = unserialize_hashtable_row_format2(dbFile, h[x]+y);
mas01mc@513 1334 #endif
mas01mc@292 1335 // Check that token is valid
mas01mc@306 1336 if( !(token==O2_SERIAL_TOKEN_T1 || token==O2_SERIAL_TOKEN_ENDTABLE) ){
mas01mc@336 1337 fclose(dbFile);
mas01mc@292 1338 error("State machine error end of row/table", "unserialize_lsh_hashtables_format2()");
mas01mc@292 1339 }
mas01mc@292 1340 // Check for end of table flag
mas01mc@306 1341 if(token==O2_SERIAL_TOKEN_ENDTABLE){
mas01mc@292 1342 x++;
mas01mc@292 1343 break;
mas01mc@292 1344 }
mas01mc@292 1345 // Check for new row flag
mas01mc@306 1346 if(token==O2_SERIAL_TOKEN_T1)
mas01mc@292 1347 H::t1 = token;
mas01mc@292 1348 }
mas01mc@292 1349 }
mas01mc@513 1350 #ifdef LSH_DUMP_CORE_TABLES
mas01mc@513 1351 dump_hashtables();
mas01mc@513 1352 #endif
mas01mc@306 1353 }
mas01mc@292 1354
mas01mc@340 1355 Uns32T G::unserialize_hashtable_row_format2(FILE* dbFile, bucket** b, Uns32T token){
mas01mc@292 1356 bool pointFound = false;
mas01mc@340 1357
mas01mc@340 1358 if(token)
mas01mc@340 1359 H::t2 = token;
mas01mc@340 1360 else if(fread(&(H::t2), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@340 1361 fclose(dbFile);
mas01mc@340 1362 error("Read error T2 token","unserialize_hashtable_row_format2");
mas01mc@340 1363 }
mas01mc@340 1364
mas01mc@306 1365 if( !(H::t2==O2_SERIAL_TOKEN_ENDTABLE || H::t2==O2_SERIAL_TOKEN_T2)){
mas01mc@336 1366 fclose(dbFile);
mas01mc@292 1367 error("State machine error: expected E or T2");
mas01mc@292 1368 }
mas01mc@340 1369
mas01mc@306 1370 while(!(H::t2==O2_SERIAL_TOKEN_ENDTABLE || H::t2==O2_SERIAL_TOKEN_T1)){
mas01mc@292 1371 pointFound=false;
mas01mc@292 1372 // Check for T2 token
mas01mc@306 1373 if(H::t2!=O2_SERIAL_TOKEN_T2){
mas01mc@336 1374 fclose(dbFile);
mas01mc@292 1375 error("State machine error T2 token", "unserialize_hashtable_row_format2()");
mas01mc@306 1376 }
mas01mc@292 1377 // Read t2 value
mas01mc@336 1378 if(fread(&(H::t2), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1379 fclose(dbFile);
mas01mc@292 1380 error("Read error t2","unserialize_hashtable_row_format2");
mas01mc@292 1381 }
mas01mc@336 1382 if(fread(&(H::p), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1383 fclose(dbFile);
mas01mc@292 1384 error("Read error H::p","unserialize_hashtable_row_format2");
mas01mc@292 1385 }
mas01mc@306 1386 while(!(H::p==O2_SERIAL_TOKEN_ENDTABLE || H::p==O2_SERIAL_TOKEN_T1 || H::p==O2_SERIAL_TOKEN_T2 )){
mas01mc@292 1387 pointFound=true;
mas01mc@292 1388 bucket_insert_point(b);
mas01mc@336 1389 if(fread(&(H::p), sizeof(Uns32T), 1, dbFile) != 1){
mas01mc@336 1390 fclose(dbFile);
mas01mc@292 1391 error("Read error H::p","unserialize_hashtable_row_format2");
mas01mc@292 1392 }
mas01mc@292 1393 }
mas01mc@292 1394 if(!pointFound)
mas01mc@292 1395 error("State machine error: point", "unserialize_hashtable_row_format2()");
mas01mc@306 1396 H::t2 = H::p; // Copy last found token to t2
mas01mc@292 1397 }
mas01mc@292 1398 return H::t2; // holds current token
mas01mc@292 1399 }
mas01mc@292 1400
mas01mc@340 1401 // Unserialize format2 hashtable row to a CORE_ARRAY pointed to
mas01mc@340 1402 // by the hashtable row pointer: rowPtr
mas01mc@340 1403 //
mas01mc@340 1404 // numElements is the total number of t2 buckets plus points
mas01mc@340 1405 // memory required is numElements+1+sizeof(hashtable ptr)
mas01mc@340 1406 //
mas01mc@340 1407 // numElements = numPoints + numBuckets
mas01mc@340 1408 //
mas01mc@340 1409 // During inserts (merges) new hashtable entries are appened at rowPtr+numPoints+numBuckets+1
mas01mc@340 1410 //
mas01mc@340 1411 // ASSUME: that LSH_LIST_HEAD_COUNTERS is set so that the first hashtable bucket is used to count
mas01mc@340 1412 // point and bucket entries
mas01mc@340 1413 //
mas01mc@340 1414 // We store the values of numPoints and numBuckets in separate fields of the first bucket
mas01mc@340 1415 // rowPtr->t2 // numPoints
mas01mc@340 1416 // (Uns32T)(rowPtr->snext) // numBuckets
mas01mc@340 1417 //
mas01mc@340 1418 // We cast the rowPtr->next pointer to (Uns32*) malloc(numElements*sizeof(Uns32T) + sizeof(bucket*))
mas01mc@340 1419 // To get to the fist bucket, we use
mas01mc@340 1420 //
mas01mc@340 1421
mas01mc@340 1422 #define READ_UNS32T(VAL,TOKENSTR) if(fread(VAL, sizeof(Uns32T), 1, dbFile) != 1){\
mas01mc@340 1423 fclose(dbFile);error("Read error unserialize_hashtable_format2",TOKENSTR);}
mas01mc@340 1424
mas01mc@340 1425 #define TEST_TOKEN(TEST, TESTSTR) if(TEST){fclose(dbFile);error("State machine error: ", TESTSTR);}
mas01mc@340 1426
mas01mc@340 1427 #define SKIP_BITS_LEFT_SHIFT_MSB (30)
mas01mc@340 1428
mas01mc@340 1429 #define SKIP_BITS_RIGHT_SHIFT_MSB (28)
mas01mc@340 1430 #define SKIP_BITS_RIGHT_SHIFT_LSB (30)
mas01mc@340 1431
mas01mc@340 1432 #define MAX_POINTS_IN_BUCKET_CORE_ARRAY (16)
mas01mc@340 1433 #define LSH_CORE_ARRAY_END_ROW_TOKEN (0xFFFFFFFD)
mas01mc@340 1434
mas01mc@340 1435 // Encode the skip bits. Zero if only one point, MAX 8 (plus first == 9)
mas01mc@340 1436 #define ENCODE_POINT_SKIP_BITS TEST_TOKEN(!numPointsThisBucket, "no points found");\
mas01mc@340 1437 if(numPointsThisBucket==1){\
mas01mc@340 1438 secondPtr=ap++;\
mas01mc@340 1439 *secondPtr=0;\
mas01mc@340 1440 numPoints++;\
mas01mc@340 1441 }\
mas01mc@340 1442 if(numPointsThisBucket>1){\
mas01mc@340 1443 *firstPtr |= ( (numPointsThisBucket-1) & 0x3 ) << SKIP_BITS_LEFT_SHIFT_MSB;\
mas01mc@340 1444 *secondPtr |= ( ( (numPointsThisBucket-1) & 0xC) >> 2 ) << SKIP_BITS_LEFT_SHIFT_MSB;}
mas01mc@340 1445
mas01mc@340 1446 Uns32T G::unserialize_hashtable_row_to_array(FILE* dbFile, bucket** rowPP, Uns32T numElements){
mas01mc@340 1447 Uns32T numPointsThisBucket = 0;
mas01mc@340 1448 Uns32T numBuckets = 0;
mas01mc@340 1449 Uns32T numPoints = 0;
mas01mc@340 1450 Uns32T* firstPtr = 0;
mas01mc@340 1451 Uns32T* secondPtr = 0;
mas01mc@340 1452
mas01mc@340 1453 // Initialize new row
mas01mc@340 1454 if(!*rowPP){
mas01mc@340 1455 *rowPP = new bucket();
mas01mc@340 1456 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@340 1457 (*rowPP)->t2 = 0; // Use t2 as a collision counter for the row
mas01mc@340 1458 (*rowPP)->next = 0;
mas01mc@340 1459 #endif
mas01mc@340 1460 }
mas01mc@340 1461 bucket* rowPtr = *rowPP;
mas01mc@340 1462
mas01mc@340 1463 READ_UNS32T(&(H::t2),"t2");
mas01mc@340 1464 TEST_TOKEN(!(H::t2==O2_SERIAL_TOKEN_ENDTABLE || H::t2==O2_SERIAL_TOKEN_T2), "expected E or T2");
mas01mc@340 1465 // Because we encode points in 16-point blocks, we sometimes allocate repeated t2 elements
mas01mc@340 1466 // So over-allocate by a factor of two and realloc later to actual numElements
mas01mc@340 1467 CR_ASSERT(rowPtr->next = (bucket*) malloc((2*numElements+1)*sizeof(Uns32T)+sizeof(bucket**)));
mas01mc@340 1468 Uns32T* ap = reinterpret_cast<Uns32T*>(rowPtr->next); // Cast pointer to Uns32T* for array format
mas01mc@340 1469 while( !(H::t2==O2_SERIAL_TOKEN_ENDTABLE || H::t2==O2_SERIAL_TOKEN_T1) ){
mas01mc@340 1470 numPointsThisBucket = 0;// reset bucket point counter
mas01mc@340 1471 secondPtr = 0; // reset second-point pointer
mas01mc@340 1472 TEST_TOKEN(H::t2!=O2_SERIAL_TOKEN_T2, "expected T2");
mas01mc@340 1473 READ_UNS32T(&(H::t2), "Read error t2");
mas01mc@340 1474 *ap++ = H::t2; // Insert t2 value into array
mas01mc@340 1475 numBuckets++;
mas01mc@340 1476 READ_UNS32T(&(H::p), "Read error H::p");
mas01mc@340 1477 while(!(H::p==O2_SERIAL_TOKEN_ENDTABLE || H::p==O2_SERIAL_TOKEN_T1 || H::p==O2_SERIAL_TOKEN_T2 )){
mas01mc@340 1478 if(numPointsThisBucket==MAX_POINTS_IN_BUCKET_CORE_ARRAY){
mas01mc@340 1479 ENCODE_POINT_SKIP_BITS;
mas01mc@340 1480 *ap++ = H::t2; // Extra element
mas01mc@340 1481 numBuckets++; // record this as a new bucket
mas01mc@340 1482 numPointsThisBucket=0; // reset bucket point counter
mas01mc@340 1483 secondPtr = 0; // reset second-point pointer
mas01mc@340 1484 }
mas01mc@340 1485 if( ++numPointsThisBucket == 1 )
mas01mc@340 1486 firstPtr = ap; // store pointer to first point to insert skip bits later on
mas01mc@340 1487 else if( numPointsThisBucket == 2 )
mas01mc@340 1488 secondPtr = ap; // store pointer to first point to insert skip bits later on
mas01mc@340 1489 numPoints++;
mas01mc@340 1490 *ap++ = H::p;
mas01mc@340 1491 READ_UNS32T(&(H::p), "Read error H::p");
mas01mc@340 1492 }
mas01mc@340 1493 ENCODE_POINT_SKIP_BITS;
mas01mc@340 1494 H::t2 = H::p; // Copy last found token to t2
mas01mc@340 1495 }
mas01mc@340 1496 // Reallocate the row to its actual size
mas01mc@340 1497 CR_ASSERT(rowPtr->next = (bucket*) realloc(rowPtr->next, (numBuckets+numPoints+1)*sizeof(Uns32T)+sizeof(bucket**)));
mas01mc@340 1498 // Record the sizes at the head of the row
mas01mc@344 1499 rowPtr->snext.numBuckets = numBuckets;
mas01mc@340 1500 rowPtr->t2 = numPoints;
mas01mc@340 1501 // Place end of row marker
mas01mc@340 1502 *ap++ = LSH_CORE_ARRAY_END_ROW_TOKEN;
mas01mc@340 1503 // Set the LSH_CORE_ARRAY_BIT to identify data structure for insertion and retrieval
mas01mc@340 1504 rowPtr->t2 |= LSH_CORE_ARRAY_BIT;
mas01mc@340 1505 // Allocate a new dynamic list head at the end of the array
mas01mc@340 1506 bucket** listPtr = reinterpret_cast<bucket**> (ap);
mas01mc@340 1507 *listPtr = 0;
mas01mc@340 1508 // Return current token
mas01mc@340 1509 return H::t2; // return H::t2 which holds current token [E or T1]
mas01mc@340 1510 }
mas01mc@340 1511
mas01mc@340 1512
mas01mc@340 1513
mas01mc@340 1514 // *p is a pointer to the beginning of a hashtable row array
mas01mc@340 1515 // The array consists of t2 hash keys and one or more point identifiers p for each hash key
mas01mc@340 1516 // Retrieval is performed by generating a hash key query_t2 for query point q
mas01mc@340 1517 // We identify the row that t2 is stored in using a secondary hash t1, this row is the entry
mas01mc@340 1518 // point for retrieve_from_core_hashtable_array
mas01mc@340 1519 #define SKIP_BITS (0xC0000000)
mas01mc@340 1520 void G::retrieve_from_core_hashtable_array(Uns32T* p, Uns32T qpos){
mas01mc@340 1521 Uns32T skip;
mas01mc@340 1522 Uns32T t2;
mas01mc@340 1523 Uns32T p1;
mas01mc@340 1524 Uns32T p2;
mas01mc@340 1525
mas01mc@340 1526 CR_ASSERT(p);
mas01mc@340 1527
mas01mc@340 1528 do{
mas01mc@340 1529 t2 = *p++;
mas01mc@340 1530 if( t2 > H::t2 )
mas01mc@340 1531 return;
mas01mc@340 1532 p1 = *p++;
mas01mc@340 1533 p2 = *p++;
mas01mc@340 1534 skip = (( p1 & SKIP_BITS ) >> SKIP_BITS_RIGHT_SHIFT_LSB) + (( p2 & SKIP_BITS ) >> SKIP_BITS_RIGHT_SHIFT_MSB);
mas01mc@340 1535 if( t2 == H::t2 ){
mas01mc@340 1536 add_point_callback(calling_instance, p1 ^ (p1 & SKIP_BITS), qpos, radius);
mas01mc@340 1537 if(skip--){
mas01mc@340 1538 add_point_callback(calling_instance, p2 ^ (p2 & SKIP_BITS), qpos, radius);
mas01mc@340 1539 while(skip-- )
mas01mc@340 1540 add_point_callback(calling_instance, *p++, qpos, radius);
mas01mc@340 1541 }
mas01mc@340 1542 }
mas01mc@340 1543 else
mas01mc@340 1544 if(*p != LSH_CORE_ARRAY_END_ROW_TOKEN)
mas01mc@340 1545 p = p + skip;
mas01mc@340 1546 }while( *p != LSH_CORE_ARRAY_END_ROW_TOKEN );
mas01mc@340 1547 }
mas01mc@513 1548
mas01mc@513 1549 void G::dump_hashtables(){
mas01mc@513 1550 for(Uns32T x = 0; x < H::L ; x++)
mas01mc@513 1551 for(Uns32T y = 0; y < H::N ; y++){
mas01mc@513 1552 bucket* bPtr = h[x][y];
mas01mc@513 1553 if(bPtr){
mas01mc@513 1554 printf("C[%d,%d]", x, y);
mas01mc@513 1555 #ifdef LSH_LIST_HEAD_COUNTERS
mas01mc@513 1556 printf("[numBuckets=%d]",bPtr->snext.numBuckets);
mas01mc@513 1557 if(bPtr->t2&LSH_CORE_ARRAY_BIT) {
mas01mc@513 1558 dump_core_hashtable_array((Uns32T*)(bPtr->next));
mas01mc@513 1559 }
mas01mc@513 1560 else {
mas01mc@513 1561 dump_hashtable_row(bPtr->next);
mas01mc@513 1562 }
mas01mc@513 1563 #else
mas01mc@513 1564 dump_hashtable_row(bPtr);
mas01mc@513 1565 #endif
mas01mc@513 1566 printf("\n");
mas01mc@513 1567 fflush(stdout);
mas01mc@513 1568 }
mas01mc@513 1569 }
mas01mc@513 1570 }
mas01mc@513 1571
mas01mc@513 1572 void G::dump_core_hashtable_array(Uns32T* p){
mas01mc@513 1573 Uns32T skip;
mas01mc@513 1574 Uns32T t2;
mas01mc@513 1575 Uns32T p1;
mas01mc@513 1576 Uns32T p2;
mas01mc@513 1577 CR_ASSERT(p);
mas01mc@513 1578 do{
mas01mc@513 1579 t2 = *p++;
mas01mc@513 1580 p1 = *p++;
mas01mc@513 1581 p2 = *p++;
mas01mc@513 1582 skip = (( p1 & SKIP_BITS ) >> SKIP_BITS_RIGHT_SHIFT_LSB) + (( p2 & SKIP_BITS ) >> SKIP_BITS_RIGHT_SHIFT_MSB);
mas01mc@513 1583 printf("(%0x, %0x)", t2, p1 ^ (p1 & SKIP_BITS));
mas01mc@513 1584 if(skip--){
mas01mc@513 1585 printf("(%0x, %0x)", t2, p2 ^ (p2 & SKIP_BITS));
mas01mc@513 1586 while(skip-- )
mas01mc@513 1587 printf("(%0x, %0x)", t2, *p++);
mas01mc@513 1588 }
mas01mc@513 1589 }while( *p != LSH_CORE_ARRAY_END_ROW_TOKEN );
mas01mc@513 1590 }
mas01mc@340 1591
mas01mc@292 1592 void G::dump_hashtable_row(bucket* p){
mas01mc@292 1593 while(p && p->t2!=IFLAG){
mas01mc@344 1594 sbucket* sbp = p->snext.ptr;
mas01mc@292 1595 while(sbp){
mas01mc@292 1596 printf("(%0X,%u)", p->t2, sbp->pointID);
mas01mc@292 1597 fflush(stdout);
mas01mc@292 1598 sbp=sbp->snext;
mas01mc@292 1599 }
mas01mc@292 1600 p=p->next;
mas01mc@292 1601 }
mas01mc@292 1602 printf("\n");
mas01mc@292 1603 }
mas01mc@292 1604
mas01mc@292 1605
mas01mc@292 1606 // G::serial_retrieve_point( ... )
mas01mc@292 1607 // retrieves (pointID) from a serialized LSH database
mas01mc@292 1608 //
mas01mc@292 1609 // inputs:
mas01mc@292 1610 // filename - file name of serialized LSH database
mas01mc@292 1611 // vv - query point set
mas01mc@292 1612 //
mas01mc@292 1613 // outputs:
mas01mc@292 1614 // inserts retrieved points into add_point() callback method
mas01mc@292 1615 void G::serial_retrieve_point_set(char* filename, vector<vector<float> >& vv, ReporterCallbackPtr add_point, void* caller)
mas01mc@292 1616 {
mas01mc@292 1617 int dbfid = serial_open(filename, 0); // open for read
mas01mc@292 1618 char* dbheader = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 0);// get database pointer
mas01mc@292 1619 serial_get_header(dbheader); // read header
mas01mc@292 1620 serial_munmap(dbheader, O2_SERIAL_HEADER_SIZE); // drop header mmap
mas01mc@292 1621
mas01mc@292 1622 if((lshHeader->flags & O2_SERIAL_FILEFORMAT2)){
mas01mc@336 1623 serial_close(dbfid);
mas01mc@292 1624 error("serial_retrieve_point_set is for SERIAL_FILEFORMAT1 only");
mas01mc@292 1625 }
mas01mc@292 1626
mas01mc@292 1627 // size of each hash table
mas01mc@292 1628 Uns32T hashTableSize=sizeof(SerialElementT)*lshHeader->numRows*lshHeader->numCols;
mas01mc@292 1629 calling_instance = caller; // class instance variable used in ...bucket_chain_point()
mas01mc@292 1630 add_point_callback = add_point;
mas01mc@292 1631
mas01mc@292 1632 for(Uns32T j=0; j<L; j++){
mas01mc@292 1633 // memory map a single hash table for random access
mas01mc@292 1634 char* db = serial_mmap(dbfid, hashTableSize, 0,
mas01mc@292 1635 align_up(get_serial_hashtable_offset()+j*hashTableSize,get_page_logn()));
mas01mc@324 1636 #ifdef __CYGWIN__
mas01mc@324 1637 // No madvise in CYGWIN
mas01mc@324 1638 #else
mas01mc@292 1639 if(madvise(db, hashTableSize, MADV_RANDOM)<0)
mas01mc@292 1640 error("could not advise local hashtable memory","","madvise");
mas01mc@324 1641 #endif
mas01mc@292 1642 SerialElementT* pe = (SerialElementT*)db ;
mas01mc@292 1643 for(Uns32T qpos=0; qpos<vv.size(); qpos++){
mas01mc@293 1644 H::compute_hash_functions(vv[qpos]);
mas01mc@293 1645 H::generate_hash_keys(*(g+j),*(r1+j),*(r2+j));
mas01mc@292 1646 serial_bucket_chain_point(pe+t1*lshHeader->numCols, qpos); // Point to correct row
mas01mc@292 1647 }
mas01mc@292 1648 serial_munmap(db, hashTableSize); // drop hashtable mmap
mas01mc@292 1649 }
mas01mc@292 1650 serial_close(dbfid);
mas01mc@292 1651 }
mas01mc@292 1652
mas01mc@292 1653 void G::serial_retrieve_point(char* filename, vector<float>& v, Uns32T qpos, ReporterCallbackPtr add_point, void* caller){
mas01mc@292 1654 int dbfid = serial_open(filename, 0); // open for read
mas01mc@292 1655 char* dbheader = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 0);// get database pointer
mas01mc@292 1656 serial_get_header(dbheader); // read header
mas01mc@292 1657 serial_munmap(dbheader, O2_SERIAL_HEADER_SIZE); // drop header mmap
mas01mc@292 1658
mas01mc@292 1659 if((lshHeader->flags & O2_SERIAL_FILEFORMAT2)){
mas01mc@336 1660 serial_close(dbfid);
mas01mc@292 1661 error("serial_retrieve_point is for SERIAL_FILEFORMAT1 only");
mas01mc@292 1662 }
mas01mc@292 1663
mas01mc@292 1664 // size of each hash table
mas01mc@292 1665 Uns32T hashTableSize=sizeof(SerialElementT)*lshHeader->numRows*lshHeader->numCols;
mas01mc@292 1666 calling_instance = caller;
mas01mc@292 1667 add_point_callback = add_point;
mas01mc@293 1668 H::compute_hash_functions(v);
mas01mc@292 1669 for(Uns32T j=0; j<L; j++){
mas01mc@292 1670 // memory map a single hash table for random access
mas01mc@292 1671 char* db = serial_mmap(dbfid, hashTableSize, 0,
mas01mc@292 1672 align_up(get_serial_hashtable_offset()+j*hashTableSize,get_page_logn()));
mas01mc@324 1673 #ifdef __CYGWIN__
mas01mc@324 1674 // No madvise in CYGWIN
mas01mc@324 1675 #else
mas01mc@292 1676 if(madvise(db, hashTableSize, MADV_RANDOM)<0)
mas01mc@292 1677 error("could not advise local hashtable memory","","madvise");
mas01mc@324 1678 #endif
mas01mc@292 1679 SerialElementT* pe = (SerialElementT*)db ;
mas01mc@293 1680 H::generate_hash_keys(*(g+j),*(r1+j),*(r2+j));
mas01mc@292 1681 serial_bucket_chain_point(pe+t1*lshHeader->numCols, qpos); // Point to correct row
mas01mc@292 1682 serial_munmap(db, hashTableSize); // drop hashtable mmap
mas01mc@292 1683 }
mas01mc@292 1684 serial_close(dbfid);
mas01mc@292 1685 }
mas01mc@292 1686
mas01mc@292 1687 void G::serial_dump_tables(char* filename){
mas01mc@292 1688 int dbfid = serial_open(filename, 0); // open for read
mas01mc@292 1689 char* dbheader = serial_mmap(dbfid, O2_SERIAL_HEADER_SIZE, 0);// get database pointer
mas01mc@292 1690 serial_get_header(dbheader); // read header
mas01mc@292 1691 serial_munmap(dbheader, O2_SERIAL_HEADER_SIZE); // drop header mmap
mas01mc@292 1692 Uns32T hashTableSize=sizeof(SerialElementT)*lshHeader->numRows*lshHeader->numCols;
mas01mc@292 1693 for(Uns32T j=0; j<L; j++){
mas01mc@292 1694 // memory map a single hash table for random access
mas01mc@292 1695 char* db = serial_mmap(dbfid, hashTableSize, 0,
mas01mc@292 1696 align_up(get_serial_hashtable_offset()+j*hashTableSize,get_page_logn()));
mas01mc@324 1697 #ifdef __CYGWIN__
mas01mc@324 1698 // No madvise in CYGWIN
mas01mc@324 1699 #else
mas01mc@292 1700 if(madvise(db, hashTableSize, MADV_SEQUENTIAL)<0)
mas01mc@292 1701 error("could not advise local hashtable memory","","madvise");
mas01mc@324 1702 #endif
mas01mc@292 1703 SerialElementT* pe = (SerialElementT*)db ;
mas01mc@292 1704 printf("*********** TABLE %d ***************\n", j);
mas01mc@292 1705 fflush(stdout);
mas01mc@292 1706 int count=0;
mas01mc@292 1707 do{
mas01mc@292 1708 printf("[%d,%d]", j, count++);
mas01mc@292 1709 fflush(stdout);
mas01mc@292 1710 serial_bucket_dump(pe);
mas01mc@292 1711 pe+=lshHeader->numCols;
mas01mc@292 1712 }while(pe<(SerialElementT*)db+lshHeader->numRows*lshHeader->numCols);
mas01mc@292 1713 }
mas01mc@292 1714
mas01mc@292 1715 }
mas01mc@292 1716
mas01mc@292 1717 void G::serial_bucket_dump(SerialElementT* pe){
mas01mc@292 1718 SerialElementT* pend = pe+lshHeader->numCols;
mas01mc@292 1719 while( !(pe->hashValue==IFLAG || pe==pend ) ){
mas01mc@292 1720 printf("(%0X,%u)",pe->hashValue,pe->pointID);
mas01mc@292 1721 pe++;
mas01mc@292 1722 }
mas01mc@292 1723 printf("\n");
mas01mc@292 1724 fflush(stdout);
mas01mc@292 1725 }
mas01mc@292 1726
mas01mc@292 1727 void G::serial_bucket_chain_point(SerialElementT* pe, Uns32T qpos){
mas01mc@292 1728 SerialElementT* pend = pe+lshHeader->numCols;
mas01mc@292 1729 while( !(pe->hashValue==IFLAG || pe==pend ) ){
mas01mc@292 1730 if(pe->hashValue==t2){ // new match
mas01mc@292 1731 add_point_callback(calling_instance, pe->pointID, qpos, radius);
mas01mc@292 1732 }
mas01mc@292 1733 pe++;
mas01mc@292 1734 }
mas01mc@292 1735 }
mas01mc@292 1736
mas01mc@292 1737 void G::bucket_chain_point(bucket* p, Uns32T qpos){
mas01mc@292 1738 if(!p || p->t2==IFLAG)
mas01mc@292 1739 return;
mas01mc@292 1740 if(p->t2==t2){ // match
mas01mc@344 1741 sbucket_chain_point(p->snext.ptr, qpos); // add to reporter
mas01mc@292 1742 }
mas01mc@292 1743 if(p->next){
mas01mc@292 1744 bucket_chain_point(p->next, qpos); // recurse
mas01mc@292 1745 }
mas01mc@292 1746 }
mas01mc@292 1747
mas01mc@292 1748 void G::sbucket_chain_point(sbucket* p, Uns32T qpos){
mas01mc@292 1749 add_point_callback(calling_instance, p->pointID, qpos, radius);
mas01mc@292 1750 if(p->snext){
mas01mc@292 1751 sbucket_chain_point(p->snext, qpos);
mas01mc@292 1752 }
mas01mc@292 1753 }
mas01mc@292 1754
mas01mc@292 1755 void G::get_lock(int fd, bool exclusive) {
mas01mc@292 1756 struct flock lock;
mas01mc@292 1757 int status;
mas01mc@292 1758 lock.l_type = exclusive ? F_WRLCK : F_RDLCK;
mas01mc@292 1759 lock.l_whence = SEEK_SET;
mas01mc@292 1760 lock.l_start = 0;
mas01mc@292 1761 lock.l_len = 0; /* "the whole file" */
mas01mc@292 1762 retry:
mas01mc@292 1763 do {
mas01mc@292 1764 status = fcntl(fd, F_SETLKW, &lock);
mas01mc@292 1765 } while (status != 0 && errno == EINTR);
mas01mc@292 1766 if (status) {
mas01mc@292 1767 if (errno == EAGAIN) {
mas01mc@292 1768 sleep(1);
mas01mc@292 1769 goto retry;
mas01mc@292 1770 } else {
mas01mc@292 1771 error("fcntl lock error", "", "fcntl");
mas01mc@292 1772 }
mas01mc@292 1773 }
mas01mc@292 1774 }
mas01mc@292 1775
mas01mc@292 1776 void G::release_lock(int fd) {
mas01mc@292 1777 struct flock lock;
mas01mc@292 1778 int status;
mas01mc@292 1779
mas01mc@292 1780 lock.l_type = F_UNLCK;
mas01mc@292 1781 lock.l_whence = SEEK_SET;
mas01mc@292 1782 lock.l_start = 0;
mas01mc@292 1783 lock.l_len = 0;
mas01mc@292 1784
mas01mc@292 1785 status = fcntl(fd, F_SETLKW, &lock);
mas01mc@292 1786
mas01mc@292 1787 if (status)
mas01mc@292 1788 error("fcntl unlock error", "", "fcntl");
mas01mc@292 1789 }