New rank/select data structure based on sadakane's code
[SXSI/XMLTree.git] / libcds / src / static_sequence / static_sequence.cpp
1 /* static_sequence.cpp
2  * Copyright (C) 2008, Francisco Claude, all rights reserved.
3  *
4  * static_sequence definition
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19  *
20  */
21  
22 #include <static_sequence.h>
23
24 static_sequence::static_sequence() {}
25 static_sequence::~static_sequence() {}
26 uint static_sequence::length() { return len; }
27
28 uint static_sequence::count(uint s) {
29   return rank(s,len-1);
30 }
31
32 static_sequence * static_sequence::load(FILE * fp) {
33   uint rd;
34   if(fread(&rd,sizeof(uint),1,fp)!=1) return NULL;
35   fseek(fp,-sizeof(uint),SEEK_CUR);
36   switch(rd) {
37     case WVTREE_HDR: return static_sequence_wvtree::load(fp);
38     case GMR_CHUNK_HDR: return static_sequence_gmr_chunk::load(fp);
39     case GMR_HDR: return static_sequence_gmr::load(fp);
40     case WVTREE_NOPTRS_HDR: return static_sequence_wvtree_noptrs::load(fp);
41                 case BS_HDR: return static_sequence_bs::load(fp);
42   }
43   return NULL;
44 }
45
46 bool static_sequence::test(uint * seq, uint n) {
47         uint sigma = 0;
48         for(uint i=0;i<n;i++)
49                 sigma = max(sigma,seq[i]);
50         uint * occ = new uint[sigma+1];
51         for(uint i=0;i<=sigma;i++)
52                 occ[i] = 0;
53         for(uint i=0;i<n;i++) {
54                 occ[seq[i]]++;
55                 if(rank(seq[i],i)!=occ[seq[i]]) {
56                         cout << "rank failed!" << endl;
57                         cout << "rank("<<seq[i]<<","<<i<<")="<<rank(seq[i],i)<<endl;
58                         cout << "expected result: " << occ[seq[i]] << endl;
59                         delete [] occ;
60                         return false;
61                 }
62                 if(i>0 && rank(seq[i],i-1)!=occ[seq[i]]-1) {
63                         cout << "rank-1 failed!" << endl;
64                         delete [] occ;
65                         return false;
66                 }
67                 if(select(seq[i],occ[seq[i]])!=i) {
68                         cout << "select failed!" << endl;
69                         cout << "select(" << seq[i] << "," << occ[seq[i]] << ")="<<select(seq[i],occ[seq[i]]) << endl;
70                         cout << "i=" << i << "  rank(" << seq[i] << ",i)=" << rank(seq[i],i) << endl;
71                         delete [] occ;
72                         return false;
73                 }
74                 if(access(i)!=seq[i]) {
75                         cout << "access failed!" << endl;
76                         delete [] occ;
77                         return false;
78                 }
79         }
80         delete [] occ;
81         return true;
82 }
83