delete [] right;
}
+// Deletes symbols array!
wt_node_internal::wt_node_internal(uchar * symbols, uint n, uint l, wt_coder * c, static_bitsequence_builder * bmb) {
uint * ibitmap = new uint[n/W+1];
for(uint i=0;i<n/W+1;i++)
match_left = false;
}
}
+
+ delete [] symbols;
+ symbols = 0;
+
if(count_left>0) {
if(match_left/* && c->done(left[0],l+1)*/)
- left_child = new wt_node_leaf((uint)left[0], count_left);
+ {
+ left_child = new wt_node_leaf((uint)left[0], count_left);
+ delete [] left;
+ left = 0;
+ }
else
- left_child = new wt_node_internal(left, count_left, l+1, c, bmb);
+ {
+ left_child = new wt_node_internal(left, count_left, l+1, c, bmb);
+ left = 0; // Already deleted
+ }
} else {
left_child = NULL;
}
if(count_right>0) {
if(match_right/* && c->done(right[0],l+1)*/)
- right_child = new wt_node_leaf((uint)right[0], count_right);
- else
- right_child = new wt_node_internal(right, count_right, l+1, c, bmb);
+ {
+ right_child = new wt_node_leaf((uint)right[0], count_right);
+ delete [] right;
+ right = 0;
+ }
+ else
+ {
+ right_child = new wt_node_internal(right, count_right, l+1, c, bmb);
+ right = 0; // Already deleted
+ }
} else {
right_child = NULL;
}
- delete [] left;
+ delete [] left; // already deleted if count_left > 0
delete [] right;
}
+wt_node_internal::wt_node_internal(uchar * symbols, uint n, uint l, wt_coder * c, static_bitsequence_builder * bmb, uint left, uint *done) {
+ uint * ibitmap = new uint[n/W+1];
+ for(uint i=0;i<n/W+1;i++)
+ ibitmap[i]=0;
+ for(uint i=0;i<n;i++)
+ if(c->is_set((uint)symbols[i + left],l))
+ bitset(ibitmap,i);
+ bitmap = bmb->build(ibitmap, n);
+ delete [] ibitmap;
+
+ uint count_right = bitmap->rank1(n-1);
+ uint count_left = n-count_right;
+/* uchar * leftarr = new uchar[count_left+1];
+ uchar * rightarr = new uchar[count_right+1];
+ count_right = count_left = 0;
+ for(uint i=0;i<n;i++) {
+ if(bitmap->access(i)) {
+ rightarr[count_right++]=symbols[i+left];
+ }
+ else {
+ leftarr[count_left++]=symbols[i+left];
+ }
+ }
+*/
+
+ for (uint i=0;i<n;i++)
+ set_field(done, 1, i+left, 0);
+
+ for (uint i = 0; i < n; )
+ {
+ uint j = i;
+ uchar swap = symbols[j+left];
+ while (!get_field(done, 1, j+left)) { // swapping
+ ulong k = j;
+ if (!c->is_set(swap,l))
+ j = bitmap->rank0(k)-1;
+ else
+ j = count_left + bitmap->rank1(k)-1;
+ uchar temp = symbols[j+left];
+ symbols[j+left] = swap;
+ swap = temp;
+ set_field(done,1,k+left,1);
+ }
+
+ while (get_field(done,1,i+left))
+ ++i;
+ }
+
+ // checking
+ /* for (uint i=0;i<n;i++)
+ if (!bitget(done,i+left))
+ std::cout << "not swapped: " << i << "\n";
+ for (uint i=0;i<count_left;i++)
+ if (leftarr[i] != symbols[i+left]) //c->is_set(symbols[i+left], l))
+ {
+ std::cout << symbols[i+left] << " != " << leftarr[i] << " lev = " << l << "\n";
+ exit(0);
+ }
+ for (uint i=count_left;i<n;i++)
+ if (rightarr[i-count_left] != symbols[i+left]) //!c->is_set(symbols[i+left],l))
+ std::cout << symbols[i+left] << " != " << rightarr[i-count_left] << " lev = " << l << "\n";
+ */
+ bool match_left = true, match_right = true;
+ for (uint i=1; i < count_left; i++)
+ if (symbols[i+left] != symbols[i+left-1])
+ match_left = false;
+ for (uint i=count_left + 1; i < n; i++)
+ if (symbols[i+left] != symbols[i+left-1])
+ match_right = false;
+
+
+ if(count_left>0) {
+ if(match_left/* && c->done(left[0],l+1)*/)
+ left_child = new wt_node_leaf((uint)symbols[left], count_left);
+ else
+ left_child = new wt_node_internal(symbols, count_left, l+1, c, bmb, left, done);
+ } else {
+ left_child = NULL;
+ }
+ if(count_right>0) {
+ if(match_right/* && c->done(right[0],l+1)*/)
+ right_child = new wt_node_leaf((uint)symbols[left+count_left], count_right);
+ else
+ right_child = new wt_node_internal(symbols, count_right, l+1, c, bmb, left+count_left, done);
+ } else {
+ right_child = NULL;
+ }
+}
+
wt_node_internal::wt_node_internal() { }
}
}
+// return value is rank of symbol (less or equal to the given symbol) that has rank > 0,
+// the parameter symbol is updated accordinly
+uint wt_node_internal::rankLessThan(uint &symbol, uint pos)
+{
+ uint result = -1;
+ using std::cout;
+ using std::endl;
+// cout << "pos = " << pos << ", symbol = " << symbol << endl;
+
+ if (pos == (uint)-1)
+ return (uint)-1;
+ if(right_child!=NULL)
+ result = right_child->rankLessThan(symbol, bitmap->rank1(pos)-1);
+ if(result == (uint)-1 && left_child!=NULL)
+ return left_child->rankLessThan(symbol, bitmap->rank0(pos)-1);
+ return result;
+}
+
+
uint wt_node_internal::select(uint symbol, uint pos, uint l, wt_coder * c) {
bool is_set = c->is_set(symbol, l);
+ uint ret = 0;
if(!is_set) {
if(left_child==NULL)
return (uint)(-1);
uint new_pos = left_child->select(symbol, pos, l+1,c);
if(new_pos+1==0) return (uint)(-1);
- return bitmap->select0(new_pos)+1;
+ ret = bitmap->select0(new_pos)+1;
} else {
if(right_child==NULL)
return (uint)(-1);
uint new_pos = right_child->select(symbol, pos, l+1,c);
if(new_pos+1==0) return (uint)(-1);
- return bitmap->select1(new_pos)+1;
+ ret = bitmap->select1(new_pos)+1;
}
+ if(ret==0) return (uint)-1;
+ return ret;
}
uint wt_node_internal::access(uint pos) {
}
}
+// Returns the value at given position and its rank
+uint wt_node_internal::access(uint pos, uint &rank)
+{
+ // p is the internal node we are pointing our finger at each step
+ wt_node_internal *p = this;
+
+ while(1)
+ {
+ bool is_set = p->bitmap->access(pos);
+// cout << "is_set = " << is_set << ", pos = " << pos << ", rank0 = " << bitmap->rank0(pos) << ", rank1 = " << bitmap->rank1(pos) << endl;
+ if(!is_set)
+ {
+ // recurse left
+ pos = p->bitmap->rank0(pos)-1;
+ wt_node_internal *tmp = dynamic_cast<wt_node_internal *>(p->left_child);
+ if (tmp == NULL)
+ {
+ // it's a leaf
+ rank = pos+1;
+ return p->left_child->access(0);
+ }
+ p = tmp; // new internal node
+ }
+ else
+ {
+ // recurse right
+ pos = p->bitmap->rank1(pos)-1;
+ wt_node_internal *tmp = dynamic_cast<wt_node_internal *>(p->right_child);
+ if (tmp == NULL)
+ {
+ // it's a leaf
+ rank = pos+1;
+ return p->right_child->access(0);
+ }
+ p = tmp; // new internal node
+ }
+ }
+}
+
+void wt_node_internal::access(vector<int> &result, uint i, uint j, uint min, uint max, uint l, uint pivot)
+{
+ uint symbol = pivot | (1 << l);
+// std::cout << "At l = " << l << ", [" << i << ", " << j << "], [" << min << ", " << max << "], symbol = " << symbol << std::endl;
+
+ if (j < i || max < min)
+ return;
+
+ if (min < symbol)
+ {
+ // Recurse left
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank0(i - 1);
+ uint newj = bitmap->rank0(j);
+
+ uint newmax = max < symbol - 1 ? max : symbol - 1;
+ if (left_child != NULL && newj > 0)
+ left_child->access(result, newi, newj-1, min, newmax, l-1, pivot);
+ }
+
+ if (max >= symbol)
+ {
+ // Recurse right
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank1(i - 1);
+ uint newj = bitmap->rank1(j);
+
+ uint newmin = min > symbol ? min : symbol;
+ if (right_child != NULL && newj > 0)
+ right_child->access(result, newi, newj-1, newmin, max, l-1, symbol);
+ }
+}
+
+void wt_node_internal::access(vector<int> &result, uint i, uint j)
+{
+// std::cout << "At l = " << l << ", [" << i << ", " << j << "], [" << min << ", " << max << "], symbol = " << symbol << std::endl;
+
+ if (j < i)
+ return;
+
+ {
+ // Recurse left
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank0(i - 1);
+ uint newj = bitmap->rank0(j);
+
+ if (left_child != NULL && newj > 0)
+ left_child->access(result, newi, newj-1);
+ }
+
+ {
+ // Recurse right
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank1(i - 1);
+ uint newj = bitmap->rank1(j);
+
+ if (right_child != NULL && newj > 0)
+ right_child->access(result, newi, newj-1);
+ }
+}
+
+// Count
+uint wt_node_internal::access(uint i, uint j, uint min, uint max, uint l, uint pivot)
+{
+ uint count = 0;
+ uint symbol = pivot | (1 << l);
+// std::cout << "At l = " << l << ", [" << i << ", " << j << "], [" << min << ", " << max << "], symbol = " << symbol << std::endl;
+
+ if (j < i || max < min)
+ return 0;
+
+ if (min < symbol)
+ {
+ // Recurse left
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank0(i - 1);
+ uint newj = bitmap->rank0(j);
+
+ uint newmax = max < symbol - 1 ? max : symbol - 1;
+ if (left_child != NULL && newj > 0)
+ count += left_child->access(newi, newj-1, min, newmax, l-1, pivot);
+ }
+
+ if (max >= symbol)
+ {
+ // Recurse right
+ uint newi = 0;
+ if (i > 0)
+ newi = bitmap->rank1(i - 1);
+ uint newj = bitmap->rank1(j);
+
+ uint newmin = min > symbol ? min : symbol;
+ if (right_child != NULL && newj > 0)
+ count += right_child->access(newi, newj-1, newmin, max, l-1, symbol);
+ }
+ return count;
+}
+
+
uint wt_node_internal::size() {
uint s = bitmap->size()+sizeof(wt_node_internal);
if(left_child!=NULL)